rv10.c 24.9 KB
Newer Older
F
Fabrice Bellard 已提交
1 2
/*
 * RV10 codec
F
Fabrice Bellard 已提交
3
 * Copyright (c) 2000,2001 Fabrice Bellard.
4
 * Copyright (c) 2002-2004 Michael Niedermayer
F
Fabrice Bellard 已提交
5
 *
F
Fabrice Bellard 已提交
6 7 8 9
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
F
Fabrice Bellard 已提交
10
 *
F
Fabrice Bellard 已提交
11
 * This library is distributed in the hope that it will be useful,
F
Fabrice Bellard 已提交
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
F
Fabrice Bellard 已提交
13 14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
F
Fabrice Bellard 已提交
15
 *
F
Fabrice Bellard 已提交
16 17
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
18
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
F
Fabrice Bellard 已提交
19
 */
M
Michael Niedermayer 已提交
20 21 22 23 24

/**
 * @file rv10.c
 * RV10 codec.
 */
25

F
Fabrice Bellard 已提交
26
#include "avcodec.h"
27
#include "dsputil.h"
F
Fabrice Bellard 已提交
28 29 30 31
#include "mpegvideo.h"

//#define DEBUG

32
#define DC_VLC_BITS 14 //FIXME find a better solution
M
Michael Niedermayer 已提交
33

34
static const uint16_t rv_lum_code[256] =
F
Fabrice Bellard 已提交
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
{
 0x3e7f, 0x0f00, 0x0f01, 0x0f02, 0x0f03, 0x0f04, 0x0f05, 0x0f06,
 0x0f07, 0x0f08, 0x0f09, 0x0f0a, 0x0f0b, 0x0f0c, 0x0f0d, 0x0f0e,
 0x0f0f, 0x0f10, 0x0f11, 0x0f12, 0x0f13, 0x0f14, 0x0f15, 0x0f16,
 0x0f17, 0x0f18, 0x0f19, 0x0f1a, 0x0f1b, 0x0f1c, 0x0f1d, 0x0f1e,
 0x0f1f, 0x0f20, 0x0f21, 0x0f22, 0x0f23, 0x0f24, 0x0f25, 0x0f26,
 0x0f27, 0x0f28, 0x0f29, 0x0f2a, 0x0f2b, 0x0f2c, 0x0f2d, 0x0f2e,
 0x0f2f, 0x0f30, 0x0f31, 0x0f32, 0x0f33, 0x0f34, 0x0f35, 0x0f36,
 0x0f37, 0x0f38, 0x0f39, 0x0f3a, 0x0f3b, 0x0f3c, 0x0f3d, 0x0f3e,
 0x0f3f, 0x0380, 0x0381, 0x0382, 0x0383, 0x0384, 0x0385, 0x0386,
 0x0387, 0x0388, 0x0389, 0x038a, 0x038b, 0x038c, 0x038d, 0x038e,
 0x038f, 0x0390, 0x0391, 0x0392, 0x0393, 0x0394, 0x0395, 0x0396,
 0x0397, 0x0398, 0x0399, 0x039a, 0x039b, 0x039c, 0x039d, 0x039e,
 0x039f, 0x00c0, 0x00c1, 0x00c2, 0x00c3, 0x00c4, 0x00c5, 0x00c6,
 0x00c7, 0x00c8, 0x00c9, 0x00ca, 0x00cb, 0x00cc, 0x00cd, 0x00ce,
 0x00cf, 0x0050, 0x0051, 0x0052, 0x0053, 0x0054, 0x0055, 0x0056,
 0x0057, 0x0020, 0x0021, 0x0022, 0x0023, 0x000c, 0x000d, 0x0004,
 0x0000, 0x0005, 0x000e, 0x000f, 0x0024, 0x0025, 0x0026, 0x0027,
 0x0058, 0x0059, 0x005a, 0x005b, 0x005c, 0x005d, 0x005e, 0x005f,
 0x00d0, 0x00d1, 0x00d2, 0x00d3, 0x00d4, 0x00d5, 0x00d6, 0x00d7,
 0x00d8, 0x00d9, 0x00da, 0x00db, 0x00dc, 0x00dd, 0x00de, 0x00df,
 0x03a0, 0x03a1, 0x03a2, 0x03a3, 0x03a4, 0x03a5, 0x03a6, 0x03a7,
 0x03a8, 0x03a9, 0x03aa, 0x03ab, 0x03ac, 0x03ad, 0x03ae, 0x03af,
 0x03b0, 0x03b1, 0x03b2, 0x03b3, 0x03b4, 0x03b5, 0x03b6, 0x03b7,
 0x03b8, 0x03b9, 0x03ba, 0x03bb, 0x03bc, 0x03bd, 0x03be, 0x03bf,
 0x0f40, 0x0f41, 0x0f42, 0x0f43, 0x0f44, 0x0f45, 0x0f46, 0x0f47,
 0x0f48, 0x0f49, 0x0f4a, 0x0f4b, 0x0f4c, 0x0f4d, 0x0f4e, 0x0f4f,
 0x0f50, 0x0f51, 0x0f52, 0x0f53, 0x0f54, 0x0f55, 0x0f56, 0x0f57,
 0x0f58, 0x0f59, 0x0f5a, 0x0f5b, 0x0f5c, 0x0f5d, 0x0f5e, 0x0f5f,
 0x0f60, 0x0f61, 0x0f62, 0x0f63, 0x0f64, 0x0f65, 0x0f66, 0x0f67,
 0x0f68, 0x0f69, 0x0f6a, 0x0f6b, 0x0f6c, 0x0f6d, 0x0f6e, 0x0f6f,
 0x0f70, 0x0f71, 0x0f72, 0x0f73, 0x0f74, 0x0f75, 0x0f76, 0x0f77,
 0x0f78, 0x0f79, 0x0f7a, 0x0f7b, 0x0f7c, 0x0f7d, 0x0f7e, 0x0f7f,
};

