1/* 2 * Copyright (c) 2002 The FFmpeg Project 3 * 4 * This file is part of FFmpeg. 5 * 6 * FFmpeg is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * FFmpeg is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with FFmpeg; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 19 */ 20 21#include "avcodec.h" 22#include "idctdsp.h" 23#include "mpegvideo.h" 24#include "msmpeg4data.h" 25#include "wmv2.h" 26 27 28av_cold void ff_wmv2_common_init(MpegEncContext *s) 29{ 30 WMV2Context *const w = s->private_ctx; 31 32 ff_blockdsp_init(&s->bdsp, s->avctx); 33 ff_wmv2dsp_init(&w->wdsp); 34 s->idsp.perm_type = w->wdsp.idct_perm; 35 ff_init_scantable_permutation(s->idsp.idct_permutation, 36 w->wdsp.idct_perm); 37 ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, 38 ff_wmv1_scantable[1]); 39 ff_init_scantable(s->idsp.idct_permutation, &s->intra_h_scantable, 40 ff_wmv1_scantable[2]); 41 ff_init_scantable(s->idsp.idct_permutation, &s->intra_v_scantable, 42 ff_wmv1_scantable[3]); 43 ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, 44 ff_wmv1_scantable[0]); 45 s->idsp.idct_put = w->wdsp.idct_put; 46 s->idsp.idct_add = w->wdsp.idct_add; 47 s->idsp.idct = NULL; 48} 49 50void ff_mspel_motion(MpegEncContext *s, uint8_t *dest_y, 51 uint8_t *dest_cb, uint8_t *dest_cr, 52 uint8_t **ref_picture, op_pixels_func (*pix_op)[4], 53 int motion_x, int motion_y, int h) 54{ 55 WMV2Context *const w = s->private_ctx; 56 uint8_t *ptr; 57 int dxy, mx, my, src_x, src_y, v_edge_pos; 58 ptrdiff_t offset, linesize, uvlinesize; 59 int emu = 0; 60 61 dxy = ((motion_y & 1) << 1) | (motion_x & 1); 62 dxy = 2 * dxy + w->hshift; 63 src_x = s->mb_x * 16 + (motion_x >> 1); 64 src_y = s->mb_y * 16 + (motion_y >> 1); 65 66 /* WARNING: do no forget half pels */ 67 v_edge_pos = s->v_edge_pos; 68 src_x = av_clip(src_x, -16, s->width); 69 src_y = av_clip(src_y, -16, s->height); 70 71 if (src_x <= -16 || src_x >= s->width) 72 dxy &= ~3; 73 if (src_y <= -16 || src_y >= s->height) 74 dxy &= ~4; 75 76 linesize = s->linesize; 77 uvlinesize = s->uvlinesize; 78 ptr = ref_picture[0] + (src_y * linesize) + src_x; 79 80 if (src_x < 1 || src_y < 1 || src_x + 17 >= s->h_edge_pos || 81 src_y + h + 1 >= v_edge_pos) { 82 s->vdsp.emulated_edge_mc(s->sc.edge_emu_buffer, ptr - 1 - s->linesize, 83 s->linesize, s->linesize, 19, 19, 84 src_x - 1, src_y - 1, 85 s->h_edge_pos, s->v_edge_pos); 86 ptr = s->sc.edge_emu_buffer + 1 + s->linesize; 87 emu = 1; 88 } 89 90 w->wdsp.put_mspel_pixels_tab[dxy](dest_y, ptr, linesize); 91 w->wdsp.put_mspel_pixels_tab[dxy](dest_y + 8, ptr + 8, linesize); 92 w->wdsp.put_mspel_pixels_tab[dxy](dest_y + 8 * linesize, ptr + 8 * linesize, linesize); 93 w->wdsp.put_mspel_pixels_tab[dxy](dest_y + 8 + 8 * linesize, ptr + 8 + 8 * linesize, linesize); 94 95 if (s->avctx->flags & AV_CODEC_FLAG_GRAY) 96 return; 97 98 dxy = 0; 99 if ((motion_x & 3) != 0) 100 dxy |= 1; 101 if ((motion_y & 3) != 0) 102 dxy |= 2; 103 mx = motion_x >> 2; 104 my = motion_y >> 2; 105 106 src_x = s->mb_x * 8 + mx; 107 src_y = s->mb_y * 8 + my; 108 src_x = av_clip(src_x, -8, s->width >> 1); 109 if (src_x == (s->width >> 1)) 110 dxy &= ~1; 111 src_y = av_clip(src_y, -8, s->height >> 1); 112 if (src_y == (s->height >> 1)) 113 dxy &= ~2; 114 offset = (src_y * uvlinesize) + src_x; 115 ptr = ref_picture[1] + offset; 116 if (emu) { 117 s->vdsp.emulated_edge_mc(s->sc.edge_emu_buffer, ptr, 118 s->uvlinesize, s->uvlinesize, 119 9, 9, 120 src_x, src_y, 121 s->h_edge_pos >> 1, s->v_edge_pos >> 1); 122 ptr = s->sc.edge_emu_buffer; 123 } 124 pix_op[1][dxy](dest_cb, ptr, uvlinesize, h >> 1); 125 126 ptr = ref_picture[2] + offset; 127 if (emu) { 128 s->vdsp.emulated_edge_mc(s->sc.edge_emu_buffer, ptr, 129 s->uvlinesize, s->uvlinesize, 130 9, 9, 131 src_x, src_y, 132 s->h_edge_pos >> 1, s->v_edge_pos >> 1); 133 ptr = s->sc.edge_emu_buffer; 134 } 135 pix_op[1][dxy](dest_cr, ptr, uvlinesize, h >> 1); 136} 137