RealVideo 3 decoder functions
[ffmpeg.git] / libavcodec / rv30.c
1 /*
2  * RV30 decoder
3  * Copyright (c) 2007 Konstantin Shishkov
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 /**
23  * @file rv30.c
24  * RV30 decoder.
25  */
26
27 #include "avcodec.h"
28 #include "dsputil.h"
29 #include "mpegvideo.h"
30
31 #include "rv34.h"
32 #include "rv30data.h"
33
34
35 static int rv30_parse_slice_header(RV34DecContext *r, GetBitContext *gb, SliceInfo *si)
36 {
37     int mb_bits;
38     int w = r->s.width, h = r->s.height;
39     int mb_size;
40
41     memset(si, 0, sizeof(SliceInfo));
42     skip_bits(gb, 3);
43     si->type = get_bits(gb, 2);
44     if(si->type == 1) si->type = 0;
45     if(get_bits1(gb))
46         return -1;
47     si->quant = get_bits(gb, 5);
48     skip_bits1(gb);
49     skip_bits(gb, 13); // timestamp
50     skip_bits(gb, r->rpr);
51     si->width  = w;
52     si->height = h;
53     mb_size = ((w + 15) >> 4) * ((h + 15) >> 4);
54     mb_bits = ff_rv34_get_start_offset(gb, mb_size);
55     si->start = get_bits(gb, mb_bits);
56     skip_bits1(gb);
57     return 0;
58 }
59
60 /**
61  * Decode 4x4 intra types array
62  */
63 static int rv30_decode_intra_types(RV34DecContext *r, GetBitContext *gb, int *dst)
64 {
65     int i, j, k;
66
67     for(i = 0; i < 4; i++, dst += r->s.b4_stride - 4){
68         for(j = 0; j < 4; j+= 2){
69             int code = (ff_rv34_get_gamma(gb) - 1) << 1;
70             if(code >= 81*2){
71                 av_log(r->s.avctx, AV_LOG_ERROR, "Incorrect intra prediction code\n");
72                 return -1;
73             }
74             for(k = 0; k < 2; k++){
75                 int A = dst[-r->s.b4_stride] + 1;
76                 int B = dst[-1] + 1;
77                 *dst++ = rv30_itype_from_context[A * 90 + B * 9 + rv30_itype_code[code + k]];
78                 if(dst[-1] == 9){
79                     av_log(r->s.avctx, AV_LOG_ERROR, "Incorrect intra prediction mode\n");
80                     return -1;
81                 }
82             }
83         }
84     }
85     return 0;
86 }
87
88 /**
89  * Decode macroblock information
90  */
91 static int rv30_decode_mb_info(RV34DecContext *r)
92 {
93     static const int rv30_p_types[6] = { RV34_MB_SKIP, RV34_MB_P_16x16, RV34_MB_P_8x8, -1, RV34_MB_TYPE_INTRA, RV34_MB_TYPE_INTRA16x16 };
94     static const int rv30_b_types[6] = { RV34_MB_SKIP, RV34_MB_B_DIRECT, RV34_MB_B_FORWARD, RV34_MB_B_BACKWARD, RV34_MB_TYPE_INTRA, RV34_MB_TYPE_INTRA16x16 };
95     MpegEncContext *s = &r->s;
96     GetBitContext *gb = &s->gb;
97     int code = ff_rv34_get_gamma(gb) - 1;
98
99     if(code > 11){
100         av_log(s->avctx, AV_LOG_ERROR, "Incorrect MB type code\n");
101         return -1;
102     }
103     if(code > 5){
104         av_log(s->avctx, AV_LOG_ERROR, "dquant needed\n");
105         code -= 6;
106     }
107     if(s->pict_type != B_TYPE)
108         return rv30_p_types[code];
109     else
110         return rv30_b_types[code];
111 }
112
113 /**
114  * Initialize decoder
115  */
116 static int rv30_decode_init(AVCodecContext *avctx)
117 {
118     RV34DecContext *r = avctx->priv_data;
119
120     r->rv30 = 1;
121     ff_rv34_decode_init(avctx);
122     if(avctx->extradata_size < 2){
123         av_log(avctx, AV_LOG_ERROR, "Extradata is too small\n");
124         return -1;
125     }
126     r->rpr = (avctx->extradata[1] & 7) >> 1;
127     r->rpr = FFMIN(r->rpr + 1, 3);
128     r->parse_slice_header = rv30_parse_slice_header;
129     r->decode_intra_types = rv30_decode_intra_types;
130     r->decode_mb_info     = rv30_decode_mb_info;
131     r->luma_dc_quant_i = rv30_luma_dc_quant;
132     r->luma_dc_quant_p = rv30_luma_dc_quant;
133     return 0;
134 }
135
136 AVCodec rv30_decoder = {
137     "rv30",
138     CODEC_TYPE_VIDEO,
139     CODEC_ID_RV30,
140     sizeof(RV34DecContext),
141     rv30_decode_init,
142     NULL,
143     ff_rv34_decode_end,
144     ff_rv34_decode_frame,
145 };