wmv2.c 6.41 KB
Newer Older
Michael Niedermayer's avatar
Michael Niedermayer committed
1
/*
2
 * Copyright (c) 2002 The FFmpeg Project
Michael Niedermayer's avatar
Michael Niedermayer committed
3
 *
4 5 6
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
Michael Niedermayer's avatar
Michael Niedermayer committed
7 8
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
9
 * version 2.1 of the License, or (at your option) any later version.
Michael Niedermayer's avatar
Michael Niedermayer committed
10
 *
11
 * FFmpeg is distributed in the hope that it will be useful,
Michael Niedermayer's avatar
Michael Niedermayer committed
12 13 14 15 16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * 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
Michael Niedermayer's avatar
Michael Niedermayer committed
19 20
 */

21 22 23
#include "avcodec.h"
#include "mpegvideo.h"
#include "msmpeg4data.h"
Michael Niedermayer's avatar
Michael Niedermayer committed
24
#include "simple_idct.h"
25
#include "wmv2.h"
Michael Niedermayer's avatar
Michael Niedermayer committed
26

27

28
av_cold void ff_wmv2_common_init(Wmv2Context * w){
Michael Niedermayer's avatar
Michael Niedermayer committed
29
    MpegEncContext * const s= &w->s;
30

31
    ff_wmv2dsp_init(&w->wdsp);
32
    s->dsp.idct_permutation_type = w->wdsp.idct_perm;
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
    ff_init_scantable_permutation(s->dsp.idct_permutation,
                                  w->wdsp.idct_perm);
    ff_init_scantable(s->dsp.idct_permutation, &w->abt_scantable[0],
                      ff_wmv2_scantableA);
    ff_init_scantable(s->dsp.idct_permutation, &w->abt_scantable[1],
                      ff_wmv2_scantableB);
    ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable,
                      ff_wmv1_scantable[1]);
    ff_init_scantable(s->dsp.idct_permutation, &s->intra_h_scantable,
                      ff_wmv1_scantable[2]);
    ff_init_scantable(s->dsp.idct_permutation, &s->intra_v_scantable,
                      ff_wmv1_scantable[3]);
    ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable,
                      ff_wmv1_scantable[0]);
    s->dsp.idct_put = w->wdsp.idct_put;
    s->dsp.idct_add = w->wdsp.idct_add;
    s->dsp.idct     = NULL;
Michael Niedermayer's avatar
Michael Niedermayer committed
50 51
}