70
static const uint8_t rv_lum_bits[256] =
F
Fabrice Bellard 已提交
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
{
 14, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 10,  8,  8,  8,  8,  8,  8,  8,
  8,  8,  8,  8,  8,  8,  8,  8,
  8,  7,  7,  7,  7,  7,  7,  7,
  7,  6,  6,  6,  6,  5,  5,  4,
  2,  4,  5,  5,  6,  6,  6,  6,
  7,  7,  7,  7,  7,  7,  7,  7,
  8,  8,  8,  8,  8,  8,  8,  8,
  8,  8,  8,  8,  8,  8,  8,  8,
 10, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
};

106
static const uint16_t rv_chrom_code[256] =
F
Fabrice Bellard 已提交
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
{
 0xfe7f, 0x3f00, 0x3f01, 0x3f02, 0x3f03, 0x3f04, 0x3f05, 0x3f06,
 0x3f07, 0x3f08, 0x3f09, 0x3f0a, 0x3f0b, 0x3f0c, 0x3f0d, 0x3f0e,
 0x3f0f, 0x3f10, 0x3f11, 0x3f12, 0x3f13, 0x3f14, 0x3f15, 0x3f16,
 0x3f17, 0x3f18, 0x3f19, 0x3f1a, 0x3f1b, 0x3f1c, 0x3f1d, 0x3f1e,
 0x3f1f, 0x3f20, 0x3f21, 0x3f22, 0x3f23, 0x3f24, 0x3f25, 0x3f26,
 0x3f27, 0x3f28, 0x3f29, 0x3f2a, 0x3f2b, 0x3f2c, 0x3f2d, 0x3f2e,
 0x3f2f, 0x3f30, 0x3f31, 0x3f32, 0x3f33, 0x3f34, 0x3f35, 0x3f36,
 0x3f37, 0x3f38, 0x3f39, 0x3f3a, 0x3f3b, 0x3f3c, 0x3f3d, 0x3f3e,
 0x3f3f, 0x0f80, 0x0f81, 0x0f82, 0x0f83, 0x0f84, 0x0f85, 0x0f86,
 0x0f87, 0x0f88, 0x0f89, 0x0f8a, 0x0f8b, 0x0f8c, 0x0f8d, 0x0f8e,
 0x0f8f, 0x0f90, 0x0f91, 0x0f92, 0x0f93, 0x0f94, 0x0f95, 0x0f96,
 0x0f97, 0x0f98, 0x0f99, 0x0f9a, 0x0f9b, 0x0f9c, 0x0f9d, 0x0f9e,
 0x0f9f, 0x03c0, 0x03c1, 0x03c2, 0x03c3, 0x03c4, 0x03c5, 0x03c6,
 0x03c7, 0x03c8, 0x03c9, 0x03ca, 0x03cb, 0x03cc, 0x03cd, 0x03ce,
 0x03cf, 0x00e0, 0x00e1, 0x00e2, 0x00e3, 0x00e4, 0x00e5, 0x00e6,
 0x00e7, 0x0030, 0x0031, 0x0032, 0x0033, 0x0008, 0x0009, 0x0002,
 0x0000, 0x0003, 0x000a, 0x000b, 0x0034, 0x0035, 0x0036, 0x0037,
 0x00e8, 0x00e9, 0x00ea, 0x00eb, 0x00ec, 0x00ed, 0x00ee, 0x00ef,
 0x03d0, 0x03d1, 0x03d2, 0x03d3, 0x03d4, 0x03d5, 0x03d6, 0x03d7,
 0x03d8, 0x03d9, 0x03da, 0x03db, 0x03dc, 0x03dd, 0x03de, 0x03df,
 0x0fa0, 0x0fa1, 0x0fa2, 0x0fa3, 0x0fa4, 0x0fa5, 0x0fa6, 0x0fa7,
 0x0fa8, 0x0fa9, 0x0faa, 0x0fab, 0x0fac, 0x0fad, 0x0fae, 0x0faf,
 0x0fb0, 0x0fb1, 0x0fb2, 0x0fb3, 0x0fb4, 0x0fb5, 0x0fb6, 0x0fb7,
 0x0fb8, 0x0fb9, 0x0fba, 0x0fbb, 0x0fbc, 0x0fbd, 0x0fbe, 0x0fbf,
 0x3f40, 0x3f41, 0x3f42, 0x3f43, 0x3f44, 0x3f45, 0x3f46, 0x3f47,
 0x3f48, 0x3f49, 0x3f4a, 0x3f4b, 0x3f4c, 0x3f4d, 0x3f4e, 0x3f4f,
 0x3f50, 0x3f51, 0x3f52, 0x3f53, 0x3f54, 0x3f55, 0x3f56, 0x3f57,
 0x3f58, 0x3f59, 0x3f5a, 0x3f5b, 0x3f5c, 0x3f5d, 0x3f5e, 0x3f5f,
 0x3f60, 0x3f61, 0x3f62, 0x3f63, 0x3f64, 0x3f65, 0x3f66, 0x3f67,
 0x3f68, 0x3f69, 0x3f6a, 0x3f6b, 0x3f6c, 0x3f6d, 0x3f6e, 0x3f6f,
 0x3f70, 0x3f71, 0x3f72, 0x3f73, 0x3f74, 0x3f75, 0x3f76, 0x3f77,
 0x3f78, 0x3f79, 0x3f7a, 0x3f7b, 0x3f7c, 0x3f7d, 0x3f7e, 0x3f7f,
};

