annotate h263.c @ 110:e00e5d93457c libavcodec

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