falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32471 Commits
32 Branches
570MB
Tree: 01cb62aba2
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '01cb62aba2'
${ noResults }
FFmpeg/libavcodec/mjpegdec.c

1685 lines
57KB
Raw Blame History 

  1. /*

  2.  * MJPEG decoder

  3.  * Copyright (c) 2000, 2001 Fabrice Bellard

  4.  * Copyright (c) 2003 Alex Beregszaszi

  5.  * Copyright (c) 2003-2004 Michael Niedermayer

  6.  *

  7.  * Support for external huffman table, various fixes (AVID workaround),

  8.  * aspecting, new decode_frame mechanism and apple mjpeg-b support

  9.  *                                  by Alex Beregszaszi

  10.  *

  11.  * This file is part of Libav.

  12.  *

  13.  * Libav is free software; you can redistribute it and/or

  14.  * modify it under the terms of the GNU Lesser General Public

  15.  * License as published by the Free Software Foundation; either

  16.  * version 2.1 of the License, or (at your option) any later version.

  17.  *

  18.  * Libav is distributed in the hope that it will be useful,

  19.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  20.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  21.  * Lesser General Public License for more details.

  22.  *

  23.  * You should have received a copy of the GNU Lesser General Public

  24.  * License along with Libav; if not, write to the Free Software

  25.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  26.  */

  27. 

  28. /**

  29.  * @file

  30.  * MJPEG decoder.

  31.  */

  32. 

  33. // #define DEBUG

  34. #include <assert.h>

  35. 

  36. #include "libavutil/imgutils.h"

  37. #include "libavutil/opt.h"

  38. #include "avcodec.h"

  39. #include "dsputil.h"

  40. #include "mjpeg.h"

  41. #include "mjpegdec.h"

  42. #include "jpeglsdec.h"

  43. 

  44. 

  45. static int build_vlc(VLC *vlc, const uint8_t *bits_table,

  46.                      const uint8_t *val_table, int nb_codes,

  47.                      int use_static, int is_ac)

  48. {

  49.     uint8_t huff_size[256];

  50.     uint16_t huff_code[256];

  51.     uint16_t huff_sym[256];

  52.     int i;

  53. 

  54.     assert(nb_codes <= 256);

  55. 

  56.     memset(huff_size, 0, sizeof(huff_size));

  57.     ff_mjpeg_build_huffman_codes(huff_size, huff_code, bits_table, val_table);

  58. 

  59.     for (i = 0; i < 256; i++)

  60.         huff_sym[i] = i + 16 * is_ac;

  61. 

  62.     if (is_ac)

  63.         huff_sym[0] = 16 * 256;

  64. 

  65.     return ff_init_vlc_sparse(vlc, 9, nb_codes, huff_size, 1, 1,

  66.                               huff_code, 2, 2, huff_sym, 2, 2, use_static);

  67. }

  68. 

  69. static void build_basic_mjpeg_vlc(MJpegDecodeContext *s)

  70. {

  71.     build_vlc(&s->vlcs[0][0], ff_mjpeg_bits_dc_luminance,

  72.               ff_mjpeg_val_dc, 12, 0, 0);

  73.     build_vlc(&s->vlcs[0][1], ff_mjpeg_bits_dc_chrominance,

  74.               ff_mjpeg_val_dc, 12, 0, 0);

  75.     build_vlc(&s->vlcs[1][0], ff_mjpeg_bits_ac_luminance,

  76.               ff_mjpeg_val_ac_luminance, 251, 0, 1);

  77.     build_vlc(&s->vlcs[1][1], ff_mjpeg_bits_ac_chrominance,

  78.               ff_mjpeg_val_ac_chrominance, 251, 0, 1);

  79.     build_vlc(&s->vlcs[2][0], ff_mjpeg_bits_ac_luminance,

  80.               ff_mjpeg_val_ac_luminance, 251, 0, 0);

  81.     build_vlc(&s->vlcs[2][1], ff_mjpeg_bits_ac_chrominance,

  82.               ff_mjpeg_val_ac_chrominance, 251, 0, 0);

  83. }

  84. 

  85. av_cold int ff_mjpeg_decode_init(AVCodecContext *avctx)

  86. {

  87.     MJpegDecodeContext *s = avctx->priv_data;

  88. 

  89.     if (!s->picture_ptr)

  90.         s->picture_ptr = &s->picture;

  91. 

  92.     s->avctx = avctx;

  93.     ff_dsputil_init(&s->dsp, avctx);

  94.     ff_init_scantable(s->dsp.idct_permutation, &s->scantable, ff_zigzag_direct);

  95.     s->buffer_size   = 0;

  96.     s->buffer        = NULL;

  97.     s->start_code    = -1;

  98.     s->first_picture = 1;

  99.     s->org_height    = avctx->coded_height;

  100.     avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;

  101. 

  102.     build_basic_mjpeg_vlc(s);

  103. 

  104.     if (s->extern_huff) {

  105.         av_log(avctx, AV_LOG_INFO, "mjpeg: using external huffman table\n");

  106.         init_get_bits(&s->gb, avctx->extradata, avctx->extradata_size * 8);

  107.         if (ff_mjpeg_decode_dht(s)) {

  108.             av_log(avctx, AV_LOG_ERROR,

  109.                    "mjpeg: error using external huffman table\n");

  110.             return AVERROR_INVALIDDATA;

  111.         }

  112.     }

  113.     if (avctx->field_order == AV_FIELD_BB) { /* quicktime icefloe 019 */

  114.         s->interlace_polarity = 1;           /* bottom field first */

  115.         av_log(avctx, AV_LOG_DEBUG, "mjpeg bottom field first\n");

  116.     }

  117.     if (avctx->codec->id == CODEC_ID_AMV)

  118.         s->flipped = 1;

  119. 

  120.     return 0;

  121. }

  122. 

  123. 

  124. /* quantize tables */

  125. int ff_mjpeg_decode_dqt(MJpegDecodeContext *s)

  126. {

  127.     int len, index, i, j;

  128. 

  129.     len = get_bits(&s->gb, 16) - 2;

  130. 

  131.     while (len >= 65) {

  132.         /* only 8 bit precision handled */

  133.         if (get_bits(&s->gb, 4) != 0) {

  134.             av_log(s->avctx, AV_LOG_ERROR, "dqt: 16bit precision\n");

  135.             return -1;

  136.         }

  137.         index = get_bits(&s->gb, 4);

  138.         if (index >= 4)

  139.             return -1;

  140.         av_log(s->avctx, AV_LOG_DEBUG, "index=%d\n", index);

  141.         /* read quant table */

  142.         for (i = 0; i < 64; i++) {

  143.             j = s->scantable.permutated[i];

  144.             s->quant_matrixes[index][j] = get_bits(&s->gb, 8);

  145.         }

  146. 

  147.         // XXX FIXME finetune, and perhaps add dc too

  148.         s->qscale[index] = FFMAX(s->quant_matrixes[index][s->scantable.permutated[1]],

  149.                                  s->quant_matrixes[index][s->scantable.permutated[8]]) >> 1;

  150.         av_log(s->avctx, AV_LOG_DEBUG, "qscale[%d]: %d\n",

  151.                index, s->qscale[index]);

  152.         len -= 65;

  153.     }

  154.     return 0;

  155. }

  156. 

  157. /* decode huffman tables and build VLC decoders */

  158. int ff_mjpeg_decode_dht(MJpegDecodeContext *s)

  159. {

  160.     int len, index, i, class, n, v, code_max;

  161.     uint8_t bits_table[17];

  162.     uint8_t val_table[256];

  163. 

  164.     len = get_bits(&s->gb, 16) - 2;

  165. 

  166.     while (len > 0) {

  167.         if (len < 17)

  168.             return -1;

  169.         class = get_bits(&s->gb, 4);

  170.         if (class >= 2)

  171.             return -1;

  172.         index = get_bits(&s->gb, 4);

  173.         if (index >= 4)

  174.             return -1;

  175.         n = 0;

  176.         for (i = 1; i <= 16; i++) {

  177.             bits_table[i] = get_bits(&s->gb, 8);

  178.             n += bits_table[i];

  179.         }

  180.         len -= 17;

  181.         if (len < n || n > 256)

  182.             return -1;

  183. 

  184.         code_max = 0;

  185.         for (i = 0; i < n; i++) {

  186.             v = get_bits(&s->gb, 8);

  187.             if (v > code_max)

  188.                 code_max = v;

  189.             val_table[i] = v;

  190.         }

  191.         len -= n;

  192. 

  193.         /* build VLC and flush previous vlc if present */

  194.         ff_free_vlc(&s->vlcs[class][index]);

  195.         av_log(s->avctx, AV_LOG_DEBUG, "class=%d index=%d nb_codes=%d\n",

  196.                class, index, code_max + 1);

  197.         if (build_vlc(&s->vlcs[class][index], bits_table, val_table,

  198.                       code_max + 1, 0, class > 0) < 0)

  199.             return -1;

  200. 

  201.         if (class > 0) {

  202.             ff_free_vlc(&s->vlcs[2][index]);

  203.             if (build_vlc(&s->vlcs[2][index], bits_table, val_table,

  204.                           code_max + 1, 0, 0) < 0)

  205.                 return -1;

  206.         }

  207.     }

  208.     return 0;

  209. }

  210. 

  211. int ff_mjpeg_decode_sof(MJpegDecodeContext *s)

  212. {

  213.     int len, nb_components, i, width, height, pix_fmt_id;

  214. 

  215.     /* XXX: verify len field validity */

  216.     len     = get_bits(&s->gb, 16);

  217.     s->bits = get_bits(&s->gb, 8);

  218. 

  219.     if (s->pegasus_rct)

  220.         s->bits = 9;

  221.     if (s->bits == 9 && !s->pegasus_rct)

  222.         s->rct  = 1;    // FIXME ugly

  223. 

  224.     if (s->bits != 8 && !s->lossless) {

  225.         av_log(s->avctx, AV_LOG_ERROR, "only 8 bits/component accepted\n");

  226.         return -1;

  227.     }

  228. 

  229.     height = get_bits(&s->gb, 16);

  230.     width  = get_bits(&s->gb, 16);

  231. 

  232.     // HACK for odd_height.mov

  233.     if (s->interlaced && s->width == width && s->height == height + 1)

  234.         height= s->height;

  235. 

  236.     av_log(s->avctx, AV_LOG_DEBUG, "sof0: picture: %dx%d\n", width, height);

  237.     if (av_image_check_size(width, height, 0, s->avctx))

  238.         return -1;

  239. 

  240.     nb_components = get_bits(&s->gb, 8);

  241.     if (nb_components <= 0 ||

  242.         nb_components > MAX_COMPONENTS)

  243.         return -1;

  244.     if (s->ls && !(s->bits <= 8 || nb_components == 1)) {

  245.         av_log(s->avctx, AV_LOG_ERROR,

  246.                "only <= 8 bits/component or 16-bit gray accepted for JPEG-LS\n");

  247.         return -1;

  248.     }

  249.     s->nb_components = nb_components;

  250.     s->h_max         = 1;

  251.     s->v_max         = 1;

  252.     for (i = 0; i < nb_components; i++) {

  253.         /* component id */

  254.         s->component_id[i] = get_bits(&s->gb, 8) - 1;

  255.         s->h_count[i]      = get_bits(&s->gb, 4);

  256.         s->v_count[i]      = get_bits(&s->gb, 4);

  257.         /* compute hmax and vmax (only used in interleaved case) */

  258.         if (s->h_count[i] > s->h_max)

  259.             s->h_max = s->h_count[i];

  260.         if (s->v_count[i] > s->v_max)

  261.             s->v_max = s->v_count[i];

  262.         s->quant_index[i] = get_bits(&s->gb, 8);

  263.         if (s->quant_index[i] >= 4)

  264.             return -1;

  265.         av_log(s->avctx, AV_LOG_DEBUG, "component %d %d:%d id: %d quant:%d\n",

  266.                i, s->h_count[i], s->v_count[i],

  267.                s->component_id[i], s->quant_index[i]);

  268.     }

  269. 

  270.     if (s->ls && (s->h_max > 1 || s->v_max > 1)) {

  271.         av_log(s->avctx, AV_LOG_ERROR,

  272.                "Subsampling in JPEG-LS is not supported.\n");

  273.         return -1;

  274.     }

  275. 

  276.     if (s->v_max == 1 && s->h_max == 1 && s->lossless == 1)

  277.         s->rgb = 1;

  278. 

  279.     /* if different size, realloc/alloc picture */

  280.     /* XXX: also check h_count and v_count */

  281.     if (width != s->width || height != s->height) {

  282.         av_freep(&s->qscale_table);

  283. 

  284.         s->width      = width;

  285.         s->height     = height;

  286.         s->interlaced = 0;

  287. 

  288.         /* test interlaced mode */

  289.         if (s->first_picture   &&

  290.             s->org_height != 0 &&

  291.             s->height < ((s->org_height * 3) / 4)) {

  292.             s->interlaced                    = 1;

  293.             s->bottom_field                  = s->interlace_polarity;

  294.             s->picture_ptr->interlaced_frame = 1;

  295.             s->picture_ptr->top_field_first  = !s->interlace_polarity;

  296.             height *= 2;

  297.         }

  298. 

  299.         avcodec_set_dimensions(s->avctx, width, height);

  300. 

  301.         s->qscale_table  = av_mallocz((s->width + 15) / 16);

  302.         s->first_picture = 0;

  303.     }

  304. 

  305.     if (s->interlaced && (s->bottom_field == !s->interlace_polarity))

  306.         return 0;

  307. 

  308.     /* XXX: not complete test ! */

  309.     pix_fmt_id = (s->h_count[0] << 28) | (s->v_count[0] << 24) |

  310.                  (s->h_count[1] << 20) | (s->v_count[1] << 16) |

  311.                  (s->h_count[2] << 12) | (s->v_count[2] <<  8) |

  312.                  (s->h_count[3] <<  4) |  s->v_count[3];

  313.     av_log(s->avctx, AV_LOG_DEBUG, "pix fmt id %x\n", pix_fmt_id);

  314.     /* NOTE we do not allocate pictures large enough for the possible

  315.      * padding of h/v_count being 4 */

  316.     if (!(pix_fmt_id & 0xD0D0D0D0))

  317.         pix_fmt_id -= (pix_fmt_id & 0xF0F0F0F0) >> 1;

  318.     if (!(pix_fmt_id & 0x0D0D0D0D))

  319.         pix_fmt_id -= (pix_fmt_id & 0x0F0F0F0F) >> 1;

  320. 

  321.     switch (pix_fmt_id) {

  322.     case 0x11111100:

  323.         if (s->rgb)

  324.             s->avctx->pix_fmt = PIX_FMT_BGRA;

  325.         else

  326.             s->avctx->pix_fmt = s->cs_itu601 ? PIX_FMT_YUV444P : PIX_FMT_YUVJ444P;

  327.         assert(s->nb_components == 3);

  328.         break;

  329.     case 0x11000000:

  330.         s->avctx->pix_fmt = PIX_FMT_GRAY8;

  331.         break;

  332.     case 0x12111100:

  333.         s->avctx->pix_fmt = s->cs_itu601 ? PIX_FMT_YUV440P : PIX_FMT_YUVJ440P;

  334.         break;

  335.     case 0x21111100:

  336.         s->avctx->pix_fmt = s->cs_itu601 ? PIX_FMT_YUV422P : PIX_FMT_YUVJ422P;

  337.         break;

  338.     case 0x22111100:

  339.         s->avctx->pix_fmt = s->cs_itu601 ? PIX_FMT_YUV420P : PIX_FMT_YUVJ420P;

  340.         break;

  341.     default:

  342.         av_log(s->avctx, AV_LOG_ERROR, "Unhandled pixel format 0x%x\n", pix_fmt_id);

  343.         return -1;

  344.     }

  345.     if (s->ls) {

  346.         if (s->nb_components > 1)

  347.             s->avctx->pix_fmt = PIX_FMT_RGB24;

  348.         else if (s->bits <= 8)

  349.             s->avctx->pix_fmt = PIX_FMT_GRAY8;

  350.         else

  351.             s->avctx->pix_fmt = PIX_FMT_GRAY16;

  352.     }

  353. 

  354.     if (s->picture_ptr->data[0])

  355.         s->avctx->release_buffer(s->avctx, s->picture_ptr);

  356. 

  357.     if (s->avctx->get_buffer(s->avctx, s->picture_ptr) < 0) {

  358.         av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed\n");

  359.         return -1;

  360.     }

  361.     s->picture_ptr->pict_type = AV_PICTURE_TYPE_I;

  362.     s->picture_ptr->key_frame = 1;

  363.     s->got_picture            = 1;

  364. 

  365.     for (i = 0; i < 3; i++)

  366.         s->linesize[i] = s->picture_ptr->linesize[i] << s->interlaced;

  367. 

  368. //    printf("%d %d %d %d %d %d\n",

  369. //           s->width, s->height, s->linesize[0], s->linesize[1],

  370. //           s->interlaced, s->avctx->height);

  371. 

  372.     if (len != (8 + (3 * nb_components)))

  373.         av_log(s->avctx, AV_LOG_DEBUG, "decode_sof0: error, len(%d) mismatch\n", len);

  374. 

  375.     /* totally blank picture as progressive JPEG will only add details to it */

  376.     if (s->progressive) {

  377.         int bw = (width  + s->h_max * 8 - 1) / (s->h_max * 8);

  378.         int bh = (height + s->v_max * 8 - 1) / (s->v_max * 8);

  379.         for (i = 0; i < s->nb_components; i++) {

  380.             int size = bw * bh * s->h_count[i] * s->v_count[i];

  381.             av_freep(&s->blocks[i]);

  382.             av_freep(&s->last_nnz[i]);

  383.             s->blocks[i]       = av_malloc(size * sizeof(**s->blocks));

  384.             s->last_nnz[i]     = av_mallocz(size * sizeof(**s->last_nnz));

  385.             s->block_stride[i] = bw * s->h_count[i];

  386.         }

  387.         memset(s->coefs_finished, 0, sizeof(s->coefs_finished));

  388.     }

  389.     return 0;

  390. }

  391. 

  392. static inline int mjpeg_decode_dc(MJpegDecodeContext *s, int dc_index)

  393. {

  394.     int code;

  395.     code = get_vlc2(&s->gb, s->vlcs[0][dc_index].table, 9, 2);

  396.     if (code < 0) {

  397.         av_log(s->avctx, AV_LOG_WARNING,

  398.                "mjpeg_decode_dc: bad vlc: %d:%d (%p)\n",

  399.                0, dc_index, &s->vlcs[0][dc_index]);

  400.         return 0xffff;

  401.     }

  402. 

  403.     if (code)

  404.         return get_xbits(&s->gb, code);

  405.     else

  406.         return 0;

  407. }

  408. 

  409. /* decode block and dequantize */

  410. static int decode_block(MJpegDecodeContext *s, DCTELEM *block, int component,

  411.                         int dc_index, int ac_index, int16_t *quant_matrix)

  412. {

  413.     int code, i, j, level, val;

  414. 

  415.     /* DC coef */

  416.     val = mjpeg_decode_dc(s, dc_index);

  417.     if (val == 0xffff) {

  418.         av_log(s->avctx, AV_LOG_ERROR, "error dc\n");

  419.         return -1;

  420.     }

  421.     val = val * quant_matrix[0] + s->last_dc[component];

  422.     s->last_dc[component] = val;

  423.     block[0] = val;

  424.     /* AC coefs */

  425.     i = 0;

  426.     {OPEN_READER(re, &s->gb);

  427.     do {

  428.         UPDATE_CACHE(re, &s->gb);

  429.         GET_VLC(code, re, &s->gb, s->vlcs[1][ac_index].table, 9, 2);

  430. 

  431.         i += ((unsigned)code) >> 4;

  432.             code &= 0xf;

  433.         if (code) {

  434.             if (code > MIN_CACHE_BITS - 16)

  435.                 UPDATE_CACHE(re, &s->gb);

  436. 

  437.             {

  438.                 int cache = GET_CACHE(re, &s->gb);

  439.                 int sign  = (~cache) >> 31;

  440.                 level     = (NEG_USR32(sign ^ cache,code) ^ sign) - sign;

  441.             }

  442. 

  443.             LAST_SKIP_BITS(re, &s->gb, code);

  444. 

  445.             if (i > 63) {

  446.                 av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i);

  447.                 return -1;

  448.             }

  449.             j        = s->scantable.permutated[i];

  450.             block[j] = level * quant_matrix[j];

  451.         }

  452.     } while (i < 63);

  453.     CLOSE_READER(re, &s->gb);}

  454. 

  455.     return 0;

  456. }

  457. 

  458. static int decode_dc_progressive(MJpegDecodeContext *s, DCTELEM *block,

  459.                                  int component, int dc_index,

  460.                                  int16_t *quant_matrix, int Al)

  461. {

  462.     int val;

  463.     s->dsp.clear_block(block);

  464.     val = mjpeg_decode_dc(s, dc_index);

  465.     if (val == 0xffff) {

  466.         av_log(s->avctx, AV_LOG_ERROR, "error dc\n");

  467.         return -1;

  468.     }

  469.     val = (val * quant_matrix[0] << Al) + s->last_dc[component];

  470.     s->last_dc[component] = val;

  471.     block[0] = val;

  472.     return 0;

  473. }

  474. 

  475. /* decode block and dequantize - progressive JPEG version */

  476. static int decode_block_progressive(MJpegDecodeContext *s, DCTELEM *block,

  477.                                     uint8_t *last_nnz, int ac_index,

  478.                                     int16_t *quant_matrix,

  479.                                     int ss, int se, int Al, int *EOBRUN)

  480. {

  481.     int code, i, j, level, val, run;

  482. 

  483.     if (*EOBRUN) {

  484.         (*EOBRUN)--;

  485.         return 0;

  486.     }

  487. 

  488.     {

  489.         OPEN_READER(re, &s->gb);

  490.         for (i = ss; ; i++) {

  491.             UPDATE_CACHE(re, &s->gb);

  492.             GET_VLC(code, re, &s->gb, s->vlcs[2][ac_index].table, 9, 2);

  493. 

  494.             run = ((unsigned) code) >> 4;

  495.             code &= 0xF;

  496.             if (code) {

  497.                 i += run;

  498.                 if (code > MIN_CACHE_BITS - 16)

  499.                     UPDATE_CACHE(re, &s->gb);

  500. 

  501.                 {

  502.                     int cache = GET_CACHE(re, &s->gb);

  503.                     int sign  = (~cache) >> 31;

  504.                     level     = (NEG_USR32(sign ^ cache,code) ^ sign) - sign;

  505.                 }

  506. 

  507.                 LAST_SKIP_BITS(re, &s->gb, code);

  508. 

  509.                 if (i >= se) {

  510.                     if (i == se) {

  511.                         j = s->scantable.permutated[se];

  512.                         block[j] = level * quant_matrix[j] << Al;

  513.                         break;

  514.                     }

  515.                     av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i);

  516.                     return -1;

  517.                 }

  518.                 j = s->scantable.permutated[i];

  519.                 block[j] = level * quant_matrix[j] << Al;

  520.             } else {

  521.                 if (run == 0xF) {// ZRL - skip 15 coefficients

  522.                     i += 15;

  523.                     if (i >= se) {

  524.                         av_log(s->avctx, AV_LOG_ERROR, "ZRL overflow: %d\n", i);

  525.                         return -1;

  526.                     }

  527.                 } else {

  528.                     val = (1 << run);

  529.                     if (run) {

  530.                         UPDATE_CACHE(re, &s->gb);

  531.                         val += NEG_USR32(GET_CACHE(re, &s->gb), run);

  532.                         LAST_SKIP_BITS(re, &s->gb, run);

  533.                     }

  534.                     *EOBRUN = val - 1;

  535.                     break;

  536.                 }

  537.             }

  538.         }

  539.         CLOSE_READER(re, &s->gb);

  540.     }

  541. 

  542.     if (i > *last_nnz)

  543.         *last_nnz = i;

  544. 

  545.     return 0;

  546. }

  547. 

  548. #define REFINE_BIT(j) {                                             \

  549.     UPDATE_CACHE(re, &s->gb);                                       \

  550.     sign = block[j] >> 15;                                          \

  551.     block[j] += SHOW_UBITS(re, &s->gb, 1) *                         \

  552.                 ((quant_matrix[j] ^ sign) - sign) << Al;            \

  553.     LAST_SKIP_BITS(re, &s->gb, 1);                                  \

  554. }

  555. 

  556. #define ZERO_RUN                                                    \

  557. for (; ; i++) {                                                     \

  558.     if (i > last) {                                                 \

  559.         i += run;                                                   \

  560.         if (i > se) {                                               \

  561.             av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i); \

  562.             return -1;                                              \

  563.         }                                                           \

  564.         break;                                                      \

  565.     }                                                               \

  566.     j = s->scantable.permutated[i];                                 \

  567.     if (block[j])                                                   \

  568.         REFINE_BIT(j)                                               \

  569.     else if (run-- == 0)                                            \

  570.         break;                                                      \

  571. }

  572. 

  573. /* decode block and dequantize - progressive JPEG refinement pass */

  574. static int decode_block_refinement(MJpegDecodeContext *s, DCTELEM *block,

  575.                                    uint8_t *last_nnz,

  576.                                    int ac_index, int16_t *quant_matrix,

  577.                                    int ss, int se, int Al, int *EOBRUN)

  578. {

  579.     int code, i = ss, j, sign, val, run;

  580.     int last    = FFMIN(se, *last_nnz);

  581. 

  582.     OPEN_READER(re, &s->gb);

  583.     if (*EOBRUN) {

  584.         (*EOBRUN)--;

  585.     } else {

  586.         for (; ; i++) {

  587.             UPDATE_CACHE(re, &s->gb);

  588.             GET_VLC(code, re, &s->gb, s->vlcs[2][ac_index].table, 9, 2);

  589. 

  590.             if (code & 0xF) {

  591.                 run = ((unsigned) code) >> 4;

  592.                 UPDATE_CACHE(re, &s->gb);

  593.                 val = SHOW_UBITS(re, &s->gb, 1);

  594.                 LAST_SKIP_BITS(re, &s->gb, 1);

  595.                 ZERO_RUN;

  596.                 j = s->scantable.permutated[i];

  597.                 val--;

  598.                 block[j] = ((quant_matrix[j]^val) - val) << Al;

  599.                 if (i == se) {

  600.                     if (i > *last_nnz)

  601.                         *last_nnz = i;

  602.                     CLOSE_READER(re, &s->gb);

  603.                     return 0;

  604.                 }

  605.             } else {

  606.                 run = ((unsigned) code) >> 4;

  607.                 if (run == 0xF) {

  608.                     ZERO_RUN;

  609.                 } else {

  610.                     val = run;

  611.                     run = (1 << run);

  612.                     if (val) {

  613.                         UPDATE_CACHE(re, &s->gb);

  614.                         run += SHOW_UBITS(re, &s->gb, val);

  615.                         LAST_SKIP_BITS(re, &s->gb, val);

  616.                     }

  617.                     *EOBRUN = run - 1;

  618.                     break;

  619.                 }

  620.             }

  621.         }

  622. 

  623.         if (i > *last_nnz)

  624.             *last_nnz = i;

  625.     }

  626. 

  627.     for (; i <= last; i++) {

  628.         j = s->scantable.permutated[i];

  629.         if (block[j])

  630.             REFINE_BIT(j)

  631.     }

  632.     CLOSE_READER(re, &s->gb);

  633. 

  634.     return 0;

  635. }

  636. #undef REFINE_BIT

  637. #undef ZERO_RUN

  638. 

  639. static int ljpeg_decode_rgb_scan(MJpegDecodeContext *s, int predictor,

  640.                                  int point_transform)

  641. {

  642.     int i, mb_x, mb_y;

  643.     uint16_t (*buffer)[4];

  644.     int left[3], top[3], topleft[3];

  645.     const int linesize = s->linesize[0];

  646.     const int mask     = (1 << s->bits) - 1;

  647. 

  648.     av_fast_malloc(&s->ljpeg_buffer, &s->ljpeg_buffer_size,

  649.                    (unsigned)s->mb_width * 4 * sizeof(s->ljpeg_buffer[0][0]));

  650.     buffer = s->ljpeg_buffer;

  651. 

  652.     for (i = 0; i < 3; i++)

  653.         buffer[0][i] = 1 << (s->bits + point_transform - 1);

  654. 

  655.     for (mb_y = 0; mb_y < s->mb_height; mb_y++) {

  656.         const int modified_predictor = mb_y ? predictor : 1;

  657.         uint8_t *ptr = s->picture_ptr->data[0] + (linesize * mb_y);

  658. 

  659.         if (s->interlaced && s->bottom_field)

  660.             ptr += linesize >> 1;

  661. 

  662.         for (i = 0; i < 3; i++)

  663.             top[i] = left[i] = topleft[i] = buffer[0][i];

  664. 

  665.         for (mb_x = 0; mb_x < s->mb_width; mb_x++) {

  666.             if (s->restart_interval && !s->restart_count)

  667.                 s->restart_count = s->restart_interval;

  668. 

  669.             for (i = 0; i < 3; i++) {

  670.                 int pred;

  671. 

  672.                 topleft[i] = top[i];

  673.                 top[i]     = buffer[mb_x][i];

  674. 

  675.                 PREDICT(pred, topleft[i], top[i], left[i], modified_predictor);

  676. 

  677.                 left[i] = buffer[mb_x][i] =

  678.                     mask & (pred + (mjpeg_decode_dc(s, s->dc_index[i]) << point_transform));

  679.             }

  680. 

  681.             if (s->restart_interval && !--s->restart_count) {

  682.                 align_get_bits(&s->gb);

  683.                 skip_bits(&s->gb, 16); /* skip RSTn */

  684.             }

  685.         }

  686. 

  687.         if (s->rct) {

  688.             for (mb_x = 0; mb_x < s->mb_width; mb_x++) {

  689.                 ptr[4 * mb_x + 1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2] - 0x200) >> 2);

  690.                 ptr[4 * mb_x + 0] = buffer[mb_x][1] + ptr[4 * mb_x + 1];

  691.                 ptr[4 * mb_x + 2] = buffer[mb_x][2] + ptr[4 * mb_x + 1];

  692.             }

  693.         } else if (s->pegasus_rct) {

  694.             for (mb_x = 0; mb_x < s->mb_width; mb_x++) {

  695.                 ptr[4 * mb_x + 1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2]) >> 2);

  696.                 ptr[4 * mb_x + 0] = buffer[mb_x][1] + ptr[4 * mb_x + 1];

  697.                 ptr[4 * mb_x + 2] = buffer[mb_x][2] + ptr[4 * mb_x + 1];

  698.             }

  699.         } else {

  700.             for (mb_x = 0; mb_x < s->mb_width; mb_x++) {

  701.                 ptr[4 * mb_x + 0] = buffer[mb_x][2];

  702.                 ptr[4 * mb_x + 1] = buffer[mb_x][1];

  703.                 ptr[4 * mb_x + 2] = buffer[mb_x][0];

  704.             }

  705.         }

  706.     }

  707.     return 0;

  708. }

  709. 

  710. static int ljpeg_decode_yuv_scan(MJpegDecodeContext *s, int predictor,

  711.                                  int point_transform)

  712. {

  713.     int i, mb_x, mb_y;

  714.     const int nb_components = 3;

  715. 

  716.     for (mb_y = 0; mb_y < s->mb_height; mb_y++) {

  717.         for (mb_x = 0; mb_x < s->mb_width; mb_x++) {

  718.             if (s->restart_interval && !s->restart_count)

  719.                 s->restart_count = s->restart_interval;

  720. 

  721.             if (mb_x == 0 || mb_y == 0 || s->interlaced) {

  722.                 for (i = 0; i < nb_components; i++) {

  723.                     uint8_t *ptr;

  724.                     int n, h, v, x, y, c, j, linesize;

  725.                     n        = s->nb_blocks[i];

  726.                     c        = s->comp_index[i];

  727.                     h        = s->h_scount[i];

  728.                     v        = s->v_scount[i];

  729.                     x        = 0;

  730.                     y        = 0;

  731.                     linesize = s->linesize[c];

  732. 

  733.                     for (j = 0; j < n; j++) {

  734.                         int pred;

  735.                         // FIXME optimize this crap

  736.                         ptr = s->picture_ptr->data[c] +

  737.                               (linesize * (v * mb_y + y)) +

  738.                               (h * mb_x + x);

  739.                         if (y == 0 && mb_y == 0) {

  740.                             if (x == 0 && mb_x == 0)

  741.                                 pred = 128 << point_transform;

  742.                             else

  743.                                 pred = ptr[-1];

  744.                         } else {

  745.                             if (x == 0 && mb_x == 0)

  746.                                 pred = ptr[-linesize];

  747.                             else

  748.                                 PREDICT(pred, ptr[-linesize - 1],

  749.                                         ptr[-linesize], ptr[-1], predictor);

  750.                        }

  751. 

  752.                         if (s->interlaced && s->bottom_field)

  753.                             ptr += linesize >> 1;

  754.                         *ptr = pred + (mjpeg_decode_dc(s, s->dc_index[i]) << point_transform);

  755. 

  756.                         if (++x == h) {

  757.                             x = 0;

  758.                             y++;

  759.                         }

  760.                     }

  761.                 }

  762.             } else {

  763.                 for (i = 0; i < nb_components; i++) {

  764.                     uint8_t *ptr;

  765.                     int n, h, v, x, y, c, j, linesize;

  766.                     n        = s->nb_blocks[i];

  767.                     c        = s->comp_index[i];

  768.                     h        = s->h_scount[i];

  769.                     v        = s->v_scount[i];

  770.                     x        = 0;

  771.                     y        = 0;

  772.                     linesize = s->linesize[c];

  773. 

  774.                     for (j = 0; j < n; j++) {

  775.                         int pred;

  776. 

  777.                         // FIXME optimize this crap

  778.                         ptr = s->picture_ptr->data[c] +

  779.                               (linesize * (v * mb_y + y)) +

  780.                               (h * mb_x + x);

  781.                         PREDICT(pred, ptr[-linesize - 1],

  782.                                 ptr[-linesize], ptr[-1], predictor);

  783.                         *ptr = pred + (mjpeg_decode_dc(s, s->dc_index[i]) << point_transform);

  784.                         if (++x == h) {

  785.                             x = 0;

  786.                             y++;

  787.                         }

  788.                     }

  789.                 }

  790.             }

  791.             if (s->restart_interval && !--s->restart_count) {

  792.                 align_get_bits(&s->gb);

  793.                 skip_bits(&s->gb, 16); /* skip RSTn */

  794.             }

  795.         }

  796.     }

  797.     return 0;

  798. }

  799. 

  800. static av_always_inline void mjpeg_copy_block(uint8_t *dst, const uint8_t *src,

  801.                                               int linesize, int lowres)

  802. {

  803.     switch (lowres) {

  804.     case 0: copy_block8(dst, src, linesize, linesize, 8);

  805.         break;

  806.     case 1: copy_block4(dst, src, linesize, linesize, 4);

  807.         break;

  808.     case 2: copy_block2(dst, src, linesize, linesize, 2);

  809.         break;

  810.     case 3: *dst = *src;

  811.         break;

  812.     }

  813. }

  814. 

  815. static int mjpeg_decode_scan(MJpegDecodeContext *s, int nb_components, int Ah,

  816.                              int Al, const uint8_t *mb_bitmask,

  817.                              const AVFrame *reference)

  818. {

  819.     int i, mb_x, mb_y;

  820.     uint8_t *data[MAX_COMPONENTS];

  821.     const uint8_t *reference_data[MAX_COMPONENTS];

  822.     int linesize[MAX_COMPONENTS];

  823.     GetBitContext mb_bitmask_gb;

  824. 

  825.     if (mb_bitmask)

  826.         init_get_bits(&mb_bitmask_gb, mb_bitmask, s->mb_width * s->mb_height);

  827. 

  828.     if (s->flipped && s->avctx->flags & CODEC_FLAG_EMU_EDGE) {

  829.         av_log(s->avctx, AV_LOG_ERROR,

  830.                "Can not flip image with CODEC_FLAG_EMU_EDGE set!\n");

  831.         s->flipped = 0;

  832.     }

  833. 

  834.     for (i = 0; i < nb_components; i++) {

  835.         int c   = s->comp_index[i];

  836.         data[c] = s->picture_ptr->data[c];

  837.         reference_data[c] = reference ? reference->data[c] : NULL;

  838.         linesize[c] = s->linesize[c];

  839.         s->coefs_finished[c] |= 1;

  840.         if (s->flipped) {

  841.             // picture should be flipped upside-down for this codec

  842.             int offset = (linesize[c] * (s->v_scount[i] *

  843.                          (8 * s->mb_height - ((s->height / s->v_max) & 7)) - 1));

  844.             data[c]           += offset;

  845.             reference_data[c] += offset;

  846.             linesize[c]       *= -1;

  847.         }

  848.     }

  849. 

  850.     for (mb_y = 0; mb_y < s->mb_height; mb_y++) {

  851.         for (mb_x = 0; mb_x < s->mb_width; mb_x++) {

  852.             const int copy_mb = mb_bitmask && !get_bits1(&mb_bitmask_gb);

  853. 

  854.             if (s->restart_interval && !s->restart_count)

  855.                 s->restart_count = s->restart_interval;

  856. 

  857.             if (get_bits_count(&s->gb)>s->gb.size_in_bits) {

  858.                 av_log(s->avctx, AV_LOG_ERROR, "overread %d\n",

  859.                        get_bits_count(&s->gb) - s->gb.size_in_bits);

  860.                 return -1;

  861.             }

  862.             for (i = 0; i < nb_components; i++) {

  863.                 uint8_t *ptr;

  864.                 int n, h, v, x, y, c, j;

  865.                 int block_offset;

  866.                 n = s->nb_blocks[i];

  867.                 c = s->comp_index[i];

  868.                 h = s->h_scount[i];

  869.                 v = s->v_scount[i];

  870.                 x = 0;

  871.                 y = 0;

  872.                 for (j = 0; j < n; j++) {

  873.                     block_offset = (((linesize[c] * (v * mb_y + y) * 8) +

  874.                                      (h * mb_x + x) * 8) >> s->avctx->lowres);

  875. 

  876.                     if (s->interlaced && s->bottom_field)

  877.                         block_offset += linesize[c] >> 1;

  878.                     ptr = data[c] + block_offset;

  879.                     if (!s->progressive) {

  880.                         if (copy_mb)

  881.                             mjpeg_copy_block(ptr, reference_data[c] + block_offset,

  882.                                              linesize[c], s->avctx->lowres);

  883.                         else {

  884.                             s->dsp.clear_block(s->block);

  885.                             if (decode_block(s, s->block, i,

  886.                                              s->dc_index[i], s->ac_index[i],

  887.                                              s->quant_matrixes[s->quant_index[c]]) < 0) {

  888.                                 av_log(s->avctx, AV_LOG_ERROR,

  889.                                        "error y=%d x=%d\n", mb_y, mb_x);

  890.                                 return -1;

  891.                             }

  892.                             s->dsp.idct_put(ptr, linesize[c], s->block);

  893.                         }

  894.                     } else {

  895.                         int block_idx  = s->block_stride[c] * (v * mb_y + y) +

  896.                                          (h * mb_x + x);

  897.                         DCTELEM *block = s->blocks[c][block_idx];

  898.                         if (Ah)

  899.                             block[0] += get_bits1(&s->gb) *

  900.                                         s->quant_matrixes[s->quant_index[c]][0] << Al;

  901.                         else if (decode_dc_progressive(s, block, i, s->dc_index[i],

  902.                                                        s->quant_matrixes[s->quant_index[c]],

  903.                                                        Al) < 0) {

  904.                             av_log(s->avctx, AV_LOG_ERROR,

  905.                                    "error y=%d x=%d\n", mb_y, mb_x);

  906.                             return -1;

  907.                         }

  908.                     }

  909.                     // av_log(s->avctx, AV_LOG_DEBUG, "mb: %d %d processed\n",

  910.                     //        mb_y, mb_x);

  911.                     // av_log(NULL, AV_LOG_DEBUG, "%d %d %d %d %d %d %d %d \n",

  912.                     //        mb_x, mb_y, x, y, c, s->bottom_field,

  913.                     //        (v * mb_y + y) * 8, (h * mb_x + x) * 8);

  914.                     if (++x == h) {

  915.                         x = 0;

  916.                         y++;

  917.                     }

  918.                 }

  919.             }

  920. 

  921.             if (s->restart_interval) {

  922.                 s->restart_count--;

  923.                 i = 8 + ((-get_bits_count(&s->gb)) & 7);

  924.                 /* skip RSTn */

  925.                 if (show_bits(&s->gb, i) == (1 << i) - 1) {

  926.                     int pos = get_bits_count(&s->gb);

  927.                     align_get_bits(&s->gb);

  928.                     while (get_bits_left(&s->gb) >= 8 && show_bits(&s->gb, 8) == 0xFF)

  929.                         skip_bits(&s->gb, 8);

  930.                     if ((get_bits(&s->gb, 8) & 0xF8) == 0xD0) {

  931.                         for (i = 0; i < nb_components; i++) /* reset dc */

  932.                             s->last_dc[i] = 1024;

  933.                     } else

  934.                         skip_bits_long(&s->gb, pos - get_bits_count(&s->gb));

  935.                 }

  936.             }

  937.         }

  938.     }

  939.     return 0;

  940. }

  941. 

  942. static int mjpeg_decode_scan_progressive_ac(MJpegDecodeContext *s, int ss,

  943.                                             int se, int Ah, int Al,

  944.                                             const uint8_t *mb_bitmask,

  945.                                             const AVFrame *reference)

  946. {

  947.     int mb_x, mb_y;

  948.     int EOBRUN = 0;

  949.     int c = s->comp_index[0];

  950.     uint8_t *data = s->picture_ptr->data[c];

  951.     const uint8_t *reference_data = reference ? reference->data[c] : NULL;

  952.     int linesize  = s->linesize[c];

  953.     int last_scan = 0;

  954.     int16_t *quant_matrix = s->quant_matrixes[s->quant_index[c]];

  955.     GetBitContext mb_bitmask_gb;

  956. 

  957.     if (mb_bitmask)

  958.         init_get_bits(&mb_bitmask_gb, mb_bitmask, s->mb_width * s->mb_height);

  959. 

  960.     if (!Al) {

  961.         s->coefs_finished[c] |= (1LL << (se + 1)) - (1LL << ss);

  962.         last_scan = !~s->coefs_finished[c];

  963.     }

  964. 

  965.     if (s->interlaced && s->bottom_field) {

  966.         int offset      = linesize >> 1;

  967.         data           += offset;

  968.         reference_data += offset;

  969.     }

  970. 

  971.     for (mb_y = 0; mb_y < s->mb_height; mb_y++) {

  972.         int block_offset = (mb_y * linesize * 8 >> s->avctx->lowres);

  973.         uint8_t *ptr     = data + block_offset;

  974.         int block_idx    = mb_y * s->block_stride[c];

  975.         DCTELEM (*block)[64] = &s->blocks[c][block_idx];

  976.         uint8_t *last_nnz    = &s->last_nnz[c][block_idx];

  977.         for (mb_x = 0; mb_x < s->mb_width; mb_x++, block++, last_nnz++) {

  978.             const int copy_mb = mb_bitmask && !get_bits1(&mb_bitmask_gb);

  979. 

  980.             if (!copy_mb) {

  981.                 int ret;

  982.                 if (Ah)

  983.                     ret = decode_block_refinement(s, *block, last_nnz, s->ac_index[0],

  984.                                                   quant_matrix, ss, se, Al, &EOBRUN);

  985.                 else

  986.                     ret = decode_block_progressive(s, *block, last_nnz, s->ac_index[0],

  987.                                                    quant_matrix, ss, se, Al, &EOBRUN);

  988.                 if (ret < 0) {

  989.                     av_log(s->avctx, AV_LOG_ERROR,

  990.                            "error y=%d x=%d\n", mb_y, mb_x);

  991.                     return -1;

  992.                 }

  993.             }

  994. 

  995.             if (last_scan) {

  996.                 if (copy_mb) {

  997.                     mjpeg_copy_block(ptr, reference_data + block_offset,

  998.                                      linesize, s->avctx->lowres);

  999.                 } else {

  1000.                     s->dsp.idct_put(ptr, linesize, *block);

  1001.                     ptr += 8 >> s->avctx->lowres;

  1002.                 }

  1003.             }

  1004.         }

  1005.     }

  1006.     return 0;

  1007. }

  1008. 

  1009. int ff_mjpeg_decode_sos(MJpegDecodeContext *s, const uint8_t *mb_bitmask,

  1010.                         const AVFrame *reference)

  1011. {

  1012.     int len, nb_components, i, h, v, predictor, point_transform;

  1013.     int index, id;

  1014.     const int block_size = s->lossless ? 1 : 8;

  1015.     int ilv, prev_shift;

  1016. 

  1017.     /* XXX: verify len field validity */

  1018.     len = get_bits(&s->gb, 16);

  1019.     nb_components = get_bits(&s->gb, 8);

  1020.     if (nb_components == 0 || nb_components > MAX_COMPONENTS) {

  1021.         av_log(s->avctx, AV_LOG_ERROR,

  1022.                "decode_sos: nb_components (%d) unsupported\n", nb_components);

  1023.         return -1;

  1024.     }

  1025.     if (len != 6 + 2 * nb_components) {

  1026.         av_log(s->avctx, AV_LOG_ERROR, "decode_sos: invalid len (%d)\n", len);

  1027.         return -1;

  1028.     }

  1029.     for (i = 0; i < nb_components; i++) {

  1030.         id = get_bits(&s->gb, 8) - 1;

  1031.         av_log(s->avctx, AV_LOG_DEBUG, "component: %d\n", id);

  1032.         /* find component index */

  1033.         for (index = 0; index < s->nb_components; index++)

  1034.             if (id == s->component_id[index])

  1035.                 break;

  1036.         if (index == s->nb_components) {

  1037.             av_log(s->avctx, AV_LOG_ERROR,

  1038.                    "decode_sos: index(%d) out of components\n", index);

  1039.             return -1;

  1040.         }

  1041.         /* Metasoft MJPEG codec has Cb and Cr swapped */

  1042.         if (s->avctx->codec_tag == MKTAG('M', 'T', 'S', 'J')

  1043.             && nb_components == 3 && s->nb_components == 3 && i)

  1044.             index = 3 - i;

  1045. 

  1046.         s->comp_index[i] = index;

  1047. 

  1048.         s->nb_blocks[i] = s->h_count[index] * s->v_count[index];

  1049.         s->h_scount[i]  = s->h_count[index];

  1050.         s->v_scount[i]  = s->v_count[index];

  1051. 

  1052.         s->dc_index[i] = get_bits(&s->gb, 4);

  1053.         s->ac_index[i] = get_bits(&s->gb, 4);

  1054. 

  1055.         if (s->dc_index[i] <  0 || s->ac_index[i] < 0 ||

  1056.             s->dc_index[i] >= 4 || s->ac_index[i] >= 4)

  1057.             goto out_of_range;

  1058.         if (!s->vlcs[0][s->dc_index[i]].table ||

  1059.             !s->vlcs[1][s->ac_index[i]].table)

  1060.             goto out_of_range;

  1061.     }

  1062. 

  1063.     predictor = get_bits(&s->gb, 8);       /* JPEG Ss / lossless JPEG predictor /JPEG-LS NEAR */

  1064.     ilv = get_bits(&s->gb, 8);             /* JPEG Se / JPEG-LS ILV */

  1065.     prev_shift      = get_bits(&s->gb, 4); /* Ah */

  1066.     point_transform = get_bits(&s->gb, 4); /* Al */

  1067. 

  1068.     for (i = 0; i < nb_components; i++)

  1069.         s->last_dc[i] = 1024;

  1070. 

  1071.     if (nb_components > 1) {

  1072.         /* interleaved stream */

  1073.         s->mb_width  = (s->width  + s->h_max * block_size - 1) / (s->h_max * block_size);

  1074.         s->mb_height = (s->height + s->v_max * block_size - 1) / (s->v_max * block_size);

  1075.     } else if (!s->ls) { /* skip this for JPEG-LS */

  1076.         h = s->h_max / s->h_scount[0];

  1077.         v = s->v_max / s->v_scount[0];

  1078.         s->mb_width     = (s->width  + h * block_size - 1) / (h * block_size);

  1079.         s->mb_height    = (s->height + v * block_size - 1) / (v * block_size);

  1080.         s->nb_blocks[0] = 1;

  1081.         s->h_scount[0]  = 1;

  1082.         s->v_scount[0]  = 1;

  1083.     }

  1084. 

  1085.     if (s->avctx->debug & FF_DEBUG_PICT_INFO)

  1086.         av_log(s->avctx, AV_LOG_DEBUG, "%s %s p:%d >>:%d ilv:%d bits:%d %s\n",

  1087.                s->lossless ? "lossless" : "sequential DCT", s->rgb ? "RGB" : "",

  1088.                predictor, point_transform, ilv, s->bits,

  1089.                s->pegasus_rct ? "PRCT" : (s->rct ? "RCT" : ""));

  1090. 

  1091. 

  1092.     /* mjpeg-b can have padding bytes between sos and image data, skip them */

  1093.     for (i = s->mjpb_skiptosod; i > 0; i--)

  1094.         skip_bits(&s->gb, 8);

  1095. 

  1096.     if (s->lossless) {

  1097.         if (CONFIG_JPEGLS_DECODER && s->ls) {

  1098. //            for () {

  1099. //            reset_ls_coding_parameters(s, 0);

  1100. 

  1101.             if (ff_jpegls_decode_picture(s, predictor, point_transform, ilv) < 0)

  1102.                 return -1;

  1103.         } else {

  1104.             if (s->rgb) {

  1105.                 if (ljpeg_decode_rgb_scan(s, predictor, point_transform) < 0)

  1106.                     return -1;

  1107.             } else {

  1108.                 if (ljpeg_decode_yuv_scan(s, predictor, point_transform) < 0)

  1109.                     return -1;

  1110.             }

  1111.         }

  1112.     } else {

  1113.         if (s->progressive && predictor) {

  1114.             if (mjpeg_decode_scan_progressive_ac(s, predictor, ilv, prev_shift,

  1115.                                                  point_transform,

  1116.                                                  mb_bitmask, reference) < 0)

  1117.                 return -1;

  1118.         } else {

  1119.             if (mjpeg_decode_scan(s, nb_components, prev_shift, point_transform,

  1120.                                   mb_bitmask, reference) < 0)

  1121.                 return -1;

  1122.         }

  1123.     }

  1124.     emms_c();

  1125.     return 0;

  1126.  out_of_range:

  1127.     av_log(s->avctx, AV_LOG_ERROR, "decode_sos: ac/dc index out of range\n");

  1128.     return -1;

  1129. }

  1130. 

  1131. static int mjpeg_decode_dri(MJpegDecodeContext *s)

  1132. {

  1133.     if (get_bits(&s->gb, 16) != 4)

  1134.         return -1;

  1135.     s->restart_interval = get_bits(&s->gb, 16);

  1136.     s->restart_count    = 0;

  1137.     av_log(s->avctx, AV_LOG_DEBUG, "restart interval: %d\n",

  1138.            s->restart_interval);

  1139. 

  1140.     return 0;

  1141. }

  1142. 

  1143. static int mjpeg_decode_app(MJpegDecodeContext *s)

  1144. {

  1145.     int len, id, i;

  1146. 

  1147.     len = get_bits(&s->gb, 16);

  1148.     if (len < 5)

  1149.         return -1;

  1150.     if (8 * len + get_bits_count(&s->gb) > s->gb.size_in_bits)

  1151.         return -1;

  1152. 

  1153.     id   = get_bits_long(&s->gb, 32);

  1154.     id   = av_be2ne32(id);

  1155.     len -= 6;

  1156. 

  1157.     if (s->avctx->debug & FF_DEBUG_STARTCODE)

  1158.         av_log(s->avctx, AV_LOG_DEBUG, "APPx %8X\n", id);

  1159. 

  1160.     /* Buggy AVID, it puts EOI only at every 10th frame. */

  1161.     /* Also, this fourcc is used by non-avid files too, it holds some

  1162.        information, but it's always present in AVID-created files. */

  1163.     if (id == AV_RL32("AVI1")) {

  1164.         /* structure:

  1165.             4bytes      AVI1

  1166.             1bytes      polarity

  1167.             1bytes      always zero

  1168.             4bytes      field_size

  1169.             4bytes      field_size_less_padding

  1170.         */

  1171.         s->buggy_avid = 1;

  1172. //      if (s->first_picture)

  1173. //          printf("mjpeg: workarounding buggy AVID\n");

  1174.         i = get_bits(&s->gb, 8);

  1175.         if (i == 2)

  1176.             s->bottom_field = 1;

  1177.         else if (i == 1)

  1178.             s->bottom_field = 0;

  1179. #if 0

  1180.         skip_bits(&s->gb, 8);

  1181.         skip_bits(&s->gb, 32);

  1182.         skip_bits(&s->gb, 32);

  1183.         len -= 10;

  1184. #endif

  1185. //        if (s->interlace_polarity)

  1186. //            printf("mjpeg: interlace polarity: %d\n", s->interlace_polarity);

  1187.         goto out;

  1188.     }

  1189. 

  1190. //    len -= 2;

  1191. 

  1192.     if (id == AV_RL32("JFIF")) {

  1193.         int t_w, t_h, v1, v2;

  1194.         skip_bits(&s->gb, 8); /* the trailing zero-byte */

  1195.         v1 = get_bits(&s->gb, 8);

  1196.         v2 = get_bits(&s->gb, 8);

  1197.         skip_bits(&s->gb, 8);

  1198. 

  1199.         s->avctx->sample_aspect_ratio.num = get_bits(&s->gb, 16);

  1200.         s->avctx->sample_aspect_ratio.den = get_bits(&s->gb, 16);

  1201. 

  1202.         if (s->avctx->debug & FF_DEBUG_PICT_INFO)

  1203.             av_log(s->avctx, AV_LOG_INFO,

  1204.                    "mjpeg: JFIF header found (version: %x.%x) SAR=%d/%d\n",

  1205.                    v1, v2,

  1206.                    s->avctx->sample_aspect_ratio.num,

  1207.                    s->avctx->sample_aspect_ratio.den);

  1208. 

  1209.         t_w = get_bits(&s->gb, 8);

  1210.         t_h = get_bits(&s->gb, 8);

  1211.         if (t_w && t_h) {

  1212.             /* skip thumbnail */

  1213.             if (len -10 - (t_w * t_h * 3) > 0)

  1214.                 len -= t_w * t_h * 3;

  1215.         }

  1216.         len -= 10;

  1217.         goto out;

  1218.     }

  1219. 

  1220.     if (id == AV_RL32("Adob") && (get_bits(&s->gb, 8) == 'e')) {

  1221.         if (s->avctx->debug & FF_DEBUG_PICT_INFO)

  1222.             av_log(s->avctx, AV_LOG_INFO, "mjpeg: Adobe header found\n");

  1223.         skip_bits(&s->gb, 16); /* version */

  1224.         skip_bits(&s->gb, 16); /* flags0 */

  1225.         skip_bits(&s->gb, 16); /* flags1 */

  1226.         skip_bits(&s->gb,  8); /* transform */

  1227.         len -= 7;

  1228.         goto out;

  1229.     }

  1230. 

  1231.     if (id == AV_RL32("LJIF")) {

  1232.         if (s->avctx->debug & FF_DEBUG_PICT_INFO)

  1233.             av_log(s->avctx, AV_LOG_INFO,

  1234.                    "Pegasus lossless jpeg header found\n");

  1235.         skip_bits(&s->gb, 16); /* version ? */

  1236.         skip_bits(&s->gb, 16); /* unknwon always 0? */

  1237.         skip_bits(&s->gb, 16); /* unknwon always 0? */

  1238.         skip_bits(&s->gb, 16); /* unknwon always 0? */

  1239.         switch (get_bits(&s->gb, 8)) {

  1240.         case 1:

  1241.             s->rgb         = 1;

  1242.             s->pegasus_rct = 0;

  1243.             break;

  1244.         case 2:

  1245.             s->rgb         = 1;

  1246.             s->pegasus_rct = 1;

  1247.             break;

  1248.         default:

  1249.             av_log(s->avctx, AV_LOG_ERROR, "unknown colorspace\n");

  1250.         }

  1251.         len -= 9;

  1252.         goto out;

  1253.     }

  1254. 

  1255.     /* Apple MJPEG-A */

  1256.     if ((s->start_code == APP1) && (len > (0x28 - 8))) {

  1257.         id   = get_bits_long(&s->gb, 32);

  1258.         id   = av_be2ne32(id);

  1259.         len -= 4;

  1260.         /* Apple MJPEG-A */

  1261.         if (id == AV_RL32("mjpg")) {

  1262. #if 0

  1263.             skip_bits(&s->gb, 32); /* field size */

  1264.             skip_bits(&s->gb, 32); /* pad field size */

  1265.             skip_bits(&s->gb, 32); /* next off */

  1266.             skip_bits(&s->gb, 32); /* quant off */

  1267.             skip_bits(&s->gb, 32); /* huff off */

  1268.             skip_bits(&s->gb, 32); /* image off */

  1269.             skip_bits(&s->gb, 32); /* scan off */

  1270.             skip_bits(&s->gb, 32); /* data off */

  1271. #endif

  1272.             if (s->avctx->debug & FF_DEBUG_PICT_INFO)

  1273.                 av_log(s->avctx, AV_LOG_INFO, "mjpeg: Apple MJPEG-A header found\n");

  1274.         }

  1275.     }

  1276. 

  1277. out:

  1278.     /* slow but needed for extreme adobe jpegs */

  1279.     if (len < 0)

  1280.         av_log(s->avctx, AV_LOG_ERROR,

  1281.                "mjpeg: error, decode_app parser read over the end\n");

  1282.     while (--len > 0)

  1283.         skip_bits(&s->gb, 8);

  1284. 

  1285.     return 0;

  1286. }

  1287. 

  1288. static int mjpeg_decode_com(MJpegDecodeContext *s)

  1289. {

  1290.     int len = get_bits(&s->gb, 16);

  1291.     if (len >= 2 &&

  1292.         8 * len - 16 + get_bits_count(&s->gb) <= s->gb.size_in_bits) {

  1293.         char *cbuf = av_malloc(len - 1);

  1294.         if (cbuf) {

  1295.             int i;

  1296.             for (i = 0; i < len - 2; i++)

  1297.                 cbuf[i] = get_bits(&s->gb, 8);

  1298.             if (i > 0 && cbuf[i - 1] == '\n')

  1299.                 cbuf[i - 1] = 0;

  1300.             else

  1301.                 cbuf[i] = 0;

  1302. 

  1303.             if (s->avctx->debug & FF_DEBUG_PICT_INFO)

  1304.                 av_log(s->avctx, AV_LOG_INFO, "mjpeg comment: '%s'\n", cbuf);

  1305. 

  1306.             /* buggy avid, it puts EOI only at every 10th frame */

  1307.             if (!strcmp(cbuf, "AVID")) {

  1308.                 s->buggy_avid = 1;

  1309.                 // if (s->first_picture)

  1310.                 // printf("mjpeg: workarounding buggy AVID\n");

  1311.             } else if (!strcmp(cbuf, "CS=ITU601"))

  1312.                 s->cs_itu601 = 1;

  1313.             else if ((len > 20 && !strncmp(cbuf, "Intel(R) JPEG Library", 21)) ||

  1314.                      (len > 19 && !strncmp(cbuf, "Metasoft MJPEG Codec", 20)))

  1315.                 s->flipped = 1;

  1316. 

  1317.             av_free(cbuf);

  1318.         }

  1319.     }

  1320. 

  1321.     return 0;

  1322. }

  1323. 

  1324. /* return the 8 bit start code value and update the search

  1325.    state. Return -1 if no start code found */

  1326. static int find_marker(const uint8_t **pbuf_ptr, const uint8_t *buf_end)

  1327. {

  1328.     const uint8_t *buf_ptr;

  1329.     unsigned int v, v2;

  1330.     int val;

  1331. #ifdef DEBUG

  1332.     int skipped = 0;

  1333. #endif

  1334. 

  1335.     buf_ptr = *pbuf_ptr;

  1336.     while (buf_ptr < buf_end) {

  1337.         v  = *buf_ptr++;

  1338.         v2 = *buf_ptr;

  1339.         if ((v == 0xff) && (v2 >= 0xc0) && (v2 <= 0xfe) && buf_ptr < buf_end) {

  1340.             val = *buf_ptr++;

  1341.             goto found;

  1342.         }

  1343. #ifdef DEBUG

  1344.         skipped++;

  1345. #endif

  1346.     }

  1347.     val = -1;

  1348. found:

  1349.     av_dlog(NULL, "find_marker skipped %d bytes\n", skipped);

  1350.     *pbuf_ptr = buf_ptr;

  1351.     return val;

  1352. }

  1353. 

  1354. int ff_mjpeg_find_marker(MJpegDecodeContext *s,

  1355.                          const uint8_t **buf_ptr, const uint8_t *buf_end,

  1356.                          const uint8_t **unescaped_buf_ptr,

  1357.                          int *unescaped_buf_size)

  1358. {

  1359.     int start_code;

  1360.     start_code = find_marker(buf_ptr, buf_end);

  1361. 

  1362.     av_fast_padded_malloc(&s->buffer, &s->buffer_size, buf_end - *buf_ptr);

  1363.     if (!s->buffer)

  1364.         return AVERROR(ENOMEM);

  1365. 

  1366.     /* unescape buffer of SOS, use special treatment for JPEG-LS */

  1367.     if (start_code == SOS && !s->ls) {

  1368.         const uint8_t *src = *buf_ptr;

  1369.         uint8_t *dst = s->buffer;

  1370. 

  1371.         while (src < buf_end) {

  1372.             uint8_t x = *(src++);

  1373. 

  1374.             *(dst++) = x;

  1375.             if (s->avctx->codec_id != CODEC_ID_THP) {

  1376.                 if (x == 0xff) {

  1377.                     while (src < buf_end && x == 0xff)

  1378.                         x = *(src++);

  1379. 

  1380.                     if (x >= 0xd0 && x <= 0xd7)

  1381.                         *(dst++) = x;

  1382.                     else if (x)

  1383.                         break;

  1384.                 }

  1385.             }

  1386.         }

  1387.         *unescaped_buf_ptr  = s->buffer;

  1388.         *unescaped_buf_size = dst - s->buffer;

  1389. 

  1390.         av_log(s->avctx, AV_LOG_DEBUG, "escaping removed %td bytes\n",

  1391.                (buf_end - *buf_ptr) - (dst - s->buffer));

  1392.     } else if (start_code == SOS && s->ls) {

  1393.         const uint8_t *src = *buf_ptr;

  1394.         uint8_t *dst  = s->buffer;

  1395.         int bit_count = 0;

  1396.         int t = 0, b = 0;

  1397.         PutBitContext pb;

  1398. 

  1399.         s->cur_scan++;

  1400. 

  1401.         /* find marker */

  1402.         while (src + t < buf_end) {

  1403.             uint8_t x = src[t++];

  1404.             if (x == 0xff) {

  1405.                 while ((src + t < buf_end) && x == 0xff)

  1406.                     x = src[t++];

  1407.                 if (x & 0x80) {

  1408.                     t -= 2;

  1409.                     break;

  1410.                 }

  1411.             }

  1412.         }

  1413.         bit_count = t * 8;

  1414.         init_put_bits(&pb, dst, t);

  1415. 

  1416.         /* unescape bitstream */

  1417.         while (b < t) {

  1418.             uint8_t x = src[b++];

  1419.             put_bits(&pb, 8, x);

  1420.             if (x == 0xFF) {

  1421.                 x = src[b++];

  1422.                 put_bits(&pb, 7, x);

  1423.                 bit_count--;

  1424.             }

  1425.         }

  1426.         flush_put_bits(&pb);

  1427. 

  1428.         *unescaped_buf_ptr  = dst;

  1429.         *unescaped_buf_size = (bit_count + 7) >> 3;

  1430.     } else {

  1431.         *unescaped_buf_ptr  = *buf_ptr;

  1432.         *unescaped_buf_size = buf_end - *buf_ptr;

  1433.     }

  1434. 

  1435.     return start_code;

  1436. }

  1437. 

  1438. int ff_mjpeg_decode_frame(AVCodecContext *avctx, void *data, int *data_size,

  1439.                           AVPacket *avpkt)

  1440. {

  1441.     const uint8_t *buf = avpkt->data;

  1442.     int buf_size       = avpkt->size;

  1443.     MJpegDecodeContext *s = avctx->priv_data;

  1444.     const uint8_t *buf_end, *buf_ptr;

  1445.     const uint8_t *unescaped_buf_ptr;

  1446.     int unescaped_buf_size;

  1447.     int start_code;

  1448.     AVFrame *picture = data;

  1449. 

  1450.     s->got_picture = 0; // picture from previous image can not be reused

  1451.     buf_ptr = buf;

  1452.     buf_end = buf + buf_size;

  1453.     while (buf_ptr < buf_end) {

  1454.         /* find start next marker */

  1455.         start_code = ff_mjpeg_find_marker(s, &buf_ptr, buf_end,

  1456.                                           &unescaped_buf_ptr,

  1457.                                           &unescaped_buf_size);

  1458.         /* EOF */

  1459.         if (start_code < 0) {

  1460.             goto the_end;

  1461.         } else {

  1462.             av_log(avctx, AV_LOG_DEBUG, "marker=%x avail_size_in_buf=%td\n",

  1463.                    start_code, buf_end - buf_ptr);

  1464. 

  1465.             init_get_bits(&s->gb, unescaped_buf_ptr, unescaped_buf_size * 8);

  1466. 

  1467.             s->start_code = start_code;

  1468.             if (s->avctx->debug & FF_DEBUG_STARTCODE)

  1469.                 av_log(avctx, AV_LOG_DEBUG, "startcode: %X\n", start_code);

  1470. 

  1471.             /* process markers */

  1472.             if (start_code >= 0xd0 && start_code <= 0xd7)

  1473.                 av_log(avctx, AV_LOG_DEBUG,

  1474.                        "restart marker: %d\n", start_code & 0x0f);

  1475.                 /* APP fields */

  1476.             else if (start_code >= APP0 && start_code <= APP15)

  1477.                 mjpeg_decode_app(s);

  1478.                 /* Comment */

  1479.             else if (start_code == COM)

  1480.                 mjpeg_decode_com(s);

  1481. 

  1482.             switch (start_code) {

  1483.             case SOI:

  1484.                 s->restart_interval = 0;

  1485.                 s->restart_count    = 0;

  1486.                 /* nothing to do on SOI */

  1487.                 break;

  1488.             case DQT:

  1489.                 ff_mjpeg_decode_dqt(s);

  1490.                 break;

  1491.             case DHT:

  1492.                 if (ff_mjpeg_decode_dht(s) < 0) {

  1493.                     av_log(avctx, AV_LOG_ERROR, "huffman table decode error\n");

  1494.                     return -1;

  1495.                 }

  1496.                 break;

  1497.             case SOF0:

  1498.             case SOF1:

  1499.                 s->lossless    = 0;

  1500.                 s->ls          = 0;

  1501.                 s->progressive = 0;

  1502.                 if (ff_mjpeg_decode_sof(s) < 0)

  1503.                     return -1;

  1504.                 break;

  1505.             case SOF2:

  1506.                 s->lossless    = 0;

  1507.                 s->ls          = 0;

  1508.                 s->progressive = 1;

  1509.                 if (ff_mjpeg_decode_sof(s) < 0)

  1510.                     return -1;

  1511.                 break;

  1512.             case SOF3:

  1513.                 s->lossless    = 1;

  1514.                 s->ls          = 0;

  1515.                 s->progressive = 0;

  1516.                 if (ff_mjpeg_decode_sof(s) < 0)

  1517.                     return -1;

  1518.                 break;

  1519.             case SOF48:

  1520.                 s->lossless    = 1;

  1521.                 s->ls          = 1;

  1522.                 s->progressive = 0;

  1523.                 if (ff_mjpeg_decode_sof(s) < 0)

  1524.                     return -1;

  1525.                 break;

  1526.             case LSE:

  1527.                 if (!CONFIG_JPEGLS_DECODER || ff_jpegls_decode_lse(s) < 0)

  1528.                     return -1;

  1529.                 break;

  1530.             case EOI:

  1531.                 s->cur_scan = 0;

  1532.                 if ((s->buggy_avid && !s->interlaced) || s->restart_interval)

  1533.                     break;

  1534. eoi_parser:

  1535.                 if (!s->got_picture) {

  1536.                     av_log(avctx, AV_LOG_WARNING,

  1537.                            "Found EOI before any SOF, ignoring\n");

  1538.                     break;

  1539.                     }

  1540.                 if (s->interlaced) {

  1541.                     s->bottom_field ^= 1;

  1542.                     /* if not bottom field, do not output image yet */

  1543.                     if (s->bottom_field == !s->interlace_polarity)

  1544.                         goto not_the_end;

  1545.                     }

  1546.                     *picture   = *s->picture_ptr;

  1547.                     *data_size = sizeof(AVFrame);

  1548. 

  1549.                     if (!s->lossless) {

  1550.                         picture->quality      = FFMAX3(s->qscale[0],

  1551.                                                        s->qscale[1],

  1552.                                                        s->qscale[2]);

  1553.                         picture->qstride      = 0;

  1554.                         picture->qscale_table = s->qscale_table;

  1555.                         memset(picture->qscale_table, picture->quality,

  1556.                                (s->width + 15) / 16);

  1557.                         if (avctx->debug & FF_DEBUG_QP)

  1558.                             av_log(avctx, AV_LOG_DEBUG,

  1559.                                    "QP: %d\n", picture->quality);

  1560.                         picture->quality *= FF_QP2LAMBDA;

  1561.                     }

  1562. 

  1563.                 goto the_end;

  1564.             case SOS:

  1565.                 if (!s->got_picture) {

  1566.                     av_log(avctx, AV_LOG_WARNING,

  1567.                            "Can not process SOS before SOF, skipping\n");

  1568.                     break;

  1569.                     }

  1570.                 if (ff_mjpeg_decode_sos(s, NULL, NULL) < 0 &&

  1571.                     (avctx->err_recognition & AV_EF_EXPLODE))

  1572.                     return AVERROR_INVALIDDATA;

  1573.                 /* buggy avid puts EOI every 10-20th frame */

  1574.                 /* if restart period is over process EOI */

  1575.                 if ((s->buggy_avid && !s->interlaced) || s->restart_interval)

  1576.                     goto eoi_parser;

  1577.                 break;

  1578.             case DRI:

  1579.                 mjpeg_decode_dri(s);

  1580.                 break;

  1581.             case SOF5:

  1582.             case SOF6:

  1583.             case SOF7:

  1584.             case SOF9:

  1585.             case SOF10:

  1586.             case SOF11:

  1587.             case SOF13:

  1588.             case SOF14:

  1589.             case SOF15:

  1590.             case JPG:

  1591.                 av_log(avctx, AV_LOG_ERROR,

  1592.                        "mjpeg: unsupported coding type (%x)\n", start_code);

  1593.                 break;

  1594. //              default:

  1595. //              printf("mjpeg: unsupported marker (%x)\n", start_code);

  1596. //              break;

  1597.             }

  1598. 

  1599. not_the_end:

  1600.             /* eof process start code */

  1601.             buf_ptr += (get_bits_count(&s->gb) + 7) / 8;

  1602.             av_log(avctx, AV_LOG_DEBUG,

  1603.                    "marker parser used %d bytes (%d bits)\n",

  1604.                    (get_bits_count(&s->gb) + 7) / 8, get_bits_count(&s->gb));

  1605.         }

  1606.     }

  1607.     if (s->got_picture) {

  1608.         av_log(avctx, AV_LOG_WARNING, "EOI missing, emulating\n");

  1609.         goto eoi_parser;

  1610.     }

  1611.     av_log(avctx, AV_LOG_FATAL, "No JPEG data found in image\n");

  1612.     return -1;

  1613. the_end:

  1614.     av_log(avctx, AV_LOG_DEBUG, "mjpeg decode frame unused %td bytes\n",

  1615.            buf_end - buf_ptr);

  1616. //  return buf_end - buf_ptr;

  1617.     return buf_ptr - buf;

  1618. }

  1619. 

  1620. av_cold int ff_mjpeg_decode_end(AVCodecContext *avctx)

  1621. {

  1622.     MJpegDecodeContext *s = avctx->priv_data;

  1623.     int i, j;

  1624. 

  1625.     if (s->picture_ptr && s->picture_ptr->data[0])

  1626.         avctx->release_buffer(avctx, s->picture_ptr);

  1627. 

  1628.     av_free(s->buffer);

  1629.     av_free(s->qscale_table);

  1630.     av_freep(&s->ljpeg_buffer);

  1631.     s->ljpeg_buffer_size = 0;

  1632. 

  1633.     for (i = 0; i < 3; i++) {

  1634.         for (j = 0; j < 4; j++)

  1635.             ff_free_vlc(&s->vlcs[i][j]);

  1636.     }

  1637.     for (i = 0; i < MAX_COMPONENTS; i++) {

  1638.         av_freep(&s->blocks[i]);

  1639.         av_freep(&s->last_nnz[i]);

  1640.     }

  1641.     return 0;

  1642. }

  1643. 

  1644. #define OFFSET(x) offsetof(MJpegDecodeContext, x)

  1645. #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM

  1646. static const AVOption options[] = {

  1647.     { "extern_huff", "Use external huffman table.",

  1648.       OFFSET(extern_huff), AV_OPT_TYPE_INT, { 0 }, 0, 1, VD },

  1649.     { NULL },

  1650. };

  1651. 

  1652. static const AVClass mjpegdec_class = {

  1653.     .class_name = "MJPEG decoder",

  1654.     .item_name  = av_default_item_name,

  1655.     .option     = options,

  1656.     .version    = LIBAVUTIL_VERSION_INT,

  1657. };

  1658. 

  1659. AVCodec ff_mjpeg_decoder = {

  1660.     .name           = "mjpeg",

  1661.     .type           = AVMEDIA_TYPE_VIDEO,

  1662.     .id             = CODEC_ID_MJPEG,

  1663.     .priv_data_size = sizeof(MJpegDecodeContext),

  1664.     .init           = ff_mjpeg_decode_init,

  1665.     .close          = ff_mjpeg_decode_end,

  1666.     .decode         = ff_mjpeg_decode_frame,

  1667.     .capabilities   = CODEC_CAP_DR1,

  1668.     .max_lowres     = 3,

  1669.     .long_name      = NULL_IF_CONFIG_SMALL("MJPEG (Motion JPEG)"),

  1670.     .priv_class     = &mjpegdec_class,

  1671. };

  1672. 

  1673. AVCodec ff_thp_decoder = {

  1674.     .name           = "thp",

  1675.     .type           = AVMEDIA_TYPE_VIDEO,

  1676.     .id             = CODEC_ID_THP,

  1677.     .priv_data_size = sizeof(MJpegDecodeContext),

  1678.     .init           = ff_mjpeg_decode_init,

  1679.     .close          = ff_mjpeg_decode_end,

  1680.     .decode         = ff_mjpeg_decode_frame,

  1681.     .capabilities   = CODEC_CAP_DR1,

  1682.     .max_lowres     = 3,

  1683.     .long_name      = NULL_IF_CONFIG_SMALL("Nintendo Gamecube THP video"),

  1684. };