Diego Biurrun's avatar
Diego Biurrun committed
52
static void wmv2_add_block(Wmv2Context *w, int16_t *block1, uint8_t *dst, int stride, int n){
Michael Niedermayer's avatar
Michael Niedermayer committed
53
    MpegEncContext * const s= &w->s;
54

55
  if (s->block_last_index[n] >= 0) {
Michael Niedermayer's avatar
Michael Niedermayer committed
56 57
    switch(w->abt_type_table[n]){
    case 0:
58
        w->wdsp.idct_add(dst, stride, block1);
Michael Niedermayer's avatar
Michael Niedermayer committed
59 60
        break;
    case 1:
61 62
        ff_simple_idct84_add(dst           , stride, block1);
        ff_simple_idct84_add(dst + 4*stride, stride, w->abt_block2[n]);
Loren Merritt's avatar
Loren Merritt committed
63
        s->dsp.clear_block(w->abt_block2[n]);
Michael Niedermayer's avatar
Michael Niedermayer committed
64 65
        break;
    case 2:
66 67
        ff_simple_idct48_add(dst           , stride, block1);
        ff_simple_idct48_add(dst + 4       , stride, w->abt_block2[n]);
Loren Merritt's avatar
Loren Merritt committed
68
        s->dsp.clear_block(w->abt_block2[n]);
Michael Niedermayer's avatar
Michael Niedermayer committed
69 70
        break;
    default:
71
        av_log(s->avctx, AV_LOG_ERROR, "internal error in WMV2 abt\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
72
    }
73
  }
Michael Niedermayer's avatar
Michael Niedermayer committed
74 75
}

Diego Biurrun's avatar
Diego Biurrun committed
76
void ff_wmv2_add_mb(MpegEncContext *s, int16_t block1[6][64], uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr){
Michael Niedermayer's avatar
Michael Niedermayer committed
77 78 79 80 81 82
    Wmv2Context * const w= (Wmv2Context*)s;

    wmv2_add_block(w, block1[0], dest_y                    , s->linesize, 0);
    wmv2_add_block(w, block1[1], dest_y + 8                , s->linesize, 1);
    wmv2_add_block(w, block1[2], dest_y +     8*s->linesize, s->linesize, 2);
    wmv2_add_block(w, block1[3], dest_y + 8 + 8*s->linesize, s->linesize, 3);
83

Michael Niedermayer's avatar
Michael Niedermayer committed
84
    if(s->flags&CODEC_FLAG_GRAY) return;
85

Michael Niedermayer's avatar
Michael Niedermayer committed
86 87 88 89 90
    wmv2_add_block(w, block1[4], dest_cb                   , s->uvlinesize, 4);
    wmv2_add_block(w, block1[5], dest_cr                   , s->uvlinesize, 5);
}

void ff_mspel_motion(MpegEncContext *s,
91 92
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
                               uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
Michael Niedermayer's avatar
Michael Niedermayer committed
93 94 95
                               int motion_x, int motion_y, int h)
{
    Wmv2Context * const w= (Wmv2Context*)s;
96
    uint8_t *ptr;
Michael Niedermayer's avatar
Michael Niedermayer committed
97 98
    int dxy, offset, mx, my, src_x, src_y, v_edge_pos, linesize, uvlinesize;
    int emu=0;
99

Michael Niedermayer's avatar
Michael Niedermayer committed
100 101 102 103
    dxy = ((motion_y & 1) << 1) | (motion_x & 1);
    dxy = 2*dxy + w->hshift;
    src_x = s->mb_x * 16 + (motion_x >> 1);
    src_y = s->mb_y * 16 + (motion_y >> 1);
104

Michael Niedermayer's avatar
Michael Niedermayer committed
105 106
    /* WARNING: do no forget half pels */
    v_edge_pos = s->v_edge_pos;
107 108
    src_x = av_clip(src_x, -16, s->width);
    src_y = av_clip(src_y, -16, s->height);
109 110 111 112 113 114

    if(src_x<=-16 || src_x >= s->width)
        dxy &= ~3;
    if(src_y<=-16 || src_y >= s->height)
        dxy &= ~4;

Michael Niedermayer's avatar
Michael Niedermayer committed
115 116 117 118 119 120 121
    linesize   = s->linesize;
    uvlinesize = s->uvlinesize;
    ptr = ref_picture[0] + (src_y * linesize) + src_x;

    if(s->flags&CODEC_FLAG_EMU_EDGE){
        if(src_x<1 || src_y<1 || src_x + 17  >= s->h_edge_pos
                              || src_y + h+1 >= v_edge_pos){
122
            s->vdsp.emulated_edge_mc(s->edge_emu_buffer, ptr - 1 - s->linesize, s->linesize, 19, 19,
Michael Niedermayer's avatar
Michael Niedermayer committed
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
                             src_x-1, src_y-1, s->h_edge_pos, s->v_edge_pos);
            ptr= s->edge_emu_buffer + 1 + s->linesize;
            emu=1;
        }
    }

    s->dsp.put_mspel_pixels_tab[dxy](dest_y             , ptr             , linesize);
    s->dsp.put_mspel_pixels_tab[dxy](dest_y+8           , ptr+8           , linesize);
    s->dsp.put_mspel_pixels_tab[dxy](dest_y  +8*linesize, ptr  +8*linesize, linesize);
    s->dsp.put_mspel_pixels_tab[dxy](dest_y+8+8*linesize, ptr+8+8*linesize, linesize);

    if(s->flags&CODEC_FLAG_GRAY) return;

    if (s->out_format == FMT_H263) {
        dxy = 0;
        if ((motion_x & 3) != 0)
            dxy |= 1;
        if ((motion_y & 3) != 0)
            dxy |= 2;
        mx = motion_x >> 2;
        my = motion_y >> 2;
    } else {
        mx = motion_x / 2;
        my = motion_y / 2;
        dxy = ((my & 1) << 1) | (mx & 1);
        mx >>= 1;
        my >>= 1;
    }
151

Michael Niedermayer's avatar
Michael Niedermayer committed
152 153
    src_x = s->mb_x * 8 + mx;
    src_y = s->mb_y * 8 + my;
154
    src_x = av_clip(src_x, -8, s->width >> 1);
Michael Niedermayer's avatar
Michael Niedermayer committed
155 156
    if (src_x == (s->width >> 1))
        dxy &= ~1;
157
    src_y = av_clip(src_y, -8, s->height >> 1);
Michael Niedermayer's avatar
Michael Niedermayer committed
158 159 160 161 162
    if (src_y == (s->height >> 1))
        dxy &= ~2;
    offset = (src_y * uvlinesize) + src_x;
    ptr = ref_picture[1] + offset;
    if(emu){
163
        s->vdsp.emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9,
Michael Niedermayer's avatar
Michael Niedermayer committed
164 165 166 167 168 169 170
                         src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
        ptr= s->edge_emu_buffer;
    }
    pix_op[1][dxy](dest_cb, ptr, uvlinesize, h >> 1);

    ptr = ref_picture[2] + offset;
    if(emu){
171
        s->vdsp.emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9,
Michael Niedermayer's avatar
Michael Niedermayer committed
172 173 174 175 176
                         src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
        ptr= s->edge_emu_buffer;
    }
    pix_op[1][dxy](dest_cr, ptr, uvlinesize, h >> 1);
}