142
static const uint8_t rv_chrom_bits[256] =
F
Fabrice Bellard 已提交
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
{
 16, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 10,  8,  8,  8,  8,  8,  8,  8,
  8,  6,  6,  6,  6,  4,  4,  3,
  2,  3,  4,  4,  6,  6,  6,  6,
  8,  8,  8,  8,  8,  8,  8,  8,
 10, 10, 10, 10, 10, 10, 10, 10,
 10, 10, 10, 10, 10, 10, 10, 10,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 12, 12, 12, 12, 12, 12, 12, 12,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
 14, 14, 14, 14, 14, 14, 14, 14,
};

static VLC rv_dc_lum, rv_dc_chrom;

int rv_decode_dc(MpegEncContext *s, int n)
{
    int code;

    if (n < 4) {
M
Michael Niedermayer 已提交
185
        code = get_vlc2(&s->gb, rv_dc_lum.table, DC_VLC_BITS, 2);
F
Fabrice Bellard 已提交
186 187 188 189 190 191
        if (code < 0) {
            /* XXX: I don't understand why they use LONGER codes than
               necessary. The following code would be completely useless
               if they had thought about it !!! */
            code = get_bits(&s->gb, 7);
            if (code == 0x7c) {
192
                code = (int8_t)(get_bits(&s->gb, 7) + 1);
F
Fabrice Bellard 已提交
193 194 195 196
            } else if (code == 0x7d) {
                code = -128 + get_bits(&s->gb, 7);
            } else if (code == 0x7e) {
                if (get_bits(&s->gb, 1) == 0)
197
                    code = (int8_t)(get_bits(&s->gb, 8) + 1);
F
Fabrice Bellard 已提交
198
                else
199
                    code = (int8_t)(get_bits(&s->gb, 8));
F
Fabrice Bellard 已提交
200 201 202 203 204 205 206 207
            } else if (code == 0x7f) {
                get_bits(&s->gb, 11);
                code = 1;
            }
        } else {
            code -= 128;
        }
    } else {
M
Michael Niedermayer 已提交
208
        code = get_vlc2(&s->gb, rv_dc_chrom.table, DC_VLC_BITS, 2);
F
Fabrice Bellard 已提交
209 210 211 212
        /* same remark */
        if (code < 0) {
            code = get_bits(&s->gb, 9);
            if (code == 0x1fc) {
213
                code = (int8_t)(get_bits(&s->gb, 7) + 1);
F
Fabrice Bellard 已提交
214 215 216 217 218 219
            } else if (code == 0x1fd) {
                code = -128 + get_bits(&s->gb, 7);
            } else if (code == 0x1fe) {
                get_bits(&s->gb, 9);
                code = 1;
            } else {
220
                av_log(s->avctx, AV_LOG_ERROR, "chroma dc error\n");
F
Fabrice Bellard 已提交
221 222 223 224 225 226 227 228 229
                return 0xffff;
            }
        } else {
            code -= 128;
        }
    }
    return -code;
}

230 231
#ifdef CONFIG_ENCODERS

F
Fabrice Bellard 已提交
232 233 234
/* write RV 1.0 compatible frame header */
void rv10_encode_picture_header(MpegEncContext *s, int picture_number)
{
235
    int full_frame= 0;
236

F
Fabrice Bellard 已提交
237
    align_put_bits(&s->pb);
238

239
    put_bits(&s->pb, 1, 1);     /* marker */
F
Fabrice Bellard 已提交
240 241 242

    put_bits(&s->pb, 1, (s->pict_type == P_TYPE));

243
    put_bits(&s->pb, 1, 0);     /* not PB frame */
F
Fabrice Bellard 已提交
244 245 246 247

    put_bits(&s->pb, 5, s->qscale);

    if (s->pict_type == I_TYPE) {
248
        /* specific MPEG like DC coding not used */
F
Fabrice Bellard 已提交
249 250 251
    }
    /* if multiple packets per frame are sent, the position at which
       to display the macro blocks is coded here */
252
    if(!full_frame){
253 254
        put_bits(&s->pb, 6, 0); /* mb_x */
        put_bits(&s->pb, 6, 0); /* mb_y */
255 256
        put_bits(&s->pb, 12, s->mb_width * s->mb_height);
    }
F
Fabrice Bellard 已提交
257

258
    put_bits(&s->pb, 3, 0);     /* ignored */
F
Fabrice Bellard 已提交
259 260
}

