00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022 #include "libavcore/imgutils.h"
00023 #include "avcodec.h"
00024 #include "dsputil.h"
00025 #include "binkdata.h"
00026 #include "mathops.h"
00027
00028 #define ALT_BITSTREAM_READER_LE
00029 #include "get_bits.h"
00030
00031 #define BINK_FLAG_ALPHA 0x00100000
00032 #define BINK_FLAG_GRAY 0x00020000
00033
00034 static VLC bink_trees[16];
00035
00039 enum Sources {
00040 BINK_SRC_BLOCK_TYPES = 0,
00041 BINK_SRC_SUB_BLOCK_TYPES,
00042 BINK_SRC_COLORS,
00043 BINK_SRC_PATTERN,
00044 BINK_SRC_X_OFF,
00045 BINK_SRC_Y_OFF,
00046 BINK_SRC_INTRA_DC,
00047 BINK_SRC_INTER_DC,
00048 BINK_SRC_RUN,
00049
00050 BINK_NB_SRC
00051 };
00052
00056 typedef struct Tree {
00057 int vlc_num;
00058 uint8_t syms[16];
00059 } Tree;
00060
00061 #define GET_HUFF(gb, tree) (tree).syms[get_vlc2(gb, bink_trees[(tree).vlc_num].table,\
00062 bink_trees[(tree).vlc_num].bits, 1)]
00063
00067 typedef struct Bundle {
00068 int len;
00069 Tree tree;
00070 uint8_t *data;
00071 uint8_t *data_end;
00072 uint8_t *cur_dec;
00073 uint8_t *cur_ptr;
00074 } Bundle;
00075
00076
00077
00078
00079 typedef struct BinkContext {
00080 AVCodecContext *avctx;
00081 DSPContext dsp;
00082 AVFrame pic, last;
00083 int version;
00084 int has_alpha;
00085 int swap_planes;
00086 ScanTable scantable;
00087
00088 Bundle bundle[BINK_NB_SRC];
00089 Tree col_high[16];
00090 int col_lastval;
00091 } BinkContext;
00092
00096 enum BlockTypes {
00097 SKIP_BLOCK = 0,
00098 SCALED_BLOCK,
00099 MOTION_BLOCK,
00100 RUN_BLOCK,
00101 RESIDUE_BLOCK,
00102 INTRA_BLOCK,
00103 FILL_BLOCK,
00104 INTER_BLOCK,
00105 PATTERN_BLOCK,
00106 RAW_BLOCK,
00107 };
00108
00116 static void init_lengths(BinkContext *c, int width, int bw)
00117 {
00118 c->bundle[BINK_SRC_BLOCK_TYPES].len = av_log2((width >> 3) + 511) + 1;
00119
00120 c->bundle[BINK_SRC_SUB_BLOCK_TYPES].len = av_log2((width >> 4) + 511) + 1;
00121
00122 c->bundle[BINK_SRC_COLORS].len = av_log2(bw*64 + 511) + 1;
00123
00124 c->bundle[BINK_SRC_INTRA_DC].len =
00125 c->bundle[BINK_SRC_INTER_DC].len =
00126 c->bundle[BINK_SRC_X_OFF].len =
00127 c->bundle[BINK_SRC_Y_OFF].len = av_log2((width >> 3) + 511) + 1;
00128
00129 c->bundle[BINK_SRC_PATTERN].len = av_log2((bw << 3) + 511) + 1;
00130
00131 c->bundle[BINK_SRC_RUN].len = av_log2(bw*48 + 511) + 1;
00132 }
00133
00139 static av_cold void init_bundles(BinkContext *c)
00140 {
00141 int bw, bh, blocks;
00142 int i;
00143
00144 bw = (c->avctx->width + 7) >> 3;
00145 bh = (c->avctx->height + 7) >> 3;
00146 blocks = bw * bh;
00147
00148 for (i = 0; i < BINK_NB_SRC; i++) {
00149 c->bundle[i].data = av_malloc(blocks * 64);
00150 c->bundle[i].data_end = c->bundle[i].data + blocks * 64;
00151 }
00152 }
00153
00159 static av_cold void free_bundles(BinkContext *c)
00160 {
00161 int i;
00162 for (i = 0; i < BINK_NB_SRC; i++)
00163 av_freep(&c->bundle[i].data);
00164 }
00165
00174 static void merge(GetBitContext *gb, uint8_t *dst, uint8_t *src, int size)
00175 {
00176 uint8_t *src2 = src + size;
00177 int size2 = size;
00178
00179 do {
00180 if (!get_bits1(gb)) {
00181 *dst++ = *src++;
00182 size--;
00183 } else {
00184 *dst++ = *src2++;
00185 size2--;
00186 }
00187 } while (size && size2);
00188
00189 while (size--)
00190 *dst++ = *src++;
00191 while (size2--)
00192 *dst++ = *src2++;
00193 }
00194
00201 static void read_tree(GetBitContext *gb, Tree *tree)
00202 {
00203 uint8_t tmp1[16], tmp2[16], *in = tmp1, *out = tmp2;
00204 int i, t, len;
00205
00206 tree->vlc_num = get_bits(gb, 4);
00207 if (!tree->vlc_num) {
00208 for (i = 0; i < 16; i++)
00209 tree->syms[i] = i;
00210 return;
00211 }
00212 if (get_bits1(gb)) {
00213 len = get_bits(gb, 3);
00214 memset(tmp1, 0, sizeof(tmp1));
00215 for (i = 0; i <= len; i++) {
00216 tree->syms[i] = get_bits(gb, 4);
00217 tmp1[tree->syms[i]] = 1;
00218 }
00219 for (i = 0; i < 16; i++)
00220 if (!tmp1[i])
00221 tree->syms[++len] = i;
00222 } else {
00223 len = get_bits(gb, 2);
00224 for (i = 0; i < 16; i++)
00225 in[i] = i;
00226 for (i = 0; i <= len; i++) {
00227 int size = 1 << i;
00228 for (t = 0; t < 16; t += size << 1)
00229 merge(gb, out + t, in + t, size);
00230 FFSWAP(uint8_t*, in, out);
00231 }
00232 memcpy(tree->syms, in, 16);
00233 }
00234 }
00235
00243 static void read_bundle(GetBitContext *gb, BinkContext *c, int bundle_num)
00244 {
00245 int i;
00246
00247 if (bundle_num == BINK_SRC_COLORS) {
00248 for (i = 0; i < 16; i++)
00249 read_tree(gb, &c->col_high[i]);
00250 c->col_lastval = 0;
00251 }
00252 if (bundle_num != BINK_SRC_INTRA_DC && bundle_num != BINK_SRC_INTER_DC)
00253 read_tree(gb, &c->bundle[bundle_num].tree);
00254 c->bundle[bundle_num].cur_dec =
00255 c->bundle[bundle_num].cur_ptr = c->bundle[bundle_num].data;
00256 }
00257
00265 #define CHECK_READ_VAL(gb, b, t) \
00266 if (!b->cur_dec || (b->cur_dec > b->cur_ptr)) \
00267 return 0; \
00268 t = get_bits(gb, b->len); \
00269 if (!t) { \
00270 b->cur_dec = NULL; \
00271 return 0; \
00272 } \
00273
00274 static int read_runs(AVCodecContext *avctx, GetBitContext *gb, Bundle *b)
00275 {
00276 int t, v;
00277 const uint8_t *dec_end;
00278
00279 CHECK_READ_VAL(gb, b, t);
00280 dec_end = b->cur_dec + t;
00281 if (dec_end > b->data_end) {
00282 av_log(avctx, AV_LOG_ERROR, "Run value went out of bounds\n");
00283 return -1;
00284 }
00285 if (get_bits1(gb)) {
00286 v = get_bits(gb, 4);
00287 memset(b->cur_dec, v, t);
00288 b->cur_dec += t;
00289 } else {
00290 while (b->cur_dec < dec_end)
00291 *b->cur_dec++ = GET_HUFF(gb, b->tree);
00292 }
00293 return 0;
00294 }
00295
00296 static int read_motion_values(AVCodecContext *avctx, GetBitContext *gb, Bundle *b)
00297 {
00298 int t, sign, v;
00299 const uint8_t *dec_end;
00300
00301 CHECK_READ_VAL(gb, b, t);
00302 dec_end = b->cur_dec + t;
00303 if (dec_end > b->data_end) {
00304 av_log(avctx, AV_LOG_ERROR, "Too many motion values\n");
00305 return -1;
00306 }
00307 if (get_bits1(gb)) {
00308 v = get_bits(gb, 4);
00309 if (v) {
00310 sign = -get_bits1(gb);
00311 v = (v ^ sign) - sign;
00312 }
00313 memset(b->cur_dec, v, t);
00314 b->cur_dec += t;
00315 } else {
00316 do {
00317 v = GET_HUFF(gb, b->tree);
00318 if (v) {
00319 sign = -get_bits1(gb);
00320 v = (v ^ sign) - sign;
00321 }
00322 *b->cur_dec++ = v;
00323 } while (b->cur_dec < dec_end);
00324 }
00325 return 0;
00326 }
00327
00328 static const uint8_t bink_rlelens[4] = { 4, 8, 12, 32 };
00329
00330 static int read_block_types(AVCodecContext *avctx, GetBitContext *gb, Bundle *b)
00331 {
00332 int t, v;
00333 int last = 0;
00334 const uint8_t *dec_end;
00335
00336 CHECK_READ_VAL(gb, b, t);
00337 dec_end = b->cur_dec + t;
00338 if (dec_end > b->data_end) {
00339 av_log(avctx, AV_LOG_ERROR, "Too many block type values\n");
00340 return -1;
00341 }
00342 if (get_bits1(gb)) {
00343 v = get_bits(gb, 4);
00344 memset(b->cur_dec, v, t);
00345 b->cur_dec += t;
00346 } else {
00347 do {
00348 v = GET_HUFF(gb, b->tree);
00349 if (v < 12) {
00350 last = v;
00351 *b->cur_dec++ = v;
00352 } else {
00353 int run = bink_rlelens[v - 12];
00354
00355 memset(b->cur_dec, last, run);
00356 b->cur_dec += run;
00357 }
00358 } while (b->cur_dec < dec_end);
00359 }
00360 return 0;
00361 }
00362
00363 static int read_patterns(AVCodecContext *avctx, GetBitContext *gb, Bundle *b)
00364 {
00365 int t, v;
00366 const uint8_t *dec_end;
00367
00368 CHECK_READ_VAL(gb, b, t);
00369 dec_end = b->cur_dec + t;
00370 if (dec_end > b->data_end) {
00371 av_log(avctx, AV_LOG_ERROR, "Too many pattern values\n");
00372 return -1;
00373 }
00374 while (b->cur_dec < dec_end) {
00375 v = GET_HUFF(gb, b->tree);
00376 v |= GET_HUFF(gb, b->tree) << 4;
00377 *b->cur_dec++ = v;
00378 }
00379
00380 return 0;
00381 }
00382
00383 static int read_colors(GetBitContext *gb, Bundle *b, BinkContext *c)
00384 {
00385 int t, sign, v;
00386 const uint8_t *dec_end;
00387
00388 CHECK_READ_VAL(gb, b, t);
00389 dec_end = b->cur_dec + t;
00390 if (dec_end > b->data_end) {
00391 av_log(c->avctx, AV_LOG_ERROR, "Too many color values\n");
00392 return -1;
00393 }
00394 if (get_bits1(gb)) {
00395 c->col_lastval = GET_HUFF(gb, c->col_high[c->col_lastval]);
00396 v = GET_HUFF(gb, b->tree);
00397 v = (c->col_lastval << 4) | v;
00398 if (c->version < 'i') {
00399 sign = ((int8_t) v) >> 7;
00400 v = ((v & 0x7F) ^ sign) - sign;
00401 v += 0x80;
00402 }
00403 memset(b->cur_dec, v, t);
00404 b->cur_dec += t;
00405 } else {
00406 while (b->cur_dec < dec_end) {
00407 c->col_lastval = GET_HUFF(gb, c->col_high[c->col_lastval]);
00408 v = GET_HUFF(gb, b->tree);
00409 v = (c->col_lastval << 4) | v;
00410 if (c->version < 'i') {
00411 sign = ((int8_t) v) >> 7;
00412 v = ((v & 0x7F) ^ sign) - sign;
00413 v += 0x80;
00414 }
00415 *b->cur_dec++ = v;
00416 }
00417 }
00418 return 0;
00419 }
00420
00422 #define DC_START_BITS 11
00423
00424 static int read_dcs(AVCodecContext *avctx, GetBitContext *gb, Bundle *b,
00425 int start_bits, int has_sign)
00426 {
00427 int i, j, len, len2, bsize, sign, v, v2;
00428 int16_t *dst = (int16_t*)b->cur_dec;
00429
00430 CHECK_READ_VAL(gb, b, len);
00431 v = get_bits(gb, start_bits - has_sign);
00432 if (v && has_sign) {
00433 sign = -get_bits1(gb);
00434 v = (v ^ sign) - sign;
00435 }
00436 *dst++ = v;
00437 len--;
00438 for (i = 0; i < len; i += 8) {
00439 len2 = FFMIN(len - i, 8);
00440 bsize = get_bits(gb, 4);
00441 if (bsize) {
00442 for (j = 0; j < len2; j++) {
00443 v2 = get_bits(gb, bsize);
00444 if (v2) {
00445 sign = -get_bits1(gb);
00446 v2 = (v2 ^ sign) - sign;
00447 }
00448 v += v2;
00449 *dst++ = v;
00450 if (v < -32768 || v > 32767) {
00451 av_log(avctx, AV_LOG_ERROR, "DC value went out of bounds: %d\n", v);
00452 return -1;
00453 }
00454 }
00455 } else {
00456 for (j = 0; j < len2; j++)
00457 *dst++ = v;
00458 }
00459 }
00460
00461 b->cur_dec = (uint8_t*)dst;
00462 return 0;
00463 }
00464
00471 static inline int get_value(BinkContext *c, int bundle)
00472 {
00473 int16_t ret;
00474
00475 if (bundle < BINK_SRC_X_OFF || bundle == BINK_SRC_RUN)
00476 return *c->bundle[bundle].cur_ptr++;
00477 if (bundle == BINK_SRC_X_OFF || bundle == BINK_SRC_Y_OFF)
00478 return (int8_t)*c->bundle[bundle].cur_ptr++;
00479 ret = *(int16_t*)c->bundle[bundle].cur_ptr;
00480 c->bundle[bundle].cur_ptr += 2;
00481 return ret;
00482 }
00483
00493 static int read_dct_coeffs(GetBitContext *gb, DCTELEM block[64], const uint8_t *scan,
00494 int is_intra)
00495 {
00496 int coef_list[128];
00497 int mode_list[128];
00498 int i, t, mask, bits, ccoef, mode, sign;
00499 int list_start = 64, list_end = 64, list_pos;
00500 int coef_count = 0;
00501 int coef_idx[64];
00502 int quant_idx;
00503 const uint32_t *quant;
00504
00505 coef_list[list_end] = 4; mode_list[list_end++] = 0;
00506 coef_list[list_end] = 24; mode_list[list_end++] = 0;
00507 coef_list[list_end] = 44; mode_list[list_end++] = 0;
00508 coef_list[list_end] = 1; mode_list[list_end++] = 3;
00509 coef_list[list_end] = 2; mode_list[list_end++] = 3;
00510 coef_list[list_end] = 3; mode_list[list_end++] = 3;
00511
00512 bits = get_bits(gb, 4) - 1;
00513 for (mask = 1 << bits; bits >= 0; mask >>= 1, bits--) {
00514 list_pos = list_start;
00515 while (list_pos < list_end) {
00516 if (!(mode_list[list_pos] | coef_list[list_pos]) || !get_bits1(gb)) {
00517 list_pos++;
00518 continue;
00519 }
00520 ccoef = coef_list[list_pos];
00521 mode = mode_list[list_pos];
00522 switch (mode) {
00523 case 0:
00524 coef_list[list_pos] = ccoef + 4;
00525 mode_list[list_pos] = 1;
00526 case 2:
00527 if (mode == 2) {
00528 coef_list[list_pos] = 0;
00529 mode_list[list_pos++] = 0;
00530 }
00531 for (i = 0; i < 4; i++, ccoef++) {
00532 if (get_bits1(gb)) {
00533 coef_list[--list_start] = ccoef;
00534 mode_list[ list_start] = 3;
00535 } else {
00536 int t;
00537 if (!bits) {
00538 t = 1 - (get_bits1(gb) << 1);
00539 } else {
00540 t = get_bits(gb, bits) | mask;
00541 sign = -get_bits1(gb);
00542 t = (t ^ sign) - sign;
00543 }
00544 block[scan[ccoef]] = t;
00545 coef_idx[coef_count++] = ccoef;
00546 }
00547 }
00548 break;
00549 case 1:
00550 mode_list[list_pos] = 2;
00551 for (i = 0; i < 3; i++) {
00552 ccoef += 4;
00553 coef_list[list_end] = ccoef;
00554 mode_list[list_end++] = 2;
00555 }
00556 break;
00557 case 3:
00558 if (!bits) {
00559 t = 1 - (get_bits1(gb) << 1);
00560 } else {
00561 t = get_bits(gb, bits) | mask;
00562 sign = -get_bits1(gb);
00563 t = (t ^ sign) - sign;
00564 }
00565 block[scan[ccoef]] = t;
00566 coef_idx[coef_count++] = ccoef;
00567 coef_list[list_pos] = 0;
00568 mode_list[list_pos++] = 0;
00569 break;
00570 }
00571 }
00572 }
00573
00574 quant_idx = get_bits(gb, 4);
00575 quant = is_intra ? bink_intra_quant[quant_idx]
00576 : bink_inter_quant[quant_idx];
00577 block[0] = (block[0] * quant[0]) >> 11;
00578 for (i = 0; i < coef_count; i++) {
00579 int idx = coef_idx[i];
00580 block[scan[idx]] = (block[scan[idx]] * quant[idx]) >> 11;
00581 }
00582
00583 return 0;
00584 }
00585
00594 static int read_residue(GetBitContext *gb, DCTELEM block[64], int masks_count)
00595 {
00596 int coef_list[128];
00597 int mode_list[128];
00598 int i, sign, mask, ccoef, mode;
00599 int list_start = 64, list_end = 64, list_pos;
00600 int nz_coeff[64];
00601 int nz_coeff_count = 0;
00602
00603 coef_list[list_end] = 4; mode_list[list_end++] = 0;
00604 coef_list[list_end] = 24; mode_list[list_end++] = 0;
00605 coef_list[list_end] = 44; mode_list[list_end++] = 0;
00606 coef_list[list_end] = 0; mode_list[list_end++] = 2;
00607
00608 for (mask = 1 << get_bits(gb, 3); mask; mask >>= 1) {
00609 for (i = 0; i < nz_coeff_count; i++) {
00610 if (!get_bits1(gb))
00611 continue;
00612 if (block[nz_coeff[i]] < 0)
00613 block[nz_coeff[i]] -= mask;
00614 else
00615 block[nz_coeff[i]] += mask;
00616 masks_count--;
00617 if (masks_count < 0)
00618 return 0;
00619 }
00620 list_pos = list_start;
00621 while (list_pos < list_end) {
00622 if (!(coef_list[list_pos] | mode_list[list_pos]) || !get_bits1(gb)) {
00623 list_pos++;
00624 continue;
00625 }
00626 ccoef = coef_list[list_pos];
00627 mode = mode_list[list_pos];
00628 switch (mode) {
00629 case 0:
00630 coef_list[list_pos] = ccoef + 4;
00631 mode_list[list_pos] = 1;
00632 case 2:
00633 if (mode == 2) {
00634 coef_list[list_pos] = 0;
00635 mode_list[list_pos++] = 0;
00636 }
00637 for (i = 0; i < 4; i++, ccoef++) {
00638 if (get_bits1(gb)) {
00639 coef_list[--list_start] = ccoef;
00640 mode_list[ list_start] = 3;
00641 } else {
00642 nz_coeff[nz_coeff_count++] = bink_scan[ccoef];
00643 sign = -get_bits1(gb);
00644 block[bink_scan[ccoef]] = (mask ^ sign) - sign;
00645 masks_count--;
00646 if (masks_count < 0)
00647 return 0;
00648 }
00649 }
00650 break;
00651 case 1:
00652 mode_list[list_pos] = 2;
00653 for (i = 0; i < 3; i++) {
00654 ccoef += 4;
00655 coef_list[list_end] = ccoef;
00656 mode_list[list_end++] = 2;
00657 }
00658 break;
00659 case 3:
00660 nz_coeff[nz_coeff_count++] = bink_scan[ccoef];
00661 sign = -get_bits1(gb);
00662 block[bink_scan[ccoef]] = (mask ^ sign) - sign;
00663 coef_list[list_pos] = 0;
00664 mode_list[list_pos++] = 0;
00665 masks_count--;
00666 if (masks_count < 0)
00667 return 0;
00668 break;
00669 }
00670 }
00671 }
00672
00673 return 0;
00674 }
00675
00676 static int bink_decode_plane(BinkContext *c, GetBitContext *gb, int plane_idx,
00677 int is_chroma)
00678 {
00679 int blk;
00680 int i, j, bx, by;
00681 uint8_t *dst, *prev, *ref, *ref_start, *ref_end;
00682 int v, col[2];
00683 const uint8_t *scan;
00684 int xoff, yoff;
00685 LOCAL_ALIGNED_16(DCTELEM, block, [64]);
00686 LOCAL_ALIGNED_16(uint8_t, ublock, [64]);
00687 int coordmap[64];
00688
00689 const int stride = c->pic.linesize[plane_idx];
00690 int bw = is_chroma ? (c->avctx->width + 15) >> 4 : (c->avctx->width + 7) >> 3;
00691 int bh = is_chroma ? (c->avctx->height + 15) >> 4 : (c->avctx->height + 7) >> 3;
00692 int width = c->avctx->width >> is_chroma;
00693
00694 init_lengths(c, FFMAX(width, 8), bw);
00695 for (i = 0; i < BINK_NB_SRC; i++)
00696 read_bundle(gb, c, i);
00697
00698 ref_start = c->last.data[plane_idx];
00699 ref_end = c->last.data[plane_idx]
00700 + (bw - 1 + c->last.linesize[plane_idx] * (bh - 1)) * 8;
00701
00702 for (i = 0; i < 64; i++)
00703 coordmap[i] = (i & 7) + (i >> 3) * stride;
00704
00705 for (by = 0; by < bh; by++) {
00706 if (read_block_types(c->avctx, gb, &c->bundle[BINK_SRC_BLOCK_TYPES]) < 0)
00707 return -1;
00708 if (read_block_types(c->avctx, gb, &c->bundle[BINK_SRC_SUB_BLOCK_TYPES]) < 0)
00709 return -1;
00710 if (read_colors(gb, &c->bundle[BINK_SRC_COLORS], c) < 0)
00711 return -1;
00712 if (read_patterns(c->avctx, gb, &c->bundle[BINK_SRC_PATTERN]) < 0)
00713 return -1;
00714 if (read_motion_values(c->avctx, gb, &c->bundle[BINK_SRC_X_OFF]) < 0)
00715 return -1;
00716 if (read_motion_values(c->avctx, gb, &c->bundle[BINK_SRC_Y_OFF]) < 0)
00717 return -1;
00718 if (read_dcs(c->avctx, gb, &c->bundle[BINK_SRC_INTRA_DC], DC_START_BITS, 0) < 0)
00719 return -1;
00720 if (read_dcs(c->avctx, gb, &c->bundle[BINK_SRC_INTER_DC], DC_START_BITS, 1) < 0)
00721 return -1;
00722 if (read_runs(c->avctx, gb, &c->bundle[BINK_SRC_RUN]) < 0)
00723 return -1;
00724
00725 if (by == bh)
00726 break;
00727 dst = c->pic.data[plane_idx] + 8*by*stride;
00728 prev = c->last.data[plane_idx] + 8*by*stride;
00729 for (bx = 0; bx < bw; bx++, dst += 8, prev += 8) {
00730 blk = get_value(c, BINK_SRC_BLOCK_TYPES);
00731
00732 if ((by & 1) && blk == SCALED_BLOCK) {
00733 bx++;
00734 dst += 8;
00735 prev += 8;
00736 continue;
00737 }
00738 switch (blk) {
00739 case SKIP_BLOCK:
00740 c->dsp.put_pixels_tab[1][0](dst, prev, stride, 8);
00741 break;
00742 case SCALED_BLOCK:
00743 blk = get_value(c, BINK_SRC_SUB_BLOCK_TYPES);
00744 switch (blk) {
00745 case RUN_BLOCK:
00746 scan = bink_patterns[get_bits(gb, 4)];
00747 i = 0;
00748 do {
00749 int run = get_value(c, BINK_SRC_RUN) + 1;
00750
00751 i += run;
00752 if (i > 64) {
00753 av_log(c->avctx, AV_LOG_ERROR, "Run went out of bounds\n");
00754 return -1;
00755 }
00756 if (get_bits1(gb)) {
00757 v = get_value(c, BINK_SRC_COLORS);
00758 for (j = 0; j < run; j++)
00759 ublock[*scan++] = v;
00760 } else {
00761 for (j = 0; j < run; j++)
00762 ublock[*scan++] = get_value(c, BINK_SRC_COLORS);
00763 }
00764 } while (i < 63);
00765 if (i == 63)
00766 ublock[*scan++] = get_value(c, BINK_SRC_COLORS);
00767 break;
00768 case INTRA_BLOCK:
00769 c->dsp.clear_block(block);
00770 block[0] = get_value(c, BINK_SRC_INTRA_DC);
00771 read_dct_coeffs(gb, block, c->scantable.permutated, 1);
00772 c->dsp.idct(block);
00773 c->dsp.put_pixels_nonclamped(block, ublock, 8);
00774 break;
00775 case FILL_BLOCK:
00776 v = get_value(c, BINK_SRC_COLORS);
00777 c->dsp.fill_block_tab[0](dst, v, stride, 16);
00778 break;
00779 case PATTERN_BLOCK:
00780 for (i = 0; i < 2; i++)
00781 col[i] = get_value(c, BINK_SRC_COLORS);
00782 for (j = 0; j < 8; j++) {
00783 v = get_value(c, BINK_SRC_PATTERN);
00784 for (i = 0; i < 8; i++, v >>= 1)
00785 ublock[i + j*8] = col[v & 1];
00786 }
00787 break;
00788 case RAW_BLOCK:
00789 for (j = 0; j < 8; j++)
00790 for (i = 0; i < 8; i++)
00791 ublock[i + j*8] = get_value(c, BINK_SRC_COLORS);
00792 break;
00793 default:
00794 av_log(c->avctx, AV_LOG_ERROR, "Incorrect 16x16 block type %d\n", blk);
00795 return -1;
00796 }
00797 if (blk != FILL_BLOCK)
00798 c->dsp.scale_block(ublock, dst, stride);
00799 bx++;
00800 dst += 8;
00801 prev += 8;
00802 break;
00803 case MOTION_BLOCK:
00804 xoff = get_value(c, BINK_SRC_X_OFF);
00805 yoff = get_value(c, BINK_SRC_Y_OFF);
00806 ref = prev + xoff + yoff * stride;
00807 if (ref < ref_start || ref > ref_end) {
00808 av_log(c->avctx, AV_LOG_ERROR, "Copy out of bounds @%d, %d\n",
00809 bx*8 + xoff, by*8 + yoff);
00810 return -1;
00811 }
00812 c->dsp.put_pixels_tab[1][0](dst, ref, stride, 8);
00813 break;
00814 case RUN_BLOCK:
00815 scan = bink_patterns[get_bits(gb, 4)];
00816 i = 0;
00817 do {
00818 int run = get_value(c, BINK_SRC_RUN) + 1;
00819
00820 i += run;
00821 if (i > 64) {
00822 av_log(c->avctx, AV_LOG_ERROR, "Run went out of bounds\n");
00823 return -1;
00824 }
00825 if (get_bits1(gb)) {
00826 v = get_value(c, BINK_SRC_COLORS);
00827 for (j = 0; j < run; j++)
00828 dst[coordmap[*scan++]] = v;
00829 } else {
00830 for (j = 0; j < run; j++)
00831 dst[coordmap[*scan++]] = get_value(c, BINK_SRC_COLORS);
00832 }
00833 } while (i < 63);
00834 if (i == 63)
00835 dst[coordmap[*scan++]] = get_value(c, BINK_SRC_COLORS);
00836 break;
00837 case RESIDUE_BLOCK:
00838 xoff = get_value(c, BINK_SRC_X_OFF);
00839 yoff = get_value(c, BINK_SRC_Y_OFF);
00840 ref = prev + xoff + yoff * stride;
00841 if (ref < ref_start || ref > ref_end) {
00842 av_log(c->avctx, AV_LOG_ERROR, "Copy out of bounds @%d, %d\n",
00843 bx*8 + xoff, by*8 + yoff);
00844 return -1;
00845 }
00846 c->dsp.put_pixels_tab[1][0](dst, ref, stride, 8);
00847 c->dsp.clear_block(block);
00848 v = get_bits(gb, 7);
00849 read_residue(gb, block, v);
00850 c->dsp.add_pixels8(dst, block, stride);
00851 break;
00852 case INTRA_BLOCK:
00853 c->dsp.clear_block(block);
00854 block[0] = get_value(c, BINK_SRC_INTRA_DC);
00855 read_dct_coeffs(gb, block, c->scantable.permutated, 1);
00856 c->dsp.idct_put(dst, stride, block);
00857 break;
00858 case FILL_BLOCK:
00859 v = get_value(c, BINK_SRC_COLORS);
00860 c->dsp.fill_block_tab[1](dst, v, stride, 8);
00861 break;
00862 case INTER_BLOCK:
00863 xoff = get_value(c, BINK_SRC_X_OFF);
00864 yoff = get_value(c, BINK_SRC_Y_OFF);
00865 ref = prev + xoff + yoff * stride;
00866 c->dsp.put_pixels_tab[1][0](dst, ref, stride, 8);
00867 c->dsp.clear_block(block);
00868 block[0] = get_value(c, BINK_SRC_INTER_DC);
00869 read_dct_coeffs(gb, block, c->scantable.permutated, 0);
00870 c->dsp.idct_add(dst, stride, block);
00871 break;
00872 case PATTERN_BLOCK:
00873 for (i = 0; i < 2; i++)
00874 col[i] = get_value(c, BINK_SRC_COLORS);
00875 for (i = 0; i < 8; i++) {
00876 v = get_value(c, BINK_SRC_PATTERN);
00877 for (j = 0; j < 8; j++, v >>= 1)
00878 dst[i*stride + j] = col[v & 1];
00879 }
00880 break;
00881 case RAW_BLOCK:
00882 for (i = 0; i < 8; i++)
00883 memcpy(dst + i*stride, c->bundle[BINK_SRC_COLORS].cur_ptr + i*8, 8);
00884 c->bundle[BINK_SRC_COLORS].cur_ptr += 64;
00885 break;
00886 default:
00887 av_log(c->avctx, AV_LOG_ERROR, "Unknown block type %d\n", blk);
00888 return -1;
00889 }
00890 }
00891 }
00892 if (get_bits_count(gb) & 0x1F)
00893 skip_bits_long(gb, 32 - (get_bits_count(gb) & 0x1F));
00894
00895 return 0;
00896 }
00897
00898 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, AVPacket *pkt)
00899 {
00900 BinkContext * const c = avctx->priv_data;
00901 GetBitContext gb;
00902 int plane, plane_idx;
00903 int bits_count = pkt->size << 3;
00904
00905 if(c->pic.data[0])
00906 avctx->release_buffer(avctx, &c->pic);
00907
00908 if(avctx->get_buffer(avctx, &c->pic) < 0){
00909 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
00910 return -1;
00911 }
00912
00913 init_get_bits(&gb, pkt->data, bits_count);
00914 if (c->has_alpha) {
00915 if (c->version >= 'i')
00916 skip_bits_long(&gb, 32);
00917 if (bink_decode_plane(c, &gb, 3, 0) < 0)
00918 return -1;
00919 }
00920 if (c->version >= 'i')
00921 skip_bits_long(&gb, 32);
00922
00923 for (plane = 0; plane < 3; plane++) {
00924 plane_idx = (!plane || !c->swap_planes) ? plane : (plane ^ 3);
00925
00926 if (bink_decode_plane(c, &gb, plane_idx, !!plane) < 0)
00927 return -1;
00928 if (get_bits_count(&gb) >= bits_count)
00929 break;
00930 }
00931 emms_c();
00932
00933 *data_size = sizeof(AVFrame);
00934 *(AVFrame*)data = c->pic;
00935
00936 FFSWAP(AVFrame, c->pic, c->last);
00937
00938
00939 return pkt->size;
00940 }
00941
00942 static av_cold int decode_init(AVCodecContext *avctx)
00943 {
00944 BinkContext * const c = avctx->priv_data;
00945 static VLC_TYPE table[16 * 128][2];
00946 int i;
00947 int flags;
00948
00949 c->version = avctx->codec_tag >> 24;
00950 if (c->version < 'c') {
00951 av_log(avctx, AV_LOG_ERROR, "Too old version '%c'\n", c->version);
00952 return -1;
00953 }
00954 if (avctx->extradata_size < 4) {
00955 av_log(avctx, AV_LOG_ERROR, "Extradata missing or too short\n");
00956 return -1;
00957 }
00958 flags = AV_RL32(avctx->extradata);
00959 c->has_alpha = flags & BINK_FLAG_ALPHA;
00960 c->swap_planes = c->version >= 'h';
00961 if (!bink_trees[15].table) {
00962 for (i = 0; i < 16; i++) {
00963 const int maxbits = bink_tree_lens[i][15];
00964 bink_trees[i].table = table + i*128;
00965 bink_trees[i].table_allocated = 1 << maxbits;
00966 init_vlc(&bink_trees[i], maxbits, 16,
00967 bink_tree_lens[i], 1, 1,
00968 bink_tree_bits[i], 1, 1, INIT_VLC_USE_NEW_STATIC | INIT_VLC_LE);
00969 }
00970 }
00971 c->avctx = avctx;
00972
00973 c->pic.data[0] = NULL;
00974
00975 if (av_image_check_size(avctx->width, avctx->height, 0, avctx) < 0) {
00976 return 1;
00977 }
00978
00979 avctx->pix_fmt = c->has_alpha ? PIX_FMT_YUVA420P : PIX_FMT_YUV420P;
00980
00981 avctx->idct_algo = FF_IDCT_BINK;
00982 dsputil_init(&c->dsp, avctx);
00983 ff_init_scantable(c->dsp.idct_permutation, &c->scantable, bink_scan);
00984
00985 init_bundles(c);
00986
00987 return 0;
00988 }
00989
00990 static av_cold int decode_end(AVCodecContext *avctx)
00991 {
00992 BinkContext * const c = avctx->priv_data;
00993
00994 if (c->pic.data[0])
00995 avctx->release_buffer(avctx, &c->pic);
00996 if (c->last.data[0])
00997 avctx->release_buffer(avctx, &c->last);
00998
00999 free_bundles(c);
01000 return 0;
01001 }
01002
01003 AVCodec bink_decoder = {
01004 "binkvideo",
01005 AVMEDIA_TYPE_VIDEO,
01006 CODEC_ID_BINKVIDEO,
01007 sizeof(BinkContext),
01008 decode_init,
01009 NULL,
01010 decode_end,
01011 decode_frame,
01012 .long_name = NULL_IF_CONFIG_SMALL("Bink video"),
01013 };
