47 const uint8_t *val_table,
int nb_codes,
48 int use_static,
int is_ac)
51 uint16_t huff_code[256];
52 uint16_t huff_sym[256];
55 assert(nb_codes <= 256);
59 for (i = 0; i < 256; i++)
60 huff_sym[i] = i + 16 * is_ac;
63 huff_sym[0] = 16 * 256;
66 huff_code, 2, 2, huff_sym, 2, 2, use_static);
110 "mjpeg: error using external huffman table\n");
143 for (i = 0; i < 64; i++) {
161 int len,
index, i,
class, n, v, code_max;
178 for (i = 1; i <= 16; i++) {
183 if (len < n || n > 256)
187 for (i = 0; i < n; i++) {
198 class, index, code_max + 1);
199 if ((ret =
build_vlc(&s->
vlcs[
class][index], bits_table, val_table,
200 code_max + 1, 0,
class > 0)) < 0)
205 if ((ret =
build_vlc(&s->
vlcs[2][index], bits_table, val_table,
206 code_max + 1, 0, 0)) < 0)
243 if (nb_components <= 0 ||
249 "nb_components changing in interlaced picture\n");
253 if (s->
ls && !(s->
bits <= 8 || nb_components == 1)) {
255 "For JPEG-LS anything except <= 8 bits/component"
256 " or 16-bit gray", 0);
262 for (i = 0; i < nb_components; i++) {
277 "Invalid sampling factor in component %d %d:%d\n",
330 if (!(pix_fmt_id & 0xD0D0D0D0))
331 pix_fmt_id -= (pix_fmt_id & 0xF0F0F0F0) >> 1;
332 if (!(pix_fmt_id & 0x0D0D0D0D))
333 pix_fmt_id -= (pix_fmt_id & 0x0F0F0F0F) >> 1;
335 switch (pix_fmt_id) {
362 else if (s->
bits <= 8)
379 for (i = 0; i < 3; i++)
386 if (len != (8 + (3 * nb_components)))
392 int bw = (width + s->
h_max * 8 - 1) / (s->
h_max * 8);
393 int bh = (height + s->
v_max * 8 - 1) / (s->
v_max * 8);
413 "mjpeg_decode_dc: bad vlc: %d:%d (%p)\n",
414 0, dc_index, &s->
vlcs[0][dc_index]);
426 int dc_index,
int ac_index, int16_t *quant_matrix)
428 int code, i, j,
level, val;
436 val = val * quant_matrix[0] + s->
last_dc[component];
446 i += ((unsigned)code) >> 4;
454 int sign = (~cache) >> 31;
455 level = (
NEG_USR32(sign ^ cache,code) ^ sign) - sign;
465 block[j] = level * quant_matrix[j];
474 int component,
int dc_index,
475 int16_t *quant_matrix,
int Al)
484 val = (val * quant_matrix[0] << Al) + s->
last_dc[component];
492 uint8_t *last_nnz,
int ac_index,
493 int16_t *quant_matrix,
494 int ss,
int se,
int Al,
int *EOBRUN)
505 for (i = ss; ; i++) {
509 run = ((unsigned) code) >> 4;
518 int sign = (~cache) >> 31;
519 level = (
NEG_USR32(sign ^ cache,code) ^ sign) - sign;
527 block[j] = level * quant_matrix[j] << Al;
534 block[j] = level * quant_matrix[j] << Al;
563 #define REFINE_BIT(j) { \
564 UPDATE_CACHE(re, &s->gb); \
565 sign = block[j] >> 15; \
566 block[j] += SHOW_UBITS(re, &s->gb, 1) * \
567 ((quant_matrix[j] ^ sign) - sign) << Al; \
568 LAST_SKIP_BITS(re, &s->gb, 1); \
576 av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i); \
581 j = s->scantable.permutated[i]; \
584 else if (run-- == 0) \
591 int ac_index, int16_t *quant_matrix,
592 int ss,
int se,
int Al,
int *EOBRUN)
594 int code, i = ss, j, sign, val,
run;
595 int last =
FFMIN(se, *last_nnz);
606 run = ((unsigned) code) >> 4;
613 block[j] = ((quant_matrix[j]^val) - val) << Al;
621 run = ((unsigned) code) >> 4;
642 for (; i <= last; i++) {
659 int left[3], top[3], topleft[3];
660 const int linesize = s->
linesize[0];
661 const int mask = (1 << s->
bits) - 1;
667 for (i = 0; i < 3; i++)
668 buffer[0][i] = 1 << (s->
bits + point_transform - 1);
670 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
671 const int modified_predictor = mb_y ? predictor : 1;
675 ptr += linesize >> 1;
677 for (i = 0; i < 3; i++)
678 top[i] = left[i] = topleft[i] =
buffer[0][i];
680 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
684 for (i = 0; i < 3; i++) {
690 PREDICT(pred, topleft[i], top[i], left[i], modified_predictor);
692 left[i] =
buffer[mb_x][i] =
703 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
704 ptr[4 * mb_x + 1] =
buffer[mb_x][0] - ((
buffer[mb_x][1] +
buffer[mb_x][2] - 0x200) >> 2);
705 ptr[4 * mb_x + 0] =
buffer[mb_x][1] + ptr[4 * mb_x + 1];
706 ptr[4 * mb_x + 2] =
buffer[mb_x][2] + ptr[4 * mb_x + 1];
709 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
711 ptr[4 * mb_x + 0] =
buffer[mb_x][1] + ptr[4 * mb_x + 1];
712 ptr[4 * mb_x + 2] =
buffer[mb_x][2] + ptr[4 * mb_x + 1];
715 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
716 ptr[4 * mb_x + 0] =
buffer[mb_x][2];
717 ptr[4 * mb_x + 1] =
buffer[mb_x][1];
718 ptr[4 * mb_x + 2] =
buffer[mb_x][0];
726 int point_transform,
int nb_components)
730 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
731 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
735 if (mb_x == 0 || mb_y == 0 || s->
interlaced) {
736 for (i = 0; i < nb_components; i++) {
738 int n, h, v, x, y, c, j, linesize;
747 for (j = 0; j < n; j++) {
751 (linesize * (v * mb_y + y)) +
753 if (y == 0 && mb_y == 0) {
754 if (x == 0 && mb_x == 0)
755 pred = 128 << point_transform;
759 if (x == 0 && mb_x == 0)
760 pred = ptr[-linesize];
762 PREDICT(pred, ptr[-linesize - 1],
763 ptr[-linesize], ptr[-1], predictor);
767 ptr += linesize >> 1;
777 for (i = 0; i < nb_components; i++) {
779 int n, h, v, x, y, c, j, linesize;
788 for (j = 0; j < n; j++) {
793 (linesize * (v * mb_y + y)) +
795 PREDICT(pred, ptr[-linesize - 1],
796 ptr[-linesize], ptr[-1], predictor);
815 int Al,
const uint8_t *mb_bitmask,
829 "Can not flip image with CODEC_FLAG_EMU_EDGE set!\n");
833 for (i = 0; i < nb_components; i++) {
836 reference_data[c] = reference ? reference->
data[c] :
NULL;
841 int offset = (linesize[c] * (s->
v_scount[i] *
844 reference_data[c] += offset;
849 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
850 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
851 const int copy_mb = mb_bitmask && !
get_bits1(&mb_bitmask_gb);
861 for (i = 0; i < nb_components; i++) {
863 int n, h, v, x, y, c, j;
871 for (j = 0; j < n; j++) {
872 block_offset = ((linesize[c] * (v * mb_y + y) * 8) +
876 block_offset += linesize[c] >> 1;
877 ptr = data[c] + block_offset;
881 linesize[c], linesize[c], 8);
888 "error y=%d x=%d\n", mb_y, mb_x);
904 "error y=%d x=%d\n", mb_y, mb_x);
911 (v * mb_y + y) * 8, (h * mb_x + x) * 8);
929 for (i = 0; i < nb_components; i++)
941 int se,
int Ah,
int Al,
955 if (ss < 0 || ss >= 64 ||
956 se < ss || se >= 64 ||
969 int offset = linesize >> 1;
971 reference_data += offset;
974 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
975 int block_offset = mb_y * linesize * 8;
976 uint8_t *ptr = data + block_offset;
980 for (mb_x = 0; mb_x < s->
mb_width; mb_x++,
block++, last_nnz++) {
981 const int copy_mb = mb_bitmask && !
get_bits1(&mb_bitmask_gb);
987 quant_matrix, ss, se, Al, &EOBRUN);
990 quant_matrix, ss, se, Al, &EOBRUN);
993 "error y=%d x=%d\n", mb_y, mb_x);
1001 linesize, linesize, 8);
1015 int len, nb_components, i, h, v, predictor, point_transform;
1017 const int block_size = s->
lossless ? 1 : 8;
1018 int ilv, prev_shift;
1025 "decode_sos: nb_components (%d) unsupported\n", nb_components);
1028 if (len != 6 + 2 * nb_components) {
1032 for (i = 0; i < nb_components; i++) {
1041 "decode_sos: index(%d) out of components\n", index);
1071 if (nb_components > 1) {
1075 }
else if (!s->
ls) {
1078 s->
mb_width = (s->
width + h * block_size - 1) / (h * block_size);
1087 s->
lossless ?
"lossless" :
"sequential DCT", s->
rgb ?
"RGB" :
"",
1088 predictor, point_transform, ilv, s->
bits,
1097 for (i = 0; i < nb_components; i++)
1106 point_transform, ilv)) < 0)
1111 point_transform)) < 0)
1116 nb_components)) < 0)
1130 prev_shift, point_transform,
1131 mb_bitmask, reference)) < 0)
1142 av_dlog(s->
avctx,
"AVRn interlaced picture marker found\n");
1216 int t_w, t_h, v1, v2;
1227 "mjpeg: JFIF header found (version: %x.%x) SAR=%d/%d\n",
1236 if (len -10 - (t_w * t_h * 3) > 0)
1237 len -= t_w * t_h * 3;
1257 "Pegasus lossless jpeg header found\n");
1304 "mjpeg: error, decode_app parser read over the end\n");
1318 for (i = 0; i < len - 2; i++)
1320 if (i > 0 && cbuf[i - 1] ==
'\n')
1329 if (!strcmp(cbuf,
"AVID")) {
1331 }
else if (!strcmp(cbuf,
"CS=ITU601"))
1333 else if ((len > 20 && !strncmp(cbuf,
"Intel(R) JPEG Library", 21)) ||
1334 (len > 19 && !strncmp(cbuf,
"Metasoft MJPEG Codec", 20)))
1355 buf_ptr = *pbuf_ptr;
1356 while (buf_ptr < buf_end) {
1359 if ((v == 0xff) && (v2 >= 0xc0) && (v2 <= 0xfe) && buf_ptr < buf_end) {
1369 av_dlog(
NULL,
"find_marker skipped %d bytes\n", skipped);
1370 *pbuf_ptr = buf_ptr;
1376 const uint8_t **unescaped_buf_ptr,
1377 int *unescaped_buf_size)
1387 if (start_code ==
SOS && !s->
ls) {
1388 const uint8_t *src = *buf_ptr;
1391 while (src < buf_end) {
1397 while (src < buf_end && x == 0xff)
1400 if (x >= 0xd0 && x <= 0xd7)
1407 *unescaped_buf_ptr = s->
buffer;
1408 *unescaped_buf_size = dst - s->
buffer;
1409 memset(s->
buffer + *unescaped_buf_size, 0,
1413 (buf_end - *buf_ptr) - (dst - s->
buffer));
1414 }
else if (start_code ==
SOS && s->
ls) {
1415 const uint8_t *src = *buf_ptr;
1424 while (src + t < buf_end) {
1427 while ((src + t < buf_end) && x == 0xff)
1450 *unescaped_buf_ptr = dst;
1451 *unescaped_buf_size = (bit_count + 7) >> 3;
1452 memset(s->
buffer + *unescaped_buf_size, 0,
1455 *unescaped_buf_ptr = *buf_ptr;
1456 *unescaped_buf_size = buf_end - *buf_ptr;
1466 int buf_size = avpkt->
size;
1468 const uint8_t *buf_end, *buf_ptr;
1469 const uint8_t *unescaped_buf_ptr;
1470 int unescaped_buf_size;
1477 buf_end = buf + buf_size;
1478 while (buf_ptr < buf_end) {
1482 &unescaped_buf_size);
1484 if (start_code < 0) {
1486 }
else if (unescaped_buf_size > INT_MAX / 8) {
1488 "MJPEG packet 0x%x too big (%d/%d), corrupt data?\n",
1489 start_code, unescaped_buf_size, buf_size);
1493 start_code, buf_end - buf_ptr);
1496 unescaped_buf_size * 8);
1506 if (start_code >= 0xd0 && start_code <= 0xd7)
1508 "restart marker: %d\n", start_code & 0x0f);
1510 else if (start_code >=
APP0 && start_code <=
APP15)
1513 else if (start_code ==
COM)
1517 (start_code ==
SOF48 || start_code ==
LSE)) {
1522 switch (start_code) {
1578 "Found EOI before any SOF, ignoring\n");
1597 (s->
width + 15) / 16);
1600 "QP: %d\n", picture->
quality);
1608 "Can not process SOS before SOF, skipping\n");
1633 "mjpeg: unsupported coding type (%x)\n", start_code);
1641 "marker parser used %d bytes (%d bits)\n",
1655 return buf_ptr - buf;
1671 for (i = 0; i < 3; i++) {
1672 for (j = 0; j < 4; j++)
1682 #define OFFSET(x) offsetof(MJpegDecodeContext, x)
1683 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
1685 {
"extern_huff",
"Use external huffman table.",