M
Michael Niedermayer 已提交
261 262
void rv20_encode_picture_header(MpegEncContext *s, int picture_number){
    put_bits(&s->pb, 2, s->pict_type); //I 0 vs. 1 ?
263
    put_bits(&s->pb, 1, 0);     /* unknown bit */
M
Michael Niedermayer 已提交
264
    put_bits(&s->pb, 5, s->qscale);
265

M
Michael Niedermayer 已提交
266 267 268
    put_bits(&s->pb, 8, picture_number&0xFF); //FIXME wrong, but correct is not known
    s->mb_x= s->mb_y= 0;
    ff_h263_encode_mba(s);
269

M
Michael Niedermayer 已提交
270
    put_bits(&s->pb, 1, s->no_rounding);
271

M
Michael Niedermayer 已提交
272 273 274 275 276 277 278 279 280 281
    assert(s->f_code == 1);
    assert(s->unrestricted_mv == 1);
//    assert(s->h263_aic== (s->pict_type == I_TYPE));
    assert(s->alt_inter_vlc == 0);
    assert(s->umvplus == 0);
    assert(s->modified_quant==1);
    assert(s->loop_filter==1);

    s->h263_aic= s->pict_type == I_TYPE;
    if(s->h263_aic){
282
        s->y_dc_scale_table=
M
Michael Niedermayer 已提交
283 284 285 286 287 288 289
        s->c_dc_scale_table= ff_aic_dc_scale_table;
    }else{
        s->y_dc_scale_table=
        s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
    }
}

290
#if 0 /* unused, remove? */
F
Fabrice Bellard 已提交
291 292 293 294 295 296 297 298 299 300 301 302
static int get_num(GetBitContext *gb)
{
    int n, n1;

    n = get_bits(gb, 16);
    if (n >= 0x4000) {
        return n - 0x4000;
    } else {
        n1 = get_bits(gb, 16);
        return (n << 16) | n1;
    }
}
303
#endif
F
Fabrice Bellard 已提交
304

305 306
#endif //CONFIG_ENCODERS

F
Fabrice Bellard 已提交
307 308 309
/* read RV 1.0 compatible frame header */
static int rv10_decode_picture_header(MpegEncContext *s)
{
310
    int mb_count, pb_frame, marker, unk, mb_xy;
311

312
//printf("ff:%d\n", full_frame);
F
Fabrice Bellard 已提交
313 314 315 316 317 318
    marker = get_bits(&s->gb, 1);

    if (get_bits(&s->gb, 1))
        s->pict_type = P_TYPE;
    else
        s->pict_type = I_TYPE;
319
//printf("h:%X ver:%d\n",h,s->rv10_version);
320
    if(!marker) av_log(s->avctx, AV_LOG_ERROR, "marker missing\n");
F
Fabrice Bellard 已提交
321 322 323
    pb_frame = get_bits(&s->gb, 1);

#ifdef DEBUG
S
Steve L'Homme 已提交
324
    av_log(s->avctx, AV_LOG_DEBUG, "pict_type=%d pb_frame=%d\n", s->pict_type, pb_frame);
F
Fabrice Bellard 已提交
325
#endif
326

327
    if (pb_frame){
328
        av_log(s->avctx, AV_LOG_ERROR, "pb frame not supported\n");
F
Fabrice Bellard 已提交
329
        return -1;
330
    }
F
Fabrice Bellard 已提交
331 332

    s->qscale = get_bits(&s->gb, 5);
M
Michael Niedermayer 已提交
333
    if(s->qscale==0){
334
        av_log(s->avctx, AV_LOG_ERROR, "error, qscale:0\n");
M
Michael Niedermayer 已提交
335 336
        return -1;
    }
F
Fabrice Bellard 已提交
337 338 339 340 341 342 343 344

    if (s->pict_type == I_TYPE) {
        if (s->rv10_version == 3) {
            /* specific MPEG like DC coding not used */
            s->last_dc[0] = get_bits(&s->gb, 8);
            s->last_dc[1] = get_bits(&s->gb, 8);
            s->last_dc[2] = get_bits(&s->gb, 8);
#ifdef DEBUG
S
Steve L'Homme 已提交
345
            av_log(s->avctx, AV_LOG_DEBUG, "DC:%d %d %d\n",
F
Fabrice Bellard 已提交
346 347 348 349 350 351 352 353
                   s->last_dc[0],
                   s->last_dc[1],
                   s->last_dc[2]);
#endif
        }
    }
    /* if multiple packets per frame are sent, the position at which
       to display the macro blocks is coded here */
354 355 356

    mb_xy= s->mb_x + s->mb_y*s->mb_width;
    if(show_bits(&s->gb, 12)==0 || (mb_xy && mb_xy < s->mb_num)){
357 358
        s->mb_x = get_bits(&s->gb, 6); /* mb_x */
        s->mb_y = get_bits(&s->gb, 6); /* mb_y */
F
Fabrice Bellard 已提交
359 360 361 362 363 364
        mb_count = get_bits(&s->gb, 12);
    } else {
        s->mb_x = 0;
        s->mb_y = 0;
        mb_count = s->mb_width * s->mb_height;
    }
365
    unk= get_bits(&s->gb, 3);   /* ignored */
366
//printf("%d\n", unk);
F
Fabrice Bellard 已提交
367 368
    s->f_code = 1;
    s->unrestricted_mv = 1;
369

F
Fabrice Bellard 已提交
370 371 372
    return mb_count;
}

