45 const uint8_t *val_table,
int nb_codes,
46 int use_static,
int is_ac)
49 uint16_t huff_code[256];
50 uint16_t huff_sym[256];
53 assert(nb_codes <= 256);
57 for (i = 0; i < 256; i++)
58 huff_sym[i] = i + 16 * is_ac;
61 huff_sym[0] = 16 * 256;
64 huff_code, 2, 2, huff_sym, 2, 2, use_static);
113 "mjpeg: error using external huffman table\n");
146 for (i = 0; i < 64; i++) {
164 int len,
index, i,
class, n, v, code_max;
181 for (i = 1; i <= 16; i++) {
186 if (len < n || n > 256)
190 for (i = 0; i < n; i++) {
201 class, index, code_max + 1);
202 if ((ret =
build_vlc(&s->
vlcs[
class][index], bits_table, val_table,
203 code_max + 1, 0,
class > 0)) < 0)
208 if ((ret =
build_vlc(&s->
vlcs[2][index], bits_table, val_table,
209 code_max + 1, 0, 0)) < 0)
246 if (nb_components <= 0 ||
252 "nb_components changing in interlaced picture\n");
256 if (s->
ls && !(s->
bits <= 8 || nb_components == 1)) {
258 "JPEG-LS that is not <= 8 "
259 "bits/component or 16-bit gray");
265 for (i = 0; i < nb_components; i++) {
280 "Invalid sampling factor in component %d %d:%d\n",
332 if (!(pix_fmt_id & 0xD0D0D0D0))
333 pix_fmt_id -= (pix_fmt_id & 0xF0F0F0F0) >> 1;
334 if (!(pix_fmt_id & 0x0D0D0D0D))
335 pix_fmt_id -= (pix_fmt_id & 0x0F0F0F0F) >> 1;
337 switch (pix_fmt_id) {
364 else if (s->
bits <= 8)
385 for (i = 0; i < 3; i++)
392 if (len != (8 + (3 * nb_components)))
398 int bw = (width + s->
h_max * 8 - 1) / (s->
h_max * 8);
399 int bh = (height + s->
v_max * 8 - 1) / (s->
v_max * 8);
419 "mjpeg_decode_dc: bad vlc: %d:%d (%p)\n",
420 0, dc_index, &s->
vlcs[0][dc_index]);
432 int dc_index,
int ac_index, int16_t *quant_matrix)
434 int code, i, j,
level, val;
442 val = val * quant_matrix[0] + s->
last_dc[component];
452 i += ((unsigned)code) >> 4;
460 int sign = (~cache) >> 31;
461 level = (
NEG_USR32(sign ^ cache,code) ^ sign) - sign;
471 block[j] = level * quant_matrix[j];
480 int component,
int dc_index,
481 int16_t *quant_matrix,
int Al)
490 val = (val * quant_matrix[0] << Al) + s->
last_dc[component];
498 uint8_t *last_nnz,
int ac_index,
499 int16_t *quant_matrix,
500 int ss,
int se,
int Al,
int *EOBRUN)
511 for (i = ss; ; i++) {
515 run = ((unsigned) code) >> 4;
524 int sign = (~cache) >> 31;
525 level = (
NEG_USR32(sign ^ cache,code) ^ sign) - sign;
533 block[j] = level * quant_matrix[j] << Al;
540 block[j] = level * quant_matrix[j] << Al;
569 #define REFINE_BIT(j) { \
570 UPDATE_CACHE(re, &s->gb); \
571 sign = block[j] >> 15; \
572 block[j] += SHOW_UBITS(re, &s->gb, 1) * \
573 ((quant_matrix[j] ^ sign) - sign) << Al; \
574 LAST_SKIP_BITS(re, &s->gb, 1); \
582 av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i); \
587 j = s->scantable.permutated[i]; \
590 else if (run-- == 0) \
597 int ac_index, int16_t *quant_matrix,
598 int ss,
int se,
int Al,
int *EOBRUN)
600 int code, i = ss, j, sign, val,
run;
601 int last =
FFMIN(se, *last_nnz);
612 run = ((unsigned) code) >> 4;
619 block[j] = ((quant_matrix[j]^val) - val) << Al;
627 run = ((unsigned) code) >> 4;
648 for (; i <= last; i++) {
665 int left[3], top[3], topleft[3];
666 const int linesize = s->
linesize[0];
667 const int mask = (1 << s->
bits) - 1;
673 for (i = 0; i < 3; i++)
674 buffer[0][i] = 1 << (s->
bits + point_transform - 1);
676 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
677 const int modified_predictor = mb_y ? predictor : 1;
681 ptr += linesize >> 1;
683 for (i = 0; i < 3; i++)
684 top[i] = left[i] = topleft[i] =
buffer[0][i];
686 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
690 for (i = 0; i < 3; i++) {
696 PREDICT(pred, topleft[i], top[i], left[i], modified_predictor);
698 left[i] =
buffer[mb_x][i] =
709 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
710 ptr[4 * mb_x + 1] =
buffer[mb_x][0] - ((
buffer[mb_x][1] +
buffer[mb_x][2] - 0x200) >> 2);
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++) {
717 ptr[4 * mb_x + 0] =
buffer[mb_x][1] + ptr[4 * mb_x + 1];
718 ptr[4 * mb_x + 2] =
buffer[mb_x][2] + ptr[4 * mb_x + 1];
721 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
722 ptr[4 * mb_x + 0] =
buffer[mb_x][2];
723 ptr[4 * mb_x + 1] =
buffer[mb_x][1];
724 ptr[4 * mb_x + 2] =
buffer[mb_x][0];
732 int point_transform,
int nb_components)
736 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
737 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
741 if (mb_x == 0 || mb_y == 0 || s->
interlaced) {
742 for (i = 0; i < nb_components; i++) {
744 int n, h, v, x, y, c, j, linesize;
753 for (j = 0; j < n; j++) {
757 (linesize * (v * mb_y + y)) +
759 if (y == 0 && mb_y == 0) {
760 if (x == 0 && mb_x == 0)
761 pred = 128 << point_transform;
765 if (x == 0 && mb_x == 0)
766 pred = ptr[-linesize];
768 PREDICT(pred, ptr[-linesize - 1],
769 ptr[-linesize], ptr[-1], predictor);
773 ptr += linesize >> 1;
783 for (i = 0; i < nb_components; i++) {
785 int n, h, v, x, y, c, j, linesize;
794 for (j = 0; j < n; j++) {
799 (linesize * (v * mb_y + y)) +
801 PREDICT(pred, ptr[-linesize - 1],
802 ptr[-linesize], ptr[-1], predictor);
821 int Al,
const uint8_t *mb_bitmask,
833 for (i = 0; i < nb_components; i++) {
836 reference_data[c] = reference ? reference->
data[c] :
NULL;
841 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
842 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
843 const int copy_mb = mb_bitmask && !
get_bits1(&mb_bitmask_gb);
853 for (i = 0; i < nb_components; i++) {
855 int n, h, v, x, y, c, j;
863 for (j = 0; j < n; j++) {
864 block_offset = ((linesize[c] * (v * mb_y + y) * 8) +
868 block_offset += linesize[c] >> 1;
869 ptr = data[c] + block_offset;
873 reference_data[c] + block_offset,
881 "error y=%d x=%d\n", mb_y, mb_x);
897 "error y=%d x=%d\n", mb_y, mb_x);
904 (v * mb_y + y) * 8, (h * mb_x + x) * 8);
922 for (i = 0; i < nb_components; i++)
934 int se,
int Ah,
int Al,
948 if (ss < 0 || ss >= 64 ||
949 se < ss || se >= 64 ||
962 int offset = linesize >> 1;
964 reference_data += offset;
967 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
968 int block_offset = mb_y * linesize * 8;
969 uint8_t *ptr = data + block_offset;
973 for (mb_x = 0; mb_x < s->
mb_width; mb_x++,
block++, last_nnz++) {
974 const int copy_mb = mb_bitmask && !
get_bits1(&mb_bitmask_gb);
980 quant_matrix, ss, se, Al, &EOBRUN);
983 quant_matrix, ss, se, Al, &EOBRUN);
986 "error y=%d x=%d\n", mb_y, mb_x);
994 reference_data + block_offset,
1009 int len, nb_components, i, h, v, predictor, point_transform;
1011 const int block_size = s->
lossless ? 1 : 8;
1012 int ilv, prev_shift;
1019 "decode_sos: nb_components (%d) unsupported\n", nb_components);
1022 if (len != 6 + 2 * nb_components) {
1026 for (i = 0; i < nb_components; i++) {
1035 "decode_sos: index(%d) out of components\n", index);
1065 if (nb_components > 1) {
1069 }
else if (!s->
ls) {
1072 s->
mb_width = (s->
width + h * block_size - 1) / (h * block_size);
1081 s->
lossless ?
"lossless" :
"sequential DCT", s->
rgb ?
"RGB" :
"",
1082 predictor, point_transform, ilv, s->
bits,
1091 for (i = 0; i < nb_components; i++)
1100 point_transform, ilv)) < 0)
1105 point_transform)) < 0)
1110 nb_components)) < 0)
1124 prev_shift, point_transform,
1125 mb_bitmask, reference)) < 0)
1136 av_dlog(s->
avctx,
"AVRn interlaced picture marker found\n");
1210 int t_w, t_h, v1, v2;
1221 "mjpeg: JFIF header found (version: %x.%x) SAR=%d/%d\n",
1230 if (len -10 - (t_w * t_h * 3) > 0)
1231 len -= t_w * t_h * 3;
1251 "Pegasus lossless jpeg header found\n");
1298 "mjpeg: error, decode_app parser read over the end\n");
1312 for (i = 0; i < len - 2; i++)
1314 if (i > 0 && cbuf[i - 1] ==
'\n')
1323 if (!strcmp(cbuf,
"AVID")) {
1325 }
else if (!strcmp(cbuf,
"CS=ITU601"))
1327 else if ((len > 20 && !strncmp(cbuf,
"Intel(R) JPEG Library", 21)) ||
1328 (len > 19 && !strncmp(cbuf,
"Metasoft MJPEG Codec", 20)))
1349 buf_ptr = *pbuf_ptr;
1350 while (buf_ptr < buf_end) {
1353 if ((v == 0xff) && (v2 >= 0xc0) && (v2 <= 0xfe) && buf_ptr < buf_end) {
1363 av_dlog(
NULL,
"find_marker skipped %d bytes\n", skipped);
1364 *pbuf_ptr = buf_ptr;
1370 const uint8_t **unescaped_buf_ptr,
1371 int *unescaped_buf_size)
1381 if (start_code ==
SOS && !s->
ls) {
1382 const uint8_t *src = *buf_ptr;
1385 while (src < buf_end) {
1391 while (src < buf_end && x == 0xff)
1394 if (x >= 0xd0 && x <= 0xd7)
1401 *unescaped_buf_ptr = s->
buffer;
1402 *unescaped_buf_size = dst - s->
buffer;
1403 memset(s->
buffer + *unescaped_buf_size, 0,
1407 (buf_end - *buf_ptr) - (dst - s->
buffer));
1408 }
else if (start_code ==
SOS && s->
ls) {
1409 const uint8_t *src = *buf_ptr;
1418 while (src + t < buf_end) {
1421 while ((src + t < buf_end) && x == 0xff)
1444 *unescaped_buf_ptr = dst;
1445 *unescaped_buf_size = (bit_count + 7) >> 3;
1446 memset(s->
buffer + *unescaped_buf_size, 0,
1449 *unescaped_buf_ptr = *buf_ptr;
1450 *unescaped_buf_size = buf_end - *buf_ptr;
1461 int buf_size = avpkt->
size;
1463 const uint8_t *buf_end, *buf_ptr;
1464 const uint8_t *unescaped_buf_ptr;
1465 int unescaped_buf_size;
1471 buf_end = buf + buf_size;
1472 while (buf_ptr < buf_end) {
1476 &unescaped_buf_size);
1478 if (start_code < 0) {
1480 }
else if (unescaped_buf_size > INT_MAX / 8) {
1482 "MJPEG packet 0x%x too big (%d/%d), corrupt data?\n",
1483 start_code, unescaped_buf_size, buf_size);
1488 start_code, buf_end - buf_ptr);
1491 unescaped_buf_size * 8);
1500 if (start_code >= 0xd0 && start_code <= 0xd7)
1502 "restart marker: %d\n", start_code & 0x0f);
1504 else if (start_code >=
APP0 && start_code <=
APP15)
1507 else if (start_code ==
COM)
1511 (start_code ==
SOF48 || start_code ==
LSE)) {
1516 switch (start_code) {
1572 "Found EOI before any SOF, ignoring\n");
1585 for (i = 0; frame->
data[i]; i++) {
1586 int h = frame->
height >> ((i == 1 || i == 2) ?
1606 "Can not process SOS before SOF, skipping\n");
1631 "mjpeg: unsupported coding type (%x)\n", start_code);
1639 "marker parser used %d bytes (%d bits)\n",
1652 return buf_ptr - buf;
1670 for (i = 0; i < 3; i++) {
1671 for (j = 0; j < 4; j++)
1681 #define OFFSET(x) offsetof(MJpegDecodeContext, x)
1682 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
1684 {
"extern_huff",
"Use external huffman table.",
1706 .priv_class = &mjpegdec_class,