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