M
Michael Niedermayer 已提交
373 374
static int rv20_decode_picture_header(MpegEncContext *s)
{
375
    int seq, mb_pos, i;
376

M
Michael Niedermayer 已提交
377 378 379 380 381 382 383 384
#if 0
    GetBitContext gb= s->gb;
    for(i=0; i<64; i++){
        av_log(s->avctx, AV_LOG_DEBUG, "%d", get_bits1(&gb));
        if(i%4==3) av_log(s->avctx, AV_LOG_DEBUG, " ");
    }
    av_log(s->avctx, AV_LOG_DEBUG, "\n");
#endif
385
#if 0
386
    av_log(s->avctx, AV_LOG_DEBUG, "%3dx%03d/%02Xx%02X ", s->width, s->height, s->width/4, s->height/4);
387
    for(i=0; i<s->avctx->extradata_size; i++){
388
        av_log(s->avctx, AV_LOG_DEBUG, "%02X ", ((uint8_t*)s->avctx->extradata)[i]);
389 390 391 392
        if(i%4==3) av_log(s->avctx, AV_LOG_DEBUG, " ");
    }
    av_log(s->avctx, AV_LOG_DEBUG, "\n");
#endif
393

394 395 396 397
    if(s->avctx->sub_id == 0x30202002 || s->avctx->sub_id == 0x30203002){
        if (get_bits(&s->gb, 3)){
            av_log(s->avctx, AV_LOG_ERROR, "unknown triplet set\n");
            return -1;
398 399
        }
    }
400 401 402 403

    i= get_bits(&s->gb, 2);
    switch(i){
    case 0: s->pict_type= I_TYPE; break;
M
Michael Niedermayer 已提交
404
    case 1: s->pict_type= I_TYPE; break; //hmm ...
405 406
    case 2: s->pict_type= P_TYPE; break;
    case 3: s->pict_type= B_TYPE; break;
407
    default:
408 409 410
        av_log(s->avctx, AV_LOG_ERROR, "unknown frame type\n");
        return -1;
    }
411

412 413 414 415
    if(s->last_picture_ptr==NULL && s->pict_type==B_TYPE){
        av_log(s->avctx, AV_LOG_ERROR, "early B pix\n");
        return -1;
    }
416

417 418
    if (get_bits(&s->gb, 1)){
        av_log(s->avctx, AV_LOG_ERROR, "unknown bit set\n");
M
Michael Niedermayer 已提交
419 420 421 422 423 424 425 426
        return -1;
    }

    s->qscale = get_bits(&s->gb, 5);
    if(s->qscale==0){
        av_log(s->avctx, AV_LOG_ERROR, "error, qscale:0\n");
        return -1;
    }
427 428 429 430 431 432
    if(s->avctx->sub_id == 0x30203002){
        if (get_bits(&s->gb, 1)){
            av_log(s->avctx, AV_LOG_ERROR, "unknown bit2 set\n");
            return -1;
        }
    }
433

434
    if(s->avctx->has_b_frames){
435
        int f, new_w, new_h;
436
        int v= s->avctx->extradata_size >= 4 ? ((uint8_t*)s->avctx->extradata)[1] : 0;
M
Michael Niedermayer 已提交
437

438
        if (get_bits(&s->gb, 1)){
M
Michael Niedermayer 已提交
439
            av_log(s->avctx, AV_LOG_ERROR, "unknown bit3 set\n");
M
Michael Niedermayer 已提交
440
//            return -1;
441
        }
442
        seq= get_bits(&s->gb, 13)<<2;
443

444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
        f= get_bits(&s->gb, av_log2(v)+1);

        if(f){
            new_w= 4*((uint8_t*)s->avctx->extradata)[6+2*f];
            new_h= 4*((uint8_t*)s->avctx->extradata)[7+2*f];
        }else{
            new_w= s->width; //FIXME wrong we of course must save the original in the context
            new_h= s->height;
        }
        if(new_w != s->width || new_h != s->height){
            av_log(s->avctx, AV_LOG_DEBUG, "attempting to change resolution to %dx%d\n", new_w, new_h);
            MPV_common_end(s);
            s->width  = s->avctx->width = new_w;
            s->height = s->avctx->height= new_h;
            if (MPV_common_init(s) < 0)
                return -1;
        }
M
Michael Niedermayer 已提交
461

M
Michael Niedermayer 已提交
462
        if(s->avctx->debug & FF_DEBUG_PICT_INFO){
463
            av_log(s->avctx, AV_LOG_DEBUG, "F %d/%d\n", f, v);
M
Michael Niedermayer 已提交
464
        }
M
Michael Niedermayer 已提交
465 466 467
    }else{
        seq= get_bits(&s->gb, 8)*128;
    }
468

469
//     if(s->avctx->sub_id <= 0x20201002){ //0x20201002 definitely needs this
M
Michael Niedermayer 已提交
470 471
    mb_pos= ff_h263_decode_mba(s);
/*    }else{
472 473 474
        mb_pos= get_bits(&s->gb, av_log2(s->mb_num-1)+1);
        s->mb_x= mb_pos % s->mb_width;
        s->mb_y= mb_pos / s->mb_width;
M
Michael Niedermayer 已提交
475
    }*/
M
Michael Niedermayer 已提交
476
//av_log(s->avctx, AV_LOG_DEBUG, "%d\n", seq);
477 478 479
    seq |= s->time &~0x7FFF;
    if(seq - s->time >  0x4000) seq -= 0x8000;
    if(seq - s->time < -0x4000) seq += 0x8000;
480
    if(seq != s->time){
481 482 483 484 485 486 487 488
        if(s->pict_type!=B_TYPE){
            s->time= seq;
            s->pp_time= s->time - s->last_non_b_time;
            s->last_non_b_time= s->time;
        }else{
            s->time= seq;
            s->pb_time= s->pp_time - (s->last_non_b_time - s->time);
            if(s->pp_time <=s->pb_time || s->pp_time <= s->pp_time - s->pb_time || s->pp_time<=0){
489 490
                av_log(s->avctx, AV_LOG_DEBUG, "messed up order, possible from seeking? skipping current b frame\n");
                return FRAME_SKIPPED;
491
            }
492
            ff_mpeg4_init_direct_mv(s);
493 494 495
        }
    }
//    printf("%d %d %d %d %d\n", seq, (int)s->time, (int)s->last_non_b_time, s->pp_time, s->pb_time);
M
Michael Niedermayer 已提交
496 497 498 499
/*for(i=0; i<32; i++){
    av_log(s->avctx, AV_LOG_DEBUG, "%d", get_bits1(&s->gb));
}
av_log(s->avctx, AV_LOG_DEBUG, "\n");*/
M
Michael Niedermayer 已提交
500
    s->no_rounding= get_bits1(&s->gb);
501

M
Michael Niedermayer 已提交
502 503 504 505 506 507
    s->f_code = 1;
    s->unrestricted_mv = 1;
    s->h263_aic= s->pict_type == I_TYPE;
//    s->alt_inter_vlc=1;
//    s->obmc=1;
//    s->umvplus=1;
M
Michael Niedermayer 已提交
508 509
    s->modified_quant=1;
    s->loop_filter=1;
510

M
Michael Niedermayer 已提交
511
    if(s->avctx->debug & FF_DEBUG_PICT_INFO){
512
            av_log(s->avctx, AV_LOG_INFO, "num:%5d x:%2d y:%2d type:%d qscale:%2d rnd:%d\n",
M
Michael Niedermayer 已提交
513 514 515
                   seq, s->mb_x, s->mb_y, s->pict_type, s->qscale, s->no_rounding);
    }

516
    assert(s->pict_type != B_TYPE || !s->low_delay);
517

M
Michael Niedermayer 已提交
518 519 520
    return s->mb_width*s->mb_height - mb_pos;
}

