FFmpeg
mjpegdec.c
Go to the documentation of this file.
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 FFmpeg.
12  *
13  * FFmpeg 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  * FFmpeg 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 FFmpeg; 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 #include "config_components.h"
34 
35 #include "libavutil/attributes.h"
36 #include "libavutil/imgutils.h"
37 #include "libavutil/avassert.h"
38 #include "libavutil/mem.h"
39 #include "libavutil/opt.h"
40 #include "avcodec.h"
41 #include "blockdsp.h"
42 #include "codec_internal.h"
43 #include "copy_block.h"
44 #include "decode.h"
45 #include "exif.h"
46 #include "hwaccel_internal.h"
47 #include "hwconfig.h"
48 #include "idctdsp.h"
49 #include "internal.h"
50 #include "jpegtables.h"
51 #include "mjpeg.h"
52 #include "mjpegdec.h"
53 #include "jpeglsdec.h"
54 #include "profiles.h"
55 #include "put_bits.h"
56 
57 
58 static void mjpeg_find_raw_scan_data(MJpegDecodeContext *s,
59  const uint8_t **pbuf_ptr, size_t *pbuf_size);
60 
61 static int init_default_huffman_tables(MJpegDecodeContext *s)
62 {
63  static const struct {
64  int class;
65  int index;
66  const uint8_t *bits;
67  const uint8_t *values;
68  int length;
69  } ht[] = {
70  { 0, 0, ff_mjpeg_bits_dc_luminance,
71  ff_mjpeg_val_dc, 12 },
72  { 0, 1, ff_mjpeg_bits_dc_chrominance,
73  ff_mjpeg_val_dc, 12 },
74  { 1, 0, ff_mjpeg_bits_ac_luminance,
75  ff_mjpeg_val_ac_luminance, 162 },
76  { 1, 1, ff_mjpeg_bits_ac_chrominance,
77  ff_mjpeg_val_ac_chrominance, 162 },
78  { 2, 0, ff_mjpeg_bits_ac_luminance,
79  ff_mjpeg_val_ac_luminance, 162 },
80  { 2, 1, ff_mjpeg_bits_ac_chrominance,
81  ff_mjpeg_val_ac_chrominance, 162 },
82  };
83  int i, ret;
84 
85  for (i = 0; i < FF_ARRAY_ELEMS(ht); i++) {
86  ff_vlc_free(&s->vlcs[ht[i].class][ht[i].index]);
87  ret = ff_mjpeg_build_vlc(&s->vlcs[ht[i].class][ht[i].index],
88  ht[i].bits, ht[i].values,
89  ht[i].class == 1, s->avctx);
90  if (ret < 0)
91  return ret;
92 
93  if (ht[i].class < 2) {
94  memcpy(s->raw_huffman_lengths[ht[i].class][ht[i].index],
95  ht[i].bits + 1, 16);
96  memcpy(s->raw_huffman_values[ht[i].class][ht[i].index],
97  ht[i].values, ht[i].length);
98  }
99  }
100 
101  return 0;
102 }
103 
104 static void parse_avid(MJpegDecodeContext *s, uint8_t *buf, int len)
105 {
106  if (len > 12 && buf[12] == 1) /* 1 - NTSC */
107  s->interlace_polarity = 1;
108  if (len > 12 && buf[12] == 2) /* 2 - PAL */
109  s->interlace_polarity = 0;
110  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
111  av_log(s->avctx, AV_LOG_INFO, "AVID: len:%d %d\n", len, len > 12 ? buf[12] : -1);
112 }
113 
114 static void init_idct(AVCodecContext *avctx)
115 {
116  MJpegDecodeContext *s = avctx->priv_data;
117 
118  ff_idctdsp_init(&s->idsp, avctx);
119  ff_permute_scantable(s->permutated_scantable, ff_zigzag_direct,
120  s->idsp.idct_permutation);
121 }
122 
123 av_cold int ff_mjpeg_decode_init(AVCodecContext *avctx)
124 {
125  MJpegDecodeContext *s = avctx->priv_data;
126  int ret;
127 
128  if (!s->picture_ptr) {
129  s->picture = av_frame_alloc();
130  if (!s->picture)
131  return AVERROR(ENOMEM);
132  s->picture_ptr = s->picture;
133  }
134 
135  s->avctx = avctx;
136  ff_blockdsp_init(&s->bdsp);
137  init_idct(avctx);
138  s->buffer_size = 0;
139  s->buffer = NULL;
140  s->first_picture = 1;
141  s->got_picture = 0;
142  s->orig_height = avctx->coded_height;
143  avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
144  avctx->colorspace = AVCOL_SPC_BT470BG;
145  s->hwaccel_pix_fmt = s->hwaccel_sw_pix_fmt = AV_PIX_FMT_NONE;
146 
147  if ((ret = init_default_huffman_tables(s)) < 0)
148  return ret;
149 
150  if (s->extern_huff && avctx->extradata) {
151  av_log(avctx, AV_LOG_INFO, "using external huffman table\n");
152  bytestream2_init(&s->gB, avctx->extradata, avctx->extradata_size);
153  if (ff_mjpeg_decode_dht(s)) {
154  av_log(avctx, AV_LOG_ERROR,
155  "error using external huffman table, switching back to internal\n");
156  if ((ret = init_default_huffman_tables(s)) < 0)
157  return ret;
158  }
159  }
160  if (avctx->field_order == AV_FIELD_BB) { /* quicktime icefloe 019 */
161  s->interlace_polarity = 1; /* bottom field first */
162  av_log(avctx, AV_LOG_DEBUG, "bottom field first\n");
163  } else if (avctx->field_order == AV_FIELD_UNKNOWN) {
164  if (avctx->codec_tag == AV_RL32("MJPG"))
165  s->interlace_polarity = 1;
166  }
167 
168  if (avctx->codec_id == AV_CODEC_ID_SMVJPEG) {
169  if (avctx->extradata_size >= 4)
170  s->smv_frames_per_jpeg = AV_RL32(avctx->extradata);
171 
172  if (s->smv_frames_per_jpeg <= 0) {
173  av_log(avctx, AV_LOG_ERROR, "Invalid number of frames per jpeg.\n");
174  return AVERROR_INVALIDDATA;
175  }
176 
177  s->smv_frame = av_frame_alloc();
178  if (!s->smv_frame)
179  return AVERROR(ENOMEM);
180  } else if (avctx->extradata_size > 8
181  && AV_RL32(avctx->extradata) == 0x2C
182  && AV_RL32(avctx->extradata + 4) == 0x18) {
183  parse_avid(s, avctx->extradata, avctx->extradata_size);
184  }
185 
186  if (avctx->codec->id == AV_CODEC_ID_AMV)
187  s->flipped = 1;
188 
189  return 0;
190 }
191 
192 
193 static int mjpeg_parse_len(MJpegDecodeContext *s, int *plen, const char *name)
194 {
195  int len = bytestream2_get_be16u(&s->gB);
196  if (len < 2 || bytestream2_get_bytes_left(&s->gB) < (len - 2)) {
197  av_log(s->avctx, AV_LOG_ERROR, "%s: invalid len %d\n", name, len);
198  return AVERROR_INVALIDDATA;
199  }
200  *plen = len - 2;
201  return 0;
202 }
203 
204 /* quantize tables */
205 int ff_mjpeg_decode_dqt(MJpegDecodeContext *s)
206 {
207  int len, index, i;
208 
209  int ret = mjpeg_parse_len(s, &len, "dqt");
210  if (ret < 0)
211  return ret;
212 
213  while (len >= 65) {
214  uint8_t b = bytestream2_get_byteu(&s->gB);
215  int pr = b >> 4;
216  if (pr > 1) {
217  av_log(s->avctx, AV_LOG_ERROR, "dqt: invalid precision\n");
218  return AVERROR_INVALIDDATA;
219  }
220  if (len < (1 + 64 * (1 + pr)))
221  return AVERROR_INVALIDDATA;
222  index = b & 0x0F;
223  if (index >= 4)
224  return AVERROR_INVALIDDATA;
225  av_log(s->avctx, AV_LOG_DEBUG, "index=%d\n", index);
226  /* read quant table */
227  for (i = 0; i < 64; i++) {
228  s->quant_matrixes[index][i] = pr ? bytestream2_get_be16u(&s->gB) : bytestream2_get_byteu(&s->gB);
229  if (s->quant_matrixes[index][i] == 0) {
230  int log_level = s->avctx->err_recognition & AV_EF_EXPLODE ? AV_LOG_ERROR : AV_LOG_WARNING;
231  av_log(s->avctx, log_level, "dqt: 0 quant value\n");
232  if (s->avctx->err_recognition & AV_EF_EXPLODE)
233  return AVERROR_INVALIDDATA;
234  }
235  }
236 
237  // XXX FIXME fine-tune, and perhaps add dc too
238  s->qscale[index] = FFMAX(s->quant_matrixes[index][1],
239  s->quant_matrixes[index][8]) >> 1;
240  av_log(s->avctx, AV_LOG_DEBUG, "qscale[%d]: %d\n",
241  index, s->qscale[index]);
242  len -= 1 + 64 * (1 + pr);
243  }
244  return 0;
245 }
246 
247 /* decode huffman tables and build VLC decoders */
248 int ff_mjpeg_decode_dht(MJpegDecodeContext *s)
249 {
250  int len, index, i, class, n, v;
251  uint8_t bits_table[17];
252  uint8_t val_table[256];
253  int ret = 0;
254 
255  ret = mjpeg_parse_len(s, &len, "dht");
256  if (ret < 0)
257  return ret;
258 
259  while (len > 0) {
260  if (len < 17)
261  return AVERROR_INVALIDDATA;
262  uint8_t b = bytestream2_get_byteu(&s->gB);
263  class = b >> 4;
264  if (class >= 2)
265  return AVERROR_INVALIDDATA;
266  index = b & 0x0F;
267  if (index >= 4)
268  return AVERROR_INVALIDDATA;
269  n = 0;
270  for (i = 1; i <= 16; i++) {
271  bits_table[i] = bytestream2_get_byteu(&s->gB);
272  n += bits_table[i];
273  }
274  len -= 17;
275  if (len < n || n > 256)
276  return AVERROR_INVALIDDATA;
277 
278  for (i = 0; i < n; i++) {
279  v = bytestream2_get_byteu(&s->gB);
280  val_table[i] = v;
281  }
282  len -= n;
283 
284  /* build VLC and flush previous vlc if present */
285  ff_vlc_free(&s->vlcs[class][index]);
286  av_log(s->avctx, AV_LOG_DEBUG, "class=%d index=%d nb_codes=%d\n",
287  class, index, n);
288  if ((ret = ff_mjpeg_build_vlc(&s->vlcs[class][index], bits_table,
289  val_table, class > 0, s->avctx)) < 0)
290  return ret;
291 
292  if (class > 0) {
293  ff_vlc_free(&s->vlcs[2][index]);
294  if ((ret = ff_mjpeg_build_vlc(&s->vlcs[2][index], bits_table,
295  val_table, 0, s->avctx)) < 0)
296  return ret;
297  }
298 
299  for (i = 0; i < 16; i++)
300  s->raw_huffman_lengths[class][index][i] = bits_table[i + 1];
301  for (i = 0; i < 256; i++)
302  s->raw_huffman_values[class][index][i] = val_table[i];
303  }
304  return 0;
305 }
306 
307 int ff_mjpeg_decode_sof(MJpegDecodeContext *s)
308 {
309  int len, nb_components, i, width, height, bits, ret, size_change;
310  unsigned pix_fmt_id;
311  int h_count[MAX_COMPONENTS] = { 0 };
312  int v_count[MAX_COMPONENTS] = { 0 };
313 
314  s->cur_scan = 0;
315  memset(s->upscale_h, 0, sizeof(s->upscale_h));
316  memset(s->upscale_v, 0, sizeof(s->upscale_v));
317 
318  ret = mjpeg_parse_len(s, &len, "sof");
319  if (ret < 0)
320  return ret;
321  if (len < 6)
322  return AVERROR_INVALIDDATA;
323  bits = bytestream2_get_byteu(&s->gB);
324 
325  if (bits > 16 || bits < 1) {
326  av_log(s->avctx, AV_LOG_ERROR, "bits %d is invalid\n", bits);
327  return AVERROR_INVALIDDATA;
328  }
329 
330  if (s->avctx->bits_per_raw_sample != bits) {
331  av_log(s->avctx, s->avctx->bits_per_raw_sample > 0 ? AV_LOG_INFO : AV_LOG_DEBUG, "Changing bps from %d to %d\n", s->avctx->bits_per_raw_sample, bits);
332  s->avctx->bits_per_raw_sample = bits;
333  init_idct(s->avctx);
334  }
335  if (s->pegasus_rct)
336  bits = 9;
337  if (bits == 9 && !s->pegasus_rct)
338  s->rct = 1; // FIXME ugly
339 
340  if (s->lossless && s->avctx->lowres) {
341  av_log(s->avctx, AV_LOG_ERROR, "lowres is not possible with lossless jpeg\n");
342  return AVERROR(ENOSYS);
343  }
344 
345  height = bytestream2_get_be16u(&s->gB);
346  width = bytestream2_get_be16u(&s->gB);
347 
348  // HACK for odd_height.mov
349  if (s->interlaced && s->width == width && s->height == height + 1)
350  height = s->height;
351 
352  av_log(s->avctx, AV_LOG_DEBUG, "sof0: picture: %dx%d\n", width, height);
353  if (av_image_check_size(width, height, 0, s->avctx) < 0)
354  return AVERROR_INVALIDDATA;
355 
356  if (!s->progressive && !s->ls) {
357  // A valid frame requires at least 1 bit for DC + 1 bit for AC for each 8x8 block.
358  if (s->buf_size && (width + 7) / 8 * ((height + 7) / 8) > s->buf_size * 4LL)
359  return AVERROR_INVALIDDATA;
360  }
361 
362  nb_components = bytestream2_get_byteu(&s->gB);
363  if (nb_components <= 0 ||
364  nb_components > MAX_COMPONENTS)
365  return AVERROR_INVALIDDATA;
366  if (s->interlaced && (s->bottom_field == !s->interlace_polarity)) {
367  if (nb_components != s->nb_components) {
368  av_log(s->avctx, AV_LOG_ERROR,
369  "nb_components changing in interlaced picture\n");
370  return AVERROR_INVALIDDATA;
371  }
372  }
373  if (s->ls && !(bits <= 8 || nb_components == 1)) {
374  avpriv_report_missing_feature(s->avctx,
375  "JPEG-LS that is not <= 8 "
376  "bits/component or 16-bit gray");
377  return AVERROR_PATCHWELCOME;
378  }
379  len -= 6;
380  if (len != 3 * nb_components) {
381  av_log(s->avctx, AV_LOG_ERROR, "decode_sof0: error, len(%d) mismatch %d components\n", len, nb_components);
382  return AVERROR_INVALIDDATA;
383  }
384 
385  s->nb_components = nb_components;
386  s->h_max = 1;
387  s->v_max = 1;
388  for (i = 0; i < nb_components; i++) {
389  /* component id */
390  s->component_id[i] = bytestream2_get_byteu(&s->gB);
391  uint8_t b = bytestream2_get_byteu(&s->gB);
392  h_count[i] = b >> 4;
393  v_count[i] = b & 0x0F;
394  /* compute hmax and vmax (only used in interleaved case) */
395  if (h_count[i] > s->h_max)
396  s->h_max = h_count[i];
397  if (v_count[i] > s->v_max)
398  s->v_max = v_count[i];
399  s->quant_index[i] = bytestream2_get_byteu(&s->gB);
400  if (s->quant_index[i] >= 4) {
401  av_log(s->avctx, AV_LOG_ERROR, "quant_index is invalid\n");
402  return AVERROR_INVALIDDATA;
403  }
404  if (!h_count[i] || !v_count[i]) {
405  av_log(s->avctx, AV_LOG_ERROR,
406  "Invalid sampling factor in component %d %d:%d\n",
407  i, h_count[i], v_count[i]);
408  return AVERROR_INVALIDDATA;
409  }
410 
411  av_log(s->avctx, AV_LOG_DEBUG, "component %d %d:%d id: %d quant:%d\n",
412  i, h_count[i], v_count[i],
413  s->component_id[i], s->quant_index[i]);
414  }
415  if ( nb_components == 4
416  && s->component_id[0] == 'C'
417  && s->component_id[1] == 'M'
418  && s->component_id[2] == 'Y'
419  && s->component_id[3] == 'K')
420  s->adobe_transform = 0;
421 
422  if (s->ls && (s->h_max > 1 || s->v_max > 1)) {
423  avpriv_report_missing_feature(s->avctx, "Subsampling in JPEG-LS");
424  return AVERROR_PATCHWELCOME;
425  }
426 
427  if (s->bayer) {
428  if (nb_components == 2) {
429  /* Bayer images embedded in DNGs can contain 2 interleaved components and the
430  width stored in their SOF3 markers is the width of each one. We only output
431  a single component, therefore we need to adjust the output image width. We
432  handle the deinterleaving (but not the debayering) in this file. */
433  width *= 2;
434  }
435  /* They can also contain 1 component, which is double the width and half the height
436  of the final image (rows are interleaved). We don't handle the decoding in this
437  file, but leave that to the TIFF/DNG decoder. */
438  }
439 
440  /* if different size, realloc/alloc picture */
441  if (width != s->width || height != s->height || bits != s->bits ||
442  memcmp(s->h_count, h_count, sizeof(h_count)) ||
443  memcmp(s->v_count, v_count, sizeof(v_count))) {
444  size_change = 1;
445 
446  s->width = width;
447  s->height = height;
448  s->bits = bits;
449  memcpy(s->h_count, h_count, sizeof(h_count));
450  memcpy(s->v_count, v_count, sizeof(v_count));
451  s->interlaced = 0;
452  s->got_picture = 0;
453 
454  /* test interlaced mode */
455  if (s->first_picture &&
456  (s->multiscope != 2 || s->avctx->pkt_timebase.den >= 25 * s->avctx->pkt_timebase.num) &&
457  s->orig_height != 0 &&
458  s->height < ((s->orig_height * 3) / 4)) {
459  s->interlaced = 1;
460  s->bottom_field = s->interlace_polarity;
461  s->picture_ptr->flags |= AV_FRAME_FLAG_INTERLACED;
462  s->picture_ptr->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST * !s->interlace_polarity;
463  height *= 2;
464  }
465 
466  ret = ff_set_dimensions(s->avctx, width, height);
467  if (ret < 0)
468  return ret;
469 
470  if (s->avctx->codec_id != AV_CODEC_ID_SMVJPEG &&
471  (s->avctx->codec_tag == MKTAG('A', 'V', 'R', 'n') ||
472  s->avctx->codec_tag == MKTAG('A', 'V', 'D', 'J')) &&
473  s->orig_height < height)
474  s->avctx->height = AV_CEIL_RSHIFT(s->orig_height, s->avctx->lowres);
475 
476  s->first_picture = 0;
477  } else {
478  size_change = 0;
479  }
480 
481  if (s->avctx->codec_id == AV_CODEC_ID_SMVJPEG) {
482  s->avctx->height = s->avctx->coded_height / s->smv_frames_per_jpeg;
483  if (s->avctx->height <= 0)
484  return AVERROR_INVALIDDATA;
485  }
486  if (s->bayer && s->progressive) {
487  avpriv_request_sample(s->avctx, "progressively coded bayer picture");
488  return AVERROR_INVALIDDATA;
489  }
490 
491  if (s->got_picture && s->interlaced && (s->bottom_field == !s->interlace_polarity)) {
492  if (s->progressive) {
493  avpriv_request_sample(s->avctx, "progressively coded interlaced picture");
494  return AVERROR_INVALIDDATA;
495  }
496  } else {
497  if (s->v_max == 1 && s->h_max == 1 && s->lossless == 1 && (nb_components == 3 || nb_components == 4))
498  s->rgb = 1;
499  else if (!s->lossless)
500  s->rgb = 0;
501  /* XXX: not complete test ! */
502  pix_fmt_id = ((unsigned)s->h_count[0] << 28) | (s->v_count[0] << 24) |
503  (s->h_count[1] << 20) | (s->v_count[1] << 16) |
504  (s->h_count[2] << 12) | (s->v_count[2] << 8) |
505  (s->h_count[3] << 4) | s->v_count[3];
506  av_log(s->avctx, AV_LOG_DEBUG, "pix fmt id %x\n", pix_fmt_id);
507  /* NOTE we do not allocate pictures large enough for the possible
508  * padding of h/v_count being 4 */
509  if (!(pix_fmt_id & 0xD0D0D0D0))
510  pix_fmt_id -= (pix_fmt_id & 0xF0F0F0F0) >> 1;
511  if (!(pix_fmt_id & 0x0D0D0D0D))
512  pix_fmt_id -= (pix_fmt_id & 0x0F0F0F0F) >> 1;
513 
514  for (i = 0; i < 8; i++) {
515  int j = 6 + (i & 1) - (i & 6);
516  int is = (pix_fmt_id >> (4 * i)) & 0xF;
517  int js = (pix_fmt_id >> (4 * j)) & 0xF;
518 
519  if (is == 1 && js != 2 && (i < 2 || i > 5))
520  js = (pix_fmt_id >> ( 8 + 4 * (i & 1))) & 0xF;
521  if (is == 1 && js != 2 && (i < 2 || i > 5))
522  js = (pix_fmt_id >> (16 + 4 * (i & 1))) & 0xF;
523 
524  if (is == 1 && js == 2) {
525  if (i & 1) s->upscale_h[j / 2] = 1;
526  else s->upscale_v[j / 2] = 1;
527  }
528  }
529 
530  if (s->bayer) {
531  if (pix_fmt_id != 0x11110000 && pix_fmt_id != 0x11000000)
532  goto unk_pixfmt;
533  }
534 
535  switch (pix_fmt_id) {
536  case 0x11110000: /* for bayer-encoded huffman lossless JPEGs embedded in DNGs */
537  if (!s->bayer)
538  goto unk_pixfmt;
539  s->avctx->pix_fmt = AV_PIX_FMT_GRAY16LE;
540  break;
541  case 0x11111100:
542  if (s->rgb)
543  s->avctx->pix_fmt = s->bits <= 9 ? AV_PIX_FMT_BGR24 : AV_PIX_FMT_BGR48;
544  else {
545  if ( s->adobe_transform == 0
546  || s->component_id[0] == 'R' && s->component_id[1] == 'G' && s->component_id[2] == 'B') {
547  s->avctx->pix_fmt = s->bits <= 8 ? AV_PIX_FMT_GBRP : AV_PIX_FMT_GBRP16;
548  } else {
549  if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV444P : AV_PIX_FMT_YUVJ444P;
550  else s->avctx->pix_fmt = AV_PIX_FMT_YUV444P16;
551  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
552  }
553  }
554  av_assert0(s->nb_components == 3);
555  break;
556  case 0x11111111:
557  if (s->rgb)
558  s->avctx->pix_fmt = s->bits <= 9 ? AV_PIX_FMT_ABGR : AV_PIX_FMT_RGBA64;
559  else {
560  if (s->adobe_transform == 0 && s->bits <= 8) {
561  s->avctx->pix_fmt = AV_PIX_FMT_GBRAP;
562  } else {
563  s->avctx->pix_fmt = s->bits <= 8 ? AV_PIX_FMT_YUVA444P : AV_PIX_FMT_YUVA444P16;
564  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
565  }
566  }
567  av_assert0(s->nb_components == 4);
568  break;
569  case 0x11412100:
570  if (s->bits > 8)
571  goto unk_pixfmt;
572  if (s->component_id[0] == 'R' && s->component_id[1] == 'G' && s->component_id[2] == 'B') {
573  s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
574  s->upscale_h[0] = 4;
575  s->upscale_h[1] = 0;
576  s->upscale_h[2] = 1;
577  } else {
578  goto unk_pixfmt;
579  }
580  break;
581  case 0x22111122:
582  case 0x22111111:
583  if (s->adobe_transform == 0 && s->bits <= 8) {
584  s->avctx->pix_fmt = AV_PIX_FMT_GBRAP;
585  s->upscale_v[1] = s->upscale_v[2] = 1;
586  s->upscale_h[1] = s->upscale_h[2] = 1;
587  } else if (s->adobe_transform == 2 && s->bits <= 8) {
588  s->avctx->pix_fmt = AV_PIX_FMT_YUVA444P;
589  s->upscale_v[1] = s->upscale_v[2] = 1;
590  s->upscale_h[1] = s->upscale_h[2] = 1;
591  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
592  } else {
593  if (s->bits <= 8) s->avctx->pix_fmt = AV_PIX_FMT_YUVA420P;
594  else s->avctx->pix_fmt = AV_PIX_FMT_YUVA420P16;
595  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
596  }
597  av_assert0(s->nb_components == 4);
598  break;
599  case 0x12121100:
600  case 0x22122100:
601  case 0x21211100:
602  case 0x21112100:
603  case 0x22211200:
604  case 0x22221100:
605  case 0x22112200:
606  case 0x11222200:
607  if (s->bits > 8)
608  goto unk_pixfmt;
609  if (s->adobe_transform == 0 || s->component_id[0] == 'R' &&
610  s->component_id[1] == 'G' && s->component_id[2] == 'B') {
611  s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
612  } else {
613  s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV444P : AV_PIX_FMT_YUVJ444P;
614  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
615  }
616  break;
617  case 0x11000000:
618  case 0x13000000:
619  case 0x14000000:
620  case 0x31000000:
621  case 0x33000000:
622  case 0x34000000:
623  case 0x41000000:
624  case 0x43000000:
625  case 0x44000000:
626  if (s->bits <= 8)
627  s->avctx->pix_fmt = s->force_pal8 ? AV_PIX_FMT_PAL8 : AV_PIX_FMT_GRAY8;
628  else
629  s->avctx->pix_fmt = AV_PIX_FMT_GRAY16;
630  break;
631  case 0x12111100:
632  case 0x14121200:
633  case 0x14111100:
634  case 0x22211100:
635  case 0x22112100:
636  if (s->component_id[0] == 'R' && s->component_id[1] == 'G' && s->component_id[2] == 'B') {
637  if (s->bits <= 8) s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
638  else
639  goto unk_pixfmt;
640  s->upscale_v[1] = s->upscale_v[2] = 1;
641  } else {
642  if (pix_fmt_id == 0x14111100)
643  s->upscale_v[1] = s->upscale_v[2] = 1;
644  if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV440P : AV_PIX_FMT_YUVJ440P;
645  else
646  goto unk_pixfmt;
647  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
648  }
649  break;
650  case 0x21111100:
651  if (s->component_id[0] == 'R' && s->component_id[1] == 'G' && s->component_id[2] == 'B') {
652  if (s->bits <= 8) s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
653  else
654  goto unk_pixfmt;
655  s->upscale_h[1] = s->upscale_h[2] = 1;
656  } else {
657  if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV422P : AV_PIX_FMT_YUVJ422P;
658  else s->avctx->pix_fmt = AV_PIX_FMT_YUV422P16;
659  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
660  }
661  break;
662  case 0x11311100:
663  if (s->bits > 8)
664  goto unk_pixfmt;
665  if (s->component_id[0] == 'R' && s->component_id[1] == 'G' && s->component_id[2] == 'B')
666  s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
667  else
668  goto unk_pixfmt;
669  s->upscale_h[0] = s->upscale_h[2] = 2;
670  break;
671  case 0x31111100:
672  if (s->bits > 8)
673  goto unk_pixfmt;
674  s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV444P : AV_PIX_FMT_YUVJ444P;
675  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
676  s->upscale_h[1] = s->upscale_h[2] = 2;
677  break;
678  case 0x22121100:
679  case 0x22111200:
680  case 0x41211100:
681  if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV422P : AV_PIX_FMT_YUVJ422P;
682  else
683  goto unk_pixfmt;
684  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
685  break;
686  case 0x22111100:
687  case 0x23111100:
688  case 0x42111100:
689  case 0x24111100:
690  if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV420P : AV_PIX_FMT_YUVJ420P;
691  else s->avctx->pix_fmt = AV_PIX_FMT_YUV420P16;
692  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
693  if (pix_fmt_id == 0x42111100) {
694  if (s->bits > 8)
695  goto unk_pixfmt;
696  s->upscale_h[1] = s->upscale_h[2] = 1;
697  } else if (pix_fmt_id == 0x24111100) {
698  if (s->bits > 8)
699  goto unk_pixfmt;
700  s->upscale_v[1] = s->upscale_v[2] = 1;
701  } else if (pix_fmt_id == 0x23111100) {
702  if (s->bits > 8)
703  goto unk_pixfmt;
704  s->upscale_v[1] = s->upscale_v[2] = 2;
705  }
706  break;
707  case 0x41111100:
708  if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV411P : AV_PIX_FMT_YUVJ411P;
709  else
710  goto unk_pixfmt;
711  s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
712  break;
713  default:
714  unk_pixfmt:
715  avpriv_report_missing_feature(s->avctx, "Pixel format 0x%x bits:%d", pix_fmt_id, s->bits);
716  memset(s->upscale_h, 0, sizeof(s->upscale_h));
717  memset(s->upscale_v, 0, sizeof(s->upscale_v));
718  return AVERROR_PATCHWELCOME;
719  }
720  if ((AV_RB32(s->upscale_h) || AV_RB32(s->upscale_v)) && s->avctx->lowres) {
721  avpriv_report_missing_feature(s->avctx, "Lowres for weird subsampling");
722  return AVERROR_PATCHWELCOME;
723  }
724  if (s->ls) {
725  memset(s->upscale_h, 0, sizeof(s->upscale_h));
726  memset(s->upscale_v, 0, sizeof(s->upscale_v));
727  if (s->nb_components == 3) {
728  s->avctx->pix_fmt = AV_PIX_FMT_RGB24;
729  } else if (s->nb_components != 1) {
730  av_log(s->avctx, AV_LOG_ERROR, "Unsupported number of components %d\n", s->nb_components);
731  return AVERROR_PATCHWELCOME;
732  } else if ((s->palette_index || s->force_pal8) && s->bits <= 8)
733  s->avctx->pix_fmt = AV_PIX_FMT_PAL8;
734  else if (s->bits <= 8)
735  s->avctx->pix_fmt = AV_PIX_FMT_GRAY8;
736  else
737  s->avctx->pix_fmt = AV_PIX_FMT_GRAY16;
738  }
739 
740  s->pix_desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
741  if (!s->pix_desc) {
742  av_log(s->avctx, AV_LOG_ERROR, "Could not get a pixel format descriptor.\n");
743  return AVERROR_BUG;
744  }
745 
746  if (s->avctx->pix_fmt == s->hwaccel_sw_pix_fmt && !size_change) {
747  s->avctx->pix_fmt = s->hwaccel_pix_fmt;
748  } else {
749  enum AVPixelFormat pix_fmts[] = {
750 #if CONFIG_MJPEG_NVDEC_HWACCEL
751  AV_PIX_FMT_CUDA,
752 #endif
753 #if CONFIG_MJPEG_VAAPI_HWACCEL
754  AV_PIX_FMT_VAAPI,
755 #endif
756  s->avctx->pix_fmt,
757  AV_PIX_FMT_NONE,
758  };
759  s->hwaccel_pix_fmt = ff_get_format(s->avctx, pix_fmts);
760  if (s->hwaccel_pix_fmt < 0)
761  return AVERROR(EINVAL);
762 
763  s->hwaccel_sw_pix_fmt = s->avctx->pix_fmt;
764  s->avctx->pix_fmt = s->hwaccel_pix_fmt;
765  }
766 
767  if (s->avctx->skip_frame == AVDISCARD_ALL) {
768  s->picture_ptr->pict_type = AV_PICTURE_TYPE_I;
769  s->picture_ptr->flags |= AV_FRAME_FLAG_KEY;
770  s->got_picture = 1;
771  return 0;
772  }
773 
774  av_frame_unref(s->picture_ptr);
775  ret = ff_get_buffer(s->avctx, s->picture_ptr, AV_GET_BUFFER_FLAG_REF);
776  if (ret < 0)
777  return ret;
778  s->picture_ptr->pict_type = AV_PICTURE_TYPE_I;
779  s->picture_ptr->flags |= AV_FRAME_FLAG_KEY;
780  s->got_picture = 1;
781 
782  // Lets clear the palette to avoid leaving uninitialized values in it
783  if (s->avctx->pix_fmt == AV_PIX_FMT_PAL8)
784  memset(s->picture_ptr->data[1], 0, 1024);
785 
786  for (i = 0; i < 4; i++)
787  s->linesize[i] = s->picture_ptr->linesize[i] << s->interlaced;
788 
789  ff_dlog(s->avctx, "%d %d %d %d %d %d\n",
790  s->width, s->height, s->linesize[0], s->linesize[1],
791  s->interlaced, s->avctx->height);
792 
793  }
794 
795  if ((s->rgb && !s->lossless && !s->ls) ||
796  (!s->rgb && s->ls && s->nb_components > 1) ||
797  (s->avctx->pix_fmt == AV_PIX_FMT_PAL8 && !s->ls)
798  ) {
799  av_log(s->avctx, AV_LOG_ERROR, "Unsupported coding and pixel format combination\n");
800  return AVERROR_PATCHWELCOME;
801  }
802 
803  /* totally blank picture as progressive JPEG will only add details to it */
804  if (s->progressive) {
805  int bw = (width + s->h_max * 8 - 1) / (s->h_max * 8);
806  int bh = (height + s->v_max * 8 - 1) / (s->v_max * 8);
807  for (i = 0; i < s->nb_components; i++) {
808  int size = bw * bh * s->h_count[i] * s->v_count[i];
809  av_freep(&s->blocks[i]);
810  av_freep(&s->last_nnz[i]);
811  s->blocks[i] = av_calloc(size, sizeof(**s->blocks));
812  s->last_nnz[i] = av_calloc(size, sizeof(**s->last_nnz));
813  if (!s->blocks[i] || !s->last_nnz[i])
814  return AVERROR(ENOMEM);
815  s->block_stride[i] = bw * s->h_count[i];
816  }
817  memset(s->coefs_finished, 0, sizeof(s->coefs_finished));
818  }
819 
820  if (s->avctx->hwaccel) {
821  const FFHWAccel *hwaccel = ffhwaccel(s->avctx->hwaccel);
822  s->hwaccel_picture_private =
823  av_mallocz(hwaccel->frame_priv_data_size);
824  if (!s->hwaccel_picture_private)
825  return AVERROR(ENOMEM);
826 
827  ret = hwaccel->start_frame(s->avctx, NULL, s->raw_image_buffer,
828  s->raw_image_buffer_size);
829  if (ret < 0)
830  return ret;
831  }
832 
833  return 0;
834 }
835 
836 static inline int mjpeg_decode_dc(MJpegDecodeContext *s, int dc_index, int *val)
837 {
838  int code;
839  code = get_vlc2(&s->gb, s->vlcs[0][dc_index].table, 9, 2);
840  if (code < 0 || code > 16) {
841  av_log(s->avctx, AV_LOG_ERROR,
842  "mjpeg_decode_dc: bad vlc: %d\n", dc_index);
843  return AVERROR_INVALIDDATA;
844  }
845 
846  *val = code ? get_xbits(&s->gb, code) : 0;
847  return 0;
848 }
849 
850 /* decode block and dequantize */
851 static int decode_block(MJpegDecodeContext *s, int16_t *block, int component,
852  int dc_index, int ac_index, uint16_t *quant_matrix)
853 {
854  int code, i, j, level, val;
855 
856  /* DC coef */
857  int ret = mjpeg_decode_dc(s, dc_index, &val);
858  if (ret < 0)
859  return ret;
860 
861  val = val * (unsigned)quant_matrix[0] + s->last_dc[component];
862  s->last_dc[component] = val;
863  block[0] = av_clip_int16(val);
864  /* AC coefs */
865  i = 0;
866  {
867  OPEN_READER(re, &s->gb);
868  do {
869  UPDATE_CACHE(re, &s->gb);
870  GET_VLC(code, re, &s->gb, s->vlcs[1][ac_index].table, 9, 2);
871 
872  i += ((unsigned)code) >> 4;
873  code &= 0xf;
874  if (code) {
875  // GET_VLC updates the cache if parsing reaches the second stage.
876  // So we have at least MIN_CACHE_BITS - 9 > 15 bits left here
877  // and don't need to refill the cache.
878  {
879  int cache = GET_CACHE(re, &s->gb);
880  int sign = (~cache) >> 31;
881  level = (NEG_USR32(sign ^ cache,code) ^ sign) - sign;
882  }
883 
884  LAST_SKIP_BITS(re, &s->gb, code);
885 
886  if (i > 63) {
887  av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i);
888  return AVERROR_INVALIDDATA;
889  }
890  j = s->permutated_scantable[i];
891  block[j] = level * quant_matrix[i];
892  }
893  } while (i < 63);
894  CLOSE_READER(re, &s->gb);
895  }
896 
897  return 0;
898 }
899 
900 static int decode_dc_progressive(MJpegDecodeContext *s, int16_t *block,
901  int component, int dc_index,
902  uint16_t *quant_matrix, int Al)
903 {
904  unsigned val;
905  s->bdsp.clear_block(block);
906  int ret = mjpeg_decode_dc(s, dc_index, &val);
907  if (ret < 0)
908  return ret;
909 
910  val = (val * (quant_matrix[0] << Al)) + s->last_dc[component];
911  s->last_dc[component] = val;
912  block[0] = val;
913  return 0;
914 }
915 
916 /* decode block and dequantize - progressive JPEG version */
917 static int decode_block_progressive(MJpegDecodeContext *s, int16_t *block,
918  uint8_t *last_nnz, int ac_index,
919  uint16_t *quant_matrix,
920  int Ss, int Se, int Al, int *EOBRUN)
921 {
922  int code, i, j, val, run;
923  unsigned level;
924 
925  if (*EOBRUN) {
926  (*EOBRUN)--;
927  return 0;
928  }
929 
930  {
931  OPEN_READER(re, &s->gb);
932  for (i = Ss; ; i++) {
933  UPDATE_CACHE(re, &s->gb);
934  GET_VLC(code, re, &s->gb, s->vlcs[2][ac_index].table, 9, 2);
935 
936  run = ((unsigned) code) >> 4;
937  code &= 0xF;
938  if (code) {
939  i += run;
940 
941  {
942  int cache = GET_CACHE(re, &s->gb);
943  int sign = (~cache) >> 31;
944  level = (NEG_USR32(sign ^ cache,code) ^ sign) - sign;
945  }
946 
947  LAST_SKIP_BITS(re, &s->gb, code);
948 
949  if (i >= Se) {
950  if (i == Se) {
951  j = s->permutated_scantable[Se];
952  block[j] = level * (quant_matrix[Se] << Al);
953  break;
954  }
955  av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i);
956  return AVERROR_INVALIDDATA;
957  }
958  j = s->permutated_scantable[i];
959  block[j] = level * (quant_matrix[i] << Al);
960  } else {
961  if (run == 0xF) { // ZRL - skip 15 coefficients
962  i += 15;
963  if (i >= Se) {
964  av_log(s->avctx, AV_LOG_ERROR, "ZRL overflow: %d\n", i);
965  return AVERROR_INVALIDDATA;
966  }
967  } else {
968  val = (1 << run);
969  if (run) {
970  // Given that GET_VLC reloads internally, we always
971  // have at least 16 bits in the cache here.
972  val += NEG_USR32(GET_CACHE(re, &s->gb), run);
973  LAST_SKIP_BITS(re, &s->gb, run);
974  }
975  *EOBRUN = val - 1;
976  break;
977  }
978  }
979  }
980  CLOSE_READER(re, &s->gb);
981  }
982 
983  if (i > *last_nnz)
984  *last_nnz = i;
985 
986  return 0;
987 }
988 
989 #define REFINE_BIT(j) { \
990  UPDATE_CACHE(re, &s->gb); \
991  sign = block[j] >> 15; \
992  block[j] += SHOW_UBITS(re, &s->gb, 1) * \
993  ((quant_matrix[i] ^ sign) - sign) << Al; \
994  LAST_SKIP_BITS(re, &s->gb, 1); \
995 }
996 
997 #define ZERO_RUN \
998 for (; ; i++) { \
999  if (i > last) { \
1000  i += run; \
1001  if (i > Se) { \
1002  av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i); \
1003  return -1; \
1004  } \
1005  break; \
1006  } \
1007  j = s->permutated_scantable[i]; \
1008  if (block[j]) \
1009  REFINE_BIT(j) \
1010  else if (run-- == 0) \
1011  break; \
1012 }
1013 
1014 /* decode block and dequantize - progressive JPEG refinement pass */
1015 static int decode_block_refinement(MJpegDecodeContext *s, int16_t *block,
1016  uint8_t *last_nnz,
1017  int ac_index, uint16_t *quant_matrix,
1018  int Ss, int Se, int Al, int *EOBRUN)
1019 {
1020  int code, i = Ss, j, sign, val, run;
1021  int last = FFMIN(Se, *last_nnz);
1022 
1023  OPEN_READER(re, &s->gb);
1024  if (*EOBRUN) {
1025  (*EOBRUN)--;
1026  } else {
1027  for (; ; i++) {
1028  UPDATE_CACHE(re, &s->gb);
1029  GET_VLC(code, re, &s->gb, s->vlcs[2][ac_index].table, 9, 2);
1030 
1031  if (code & 0xF) {
1032  run = ((unsigned) code) >> 4;
1033  val = SHOW_UBITS(re, &s->gb, 1);
1034  LAST_SKIP_BITS(re, &s->gb, 1);
1035  ZERO_RUN;
1036  j = s->permutated_scantable[i];
1037  val--;
1038  block[j] = ((quant_matrix[i] << Al) ^ val) - val;
1039  if (i == Se) {
1040  if (i > *last_nnz)
1041  *last_nnz = i;
1042  CLOSE_READER(re, &s->gb);
1043  return 0;
1044  }
1045  } else {
1046  run = ((unsigned) code) >> 4;
1047  if (run == 0xF) {
1048  ZERO_RUN;
1049  } else {
1050  val = run;
1051  run = (1 << run);
1052  if (val) {
1053  // Given that GET_VLC reloads internally, we always
1054  // have at least 16 bits in the cache here.
1055  run += SHOW_UBITS(re, &s->gb, val);
1056  LAST_SKIP_BITS(re, &s->gb, val);
1057  }
1058  *EOBRUN = run - 1;
1059  break;
1060  }
1061  }
1062  }
1063 
1064  if (i > *last_nnz)
1065  *last_nnz = i;
1066  }
1067 
1068  for (; i <= last; i++) {
1069  j = s->permutated_scantable[i];
1070  if (block[j])
1071  REFINE_BIT(j)
1072  }
1073  CLOSE_READER(re, &s->gb);
1074 
1075  return 0;
1076 }
1077 #undef REFINE_BIT
1078 #undef ZERO_RUN
1079 
1080 /* Handles 1 to 4 components */
1081 static int ljpeg_decode_rgb_scan(MJpegDecodeContext *s)
1082 {
1083  int nb_components = s->nb_components_sos;
1084  int predictor = s->Ss;
1085  int point_transform = s->Al;
1086  int i, mb_x, mb_y;
1087  unsigned width;
1088  uint16_t (*buffer)[4];
1089  int left[4], top[4], topleft[4];
1090  const int linesize = s->linesize[0];
1091  const int mask = ((1 << s->bits) - 1) << point_transform;
1092  int resync_mb_y = 0;
1093  int resync_mb_x = 0;
1094  int vpred[6];
1095  int ret;
1096 
1097  if (!s->bayer && s->nb_components < 3)
1098  return AVERROR_INVALIDDATA;
1099  if (s->bayer && s->nb_components > 2)
1100  return AVERROR_INVALIDDATA;
1101  if (s->nb_components <= 0 || s->nb_components > 4)
1102  return AVERROR_INVALIDDATA;
1103  if (s->v_max != 1 || s->h_max != 1 || !s->lossless)
1104  return AVERROR_INVALIDDATA;
1105  if (s->bayer) {
1106  if (s->rct || s->pegasus_rct)
1107  return AVERROR_INVALIDDATA;
1108  }
1109 
1110 
1111  for (i = 0; i < 6; i++)
1112  vpred[i] = 1 << (s->bits - 1);
1113 
1114  if (s->bayer)
1115  width = s->mb_width / nb_components; /* Interleaved, width stored is the total so need to divide */
1116  else
1117  width = s->mb_width;
1118 
1119  av_fast_malloc(&s->ljpeg_buffer, &s->ljpeg_buffer_size, width * 4 * sizeof(s->ljpeg_buffer[0][0]));
1120  if (!s->ljpeg_buffer)
1121  return AVERROR(ENOMEM);
1122 
1123  buffer = s->ljpeg_buffer;
1124 
1125  for (i = 0; i < 4; i++)
1126  buffer[0][i] = 1 << (s->bits - 1);
1127 
1128  s->restart_count = -1;
1129 
1130  for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1131  uint8_t *ptr = s->picture_ptr->data[0] + (linesize * mb_y);
1132 
1133  if (s->interlaced && s->bottom_field)
1134  ptr += linesize >> 1;
1135 
1136  for (i = 0; i < 4; i++)
1137  top[i] = left[i] = topleft[i] = buffer[0][i];
1138 
1139  for (mb_x = 0; mb_x < width; mb_x++) {
1140  int modified_predictor = predictor;
1141  int restart;
1142 
1143  ret = ff_mjpeg_handle_restart(s, &restart);
1144  if (ret < 0)
1145  return ret;
1146  if (restart) {
1147  resync_mb_x = mb_x;
1148  resync_mb_y = mb_y;
1149  for (i = 0; i < 4; i++)
1150  top[i] = left[i] = topleft[i] = 1 << (s->bits - 1);
1151  }
1152 
1153  if (get_bits_left(&s->gb) < 1) {
1154  av_log(s->avctx, AV_LOG_ERROR, "bitstream end in rgb_scan\n");
1155  return AVERROR_INVALIDDATA;
1156  }
1157 
1158  if (mb_y == resync_mb_y || mb_y == resync_mb_y + 1 && mb_x < resync_mb_x || !mb_x)
1159  modified_predictor = 1;
1160 
1161  for (i = 0; i < nb_components; i++) {
1162  int pred, dc;
1163 
1164  topleft[i] = top[i];
1165  top[i] = buffer[mb_x][i];
1166 
1167  ret = mjpeg_decode_dc(s, s->dc_index[i], &dc);
1168  if (ret < 0)
1169  return ret;
1170 
1171  if (!s->bayer || mb_x) {
1172  pred = left[i];
1173  } else { /* This path runs only for the first line in bayer images */
1174  vpred[i] += dc;
1175  pred = vpred[i] - dc;
1176  }
1177 
1178  PREDICT(pred, topleft[i], top[i], pred, modified_predictor);
1179 
1180  left[i] = buffer[mb_x][i] =
1181  mask & (pred + (unsigned)(dc * (1 << point_transform)));
1182  }
1183  }
1184  if (s->rct && s->nb_components == 4) {
1185  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1186  ptr[4 * mb_x + 2] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2] - 0x200) >> 2);
1187  ptr[4 * mb_x + 1] = buffer[mb_x][1] + ptr[4 * mb_x + 2];
1188  ptr[4 * mb_x + 3] = buffer[mb_x][2] + ptr[4 * mb_x + 2];
1189  ptr[4 * mb_x + 0] = buffer[mb_x][3];
1190  }
1191  } else if (s->nb_components == 4) {
1192  for (i = 0; i < nb_components; i++) {
1193  int c = s->comp_index[i];
1194  if (s->bits <= 8) {
1195  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1196  ptr[4 * mb_x + 3 - c] = buffer[mb_x][i];
1197  }
1198  } else if (s->bits == 9) {
1199  return AVERROR_PATCHWELCOME;
1200  } else {
1201  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1202  ((uint16_t*)ptr)[4 * mb_x + c] = buffer[mb_x][i];
1203  }
1204  }
1205  }
1206  } else if (s->rct) {
1207  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1208  ptr[3 * mb_x + 1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2] - 0x200) >> 2);
1209  ptr[3 * mb_x + 0] = buffer[mb_x][1] + ptr[3 * mb_x + 1];
1210  ptr[3 * mb_x + 2] = buffer[mb_x][2] + ptr[3 * mb_x + 1];
1211  }
1212  } else if (s->pegasus_rct) {
1213  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1214  ptr[3 * mb_x + 1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2]) >> 2);
1215  ptr[3 * mb_x + 0] = buffer[mb_x][1] + ptr[3 * mb_x + 1];
1216  ptr[3 * mb_x + 2] = buffer[mb_x][2] + ptr[3 * mb_x + 1];
1217  }
1218  } else if (s->bayer) {
1219  if (s->bits <= 8)
1220  return AVERROR_PATCHWELCOME;
1221  if (nb_components == 1) {
1222  /* Leave decoding to the TIFF/DNG decoder (see comment in ff_mjpeg_decode_sof) */
1223  for (mb_x = 0; mb_x < width; mb_x++)
1224  ((uint16_t*)ptr)[mb_x] = buffer[mb_x][0];
1225  } else if (nb_components == 2) {
1226  for (mb_x = 0; mb_x < width; mb_x++) {
1227  ((uint16_t*)ptr)[2 * mb_x + 0] = buffer[mb_x][0];
1228  ((uint16_t*)ptr)[2 * mb_x + 1] = buffer[mb_x][1];
1229  }
1230  }
1231  } else {
1232  for (i = 0; i < nb_components; i++) {
1233  int c = s->comp_index[i];
1234  if (s->bits <= 8) {
1235  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1236  ptr[3 * mb_x + 2 - c] = buffer[mb_x][i];
1237  }
1238  } else if (s->bits == 9) {
1239  return AVERROR_PATCHWELCOME;
1240  } else {
1241  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1242  ((uint16_t*)ptr)[3 * mb_x + 2 - c] = buffer[mb_x][i];
1243  }
1244  }
1245  }
1246  }
1247  }
1248  return 0;
1249 }
1250 
1251 static int ljpeg_decode_yuv_scan(MJpegDecodeContext *s)
1252 {
1253  int predictor = s->Ss;
1254  int point_transform = s->Al;
1255  int nb_components = s->nb_components_sos;
1256  int i, mb_x, mb_y, mask;
1257  int bits = (s->bits + 7) & ~7;
1258  int resync_mb_y = 0;
1259  int resync_mb_x = 0;
1260  int ret;
1261 
1262  point_transform += bits - s->bits;
1263  mask = ((1 << s->bits) - 1) << point_transform;
1264 
1265  av_assert0(nb_components >= 1 && nb_components <= 4);
1266 
1267  s->restart_count = -1;
1268 
1269  for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1270  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1271  int restart;
1272  ret = ff_mjpeg_handle_restart(s, &restart);
1273  if (ret < 0)
1274  return ret;
1275  if (restart) {
1276  resync_mb_x = mb_x;
1277  resync_mb_y = mb_y;
1278  }
1279 
1280  if (get_bits_left(&s->gb) < 1) {
1281  av_log(s->avctx, AV_LOG_ERROR, "bitstream end in yuv_scan\n");
1282  return AVERROR_INVALIDDATA;
1283  }
1284 
1285  if (!mb_x || mb_y == resync_mb_y || mb_y == resync_mb_y + 1 && mb_x < resync_mb_x || s->interlaced) {
1286  int toprow = mb_y == resync_mb_y || mb_y == resync_mb_y + 1 && mb_x < resync_mb_x;
1287  int leftcol = !mb_x || mb_y == resync_mb_y && mb_x == resync_mb_x;
1288  for (i = 0; i < nb_components; i++) {
1289  uint8_t *ptr;
1290  uint16_t *ptr16;
1291  int n, h, v, x, y, c, j, linesize;
1292  n = s->nb_blocks[i];
1293  c = s->comp_index[i];
1294  h = s->h_scount[i];
1295  v = s->v_scount[i];
1296  x = 0;
1297  y = 0;
1298  linesize = s->linesize[c];
1299 
1300  if (bits > 8) linesize /= 2;
1301 
1302  for (j = 0; j < n; j++) {
1303  int pred, dc;
1304 
1305  ret = mjpeg_decode_dc(s, s->dc_index[i], &dc);
1306  if (ret < 0)
1307  return ret;
1308 
1309  if ( h * mb_x + x >= s->width
1310  || v * mb_y + y >= s->height) {
1311  // Nothing to do
1312  } else if (bits <= 8) {
1313  ptr = s->picture_ptr->data[c] + (linesize * (v * mb_y + y)) + (h * mb_x + x); // FIXME optimize this crap
1314  if (y == 0 && toprow) {
1315  if (x == 0 && leftcol) {
1316  pred = 1 << (bits - 1);
1317  } else {
1318  pred = ptr[-1];
1319  }
1320  } else {
1321  if (x == 0 && leftcol) {
1322  pred = ptr[-linesize];
1323  } else {
1324  PREDICT(pred, ptr[-linesize - 1], ptr[-linesize], ptr[-1], predictor);
1325  }
1326  }
1327 
1328  if (s->interlaced && s->bottom_field)
1329  ptr += linesize >> 1;
1330  pred &= mask;
1331  *ptr = pred + ((unsigned)dc << point_transform);
1332  } else {
1333  ptr16 = (uint16_t*)(s->picture_ptr->data[c] + 2 * (linesize * (v * mb_y + y)) + 2 * (h * mb_x + x)); // FIXME optimize this crap
1334  if (y == 0 && toprow) {
1335  if (x == 0 && leftcol) {
1336  pred = 1 << (bits - 1);
1337  } else {
1338  pred = ptr16[-1];
1339  }
1340  } else {
1341  if (x == 0 && leftcol) {
1342  pred = ptr16[-linesize];
1343  } else {
1344  PREDICT(pred, ptr16[-linesize - 1], ptr16[-linesize], ptr16[-1], predictor);
1345  }
1346  }
1347 
1348  if (s->interlaced && s->bottom_field)
1349  ptr16 += linesize >> 1;
1350  pred &= mask;
1351  *ptr16 = pred + ((unsigned)dc << point_transform);
1352  }
1353  if (++x == h) {
1354  x = 0;
1355  y++;
1356  }
1357  }
1358  }
1359  } else {
1360  for (i = 0; i < nb_components; i++) {
1361  uint8_t *ptr;
1362  uint16_t *ptr16;
1363  int n, h, v, x, y, c, j, linesize, dc;
1364  n = s->nb_blocks[i];
1365  c = s->comp_index[i];
1366  h = s->h_scount[i];
1367  v = s->v_scount[i];
1368  x = 0;
1369  y = 0;
1370  linesize = s->linesize[c];
1371 
1372  if (bits > 8) linesize /= 2;
1373 
1374  for (j = 0; j < n; j++) {
1375  int pred;
1376 
1377  ret = mjpeg_decode_dc(s, s->dc_index[i], &dc);
1378  if (ret < 0)
1379  return ret;
1380 
1381  if ( h * mb_x + x >= s->width
1382  || v * mb_y + y >= s->height) {
1383  // Nothing to do
1384  } else if (bits <= 8) {
1385  ptr = s->picture_ptr->data[c] +
1386  (linesize * (v * mb_y + y)) +
1387  (h * mb_x + x); // FIXME optimize this crap
1388  PREDICT(pred, ptr[-linesize - 1], ptr[-linesize], ptr[-1], predictor);
1389 
1390  pred &= mask;
1391  *ptr = pred + ((unsigned)dc << point_transform);
1392  } else {
1393  ptr16 = (uint16_t*)(s->picture_ptr->data[c] + 2 * (linesize * (v * mb_y + y)) + 2 * (h * mb_x + x)); // FIXME optimize this crap
1394  PREDICT(pred, ptr16[-linesize - 1], ptr16[-linesize], ptr16[-1], predictor);
1395 
1396  pred &= mask;
1397  *ptr16 = pred + ((unsigned)dc << point_transform);
1398  }
1399 
1400  if (++x == h) {
1401  x = 0;
1402  y++;
1403  }
1404  }
1405  }
1406  }
1407  }
1408  }
1409  return 0;
1410 }
1411 
1412 static av_always_inline void mjpeg_copy_block(MJpegDecodeContext *s,
1413  uint8_t *dst, const uint8_t *src,
1414  int linesize, int lowres)
1415 {
1416  switch (lowres) {
1417  case 0: s->copy_block(dst, src, linesize, 8);
1418  break;
1419  case 1: copy_block4(dst, src, linesize, linesize, 4);
1420  break;
1421  case 2: copy_block2(dst, src, linesize, linesize, 2);
1422  break;
1423  case 3: *dst = *src;
1424  break;
1425  }
1426 }
1427 
1428 static void shift_output(MJpegDecodeContext *s, uint8_t *ptr, int linesize)
1429 {
1430  int block_x, block_y;
1431  int size = 8 >> s->avctx->lowres;
1432  if (s->bits > 8) {
1433  for (block_y = 0; block_y < size; block_y++)
1434  for (block_x = 0; block_x < size; block_x++)
1435  *(uint16_t*)(ptr + 2 * block_x + block_y * linesize) <<= 16 - s->bits;
1436  } else {
1437  for (block_y = 0; block_y < size; block_y++)
1438  for (block_x = 0; block_x < size; block_x++)
1439  *(ptr + block_x + block_y * linesize) <<= 8 - s->bits;
1440  }
1441 }
1442 
1443 static int mjpeg_decode_scan(MJpegDecodeContext *s)
1444 {
1445  int nb_components = s->nb_components_sos;
1446  int Ah = s->Ah;
1447  int Al = s->Al;
1448  const uint8_t *mb_bitmask = NULL;
1449  const AVFrame *reference = NULL;
1450  int i, mb_x, mb_y, chroma_h_shift, chroma_v_shift, chroma_width, chroma_height;
1451  uint8_t *data[MAX_COMPONENTS];
1452  const uint8_t *reference_data[MAX_COMPONENTS];
1453  int linesize[MAX_COMPONENTS];
1454  GetBitContext mb_bitmask_gb = {0}; // initialize to silence gcc warning
1455  int bytes_per_pixel = 1 + (s->bits > 8);
1456  int ret;
1457 
1458  if (s->avctx->codec_id == AV_CODEC_ID_MXPEG) {
1459  mb_bitmask = s->mb_bitmask;
1460  reference = s->reference;
1461  }
1462 
1463  if (mb_bitmask) {
1464  if (s->mb_bitmask_size != (s->mb_width * s->mb_height + 7) >> 3) {
1465  av_log(s->avctx, AV_LOG_ERROR, "mb_bitmask_size mismatches\n");
1466  return AVERROR_INVALIDDATA;
1467  }
1468  init_get_bits(&mb_bitmask_gb, mb_bitmask, s->mb_width * s->mb_height);
1469  }
1470 
1471  av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt, &chroma_h_shift,
1472  &chroma_v_shift);
1473  chroma_width = AV_CEIL_RSHIFT(s->width, chroma_h_shift);
1474  chroma_height = AV_CEIL_RSHIFT(s->height, chroma_v_shift);
1475 
1476  for (i = 0; i < nb_components; i++) {
1477  int c = s->comp_index[i];
1478  data[c] = s->picture_ptr->data[c];
1479  reference_data[c] = reference ? reference->data[c] : NULL;
1480  linesize[c] = s->linesize[c];
1481  s->coefs_finished[c] |= 1;
1482  }
1483 
1484 next_field:
1485  s->restart_count = -1;
1486 
1487  for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1488  for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1489  const int copy_mb = mb_bitmask && !get_bits1(&mb_bitmask_gb);
1490  int restart;
1491 
1492  if (s->avctx->codec_id == AV_CODEC_ID_THP) {
1493  if (s->restart_count < 0) {
1494  ret = ff_mjpeg_unescape_sos(s);
1495  if (ret < 0)
1496  return ret;
1497  }
1498  restart = ff_mjpeg_should_restart(s);
1499  if (restart)
1500  align_get_bits(&s->gb);
1501  } else {
1502  ret = ff_mjpeg_handle_restart(s, &restart);
1503  if (ret < 0)
1504  return ret;
1505  }
1506  if (restart) {
1507  for (i = 0; i < nb_components; i++)
1508  s->last_dc[i] = (4 << s->bits);
1509  }
1510 
1511  if (get_bits_left(&s->gb) < 0) {
1512  av_log(s->avctx, AV_LOG_ERROR, "overread %d\n",
1513  -get_bits_left(&s->gb));
1514  return AVERROR_INVALIDDATA;
1515  }
1516  for (i = 0; i < nb_components; i++) {
1517  uint8_t *ptr;
1518  int n, h, v, x, y, c, j;
1519  int block_offset;
1520  n = s->nb_blocks[i];
1521  c = s->comp_index[i];
1522  h = s->h_scount[i];
1523  v = s->v_scount[i];
1524  x = 0;
1525  y = 0;
1526  for (j = 0; j < n; j++) {
1527  block_offset = (((linesize[c] * (v * mb_y + y) * 8) +
1528  (h * mb_x + x) * 8 * bytes_per_pixel) >> s->avctx->lowres);
1529 
1530  if (s->interlaced && s->bottom_field)
1531  block_offset += linesize[c] >> 1;
1532  if ( 8 * (h * mb_x + x) < ((c == 1) || (c == 2) ? chroma_width : s->width)
1533  && 8 * (v * mb_y + y) < ((c == 1) || (c == 2) ? chroma_height : s->height)) {
1534  ptr = data[c] + block_offset;
1535  } else
1536  ptr = NULL;
1537  if (!s->progressive) {
1538  if (copy_mb) {
1539  if (ptr)
1540  mjpeg_copy_block(s, ptr, reference_data[c] + block_offset,
1541  linesize[c], s->avctx->lowres);
1542 
1543  } else {
1544  s->bdsp.clear_block(s->block);
1545  if (decode_block(s, s->block, i,
1546  s->dc_index[i], s->ac_index[i],
1547  s->quant_matrixes[s->quant_sindex[i]]) < 0) {
1548  av_log(s->avctx, AV_LOG_ERROR,
1549  "error y=%d x=%d\n", mb_y, mb_x);
1550  return AVERROR_INVALIDDATA;
1551  }
1552  if (ptr && linesize[c]) {
1553  s->idsp.idct_put(ptr, linesize[c], s->block);
1554  if (s->bits & 7)
1555  shift_output(s, ptr, linesize[c]);
1556  }
1557  }
1558  } else {
1559  int block_idx = s->block_stride[c] * (v * mb_y + y) +
1560  (h * mb_x + x);
1561  int16_t *block = s->blocks[c][block_idx];
1562  if (Ah)
1563  block[0] += get_bits1(&s->gb) *
1564  s->quant_matrixes[s->quant_sindex[i]][0] << Al;
1565  else if (decode_dc_progressive(s, block, i, s->dc_index[i],
1566  s->quant_matrixes[s->quant_sindex[i]],
1567  Al) < 0) {
1568  av_log(s->avctx, AV_LOG_ERROR,
1569  "error y=%d x=%d\n", mb_y, mb_x);
1570  return AVERROR_INVALIDDATA;
1571  }
1572  }
1573  ff_dlog(s->avctx, "mb: %d %d processed\n", mb_y, mb_x);
1574  ff_dlog(s->avctx, "%d %d %d %d %d %d %d %d \n",
1575  mb_x, mb_y, x, y, c, s->bottom_field,
1576  (v * mb_y + y) * 8, (h * mb_x + x) * 8);
1577  if (++x == h) {
1578  x = 0;
1579  y++;
1580  }
1581  }
1582  }
1583  }
1584  }
1585 
1586  if (s->interlaced &&
1587  bytestream2_get_bytes_left(&s->gB) > 2 &&
1588  bytestream2_tell(&s->gB) > 2 &&
1589  s->gB.buffer[-2] == 0xFF &&
1590  s->gB.buffer[-1] == 0xD1) {
1591  av_log(s->avctx, AV_LOG_DEBUG, "AVRn interlaced picture marker found\n");
1592  s->bottom_field ^= 1;
1593 
1594  goto next_field;
1595  }
1596 
1597  return 0;
1598 }
1599 
1600 static int mjpeg_decode_scan_progressive_ac(MJpegDecodeContext *s)
1601 {
1602  int Ss = s->Ss;
1603  int Se = s->Se;
1604  int Ah = s->Ah;
1605  int Al = s->Al;
1606  int mb_x, mb_y;
1607  int EOBRUN = 0;
1608  int c = s->comp_index[0];
1609  uint16_t *quant_matrix = s->quant_matrixes[s->quant_sindex[0]];
1610 
1611  av_assert0(Ss >= 0 && Ah >= 0 && Al >= 0);
1612  if (Se < Ss || Se > 63) {
1613  av_log(s->avctx, AV_LOG_ERROR, "SS/SE %d/%d is invalid\n", Ss, Se);
1614  return AVERROR_INVALIDDATA;
1615  }
1616 
1617  // s->coefs_finished is a bitmask for coefficients coded
1618  // Ss and Se are parameters telling start and end coefficients
1619  s->coefs_finished[c] |= (2ULL << Se) - (1ULL << Ss);
1620 
1621  s->restart_count = -1;
1622 
1623  for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1624  int block_idx = mb_y * s->block_stride[c];
1625  int16_t (*block)[64] = &s->blocks[c][block_idx];
1626  uint8_t *last_nnz = &s->last_nnz[c][block_idx];
1627  for (mb_x = 0; mb_x < s->mb_width; mb_x++, block++, last_nnz++) {
1628  int ret;
1629  int restart;
1630  ret = ff_mjpeg_handle_restart(s, &restart);
1631  if (ret < 0)
1632  return ret;
1633  if (restart)
1634  EOBRUN = 0;
1635 
1636  if (Ah)
1637  ret = decode_block_refinement(s, *block, last_nnz, s->ac_index[0],
1638  quant_matrix, Ss, Se, Al, &EOBRUN);
1639  else
1640  ret = decode_block_progressive(s, *block, last_nnz, s->ac_index[0],
1641  quant_matrix, Ss, Se, Al, &EOBRUN);
1642 
1643  if (ret >= 0 && get_bits_left(&s->gb) < 0)
1644  ret = AVERROR_INVALIDDATA;
1645  if (ret < 0) {
1646  av_log(s->avctx, AV_LOG_ERROR,
1647  "error y=%d x=%d\n", mb_y, mb_x);
1648  return AVERROR_INVALIDDATA;
1649  }
1650  }
1651  }
1652  return 0;
1653 }
1654 
1655 static void mjpeg_idct_scan_progressive_ac(MJpegDecodeContext *s)
1656 {
1657  int mb_x, mb_y;
1658  int c;
1659  const int bytes_per_pixel = 1 + (s->bits > 8);
1660  const int block_size = s->lossless ? 1 : 8;
1661 
1662  for (c = 0; c < s->nb_components; c++) {
1663  uint8_t *data = s->picture_ptr->data[c];
1664  int linesize = s->linesize[c];
1665  int h = s->h_max / s->h_count[c];
1666  int v = s->v_max / s->v_count[c];
1667  int mb_width = (s->width + h * block_size - 1) / (h * block_size);
1668  int mb_height = (s->height + v * block_size - 1) / (v * block_size);
1669 
1670  if (~s->coefs_finished[c])
1671  av_log(s->avctx, AV_LOG_WARNING, "component %d is incomplete\n", c);
1672 
1673  if (s->interlaced && s->bottom_field)
1674  data += linesize >> 1;
1675 
1676  for (mb_y = 0; mb_y < mb_height; mb_y++) {
1677  uint8_t *ptr = data + (mb_y * linesize * 8 >> s->avctx->lowres);
1678  int block_idx = mb_y * s->block_stride[c];
1679  int16_t (*block)[64] = &s->blocks[c][block_idx];
1680  for (mb_x = 0; mb_x < mb_width; mb_x++, block++) {
1681  s->idsp.idct_put(ptr, linesize, *block);
1682  if (s->bits & 7)
1683  shift_output(s, ptr, linesize);
1684  ptr += bytes_per_pixel * 8 >> s->avctx->lowres;
1685  }
1686  }
1687  }
1688 }
1689 
1690 int ff_mjpeg_decode_sos(MJpegDecodeContext *s)
1691 {
1692  int len, i, h, v;
1693  int index, id, ret;
1694  const int block_size = s->lossless ? 1 : 8;
1695 
1696  if (!s->got_picture) {
1697  av_log(s->avctx, AV_LOG_WARNING,
1698  "Can not process SOS before SOF, skipping\n");
1699  return AVERROR_INVALIDDATA;
1700  }
1701 
1702  ret = mjpeg_parse_len(s, &len, "sos");
1703  if (ret < 0)
1704  return ret;
1705  if (len < 1)
1706  return AVERROR_INVALIDDATA;
1707  s->nb_components_sos = bytestream2_get_byteu(&s->gB);
1708  if (s->nb_components_sos == 0 || s->nb_components_sos > MAX_COMPONENTS) {
1709  avpriv_report_missing_feature(s->avctx,
1710  "decode_sos: nb_components (%d)",
1711  s->nb_components_sos);
1712  return AVERROR_PATCHWELCOME;
1713  }
1714  if (len != 4 + 2 * s->nb_components_sos) {
1715  av_log(s->avctx, AV_LOG_ERROR, "decode_sos: len(%d) mismatch %d components\n", len, s->nb_components_sos);
1716  return AVERROR_INVALIDDATA;
1717  }
1718  for (i = 0; i < s->nb_components_sos; i++) {
1719  id = bytestream2_get_byteu(&s->gB);
1720  av_log(s->avctx, AV_LOG_DEBUG, "component: %d\n", id);
1721  /* find component index */
1722  for (index = 0; index < s->nb_components; index++)
1723  if (id == s->component_id[index])
1724  break;
1725  if (index == s->nb_components) {
1726  av_log(s->avctx, AV_LOG_ERROR,
1727  "decode_sos: index(%d) out of components\n", index);
1728  return AVERROR_INVALIDDATA;
1729  }
1730  /* Metasoft MJPEG codec has Cb and Cr swapped */
1731  if (s->avctx->codec_tag == MKTAG('M', 'T', 'S', 'J')
1732  && s->nb_components_sos == 3 && s->nb_components == 3 && i)
1733  index = 3 - i;
1734 
1735  s->quant_sindex[i] = s->quant_index[index];
1736  s->nb_blocks[i] = s->h_count[index] * s->v_count[index];
1737  s->h_scount[i] = s->h_count[index];
1738  s->v_scount[i] = s->v_count[index];
1739 
1740  s->comp_index[i] = index;
1741 
1742  uint8_t b = bytestream2_get_byteu(&s->gB);
1743  s->dc_index[i] = b >> 4;
1744  s->ac_index[i] = b & 0x0F;
1745 
1746  if (s->dc_index[i] < 0 || s->ac_index[i] < 0 ||
1747  s->dc_index[i] >= 4 || s->ac_index[i] >= 4)
1748  goto out_of_range;
1749  if (!s->vlcs[0][s->dc_index[i]].table || !(s->progressive ? s->vlcs[2][s->ac_index[0]].table : s->vlcs[1][s->ac_index[i]].table))
1750  goto out_of_range;
1751  }
1752 
1753  s->Ss = bytestream2_get_byteu(&s->gB); /* JPEG Ss / lossless JPEG predictor / JPEG-LS NEAR */
1754  s->Se = bytestream2_get_byteu(&s->gB); /* JPEG Se / JPEG-LS ILV */
1755  uint8_t b = bytestream2_get_byteu(&s->gB);
1756  s->Ah = b >> 4; /* Ah */
1757  s->Al = b & 0x0F; /* Al */
1758 
1759  if (s->nb_components_sos > 1) {
1760  /* interleaved stream */
1761  s->mb_width = (s->width + s->h_max * block_size - 1) / (s->h_max * block_size);
1762  s->mb_height = (s->height + s->v_max * block_size - 1) / (s->v_max * block_size);
1763  } else if (!s->ls) { /* skip this for JPEG-LS */
1764  h = s->h_max / s->h_scount[0];
1765  v = s->v_max / s->v_scount[0];
1766  s->mb_width = (s->width + h * block_size - 1) / (h * block_size);
1767  s->mb_height = (s->height + v * block_size - 1) / (v * block_size);
1768  s->nb_blocks[0] = 1;
1769  s->h_scount[0] = 1;
1770  s->v_scount[0] = 1;
1771  }
1772 
1773  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1774  av_log(s->avctx, AV_LOG_DEBUG, "%s %s p:%d >>:%d ilv:%d bits:%d skip:%d %s comp:%d\n",
1775  s->lossless ? "lossless" : "sequential DCT", s->rgb ? "RGB" : "",
1776  s->Ss, s->Al, s->Se, s->bits, s->mjpb_skiptosod,
1777  s->pegasus_rct ? "PRCT" : (s->rct ? "RCT" : ""), s->nb_components_sos);
1778 
1779 
1780  /* mjpeg-b can have padding bytes between sos and image data, skip them */
1781  if (s->mjpb_skiptosod)
1782  bytestream2_skip(&s->gB, s->mjpb_skiptosod);
1783 
1784  if (s->avctx->hwaccel) {
1785  const uint8_t *buf_ptr;
1786  size_t buf_size;
1787 
1788  mjpeg_find_raw_scan_data(s, &buf_ptr, &buf_size);
1789 
1790  ret = FF_HW_CALL(s->avctx, decode_slice, buf_ptr, buf_size);
1791  if (ret < 0)
1792  return ret;
1793 
1794  } else {
1795  if (s->lossless) {
1796  av_assert0(s->picture_ptr == s->picture);
1797  if (CONFIG_JPEGLS_DECODER && s->ls) {
1798  if ((ret = ff_jpegls_decode_picture(s)) < 0)
1799  return ret;
1800  } else {
1801  if (s->rgb || s->bayer) {
1802  if ((ret = ljpeg_decode_rgb_scan(s)) < 0)
1803  return ret;
1804  } else {
1805  if ((ret = ljpeg_decode_yuv_scan(s)) < 0)
1806  return ret;
1807  }
1808  }
1809  } else {
1810  if (s->progressive && s->Ss) {
1811  av_assert0(s->picture_ptr == s->picture);
1812  if ((ret = mjpeg_decode_scan_progressive_ac(s)) < 0)
1813  return ret;
1814  } else {
1815  if ((ret = mjpeg_decode_scan(s)) < 0)
1816  return ret;
1817  }
1818  }
1819  }
1820 
1821  if (s->avctx->codec_id == AV_CODEC_ID_MEDIA100 ||
1822  s->avctx->codec_id == AV_CODEC_ID_MJPEGB ||
1823  s->avctx->codec_id == AV_CODEC_ID_THP) {
1824  /* Add the amount of bits read from the unescaped image data buffer
1825  * into the GetByteContext. */
1826  bytestream2_skipu(&s->gB, (get_bits_count(&s->gb) + 7) / 8);
1827  }
1828 
1829  return 0;
1830  out_of_range:
1831  av_log(s->avctx, AV_LOG_ERROR, "decode_sos: ac/dc index out of range\n");
1832  return AVERROR_INVALIDDATA;
1833 }
1834 
1835 static int mjpeg_decode_dri(MJpegDecodeContext *s)
1836 {
1837  if (bytestream2_get_be16u(&s->gB) != 4)
1838  return AVERROR_INVALIDDATA;
1839  s->restart_interval = bytestream2_get_be16u(&s->gB);
1840  av_log(s->avctx, AV_LOG_DEBUG, "restart interval: %d\n",
1841  s->restart_interval);
1842 
1843  return 0;
1844 }
1845 
1846 static int mjpeg_decode_app(MJpegDecodeContext *s, int start_code)
1847 {
1848  int len, id, i;
1849 
1850  int ret = mjpeg_parse_len(s, &len, "app");
1851  if (ret < 0)
1852  return AVERROR_INVALIDDATA;
1853 
1854  if (len < 4) {
1855  if (s->avctx->err_recognition & AV_EF_EXPLODE)
1856  return AVERROR_INVALIDDATA;
1857  av_log(s->avctx, AV_LOG_VERBOSE, "skipping APPx stub (len=%" PRId32 ")\n", len);
1858  goto out;
1859  }
1860 
1861  id = bytestream2_get_be32u(&s->gB);
1862  len -= 4;
1863 
1864  if (s->avctx->debug & FF_DEBUG_STARTCODE)
1865  av_log(s->avctx, AV_LOG_DEBUG, "APPx (%s / %8X) len=%d\n",
1866  av_fourcc2str(av_bswap32(id)), id, len);
1867 
1868  /* This fourcc is used by non-avid files too, it holds some
1869  information, but it's always present in AVID-created files. */
1870  if (id == AV_RB32("AVI1")) {
1871  /* structure:
1872  4bytes AVI1
1873  1bytes polarity
1874  1bytes always zero
1875  4bytes field_size
1876  4bytes field_size_less_padding
1877  */
1878  if (len < 1)
1879  goto out;
1880  i = bytestream2_get_byteu(&s->gB); len--;
1881  av_log(s->avctx, AV_LOG_DEBUG, "polarity %d\n", i);
1882  goto out;
1883  }
1884 
1885  if (id == AV_RB32("JFIF")) {
1886  int t_w, t_h, v1, v2;
1887  if (len < 8)
1888  goto out;
1889  bytestream2_skipu(&s->gB, 1); /* the trailing zero-byte */
1890  v1 = bytestream2_get_byteu(&s->gB);
1891  v2 = bytestream2_get_byteu(&s->gB);
1892  bytestream2_skipu(&s->gB, 1);
1893 
1894  s->avctx->sample_aspect_ratio.num = bytestream2_get_be16u(&s->gB);
1895  s->avctx->sample_aspect_ratio.den = bytestream2_get_be16u(&s->gB);
1896  if ( s->avctx->sample_aspect_ratio.num <= 0
1897  || s->avctx->sample_aspect_ratio.den <= 0) {
1898  s->avctx->sample_aspect_ratio.num = 0;
1899  s->avctx->sample_aspect_ratio.den = 1;
1900  }
1901 
1902  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1903  av_log(s->avctx, AV_LOG_INFO,
1904  "mjpeg: JFIF header found (version: %x.%x) SAR=%d/%d\n",
1905  v1, v2,
1906  s->avctx->sample_aspect_ratio.num,
1907  s->avctx->sample_aspect_ratio.den);
1908 
1909  len -= 8;
1910  if (len >= 2) {
1911  t_w = bytestream2_get_byteu(&s->gB);
1912  t_h = bytestream2_get_byteu(&s->gB);
1913  if (t_w && t_h) {
1914  /* skip thumbnail */
1915  if (len - 10 - (t_w * t_h * 3) > 0)
1916  len -= t_w * t_h * 3;
1917  }
1918  len -= 2;
1919  }
1920  goto out;
1921  }
1922 
1923  if ( id == AV_RB32("Adob")
1924  && len >= 8
1925  && bytestream2_peek_byteu(&s->gB) == 'e'
1926  && bytestream2_peek_be32u(&s->gB) != AV_RB32("e_CM")) {
1927  bytestream2_skipu(&s->gB, 1); /* 'e' */
1928  bytestream2_skipu(&s->gB, 2); /* version */
1929  bytestream2_skipu(&s->gB, 2); /* flags0 */
1930  bytestream2_skipu(&s->gB, 2); /* flags1 */
1931  s->adobe_transform = bytestream2_get_byteu(&s->gB);
1932  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1933  av_log(s->avctx, AV_LOG_INFO, "mjpeg: Adobe header found, transform=%d\n", s->adobe_transform);
1934  len -= 8;
1935  goto out;
1936  }
1937 
1938  if (id == AV_RB32("LJIF")) {
1939  int rgb = s->rgb;
1940  int pegasus_rct = s->pegasus_rct;
1941  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1942  av_log(s->avctx, AV_LOG_INFO,
1943  "Pegasus lossless jpeg header found\n");
1944  if (len < 9)
1945  goto out;
1946  bytestream2_skipu(&s->gB, 2); /* version ? */
1947  bytestream2_skipu(&s->gB, 2); /* unknown always 0? */
1948  bytestream2_skipu(&s->gB, 2); /* unknown always 0? */
1949  bytestream2_skipu(&s->gB, 2); /* unknown always 0? */
1950  switch (i = bytestream2_get_byteu(&s->gB)) {
1951  case 1:
1952  rgb = 1;
1953  pegasus_rct = 0;
1954  break;
1955  case 2:
1956  rgb = 1;
1957  pegasus_rct = 1;
1958  break;
1959  default:
1960  av_log(s->avctx, AV_LOG_ERROR, "unknown colorspace %d\n", i);
1961  }
1962 
1963  len -= 9;
1964  if (s->bayer)
1965  goto out;
1966  if (s->got_picture)
1967  if (rgb != s->rgb || pegasus_rct != s->pegasus_rct) {
1968  av_log(s->avctx, AV_LOG_WARNING, "Mismatching LJIF tag\n");
1969  goto out;
1970  }
1971 
1972  s->rgb = rgb;
1973  s->pegasus_rct = pegasus_rct;
1974 
1975  goto out;
1976  }
1977  if (id == AV_RL32("colr") && len > 0) {
1978  s->colr = bytestream2_get_byteu(&s->gB);
1979  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1980  av_log(s->avctx, AV_LOG_INFO, "COLR %d\n", s->colr);
1981  len--;
1982  goto out;
1983  }
1984  if (id == AV_RL32("xfrm") && len > 0) {
1985  s->xfrm = bytestream2_get_byteu(&s->gB);
1986  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1987  av_log(s->avctx, AV_LOG_INFO, "XFRM %d\n", s->xfrm);
1988  len--;
1989  goto out;
1990  }
1991 
1992  /* JPS extension by VRex */
1993  if (start_code == APP3 && id == AV_RB32("_JPS") && len >= 10) {
1994  int flags, layout, type;
1995  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1996  av_log(s->avctx, AV_LOG_INFO, "_JPSJPS_\n");
1997 
1998  bytestream2_skipu(&s->gB, 4); len -= 4; /* JPS_ */
1999  bytestream2_skipu(&s->gB, 2); len -= 2; /* block length */
2000  bytestream2_skipu(&s->gB, 1); /* reserved */
2001  flags = bytestream2_get_byteu(&s->gB);
2002  layout = bytestream2_get_byteu(&s->gB);
2003  type = bytestream2_get_byteu(&s->gB);
2004  len -= 4;
2005 
2006  av_freep(&s->stereo3d);
2007  s->stereo3d = av_stereo3d_alloc();
2008  if (!s->stereo3d) {
2009  goto out;
2010  }
2011  if (type == 0) {
2012  s->stereo3d->type = AV_STEREO3D_2D;
2013  } else if (type == 1) {
2014  switch (layout) {
2015  case 0x01:
2016  s->stereo3d->type = AV_STEREO3D_LINES;
2017  break;
2018  case 0x02:
2019  s->stereo3d->type = AV_STEREO3D_SIDEBYSIDE;
2020  break;
2021  case 0x03:
2022  s->stereo3d->type = AV_STEREO3D_TOPBOTTOM;
2023  break;
2024  }
2025  if (!(flags & 0x04)) {
2026  s->stereo3d->flags = AV_STEREO3D_FLAG_INVERT;
2027  }
2028  }
2029  goto out;
2030  }
2031 
2032  /* EXIF metadata */
2033  if (start_code == APP1 && id == AV_RB32("Exif") && len >= 2) {
2034  int ret;
2035 
2036  bytestream2_skipu(&s->gB, 2); // skip padding
2037  len -= 2;
2038 
2039  ret = av_exif_parse_buffer(s->avctx, s->gB.buffer, len, &s->exif_metadata, AV_EXIF_TIFF_HEADER);
2040  if (ret < 0) {
2041  av_log(s->avctx, AV_LOG_WARNING, "unable to parse EXIF buffer\n");
2042  goto out;
2043  }
2044 
2045  bytestream2_skipu(&s->gB, ret);
2046  len -= ret;
2047 
2048  goto out;
2049  }
2050 
2051  /* Apple MJPEG-A */
2052  if ((start_code == APP1) && (len > (0x28 - 8))) {
2053  id = bytestream2_get_be32u(&s->gB);
2054  len -= 4;
2055  /* Apple MJPEG-A */
2056  if (id == AV_RB32("mjpg")) {
2057  /* structure:
2058  4bytes field size
2059  4bytes pad field size
2060  4bytes next off
2061  4bytes quant off
2062  4bytes huff off
2063  4bytes image off
2064  4bytes scan off
2065  4bytes data off
2066  */
2067  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2068  av_log(s->avctx, AV_LOG_INFO, "mjpeg: Apple MJPEG-A header found\n");
2069  }
2070  }
2071 
2072  if (start_code == APP2 && id == AV_RB32("ICC_") && len >= 10) {
2073  int id2;
2074  unsigned seqno;
2075  unsigned nummarkers;
2076 
2077  id = bytestream2_get_be32u(&s->gB);
2078  id2 = bytestream2_get_be24u(&s->gB);
2079  len -= 7;
2080  if (id != AV_RB32("PROF") || id2 != AV_RB24("ILE")) {
2081  av_log(s->avctx, AV_LOG_WARNING, "Invalid ICC_PROFILE header in APP2\n");
2082  goto out;
2083  }
2084 
2085  bytestream2_skipu(&s->gB, 1);
2086  seqno = bytestream2_get_byteu(&s->gB);
2087  len -= 2;
2088  if (seqno == 0) {
2089  av_log(s->avctx, AV_LOG_WARNING, "Invalid sequence number in APP2\n");
2090  goto out;
2091  }
2092 
2093  nummarkers = bytestream2_get_byteu(&s->gB);
2094  len -= 1;
2095  if (nummarkers == 0) {
2096  av_log(s->avctx, AV_LOG_WARNING, "Invalid number of markers coded in APP2\n");
2097  goto out;
2098  } else if (s->iccnum != 0 && nummarkers != s->iccnum) {
2099  av_log(s->avctx, AV_LOG_WARNING, "Mismatch in coded number of ICC markers between markers\n");
2100  goto out;
2101  } else if (seqno > nummarkers) {
2102  av_log(s->avctx, AV_LOG_WARNING, "Mismatching sequence number and coded number of ICC markers\n");
2103  goto out;
2104  }
2105 
2106  /* Allocate if this is the first APP2 we've seen. */
2107  if (s->iccnum == 0) {
2108  if (!FF_ALLOCZ_TYPED_ARRAY(s->iccentries, nummarkers)) {
2109  av_log(s->avctx, AV_LOG_ERROR, "Could not allocate ICC data arrays\n");
2110  return AVERROR(ENOMEM);
2111  }
2112  s->iccnum = nummarkers;
2113  }
2114 
2115  if (s->iccentries[seqno - 1].data) {
2116  av_log(s->avctx, AV_LOG_WARNING, "Duplicate ICC sequence number\n");
2117  goto out;
2118  }
2119 
2120  s->iccentries[seqno - 1].length = len;
2121  s->iccentries[seqno - 1].data = av_malloc(len);
2122  if (!s->iccentries[seqno - 1].data) {
2123  av_log(s->avctx, AV_LOG_ERROR, "Could not allocate ICC data buffer\n");
2124  return AVERROR(ENOMEM);
2125  }
2126 
2127  bytestream2_get_bufferu(&s->gB, s->iccentries[seqno - 1].data, len);
2128  len = 0;
2129  s->iccread++;
2130 
2131  if (s->iccread > s->iccnum)
2132  av_log(s->avctx, AV_LOG_WARNING, "Read more ICC markers than are supposed to be coded\n");
2133  }
2134 
2135 out:
2136  /* slow but needed for extreme adobe jpegs */
2137  if (len < 0)
2138  av_log(s->avctx, AV_LOG_ERROR,
2139  "mjpeg: error, decode_app parser read over the end\n");
2140  if (len > 0)
2141  bytestream2_skipu(&s->gB, len);
2142 
2143  return 0;
2144 }
2145 
2146 static int mjpeg_decode_com(MJpegDecodeContext *s)
2147 {
2148  int len;
2149  int ret = mjpeg_parse_len(s, &len, "com");
2150  if (ret < 0)
2151  return ret;
2152  if (!len)
2153  return 0;
2154 
2155  int i;
2156  char *cbuf = av_malloc(len + 1);
2157  if (!cbuf)
2158  return AVERROR(ENOMEM);
2159 
2160  for (i = 0; i < len; i++)
2161  cbuf[i] = bytestream2_get_byteu(&s->gB);
2162  if (cbuf[i - 1] == '\n')
2163  cbuf[i - 1] = 0;
2164  else
2165  cbuf[i] = 0;
2166 
2167  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2168  av_log(s->avctx, AV_LOG_INFO, "comment: '%s'\n", cbuf);
2169 
2170  /* buggy avid, it puts EOI only at every 10th frame */
2171  if (!strncmp(cbuf, "AVID", 4)) {
2172  parse_avid(s, cbuf, len);
2173  } else if (!strcmp(cbuf, "CS=ITU601"))
2174  s->cs_itu601 = 1;
2175  else if ((!strncmp(cbuf, "Intel(R) JPEG Library, version 1", 32) && s->avctx->codec_tag) ||
2176  (!strncmp(cbuf, "Metasoft MJPEG Codec", 20)))
2177  s->flipped = 1;
2178  else if (!strcmp(cbuf, "MULTISCOPE II")) {
2179  s->avctx->sample_aspect_ratio = (AVRational) { 1, 2 };
2180  s->multiscope = 2;
2181  }
2182 
2183  av_free(cbuf);
2184 
2185  return 0;
2186 }
2187 
2188 /* return the 8 bit start code value and update the search
2189  state. Return -1 if no start code found */
2190 int ff_mjpeg_find_marker(const uint8_t **pbuf_ptr, const uint8_t *buf_end)
2191 {
2192  const uint8_t *buf_ptr;
2193  int val;
2194 
2195  buf_ptr = *pbuf_ptr;
2196  while ((buf_ptr = memchr(buf_ptr, 0xff, buf_end - buf_ptr))) {
2197  buf_ptr++;
2198  while (buf_ptr < buf_end) {
2199  val = *buf_ptr++;
2200  if (val != 0xff) {
2201  if ((val >= SOF0) && (val <= COM))
2202  goto found;
2203  break;
2204  }
2205  }
2206  }
2207  buf_ptr = buf_end;
2208  val = -1;
2209 found:
2210  ff_dlog(NULL, "find_marker skipped %td bytes\n",
2211  (buf_ptr - *pbuf_ptr) - (val < 0 ? 0 : 2));
2212  *pbuf_ptr = buf_ptr;
2213  return val;
2214 }
2215 
2216 static void mjpeg_find_raw_scan_data(MJpegDecodeContext *s,
2217  const uint8_t **pbuf_ptr, size_t *pbuf_size)
2218 {
2219  const uint8_t *buf_ptr = s->gB.buffer;
2220  const uint8_t *buf_end = buf_ptr + bytestream2_get_bytes_left(&s->gB);
2221 
2222  /* Find size of image data buffer (including restart markers).
2223  * No unescaping is performed. */
2224  const uint8_t *ptr = buf_ptr;
2225  while ((ptr = memchr(ptr, 0xff, buf_end - ptr))) {
2226  ptr++;
2227  if (ptr < buf_end) {
2228  uint8_t x = *ptr++;
2229  /* Discard multiple optional 0xFF fill bytes. */
2230  while (x == 0xff && ptr < buf_end)
2231  x = *ptr++;
2232  if (x && (x < RST0 || x > RST7)) {
2233  /* Non-restart marker */
2234  ptr -= 2;
2235  goto found_hw;
2236  }
2237  }
2238  }
2239  ptr = buf_end;
2240 found_hw:
2241  *pbuf_ptr = buf_ptr;
2242  *pbuf_size = ptr - buf_ptr;
2243  bytestream2_skipu(&s->gB, *pbuf_size);
2244 }
2245 
2246 int ff_mjpeg_unescape_sos(MJpegDecodeContext *s)
2247 {
2248  const uint8_t *buf_ptr = s->gB.buffer;
2249  const uint8_t *buf_end = buf_ptr + bytestream2_get_bytes_left(&s->gB);
2250  const uint8_t *unescaped_buf_ptr;
2251  size_t unescaped_buf_size;
2252 
2253  if (s->avctx->codec_id == AV_CODEC_ID_MEDIA100 ||
2254  s->avctx->codec_id == AV_CODEC_ID_MJPEGB ||
2255  s->avctx->codec_id == AV_CODEC_ID_THP) {
2256  /* The image data buffer is already unescaped. The only way to
2257  * find the size of the buffer is by fully decoding it. */
2258  unescaped_buf_ptr = buf_ptr;
2259  unescaped_buf_size = buf_end - buf_ptr;
2260  goto the_end;
2261  }
2262 
2263  av_fast_padded_malloc(&s->buffer, &s->buffer_size, buf_end - buf_ptr);
2264  if (!s->buffer)
2265  return AVERROR(ENOMEM);
2266 
2267  /* unescape buffer of SOS, use special treatment for JPEG-LS */
2268  if (!s->ls) {
2269  const uint8_t *src = buf_ptr;
2270  const uint8_t *ptr = src;
2271  uint8_t *dst = s->buffer;
2272  PutByteContext pb;
2273 
2274  bytestream2_init_writer(&pb, dst, buf_end - src);
2275 
2276  while ((ptr = memchr(ptr, 0xff, buf_end - ptr))) {
2277  ptr++;
2278  if (ptr < buf_end) {
2279  /* Copy verbatim data. */
2280  ptrdiff_t length = (ptr - 1) - src;
2281  if (length > 0)
2282  bytestream2_put_bufferu(&pb, src, length);
2283 
2284  uint8_t x = *ptr++;
2285  /* Discard multiple optional 0xFF fill bytes. */
2286  while (x == 0xff && ptr < buf_end)
2287  x = *ptr++;
2288 
2289  src = ptr;
2290  if (x == 0) {
2291  /* Stuffed zero byte */
2292  bytestream2_put_byteu(&pb, 0xff);
2293  } else if (x >= RST0 && x <= RST7) {
2294  /* Restart marker */
2295  goto found;
2296  } else {
2297  /* Non-restart marker */
2298  ptr -= 2;
2299  goto found;
2300  }
2301  }
2302  }
2303  /* Copy remaining verbatim data. */
2304  ptr = buf_end;
2305  ptrdiff_t length = ptr - src;
2306  if (length > 0)
2307  bytestream2_put_bufferu(&pb, src, length);
2308 
2309 found:
2310  unescaped_buf_ptr = s->buffer;
2311  unescaped_buf_size = bytestream2_tell_p(&pb);
2312  memset(s->buffer + unescaped_buf_size, 0,
2313  AV_INPUT_BUFFER_PADDING_SIZE);
2314 
2315  bytestream2_skipu(&s->gB, ptr - buf_ptr);
2316 
2317  av_log(s->avctx, AV_LOG_DEBUG, "escaping removed %td bytes\n",
2318  (buf_end - buf_ptr) - (unescaped_buf_size));
2319  } else {
2320  const uint8_t *src = buf_ptr;
2321  const uint8_t *ptr = src;
2322  uint8_t *dst = s->buffer;
2323  PutBitContext pb;
2324 
2325  init_put_bits(&pb, dst, buf_end - src);
2326 
2327  while ((ptr = memchr(ptr, 0xff, buf_end - ptr))) {
2328  ptr++;
2329  if (ptr < buf_end) {
2330  /* Copy verbatim data. */
2331  ptrdiff_t length = (ptr - 1) - src;
2332  if (length > 0)
2333  ff_copy_bits(&pb, src, length * 8);
2334 
2335  uint8_t x = *ptr++;
2336  /* Discard multiple optional 0xFF fill bytes. */
2337  while (x == 0xff && ptr < buf_end)
2338  x = *ptr++;
2339 
2340  src = ptr;
2341  if (!(x & 0x80)) {
2342  /* Stuffed zero bit */
2343  put_bits(&pb, 15, 0x7f80 | x);
2344  } else if (x >= RST0 && x <= RST7) {
2345  /* Restart marker */
2346  goto found_ls;
2347  } else {
2348  /* Non-restart marker */
2349  ptr -= 2;
2350  goto found_ls;
2351  }
2352  }
2353  }
2354  /* Copy remaining verbatim data. */
2355  ptr = buf_end;
2356  ptrdiff_t length = ptr - src;
2357  if (length > 0)
2358  ff_copy_bits(&pb, src, length * 8);
2359 
2360 found_ls:
2361  flush_put_bits(&pb);
2362 
2363  unescaped_buf_ptr = dst;
2364  unescaped_buf_size = put_bytes_output(&pb);
2365  memset(s->buffer + unescaped_buf_size, 0,
2366  AV_INPUT_BUFFER_PADDING_SIZE);
2367 
2368  bytestream2_skipu(&s->gB, ptr - buf_ptr);
2369  }
2370 
2371 the_end:
2372  return init_get_bits8(&s->gb, unescaped_buf_ptr, unescaped_buf_size);
2373 }
2374 
2375 static void reset_icc_profile(MJpegDecodeContext *s)
2376 {
2377  int i;
2378 
2379  if (s->iccentries) {
2380  for (i = 0; i < s->iccnum; i++)
2381  av_freep(&s->iccentries[i].data);
2382  av_freep(&s->iccentries);
2383  }
2384 
2385  s->iccread = 0;
2386  s->iccnum = 0;
2387 }
2388 
2389 int ff_mjpeg_decode_frame_from_buf(AVCodecContext *avctx, AVFrame *frame,
2390  int *got_frame, const AVPacket *avpkt,
2391  const uint8_t *buf, const int buf_size)
2392 {
2393  MJpegDecodeContext *s = avctx->priv_data;
2394  const uint8_t *buf_end, *buf_ptr;
2395  int hshift, vshift;
2396  int start_code;
2397  int index;
2398  int ret = 0;
2399  int is16bit;
2400 
2401  s->force_pal8 = 0;
2402 
2403  s->buf_size = buf_size;
2404 
2405  av_exif_free(&s->exif_metadata);
2406  av_freep(&s->stereo3d);
2407  s->adobe_transform = -1;
2408 
2409  if (s->iccnum != 0)
2410  reset_icc_profile(s);
2411 
2412 redo_for_pal8:
2413  buf_ptr = buf;
2414  buf_end = buf + buf_size;
2415  while (buf_ptr < buf_end) {
2416  /* find start next marker */
2417  start_code = ff_mjpeg_find_marker(&buf_ptr, buf_end);
2418  /* EOF */
2419  if (start_code < 0)
2420  break;
2421 
2422  ptrdiff_t bytes_left = buf_end - buf_ptr;
2423  if (bytes_left > INT_MAX / 8) {
2424  av_log(avctx, AV_LOG_ERROR,
2425  "MJPEG packet 0x%x too big (%td/%d), corrupt data?\n",
2426  start_code, bytes_left, buf_size);
2427  return AVERROR_INVALIDDATA;
2428  }
2429  av_log(avctx, AV_LOG_DEBUG, "marker=%x avail_size_in_buf=%td\n",
2430  start_code, buf_end - buf_ptr);
2431 
2432  bytestream2_init(&s->gB, buf_ptr, bytes_left);
2433 
2434  if (avctx->debug & FF_DEBUG_STARTCODE)
2435  av_log(avctx, AV_LOG_DEBUG, "startcode: %X\n", start_code);
2436 
2437  /* process markers */
2438  if (start_code >= RST0 && start_code <= RST7) {
2439  av_log(avctx, AV_LOG_DEBUG,
2440  "restart marker: %d\n", start_code & 0x0f);
2441  /* APP fields */
2442  } else if (start_code >= APP0 && start_code <= APP15) {
2443  if ((ret = mjpeg_decode_app(s, start_code)) < 0)
2444  av_log(avctx, AV_LOG_ERROR, "unable to decode APP fields: %s\n",
2445  av_err2str(ret));
2446  /* Comment */
2447  } else if (start_code == COM) {
2448  ret = mjpeg_decode_com(s);
2449  if (ret < 0)
2450  return ret;
2451  } else if (start_code == DQT) {
2452  ret = ff_mjpeg_decode_dqt(s);
2453  if (ret < 0)
2454  return ret;
2455  }
2456 
2457  ret = -1;
2458 
2459  if (!CONFIG_JPEGLS_DECODER &&
2460  (start_code == SOF55 || start_code == LSE)) {
2461  av_log(avctx, AV_LOG_ERROR, "JPEG-LS support not enabled.\n");
2462  return AVERROR(ENOSYS);
2463  }
2464 
2465  if (avctx->skip_frame == AVDISCARD_ALL) {
2466  switch (start_code) {
2467  case SOF0:
2468  case SOF1:
2469  case SOF2:
2470  case SOF3:
2471  case SOF55:
2472  break;
2473  default:
2474  goto skip;
2475  }
2476  }
2477 
2478  switch (start_code) {
2479  case SOI:
2480  s->restart_interval = 0;
2481  s->raw_image_buffer = buf_ptr;
2482  s->raw_image_buffer_size = buf_end - buf_ptr;
2483  /* nothing to do on SOI */
2484  break;
2485  case DHT:
2486  if ((ret = ff_mjpeg_decode_dht(s)) < 0) {
2487  av_log(avctx, AV_LOG_ERROR, "huffman table decode error\n");
2488  goto fail;
2489  }
2490  break;
2491  case SOF0:
2492  case SOF1:
2493  if (start_code == SOF0)
2494  avctx->profile = AV_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT;
2495  else
2496  avctx->profile = AV_PROFILE_MJPEG_HUFFMAN_EXTENDED_SEQUENTIAL_DCT;
2497  s->lossless = 0;
2498  s->ls = 0;
2499  s->progressive = 0;
2500  if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2501  goto fail;
2502  break;
2503  case SOF2:
2504  avctx->profile = AV_PROFILE_MJPEG_HUFFMAN_PROGRESSIVE_DCT;
2505  s->lossless = 0;
2506  s->ls = 0;
2507  s->progressive = 1;
2508  if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2509  goto fail;
2510  break;
2511  case SOF3:
2512  avctx->profile = AV_PROFILE_MJPEG_HUFFMAN_LOSSLESS;
2513 #if FF_API_CODEC_PROPS
2514 FF_DISABLE_DEPRECATION_WARNINGS
2515  avctx->properties |= FF_CODEC_PROPERTY_LOSSLESS;
2516 FF_ENABLE_DEPRECATION_WARNINGS
2517 #endif
2518  s->lossless = 1;
2519  s->ls = 0;
2520  s->progressive = 0;
2521  if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2522  goto fail;
2523  break;
2524  case SOF55:
2525  avctx->profile = AV_PROFILE_MJPEG_JPEG_LS;
2526 #if FF_API_CODEC_PROPS
2527 FF_DISABLE_DEPRECATION_WARNINGS
2528  avctx->properties |= FF_CODEC_PROPERTY_LOSSLESS;
2529 FF_ENABLE_DEPRECATION_WARNINGS
2530 #endif
2531  s->lossless = 1;
2532  s->ls = 1;
2533  s->progressive = 0;
2534  if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2535  goto fail;
2536  break;
2537  case LSE:
2538  if (!CONFIG_JPEGLS_DECODER ||
2539  (ret = ff_jpegls_decode_lse(s)) < 0)
2540  goto fail;
2541  if (ret == 1)
2542  goto redo_for_pal8;
2543  break;
2544  case EOI:
2545 eoi_parser:
2546  if (!avctx->hwaccel &&
2547  s->progressive && s->cur_scan && s->got_picture)
2548  mjpeg_idct_scan_progressive_ac(s);
2549  s->cur_scan = 0;
2550  if (!s->got_picture) {
2551  av_log(avctx, AV_LOG_WARNING,
2552  "Found EOI before any SOF, ignoring\n");
2553  break;
2554  }
2555  if (s->interlaced) {
2556  s->bottom_field ^= 1;
2557  /* if not bottom field, do not output image yet */
2558  if (s->bottom_field == !s->interlace_polarity)
2559  break;
2560  }
2561  if (avctx->hwaccel) {
2562  ret = FF_HW_SIMPLE_CALL(avctx, end_frame);
2563  if (ret < 0)
2564  return ret;
2565 
2566  av_freep(&s->hwaccel_picture_private);
2567  }
2568  if ((ret = av_frame_ref(frame, s->picture_ptr)) < 0)
2569  return ret;
2570  if (s->lossless)
2571  frame->flags |= AV_FRAME_FLAG_LOSSLESS;
2572  *got_frame = 1;
2573  s->got_picture = 0;
2574 
2575  if (!s->lossless && avctx->debug & FF_DEBUG_QP) {
2576  int qp = FFMAX3(s->qscale[0],
2577  s->qscale[1],
2578  s->qscale[2]);
2579 
2580  av_log(avctx, AV_LOG_DEBUG, "QP: %d\n", qp);
2581  }
2582 
2583  goto the_end;
2584  case SOS:
2585  s->cur_scan++;
2586 
2587  if ((ret = ff_mjpeg_decode_sos(s)) < 0 &&
2588  (avctx->err_recognition & AV_EF_EXPLODE))
2589  goto fail;
2590  break;
2591  case DRI:
2592  if ((ret = mjpeg_decode_dri(s)) < 0)
2593  return ret;
2594  break;
2595  case SOF5:
2596  case SOF6:
2597  case SOF7:
2598  case SOF9:
2599  case SOF10:
2600  case SOF11:
2601  case SOF13:
2602  case SOF14:
2603  case SOF15:
2604  case JPG:
2605  av_log(avctx, AV_LOG_ERROR,
2606  "mjpeg: unsupported coding type (%x)\n", start_code);
2607  break;
2608  }
2609 
2610  if (avctx->skip_frame == AVDISCARD_ALL) {
2611  switch (start_code) {
2612  case SOF0:
2613  case SOF1:
2614  case SOF2:
2615  case SOF3:
2616  case SOF55:
2617  s->got_picture = 0;
2618  goto the_end_no_picture;
2619  }
2620  }
2621 
2622 skip:
2623  /* eof process start code */
2624  buf_ptr += bytestream2_tell(&s->gB);
2625  av_log(avctx, AV_LOG_DEBUG,
2626  "marker parser used %d bytes\n",
2627  bytestream2_tell(&s->gB));
2628  }
2629  if (s->got_picture && s->cur_scan) {
2630  av_log(avctx, AV_LOG_WARNING, "EOI missing, emulating\n");
2631  goto eoi_parser;
2632  }
2633  av_log(avctx, AV_LOG_FATAL, "No JPEG data found in image\n");
2634  return AVERROR_INVALIDDATA;
2635 fail:
2636  s->got_picture = 0;
2637  return ret;
2638 the_end:
2639 
2640  is16bit = av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].step > 1;
2641 
2642  if (AV_RB32(s->upscale_h)) {
2643  int p;
2644  av_assert0(avctx->pix_fmt == AV_PIX_FMT_YUVJ444P ||
2645  avctx->pix_fmt == AV_PIX_FMT_YUV444P ||
2646  avctx->pix_fmt == AV_PIX_FMT_YUVJ440P ||
2647  avctx->pix_fmt == AV_PIX_FMT_YUV440P ||
2648  avctx->pix_fmt == AV_PIX_FMT_YUVA444P ||
2649  avctx->pix_fmt == AV_PIX_FMT_YUVJ422P ||
2650  avctx->pix_fmt == AV_PIX_FMT_YUV422P ||
2651  avctx->pix_fmt == AV_PIX_FMT_YUVJ420P ||
2652  avctx->pix_fmt == AV_PIX_FMT_YUV420P ||
2653  avctx->pix_fmt == AV_PIX_FMT_YUV420P16||
2654  avctx->pix_fmt == AV_PIX_FMT_YUVA420P ||
2655  avctx->pix_fmt == AV_PIX_FMT_YUVA420P16||
2656  avctx->pix_fmt == AV_PIX_FMT_GBRP ||
2657  avctx->pix_fmt == AV_PIX_FMT_GBRAP
2658  );
2659  ret = av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, &hshift, &vshift);
2660  if (ret)
2661  return ret;
2662 
2663  av_assert0(s->nb_components == av_pix_fmt_count_planes(s->picture_ptr->format));
2664  for (p = 0; p < s->nb_components; p++) {
2665  uint8_t *line = s->picture_ptr->data[p];
2666  int w = s->width;
2667  int h = s->height;
2668  if (!s->upscale_h[p])
2669  continue;
2670  if (p == 1 || p == 2) {
2671  w = AV_CEIL_RSHIFT(w, hshift);
2672  h = AV_CEIL_RSHIFT(h, vshift);
2673  }
2674  if (s->upscale_v[p] == 1)
2675  h = (h + 1) >> 1;
2676  av_assert0(w > 0);
2677  for (int i = 0; i < h; i++) {
2678  if (s->upscale_h[p] == 1) {
2679  if (is16bit) ((uint16_t*)line)[w - 1] = ((uint16_t*)line)[(w - 1) / 2];
2680  else line[w - 1] = line[(w - 1) / 2];
2681  for (index = w - 2; index > 0; index--) {
2682  if (is16bit)
2683  ((uint16_t*)line)[index] = (((uint16_t*)line)[index / 2] + ((uint16_t*)line)[(index + 1) / 2]) >> 1;
2684  else
2685  line[index] = (line[index / 2] + line[(index + 1) / 2]) >> 1;
2686  }
2687  } else if (s->upscale_h[p] == 2) {
2688  if (is16bit) {
2689  ((uint16_t*)line)[w - 1] = ((uint16_t*)line)[(w - 1) / 3];
2690  if (w > 1)
2691  ((uint16_t*)line)[w - 2] = ((uint16_t*)line)[w - 1];
2692  } else {
2693  line[w - 1] = line[(w - 1) / 3];
2694  if (w > 1)
2695  line[w - 2] = line[w - 1];
2696  }
2697  for (index = w - 3; index > 0; index--) {
2698  line[index] = (line[index / 3] + line[(index + 1) / 3] + line[(index + 2) / 3] + 1) / 3;
2699  }
2700  } else if (s->upscale_h[p] == 4) {
2701  if (is16bit) {
2702  uint16_t *line16 = (uint16_t *) line;
2703  line16[w - 1] = line16[(w - 1) >> 2];
2704  if (w > 1)
2705  line16[w - 2] = (line16[(w - 1) >> 2] * 3 + line16[(w - 2) >> 2]) >> 2;
2706  if (w > 2)
2707  line16[w - 3] = (line16[(w - 1) >> 2] + line16[(w - 2) >> 2]) >> 1;
2708  } else {
2709  line[w - 1] = line[(w - 1) >> 2];
2710  if (w > 1)
2711  line[w - 2] = (line[(w - 1) >> 2] * 3 + line[(w - 2) >> 2]) >> 2;
2712  if (w > 2)
2713  line[w - 3] = (line[(w - 1) >> 2] + line[(w - 2) >> 2]) >> 1;
2714  }
2715  for (index = w - 4; index > 0; index--)
2716  line[index] = (line[(index + 3) >> 2] + line[(index + 2) >> 2]
2717  + line[(index + 1) >> 2] + line[index >> 2]) >> 2;
2718  }
2719  line += s->linesize[p];
2720  }
2721  }
2722  }
2723  if (AV_RB32(s->upscale_v)) {
2724  int p;
2725  av_assert0(avctx->pix_fmt == AV_PIX_FMT_YUVJ444P ||
2726  avctx->pix_fmt == AV_PIX_FMT_YUV444P ||
2727  avctx->pix_fmt == AV_PIX_FMT_YUVJ422P ||
2728  avctx->pix_fmt == AV_PIX_FMT_YUV422P ||
2729  avctx->pix_fmt == AV_PIX_FMT_YUVJ420P ||
2730  avctx->pix_fmt == AV_PIX_FMT_YUV420P ||
2731  avctx->pix_fmt == AV_PIX_FMT_YUV440P ||
2732  avctx->pix_fmt == AV_PIX_FMT_YUVJ440P ||
2733  avctx->pix_fmt == AV_PIX_FMT_YUVA444P ||
2734  avctx->pix_fmt == AV_PIX_FMT_YUVA420P ||
2735  avctx->pix_fmt == AV_PIX_FMT_YUVA420P16||
2736  avctx->pix_fmt == AV_PIX_FMT_GBRP ||
2737  avctx->pix_fmt == AV_PIX_FMT_GBRAP
2738  );
2739  ret = av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, &hshift, &vshift);
2740  if (ret)
2741  return ret;
2742 
2743  av_assert0(s->nb_components == av_pix_fmt_count_planes(s->picture_ptr->format));
2744  for (p = 0; p < s->nb_components; p++) {
2745  uint8_t *dst;
2746  int w = s->width;
2747  int h = s->height;
2748  if (!s->upscale_v[p])
2749  continue;
2750  if (p == 1 || p == 2) {
2751  w = AV_CEIL_RSHIFT(w, hshift);
2752  h = AV_CEIL_RSHIFT(h, vshift);
2753  }
2754  dst = &((uint8_t *)s->picture_ptr->data[p])[(h - 1) * s->linesize[p]];
2755  for (int i = h - 1; i; i--) {
2756  uint8_t *src1 = &((uint8_t *)s->picture_ptr->data[p])[i * s->upscale_v[p] / (s->upscale_v[p] + 1) * s->linesize[p]];
2757  uint8_t *src2 = &((uint8_t *)s->picture_ptr->data[p])[(i + 1) * s->upscale_v[p] / (s->upscale_v[p] + 1) * s->linesize[p]];
2758  if (s->upscale_v[p] != 2 && (src1 == src2 || i == h - 1)) {
2759  memcpy(dst, src1, w);
2760  } else {
2761  for (index = 0; index < w; index++)
2762  dst[index] = (src1[index] + src2[index]) >> 1;
2763  }
2764  dst -= s->linesize[p];
2765  }
2766  }
2767  }
2768  if (s->flipped && !s->rgb) {
2769  ret = av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, &hshift, &vshift);
2770  if (ret)
2771  return ret;
2772 
2773  av_assert0(s->nb_components == av_pix_fmt_count_planes(frame->format));
2774  for (index = 0; index < s->nb_components; index++) {
2775  int h = frame->height;
2776  if (index && index < 3)
2777  h = AV_CEIL_RSHIFT(h, vshift);
2778  if (frame->data[index]) {
2779  frame->data[index] += (h - 1) * frame->linesize[index];
2780  frame->linesize[index] *= -1;
2781  }
2782  }
2783  }
2784 
2785  if (avctx->pix_fmt == AV_PIX_FMT_GBRP) {
2786  av_assert0(s->nb_components == 3);
2787  FFSWAP(uint8_t *, frame->data[0], frame->data[2]);
2788  FFSWAP(uint8_t *, frame->data[0], frame->data[1]);
2789  FFSWAP(int, frame->linesize[0], frame->linesize[2]);
2790  FFSWAP(int, frame->linesize[0], frame->linesize[1]);
2791  }
2792 
2793  if (s->adobe_transform == 0 && avctx->pix_fmt == AV_PIX_FMT_GBRAP) {
2794  int w = s->picture_ptr->width;
2795  int h = s->picture_ptr->height;
2796  av_assert0(s->nb_components == 4);
2797  for (int i = 0; i < h; i++) {
2798  int j;
2799  uint8_t *dst[4];
2800  for (index = 0; index < 4; index++) {
2801  dst[index] = s->picture_ptr->data[index]
2802  + s->picture_ptr->linesize[index]*i;
2803  }
2804  for (j = 0; j < w; j++) {
2805  int k = dst[3][j];
2806  int r = dst[0][j] * k;
2807  int g = dst[1][j] * k;
2808  int b = dst[2][j] * k;
2809  dst[0][j] = g * 257 >> 16;
2810  dst[1][j] = b * 257 >> 16;
2811  dst[2][j] = r * 257 >> 16;
2812  }
2813  memset(dst[3], 255, w);
2814  }
2815  }
2816  if (s->adobe_transform == 2 && avctx->pix_fmt == AV_PIX_FMT_YUVA444P) {
2817  int w = s->picture_ptr->width;
2818  int h = s->picture_ptr->height;
2819  av_assert0(s->nb_components == 4);
2820  for (int i = 0; i < h; i++) {
2821  int j;
2822  uint8_t *dst[4];
2823  for (index = 0; index < 4; index++) {
2824  dst[index] = s->picture_ptr->data[index]
2825  + s->picture_ptr->linesize[index]*i;
2826  }
2827  for (j = 0; j < w; j++) {
2828  int k = dst[3][j];
2829  int r = (255 - dst[0][j]) * k;
2830  int g = (128 - dst[1][j]) * k;
2831  int b = (128 - dst[2][j]) * k;
2832  dst[0][j] = r * 257 >> 16;
2833  dst[1][j] = (g * 257 >> 16) + 128;
2834  dst[2][j] = (b * 257 >> 16) + 128;
2835  }
2836  memset(dst[3], 255, w);
2837  }
2838  }
2839 
2840  if (s->stereo3d) {
2841  AVStereo3D *stereo = av_stereo3d_create_side_data(frame);
2842  if (stereo) {
2843  stereo->type = s->stereo3d->type;
2844  stereo->flags = s->stereo3d->flags;
2845  }
2846  av_freep(&s->stereo3d);
2847  }
2848 
2849  if (s->iccnum != 0 && s->iccnum == s->iccread) {
2850  AVFrameSideData *sd;
2851  size_t offset = 0;
2852  int total_size = 0;
2853 
2854  /* Sum size of all parts. */
2855  for (int i = 0; i < s->iccnum; i++)
2856  total_size += s->iccentries[i].length;
2857 
2858  ret = ff_frame_new_side_data(avctx, frame, AV_FRAME_DATA_ICC_PROFILE, total_size, &sd);
2859  if (ret < 0) {
2860  av_log(avctx, AV_LOG_ERROR, "Could not allocate frame side data\n");
2861  return ret;
2862  }
2863 
2864  if (sd) {
2865  /* Reassemble the parts, which are now in-order. */
2866  for (int i = 0; i < s->iccnum; i++) {
2867  memcpy(sd->data + offset, s->iccentries[i].data, s->iccentries[i].length);
2868  offset += s->iccentries[i].length;
2869  }
2870  }
2871  }
2872 
2873  if (s->exif_metadata.entries) {
2874  ret = ff_decode_exif_attach_ifd(avctx, frame, &s->exif_metadata);
2875  av_exif_free(&s->exif_metadata);
2876  if (ret < 0)
2877  av_log(avctx, AV_LOG_WARNING, "couldn't attach EXIF metadata\n");
2878  }
2879 
2880  if (avctx->codec_id != AV_CODEC_ID_SMVJPEG &&
2881  (avctx->codec_tag == MKTAG('A', 'V', 'R', 'n') ||
2882  avctx->codec_tag == MKTAG('A', 'V', 'D', 'J')) &&
2883  avctx->coded_height > s->orig_height) {
2884  frame->height = AV_CEIL_RSHIFT(avctx->coded_height, avctx->lowres);
2885  frame->crop_top = frame->height - avctx->height;
2886  }
2887 
2888 the_end_no_picture:
2889  av_log(avctx, AV_LOG_DEBUG, "decode frame unused %td bytes\n",
2890  buf_end - buf_ptr);
2891  return buf_ptr - buf;
2892 }
2893 
2894 int ff_mjpeg_decode_frame(AVCodecContext *avctx, AVFrame *frame, int *got_frame,
2895  AVPacket *avpkt)
2896 {
2897  return ff_mjpeg_decode_frame_from_buf(avctx, frame, got_frame,
2898  avpkt, avpkt->data, avpkt->size);
2899 }
2900 
2901 
2902 /* mxpeg may call the following function (with a blank MJpegDecodeContext)
2903  * even without having called ff_mjpeg_decode_init(). */
2904 av_cold int ff_mjpeg_decode_end(AVCodecContext *avctx)
2905 {
2906  MJpegDecodeContext *s = avctx->priv_data;
2907  int i, j;
2908 
2909  if (s->interlaced && s->bottom_field == !s->interlace_polarity && s->got_picture && !avctx->frame_num) {
2910  av_log(avctx, AV_LOG_INFO, "Single field\n");
2911  }
2912 
2913  av_frame_free(&s->picture);
2914  s->picture_ptr = NULL;
2915 
2916  av_frame_free(&s->smv_frame);
2917 
2918  av_freep(&s->buffer);
2919  av_freep(&s->stereo3d);
2920  av_freep(&s->ljpeg_buffer);
2921  s->ljpeg_buffer_size = 0;
2922 
2923  for (i = 0; i < 3; i++) {
2924  for (j = 0; j < 4; j++)
2925  ff_vlc_free(&s->vlcs[i][j]);
2926  }
2927  for (i = 0; i < MAX_COMPONENTS; i++) {
2928  av_freep(&s->blocks[i]);
2929  av_freep(&s->last_nnz[i]);
2930  }
2931  av_exif_free(&s->exif_metadata);
2932 
2933  reset_icc_profile(s);
2934 
2935  av_freep(&s->hwaccel_picture_private);
2936  av_freep(&s->jls_state);
2937 
2938  return 0;
2939 }
2940 
2941 static av_cold void decode_flush(AVCodecContext *avctx)
2942 {
2943  MJpegDecodeContext *s = avctx->priv_data;
2944  s->got_picture = 0;
2945 
2946  s->smv_next_frame = 0;
2947  av_frame_unref(s->smv_frame);
2948 }
2949 
2950 #if CONFIG_MJPEG_DECODER
2951 #define OFFSET(x) offsetof(MJpegDecodeContext, x)
2952 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
2953 static const AVOption options[] = {
2954  { "extern_huff", "Use external huffman table.",
2955  OFFSET(extern_huff), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VD },
2956  { NULL },
2957 };
2958 
2959 static const AVClass mjpegdec_class = {
2960  .class_name = "MJPEG decoder",
2961  .item_name = av_default_item_name,
2962  .option = options,
2963  .version = LIBAVUTIL_VERSION_INT,
2964 };
2965 
2966 const FFCodec ff_mjpeg_decoder = {
2967  .p.name = "mjpeg",
2968  CODEC_LONG_NAME("MJPEG (Motion JPEG)"),
2969  .p.type = AVMEDIA_TYPE_VIDEO,
2970  .p.id = AV_CODEC_ID_MJPEG,
2971  .priv_data_size = sizeof(MJpegDecodeContext),
2972  .init = ff_mjpeg_decode_init,
2973  .close = ff_mjpeg_decode_end,
2974  FF_CODEC_DECODE_CB(ff_mjpeg_decode_frame),
2975  .flush = decode_flush,
2976  .p.capabilities = AV_CODEC_CAP_DR1,
2977  .p.max_lowres = 3,
2978  .p.priv_class = &mjpegdec_class,
2979  .p.profiles = NULL_IF_CONFIG_SMALL(ff_mjpeg_profiles),
2980  .caps_internal = FF_CODEC_CAP_INIT_CLEANUP |
2981  FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM |
2982  FF_CODEC_CAP_ICC_PROFILES,
2983  .hw_configs = (const AVCodecHWConfigInternal *const []) {
2984 #if CONFIG_MJPEG_NVDEC_HWACCEL
2985  HWACCEL_NVDEC(mjpeg),
2986 #endif
2987 #if CONFIG_MJPEG_VAAPI_HWACCEL
2988  HWACCEL_VAAPI(mjpeg),
2989 #endif
2990  NULL
2991  },
2992 };
2993 #endif
2994 #if CONFIG_THP_DECODER
2995 const FFCodec ff_thp_decoder = {
2996  .p.name = "thp",
2997  CODEC_LONG_NAME("Nintendo Gamecube THP video"),
2998  .p.type = AVMEDIA_TYPE_VIDEO,
2999  .p.id = AV_CODEC_ID_THP,
3000  .priv_data_size = sizeof(MJpegDecodeContext),
3001  .init = ff_mjpeg_decode_init,
3002  .close = ff_mjpeg_decode_end,
3003  FF_CODEC_DECODE_CB(ff_mjpeg_decode_frame),
3004  .flush = decode_flush,
3005  .p.capabilities = AV_CODEC_CAP_DR1,
3006  .p.max_lowres = 3,
3007  .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
3008 };
3009 #endif
3010 
3011 #if CONFIG_SMVJPEG_DECODER
3012 // SMV JPEG just stacks several output frames into one JPEG picture
3013 // we handle that by setting up the cropping parameters appropriately
3014 static void smv_process_frame(AVCodecContext *avctx, AVFrame *frame)
3015 {
3016  MJpegDecodeContext *s = avctx->priv_data;
3017 
3018  av_assert0((s->smv_next_frame + 1) * avctx->height <= avctx->coded_height);
3019 
3020  frame->width = avctx->coded_width;
3021  frame->height = avctx->coded_height;
3022  frame->crop_top = FFMIN(s->smv_next_frame * avctx->height, frame->height);
3023  frame->crop_bottom = frame->height - (s->smv_next_frame + 1) * avctx->height;
3024 
3025  if (s->smv_frame->pts != AV_NOPTS_VALUE)
3026  s->smv_frame->pts += s->smv_frame->duration;
3027  s->smv_next_frame = (s->smv_next_frame + 1) % s->smv_frames_per_jpeg;
3028 
3029  if (s->smv_next_frame == 0)
3030  av_frame_unref(s->smv_frame);
3031 }
3032 
3033 static int smvjpeg_receive_frame(AVCodecContext *avctx, AVFrame *frame)
3034 {
3035  MJpegDecodeContext *s = avctx->priv_data;
3036  AVPacket *const pkt = avctx->internal->in_pkt;
3037  int got_frame = 0;
3038  int ret;
3039 
3040  if (s->smv_next_frame > 0)
3041  goto return_frame;
3042 
3043  ret = ff_decode_get_packet(avctx, pkt);
3044  if (ret < 0)
3045  return ret;
3046 
3047  av_frame_unref(s->smv_frame);
3048 
3049  ret = ff_mjpeg_decode_frame(avctx, s->smv_frame, &got_frame, pkt);
3050  s->smv_frame->pkt_dts = pkt->dts;
3051  av_packet_unref(pkt);
3052  if (ret < 0)
3053  return ret;
3054 
3055  if (!got_frame)
3056  return AVERROR(EAGAIN);
3057 
3058  // packet duration covers all the frames in the packet
3059  s->smv_frame->duration /= s->smv_frames_per_jpeg;
3060 
3061 return_frame:
3062  av_assert0(s->smv_frame->buf[0]);
3063  ret = av_frame_ref(frame, s->smv_frame);
3064  if (ret < 0)
3065  return ret;
3066 
3067  smv_process_frame(avctx, frame);
3068  return 0;
3069 }
3070 
3071 const FFCodec ff_smvjpeg_decoder = {
3072  .p.name = "smvjpeg",
3073  CODEC_LONG_NAME("SMV JPEG"),
3074  .p.type = AVMEDIA_TYPE_VIDEO,
3075  .p.id = AV_CODEC_ID_SMVJPEG,
3076  .priv_data_size = sizeof(MJpegDecodeContext),
3077  .init = ff_mjpeg_decode_init,
3078  .close = ff_mjpeg_decode_end,
3079  FF_CODEC_RECEIVE_FRAME_CB(smvjpeg_receive_frame),
3080  .flush = decode_flush,
3081  .p.capabilities = AV_CODEC_CAP_DR1,
3082  .caps_internal = FF_CODEC_CAP_EXPORTS_CROPPING |
3083  FF_CODEC_CAP_INIT_CLEANUP,
3084 };
3085 #endif
FF_ALLOCZ_TYPED_ARRAY
#define FF_ALLOCZ_TYPED_ARRAY(p, nelem)
Definition: internal.h:78
flags
const SwsFlags flags[]
Definition: swscale.c:61
hwconfig.h
av_packet_unref
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: packet.c:432
AVCodecContext::hwaccel
const struct AVHWAccel * hwaccel
Hardware accelerator in use.
Definition: avcodec.h:1413
FF_ENABLE_DEPRECATION_WARNINGS
#define FF_ENABLE_DEPRECATION_WARNINGS
Definition: internal.h:73
ff_decode_get_packet
int ff_decode_get_packet(AVCodecContext *avctx, AVPacket *pkt)
Called by decoders to get the next packet for decoding.
Definition: decode.c:249
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:216
AV_PIX_FMT_CUDA
@ AV_PIX_FMT_CUDA
HW acceleration through CUDA.
Definition: pixfmt.h:260
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:71
name
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default minimum maximum flags name is the option name
Definition: writing_filters.txt:88
jpegtables.h
mjpeg.h
level
uint8_t level
Definition: svq3.c:208
AV_EF_EXPLODE
#define AV_EF_EXPLODE
abort decoding on minor error detection
Definition: defs.h:51
FF_CODEC_CAP_INIT_CLEANUP
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
Definition: codec_internal.h:42
blockdsp.h
get_bits_left
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:694
r
const char * r
Definition: vf_curves.c:127
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
decode_slice
static int decode_slice(AVCodecContext *c, void *arg)
Definition: ffv1dec.c:360
opt.h
bytestream2_get_bytes_left
static av_always_inline int bytestream2_get_bytes_left(const GetByteContext *g)
Definition: bytestream.h:158
av_exif_parse_buffer
int av_exif_parse_buffer(void *logctx, const uint8_t *buf, size_t size, AVExifMetadata *ifd, enum AVExifHeaderMode header_mode)
Decodes the EXIF data provided in the buffer and writes it into the struct *ifd.
Definition: exif.c:881
AVCodecContext::colorspace
enum AVColorSpace colorspace
YUV colorspace type.
Definition: avcodec.h:667
ff_get_format
int ff_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
Select the (possibly hardware accelerated) pixel format.
Definition: decode.c:1208
out
static FILE * out
Definition: movenc.c:55
put_bytes_output
static int put_bytes_output(const PutBitContext *s)
Definition: put_bits.h:99
SOS
@ SOS
Definition: mjpeg.h:72
mjpeg_copy_block
static av_always_inline void mjpeg_copy_block(MJpegDecodeContext *s, uint8_t *dst, const uint8_t *src, int linesize, int lowres)
Definition: mjpegdec.c:1412
is
The official guide to swscale for confused that is
Definition: swscale.txt:28
APP1
@ APP1
Definition: mjpeg.h:80
bytestream2_tell
static av_always_inline int bytestream2_tell(const GetByteContext *g)
Definition: bytestream.h:192
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:3456
ZERO_RUN
#define ZERO_RUN
Definition: mjpegdec.c:997
SOF0
@ SOF0
Definition: mjpeg.h:39
src1
const pixel * src1
Definition: h264pred_template.c:420
AVCodecContext::err_recognition
int err_recognition
Error recognition; may misdetect some more or less valid parts as errors.
Definition: avcodec.h:1406
GET_VLC
#define GET_VLC(code, name, gb, table, bits, max_depth)
If the vlc code is invalid and max_depth=1, then no bits will be removed.
Definition: get_bits.h:573
bytestream2_skipu
static av_always_inline void bytestream2_skipu(GetByteContext *g, unsigned int size)
Definition: bytestream.h:174
ff_smvjpeg_decoder
const FFCodec ff_smvjpeg_decoder
init_put_bits
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:62
get_bits_count
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:254
init_idct
static void init_idct(AVCodecContext *avctx)
Definition: mjpegdec.c:114
mask
int mask
Definition: mediacodecdec_common.c:154
RST7
@ RST7
Definition: mjpeg.h:68
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:64
mjpegdec.h
start_code
static const uint8_t start_code[]
Definition: videotoolboxenc.c:230
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:427
put_bits
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:154
AV_PIX_FMT_YUVA420P16
#define AV_PIX_FMT_YUVA420P16
Definition: pixfmt.h:595
AVCOL_RANGE_JPEG
@ AVCOL_RANGE_JPEG
Full range content.
Definition: pixfmt.h:777
ff_mjpeg_decoder
const FFCodec ff_mjpeg_decoder
internal.h
AVPacket::data
uint8_t * data
Definition: packet.h:588
SOF11
@ SOF11
Definition: mjpeg.h:50
AVCodecContext::field_order
enum AVFieldOrder field_order
Field order.
Definition: avcodec.h:690
AVOption
AVOption.
Definition: opt.h:429
b
#define b
Definition: input.c:42
ljpeg_decode_rgb_scan
static int ljpeg_decode_rgb_scan(MJpegDecodeContext *s)
Definition: mjpegdec.c:1081
jpeglsdec.h
data
const char data[16]
Definition: mxf.c:149
AVComponentDescriptor::step
int step
Number of elements between 2 horizontally consecutive pixels.
Definition: pixdesc.h:40
ff_mjpeg_val_dc
const uint8_t ff_mjpeg_val_dc[]
Definition: jpegtabs.h:34
FFCodec
Definition: codec_internal.h:127
AV_LOG_VERBOSE
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:226
FF_HW_SIMPLE_CALL
#define FF_HW_SIMPLE_CALL(avctx, function)
Definition: hwaccel_internal.h:176
AV_PIX_FMT_BGR24
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:76
AV_PIX_FMT_YUV440P
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:106
AV_CODEC_ID_MEDIA100
@ AV_CODEC_ID_MEDIA100
Definition: codec_id.h:322
UPDATE_CACHE
#define UPDATE_CACHE(name, gb)
Definition: get_bits.h:213
FFMAX
#define FFMAX(a, b)
Definition: macros.h:47
ff_mjpeg_bits_ac_chrominance
const uint8_t ff_mjpeg_bits_ac_chrominance[]
Definition: jpegtabs.h:66
AV_CODEC_ID_THP
@ AV_CODEC_ID_THP
Definition: codec_id.h:152
AV_CODEC_ID_MXPEG
@ AV_CODEC_ID_MXPEG
Definition: codec_id.h:198
ff_set_dimensions
int ff_set_dimensions(AVCodecContext *s, int width, int height)
Check that the provided frame dimensions are valid and set them on the codec context.
Definition: utils.c:91
init_get_bits
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:517
ff_idctdsp_init
av_cold void ff_idctdsp_init(IDCTDSPContext *c, AVCodecContext *avctx)
Definition: idctdsp.c:228
FF_DEBUG_PICT_INFO
#define FF_DEBUG_PICT_INFO
Definition: avcodec.h:1383
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:448
AV_FRAME_FLAG_TOP_FIELD_FIRST
#define AV_FRAME_FLAG_TOP_FIELD_FIRST
A flag to mark frames where the top field is displayed first if the content is interlaced.
Definition: frame.h:655
APP15
@ APP15
Definition: mjpeg.h:94
GET_CACHE
#define GET_CACHE(name, gb)
Definition: get_bits.h:251
ff_permute_scantable
av_cold void ff_permute_scantable(uint8_t dst[64], const uint8_t src[64], const uint8_t permutation[64])
Definition: idctdsp.c:30
close
static av_cold void close(AVCodecParserContext *s)
Definition: apv_parser.c:197
AV_STEREO3D_SIDEBYSIDE
@ AV_STEREO3D_SIDEBYSIDE
Views are next to each other.
Definition: stereo3d.h:64
bytestream2_skip
static av_always_inline void bytestream2_skip(GetByteContext *g, unsigned int size)
Definition: bytestream.h:168
av_pix_fmt_count_planes
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:3496
AVCOL_SPC_BT470BG
@ AVCOL_SPC_BT470BG
also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM / IEC 61966-2-4 xvYCC601
Definition: pixfmt.h:706
rgb
Definition: rpzaenc.c:60
ff_mjpeg_decode_dht
int ff_mjpeg_decode_dht(MJpegDecodeContext *s)
Definition: mjpegdec.c:248
ff_copy_bits
void ff_copy_bits(PutBitContext *pb, const uint8_t *src, int length)
Copy the content of src to the bitstream.
Definition: bitstream.c:49
shift_output
static void shift_output(MJpegDecodeContext *s, uint8_t *ptr, int linesize)
Definition: mjpegdec.c:1428
FFCodec::p
AVCodec p
The public AVCodec.
Definition: codec_internal.h:131
FFHWAccel
Definition: hwaccel_internal.h:34
AV_PIX_FMT_GBRAP
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:212
AVCodecContext::codec
const struct AVCodec * codec
Definition: avcodec.h:448
ff_mjpeg_decode_init
av_cold int ff_mjpeg_decode_init(AVCodecContext *avctx)
Definition: mjpegdec.c:123
AVCodecContext::skip_frame
enum AVDiscard skip_frame
Skip decoding for selected frames.
Definition: avcodec.h:1670
fail
#define fail()
Definition: checkasm.h:218
AV_STEREO3D_2D
@ AV_STEREO3D_2D
Video is not stereoscopic (and metadata has to be there).
Definition: stereo3d.h:52
AV_PIX_FMT_YUVA444P16
#define AV_PIX_FMT_YUVA444P16
Definition: pixfmt.h:597
SOF3
@ SOF3
Definition: mjpeg.h:42
GetBitContext
Definition: get_bits.h:109
ff_mjpeg_decode_frame_from_buf
int ff_mjpeg_decode_frame_from_buf(AVCodecContext *avctx, AVFrame *frame, int *got_frame, const AVPacket *avpkt, const uint8_t *buf, const int buf_size)
Definition: mjpegdec.c:2389
mjpeg_decode_com
static int mjpeg_decode_com(MJpegDecodeContext *s)
Definition: mjpegdec.c:2146
init_default_huffman_tables
static int init_default_huffman_tables(MJpegDecodeContext *s)
Definition: mjpegdec.c:61
ff_mjpeg_find_marker
int ff_mjpeg_find_marker(const uint8_t **pbuf_ptr, const uint8_t *buf_end)
Definition: mjpegdec.c:2190
av_exif_free
void av_exif_free(AVExifMetadata *ifd)
Frees all resources associated with the given EXIF metadata struct.
Definition: exif.c:658
val
static double val(void *priv, double ch)
Definition: aeval.c:77
av_pix_fmt_get_chroma_sub_sample
int av_pix_fmt_get_chroma_sub_sample(enum AVPixelFormat pix_fmt, int *h_shift, int *v_shift)
Utility function to access log2_chroma_w log2_chroma_h from the pixel format AVPixFmtDescriptor.
Definition: pixdesc.c:3484
mjpeg_decode_scan_progressive_ac
static int mjpeg_decode_scan_progressive_ac(MJpegDecodeContext *s)
Definition: mjpegdec.c:1600
type
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf type
Definition: writing_filters.txt:86
AVCodecContext::coded_height
int coded_height
Definition: avcodec.h:615
AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:522
ff_mjpeg_handle_restart
static int ff_mjpeg_handle_restart(MJpegDecodeContext *s, int *restart)
Definition: mjpegdec.h:214
av_frame_alloc
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:52
AV_PIX_FMT_YUVJ411P
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
Definition: pixfmt.h:283
ff_mjpeg_decode_sos
int ff_mjpeg_decode_sos(MJpegDecodeContext *s)
Definition: mjpegdec.c:1690
ff_mjpeg_profiles
const AVProfile ff_mjpeg_profiles[]
Definition: profiles.c:191
avassert.h
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:210
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
av_cold
#define av_cold
Definition: attributes.h:106
decode_dc_progressive
static int decode_dc_progressive(MJpegDecodeContext *s, int16_t *block, int component, int dc_index, uint16_t *quant_matrix, int Al)
Definition: mjpegdec.c:900
AV_PIX_FMT_YUV422P16
#define AV_PIX_FMT_YUV422P16
Definition: pixfmt.h:551
init_get_bits8
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
Definition: get_bits.h:544
FF_CODEC_PROPERTY_LOSSLESS
#define FF_CODEC_PROPERTY_LOSSLESS
Definition: avcodec.h:1646
AV_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT
#define AV_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT
Definition: defs.h:173
COM
@ COM
Definition: mjpeg.h:111
AV_FRAME_FLAG_KEY
#define AV_FRAME_FLAG_KEY
A flag to mark frames that are keyframes.
Definition: frame.h:642
AV_FIELD_UNKNOWN
@ AV_FIELD_UNKNOWN
Definition: defs.h:212
bytestream2_init_writer
static av_always_inline void bytestream2_init_writer(PutByteContext *p, uint8_t *buf, int buf_size)
Definition: bytestream.h:147
AV_PIX_FMT_YUVJ422P
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:86
CLOSE_READER
#define CLOSE_READER(name, gb)
Definition: get_bits.h:189
SOF5
@ SOF5
Definition: mjpeg.h:44
AVCodecContext::extradata_size
int extradata_size
Definition: avcodec.h:523
FF_CODEC_DECODE_CB
#define FF_CODEC_DECODE_CB(func)
Definition: codec_internal.h:347
AV_STEREO3D_LINES
@ AV_STEREO3D_LINES
Views are packed per line, as if interlaced.
Definition: stereo3d.h:126
ff_blockdsp_init
av_cold void ff_blockdsp_init(BlockDSPContext *c)
Definition: blockdsp.c:58
s
#define s(width, name)
Definition: cbs_vp9.c:198
AV_PIX_FMT_YUVA420P
@ AV_PIX_FMT_YUVA420P
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
Definition: pixfmt.h:108
ff_mjpeg_should_restart
static int ff_mjpeg_should_restart(MJpegDecodeContext *s)
Definition: mjpegdec.h:196
parse_avid
static void parse_avid(MJpegDecodeContext *s, uint8_t *buf, int len)
Definition: mjpegdec.c:104
AV_PIX_FMT_YUV444P16
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:552
AV_CEIL_RSHIFT
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:60
g
const char * g
Definition: vf_curves.c:128
APP3
@ APP3
Definition: mjpeg.h:82
AV_GET_BUFFER_FLAG_REF
#define AV_GET_BUFFER_FLAG_REF
The decoder will keep a reference to the frame and may reuse it later.
Definition: avcodec.h:411
bytestream2_tell_p
static av_always_inline int bytestream2_tell_p(const PutByteContext *p)
Definition: bytestream.h:197
bits
uint8_t bits
Definition: vp3data.h:128
av_assert0
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:42
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:296
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:231
AV_PIX_FMT_YUV420P16
#define AV_PIX_FMT_YUV420P16
Definition: pixfmt.h:550
RST0
@ RST0
Definition: mjpeg.h:61
decode.h
reset_icc_profile
static void reset_icc_profile(MJpegDecodeContext *s)
Definition: mjpegdec.c:2375
ff_mjpeg_decode_end
av_cold int ff_mjpeg_decode_end(AVCodecContext *avctx)
Definition: mjpegdec.c:2904
mjpeg_find_raw_scan_data
static void mjpeg_find_raw_scan_data(MJpegDecodeContext *s, const uint8_t **pbuf_ptr, size_t *pbuf_size)
Definition: mjpegdec.c:2216
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:73
av_mallocz
#define av_mallocz(s)
Definition: tableprint_vlc.h:31
SOF55
@ SOF55
JPEG-LS.
Definition: mjpeg.h:103
PutBitContext
Definition: put_bits.h:50
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:332
AV_PIX_FMT_YUVJ444P
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:87
AVCodecContext::codec_id
enum AVCodecID codec_id
Definition: avcodec.h:449
AVStereo3D::flags
int flags
Additional information about the frame packing.
Definition: stereo3d.h:212
mjpeg_parse_len
static int mjpeg_parse_len(MJpegDecodeContext *s, int *plen, const char *name)
Definition: mjpegdec.c:193
if
if(ret)
Definition: filter_design.txt:179
AVDISCARD_ALL
@ AVDISCARD_ALL
discard all
Definition: defs.h:232
AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:561
AV_PIX_FMT_RGBA64
#define AV_PIX_FMT_RGBA64
Definition: pixfmt.h:529
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
ff_decode_exif_attach_ifd
int ff_decode_exif_attach_ifd(AVCodecContext *avctx, AVFrame *frame, const AVExifMetadata *ifd)
Definition: decode.c:2430
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:76
av_clip_int16
#define av_clip_int16
Definition: common.h:115
AV_PIX_FMT_BGR48
#define AV_PIX_FMT_BGR48
Definition: pixfmt.h:530
NULL
#define NULL
Definition: coverity.c:32
mjpeg_idct_scan_progressive_ac
static void mjpeg_idct_scan_progressive_ac(MJpegDecodeContext *s)
Definition: mjpegdec.c:1655
copy_block2
static void copy_block2(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride, int h)
Definition: copy_block.h:27
AVERROR_PATCHWELCOME
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:64
run
uint8_t run
Definition: svq3.c:207
AV_EXIF_TIFF_HEADER
@ AV_EXIF_TIFF_HEADER
The TIFF header starts with 0x49492a00, or 0x4d4d002a.
Definition: exif.h:63
hwaccel_internal.h
AV_PROFILE_MJPEG_HUFFMAN_EXTENDED_SEQUENTIAL_DCT
#define AV_PROFILE_MJPEG_HUFFMAN_EXTENDED_SEQUENTIAL_DCT
Definition: defs.h:174
AVRational
Rational number (pair of numerator and denominator).
Definition: rational.h:58
ff_mjpeg_decode_dqt
int ff_mjpeg_decode_dqt(MJpegDecodeContext *s)
Definition: mjpegdec.c:205
SOF13
@ SOF13
Definition: mjpeg.h:52
AVCodecContext::internal
struct AVCodecInternal * internal
Private context used for internal data.
Definition: avcodec.h:474
AV_PIX_FMT_YUVJ420P
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:85
mjpeg_decode_dc
static int mjpeg_decode_dc(MJpegDecodeContext *s, int dc_index, int *val)
Definition: mjpegdec.c:836
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:242
get_bits1
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:391
AV_PICTURE_TYPE_I
@ AV_PICTURE_TYPE_I
Intra.
Definition: avutil.h:278
profiles.h
AV_FRAME_DATA_ICC_PROFILE
@ AV_FRAME_DATA_ICC_PROFILE
The data contains an ICC profile as an opaque octet buffer following the format described by ISO 1507...
Definition: frame.h:144
decode_block_progressive
static int decode_block_progressive(MJpegDecodeContext *s, int16_t *block, uint8_t *last_nnz, int ac_index, uint16_t *quant_matrix, int Ss, int Se, int Al, int *EOBRUN)
Definition: mjpegdec.c:917
options
Definition: swscale.c:43
LAST_SKIP_BITS
#define LAST_SKIP_BITS(name, gb, num)
Definition: get_bits.h:235
MJpegDecodeContext
Definition: mjpegdec.h:56
lowres
static int lowres
Definition: ffplay.c:332
ff_mjpeg_val_ac_chrominance
const uint8_t ff_mjpeg_val_ac_chrominance[]
Definition: jpegtabs.h:69
AV_PIX_FMT_GRAY8
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:81
get_vlc2
static av_always_inline int get_vlc2(GetBitContext *s, const VLCElem *table, int bits, int max_depth)
Parse a vlc code.
Definition: get_bits.h:651
AV_PIX_FMT_ABGR
@ AV_PIX_FMT_ABGR
packed ABGR 8:8:8:8, 32bpp, ABGRABGR...
Definition: pixfmt.h:101
DRI
@ DRI
Definition: mjpeg.h:75
index
int index
Definition: gxfenc.c:90
c
Undefined Behavior In the C some operations are like signed integer dereferencing freed accessing outside allocated Undefined Behavior must not occur in a C it is not safe even if the output of undefined operations is unused The unsafety may seem nit picking but Optimizing compilers have in fact optimized code on the assumption that no undefined Behavior occurs Optimizing code based on wrong assumptions can and has in some cases lead to effects beyond the output of computations The signed integer overflow problem in speed critical code Code which is highly optimized and works with signed integers sometimes has the problem that often the output of the computation does not c
Definition: undefined.txt:32
mjpeg_decode_app
static int mjpeg_decode_app(MJpegDecodeContext *s, int start_code)
Definition: mjpegdec.c:1846
AV_CODEC_ID_MJPEGB
@ AV_CODEC_ID_MJPEGB
Definition: codec_id.h:60
ff_dlog
#define ff_dlog(a,...)
Definition: tableprint_vlc.h:28
PutByteContext
Definition: bytestream.h:37
AVCodecContext::lowres
int lowres
low resolution decoding, 1-> 1/2 size, 2->1/4 size
Definition: avcodec.h:1705
options
const OptionDef options[]
copy_mb
static void copy_mb(CinepakEncContext *s, uint8_t *a_data[4], int a_linesize[4], uint8_t *b_data[4], int b_linesize[4])
Definition: cinepakenc.c:506
ff_get_buffer
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: decode.c:1729
init
int(* init)(AVBSFContext *ctx)
Definition: dts2pts.c:550
AV_PIX_FMT_RGB24
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
Definition: pixfmt.h:75
AV_CODEC_CAP_DR1
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
Definition: codec.h:52
ff_mjpeg_val_ac_luminance
const uint8_t ff_mjpeg_val_ac_luminance[]
Definition: jpegtabs.h:42
AVPacket::size
int size
Definition: packet.h:589
dc
Tag MUST be and< 10hcoeff half pel interpolation filter coefficients, hcoeff[0] are the 2 middle coefficients[1] are the next outer ones and so on, resulting in a filter like:...eff[2], hcoeff[1], hcoeff[0], hcoeff[0], hcoeff[1], hcoeff[2] ... the sign of the coefficients is not explicitly stored but alternates after each coeff and coeff[0] is positive, so ...,+,-,+,-,+,+,-,+,-,+,... hcoeff[0] is not explicitly stored but found by subtracting the sum of all stored coefficients with signs from 32 hcoeff[0]=32 - hcoeff[1] - hcoeff[2] - ... a good choice for hcoeff and htaps is htaps=6 hcoeff={40,-10, 2} an alternative which requires more computations at both encoder and decoder side and may or may not be better is htaps=8 hcoeff={42,-14, 6,-2}ref_frames minimum of the number of available reference frames and max_ref_frames for example the first frame after a key frame always has ref_frames=1spatial_decomposition_type wavelet type 0 is a 9/7 symmetric compact integer wavelet 1 is a 5/3 symmetric compact integer wavelet others are reserved stored as delta from last, last is reset to 0 if always_reset||keyframeqlog quality(logarithmic quantizer scale) stored as delta from last, last is reset to 0 if always_reset||keyframemv_scale stored as delta from last, last is reset to 0 if always_reset||keyframe FIXME check that everything works fine if this changes between framesqbias dequantization bias stored as delta from last, last is reset to 0 if always_reset||keyframeblock_max_depth maximum depth of the block tree stored as delta from last, last is reset to 0 if always_reset||keyframequant_table quantization tableHighlevel bitstream structure:==============================--------------------------------------------|Header|--------------------------------------------|------------------------------------|||Block0||||split?||||yes no||||......... intra?||||:Block01 :yes no||||:Block02 :....... ..........||||:Block03 ::y DC ::ref index:||||:Block04 ::cb DC ::motion x :||||......... :cr DC ::motion y :||||....... ..........|||------------------------------------||------------------------------------|||Block1|||...|--------------------------------------------|------------ ------------ ------------|||Y subbands||Cb subbands||Cr subbands||||--- ---||--- ---||--- ---|||||LL0||HL0||||LL0||HL0||||LL0||HL0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||LH0||HH0||||LH0||HH0||||LH0||HH0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HL1||LH1||||HL1||LH1||||HL1||LH1|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HH1||HL2||||HH1||HL2||||HH1||HL2|||||...||...||...|||------------ ------------ ------------|--------------------------------------------Decoding process:=================------------|||Subbands|------------||||------------|Intra DC||||LL0 subband prediction ------------|\ Dequantization ------------------- \||Reference frames|\ IDWT|------- -------|Motion \|||Frame 0||Frame 1||Compensation . OBMC v -------|------- -------|--------------. \------> Frame n output Frame Frame<----------------------------------/|...|------------------- Range Coder:============Binary Range Coder:------------------- The implemented range coder is an adapted version based upon "Range encoding: an algorithm for removing redundancy from a digitised message." by G. N. N. Martin. The symbols encoded by the Snow range coder are bits(0|1). The associated probabilities are not fix but change depending on the symbol mix seen so far. bit seen|new state ---------+----------------------------------------------- 0|256 - state_transition_table[256 - old_state];1|state_transition_table[old_state];state_transition_table={ 0, 0, 0, 0, 0, 0, 0, 0, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 190, 191, 192, 194, 194, 195, 196, 197, 198, 199, 200, 201, 202, 202, 204, 205, 206, 207, 208, 209, 209, 210, 211, 212, 213, 215, 215, 216, 217, 218, 219, 220, 220, 222, 223, 224, 225, 226, 227, 227, 229, 229, 230, 231, 232, 234, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 248, 0, 0, 0, 0, 0, 0, 0};FIXME Range Coding of integers:------------------------- FIXME Neighboring Blocks:===================left and top are set to the respective blocks unless they are outside of the image in which case they are set to the Null block top-left is set to the top left block unless it is outside of the image in which case it is set to the left block if this block has no larger parent block or it is at the left side of its parent block and the top right block is not outside of the image then the top right block is used for top-right else the top-left block is used Null block y, cb, cr are 128 level, ref, mx and my are 0 Motion Vector Prediction:=========================1. the motion vectors of all the neighboring blocks are scaled to compensate for the difference of reference frames scaled_mv=(mv *(256 *(current_reference+1)/(mv.reference+1))+128)> the median of the scaled top and top right vectors is used as motion vector prediction the used motion vector is the sum of the predictor and(mvx_diff, mvy_diff) *mv_scale Intra DC Prediction block[y][x] dc[1]
Definition: snow.txt:400
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:94
height
#define height
Definition: dsp.h:89
av_frame_ref
int av_frame_ref(AVFrame *dst, const AVFrame *src)
Set up a new reference to the data described by the source frame.
Definition: frame.c:278
codec_internal.h
SOF14
@ SOF14
Definition: mjpeg.h:53
ff_jpegls_decode_lse
int ff_jpegls_decode_lse(MJpegDecodeContext *s)
Decode LSE block with initialization parameters.
Definition: jpeglsdec.c:51
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition: dsp.h:87
ff_mjpeg_decode_frame
int ff_mjpeg_decode_frame(AVCodecContext *avctx, AVFrame *frame, int *got_frame, AVPacket *avpkt)
Definition: mjpegdec.c:2894
av_bswap32
#define av_bswap32
Definition: bswap.h:47
av_err2str
#define av_err2str(errnum)
Convenience macro, the return value should be used only directly in function arguments but never stan...
Definition: error.h:122
for
for(k=2;k<=8;++k)
Definition: h264pred_template.c:424
AV_PROFILE_MJPEG_JPEG_LS
#define AV_PROFILE_MJPEG_JPEG_LS
Definition: defs.h:177
ff_mjpeg_bits_ac_luminance
const uint8_t ff_mjpeg_bits_ac_luminance[]
Definition: jpegtabs.h:40
FF_CODEC_CAP_EXPORTS_CROPPING
#define FF_CODEC_CAP_EXPORTS_CROPPING
The decoder sets the cropping fields in the output frames manually.
Definition: codec_internal.h:60
size
int size
Definition: twinvq_data.h:10344
AV_CODEC_ID_SMVJPEG
@ AV_CODEC_ID_SMVJPEG
Definition: codec_id.h:268
AV_NOPTS_VALUE
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:247
ff_frame_new_side_data
int ff_frame_new_side_data(const AVCodecContext *avctx, AVFrame *frame, enum AVFrameSideDataType type, size_t size, AVFrameSideData **psd)
Wrapper around av_frame_new_side_data, which rejects side data overridden by the demuxer.
Definition: decode.c:2127
AV_RB32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_RB32
Definition: bytestream.h:96
FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM
#define FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM
The decoder extracts and fills its parameters even if the frame is skipped due to the skip_frame sett...
Definition: codec_internal.h:54
avpriv_report_missing_feature
void avpriv_report_missing_feature(void *avc, const char *msg,...) av_printf_format(2
Log a generic warning message about a missing feature.
AVFrameSideData::data
uint8_t * data
Definition: frame.h:284
SOF15
@ SOF15
Definition: mjpeg.h:54
AVCodecHWConfigInternal
Definition: hwconfig.h:25
OPEN_READER
#define OPEN_READER(name, gb)
Definition: get_bits.h:177
AVPacket::dts
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed.
Definition: packet.h:587
AV_PIX_FMT_YUVA444P
@ AV_PIX_FMT_YUVA444P
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
Definition: pixfmt.h:174
offset
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf offset
Definition: writing_filters.txt:86
line
Definition: graph2dot.c:48
attributes.h
get_xbits
static int get_xbits(GetBitContext *s, int n)
Read MPEG-1 dc-style VLC (sign bit + mantissa with no MSB).
Definition: get_bits.h:294
HWACCEL_NVDEC
#define HWACCEL_NVDEC(codec)
Definition: hwconfig.h:68
predictor
static void predictor(uint8_t *src, ptrdiff_t size)
Definition: exrenc.c:170
AV_STEREO3D_FLAG_INVERT
#define AV_STEREO3D_FLAG_INVERT
Inverted views, Right/Bottom represents the left view.
Definition: stereo3d.h:194
AV_PIX_FMT_VAAPI
@ AV_PIX_FMT_VAAPI
Hardware acceleration through VA-API, data[3] contains a VASurfaceID.
Definition: pixfmt.h:126
DQT
@ DQT
Definition: mjpeg.h:73
AV_LOG_INFO
#define AV_LOG_INFO
Standard information.
Definition: log.h:221
ff_thp_decoder
const FFCodec ff_thp_decoder
AVCodec::id
enum AVCodecID id
Definition: codec.h:186
layout
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel layout
Definition: filter_design.txt:18
SOF10
@ SOF10
Definition: mjpeg.h:49
AV_CODEC_ID_MJPEG
@ AV_CODEC_ID_MJPEG
Definition: codec_id.h:59
NEG_USR32
#define NEG_USR32(a, s)
Definition: mathops.h:180
copy_block4
static void copy_block4(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride, int h)
Definition: copy_block.h:37
interlaced
uint8_t interlaced
Definition: mxfenc.c:2334
decode_block
static int decode_block(MJpegDecodeContext *s, int16_t *block, int component, int dc_index, int ac_index, uint16_t *quant_matrix)
Definition: mjpegdec.c:851
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
code
and forward the test the status of outputs and forward it to the corresponding return FFERROR_NOT_READY If the filters stores internally one or a few frame for some it can consider them to be part of the FIFO and delay acknowledging a status change accordingly Example code
Definition: filter_design.txt:178
EOI
@ EOI
Definition: mjpeg.h:71
copy_block.h
AVCodecContext::extradata
uint8_t * extradata
Out-of-band global headers that may be used by some codecs.
Definition: avcodec.h:522
AV_PROFILE_MJPEG_HUFFMAN_LOSSLESS
#define AV_PROFILE_MJPEG_HUFFMAN_LOSSLESS
Definition: defs.h:176
VD
#define VD
Definition: amfdec.c:664
src2
const pixel * src2
Definition: h264pred_template.c:421
ff_jpegls_decode_picture
int ff_jpegls_decode_picture(MJpegDecodeContext *s)
Definition: jpeglsdec.c:355
AV_FIELD_BB
@ AV_FIELD_BB
Bottom coded first, bottom displayed first.
Definition: defs.h:215
mjpeg_decode_scan
static int mjpeg_decode_scan(MJpegDecodeContext *s)
Definition: mjpegdec.c:1443
AV_STEREO3D_TOPBOTTOM
@ AV_STEREO3D_TOPBOTTOM
Views are on top of each other.
Definition: stereo3d.h:76
mjpeg_decode_dri
static int mjpeg_decode_dri(MJpegDecodeContext *s)
Definition: mjpegdec.c:1835
AVCodecInternal::in_pkt
AVPacket * in_pkt
This packet is used to hold the packet given to decoders implementing the .decode API; it is unused b...
Definition: internal.h:83
av_fast_padded_malloc
void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
Same behaviour av_fast_malloc but the buffer has additional AV_INPUT_BUFFER_PADDING_SIZE at the end w...
Definition: utils.c:53
ff_mjpeg_unescape_sos
int ff_mjpeg_unescape_sos(MJpegDecodeContext *s)
Definition: mjpegdec.c:2246
SOF9
@ SOF9
Definition: mjpeg.h:48
av_always_inline
#define av_always_inline
Definition: attributes.h:63
decode_flush
static av_cold void decode_flush(AVCodecContext *avctx)
Definition: mjpegdec.c:2941
FF_DEBUG_STARTCODE
#define FF_DEBUG_STARTCODE
Definition: avcodec.h:1390
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
AV_PIX_FMT_YUVJ440P
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
Definition: pixfmt.h:107
av_frame_unref
void av_frame_unref(AVFrame *frame)
Unreference all the buffers referenced by frame and reset the frame fields.
Definition: frame.c:496
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:179
AVCodecContext::chroma_sample_location
enum AVChromaLocation chroma_sample_location
This defines the location of chroma samples.
Definition: avcodec.h:684
len
int len
Definition: vorbis_enc_data.h:426
exif.h
DHT
@ DHT
Definition: mjpeg.h:56
AVCodecContext::height
int height
Definition: avcodec.h:600
AVCodecContext::pix_fmt
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:639
AV_FRAME_FLAG_INTERLACED
#define AV_FRAME_FLAG_INTERLACED
A flag to mark frames whose content is interlaced.
Definition: frame.h:650
AVCOL_RANGE_MPEG
@ AVCOL_RANGE_MPEG
Narrow or limited range content.
Definition: pixfmt.h:760
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:264
FF_CODEC_CAP_ICC_PROFILES
#define FF_CODEC_CAP_ICC_PROFILES
Codec supports embedded ICC profiles (AV_FRAME_DATA_ICC_PROFILE).
Definition: codec_internal.h:81
idctdsp.h
avcodec.h
ff_zigzag_direct
const uint8_t ff_zigzag_direct[64]
Definition: mathtables.c:137
AV_PIX_FMT_PAL8
@ AV_PIX_FMT_PAL8
8 bits with AV_PIX_FMT_RGB32 palette
Definition: pixfmt.h:84
AVCodecContext::frame_num
int64_t frame_num
Frame counter, set by libavcodec.
Definition: avcodec.h:1886
REFINE_BIT
#define REFINE_BIT(j)
Definition: mjpegdec.c:989
ff_vlc_free
void ff_vlc_free(VLC *vlc)
Definition: vlc.c:580
decode_block_refinement
static int decode_block_refinement(MJpegDecodeContext *s, int16_t *block, uint8_t *last_nnz, int ac_index, uint16_t *quant_matrix, int Ss, int Se, int Al, int *EOBRUN)
Definition: mjpegdec.c:1015
ret
ret
Definition: filter_design.txt:187
AV_LOG_FATAL
#define AV_LOG_FATAL
Something went wrong and recovery is not possible.
Definition: log.h:204
pred
static const float pred[4]
Definition: siprdata.h:259
av_stereo3d_alloc
AVStereo3D * av_stereo3d_alloc(void)
Allocate an AVStereo3D structure and set its fields to default values.
Definition: stereo3d.c:35
FFSWAP
#define FFSWAP(type, a, b)
Definition: macros.h:52
AVClass::class_name
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:81
frame
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
Definition: filter_design.txt:265
av_malloc
void * av_malloc(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:98
AVStereo3D::type
enum AVStereo3DType type
How views are packed within the video.
Definition: stereo3d.h:207
SOF2
@ SOF2
Definition: mjpeg.h:41
align_get_bits
static const uint8_t * align_get_bits(GetBitContext *s)
Definition: get_bits.h:560
hwaccel
static const char * hwaccel
Definition: ffplay.c:356
LSE
@ LSE
JPEG-LS extension parameters.
Definition: mjpeg.h:104
FF_DEBUG_QP
#define FF_DEBUG_QP
Definition: avcodec.h:1387
AV_INPUT_BUFFER_PADDING_SIZE
#define AV_INPUT_BUFFER_PADDING_SIZE
Definition: defs.h:40
id
enum AVCodecID id
Definition: dts2pts.c:549
left
Tag MUST be and< 10hcoeff half pel interpolation filter coefficients, hcoeff[0] are the 2 middle coefficients[1] are the next outer ones and so on, resulting in a filter like:...eff[2], hcoeff[1], hcoeff[0], hcoeff[0], hcoeff[1], hcoeff[2] ... the sign of the coefficients is not explicitly stored but alternates after each coeff and coeff[0] is positive, so ...,+,-,+,-,+,+,-,+,-,+,... hcoeff[0] is not explicitly stored but found by subtracting the sum of all stored coefficients with signs from 32 hcoeff[0]=32 - hcoeff[1] - hcoeff[2] - ... a good choice for hcoeff and htaps is htaps=6 hcoeff={40,-10, 2} an alternative which requires more computations at both encoder and decoder side and may or may not be better is htaps=8 hcoeff={42,-14, 6,-2}ref_frames minimum of the number of available reference frames and max_ref_frames for example the first frame after a key frame always has ref_frames=1spatial_decomposition_type wavelet type 0 is a 9/7 symmetric compact integer wavelet 1 is a 5/3 symmetric compact integer wavelet others are reserved stored as delta from last, last is reset to 0 if always_reset||keyframeqlog quality(logarithmic quantizer scale) stored as delta from last, last is reset to 0 if always_reset||keyframemv_scale stored as delta from last, last is reset to 0 if always_reset||keyframe FIXME check that everything works fine if this changes between framesqbias dequantization bias stored as delta from last, last is reset to 0 if always_reset||keyframeblock_max_depth maximum depth of the block tree stored as delta from last, last is reset to 0 if always_reset||keyframequant_table quantization tableHighlevel bitstream structure:==============================--------------------------------------------|Header|--------------------------------------------|------------------------------------|||Block0||||split?||||yes no||||......... intra?||||:Block01 :yes no||||:Block02 :....... ..........||||:Block03 ::y DC ::ref index:||||:Block04 ::cb DC ::motion x :||||......... :cr DC ::motion y :||||....... ..........|||------------------------------------||------------------------------------|||Block1|||...|--------------------------------------------|------------ ------------ ------------|||Y subbands||Cb subbands||Cr subbands||||--- ---||--- ---||--- ---|||||LL0||HL0||||LL0||HL0||||LL0||HL0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||LH0||HH0||||LH0||HH0||||LH0||HH0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HL1||LH1||||HL1||LH1||||HL1||LH1|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HH1||HL2||||HH1||HL2||||HH1||HL2|||||...||...||...|||------------ ------------ ------------|--------------------------------------------Decoding process:=================------------|||Subbands|------------||||------------|Intra DC||||LL0 subband prediction ------------|\ Dequantization ------------------- \||Reference frames|\ IDWT|------- -------|Motion \|||Frame 0||Frame 1||Compensation . OBMC v -------|------- -------|--------------. \------> Frame n output Frame Frame<----------------------------------/|...|------------------- Range Coder:============Binary Range Coder:------------------- The implemented range coder is an adapted version based upon "Range encoding: an algorithm for removing redundancy from a digitised message." by G. N. N. Martin. The symbols encoded by the Snow range coder are bits(0|1). The associated probabilities are not fix but change depending on the symbol mix seen so far. bit seen|new state ---------+----------------------------------------------- 0|256 - state_transition_table[256 - old_state];1|state_transition_table[old_state];state_transition_table={ 0, 0, 0, 0, 0, 0, 0, 0, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 190, 191, 192, 194, 194, 195, 196, 197, 198, 199, 200, 201, 202, 202, 204, 205, 206, 207, 208, 209, 209, 210, 211, 212, 213, 215, 215, 216, 217, 218, 219, 220, 220, 222, 223, 224, 225, 226, 227, 227, 229, 229, 230, 231, 232, 234, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 248, 0, 0, 0, 0, 0, 0, 0};FIXME Range Coding of integers:------------------------- FIXME Neighboring Blocks:===================left and top are set to the respective blocks unless they are outside of the image in which case they are set to the Null block top-left is set to the top left block unless it is outside of the image in which case it is set to the left block if this block has no larger parent block or it is at the left side of its parent block and the top right block is not outside of the image then the top right block is used for top-right else the top-left block is used Null block y, cb, cr are 128 level, ref, mx and my are 0 Motion Vector Prediction:=========================1. the motion vectors of all the neighboring blocks are scaled to compensate for the difference of reference frames scaled_mv=(mv *(256 *(current_reference+1)/(mv.reference+1))+128)> the median of the scaled left
Definition: snow.txt:386
AV_RL32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:92
AV_CODEC_ID_AMV
@ AV_CODEC_ID_AMV
Definition: codec_id.h:159
OFFSET
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default minimum maximum flags name is the option keep it simple and lowercase description are in without and describe what they for example set the foo of the bar offset is the offset of the field in your see the OFFSET() macro
AVCodecContext
main external API structure.
Definition: avcodec.h:439
FF_CODEC_RECEIVE_FRAME_CB
#define FF_CODEC_RECEIVE_FRAME_CB(func)
Definition: codec_internal.h:355
SHOW_UBITS
#define SHOW_UBITS(name, gb, num)
Definition: get_bits.h:247
buffer
the frame and frame reference mechanism is intended to as much as expensive copies of that data while still allowing the filters to produce correct results The data is stored in buffers represented by AVFrame structures Several references can point to the same frame buffer
Definition: filter_design.txt:49
AVCHROMA_LOC_CENTER
@ AVCHROMA_LOC_CENTER
MPEG-1 4:2:0, JPEG 4:2:0, H.263 4:2:0.
Definition: pixfmt.h:799
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:72
APP2
@ APP2
Definition: mjpeg.h:81
FF_HW_CALL
#define FF_HW_CALL(avctx, function,...)
Definition: hwaccel_internal.h:173
AVCodecContext::profile
int profile
profile
Definition: avcodec.h:1626
ffhwaccel
static const FFHWAccel * ffhwaccel(const AVHWAccel *codec)
Definition: hwaccel_internal.h:168
values
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return values
Definition: filter_design.txt:264
AVPixFmtDescriptor::comp
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
Definition: pixdesc.h:105
Windows::Graphics::DirectX::Direct3D11::p
IDirect3DDxgiInterfaceAccess _COM_Outptr_ void ** p
Definition: vsrc_gfxcapture_winrt.hpp:53
AV_PIX_FMT_YUV444P
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:78
ff_mjpeg_bits_dc_chrominance
const uint8_t ff_mjpeg_bits_dc_chrominance[]
Definition: jpegtabs.h:37
AVCodecContext::debug
int debug
debug
Definition: avcodec.h:1382
ff_mjpeg_decode_sof
int ff_mjpeg_decode_sof(MJpegDecodeContext *s)
Definition: mjpegdec.c:307
APP0
@ APP0
Definition: mjpeg.h:79
FF_DISABLE_DEPRECATION_WARNINGS
#define FF_DISABLE_DEPRECATION_WARNINGS
Definition: internal.h:72
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:165
AVCodecContext::coded_width
int coded_width
Bitstream width / height, may be different from width/height e.g.
Definition: avcodec.h:615
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:200
AV_PIX_FMT_GRAY16LE
@ AV_PIX_FMT_GRAY16LE
Y , 16bpp, little-endian.
Definition: pixfmt.h:105
AV_PIX_FMT_YUV422P
@ AV_PIX_FMT_YUV422P
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:77
mem.h
bytestream2_get_bufferu
static av_always_inline unsigned int bytestream2_get_bufferu(GetByteContext *g, uint8_t *dst, unsigned int size)
Definition: bytestream.h:277
SOI
@ SOI
Definition: mjpeg.h:70
av_stereo3d_create_side_data
AVStereo3D * av_stereo3d_create_side_data(AVFrame *frame)
Allocate a complete AVFrameSideData and add it to the frame.
Definition: stereo3d.c:54
avpriv_request_sample
#define avpriv_request_sample(...)
Definition: tableprint_vlc.h:37
AVFrameSideData
Structure to hold side data for an AVFrame.
Definition: frame.h:282
flush_put_bits
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:153
SOF1
@ SOF1
Definition: mjpeg.h:40
w
uint8_t w
Definition: llvidencdsp.c:39
av_free
#define av_free(p)
Definition: tableprint_vlc.h:34
AVCodecContext::codec_tag
unsigned int codec_tag
fourcc (LSB first, so "ABCD" -> ('D'<<24) + ('C'<<16) + ('B'<<8) + 'A').
Definition: avcodec.h:464
bytestream2_put_bufferu
static av_always_inline unsigned int bytestream2_put_bufferu(PutByteContext *p, const uint8_t *src, unsigned int size)
Definition: bytestream.h:301
ff_mjpeg_bits_dc_luminance
const FF_VISIBILITY_PUSH_HIDDEN uint8_t ff_mjpeg_bits_dc_luminance[]
Definition: jpegtabs.h:32
ff_mjpeg_build_vlc
int ff_mjpeg_build_vlc(VLC *vlc, const uint8_t *bits_table, const uint8_t *val_table, int is_ac, void *logctx)
Definition: mjpegdec_common.c:41
AVPacket
This structure stores compressed data.
Definition: packet.h:565
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:466
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Underlying C type is int.
Definition: opt.h:327
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:35
av_fast_malloc
void av_fast_malloc(void *ptr, unsigned int *size, size_t min_size)
Allocate a buffer, reusing the given one if large enough.
Definition: mem.c:557
AV_PIX_FMT_YUV411P
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:80
HWACCEL_VAAPI
#define HWACCEL_VAAPI(codec)
Definition: hwconfig.h:70
FFMAX3
#define FFMAX3(a, b, c)
Definition: macros.h:48
imgutils.h
bytestream2_init
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
Definition: bytestream.h:137
AVERROR_BUG
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
Definition: error.h:52
AVCodecContext::properties
attribute_deprecated unsigned properties
Properties of the stream that gets decoded.
Definition: avcodec.h:1645
MAX_COMPONENTS
#define MAX_COMPONENTS
Definition: mjpegdec.h:47
rgb
static const SheerTable rgb[2]
Definition: sheervideodata.h:32
block
The exact code depends on how similar the blocks are and how related they are to the block
Definition: filter_design.txt:207
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
ljpeg_decode_yuv_scan
static int ljpeg_decode_yuv_scan(MJpegDecodeContext *s)
Definition: mjpegdec.c:1251
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
MKTAG
#define MKTAG(a, b, c, d)
Definition: macros.h:55
h
h
Definition: vp9dsp_template.c:2070
SOF7
@ SOF7
Definition: mjpeg.h:46
AVStereo3D
Stereo 3D type: this structure describes how two videos are packed within a single video surface,...
Definition: stereo3d.h:203
pkt
static AVPacket * pkt
Definition: demux_decode.c:55
av_image_check_size
int av_image_check_size(unsigned int w, unsigned int h, int log_offset, void *log_ctx)
Check if the given dimension of an image is valid, meaning that all bytes of the image can be address...
Definition: imgutils.c:318
width
#define width
Definition: dsp.h:89
AV_PROFILE_MJPEG_HUFFMAN_PROGRESSIVE_DCT
#define AV_PROFILE_MJPEG_HUFFMAN_PROGRESSIVE_DCT
Definition: defs.h:175
AV_RB24
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_RB24
Definition: bytestream.h:97
PREDICT
#define PREDICT(ret, topleft, top, left, predictor)
Definition: mjpeg.h:118
put_bits.h
return_frame
static int return_frame(AVFilterContext *ctx, int is_second)
Definition: yadif_common.c:28
AV_FRAME_FLAG_LOSSLESS
#define AV_FRAME_FLAG_LOSSLESS
A decoder can use this flag to mark frames which were originally encoded losslessly.
Definition: frame.h:663
SOF6
@ SOF6
Definition: mjpeg.h:45
skip
static void BS_FUNC() skip(BSCTX *bc, unsigned int n)
Skip n bits in the buffer.
Definition: bitstream_template.h:383
src
#define src
Definition: vp8dsp.c:248
JPG
@ JPG
Definition: mjpeg.h:47
av_fourcc2str
#define av_fourcc2str(fourcc)
Definition: avutil.h:347
Generated on Sun Feb 22 2026 19:22:15 for FFmpeg by   doxygen 1.8.17