annotate h263.c @ 7:1d3ac9654178 libavcodec

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