F
Fabrice Bellard 已提交
521 522 523
static int rv10_decode_init(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;
524
    static int done=0;
F
Fabrice Bellard 已提交
525

M
Michael Niedermayer 已提交
526
    MPV_decode_defaults(s);
527

M
Michael Niedermayer 已提交
528
    s->avctx= avctx;
F
Fabrice Bellard 已提交
529
    s->out_format = FMT_H263;
M
Michael Niedermayer 已提交
530
    s->codec_id= avctx->codec_id;
F
Fabrice Bellard 已提交
531 532 533 534

    s->width = avctx->width;
    s->height = avctx->height;

M
Michael Niedermayer 已提交
535 536
    s->h263_long_vectors= !!(*(uint32_t*)avctx->extradata & 0x1000000);

537 538 539
    switch(avctx->sub_id){
    case 0x10000000:
        s->rv10_version= 0;
540
        s->low_delay=1;
541
        break;
M
Michael Niedermayer 已提交
542 543 544 545 546
    case 0x10002000:
        s->rv10_version= 3;
        s->low_delay=1;
        s->obmc=1;
        break;
547 548
    case 0x10003000:
        s->rv10_version= 3;
549
        s->low_delay=1;
550 551 552
        break;
    case 0x10003001:
        s->rv10_version= 3;
553
        s->low_delay=1;
554
        break;
555 556
    case 0x20001000: /* real rv20 decoder fail on this id */
    /*case 0x20100001:
557
    case 0x20101001:
558 559
    case 0x20103001:*/
    case 0x20100000 ... 0x2019ffff:
560 561
        s->low_delay=1;
        break;
562
    /*case 0x20200002:
M
Michael Niedermayer 已提交
563
    case 0x20201002:
564 565
    case 0x20203002:*/
    case 0x20200002 ... 0x202fffff:
566 567 568 569 570
    case 0x30202002:
    case 0x30203002:
        s->low_delay=0;
        s->avctx->has_b_frames=1;
        break;
571
    default:
572
        av_log(s->avctx, AV_LOG_ERROR, "unknown header %X\n", avctx->sub_id);
573
    }
574

575 576 577
    if(avctx->debug & FF_DEBUG_PICT_INFO){
        av_log(avctx, AV_LOG_DEBUG, "ver:%X ver0:%X\n", avctx->sub_id, avctx->extradata_size >= 4 ? ((uint32_t*)avctx->extradata)[0] : -1);
    }
578 579 580

    avctx->pix_fmt = PIX_FMT_YUV420P;

F
Fabrice Bellard 已提交
581 582 583 584 585 586 587
    if (MPV_common_init(s) < 0)
        return -1;

    h263_decode_init_vlc(s);

    /* init rv vlc */
    if (!done) {
588
        init_vlc(&rv_dc_lum, DC_VLC_BITS, 256,
F
Fabrice Bellard 已提交
589
                 rv_lum_bits, 1, 1,
590
                 rv_lum_code, 2, 2, 1);
591
        init_vlc(&rv_dc_chrom, DC_VLC_BITS, 256,
F
Fabrice Bellard 已提交
592
                 rv_chrom_bits, 1, 1,
593
                 rv_chrom_code, 2, 2, 1);
F
Fabrice Bellard 已提交
594 595 596 597 598 599 600 601 602 603 604 605 606 607
        done = 1;
    }

    return 0;
}

