annotate h263.c @ 70:c7767df463f4 libavcodec

log2 to av_log2
author glantau
date Wed, 15 Aug 2001 13:11:31 +0000
parents 8e2d8dbf89a5
children 92ed1c4f94fa
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1 /*
986e461dc072 Initial revision
glantau
parents:
diff changeset
2 * H263/MPEG4 backend for ffmpeg encoder and decoder
986e461dc072 Initial revision
glantau
parents:
diff changeset
3 * Copyright (c) 2000,2001 Gerard Lantau.
986e461dc072 Initial revision
glantau
parents:
diff changeset
4 * H263+ support for custom picture format.
986e461dc072 Initial revision
glantau
parents:
diff changeset
5 * Copyright (c) 2001 Juan J. Sierralta P.
986e461dc072 Initial revision
glantau
parents:
diff changeset
6 *
986e461dc072 Initial revision
glantau
parents:
diff changeset
7 * This program is free software; you can redistribute it and/or modify
986e461dc072 Initial revision
glantau
parents:
diff changeset
8 * it under the terms of the GNU General Public License as published by
986e461dc072 Initial revision
glantau
parents:
diff changeset
9 * the Free Software Foundation; either version 2 of the License, or
986e461dc072 Initial revision
glantau
parents:
diff changeset
10 * (at your option) any later version.
986e461dc072 Initial revision
glantau
parents:
diff changeset
11 *
986e461dc072 Initial revision
glantau
parents:
diff changeset
12 * This program is distributed in the hope that it will be useful,
986e461dc072 Initial revision
glantau
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
986e461dc072 Initial revision
glantau
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
986e461dc072 Initial revision
glantau
parents:
diff changeset
15 * GNU General Public License for more details.
986e461dc072 Initial revision
glantau
parents:
diff changeset
16 *
986e461dc072 Initial revision
glantau
parents:
diff changeset
17 * You should have received a copy of the GNU General Public License
986e461dc072 Initial revision
glantau
parents:
diff changeset
18 * along with this program; if not, write to the Free Software
986e461dc072 Initial revision
glantau
parents:
diff changeset
19 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
986e461dc072 Initial revision
glantau
parents:
diff changeset
20 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
21 #include "common.h"
986e461dc072 Initial revision
glantau
parents:
diff changeset
22 #include "dsputil.h"
986e461dc072 Initial revision
glantau
parents:
diff changeset
23 #include "avcodec.h"
986e461dc072 Initial revision
glantau
parents:
diff changeset
24 #include "mpegvideo.h"
986e461dc072 Initial revision
glantau
parents:
diff changeset
25 #include "h263data.h"
986e461dc072 Initial revision
glantau
parents:
diff changeset
26 #include "mpeg4data.h"
986e461dc072 Initial revision
glantau
parents:
diff changeset
27
986e461dc072 Initial revision
glantau
parents:
diff changeset
28 static void h263_encode_block(MpegEncContext * s, DCTELEM * block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
29 int n);
986e461dc072 Initial revision
glantau
parents:
diff changeset
30 static void h263_encode_motion(MpegEncContext * s, int val);
986e461dc072 Initial revision
glantau
parents:
diff changeset
31 static void mpeg4_encode_block(MpegEncContext * s, DCTELEM * block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
32 int n);
986e461dc072 Initial revision
glantau
parents:
diff changeset
33 static int h263_decode_motion(MpegEncContext * s, int pred);
986e461dc072 Initial revision
glantau
parents:
diff changeset
34 static int h263_decode_block(MpegEncContext * s, DCTELEM * block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
35 int n, int coded);
986e461dc072 Initial revision
glantau
parents:
diff changeset
36 static int mpeg4_decode_block(MpegEncContext * s, DCTELEM * block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
37 int n, int coded);
986e461dc072 Initial revision
glantau
parents:
diff changeset
38
986e461dc072 Initial revision
glantau
parents:
diff changeset
39 int h263_get_picture_format(int width, int height)
986e461dc072 Initial revision
glantau
parents:
diff changeset
40 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
41 int format;
986e461dc072 Initial revision
glantau
parents:
diff changeset
42
986e461dc072 Initial revision
glantau
parents:
diff changeset
43 if (width == 128 && height == 96)
986e461dc072 Initial revision
glantau
parents:
diff changeset
44 format = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
45 else if (width == 176 && height == 144)
986e461dc072 Initial revision
glantau
parents:
diff changeset
46 format = 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
47 else if (width == 352 && height == 288)
986e461dc072 Initial revision
glantau
parents:
diff changeset
48 format = 3;
986e461dc072 Initial revision
glantau
parents:
diff changeset
49 else if (width == 704 && height == 576)
986e461dc072 Initial revision
glantau
parents:
diff changeset
50 format = 4;
986e461dc072 Initial revision
glantau
parents:
diff changeset
51 else if (width == 1408 && height == 1152)
986e461dc072 Initial revision
glantau
parents:
diff changeset
52 format = 5;
986e461dc072 Initial revision
glantau
parents:
diff changeset
53 else
986e461dc072 Initial revision
glantau
parents:
diff changeset
54 format = 7;
986e461dc072 Initial revision
glantau
parents:
diff changeset
55 return format;
986e461dc072 Initial revision
glantau
parents:
diff changeset
56 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
57
986e461dc072 Initial revision
glantau
parents:
diff changeset
58 void h263_encode_picture_header(MpegEncContext * s, int picture_number)
986e461dc072 Initial revision
glantau
parents:
diff changeset
59 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
60 int format, umvplus;
986e461dc072 Initial revision
glantau
parents:
diff changeset
61
986e461dc072 Initial revision
glantau
parents:
diff changeset
62 align_put_bits(&s->pb);
986e461dc072 Initial revision
glantau
parents:
diff changeset
63 put_bits(&s->pb, 22, 0x20);
986e461dc072 Initial revision
glantau
parents:
diff changeset
64 put_bits(&s->pb, 8, ((s->picture_number * 30 * FRAME_RATE_BASE) /
986e461dc072 Initial revision
glantau
parents:
diff changeset
65 s->frame_rate) & 0xff);
986e461dc072 Initial revision
glantau
parents:
diff changeset
66
986e461dc072 Initial revision
glantau
parents:
diff changeset
67 put_bits(&s->pb, 1, 1); /* marker */
986e461dc072 Initial revision
glantau
parents:
diff changeset
68 put_bits(&s->pb, 1, 0); /* h263 id */
986e461dc072 Initial revision
glantau
parents:
diff changeset
69 put_bits(&s->pb, 1, 0); /* split screen off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
70 put_bits(&s->pb, 1, 0); /* camera off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
71 put_bits(&s->pb, 1, 0); /* freeze picture release off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
72
986e461dc072 Initial revision
glantau
parents:
diff changeset
73 if (!s->h263_plus) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
74 /* H.263v1 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
75 format = h263_get_picture_format(s->width, s->height);
986e461dc072 Initial revision
glantau
parents:
diff changeset
76 put_bits(&s->pb, 3, format);
986e461dc072 Initial revision
glantau
parents:
diff changeset
77 put_bits(&s->pb, 1, (s->pict_type == P_TYPE));
986e461dc072 Initial revision
glantau
parents:
diff changeset
78 /* By now UMV IS DISABLED ON H.263v1, since the restrictions
986e461dc072 Initial revision
glantau
parents:
diff changeset
79 of H.263v1 UMV implies to check the predicted MV after
986e461dc072 Initial revision
glantau
parents:
diff changeset
80 calculation of the current MB to see if we're on the limits */
986e461dc072 Initial revision
glantau
parents:
diff changeset
81 put_bits(&s->pb, 1, 0); /* unrestricted motion vector: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
82 put_bits(&s->pb, 1, 0); /* SAC: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
83 put_bits(&s->pb, 1, 0); /* advanced prediction mode: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
84 put_bits(&s->pb, 1, 0); /* not PB frame */
986e461dc072 Initial revision
glantau
parents:
diff changeset
85 put_bits(&s->pb, 5, s->qscale);
986e461dc072 Initial revision
glantau
parents:
diff changeset
86 put_bits(&s->pb, 1, 0); /* Continuous Presence Multipoint mode: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
87 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
88 /* H.263v2 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
89 /* H.263 Plus PTYPE */
986e461dc072 Initial revision
glantau
parents:
diff changeset
90 put_bits(&s->pb, 3, 7);
986e461dc072 Initial revision
glantau
parents:
diff changeset
91 put_bits(&s->pb,3,1); /* Update Full Extended PTYPE */
986e461dc072 Initial revision
glantau
parents:
diff changeset
92 put_bits(&s->pb,3,6); /* Custom Source Format */
986e461dc072 Initial revision
glantau
parents:
diff changeset
93 put_bits(&s->pb,1,0); /* Custom PCF: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
94 umvplus = (s->pict_type == P_TYPE) && s->unrestricted_mv;
986e461dc072 Initial revision
glantau
parents:
diff changeset
95 put_bits(&s->pb, 1, umvplus); /* Unrestricted Motion Vector */
986e461dc072 Initial revision
glantau
parents:
diff changeset
96 put_bits(&s->pb,1,0); /* SAC: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
97 put_bits(&s->pb,1,0); /* Advanced Prediction Mode: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
98 put_bits(&s->pb,1,0); /* Advanced Intra Coding: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
99 put_bits(&s->pb,1,0); /* Deblocking Filter: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
100 put_bits(&s->pb,1,0); /* Slice Structured: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
101 put_bits(&s->pb,1,0); /* Reference Picture Selection: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
102 put_bits(&s->pb,1,0); /* Independent Segment Decoding: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
103 put_bits(&s->pb,1,0); /* Alternative Inter VLC: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
104 put_bits(&s->pb,1,0); /* Modified Quantization: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
105 put_bits(&s->pb,1,1); /* "1" to prevent start code emulation */
986e461dc072 Initial revision
glantau
parents:
diff changeset
106 put_bits(&s->pb,3,0); /* Reserved */
986e461dc072 Initial revision
glantau
parents:
diff changeset
107
986e461dc072 Initial revision
glantau
parents:
diff changeset
108 put_bits(&s->pb, 3, s->pict_type == P_TYPE);
986e461dc072 Initial revision
glantau
parents:
diff changeset
109
986e461dc072 Initial revision
glantau
parents:
diff changeset
110 put_bits(&s->pb,1,0); /* Reference Picture Resampling: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
111 put_bits(&s->pb,1,0); /* Reduced-Resolution Update: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
112 put_bits(&s->pb,1,0); /* Rounding Type */
986e461dc072 Initial revision
glantau
parents:
diff changeset
113 put_bits(&s->pb,2,0); /* Reserved */
986e461dc072 Initial revision
glantau
parents:
diff changeset
114 put_bits(&s->pb,1,1); /* "1" to prevent start code emulation */
986e461dc072 Initial revision
glantau
parents:
diff changeset
115
986e461dc072 Initial revision
glantau
parents:
diff changeset
116 /* This should be here if PLUSPTYPE */
986e461dc072 Initial revision
glantau
parents:
diff changeset
117 put_bits(&s->pb, 1, 0); /* Continuous Presence Multipoint mode: off */
986e461dc072 Initial revision
glantau
parents:
diff changeset
118
986e461dc072 Initial revision
glantau
parents:
diff changeset
119 /* Custom Picture Format (CPFMT) */
986e461dc072 Initial revision
glantau
parents:
diff changeset
120
986e461dc072 Initial revision
glantau
parents:
diff changeset
121 put_bits(&s->pb,4,2); /* Aspect ratio: CIF 12:11 (4:3) picture */
986e461dc072 Initial revision
glantau
parents:
diff changeset
122 put_bits(&s->pb,9,(s->width >> 2) - 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
123 put_bits(&s->pb,1,1); /* "1" to prevent start code emulation */
986e461dc072 Initial revision
glantau
parents:
diff changeset
124 put_bits(&s->pb,9,(s->height >> 2));
986e461dc072 Initial revision
glantau
parents:
diff changeset
125 /* Unlimited Unrestricted Motion Vectors Indicator (UUI) */
986e461dc072 Initial revision
glantau
parents:
diff changeset
126 if (umvplus)
986e461dc072 Initial revision
glantau
parents:
diff changeset
127 put_bits(&s->pb,1,1); /* Limited according tables of Annex D */
986e461dc072 Initial revision
glantau
parents:
diff changeset
128 put_bits(&s->pb, 5, s->qscale);
986e461dc072 Initial revision
glantau
parents:
diff changeset
129 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
130
986e461dc072 Initial revision
glantau
parents:
diff changeset
131 put_bits(&s->pb, 1, 0); /* no PEI */
986e461dc072 Initial revision
glantau
parents:
diff changeset
132 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
133
986e461dc072 Initial revision
glantau
parents:
diff changeset
134 void h263_encode_mb(MpegEncContext * s,
986e461dc072 Initial revision
glantau
parents:
diff changeset
135 DCTELEM block[6][64],
986e461dc072 Initial revision
glantau
parents:
diff changeset
136 int motion_x, int motion_y)
986e461dc072 Initial revision
glantau
parents:
diff changeset
137 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
138 int cbpc, cbpy, i, cbp, pred_x, pred_y;
986e461dc072 Initial revision
glantau
parents:
diff changeset
139
986e461dc072 Initial revision
glantau
parents:
diff changeset
140 // printf("**mb x=%d y=%d\n", s->mb_x, s->mb_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
141 if (!s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
142 /* compute cbp */
986e461dc072 Initial revision
glantau
parents:
diff changeset
143 cbp = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
144 for (i = 0; i < 6; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
145 if (s->block_last_index[i] >= 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
146 cbp |= 1 << (5 - i);
986e461dc072 Initial revision
glantau
parents:
diff changeset
147 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
148 if ((cbp | motion_x | motion_y) == 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
149 /* skip macroblock */
986e461dc072 Initial revision
glantau
parents:
diff changeset
150 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
151 return;
986e461dc072 Initial revision
glantau
parents:
diff changeset
152 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
153 put_bits(&s->pb, 1, 0); /* mb coded */
986e461dc072 Initial revision
glantau
parents:
diff changeset
154 cbpc = cbp & 3;
986e461dc072 Initial revision
glantau
parents:
diff changeset
155 put_bits(&s->pb,
986e461dc072 Initial revision
glantau
parents:
diff changeset
156 inter_MCBPC_bits[cbpc],
986e461dc072 Initial revision
glantau
parents:
diff changeset
157 inter_MCBPC_code[cbpc]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
158 cbpy = cbp >> 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
159 cbpy ^= 0xf;
986e461dc072 Initial revision
glantau
parents:
diff changeset
160 put_bits(&s->pb, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
161
986e461dc072 Initial revision
glantau
parents:
diff changeset
162 /* motion vectors: 16x16 mode only now */
986e461dc072 Initial revision
glantau
parents:
diff changeset
163 h263_pred_motion(s, 0, &pred_x, &pred_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
164
986e461dc072 Initial revision
glantau
parents:
diff changeset
165 h263_encode_motion(s, motion_x - pred_x);
986e461dc072 Initial revision
glantau
parents:
diff changeset
166 h263_encode_motion(s, motion_y - pred_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
167 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
168 /* compute cbp */
986e461dc072 Initial revision
glantau
parents:
diff changeset
169 cbp = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
170 for (i = 0; i < 6; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
171 if (s->block_last_index[i] >= 1)
986e461dc072 Initial revision
glantau
parents:
diff changeset
172 cbp |= 1 << (5 - i);
986e461dc072 Initial revision
glantau
parents:
diff changeset
173 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
174
986e461dc072 Initial revision
glantau
parents:
diff changeset
175 cbpc = cbp & 3;
986e461dc072 Initial revision
glantau
parents:
diff changeset
176 if (s->pict_type == I_TYPE) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
177 put_bits(&s->pb,
986e461dc072 Initial revision
glantau
parents:
diff changeset
178 intra_MCBPC_bits[cbpc],
986e461dc072 Initial revision
glantau
parents:
diff changeset
179 intra_MCBPC_code[cbpc]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
180 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
181 put_bits(&s->pb, 1, 0); /* mb coded */
986e461dc072 Initial revision
glantau
parents:
diff changeset
182 put_bits(&s->pb,
986e461dc072 Initial revision
glantau
parents:
diff changeset
183 inter_MCBPC_bits[cbpc + 4],
986e461dc072 Initial revision
glantau
parents:
diff changeset
184 inter_MCBPC_code[cbpc + 4]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
185 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
186 if (s->h263_pred) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
187 /* XXX: currently, we do not try to use ac prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
188 put_bits(&s->pb, 1, 0); /* no ac prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
189 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
190 cbpy = cbp >> 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
191 put_bits(&s->pb, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
192 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
193
986e461dc072 Initial revision
glantau
parents:
diff changeset
194 /* encode each block */
986e461dc072 Initial revision
glantau
parents:
diff changeset
195 if (s->h263_pred) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
196 for (i = 0; i < 6; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
197 mpeg4_encode_block(s, block[i], i);
986e461dc072 Initial revision
glantau
parents:
diff changeset
198 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
199 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
200 for (i = 0; i < 6; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
201 h263_encode_block(s, block[i], i);
986e461dc072 Initial revision
glantau
parents:
diff changeset
202 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
203 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
204 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
205
986e461dc072 Initial revision
glantau
parents:
diff changeset
206 static inline int mid_pred(int a, int b, int c)
986e461dc072 Initial revision
glantau
parents:
diff changeset
207 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
208 int vmin, vmax;
986e461dc072 Initial revision
glantau
parents:
diff changeset
209 vmin = a;
986e461dc072 Initial revision
glantau
parents:
diff changeset
210 if (b < vmin)
986e461dc072 Initial revision
glantau
parents:
diff changeset
211 vmin = b;
986e461dc072 Initial revision
glantau
parents:
diff changeset
212 if (c < vmin)
986e461dc072 Initial revision
glantau
parents:
diff changeset
213 vmin = c;
986e461dc072 Initial revision
glantau
parents:
diff changeset
214
986e461dc072 Initial revision
glantau
parents:
diff changeset
215 vmax = a;
986e461dc072 Initial revision
glantau
parents:
diff changeset
216 if (b > vmax)
986e461dc072 Initial revision
glantau
parents:
diff changeset
217 vmax = b;
986e461dc072 Initial revision
glantau
parents:
diff changeset
218 if (c > vmax)
986e461dc072 Initial revision
glantau
parents:
diff changeset
219 vmax = c;
986e461dc072 Initial revision
glantau
parents:
diff changeset
220
986e461dc072 Initial revision
glantau
parents:
diff changeset
221 return a + b + c - vmin - vmax;
986e461dc072 Initial revision
glantau
parents:
diff changeset
222 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
223
986e461dc072 Initial revision
glantau
parents:
diff changeset
224 INT16 *h263_pred_motion(MpegEncContext * s, int block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
225 int *px, int *py)
986e461dc072 Initial revision
glantau
parents:
diff changeset
226 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
227 int x, y, wrap;
986e461dc072 Initial revision
glantau
parents:
diff changeset
228 INT16 *A, *B, *C, *mot_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
229
986e461dc072 Initial revision
glantau
parents:
diff changeset
230 x = 2 * s->mb_x + 1 + (block & 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
231 y = 2 * s->mb_y + 1 + ((block >> 1) & 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
232 wrap = 2 * s->mb_width + 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
233
986e461dc072 Initial revision
glantau
parents:
diff changeset
234 mot_val = s->motion_val[(x) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
235
986e461dc072 Initial revision
glantau
parents:
diff changeset
236 /* special case for first line */
986e461dc072 Initial revision
glantau
parents:
diff changeset
237 if (y == 1 || s->first_slice_line) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
238 A = s->motion_val[(x-1) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
239 *px = A[0];
986e461dc072 Initial revision
glantau
parents:
diff changeset
240 *py = A[1];
986e461dc072 Initial revision
glantau
parents:
diff changeset
241 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
242 switch(block) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
243 default:
986e461dc072 Initial revision
glantau
parents:
diff changeset
244 case 0:
986e461dc072 Initial revision
glantau
parents:
diff changeset
245 A = s->motion_val[(x-1) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
246 B = s->motion_val[(x) + (y-1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
247 C = s->motion_val[(x+2) + (y-1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
248 break;
986e461dc072 Initial revision
glantau
parents:
diff changeset
249 case 1:
986e461dc072 Initial revision
glantau
parents:
diff changeset
250 case 2:
986e461dc072 Initial revision
glantau
parents:
diff changeset
251 A = s->motion_val[(x-1) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
252 B = s->motion_val[(x) + (y-1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
253 C = s->motion_val[(x+1) + (y-1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
254 break;
986e461dc072 Initial revision
glantau
parents:
diff changeset
255 case 3:
986e461dc072 Initial revision
glantau
parents:
diff changeset
256 A = s->motion_val[(x-1) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
257 B = s->motion_val[(x-1) + (y-1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
258 C = s->motion_val[(x) + (y-1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
259 break;
986e461dc072 Initial revision
glantau
parents:
diff changeset
260 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
261 *px = mid_pred(A[0], B[0], C[0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
262 *py = mid_pred(A[1], B[1], C[1]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
263 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
264 return mot_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
265 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
266
986e461dc072 Initial revision
glantau
parents:
diff changeset
267
986e461dc072 Initial revision
glantau
parents:
diff changeset
268 static void h263_encode_motion(MpegEncContext * s, int val)
986e461dc072 Initial revision
glantau
parents:
diff changeset
269 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
270 int range, l, m, bit_size, sign, code, bits;
986e461dc072 Initial revision
glantau
parents:
diff changeset
271
986e461dc072 Initial revision
glantau
parents:
diff changeset
272 if (val == 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
273 /* zero vector */
986e461dc072 Initial revision
glantau
parents:
diff changeset
274 code = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
275 put_bits(&s->pb, mvtab[code][1], mvtab[code][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
276 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
277 bit_size = s->f_code - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
278 range = 1 << bit_size;
986e461dc072 Initial revision
glantau
parents:
diff changeset
279 /* modulo encoding */
986e461dc072 Initial revision
glantau
parents:
diff changeset
280 l = range * 32;
986e461dc072 Initial revision
glantau
parents:
diff changeset
281 m = 2 * l;
986e461dc072 Initial revision
glantau
parents:
diff changeset
282 if (val < -l) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
283 val += m;
986e461dc072 Initial revision
glantau
parents:
diff changeset
284 } else if (val >= l) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
285 val -= m;
986e461dc072 Initial revision
glantau
parents:
diff changeset
286 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
287
986e461dc072 Initial revision
glantau
parents:
diff changeset
288 if (val >= 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
289 val--;
986e461dc072 Initial revision
glantau
parents:
diff changeset
290 code = (val >> bit_size) + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
291 bits = val & (range - 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
292 sign = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
293 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
294 val = -val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
295 val--;
986e461dc072 Initial revision
glantau
parents:
diff changeset
296 code = (val >> bit_size) + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
297 bits = val & (range - 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
298 sign = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
299 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
300
986e461dc072 Initial revision
glantau
parents:
diff changeset
301 put_bits(&s->pb, mvtab[code][1] + 1, (mvtab[code][0] << 1) | sign);
986e461dc072 Initial revision
glantau
parents:
diff changeset
302 if (bit_size > 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
303 put_bits(&s->pb, bit_size, bits);
986e461dc072 Initial revision
glantau
parents:
diff changeset
304 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
305 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
306 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
307
986e461dc072 Initial revision
glantau
parents:
diff changeset
308 void h263_encode_init_vlc(MpegEncContext *s)
986e461dc072 Initial revision
glantau
parents:
diff changeset
309 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
310 static int done = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
311
986e461dc072 Initial revision
glantau
parents:
diff changeset
312 if (!done) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
313 done = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
314 init_rl(&rl_inter);
986e461dc072 Initial revision
glantau
parents:
diff changeset
315 init_rl(&rl_intra);
986e461dc072 Initial revision
glantau
parents:
diff changeset
316 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
317 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
318
986e461dc072 Initial revision
glantau
parents:
diff changeset
319 static void h263_encode_block(MpegEncContext * s, DCTELEM * block, int n)
986e461dc072 Initial revision
glantau
parents:
diff changeset
320 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
321 int level, run, last, i, j, last_index, last_non_zero, sign, slevel;
986e461dc072 Initial revision
glantau
parents:
diff changeset
322 int code;
986e461dc072 Initial revision
glantau
parents:
diff changeset
323 RLTable *rl = &rl_inter;
986e461dc072 Initial revision
glantau
parents:
diff changeset
324
986e461dc072 Initial revision
glantau
parents:
diff changeset
325 if (s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
326 /* DC coef */
986e461dc072 Initial revision
glantau
parents:
diff changeset
327 level = block[0];
986e461dc072 Initial revision
glantau
parents:
diff changeset
328 /* 255 cannot be represented, so we clamp */
986e461dc072 Initial revision
glantau
parents:
diff changeset
329 if (level > 254) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
330 level = 254;
986e461dc072 Initial revision
glantau
parents:
diff changeset
331 block[0] = 254;
986e461dc072 Initial revision
glantau
parents:
diff changeset
332 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
333 if (level == 128)
986e461dc072 Initial revision
glantau
parents:
diff changeset
334 put_bits(&s->pb, 8, 0xff);
986e461dc072 Initial revision
glantau
parents:
diff changeset
335 else
986e461dc072 Initial revision
glantau
parents:
diff changeset
336 put_bits(&s->pb, 8, level & 0xff);
986e461dc072 Initial revision
glantau
parents:
diff changeset
337 i = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
338 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
339 i = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
340 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
341
986e461dc072 Initial revision
glantau
parents:
diff changeset
342 /* AC coefs */
986e461dc072 Initial revision
glantau
parents:
diff changeset
343 last_index = s->block_last_index[n];
986e461dc072 Initial revision
glantau
parents:
diff changeset
344 last_non_zero = i - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
345 for (; i <= last_index; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
346 j = zigzag_direct[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
347 level = block[j];
986e461dc072 Initial revision
glantau
parents:
diff changeset
348 if (level) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
349 run = i - last_non_zero - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
350 last = (i == last_index);
986e461dc072 Initial revision
glantau
parents:
diff changeset
351 sign = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
352 slevel = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
353 if (level < 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
354 sign = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
355 level = -level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
356 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
357 code = get_rl_index(rl, last, run, level);
986e461dc072 Initial revision
glantau
parents:
diff changeset
358 put_bits(&s->pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
359 if (code == rl->n) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
360 put_bits(&s->pb, 1, last);
986e461dc072 Initial revision
glantau
parents:
diff changeset
361 put_bits(&s->pb, 6, run);
986e461dc072 Initial revision
glantau
parents:
diff changeset
362 put_bits(&s->pb, 8, slevel & 0xff);
986e461dc072 Initial revision
glantau
parents:
diff changeset
363 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
364 put_bits(&s->pb, 1, sign);
986e461dc072 Initial revision
glantau
parents:
diff changeset
365 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
366 last_non_zero = i;
986e461dc072 Initial revision
glantau
parents:
diff changeset
367 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
368 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
369 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
370
986e461dc072 Initial revision
glantau
parents:
diff changeset
371 /***************************************************/
986e461dc072 Initial revision
glantau
parents:
diff changeset
372
986e461dc072 Initial revision
glantau
parents:
diff changeset
373 /* write mpeg4 VOP header */
986e461dc072 Initial revision
glantau
parents:
diff changeset
374 void mpeg4_encode_picture_header(MpegEncContext * s, int picture_number)
986e461dc072 Initial revision
glantau
parents:
diff changeset
375 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
376 align_put_bits(&s->pb);
986e461dc072 Initial revision
glantau
parents:
diff changeset
377
986e461dc072 Initial revision
glantau
parents:
diff changeset
378 put_bits(&s->pb, 32, 0x1B6); /* vop header */
986e461dc072 Initial revision
glantau
parents:
diff changeset
379 put_bits(&s->pb, 2, s->pict_type - 1); /* pict type: I = 0 , P = 1 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
380 /* XXX: time base + 1 not always correct */
986e461dc072 Initial revision
glantau
parents:
diff changeset
381 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
382 put_bits(&s->pb, 1, 0);
986e461dc072 Initial revision
glantau
parents:
diff changeset
383
986e461dc072 Initial revision
glantau
parents:
diff changeset
384 put_bits(&s->pb, 1, 1); /* marker */
986e461dc072 Initial revision
glantau
parents:
diff changeset
385 put_bits(&s->pb, 4, 1); /* XXX: correct time increment */
986e461dc072 Initial revision
glantau
parents:
diff changeset
386 put_bits(&s->pb, 1, 1); /* marker */
986e461dc072 Initial revision
glantau
parents:
diff changeset
387 put_bits(&s->pb, 1, 1); /* vop coded */
986e461dc072 Initial revision
glantau
parents:
diff changeset
388 if (s->pict_type == P_TYPE) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
389 s->no_rounding = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
390 put_bits(&s->pb, 1, s->no_rounding); /* rounding type */
986e461dc072 Initial revision
glantau
parents:
diff changeset
391 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
392 put_bits(&s->pb, 3, 0); /* intra dc VLC threshold */
986e461dc072 Initial revision
glantau
parents:
diff changeset
393
986e461dc072 Initial revision
glantau
parents:
diff changeset
394 put_bits(&s->pb, 5, s->qscale);
986e461dc072 Initial revision
glantau
parents:
diff changeset
395
986e461dc072 Initial revision
glantau
parents:
diff changeset
396 if (s->pict_type != I_TYPE)
986e461dc072 Initial revision
glantau
parents:
diff changeset
397 put_bits(&s->pb, 3, s->f_code); /* fcode_for */
986e461dc072 Initial revision
glantau
parents:
diff changeset
398 // printf("****frame %d\n", picture_number);
986e461dc072 Initial revision
glantau
parents:
diff changeset
399 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
400
986e461dc072 Initial revision
glantau
parents:
diff changeset
401 void h263_dc_scale(MpegEncContext * s)
986e461dc072 Initial revision
glantau
parents:
diff changeset
402 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
403 int quant;
986e461dc072 Initial revision
glantau
parents:
diff changeset
404
986e461dc072 Initial revision
glantau
parents:
diff changeset
405 quant = s->qscale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
406 /* luminance */
986e461dc072 Initial revision
glantau
parents:
diff changeset
407 if (quant < 5)
986e461dc072 Initial revision
glantau
parents:
diff changeset
408 s->y_dc_scale = 8;
986e461dc072 Initial revision
glantau
parents:
diff changeset
409 else if (quant > 4 && quant < 9)
986e461dc072 Initial revision
glantau
parents:
diff changeset
410 s->y_dc_scale = (2 * quant);
986e461dc072 Initial revision
glantau
parents:
diff changeset
411 else if (quant > 8 && quant < 25)
986e461dc072 Initial revision
glantau
parents:
diff changeset
412 s->y_dc_scale = (quant + 8);
986e461dc072 Initial revision
glantau
parents:
diff changeset
413 else
986e461dc072 Initial revision
glantau
parents:
diff changeset
414 s->y_dc_scale = (2 * quant - 16);
986e461dc072 Initial revision
glantau
parents:
diff changeset
415 /* chrominance */
986e461dc072 Initial revision
glantau
parents:
diff changeset
416 if (quant < 5)
986e461dc072 Initial revision
glantau
parents:
diff changeset
417 s->c_dc_scale = 8;
986e461dc072 Initial revision
glantau
parents:
diff changeset
418 else if (quant > 4 && quant < 25)
986e461dc072 Initial revision
glantau
parents:
diff changeset
419 s->c_dc_scale = ((quant + 13) / 2);
986e461dc072 Initial revision
glantau
parents:
diff changeset
420 else
986e461dc072 Initial revision
glantau
parents:
diff changeset
421 s->c_dc_scale = (quant - 6);
986e461dc072 Initial revision
glantau
parents:
diff changeset
422 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
423
986e461dc072 Initial revision
glantau
parents:
diff changeset
424 static int mpeg4_pred_dc(MpegEncContext * s, int n, UINT16 **dc_val_ptr, int *dir_ptr)
986e461dc072 Initial revision
glantau
parents:
diff changeset
425 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
426 int a, b, c, x, y, wrap, pred, scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
427 UINT16 *dc_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
428
986e461dc072 Initial revision
glantau
parents:
diff changeset
429 /* find prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
430 if (n < 4) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
431 x = 2 * s->mb_x + 1 + (n & 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
432 y = 2 * s->mb_y + 1 + ((n & 2) >> 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
433 wrap = s->mb_width * 2 + 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
434 dc_val = s->dc_val[0];
986e461dc072 Initial revision
glantau
parents:
diff changeset
435 scale = s->y_dc_scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
436 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
437 x = s->mb_x + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
438 y = s->mb_y + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
439 wrap = s->mb_width + 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
440 dc_val = s->dc_val[n - 4 + 1];
986e461dc072 Initial revision
glantau
parents:
diff changeset
441 scale = s->c_dc_scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
442 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
443
986e461dc072 Initial revision
glantau
parents:
diff changeset
444 /* B C
986e461dc072 Initial revision
glantau
parents:
diff changeset
445 * A X
986e461dc072 Initial revision
glantau
parents:
diff changeset
446 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
447 a = dc_val[(x - 1) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
448 b = dc_val[(x - 1) + (y - 1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
449 c = dc_val[(x) + (y - 1) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
450
986e461dc072 Initial revision
glantau
parents:
diff changeset
451 if (abs(a - b) < abs(b - c)) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
452 pred = c;
986e461dc072 Initial revision
glantau
parents:
diff changeset
453 *dir_ptr = 1; /* top */
986e461dc072 Initial revision
glantau
parents:
diff changeset
454 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
455 pred = a;
986e461dc072 Initial revision
glantau
parents:
diff changeset
456 *dir_ptr = 0; /* left */
986e461dc072 Initial revision
glantau
parents:
diff changeset
457 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
458 /* we assume pred is positive */
986e461dc072 Initial revision
glantau
parents:
diff changeset
459 pred = (pred + (scale >> 1)) / scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
460
986e461dc072 Initial revision
glantau
parents:
diff changeset
461 /* prepare address for prediction update */
986e461dc072 Initial revision
glantau
parents:
diff changeset
462 *dc_val_ptr = &dc_val[(x) + (y) * wrap];
986e461dc072 Initial revision
glantau
parents:
diff changeset
463
986e461dc072 Initial revision
glantau
parents:
diff changeset
464 return pred;
986e461dc072 Initial revision
glantau
parents:
diff changeset
465 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
466
986e461dc072 Initial revision
glantau
parents:
diff changeset
467 void mpeg4_pred_ac(MpegEncContext * s, INT16 *block, int n,
986e461dc072 Initial revision
glantau
parents:
diff changeset
468 int dir)
986e461dc072 Initial revision
glantau
parents:
diff changeset
469 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
470 int x, y, wrap, i;
986e461dc072 Initial revision
glantau
parents:
diff changeset
471 INT16 *ac_val, *ac_val1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
472
986e461dc072 Initial revision
glantau
parents:
diff changeset
473 /* find prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
474 if (n < 4) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
475 x = 2 * s->mb_x + 1 + (n & 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
476 y = 2 * s->mb_y + 1 + ((n & 2) >> 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
477 wrap = s->mb_width * 2 + 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
478 ac_val = s->ac_val[0][0];
986e461dc072 Initial revision
glantau
parents:
diff changeset
479 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
480 x = s->mb_x + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
481 y = s->mb_y + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
482 wrap = s->mb_width + 2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
483 ac_val = s->ac_val[n - 4 + 1][0];
986e461dc072 Initial revision
glantau
parents:
diff changeset
484 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
485 ac_val += ((y) * wrap + (x)) * 16;
986e461dc072 Initial revision
glantau
parents:
diff changeset
486 ac_val1 = ac_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
487 if (s->ac_pred) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
488 if (dir == 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
489 /* left prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
490 ac_val -= 16;
986e461dc072 Initial revision
glantau
parents:
diff changeset
491 for(i=1;i<8;i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
492 block[i*8] += ac_val[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
493 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
494 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
495 /* top prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
496 ac_val -= 16 * wrap;
986e461dc072 Initial revision
glantau
parents:
diff changeset
497 for(i=1;i<8;i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
498 block[i] += ac_val[i + 8];
986e461dc072 Initial revision
glantau
parents:
diff changeset
499 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
500 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
501 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
502 /* left copy */
986e461dc072 Initial revision
glantau
parents:
diff changeset
503 for(i=1;i<8;i++)
986e461dc072 Initial revision
glantau
parents:
diff changeset
504 ac_val1[i] = block[i * 8];
986e461dc072 Initial revision
glantau
parents:
diff changeset
505 /* top copy */
986e461dc072 Initial revision
glantau
parents:
diff changeset
506 for(i=1;i<8;i++)
986e461dc072 Initial revision
glantau
parents:
diff changeset
507 ac_val1[8 + i] = block[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
508 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
509
986e461dc072 Initial revision
glantau
parents:
diff changeset
510 static inline void mpeg4_encode_dc(MpegEncContext * s, int level, int n, int *dir_ptr)
986e461dc072 Initial revision
glantau
parents:
diff changeset
511 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
512 int size, v, pred;
986e461dc072 Initial revision
glantau
parents:
diff changeset
513 UINT16 *dc_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
514
986e461dc072 Initial revision
glantau
parents:
diff changeset
515 pred = mpeg4_pred_dc(s, n, &dc_val, dir_ptr);
986e461dc072 Initial revision
glantau
parents:
diff changeset
516 if (n < 4) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
517 *dc_val = level * s->y_dc_scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
518 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
519 *dc_val = level * s->c_dc_scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
520 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
521
986e461dc072 Initial revision
glantau
parents:
diff changeset
522 /* do the prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
523 level -= pred;
986e461dc072 Initial revision
glantau
parents:
diff changeset
524 /* find number of bits */
986e461dc072 Initial revision
glantau
parents:
diff changeset
525 size = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
526 v = abs(level);
986e461dc072 Initial revision
glantau
parents:
diff changeset
527 while (v) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
528 v >>= 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
529 size++;
986e461dc072 Initial revision
glantau
parents:
diff changeset
530 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
531
986e461dc072 Initial revision
glantau
parents:
diff changeset
532 if (n < 4) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
533 /* luminance */
986e461dc072 Initial revision
glantau
parents:
diff changeset
534 put_bits(&s->pb, DCtab_lum[size][1], DCtab_lum[size][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
535 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
536 /* chrominance */
986e461dc072 Initial revision
glantau
parents:
diff changeset
537 put_bits(&s->pb, DCtab_chrom[size][1], DCtab_chrom[size][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
538 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
539
986e461dc072 Initial revision
glantau
parents:
diff changeset
540 /* encode remaining bits */
986e461dc072 Initial revision
glantau
parents:
diff changeset
541 if (size > 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
542 if (level < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
543 level = (-level) ^ ((1 << size) - 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
544 put_bits(&s->pb, size, level);
986e461dc072 Initial revision
glantau
parents:
diff changeset
545 if (size > 8)
986e461dc072 Initial revision
glantau
parents:
diff changeset
546 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
547 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
548 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
549
986e461dc072 Initial revision
glantau
parents:
diff changeset
550 static void mpeg4_encode_block(MpegEncContext * s, DCTELEM * block, int n)
986e461dc072 Initial revision
glantau
parents:
diff changeset
551 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
552 int level, run, last, i, j, last_index, last_non_zero, sign, slevel;
986e461dc072 Initial revision
glantau
parents:
diff changeset
553 int code, dc_pred_dir;
986e461dc072 Initial revision
glantau
parents:
diff changeset
554 const RLTable *rl;
986e461dc072 Initial revision
glantau
parents:
diff changeset
555
986e461dc072 Initial revision
glantau
parents:
diff changeset
556 if (s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
557 /* mpeg4 based DC predictor */
986e461dc072 Initial revision
glantau
parents:
diff changeset
558 mpeg4_encode_dc(s, block[0], n, &dc_pred_dir);
986e461dc072 Initial revision
glantau
parents:
diff changeset
559 i = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
560 rl = &rl_intra;
986e461dc072 Initial revision
glantau
parents:
diff changeset
561 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
562 i = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
563 rl = &rl_inter;
986e461dc072 Initial revision
glantau
parents:
diff changeset
564 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
565
986e461dc072 Initial revision
glantau
parents:
diff changeset
566 /* AC coefs */
986e461dc072 Initial revision
glantau
parents:
diff changeset
567 last_index = s->block_last_index[n];
986e461dc072 Initial revision
glantau
parents:
diff changeset
568 last_non_zero = i - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
569 for (; i <= last_index; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
570 j = zigzag_direct[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
571 level = block[j];
986e461dc072 Initial revision
glantau
parents:
diff changeset
572 if (level) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
573 run = i - last_non_zero - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
574 last = (i == last_index);
986e461dc072 Initial revision
glantau
parents:
diff changeset
575 sign = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
576 slevel = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
577 if (level < 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
578 sign = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
579 level = -level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
580 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
581 code = get_rl_index(rl, last, run, level);
986e461dc072 Initial revision
glantau
parents:
diff changeset
582 put_bits(&s->pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
583 if (code == rl->n) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
584 int level1, run1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
585 level1 = level - rl->max_level[last][run];
986e461dc072 Initial revision
glantau
parents:
diff changeset
586 if (level1 < 1)
986e461dc072 Initial revision
glantau
parents:
diff changeset
587 goto esc2;
986e461dc072 Initial revision
glantau
parents:
diff changeset
588 code = get_rl_index(rl, last, run, level1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
589 if (code == rl->n) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
590 esc2:
986e461dc072 Initial revision
glantau
parents:
diff changeset
591 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
592 if (level > MAX_LEVEL)
986e461dc072 Initial revision
glantau
parents:
diff changeset
593 goto esc3;
986e461dc072 Initial revision
glantau
parents:
diff changeset
594 run1 = run - rl->max_run[last][level] - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
595 if (run1 < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
596 goto esc3;
986e461dc072 Initial revision
glantau
parents:
diff changeset
597 code = get_rl_index(rl, last, run1, level);
986e461dc072 Initial revision
glantau
parents:
diff changeset
598 if (code == rl->n) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
599 esc3:
986e461dc072 Initial revision
glantau
parents:
diff changeset
600 /* third escape */
986e461dc072 Initial revision
glantau
parents:
diff changeset
601 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
602 put_bits(&s->pb, 1, last);
986e461dc072 Initial revision
glantau
parents:
diff changeset
603 put_bits(&s->pb, 6, run);
986e461dc072 Initial revision
glantau
parents:
diff changeset
604 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
605 put_bits(&s->pb, 12, slevel & 0xfff);
986e461dc072 Initial revision
glantau
parents:
diff changeset
606 put_bits(&s->pb, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
607 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
608 /* second escape */
986e461dc072 Initial revision
glantau
parents:
diff changeset
609 put_bits(&s->pb, 1, 0);
986e461dc072 Initial revision
glantau
parents:
diff changeset
610 put_bits(&s->pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
611 put_bits(&s->pb, 1, sign);
986e461dc072 Initial revision
glantau
parents:
diff changeset
612 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
613 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
614 /* first escape */
986e461dc072 Initial revision
glantau
parents:
diff changeset
615 put_bits(&s->pb, 1, 0);
986e461dc072 Initial revision
glantau
parents:
diff changeset
616 put_bits(&s->pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
986e461dc072 Initial revision
glantau
parents:
diff changeset
617 put_bits(&s->pb, 1, sign);
986e461dc072 Initial revision
glantau
parents:
diff changeset
618 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
619 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
620 put_bits(&s->pb, 1, sign);
986e461dc072 Initial revision
glantau
parents:
diff changeset
621 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
622 last_non_zero = i;
986e461dc072 Initial revision
glantau
parents:
diff changeset
623 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
624 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
625 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
626
986e461dc072 Initial revision
glantau
parents:
diff changeset
627
986e461dc072 Initial revision
glantau
parents:
diff changeset
628
986e461dc072 Initial revision
glantau
parents:
diff changeset
629 /***********************************************/
986e461dc072 Initial revision
glantau
parents:
diff changeset
630 /* decoding */
986e461dc072 Initial revision
glantau
parents:
diff changeset
631
986e461dc072 Initial revision
glantau
parents:
diff changeset
632 static VLC intra_MCBPC_vlc;
986e461dc072 Initial revision
glantau
parents:
diff changeset
633 static VLC inter_MCBPC_vlc;
986e461dc072 Initial revision
glantau
parents:
diff changeset
634 static VLC cbpy_vlc;
986e461dc072 Initial revision
glantau
parents:
diff changeset
635 static VLC mv_vlc;
986e461dc072 Initial revision
glantau
parents:
diff changeset
636 static VLC dc_lum, dc_chrom;
986e461dc072 Initial revision
glantau
parents:
diff changeset
637
986e461dc072 Initial revision
glantau
parents:
diff changeset
638 void init_rl(RLTable *rl)
986e461dc072 Initial revision
glantau
parents:
diff changeset
639 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
640 INT8 max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
986e461dc072 Initial revision
glantau
parents:
diff changeset
641 UINT8 index_run[MAX_RUN+1];
986e461dc072 Initial revision
glantau
parents:
diff changeset
642 int last, run, level, start, end, i;
986e461dc072 Initial revision
glantau
parents:
diff changeset
643
986e461dc072 Initial revision
glantau
parents:
diff changeset
644 /* compute max_level[], max_run[] and index_run[] */
986e461dc072 Initial revision
glantau
parents:
diff changeset
645 for(last=0;last<2;last++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
646 if (last == 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
647 start = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
648 end = rl->last;
986e461dc072 Initial revision
glantau
parents:
diff changeset
649 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
650 start = rl->last;
986e461dc072 Initial revision
glantau
parents:
diff changeset
651 end = rl->n;
986e461dc072 Initial revision
glantau
parents:
diff changeset
652 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
653
986e461dc072 Initial revision
glantau
parents:
diff changeset
654 memset(max_level, 0, MAX_RUN + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
655 memset(max_run, 0, MAX_LEVEL + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
656 memset(index_run, rl->n, MAX_RUN + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
657 for(i=start;i<end;i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
658 run = rl->table_run[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
659 level = rl->table_level[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
660 if (index_run[run] == rl->n)
986e461dc072 Initial revision
glantau
parents:
diff changeset
661 index_run[run] = i;
986e461dc072 Initial revision
glantau
parents:
diff changeset
662 if (level > max_level[run])
986e461dc072 Initial revision
glantau
parents:
diff changeset
663 max_level[run] = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
664 if (run > max_run[level])
986e461dc072 Initial revision
glantau
parents:
diff changeset
665 max_run[level] = run;
986e461dc072 Initial revision
glantau
parents:
diff changeset
666 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
667 rl->max_level[last] = malloc(MAX_RUN + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
668 memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
669 rl->max_run[last] = malloc(MAX_LEVEL + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
670 memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
671 rl->index_run[last] = malloc(MAX_RUN + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
672 memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
673 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
674 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
675
986e461dc072 Initial revision
glantau
parents:
diff changeset
676 void init_vlc_rl(RLTable *rl)
986e461dc072 Initial revision
glantau
parents:
diff changeset
677 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
678 init_vlc(&rl->vlc, 9, rl->n + 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
679 &rl->table_vlc[0][1], 4, 2,
986e461dc072 Initial revision
glantau
parents:
diff changeset
680 &rl->table_vlc[0][0], 4, 2);
986e461dc072 Initial revision
glantau
parents:
diff changeset
681 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
682
986e461dc072 Initial revision
glantau
parents:
diff changeset
683 /* init vlcs */
986e461dc072 Initial revision
glantau
parents:
diff changeset
684
986e461dc072 Initial revision
glantau
parents:
diff changeset
685 /* XXX: find a better solution to handle static init */
986e461dc072 Initial revision
glantau
parents:
diff changeset
686 void h263_decode_init_vlc(MpegEncContext *s)
986e461dc072 Initial revision
glantau
parents:
diff changeset
687 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
688 static int done = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
689
986e461dc072 Initial revision
glantau
parents:
diff changeset
690 if (!done) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
691 done = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
692
986e461dc072 Initial revision
glantau
parents:
diff changeset
693 init_vlc(&intra_MCBPC_vlc, 6, 8,
986e461dc072 Initial revision
glantau
parents:
diff changeset
694 intra_MCBPC_bits, 1, 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
695 intra_MCBPC_code, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
696 init_vlc(&inter_MCBPC_vlc, 9, 20,
986e461dc072 Initial revision
glantau
parents:
diff changeset
697 inter_MCBPC_bits, 1, 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
698 inter_MCBPC_code, 1, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
699 init_vlc(&cbpy_vlc, 6, 16,
986e461dc072 Initial revision
glantau
parents:
diff changeset
700 &cbpy_tab[0][1], 2, 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
701 &cbpy_tab[0][0], 2, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
702 init_vlc(&mv_vlc, 9, 33,
986e461dc072 Initial revision
glantau
parents:
diff changeset
703 &mvtab[0][1], 2, 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
704 &mvtab[0][0], 2, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
705 init_rl(&rl_inter);
986e461dc072 Initial revision
glantau
parents:
diff changeset
706 init_rl(&rl_intra);
986e461dc072 Initial revision
glantau
parents:
diff changeset
707 init_vlc_rl(&rl_inter);
986e461dc072 Initial revision
glantau
parents:
diff changeset
708 init_vlc_rl(&rl_intra);
986e461dc072 Initial revision
glantau
parents:
diff changeset
709 init_vlc(&dc_lum, 9, 13,
986e461dc072 Initial revision
glantau
parents:
diff changeset
710 &DCtab_lum[0][1], 2, 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
711 &DCtab_lum[0][0], 2, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
712 init_vlc(&dc_chrom, 9, 13,
986e461dc072 Initial revision
glantau
parents:
diff changeset
713 &DCtab_chrom[0][1], 2, 1,
986e461dc072 Initial revision
glantau
parents:
diff changeset
714 &DCtab_chrom[0][0], 2, 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
715 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
716 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
717
986e461dc072 Initial revision
glantau
parents:
diff changeset
718 int h263_decode_mb(MpegEncContext *s,
986e461dc072 Initial revision
glantau
parents:
diff changeset
719 DCTELEM block[6][64])
986e461dc072 Initial revision
glantau
parents:
diff changeset
720 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
721 int cbpc, cbpy, i, cbp, pred_x, pred_y, mx, my, dquant;
986e461dc072 Initial revision
glantau
parents:
diff changeset
722 INT16 *mot_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
723 static UINT8 quant_tab[4] = { -1, -2, 1, 2 };
986e461dc072 Initial revision
glantau
parents:
diff changeset
724
986e461dc072 Initial revision
glantau
parents:
diff changeset
725 if (s->pict_type == P_TYPE) {
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
726 if (get_bits1(&s->gb)) {
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
727 /* skip mb */
986e461dc072 Initial revision
glantau
parents:
diff changeset
728 s->mb_intra = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
729 for(i=0;i<6;i++)
986e461dc072 Initial revision
glantau
parents:
diff changeset
730 s->block_last_index[i] = -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
731 s->mv_dir = MV_DIR_FORWARD;
986e461dc072 Initial revision
glantau
parents:
diff changeset
732 s->mv_type = MV_TYPE_16X16;
986e461dc072 Initial revision
glantau
parents:
diff changeset
733 s->mv[0][0][0] = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
734 s->mv[0][0][1] = 0;
7
1d3ac9654178 added skip macroblock optimization (big perf win on black regions for example)
glantau
parents: 0
diff changeset
735 s->mb_skiped = 1;
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
736 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
737 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
738 cbpc = get_vlc(&s->gb, &inter_MCBPC_vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
739 if (cbpc < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
740 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
741 dquant = cbpc & 8;
986e461dc072 Initial revision
glantau
parents:
diff changeset
742 s->mb_intra = ((cbpc & 4) != 0);
986e461dc072 Initial revision
glantau
parents:
diff changeset
743 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
744 cbpc = get_vlc(&s->gb, &intra_MCBPC_vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
745 if (cbpc < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
746 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
747 dquant = cbpc & 4;
986e461dc072 Initial revision
glantau
parents:
diff changeset
748 s->mb_intra = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
749 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
750
986e461dc072 Initial revision
glantau
parents:
diff changeset
751 if (!s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
752 cbpy = get_vlc(&s->gb, &cbpy_vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
753 cbp = (cbpc & 3) | ((cbpy ^ 0xf) << 2);
986e461dc072 Initial revision
glantau
parents:
diff changeset
754 if (dquant) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
755 s->qscale += quant_tab[get_bits(&s->gb, 2)];
986e461dc072 Initial revision
glantau
parents:
diff changeset
756 if (s->qscale < 1)
986e461dc072 Initial revision
glantau
parents:
diff changeset
757 s->qscale = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
758 else if (s->qscale > 31)
986e461dc072 Initial revision
glantau
parents:
diff changeset
759 s->qscale = 31;
986e461dc072 Initial revision
glantau
parents:
diff changeset
760 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
761 s->mv_dir = MV_DIR_FORWARD;
986e461dc072 Initial revision
glantau
parents:
diff changeset
762 if ((cbpc & 16) == 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
763 /* 16x16 motion prediction */
986e461dc072 Initial revision
glantau
parents:
diff changeset
764 s->mv_type = MV_TYPE_16X16;
986e461dc072 Initial revision
glantau
parents:
diff changeset
765 h263_pred_motion(s, 0, &pred_x, &pred_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
766 mx = h263_decode_motion(s, pred_x);
986e461dc072 Initial revision
glantau
parents:
diff changeset
767 if (mx >= 0xffff)
986e461dc072 Initial revision
glantau
parents:
diff changeset
768 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
769 my = h263_decode_motion(s, pred_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
770 if (my >= 0xffff)
986e461dc072 Initial revision
glantau
parents:
diff changeset
771 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
772 s->mv[0][0][0] = mx;
986e461dc072 Initial revision
glantau
parents:
diff changeset
773 s->mv[0][0][1] = my;
986e461dc072 Initial revision
glantau
parents:
diff changeset
774 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
775 s->mv_type = MV_TYPE_8X8;
986e461dc072 Initial revision
glantau
parents:
diff changeset
776 for(i=0;i<4;i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
777 mot_val = h263_pred_motion(s, i, &pred_x, &pred_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
778 mx = h263_decode_motion(s, pred_x);
986e461dc072 Initial revision
glantau
parents:
diff changeset
779 if (mx >= 0xffff)
986e461dc072 Initial revision
glantau
parents:
diff changeset
780 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
781 my = h263_decode_motion(s, pred_y);
986e461dc072 Initial revision
glantau
parents:
diff changeset
782 if (my >= 0xffff)
986e461dc072 Initial revision
glantau
parents:
diff changeset
783 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
784 s->mv[0][i][0] = mx;
986e461dc072 Initial revision
glantau
parents:
diff changeset
785 s->mv[0][i][1] = my;
986e461dc072 Initial revision
glantau
parents:
diff changeset
786 mot_val[0] = mx;
986e461dc072 Initial revision
glantau
parents:
diff changeset
787 mot_val[1] = my;
986e461dc072 Initial revision
glantau
parents:
diff changeset
788 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
789 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
790 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
791 s->ac_pred = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
792 if (s->h263_pred) {
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
793 s->ac_pred = get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
794 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
795 cbpy = get_vlc(&s->gb, &cbpy_vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
796 cbp = (cbpc & 3) | (cbpy << 2);
986e461dc072 Initial revision
glantau
parents:
diff changeset
797 if (dquant) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
798 s->qscale += quant_tab[get_bits(&s->gb, 2)];
986e461dc072 Initial revision
glantau
parents:
diff changeset
799 if (s->qscale < 1)
986e461dc072 Initial revision
glantau
parents:
diff changeset
800 s->qscale = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
801 else if (s->qscale > 31)
986e461dc072 Initial revision
glantau
parents:
diff changeset
802 s->qscale = 31;
986e461dc072 Initial revision
glantau
parents:
diff changeset
803 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
804 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
805
986e461dc072 Initial revision
glantau
parents:
diff changeset
806 /* decode each block */
986e461dc072 Initial revision
glantau
parents:
diff changeset
807 if (s->h263_pred) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
808 for (i = 0; i < 6; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
809 if (mpeg4_decode_block(s, block[i], i, (cbp >> (5 - i)) & 1) < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
810 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
811 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
812 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
813 for (i = 0; i < 6; i++) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
814 if (h263_decode_block(s, block[i], i, (cbp >> (5 - i)) & 1) < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
815 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
816 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
817 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
818 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
819 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
820
986e461dc072 Initial revision
glantau
parents:
diff changeset
821 static int h263_decode_motion(MpegEncContext * s, int pred)
986e461dc072 Initial revision
glantau
parents:
diff changeset
822 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
823 int code, val, sign, shift, l, m;
986e461dc072 Initial revision
glantau
parents:
diff changeset
824
986e461dc072 Initial revision
glantau
parents:
diff changeset
825 code = get_vlc(&s->gb, &mv_vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
826 if (code < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
827 return 0xffff;
986e461dc072 Initial revision
glantau
parents:
diff changeset
828
986e461dc072 Initial revision
glantau
parents:
diff changeset
829 if (code == 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
830 return pred;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
831 sign = get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
832 shift = s->f_code - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
833 val = (code - 1) << shift;
986e461dc072 Initial revision
glantau
parents:
diff changeset
834 if (shift > 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
835 val |= get_bits(&s->gb, shift);
986e461dc072 Initial revision
glantau
parents:
diff changeset
836 val++;
986e461dc072 Initial revision
glantau
parents:
diff changeset
837 if (sign)
986e461dc072 Initial revision
glantau
parents:
diff changeset
838 val = -val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
839 val += pred;
986e461dc072 Initial revision
glantau
parents:
diff changeset
840
986e461dc072 Initial revision
glantau
parents:
diff changeset
841 /* modulo decoding */
986e461dc072 Initial revision
glantau
parents:
diff changeset
842 if (!s->h263_long_vectors) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
843 l = (1 << (s->f_code - 1)) * 32;
986e461dc072 Initial revision
glantau
parents:
diff changeset
844 m = 2 * l;
986e461dc072 Initial revision
glantau
parents:
diff changeset
845 if (val < -l) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
846 val += m;
986e461dc072 Initial revision
glantau
parents:
diff changeset
847 } else if (val >= l) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
848 val -= m;
986e461dc072 Initial revision
glantau
parents:
diff changeset
849 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
850 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
851 /* horrible h263 long vector mode */
986e461dc072 Initial revision
glantau
parents:
diff changeset
852 if (pred < -31 && val < -63)
986e461dc072 Initial revision
glantau
parents:
diff changeset
853 val += 64;
986e461dc072 Initial revision
glantau
parents:
diff changeset
854 if (pred > 32 && val > 63)
986e461dc072 Initial revision
glantau
parents:
diff changeset
855 val -= 64;
986e461dc072 Initial revision
glantau
parents:
diff changeset
856 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
857 return val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
858 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
859
986e461dc072 Initial revision
glantau
parents:
diff changeset
860 static int h263_decode_block(MpegEncContext * s, DCTELEM * block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
861 int n, int coded)
986e461dc072 Initial revision
glantau
parents:
diff changeset
862 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
863 int code, level, i, j, last, run;
986e461dc072 Initial revision
glantau
parents:
diff changeset
864 RLTable *rl = &rl_inter;
986e461dc072 Initial revision
glantau
parents:
diff changeset
865
986e461dc072 Initial revision
glantau
parents:
diff changeset
866 if (s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
867 /* DC coef */
986e461dc072 Initial revision
glantau
parents:
diff changeset
868 if (s->h263_rv10 && s->rv10_version == 3 && s->pict_type == I_TYPE) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
869 int component, diff;
986e461dc072 Initial revision
glantau
parents:
diff changeset
870 component = (n <= 3 ? 0 : n - 4 + 1);
986e461dc072 Initial revision
glantau
parents:
diff changeset
871 level = s->last_dc[component];
986e461dc072 Initial revision
glantau
parents:
diff changeset
872 if (s->rv10_first_dc_coded[component]) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
873 diff = rv_decode_dc(s, n);
986e461dc072 Initial revision
glantau
parents:
diff changeset
874 if (diff == 0xffff)
986e461dc072 Initial revision
glantau
parents:
diff changeset
875 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
876 level += diff;
986e461dc072 Initial revision
glantau
parents:
diff changeset
877 level = level & 0xff; /* handle wrap round */
986e461dc072 Initial revision
glantau
parents:
diff changeset
878 s->last_dc[component] = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
879 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
880 s->rv10_first_dc_coded[component] = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
881 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
882 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
883 level = get_bits(&s->gb, 8);
986e461dc072 Initial revision
glantau
parents:
diff changeset
884 if (level == 255)
986e461dc072 Initial revision
glantau
parents:
diff changeset
885 level = 128;
986e461dc072 Initial revision
glantau
parents:
diff changeset
886 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
887 block[0] = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
888 i = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
889 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
890 i = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
891 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
892 if (!coded) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
893 s->block_last_index[n] = i - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
894 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
895 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
896
986e461dc072 Initial revision
glantau
parents:
diff changeset
897 for(;;) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
898 code = get_vlc(&s->gb, &rl->vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
899 if (code < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
900 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
901 if (code == rl->n) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
902 /* escape */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
903 last = get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
904 run = get_bits(&s->gb, 6);
986e461dc072 Initial revision
glantau
parents:
diff changeset
905 level = (INT8)get_bits(&s->gb, 8);
986e461dc072 Initial revision
glantau
parents:
diff changeset
906 if (s->h263_rv10 && level == -128) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
907 /* XXX: should patch encoder too */
986e461dc072 Initial revision
glantau
parents:
diff changeset
908 level = get_bits(&s->gb, 12);
986e461dc072 Initial revision
glantau
parents:
diff changeset
909 level = (level << 20) >> 20;
986e461dc072 Initial revision
glantau
parents:
diff changeset
910 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
911 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
912 run = rl->table_run[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
913 level = rl->table_level[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
914 last = code >= rl->last;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
915 if (get_bits1(&s->gb))
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
916 level = -level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
917 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
918 i += run;
986e461dc072 Initial revision
glantau
parents:
diff changeset
919 if (i >= 64)
986e461dc072 Initial revision
glantau
parents:
diff changeset
920 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
921 j = zigzag_direct[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
922 block[j] = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
923 if (last)
986e461dc072 Initial revision
glantau
parents:
diff changeset
924 break;
986e461dc072 Initial revision
glantau
parents:
diff changeset
925 i++;
986e461dc072 Initial revision
glantau
parents:
diff changeset
926 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
927 s->block_last_index[n] = i;
986e461dc072 Initial revision
glantau
parents:
diff changeset
928 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
929 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
930
986e461dc072 Initial revision
glantau
parents:
diff changeset
931 static int mpeg4_decode_dc(MpegEncContext * s, int n, int *dir_ptr)
986e461dc072 Initial revision
glantau
parents:
diff changeset
932 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
933 int level, pred, code;
986e461dc072 Initial revision
glantau
parents:
diff changeset
934 UINT16 *dc_val;
986e461dc072 Initial revision
glantau
parents:
diff changeset
935
986e461dc072 Initial revision
glantau
parents:
diff changeset
936 if (n < 4)
986e461dc072 Initial revision
glantau
parents:
diff changeset
937 code = get_vlc(&s->gb, &dc_lum);
986e461dc072 Initial revision
glantau
parents:
diff changeset
938 else
986e461dc072 Initial revision
glantau
parents:
diff changeset
939 code = get_vlc(&s->gb, &dc_chrom);
986e461dc072 Initial revision
glantau
parents:
diff changeset
940 if (code < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
941 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
942 if (code == 0) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
943 level = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
944 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
945 level = get_bits(&s->gb, code);
986e461dc072 Initial revision
glantau
parents:
diff changeset
946 if ((level >> (code - 1)) == 0) /* if MSB not set it is negative*/
986e461dc072 Initial revision
glantau
parents:
diff changeset
947 level = - (level ^ ((1 << code) - 1));
986e461dc072 Initial revision
glantau
parents:
diff changeset
948 if (code > 8)
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
949 skip_bits1(&s->gb); /* marker */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
950 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
951
986e461dc072 Initial revision
glantau
parents:
diff changeset
952 pred = mpeg4_pred_dc(s, n, &dc_val, dir_ptr);
986e461dc072 Initial revision
glantau
parents:
diff changeset
953 level += pred;
986e461dc072 Initial revision
glantau
parents:
diff changeset
954 if (level < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
955 level = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
956 if (n < 4) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
957 *dc_val = level * s->y_dc_scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
958 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
959 *dc_val = level * s->c_dc_scale;
986e461dc072 Initial revision
glantau
parents:
diff changeset
960 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
961 return level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
962 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
963
986e461dc072 Initial revision
glantau
parents:
diff changeset
964 static int mpeg4_decode_block(MpegEncContext * s, DCTELEM * block,
986e461dc072 Initial revision
glantau
parents:
diff changeset
965 int n, int coded)
986e461dc072 Initial revision
glantau
parents:
diff changeset
966 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
967 int code, level, i, j, last, run;
986e461dc072 Initial revision
glantau
parents:
diff changeset
968 int dc_pred_dir;
986e461dc072 Initial revision
glantau
parents:
diff changeset
969 RLTable *rl;
986e461dc072 Initial revision
glantau
parents:
diff changeset
970 const UINT8 *scan_table;
986e461dc072 Initial revision
glantau
parents:
diff changeset
971
986e461dc072 Initial revision
glantau
parents:
diff changeset
972 if (s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
973 /* DC coef */
986e461dc072 Initial revision
glantau
parents:
diff changeset
974 level = mpeg4_decode_dc(s, n, &dc_pred_dir);
986e461dc072 Initial revision
glantau
parents:
diff changeset
975 if (level < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
976 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
977 block[0] = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
978 i = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
979 if (!coded)
986e461dc072 Initial revision
glantau
parents:
diff changeset
980 goto not_coded;
986e461dc072 Initial revision
glantau
parents:
diff changeset
981 rl = &rl_intra;
986e461dc072 Initial revision
glantau
parents:
diff changeset
982 if (s->ac_pred) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
983 if (dc_pred_dir == 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
984 scan_table = ff_alternate_vertical_scan; /* left */
986e461dc072 Initial revision
glantau
parents:
diff changeset
985 else
986e461dc072 Initial revision
glantau
parents:
diff changeset
986 scan_table = ff_alternate_horizontal_scan; /* top */
986e461dc072 Initial revision
glantau
parents:
diff changeset
987 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
988 scan_table = zigzag_direct;
986e461dc072 Initial revision
glantau
parents:
diff changeset
989 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
990 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
991 i = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
992 if (!coded) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
993 s->block_last_index[n] = i - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
994 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
995 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
996 rl = &rl_inter;
986e461dc072 Initial revision
glantau
parents:
diff changeset
997 scan_table = zigzag_direct;
986e461dc072 Initial revision
glantau
parents:
diff changeset
998 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
999
986e461dc072 Initial revision
glantau
parents:
diff changeset
1000 for(;;) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1001 code = get_vlc(&s->gb, &rl->vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1002 if (code < 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1003 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1004 if (code == rl->n) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1005 /* escape */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1006 if (get_bits1(&s->gb) != 0) {
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1007 if (get_bits1(&s->gb) != 0) {
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1008 /* third escape */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1009 last = get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1010 run = get_bits(&s->gb, 6);
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1011 get_bits1(&s->gb); /* marker */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1012 level = get_bits(&s->gb, 12);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1013 level = (level << 20) >> 20; /* sign extend */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1014 skip_bits1(&s->gb); /* marker */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1015 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1016 /* second escape */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1017 code = get_vlc(&s->gb, &rl->vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1018 if (code < 0 || code >= rl->n)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1019 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1020 run = rl->table_run[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1021 level = rl->table_level[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1022 last = code >= rl->last;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1023 run += rl->max_run[last][level] + 1;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1024 if (get_bits1(&s->gb))
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1025 level = -level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1026 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1027 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1028 /* first escape */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1029 code = get_vlc(&s->gb, &rl->vlc);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1030 if (code < 0 || code >= rl->n)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1031 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1032 run = rl->table_run[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1033 level = rl->table_level[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1034 last = code >= rl->last;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1035 level += rl->max_level[last][run];
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1036 if (get_bits1(&s->gb))
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1037 level = -level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1038 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1039 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1040 run = rl->table_run[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1041 level = rl->table_level[code];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1042 last = code >= rl->last;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1043 if (get_bits1(&s->gb))
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1044 level = -level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1045 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1046 i += run;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1047 if (i >= 64)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1048 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1049 j = scan_table[i];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1050 block[j] = level;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1051 i++;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1052 if (last)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1053 break;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1054 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1055 not_coded:
986e461dc072 Initial revision
glantau
parents:
diff changeset
1056 if (s->mb_intra) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1057 mpeg4_pred_ac(s, block, n, dc_pred_dir);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1058 if (s->ac_pred) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1059 i = 64; /* XXX: not optimal */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1060 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1061 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1062 s->block_last_index[n] = i - 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1063 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1064 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1065
986e461dc072 Initial revision
glantau
parents:
diff changeset
1066 /* most is hardcoded. should extend to handle all h263 streams */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1067 int h263_decode_picture_header(MpegEncContext *s)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1068 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1069 int format, width, height;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1070
986e461dc072 Initial revision
glantau
parents:
diff changeset
1071 /* picture header */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1072 if (get_bits(&s->gb, 22) != 0x20)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1073 return -1;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1074 skip_bits(&s->gb, 8); /* picture timestamp */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1075
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1076 if (get_bits1(&s->gb) != 1)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1077 return -1; /* marker */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1078 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1079 return -1; /* h263 id */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1080 skip_bits1(&s->gb); /* split screen off */
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1081 skip_bits1(&s->gb); /* camera off */
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1082 skip_bits1(&s->gb); /* freeze picture release off */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1083
986e461dc072 Initial revision
glantau
parents:
diff changeset
1084 format = get_bits(&s->gb, 3);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1085
986e461dc072 Initial revision
glantau
parents:
diff changeset
1086 if (format != 7) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1087 s->h263_plus = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1088 /* H.263v1 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1089 width = h263_format[format][0];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1090 height = h263_format[format][1];
986e461dc072 Initial revision
glantau
parents:
diff changeset
1091 if (!width)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1092 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1093
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1094 s->pict_type = I_TYPE + get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1095
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1096 s->unrestricted_mv = get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1097 s->h263_long_vectors = s->unrestricted_mv;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1098
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1099 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1100 return -1; /* SAC: off */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1101 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1102 return -1; /* advanced prediction mode: off */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1103 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1104 return -1; /* not PB frame */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1105
986e461dc072 Initial revision
glantau
parents:
diff changeset
1106 s->qscale = get_bits(&s->gb, 5);
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1107 skip_bits1(&s->gb); /* Continuous Presence Multipoint mode: off */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1108 } else {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1109 s->h263_plus = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1110 /* H.263v2 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1111 if (get_bits(&s->gb, 3) != 1)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1112 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1113 if (get_bits(&s->gb, 3) != 6) /* custom source format */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1114 return -1;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1115 skip_bits(&s->gb, 12);
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1116 skip_bits(&s->gb, 3);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1117 s->pict_type = get_bits(&s->gb, 3) + 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1118 if (s->pict_type != I_TYPE &&
986e461dc072 Initial revision
glantau
parents:
diff changeset
1119 s->pict_type != P_TYPE)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1120 return -1;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1121 skip_bits(&s->gb, 7);
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1122 skip_bits(&s->gb, 4); /* aspect ratio */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1123 width = (get_bits(&s->gb, 9) + 1) * 4;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1124 skip_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1125 height = get_bits(&s->gb, 9) * 4;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1126 if (height == 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1127 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1128 s->qscale = get_bits(&s->gb, 5);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1129 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1130 /* PEI */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1131 while (get_bits1(&s->gb) != 0) {
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1132 skip_bits(&s->gb, 8);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1133 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1134 s->f_code = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1135 s->width = width;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1136 s->height = height;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1137 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1138 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1139
986e461dc072 Initial revision
glantau
parents:
diff changeset
1140 /* decode mpeg4 VOP header */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1141 int mpeg4_decode_picture_header(MpegEncContext * s)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1142 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1143 int time_incr, startcode, state, v;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1144
986e461dc072 Initial revision
glantau
parents:
diff changeset
1145 redo:
986e461dc072 Initial revision
glantau
parents:
diff changeset
1146 /* search next start code */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1147 align_get_bits(&s->gb);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1148 state = 0xff;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1149 for(;;) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1150 v = get_bits(&s->gb, 8);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1151 if (state == 0x000001) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1152 state = ((state << 8) | v) & 0xffffff;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1153 startcode = state;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1154 break;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1155 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1156 state = ((state << 8) | v) & 0xffffff;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1157 /* XXX: really detect end of frame */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1158 if (state == 0)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1159 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1160 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1161
986e461dc072 Initial revision
glantau
parents:
diff changeset
1162 if (startcode == 0x120) {
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1163 int time_increment_resolution, width, height, vo_ver_id;
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1164
986e461dc072 Initial revision
glantau
parents:
diff changeset
1165 /* vol header */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1166 skip_bits(&s->gb, 1); /* random access */
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1167 skip_bits(&s->gb, 8); /* vo_type */
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1168 if (get_bits1(&s->gb) != 0) { /* is_ol_id */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1169 vo_ver_id = get_bits(&s->gb, 4); /* vo_ver_id */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1170 skip_bits(&s->gb, 3); /* vo_priority */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1171 } else {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1172 vo_ver_id = 1;
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1173 }
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1174
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1175 skip_bits(&s->gb, 4); /* aspect_ratio_info */
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1176 skip_bits1(&s->gb); /* vol control parameter */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1177 s->shape = get_bits(&s->gb, 2); /* vol shape */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1178 skip_bits1(&s->gb); /* marker */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1179
986e461dc072 Initial revision
glantau
parents:
diff changeset
1180 time_increment_resolution = get_bits(&s->gb, 16);
70
c7767df463f4 log2 to av_log2
glantau
parents: 63
diff changeset
1181 s->time_increment_bits = av_log2(time_increment_resolution - 1) + 1;
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1182 if (s->time_increment_bits < 1)
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1183 s->time_increment_bits = 1;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1184 skip_bits1(&s->gb); /* marker */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1185
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1186 if (get_bits1(&s->gb) != 0) { /* fixed_vop_rate */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1187 skip_bits(&s->gb, s->time_increment_bits);
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1188 }
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1189
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1190 if (s->shape != 2) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1191 if (s->shape == 0) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1192 skip_bits1(&s->gb); /* marker */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1193 width = get_bits(&s->gb, 13);
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1194 skip_bits1(&s->gb); /* marker */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1195 height = get_bits(&s->gb, 13);
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1196 skip_bits1(&s->gb); /* marker */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1197 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1198
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1199 skip_bits1(&s->gb); /* interlaced */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1200 skip_bits1(&s->gb); /* OBMC */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1201 if (vo_ver_id == 1) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1202 s->vol_sprite_usage = get_bits1(&s->gb); /* vol_sprite_usage */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1203 } else {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1204 s->vol_sprite_usage = get_bits(&s->gb, 2); /* vol_sprite_usage */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1205 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1206 if (get_bits1(&s->gb) == 1) { /* not_8_bit */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1207 s->quant_precision = get_bits(&s->gb, 4); /* quant_precision */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1208 skip_bits(&s->gb, 4); /* bits_per_pixel */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1209 } else {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1210 s->quant_precision = 5;
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1211 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1212
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1213 skip_bits1(&s->gb); /* vol_quant_type */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1214 skip_bits1(&s->gb); /* vol_quarter_pixel */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1215 skip_bits1(&s->gb); /* complexity_estimation_disabled */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1216 skip_bits1(&s->gb); /* resync_marker_disabled */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1217 skip_bits1(&s->gb); /* data_partioning_enabled */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1218 if (get_bits1(&s->gb) != 0) { /* scalability */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1219 printf("bad scalability!!!\n");
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1220 return -1;
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1221 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1222 }
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1223 goto redo;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1224 } else if (startcode != 0x1b6) {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1225 goto redo;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1226 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1227
986e461dc072 Initial revision
glantau
parents:
diff changeset
1228 s->pict_type = get_bits(&s->gb, 2) + 1; /* pict type: I = 0 , P = 1 */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1229 if (s->pict_type != I_TYPE &&
986e461dc072 Initial revision
glantau
parents:
diff changeset
1230 s->pict_type != P_TYPE)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1231 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1232
986e461dc072 Initial revision
glantau
parents:
diff changeset
1233 /* XXX: parse time base */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1234 time_incr = 0;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1235 while (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1236 time_incr++;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1237
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1238 skip_bits1(&s->gb); /* marker */
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1239 skip_bits(&s->gb, s->time_increment_bits);
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1240 skip_bits1(&s->gb); /* marker */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1241 /* vop coded */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1242 if (get_bits1(&s->gb) != 1)
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1243 goto redo;
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1244
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1245 if (s->shape != 2 && s->pict_type == P_TYPE) {
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1246 /* rounding type for motion estimation */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1247 s->no_rounding = get_bits1(&s->gb);
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1248 } else {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1249 s->no_rounding = 0;
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1250 }
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1251
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1252 if (s->shape != 0) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1253 if (s->vol_sprite_usage != 1 || s->pict_type != I_TYPE) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1254 int width, height, hor_spat_ref, ver_spat_ref;
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1255
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1256 width = get_bits(&s->gb, 13);
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1257 skip_bits1(&s->gb); /* marker */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1258 height = get_bits(&s->gb, 13);
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1259 skip_bits1(&s->gb); /* marker */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1260 hor_spat_ref = get_bits(&s->gb, 13); /* hor_spat_ref */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1261 skip_bits1(&s->gb); /* marker */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1262 ver_spat_ref = get_bits(&s->gb, 13); /* ver_spat_ref */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1263 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1264 skip_bits1(&s->gb); /* change_CR_disable */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1265
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1266 if (get_bits1(&s->gb) != 0) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1267 skip_bits(&s->gb, 8); /* constant_alpha_value */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1268 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1269 }
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1270
63
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1271 if (s->shape != 2) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1272 skip_bits(&s->gb, 3); /* intra dc VLC threshold */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1273
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1274 /* note: we do not use quant_precision to avoid problem if no
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1275 MPEG4 vol header as it is found on some old opendivx
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1276 movies */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1277 s->qscale = get_bits(&s->gb, 5);
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1278
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1279 if (s->pict_type != I_TYPE) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1280 s->f_code = get_bits(&s->gb, 3); /* fcode_for */
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1281 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1282 if (s->shape && (s->pict_type != I_TYPE)) {
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1283 skip_bits1(&s->gb); // vop shape coding type
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1284 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1285 }
8e2d8dbf89a5 better vol header parsing for mpeg4
glantau
parents: 21
diff changeset
1286 return 0;
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1287 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1288
986e461dc072 Initial revision
glantau
parents:
diff changeset
1289 /* don't understand why they choose a different header ! */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1290 int intel_h263_decode_picture_header(MpegEncContext *s)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1291 {
986e461dc072 Initial revision
glantau
parents:
diff changeset
1292 int format;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1293
986e461dc072 Initial revision
glantau
parents:
diff changeset
1294 /* picture header */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1295 if (get_bits(&s->gb, 22) != 0x20)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1296 return -1;
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1297 skip_bits(&s->gb, 8); /* picture timestamp */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1298
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1299 if (get_bits1(&s->gb) != 1)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1300 return -1; /* marker */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1301 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1302 return -1; /* h263 id */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1303 skip_bits1(&s->gb); /* split screen off */
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1304 skip_bits1(&s->gb); /* camera off */
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1305 skip_bits1(&s->gb); /* freeze picture release off */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1306
986e461dc072 Initial revision
glantau
parents:
diff changeset
1307 format = get_bits(&s->gb, 3);
986e461dc072 Initial revision
glantau
parents:
diff changeset
1308 if (format != 7)
986e461dc072 Initial revision
glantau
parents:
diff changeset
1309 return -1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1310
986e461dc072 Initial revision
glantau
parents:
diff changeset
1311 s->h263_plus = 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1312
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1313 s->pict_type = I_TYPE + get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1314
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1315 s->unrestricted_mv = get_bits1(&s->gb);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1316 s->h263_long_vectors = s->unrestricted_mv;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1317
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1318 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1319 return -1; /* SAC: off */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1320 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1321 return -1; /* advanced prediction mode: off */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1322 if (get_bits1(&s->gb) != 0)
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1323 return -1; /* not PB frame */
986e461dc072 Initial revision
glantau
parents:
diff changeset
1324
986e461dc072 Initial revision
glantau
parents:
diff changeset
1325 /* skip unknown header garbage */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1326 skip_bits(&s->gb, 41);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1327
986e461dc072 Initial revision
glantau
parents:
diff changeset
1328 s->qscale = get_bits(&s->gb, 5);
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1329 skip_bits1(&s->gb); /* Continuous Presence Multipoint mode: off */
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1330
986e461dc072 Initial revision
glantau
parents:
diff changeset
1331 /* PEI */
21
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1332 while (get_bits1(&s->gb) != 0) {
20e680e7a490 get_bits() specialization, gives 4\speedup
arpi_esp
parents: 7
diff changeset
1333 skip_bits(&s->gb, 8);
0
986e461dc072 Initial revision
glantau
parents:
diff changeset
1334 }
986e461dc072 Initial revision
glantau
parents:
diff changeset
1335 s->f_code = 1;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1336 return 0;
986e461dc072 Initial revision
glantau
parents:
diff changeset
1337 }