static int rv10_decode_end(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;

    MPV_common_end(s);
    return 0;
}

608
static int rv10_decode_packet(AVCodecContext *avctx,
609
                             uint8_t *buf, int buf_size)
F
Fabrice Bellard 已提交
610 611
{
    MpegEncContext *s = avctx->priv_data;
612
    int mb_count, mb_pos, left, start_mb_x;
F
Fabrice Bellard 已提交
613

614
    init_get_bits(&s->gb, buf, buf_size*8);
M
Michael Niedermayer 已提交
615 616 617 618
    if(s->codec_id ==CODEC_ID_RV10)
        mb_count = rv10_decode_picture_header(s);
    else
        mb_count = rv20_decode_picture_header(s);
F
Fabrice Bellard 已提交
619
    if (mb_count < 0) {
620
        av_log(s->avctx, AV_LOG_ERROR, "HEADER ERROR\n");
F
Fabrice Bellard 已提交
621 622
        return -1;
    }
623

F
Fabrice Bellard 已提交
624 625
    if (s->mb_x >= s->mb_width ||
        s->mb_y >= s->mb_height) {
626
        av_log(s->avctx, AV_LOG_ERROR, "POS ERROR %d %d\n", s->mb_x, s->mb_y);
F
Fabrice Bellard 已提交
627 628 629 630 631
        return -1;
    }
    mb_pos = s->mb_y * s->mb_width + s->mb_x;
    left = s->mb_width * s->mb_height - mb_pos;
    if (mb_count > left) {
632
        av_log(s->avctx, AV_LOG_ERROR, "COUNT ERROR\n");
F
Fabrice Bellard 已提交
633 634
        return -1;
    }
M
Michael Niedermayer 已提交
635
//if(s->pict_type == P_TYPE) return 0;
F
Fabrice Bellard 已提交
636

M
Michael Niedermayer 已提交
637
    if ((s->mb_x == 0 && s->mb_y == 0) || s->current_picture_ptr==NULL) {
M
Michael Niedermayer 已提交
638 639 640 641 642
        if(s->current_picture_ptr){ //FIXME write parser so we always have complete frames?
            ff_er_frame_end(s);
            MPV_frame_end(s);
            s->mb_x= s->mb_y = s->resync_mb_x = s->resync_mb_y= 0;
        }
643 644
        if(MPV_frame_start(s, avctx) < 0)
            return -1;
M
Michael Niedermayer 已提交
645
        ff_er_frame_start(s);
F
Fabrice Bellard 已提交
646 647 648
    }

#ifdef DEBUG
S
Steve L'Homme 已提交
649
    av_log(avctx, AV_LOG_DEBUG, "qscale=%d\n", s->qscale);
F
Fabrice Bellard 已提交
650 651 652
#endif

    /* default quantization values */
M
Michael Niedermayer 已提交
653 654 655
    if(s->codec_id== CODEC_ID_RV10){
        if(s->mb_y==0) s->first_slice_line=1;
    }else{
656
        s->first_slice_line=1;
657
        s->resync_mb_x= s->mb_x;
M
Michael Niedermayer 已提交
658
    }
659
    start_mb_x= s->mb_x;
660
    s->resync_mb_y= s->mb_y;
M
Michael Niedermayer 已提交
661
    if(s->h263_aic){
662
        s->y_dc_scale_table=
M
Michael Niedermayer 已提交
663 664 665 666 667
        s->c_dc_scale_table= ff_aic_dc_scale_table;
    }else{
        s->y_dc_scale_table=
        s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
    }
M
Michael Niedermayer 已提交
668 669 670

    if(s->modified_quant)
        s->chroma_qscale_table= ff_h263_chroma_qscale_table;
671

M
cleanup  
Michael Niedermayer 已提交
672
    ff_set_qscale(s, s->qscale);
M
Michael Niedermayer 已提交
673

F
Fabrice Bellard 已提交
674 675 676
    s->rv10_first_dc_coded[0] = 0;
    s->rv10_first_dc_coded[1] = 0;
    s->rv10_first_dc_coded[2] = 0;
677
//printf("%d %X %X\n", s->pict_type, s->current_picture.motion_val[0], s->current_picture.motion_val[1]);
M
bugfix  
Michael Niedermayer 已提交
678 679 680
    s->block_wrap[0]=
    s->block_wrap[1]=
    s->block_wrap[2]=
681
    s->block_wrap[3]= s->b8_stride;
M
bugfix  
Michael Niedermayer 已提交
682
    s->block_wrap[4]=
683
    s->block_wrap[5]= s->mb_stride;
684
    ff_init_block_index(s);
F
Fabrice Bellard 已提交
685
    /* decode each macroblock */
M
Michael Niedermayer 已提交
686 687

    for(s->mb_num_left= mb_count; s->mb_num_left>0; s->mb_num_left--) {
M
Michael Niedermayer 已提交
688
        int ret;
689
        ff_update_block_index(s);
F
Fabrice Bellard 已提交
690
#ifdef DEBUG
S
Steve L'Homme 已提交
691
        av_log(avctx, AV_LOG_DEBUG, "**mb x=%d y=%d\n", s->mb_x, s->mb_y);
F
Fabrice Bellard 已提交
692
#endif
M
Michael Niedermayer 已提交
693

F
Fabrice Bellard 已提交
694
        s->mv_dir = MV_DIR_FORWARD;
695
        s->mv_type = MV_TYPE_16X16;
M
Michael Niedermayer 已提交
696 697
        ret=ff_h263_decode_mb(s, s->block);

M
Michael Niedermayer 已提交
698
        if (ret == SLICE_ERROR || s->gb.size_in_bits < get_bits_count(&s->gb)) {
699
            av_log(s->avctx, AV_LOG_ERROR, "ERROR at MB %d %d\n", s->mb_x, s->mb_y);
F
Fabrice Bellard 已提交
700 701
            return -1;
        }
702 703
        if(s->pict_type != B_TYPE)
            ff_h263_update_motion_val(s);
704
        MPV_decode_mb(s, s->block);
M
Michael Niedermayer 已提交
705 706 707
        if(s->loop_filter)
            ff_h263_loop_filter(s);

F
Fabrice Bellard 已提交
708 709 710
        if (++s->mb_x == s->mb_width) {
            s->mb_x = 0;
            s->mb_y++;
711
            ff_init_block_index(s);
F
Fabrice Bellard 已提交
712
        }
M
Michael Niedermayer 已提交
713 714 715
        if(s->mb_x == s->resync_mb_x)
            s->first_slice_line=0;
        if(ret == SLICE_END) break;
F
Fabrice Bellard 已提交
716 717
    }

718
    ff_er_add_slice(s, start_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, AC_END|DC_END|MV_END);
M
Michael Niedermayer 已提交
719

720 721 722
    return buf_size;
}

723
static int rv10_decode_frame(AVCodecContext *avctx,
724
                             void *data, int *data_size,
725
                             uint8_t *buf, int buf_size)
726 727 728
{
    MpegEncContext *s = avctx->priv_data;
    int i;
729
    AVFrame *pict = data;
730 731

#ifdef DEBUG
S
Steve L'Homme 已提交
732
    av_log(avctx, AV_LOG_DEBUG, "*****frame %d size=%d\n", avctx->frame_number, buf_size);
733 734 735 736 737 738
#endif

    /* no supplementary picture */
    if (buf_size == 0) {
        return 0;
    }
739

740 741 742 743
    if(avctx->slice_count){
        for(i=0; i<avctx->slice_count; i++){
            int offset= avctx->slice_offset[i];
            int size;
744

745 746 747 748 749
            if(i+1 == avctx->slice_count)
                size= buf_size - offset;
            else
                size= avctx->slice_offset[i+1] - offset;

M
Michael Niedermayer 已提交
750
            rv10_decode_packet(avctx, buf+offset, size);
751 752
        }
    }else{
M
Michael Niedermayer 已提交
753
        rv10_decode_packet(avctx, buf, buf_size);
754
    }
755

756
    if(s->current_picture_ptr != NULL && s->mb_y>=s->mb_height){
M
Michael Niedermayer 已提交
757
        ff_er_frame_end(s);
F
Fabrice Bellard 已提交
758
        MPV_frame_end(s);
759

760 761 762 763
        if (s->pict_type == B_TYPE || s->low_delay) {
            *pict= *(AVFrame*)s->current_picture_ptr;
        } else if (s->last_picture_ptr != NULL) {
            *pict= *(AVFrame*)s->last_picture_ptr;
764
        }
765 766

        if(s->last_picture_ptr || s->low_delay){
M
Michael Niedermayer 已提交
767
            *data_size = sizeof(AVFrame);
768 769
            ff_print_debug_info(s, pict);
        }
M
Michael Niedermayer 已提交
770
        s->current_picture_ptr= NULL; //so we can detect if frame_end wasnt called (find some nicer solution...)
F
Fabrice Bellard 已提交
771
    }
772

F
Fabrice Bellard 已提交
773 774 775 776 777 778 779 780 781 782 783 784
    return buf_size;
}

AVCodec rv10_decoder = {
    "rv10",
    CODEC_TYPE_VIDEO,
    CODEC_ID_RV10,
    sizeof(MpegEncContext),
    rv10_decode_init,
    NULL,
    rv10_decode_end,
    rv10_decode_frame,
M
dr1  
Michael Niedermayer 已提交
785
    CODEC_CAP_DR1
F
Fabrice Bellard 已提交
786
};
M
Michael Niedermayer 已提交
787 788 789 790 791 792 793 794 795 796

AVCodec rv20_decoder = {
    "rv20",
    CODEC_TYPE_VIDEO,
    CODEC_ID_RV20,
    sizeof(MpegEncContext),
    rv10_decode_init,
    NULL,
    rv10_decode_end,
    rv10_decode_frame,
797
    CODEC_CAP_DR1 | CODEC_CAP_DELAY,
M
Michael Niedermayer 已提交
798
    .flush= ff_mpeg_flush,
M
Michael Niedermayer 已提交
799
};
800