FFmpeg  4.1.5
huffyuvdec.c
Go to the documentation of this file.
1 /*
2  * huffyuv decoder
3  *
4  * Copyright (c) 2002-2014 Michael Niedermayer <michaelni@gmx.at>
5  *
6  * see http://www.pcisys.net/~melanson/codecs/huffyuv.txt for a description of
7  * the algorithm used
8  *
9  * This file is part of FFmpeg.
10  *
11  * FFmpeg is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU Lesser General Public
13  * License as published by the Free Software Foundation; either
14  * version 2.1 of the License, or (at your option) any later version.
15  *
16  * FFmpeg is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19  * Lesser General Public License for more details.
20  *
21  * You should have received a copy of the GNU Lesser General Public
22  * License along with FFmpeg; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24  *
25  * yuva, gray, 4:4:4, 4:1:1, 4:1:0 and >8 bit per sample support sponsored by NOA
26  */
27 
28 /**
29  * @file
30  * huffyuv decoder
31  */
32 
33 #define UNCHECKED_BITSTREAM_READER 1
34 
35 #include "avcodec.h"
36 #include "get_bits.h"
37 #include "huffyuv.h"
38 #include "huffyuvdsp.h"
39 #include "lossless_videodsp.h"
40 #include "thread.h"
41 #include "libavutil/imgutils.h"
42 #include "libavutil/pixdesc.h"
43 
44 #define classic_shift_luma_table_size 42
46  34, 36, 35, 69, 135, 232, 9, 16, 10, 24, 11, 23, 12, 16, 13, 10,
47  14, 8, 15, 8, 16, 8, 17, 20, 16, 10, 207, 206, 205, 236, 11, 8,
48  10, 21, 9, 23, 8, 8, 199, 70, 69, 68, 0,
49  0,0,0,0,0,0,0,0,
50 };
51 
52 #define classic_shift_chroma_table_size 59
54  66, 36, 37, 38, 39, 40, 41, 75, 76, 77, 110, 239, 144, 81, 82, 83,
55  84, 85, 118, 183, 56, 57, 88, 89, 56, 89, 154, 57, 58, 57, 26, 141,
56  57, 56, 58, 57, 58, 57, 184, 119, 214, 245, 116, 83, 82, 49, 80, 79,
57  78, 77, 44, 75, 41, 40, 39, 38, 37, 36, 34, 0,
58  0,0,0,0,0,0,0,0,
59 };
60 
61 static const unsigned char classic_add_luma[256] = {
62  3, 9, 5, 12, 10, 35, 32, 29, 27, 50, 48, 45, 44, 41, 39, 37,
63  73, 70, 68, 65, 64, 61, 58, 56, 53, 50, 49, 46, 44, 41, 38, 36,
64  68, 65, 63, 61, 58, 55, 53, 51, 48, 46, 45, 43, 41, 39, 38, 36,
65  35, 33, 32, 30, 29, 27, 26, 25, 48, 47, 46, 44, 43, 41, 40, 39,
66  37, 36, 35, 34, 32, 31, 30, 28, 27, 26, 24, 23, 22, 20, 19, 37,
67  35, 34, 33, 31, 30, 29, 27, 26, 24, 23, 21, 20, 18, 17, 15, 29,
68  27, 26, 24, 22, 21, 19, 17, 16, 14, 26, 25, 23, 21, 19, 18, 16,
69  15, 27, 25, 23, 21, 19, 17, 16, 14, 26, 25, 23, 21, 18, 17, 14,
70  12, 17, 19, 13, 4, 9, 2, 11, 1, 7, 8, 0, 16, 3, 14, 6,
71  12, 10, 5, 15, 18, 11, 10, 13, 15, 16, 19, 20, 22, 24, 27, 15,
72  18, 20, 22, 24, 26, 14, 17, 20, 22, 24, 27, 15, 18, 20, 23, 25,
73  28, 16, 19, 22, 25, 28, 32, 36, 21, 25, 29, 33, 38, 42, 45, 49,
74  28, 31, 34, 37, 40, 42, 44, 47, 49, 50, 52, 54, 56, 57, 59, 60,
75  62, 64, 66, 67, 69, 35, 37, 39, 40, 42, 43, 45, 47, 48, 51, 52,
76  54, 55, 57, 59, 60, 62, 63, 66, 67, 69, 71, 72, 38, 40, 42, 43,
77  46, 47, 49, 51, 26, 28, 30, 31, 33, 34, 18, 19, 11, 13, 7, 8,
78 };
79 
80 static const unsigned char classic_add_chroma[256] = {
81  3, 1, 2, 2, 2, 2, 3, 3, 7, 5, 7, 5, 8, 6, 11, 9,
82  7, 13, 11, 10, 9, 8, 7, 5, 9, 7, 6, 4, 7, 5, 8, 7,
83  11, 8, 13, 11, 19, 15, 22, 23, 20, 33, 32, 28, 27, 29, 51, 77,
84  43, 45, 76, 81, 46, 82, 75, 55, 56, 144, 58, 80, 60, 74, 147, 63,
85  143, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79,
86  80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 27, 30, 21, 22,
87  17, 14, 5, 6, 100, 54, 47, 50, 51, 53, 106, 107, 108, 109, 110, 111,
88  112, 113, 114, 115, 4, 117, 118, 92, 94, 121, 122, 3, 124, 103, 2, 1,
89  0, 129, 130, 131, 120, 119, 126, 125, 136, 137, 138, 139, 140, 141, 142, 134,
90  135, 132, 133, 104, 64, 101, 62, 57, 102, 95, 93, 59, 61, 28, 97, 96,
91  52, 49, 48, 29, 32, 25, 24, 46, 23, 98, 45, 44, 43, 20, 42, 41,
92  19, 18, 99, 40, 15, 39, 38, 16, 13, 12, 11, 37, 10, 9, 8, 36,
93  7, 128, 127, 105, 123, 116, 35, 34, 33, 145, 31, 79, 42, 146, 78, 26,
94  83, 48, 49, 50, 44, 47, 26, 31, 30, 18, 17, 19, 21, 24, 25, 13,
95  14, 16, 17, 18, 20, 21, 12, 14, 15, 9, 10, 6, 9, 6, 5, 8,
96  6, 12, 8, 10, 7, 9, 6, 4, 6, 2, 2, 3, 3, 3, 3, 2,
97 };
98 
99 static int read_len_table(uint8_t *dst, GetBitContext *gb, int n)
100 {
101  int i, val, repeat;
102 
103  for (i = 0; i < n;) {
104  repeat = get_bits(gb, 3);
105  val = get_bits(gb, 5);
106  if (repeat == 0)
107  repeat = get_bits(gb, 8);
108  if (i + repeat > n || get_bits_left(gb) < 0) {
109  av_log(NULL, AV_LOG_ERROR, "Error reading huffman table\n");
110  return AVERROR_INVALIDDATA;
111  }
112  while (repeat--)
113  dst[i++] = val;
114  }
115  return 0;
116 }
117 
119 {
120  int ret;
121  uint16_t *symbols = av_mallocz(5 << VLC_BITS);
122  uint16_t *bits;
123  uint8_t *len;
124  if (!symbols)
125  return AVERROR(ENOMEM);
126  bits = symbols + (1 << VLC_BITS);
127  len = (uint8_t *)(bits + (1 << VLC_BITS));
128 
129  if (s->bitstream_bpp < 24 || s->version > 2) {
130  int p, i, y, u;
131  for (p = 0; p < 4; p++) {
132  int p0 = s->version > 2 ? p : 0;
133  for (i = y = 0; y < s->vlc_n; y++) {
134  int len0 = s->len[p0][y];
135  int limit = VLC_BITS - len0;
136  if (limit <= 0 || !len0)
137  continue;
138  if ((sign_extend(y, 8) & (s->vlc_n-1)) != y)
139  continue;
140  for (u = 0; u < s->vlc_n; u++) {
141  int len1 = s->len[p][u];
142  if (len1 > limit || !len1)
143  continue;
144  if ((sign_extend(u, 8) & (s->vlc_n-1)) != u)
145  continue;
146  av_assert0(i < (1 << VLC_BITS));
147  len[i] = len0 + len1;
148  bits[i] = (s->bits[p0][y] << len1) + s->bits[p][u];
149  symbols[i] = (y << 8) + (u & 0xFF);
150  i++;
151  }
152  }
153  ff_free_vlc(&s->vlc[4 + p]);
154  if ((ret = ff_init_vlc_sparse(&s->vlc[4 + p], VLC_BITS, i, len, 1, 1,
155  bits, 2, 2, symbols, 2, 2, 0)) < 0)
156  goto out;
157  }
158  } else {
159  uint8_t (*map)[4] = (uint8_t(*)[4]) s->pix_bgr_map;
160  int i, b, g, r, code;
161  int p0 = s->decorrelate;
162  int p1 = !s->decorrelate;
163  /* Restrict the range to +/-16 because that's pretty much guaranteed
164  * to cover all the combinations that fit in 11 bits total, and it
165  * does not matter if we miss a few rare codes. */
166  for (i = 0, g = -16; g < 16; g++) {
167  int len0 = s->len[p0][g & 255];
168  int limit0 = VLC_BITS - len0;
169  if (limit0 < 2 || !len0)
170  continue;
171  for (b = -16; b < 16; b++) {
172  int len1 = s->len[p1][b & 255];
173  int limit1 = limit0 - len1;
174  if (limit1 < 1 || !len1)
175  continue;
176  code = (s->bits[p0][g & 255] << len1) + s->bits[p1][b & 255];
177  for (r = -16; r < 16; r++) {
178  int len2 = s->len[2][r & 255];
179  if (len2 > limit1 || !len2)
180  continue;
181  av_assert0(i < (1 << VLC_BITS));
182  len[i] = len0 + len1 + len2;
183  bits[i] = (code << len2) + s->bits[2][r & 255];
184  if (s->decorrelate) {
185  map[i][G] = g;
186  map[i][B] = g + b;
187  map[i][R] = g + r;
188  } else {
189  map[i][B] = g;
190  map[i][G] = b;
191  map[i][R] = r;
192  }
193  i++;
194  }
195  }
196  }
197  ff_free_vlc(&s->vlc[4]);
198  if ((ret = init_vlc(&s->vlc[4], VLC_BITS, i, len, 1, 1,
199  bits, 2, 2, 0)) < 0)
200  goto out;
201  }
202  ret = 0;
203 out:
204  av_freep(&symbols);
205  return ret;
206 }
207 
209 {
210  GetBitContext gb;
211  int i, ret;
212  int count = 3;
213 
214  if ((ret = init_get_bits(&gb, src, length * 8)) < 0)
215  return ret;
216 
217  if (s->version > 2)
218  count = 1 + s->alpha + 2*s->chroma;
219 
220  for (i = 0; i < count; i++) {
221  if ((ret = read_len_table(s->len[i], &gb, s->vlc_n)) < 0)
222  return ret;
223  if ((ret = ff_huffyuv_generate_bits_table(s->bits[i], s->len[i], s->vlc_n)) < 0)
224  return ret;
225  ff_free_vlc(&s->vlc[i]);
226  if ((ret = init_vlc(&s->vlc[i], VLC_BITS, s->vlc_n, s->len[i], 1, 1,
227  s->bits[i], 4, 4, 0)) < 0)
228  return ret;
229  }
230 
231  if ((ret = generate_joint_tables(s)) < 0)
232  return ret;
233 
234  return (get_bits_count(&gb) + 7) / 8;
235 }
236 
238 {
239  GetBitContext gb;
240  int i, ret;
241 
244  if ((ret = read_len_table(s->len[0], &gb, 256)) < 0)
245  return ret;
246 
249  if ((ret = read_len_table(s->len[1], &gb, 256)) < 0)
250  return ret;
251 
252  for (i = 0; i < 256; i++)
253  s->bits[0][i] = classic_add_luma[i];
254  for (i = 0; i < 256; i++)
255  s->bits[1][i] = classic_add_chroma[i];
256 
257  if (s->bitstream_bpp >= 24) {
258  memcpy(s->bits[1], s->bits[0], 256 * sizeof(uint32_t));
259  memcpy(s->len[1], s->len[0], 256 * sizeof(uint8_t));
260  }
261  memcpy(s->bits[2], s->bits[1], 256 * sizeof(uint32_t));
262  memcpy(s->len[2], s->len[1], 256 * sizeof(uint8_t));
263 
264  for (i = 0; i < 4; i++) {
265  ff_free_vlc(&s->vlc[i]);
266  if ((ret = init_vlc(&s->vlc[i], VLC_BITS, 256, s->len[i], 1, 1,
267  s->bits[i], 4, 4, 0)) < 0)
268  return ret;
269  }
270 
271  if ((ret = generate_joint_tables(s)) < 0)
272  return ret;
273 
274  return 0;
275 }
276 
278 {
279  HYuvContext *s = avctx->priv_data;
280  int i;
281 
284 
285  for (i = 0; i < 8; i++)
286  ff_free_vlc(&s->vlc[i]);
287 
288  return 0;
289 }
290 
292 {
293  HYuvContext *s = avctx->priv_data;
294  int ret;
295 
296  ret = av_image_check_size(avctx->width, avctx->height, 0, avctx);
297  if (ret < 0)
298  return ret;
299 
300  ff_huffyuvdsp_init(&s->hdsp, avctx->pix_fmt);
302  memset(s->vlc, 0, 4 * sizeof(VLC));
303 
304  s->interlaced = avctx->height > 288;
305  s->bgr32 = 1;
306 
307  if (avctx->extradata_size) {
308  if ((avctx->bits_per_coded_sample & 7) &&
309  avctx->bits_per_coded_sample != 12)
310  s->version = 1; // do such files exist at all?
311  else if (avctx->extradata_size > 3 && avctx->extradata[3] == 0)
312  s->version = 2;
313  else
314  s->version = 3;
315  } else
316  s->version = 0;
317 
318  s->bps = 8;
319  s->n = 1<<s->bps;
320  s->vlc_n = FFMIN(s->n, MAX_VLC_N);
321  s->chroma = 1;
322  if (s->version >= 2) {
323  int method, interlace;
324 
325  if (avctx->extradata_size < 4)
326  return AVERROR_INVALIDDATA;
327 
328  method = avctx->extradata[0];
329  s->decorrelate = method & 64 ? 1 : 0;
330  s->predictor = method & 63;
331  if (s->version == 2) {
332  s->bitstream_bpp = avctx->extradata[1];
333  if (s->bitstream_bpp == 0)
334  s->bitstream_bpp = avctx->bits_per_coded_sample & ~7;
335  } else {
336  s->bps = (avctx->extradata[1] >> 4) + 1;
337  s->n = 1<<s->bps;
338  s->vlc_n = FFMIN(s->n, MAX_VLC_N);
339  s->chroma_h_shift = avctx->extradata[1] & 3;
340  s->chroma_v_shift = (avctx->extradata[1] >> 2) & 3;
341  s->yuv = !!(avctx->extradata[2] & 1);
342  s->chroma= !!(avctx->extradata[2] & 3);
343  s->alpha = !!(avctx->extradata[2] & 4);
344  }
345  interlace = (avctx->extradata[2] & 0x30) >> 4;
346  s->interlaced = (interlace == 1) ? 1 : (interlace == 2) ? 0 : s->interlaced;
347  s->context = avctx->extradata[2] & 0x40 ? 1 : 0;
348 
349  if ((ret = read_huffman_tables(s, avctx->extradata + 4,
350  avctx->extradata_size - 4)) < 0)
351  goto error;
352  } else {
353  switch (avctx->bits_per_coded_sample & 7) {
354  case 1:
355  s->predictor = LEFT;
356  s->decorrelate = 0;
357  break;
358  case 2:
359  s->predictor = LEFT;
360  s->decorrelate = 1;
361  break;
362  case 3:
363  s->predictor = PLANE;
364  s->decorrelate = avctx->bits_per_coded_sample >= 24;
365  break;
366  case 4:
367  s->predictor = MEDIAN;
368  s->decorrelate = 0;
369  break;
370  default:
371  s->predictor = LEFT; // OLD
372  s->decorrelate = 0;
373  break;
374  }
375  s->bitstream_bpp = avctx->bits_per_coded_sample & ~7;
376  s->context = 0;
377 
378  if ((ret = read_old_huffman_tables(s)) < 0)
379  goto error;
380  }
381 
382  if (s->version <= 2) {
383  switch (s->bitstream_bpp) {
384  case 12:
385  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
386  s->yuv = 1;
387  break;
388  case 16:
389  if (s->yuy2)
390  avctx->pix_fmt = AV_PIX_FMT_YUYV422;
391  else
392  avctx->pix_fmt = AV_PIX_FMT_YUV422P;
393  s->yuv = 1;
394  break;
395  case 24:
396  if (s->bgr32)
397  avctx->pix_fmt = AV_PIX_FMT_0RGB32;
398  else
399  avctx->pix_fmt = AV_PIX_FMT_BGR24;
400  break;
401  case 32:
402  av_assert0(s->bgr32);
403  avctx->pix_fmt = AV_PIX_FMT_RGB32;
404  s->alpha = 1;
405  break;
406  default:
407  ret = AVERROR_INVALIDDATA;
408  goto error;
409  }
411  &s->chroma_h_shift,
412  &s->chroma_v_shift);
413  } else {
414  switch ( (s->chroma<<10) | (s->yuv<<9) | (s->alpha<<8) | ((s->bps-1)<<4) | s->chroma_h_shift | (s->chroma_v_shift<<2)) {
415  case 0x070:
416  avctx->pix_fmt = AV_PIX_FMT_GRAY8;
417  break;
418  case 0x0F0:
419  avctx->pix_fmt = AV_PIX_FMT_GRAY16;
420  break;
421  case 0x470:
422  avctx->pix_fmt = AV_PIX_FMT_GBRP;
423  break;
424  case 0x480:
425  avctx->pix_fmt = AV_PIX_FMT_GBRP9;
426  break;
427  case 0x490:
428  avctx->pix_fmt = AV_PIX_FMT_GBRP10;
429  break;
430  case 0x4B0:
431  avctx->pix_fmt = AV_PIX_FMT_GBRP12;
432  break;
433  case 0x4D0:
434  avctx->pix_fmt = AV_PIX_FMT_GBRP14;
435  break;
436  case 0x4F0:
437  avctx->pix_fmt = AV_PIX_FMT_GBRP16;
438  break;
439  case 0x570:
440  avctx->pix_fmt = AV_PIX_FMT_GBRAP;
441  break;
442  case 0x670:
443  avctx->pix_fmt = AV_PIX_FMT_YUV444P;
444  break;
445  case 0x680:
446  avctx->pix_fmt = AV_PIX_FMT_YUV444P9;
447  break;
448  case 0x690:
449  avctx->pix_fmt = AV_PIX_FMT_YUV444P10;
450  break;
451  case 0x6B0:
452  avctx->pix_fmt = AV_PIX_FMT_YUV444P12;
453  break;
454  case 0x6D0:
455  avctx->pix_fmt = AV_PIX_FMT_YUV444P14;
456  break;
457  case 0x6F0:
458  avctx->pix_fmt = AV_PIX_FMT_YUV444P16;
459  break;
460  case 0x671:
461  avctx->pix_fmt = AV_PIX_FMT_YUV422P;
462  break;
463  case 0x681:
464  avctx->pix_fmt = AV_PIX_FMT_YUV422P9;
465  break;
466  case 0x691:
467  avctx->pix_fmt = AV_PIX_FMT_YUV422P10;
468  break;
469  case 0x6B1:
470  avctx->pix_fmt = AV_PIX_FMT_YUV422P12;
471  break;
472  case 0x6D1:
473  avctx->pix_fmt = AV_PIX_FMT_YUV422P14;
474  break;
475  case 0x6F1:
476  avctx->pix_fmt = AV_PIX_FMT_YUV422P16;
477  break;
478  case 0x672:
479  avctx->pix_fmt = AV_PIX_FMT_YUV411P;
480  break;
481  case 0x674:
482  avctx->pix_fmt = AV_PIX_FMT_YUV440P;
483  break;
484  case 0x675:
485  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
486  break;
487  case 0x685:
488  avctx->pix_fmt = AV_PIX_FMT_YUV420P9;
489  break;
490  case 0x695:
491  avctx->pix_fmt = AV_PIX_FMT_YUV420P10;
492  break;
493  case 0x6B5:
494  avctx->pix_fmt = AV_PIX_FMT_YUV420P12;
495  break;
496  case 0x6D5:
497  avctx->pix_fmt = AV_PIX_FMT_YUV420P14;
498  break;
499  case 0x6F5:
500  avctx->pix_fmt = AV_PIX_FMT_YUV420P16;
501  break;
502  case 0x67A:
503  avctx->pix_fmt = AV_PIX_FMT_YUV410P;
504  break;
505  case 0x770:
506  avctx->pix_fmt = AV_PIX_FMT_YUVA444P;
507  break;
508  case 0x780:
509  avctx->pix_fmt = AV_PIX_FMT_YUVA444P9;
510  break;
511  case 0x790:
513  break;
514  case 0x7F0:
516  break;
517  case 0x771:
518  avctx->pix_fmt = AV_PIX_FMT_YUVA422P;
519  break;
520  case 0x781:
521  avctx->pix_fmt = AV_PIX_FMT_YUVA422P9;
522  break;
523  case 0x791:
525  break;
526  case 0x7F1:
528  break;
529  case 0x775:
530  avctx->pix_fmt = AV_PIX_FMT_YUVA420P;
531  break;
532  case 0x785:
533  avctx->pix_fmt = AV_PIX_FMT_YUVA420P9;
534  break;
535  case 0x795:
537  break;
538  case 0x7F5:
540  break;
541  default:
542  ret = AVERROR_INVALIDDATA;
543  goto error;
544  }
545  }
546 
547  ff_huffyuv_common_init(avctx);
548 
549  if ((avctx->pix_fmt == AV_PIX_FMT_YUV422P || avctx->pix_fmt == AV_PIX_FMT_YUV420P) && avctx->width & 1) {
550  av_log(avctx, AV_LOG_ERROR, "width must be even for this colorspace\n");
551  ret = AVERROR_INVALIDDATA;
552  goto error;
553  }
554  if (s->predictor == MEDIAN && avctx->pix_fmt == AV_PIX_FMT_YUV422P &&
555  avctx->width % 4) {
556  av_log(avctx, AV_LOG_ERROR, "width must be a multiple of 4 "
557  "for this combination of colorspace and predictor type.\n");
558  ret = AVERROR_INVALIDDATA;
559  goto error;
560  }
561 
562  if ((ret = ff_huffyuv_alloc_temp(s)) < 0) {
564  goto error;
565  }
566 
567  return 0;
568  error:
569  decode_end(avctx);
570  return ret;
571 }
572 
573 #if HAVE_THREADS
574 static av_cold int decode_init_thread_copy(AVCodecContext *avctx)
575 {
576  HYuvContext *s = avctx->priv_data;
577  int i, ret;
578 
579  s->avctx = avctx;
580 
581  if ((ret = ff_huffyuv_alloc_temp(s)) < 0) {
583  return ret;
584  }
585 
586  for (i = 0; i < 8; i++)
587  s->vlc[i].table = NULL;
588 
589  if (s->version >= 2) {
590  if ((ret = read_huffman_tables(s, avctx->extradata + 4,
591  avctx->extradata_size)) < 0)
592  return ret;
593  } else {
594  if ((ret = read_old_huffman_tables(s)) < 0)
595  return ret;
596  }
597 
598  return 0;
599 }
600 #endif
601 
602 /** Subset of GET_VLC for use in hand-roller VLC code */
603 #define VLC_INTERN(dst, table, gb, name, bits, max_depth) \
604  code = table[index][0]; \
605  n = table[index][1]; \
606  if (max_depth > 1 && n < 0) { \
607  LAST_SKIP_BITS(name, gb, bits); \
608  UPDATE_CACHE(name, gb); \
609  \
610  nb_bits = -n; \
611  index = SHOW_UBITS(name, gb, nb_bits) + code; \
612  code = table[index][0]; \
613  n = table[index][1]; \
614  if (max_depth > 2 && n < 0) { \
615  LAST_SKIP_BITS(name, gb, nb_bits); \
616  UPDATE_CACHE(name, gb); \
617  \
618  nb_bits = -n; \
619  index = SHOW_UBITS(name, gb, nb_bits) + code; \
620  code = table[index][0]; \
621  n = table[index][1]; \
622  } \
623  } \
624  dst = code; \
625  LAST_SKIP_BITS(name, gb, n)
626 
627 
628 #define GET_VLC_DUAL(dst0, dst1, name, gb, dtable, table1, table2, \
629  bits, max_depth, OP) \
630  do { \
631  unsigned int index = SHOW_UBITS(name, gb, bits); \
632  int code, n = dtable[index][1]; \
633  \
634  if (n<=0) { \
635  int nb_bits; \
636  VLC_INTERN(dst0, table1, gb, name, bits, max_depth); \
637  \
638  UPDATE_CACHE(re, gb); \
639  index = SHOW_UBITS(name, gb, bits); \
640  VLC_INTERN(dst1, table2, gb, name, bits, max_depth); \
641  } else { \
642  code = dtable[index][0]; \
643  OP(dst0, dst1, code); \
644  LAST_SKIP_BITS(name, gb, n); \
645  } \
646  } while (0)
647 
648 #define OP8bits(dst0, dst1, code) dst0 = code>>8; dst1 = code
649 
650 #define READ_2PIX(dst0, dst1, plane1) \
651  UPDATE_CACHE(re, &s->gb); \
652  GET_VLC_DUAL(dst0, dst1, re, &s->gb, s->vlc[4+plane1].table, \
653  s->vlc[0].table, s->vlc[plane1].table, VLC_BITS, 3, OP8bits)
654 
656 {
657  int i, icount;
658  OPEN_READER(re, &s->gb);
659  count /= 2;
660 
661  icount = get_bits_left(&s->gb) / (32 * 4);
662  if (count >= icount) {
663  for (i = 0; i < icount; i++) {
664  READ_2PIX(s->temp[0][2 * i], s->temp[1][i], 1);
665  READ_2PIX(s->temp[0][2 * i + 1], s->temp[2][i], 2);
666  }
667  for (; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
668  READ_2PIX(s->temp[0][2 * i ], s->temp[1][i], 1);
669  if (BITS_LEFT(re, &s->gb) <= 0) break;
670  READ_2PIX(s->temp[0][2 * i + 1], s->temp[2][i], 2);
671  }
672  for (; i < count; i++)
673  s->temp[0][2 * i ] = s->temp[1][i] =
674  s->temp[0][2 * i + 1] = s->temp[2][i] = 0;
675  } else {
676  for (i = 0; i < count; i++) {
677  READ_2PIX(s->temp[0][2 * i], s->temp[1][i], 1);
678  READ_2PIX(s->temp[0][2 * i + 1], s->temp[2][i], 2);
679  }
680  }
681  CLOSE_READER(re, &s->gb);
682 }
683 
684 #define READ_2PIX_PLANE(dst0, dst1, plane, OP) \
685  UPDATE_CACHE(re, &s->gb); \
686  GET_VLC_DUAL(dst0, dst1, re, &s->gb, s->vlc[4+plane].table, \
687  s->vlc[plane].table, s->vlc[plane].table, VLC_BITS, 3, OP)
688 
689 #define OP14bits(dst0, dst1, code) dst0 = code>>8; dst1 = sign_extend(code, 8)
690 
691 /* TODO instead of restarting the read when the code isn't in the first level
692  * of the joint table, jump into the 2nd level of the individual table. */
693 #define READ_2PIX_PLANE16(dst0, dst1, plane){\
694  dst0 = get_vlc2(&s->gb, s->vlc[plane].table, VLC_BITS, 3)<<2;\
695  dst0 += get_bits(&s->gb, 2);\
696  dst1 = get_vlc2(&s->gb, s->vlc[plane].table, VLC_BITS, 3)<<2;\
697  dst1 += get_bits(&s->gb, 2);\
698 }
700 {
701  int i, count = width/2;
702 
703  if (s->bps <= 8) {
704  OPEN_READER(re, &s->gb);
705  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
706  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
707  READ_2PIX_PLANE(s->temp[0][2 * i], s->temp[0][2 * i + 1], plane, OP8bits);
708  }
709  } else {
710  for(i=0; i<count; i++){
711  READ_2PIX_PLANE(s->temp[0][2 * i], s->temp[0][2 * i + 1], plane, OP8bits);
712  }
713  }
714  if( width&1 && BITS_LEFT(re, &s->gb)>0 ) {
715  unsigned int index;
716  int nb_bits, code, n;
717  UPDATE_CACHE(re, &s->gb);
718  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
719  VLC_INTERN(s->temp[0][width-1], s->vlc[plane].table,
720  &s->gb, re, VLC_BITS, 3);
721  }
722  CLOSE_READER(re, &s->gb);
723  } else if (s->bps <= 14) {
724  OPEN_READER(re, &s->gb);
725  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
726  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
727  READ_2PIX_PLANE(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane, OP14bits);
728  }
729  } else {
730  for(i=0; i<count; i++){
731  READ_2PIX_PLANE(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane, OP14bits);
732  }
733  }
734  if( width&1 && BITS_LEFT(re, &s->gb)>0 ) {
735  unsigned int index;
736  int nb_bits, code, n;
737  UPDATE_CACHE(re, &s->gb);
738  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
739  VLC_INTERN(s->temp16[0][width-1], s->vlc[plane].table,
740  &s->gb, re, VLC_BITS, 3);
741  }
742  CLOSE_READER(re, &s->gb);
743  } else {
744  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
745  for (i = 0; i < count && get_bits_left(&s->gb) > 0; i++) {
746  READ_2PIX_PLANE16(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane);
747  }
748  } else {
749  for(i=0; i<count; i++){
750  READ_2PIX_PLANE16(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane);
751  }
752  }
753  if( width&1 && get_bits_left(&s->gb)>0 ) {
754  int dst = get_vlc2(&s->gb, s->vlc[plane].table, VLC_BITS, 3)<<2;
755  s->temp16[0][width-1] = dst + get_bits(&s->gb, 2);
756  }
757  }
758 }
759 
761 {
762  int i;
763  OPEN_READER(re, &s->gb);
764  count /= 2;
765 
766  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
767  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
768  READ_2PIX(s->temp[0][2 * i], s->temp[0][2 * i + 1], 0);
769  }
770  } else {
771  for (i = 0; i < count; i++) {
772  READ_2PIX(s->temp[0][2 * i], s->temp[0][2 * i + 1], 0);
773  }
774  }
775  CLOSE_READER(re, &s->gb);
776 }
777 
779  int decorrelate, int alpha)
780 {
781  int i;
782  OPEN_READER(re, &s->gb);
783 
784  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
785  unsigned int index;
786  int code, n, nb_bits;
787 
788  UPDATE_CACHE(re, &s->gb);
789  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
790  n = s->vlc[4].table[index][1];
791 
792  if (n>0) {
793  code = s->vlc[4].table[index][0];
794  *(uint32_t *) &s->temp[0][4 * i] = s->pix_bgr_map[code];
795  LAST_SKIP_BITS(re, &s->gb, n);
796  } else {
797  if (decorrelate) {
798  VLC_INTERN(s->temp[0][4 * i + G], s->vlc[1].table,
799  &s->gb, re, VLC_BITS, 3);
800 
801  UPDATE_CACHE(re, &s->gb);
802  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
803  VLC_INTERN(code, s->vlc[0].table, &s->gb, re, VLC_BITS, 3);
804  s->temp[0][4 * i + B] = code + s->temp[0][4 * i + G];
805 
806  UPDATE_CACHE(re, &s->gb);
807  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
808  VLC_INTERN(code, s->vlc[2].table, &s->gb, re, VLC_BITS, 3);
809  s->temp[0][4 * i + R] = code + s->temp[0][4 * i + G];
810  } else {
811  VLC_INTERN(s->temp[0][4 * i + B], s->vlc[0].table,
812  &s->gb, re, VLC_BITS, 3);
813 
814  UPDATE_CACHE(re, &s->gb);
815  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
816  VLC_INTERN(s->temp[0][4 * i + G], s->vlc[1].table,
817  &s->gb, re, VLC_BITS, 3);
818 
819  UPDATE_CACHE(re, &s->gb);
820  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
821  VLC_INTERN(s->temp[0][4 * i + R], s->vlc[2].table,
822  &s->gb, re, VLC_BITS, 3);
823  }
824  }
825  if (alpha) {
826  UPDATE_CACHE(re, &s->gb);
827  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
828  VLC_INTERN(s->temp[0][4 * i + A], s->vlc[2].table,
829  &s->gb, re, VLC_BITS, 3);
830  } else
831  s->temp[0][4 * i + A] = 0;
832  }
833  CLOSE_READER(re, &s->gb);
834 }
835 
837 {
838  if (s->decorrelate) {
839  if (s->bitstream_bpp == 24)
840  decode_bgr_1(s, count, 1, 0);
841  else
842  decode_bgr_1(s, count, 1, 1);
843  } else {
844  if (s->bitstream_bpp == 24)
845  decode_bgr_1(s, count, 0, 0);
846  else
847  decode_bgr_1(s, count, 0, 1);
848  }
849 }
850 
851 static void draw_slice(HYuvContext *s, AVFrame *frame, int y)
852 {
853  int h, cy, i;
855 
856  if (!s->avctx->draw_horiz_band)
857  return;
858 
859  h = y - s->last_slice_end;
860  y -= h;
861 
862  if (s->bitstream_bpp == 12)
863  cy = y >> 1;
864  else
865  cy = y;
866 
867  offset[0] = frame->linesize[0] * y;
868  offset[1] = frame->linesize[1] * cy;
869  offset[2] = frame->linesize[2] * cy;
870  for (i = 3; i < AV_NUM_DATA_POINTERS; i++)
871  offset[i] = 0;
872  emms_c();
873 
874  s->avctx->draw_horiz_band(s->avctx, frame, offset, y, 3, h);
875 
876  s->last_slice_end = y + h;
877 }
878 
879 static int left_prediction(HYuvContext *s, uint8_t *dst, const uint8_t *src, int w, int acc)
880 {
881  if (s->bps <= 8) {
882  return s->llviddsp.add_left_pred(dst, src, w, acc);
883  } else {
884  return s->llviddsp.add_left_pred_int16(( uint16_t *)dst, (const uint16_t *)src, s->n-1, w, acc);
885  }
886 }
887 
888 static void add_bytes(HYuvContext *s, uint8_t *dst, uint8_t *src, int w)
889 {
890  if (s->bps <= 8) {
891  s->llviddsp.add_bytes(dst, src, w);
892  } else {
893  s->hdsp.add_int16((uint16_t*)dst, (const uint16_t*)src, s->n - 1, w);
894  }
895 }
896 
897 static void add_median_prediction(HYuvContext *s, uint8_t *dst, const uint8_t *src, const uint8_t *diff, int w, int *left, int *left_top)
898 {
899  if (s->bps <= 8) {
900  s->llviddsp.add_median_pred(dst, src, diff, w, left, left_top);
901  } else {
902  s->hdsp.add_hfyu_median_pred_int16((uint16_t *)dst, (const uint16_t *)src, (const uint16_t *)diff, s->n-1, w, left, left_top);
903  }
904 }
905 static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
906  AVPacket *avpkt)
907 {
908  const uint8_t *buf = avpkt->data;
909  int buf_size = avpkt->size;
910  HYuvContext *s = avctx->priv_data;
911  const int width = s->width;
912  const int width2 = s->width >> 1;
913  const int height = s->height;
914  int fake_ystride, fake_ustride, fake_vstride;
915  ThreadFrame frame = { .f = data };
916  AVFrame *const p = data;
917  int table_size = 0, ret;
918 
919  if (buf_size < (width * height + 7)/8)
920  return AVERROR_INVALIDDATA;
921 
924  buf_size);
925  if (!s->bitstream_buffer)
926  return AVERROR(ENOMEM);
927 
928  s->bdsp.bswap_buf((uint32_t *) s->bitstream_buffer,
929  (const uint32_t *) buf, buf_size / 4);
930 
931  if ((ret = ff_thread_get_buffer(avctx, &frame, 0)) < 0)
932  return ret;
933 
934  if (s->context) {
935  table_size = read_huffman_tables(s, s->bitstream_buffer, buf_size);
936  if (table_size < 0)
937  return table_size;
938  }
939 
940  if ((unsigned) (buf_size - table_size) >= INT_MAX / 8)
941  return AVERROR_INVALIDDATA;
942 
943  if ((ret = init_get_bits(&s->gb, s->bitstream_buffer + table_size,
944  (buf_size - table_size) * 8)) < 0)
945  return ret;
946 
947  fake_ystride = s->interlaced ? p->linesize[0] * 2 : p->linesize[0];
948  fake_ustride = s->interlaced ? p->linesize[1] * 2 : p->linesize[1];
949  fake_vstride = s->interlaced ? p->linesize[2] * 2 : p->linesize[2];
950 
951  s->last_slice_end = 0;
952 
953  if (s->version > 2) {
954  int plane;
955  for(plane = 0; plane < 1 + 2*s->chroma + s->alpha; plane++) {
956  int left, lefttop, y;
957  int w = width;
958  int h = height;
959  int fake_stride = fake_ystride;
960 
961  if (s->chroma && (plane == 1 || plane == 2)) {
962  w >>= s->chroma_h_shift;
963  h >>= s->chroma_v_shift;
964  fake_stride = plane == 1 ? fake_ustride : fake_vstride;
965  }
966 
967  switch (s->predictor) {
968  case LEFT:
969  case PLANE:
970  decode_plane_bitstream(s, w, plane);
971  left = left_prediction(s, p->data[plane], s->temp[0], w, 0);
972 
973  for (y = 1; y < h; y++) {
974  uint8_t *dst = p->data[plane] + p->linesize[plane]*y;
975 
976  decode_plane_bitstream(s, w, plane);
977  left = left_prediction(s, dst, s->temp[0], w, left);
978  if (s->predictor == PLANE) {
979  if (y > s->interlaced) {
980  add_bytes(s, dst, dst - fake_stride, w);
981  }
982  }
983  }
984 
985  break;
986  case MEDIAN:
987  decode_plane_bitstream(s, w, plane);
988  left= left_prediction(s, p->data[plane], s->temp[0], w, 0);
989 
990  y = 1;
991 
992  /* second line is left predicted for interlaced case */
993  if (s->interlaced) {
994  decode_plane_bitstream(s, w, plane);
995  left = left_prediction(s, p->data[plane] + p->linesize[plane], s->temp[0], w, left);
996  y++;
997  }
998 
999  lefttop = p->data[plane][0];
1000  decode_plane_bitstream(s, w, plane);
1001  add_median_prediction(s, p->data[plane] + fake_stride, p->data[plane], s->temp[0], w, &left, &lefttop);
1002  y++;
1003 
1004  for (; y<h; y++) {
1005  uint8_t *dst;
1006 
1007  decode_plane_bitstream(s, w, plane);
1008 
1009  dst = p->data[plane] + p->linesize[plane] * y;
1010 
1011  add_median_prediction(s, dst, dst - fake_stride, s->temp[0], w, &left, &lefttop);
1012  }
1013 
1014  break;
1015  }
1016  }
1017  draw_slice(s, p, height);
1018  } else if (s->bitstream_bpp < 24) {
1019  int y, cy;
1020  int lefty, leftu, leftv;
1021  int lefttopy, lefttopu, lefttopv;
1022 
1023  if (s->yuy2) {
1024  p->data[0][3] = get_bits(&s->gb, 8);
1025  p->data[0][2] = get_bits(&s->gb, 8);
1026  p->data[0][1] = get_bits(&s->gb, 8);
1027  p->data[0][0] = get_bits(&s->gb, 8);
1028 
1029  av_log(avctx, AV_LOG_ERROR,
1030  "YUY2 output is not implemented yet\n");
1031  return AVERROR_PATCHWELCOME;
1032  } else {
1033  leftv =
1034  p->data[2][0] = get_bits(&s->gb, 8);
1035  lefty =
1036  p->data[0][1] = get_bits(&s->gb, 8);
1037  leftu =
1038  p->data[1][0] = get_bits(&s->gb, 8);
1039  p->data[0][0] = get_bits(&s->gb, 8);
1040 
1041  switch (s->predictor) {
1042  case LEFT:
1043  case PLANE:
1044  decode_422_bitstream(s, width - 2);
1045  lefty = s->llviddsp.add_left_pred(p->data[0] + 2, s->temp[0],
1046  width - 2, lefty);
1047  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1048  leftu = s->llviddsp.add_left_pred(p->data[1] + 1, s->temp[1], width2 - 1, leftu);
1049  leftv = s->llviddsp.add_left_pred(p->data[2] + 1, s->temp[2], width2 - 1, leftv);
1050  }
1051 
1052  for (cy = y = 1; y < s->height; y++, cy++) {
1053  uint8_t *ydst, *udst, *vdst;
1054 
1055  if (s->bitstream_bpp == 12) {
1056  decode_gray_bitstream(s, width);
1057 
1058  ydst = p->data[0] + p->linesize[0] * y;
1059 
1060  lefty = s->llviddsp.add_left_pred(ydst, s->temp[0],
1061  width, lefty);
1062  if (s->predictor == PLANE) {
1063  if (y > s->interlaced)
1064  s->llviddsp.add_bytes(ydst, ydst - fake_ystride, width);
1065  }
1066  y++;
1067  if (y >= s->height)
1068  break;
1069  }
1070 
1071  draw_slice(s, p, y);
1072 
1073  ydst = p->data[0] + p->linesize[0] * y;
1074  udst = p->data[1] + p->linesize[1] * cy;
1075  vdst = p->data[2] + p->linesize[2] * cy;
1076 
1077  decode_422_bitstream(s, width);
1078  lefty = s->llviddsp.add_left_pred(ydst, s->temp[0],
1079  width, lefty);
1080  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1081  leftu = s->llviddsp.add_left_pred(udst, s->temp[1], width2, leftu);
1082  leftv = s->llviddsp.add_left_pred(vdst, s->temp[2], width2, leftv);
1083  }
1084  if (s->predictor == PLANE) {
1085  if (cy > s->interlaced) {
1086  s->llviddsp.add_bytes(ydst, ydst - fake_ystride, width);
1087  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1088  s->llviddsp.add_bytes(udst, udst - fake_ustride, width2);
1089  s->llviddsp.add_bytes(vdst, vdst - fake_vstride, width2);
1090  }
1091  }
1092  }
1093  }
1094  draw_slice(s, p, height);
1095 
1096  break;
1097  case MEDIAN:
1098  /* first line except first 2 pixels is left predicted */
1099  decode_422_bitstream(s, width - 2);
1100  lefty = s->llviddsp.add_left_pred(p->data[0] + 2, s->temp[0],
1101  width - 2, lefty);
1102  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1103  leftu = s->llviddsp.add_left_pred(p->data[1] + 1, s->temp[1], width2 - 1, leftu);
1104  leftv = s->llviddsp.add_left_pred(p->data[2] + 1, s->temp[2], width2 - 1, leftv);
1105  }
1106 
1107  cy = y = 1;
1108 
1109  /* second line is left predicted for interlaced case */
1110  if (s->interlaced) {
1111  decode_422_bitstream(s, width);
1112  lefty = s->llviddsp.add_left_pred(p->data[0] + p->linesize[0],
1113  s->temp[0], width, lefty);
1114  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1115  leftu = s->llviddsp.add_left_pred(p->data[1] + p->linesize[2], s->temp[1], width2, leftu);
1116  leftv = s->llviddsp.add_left_pred(p->data[2] + p->linesize[1], s->temp[2], width2, leftv);
1117  }
1118  y++;
1119  cy++;
1120  }
1121 
1122  /* next 4 pixels are left predicted too */
1123  decode_422_bitstream(s, 4);
1124  lefty = s->llviddsp.add_left_pred(p->data[0] + fake_ystride,
1125  s->temp[0], 4, lefty);
1126  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1127  leftu = s->llviddsp.add_left_pred(p->data[1] + fake_ustride, s->temp[1], 2, leftu);
1128  leftv = s->llviddsp.add_left_pred(p->data[2] + fake_vstride, s->temp[2], 2, leftv);
1129  }
1130 
1131  /* next line except the first 4 pixels is median predicted */
1132  lefttopy = p->data[0][3];
1133  decode_422_bitstream(s, width - 4);
1134  s->llviddsp.add_median_pred(p->data[0] + fake_ystride + 4,
1135  p->data[0] + 4, s->temp[0],
1136  width - 4, &lefty, &lefttopy);
1137  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1138  lefttopu = p->data[1][1];
1139  lefttopv = p->data[2][1];
1140  s->llviddsp.add_median_pred(p->data[1] + fake_ustride + 2, p->data[1] + 2, s->temp[1], width2 - 2, &leftu, &lefttopu);
1141  s->llviddsp.add_median_pred(p->data[2] + fake_vstride + 2, p->data[2] + 2, s->temp[2], width2 - 2, &leftv, &lefttopv);
1142  }
1143  y++;
1144  cy++;
1145 
1146  for (; y < height; y++, cy++) {
1147  uint8_t *ydst, *udst, *vdst;
1148 
1149  if (s->bitstream_bpp == 12) {
1150  while (2 * cy > y) {
1151  decode_gray_bitstream(s, width);
1152  ydst = p->data[0] + p->linesize[0] * y;
1153  s->llviddsp.add_median_pred(ydst, ydst - fake_ystride,
1154  s->temp[0], width,
1155  &lefty, &lefttopy);
1156  y++;
1157  }
1158  if (y >= height)
1159  break;
1160  }
1161  draw_slice(s, p, y);
1162 
1163  decode_422_bitstream(s, width);
1164 
1165  ydst = p->data[0] + p->linesize[0] * y;
1166  udst = p->data[1] + p->linesize[1] * cy;
1167  vdst = p->data[2] + p->linesize[2] * cy;
1168 
1169  s->llviddsp.add_median_pred(ydst, ydst - fake_ystride,
1170  s->temp[0], width,
1171  &lefty, &lefttopy);
1172  if (!(s->flags & AV_CODEC_FLAG_GRAY)) {
1173  s->llviddsp.add_median_pred(udst, udst - fake_ustride, s->temp[1], width2, &leftu, &lefttopu);
1174  s->llviddsp.add_median_pred(vdst, vdst - fake_vstride, s->temp[2], width2, &leftv, &lefttopv);
1175  }
1176  }
1177 
1178  draw_slice(s, p, height);
1179  break;
1180  }
1181  }
1182  } else {
1183  int y;
1184  uint8_t left[4];
1185  const int last_line = (height - 1) * p->linesize[0];
1186 
1187  if (s->bitstream_bpp == 32) {
1188  left[A] = p->data[0][last_line + A] = get_bits(&s->gb, 8);
1189  left[R] = p->data[0][last_line + R] = get_bits(&s->gb, 8);
1190  left[G] = p->data[0][last_line + G] = get_bits(&s->gb, 8);
1191  left[B] = p->data[0][last_line + B] = get_bits(&s->gb, 8);
1192  } else {
1193  left[R] = p->data[0][last_line + R] = get_bits(&s->gb, 8);
1194  left[G] = p->data[0][last_line + G] = get_bits(&s->gb, 8);
1195  left[B] = p->data[0][last_line + B] = get_bits(&s->gb, 8);
1196  left[A] = p->data[0][last_line + A] = 255;
1197  skip_bits(&s->gb, 8);
1198  }
1199 
1200  if (s->bgr32) {
1201  switch (s->predictor) {
1202  case LEFT:
1203  case PLANE:
1204  decode_bgr_bitstream(s, width - 1);
1205  s->hdsp.add_hfyu_left_pred_bgr32(p->data[0] + last_line + 4,
1206  s->temp[0], width - 1, left);
1207 
1208  for (y = s->height - 2; y >= 0; y--) { // Yes it is stored upside down.
1209  decode_bgr_bitstream(s, width);
1210 
1211  s->hdsp.add_hfyu_left_pred_bgr32(p->data[0] + p->linesize[0] * y,
1212  s->temp[0], width, left);
1213  if (s->predictor == PLANE) {
1214  if (s->bitstream_bpp != 32)
1215  left[A] = 0;
1216  if (y < s->height - 1 - s->interlaced) {
1217  s->llviddsp.add_bytes(p->data[0] + p->linesize[0] * y,
1218  p->data[0] + p->linesize[0] * y +
1219  fake_ystride, 4 * width);
1220  }
1221  }
1222  }
1223  // just 1 large slice as this is not possible in reverse order
1224  draw_slice(s, p, height);
1225  break;
1226  default:
1227  av_log(avctx, AV_LOG_ERROR,
1228  "prediction type not supported!\n");
1229  }
1230  } else {
1231  av_log(avctx, AV_LOG_ERROR,
1232  "BGR24 output is not implemented yet\n");
1233  return AVERROR_PATCHWELCOME;
1234  }
1235  }
1236  emms_c();
1237 
1238  *got_frame = 1;
1239 
1240  return (get_bits_count(&s->gb) + 31) / 32 * 4 + table_size;
1241 }
1242 
1244  .name = "huffyuv",
1245  .long_name = NULL_IF_CONFIG_SMALL("Huffyuv / HuffYUV"),
1246  .type = AVMEDIA_TYPE_VIDEO,
1247  .id = AV_CODEC_ID_HUFFYUV,
1248  .priv_data_size = sizeof(HYuvContext),
1249  .init = decode_init,
1250  .close = decode_end,
1251  .decode = decode_frame,
1254  .init_thread_copy = ONLY_IF_THREADS_ENABLED(decode_init_thread_copy),
1255 };
1256 
1257 #if CONFIG_FFVHUFF_DECODER
1259  .name = "ffvhuff",
1260  .long_name = NULL_IF_CONFIG_SMALL("Huffyuv FFmpeg variant"),
1261  .type = AVMEDIA_TYPE_VIDEO,
1262  .id = AV_CODEC_ID_FFVHUFF,
1263  .priv_data_size = sizeof(HYuvContext),
1264  .init = decode_init,
1265  .close = decode_end,
1266  .decode = decode_frame,
1269  .init_thread_copy = ONLY_IF_THREADS_ENABLED(decode_init_thread_copy),
1270 };
1271 #endif /* CONFIG_FFVHUFF_DECODER */
int plane
Definition: avisynth_c.h:422
static void add_median_prediction(HYuvContext *s, uint8_t *dst, const uint8_t *src, const uint8_t *diff, int w, int *left, int *left_top)
Definition: huffyuvdec.c:897
#define OP14bits(dst0, dst1, code)
Definition: huffyuvdec.c:689
#define NULL
Definition: coverity.c:32
const char const char void * val
Definition: avisynth_c.h:771
#define AV_PIX_FMT_YUVA422P16
Definition: pixfmt.h:420
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int(* add_left_pred)(uint8_t *dst, const uint8_t *src, ptrdiff_t w, int left)
#define AV_NUM_DATA_POINTERS
Definition: frame.h:227
static void decode_plane_bitstream(HYuvContext *s, int width, int plane)
Definition: huffyuvdec.c:699
#define AV_PIX_FMT_YUVA422P9
Definition: pixfmt.h:414
#define READ_2PIX_PLANE(dst0, dst1, plane, OP)
Definition: huffyuvdec.c:684
This structure describes decoded (raw) audio or video data.
Definition: frame.h:226
#define AV_PIX_FMT_YUVA420P10
Definition: pixfmt.h:416
#define AV_PIX_FMT_YUV444P14
Definition: pixfmt.h:389
float re
Definition: fft.c:82
int yuy2
Definition: huffyuv.h:65
#define AV_PIX_FMT_YUVA422P10
Definition: pixfmt.h:417
int bitstream_bpp
Definition: huffyuv.h:63
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
misc image utilities
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:381
static int init_thread_copy(AVCodecContext *avctx)
Definition: tta.c:392
const char * g
Definition: vf_curves.c:115
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
int acc
Definition: yuv2rgb.c:554
static void decode_422_bitstream(HYuvContext *s, int count)
Definition: huffyuvdec.c:655
int ff_init_vlc_sparse(VLC *vlc_arg, int nb_bits, int nb_codes, const void *bits, int bits_wrap, int bits_size, const void *codes, int codes_wrap, int codes_size, const void *symbols, int symbols_wrap, int symbols_size, int flags)
Definition: bitstream.c:273
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:168
int size
Definition: avcodec.h:1446
const char * b
Definition: vf_curves.c:116
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:395
static av_always_inline void decode_bgr_1(HYuvContext *s, int count, int decorrelate, int alpha)
Definition: huffyuvdec.c:778
#define VLC_BITS
Definition: asvdec.c:37
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1743
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:70
#define AV_PIX_FMT_YUV420P12
Definition: pixfmt.h:383
#define MAX_VLC_N
Definition: huffyuv.h:47
int context
Definition: huffyuv.h:77
#define READ_2PIX_PLANE16(dst0, dst1, plane)
Definition: huffyuvdec.c:693
unsigned int bitstream_buffer_size
Definition: huffyuv.h:88
#define init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, codes, codes_wrap, codes_size, flags)
Definition: vlc.h:38
#define src
Definition: vp8dsp.c:254
static int generate_joint_tables(HYuvContext *s)
Definition: huffyuvdec.c:118
AVCodec.
Definition: avcodec.h:3424
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:42
int height
Definition: huffyuv.h:75
void(* draw_horiz_band)(struct AVCodecContext *s, const AVFrame *src, int offset[AV_NUM_DATA_POINTERS], int y, int type, int height)
If non NULL, &#39;draw_horiz_band&#39; is called by the libavcodec decoder to draw a horizontal band...
Definition: avcodec.h:1768
#define READ_2PIX(dst0, dst1, plane1)
Definition: huffyuvdec.c:650
static int read_old_huffman_tables(HYuvContext *s)
Definition: huffyuvdec.c:237
static void draw_slice(HYuvContext *s, AVFrame *frame, int y)
Definition: huffyuvdec.c:851
void(* add_int16)(uint16_t *dst, const uint16_t *src, unsigned mask, int w)
Definition: huffyuvdsp.h:39
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
uint32_t pix_bgr_map[1<< VLC_BITS]
Definition: huffyuv.h:85
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
Definition: pixfmt.h:101
uint8_t
#define av_cold
Definition: attributes.h:82
void(* bswap_buf)(uint32_t *dst, const uint32_t *src, int w)
Definition: bswapdsp.h:25
static void decode_bgr_bitstream(HYuvContext *s, int count)
Definition: huffyuvdec.c:836
int bps
Definition: huffyuv.h:67
Multithreading support functions.
#define emms_c()
Definition: internal.h:55
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1634
#define u(width, name, range_min, range_max)
Definition: cbs_h2645.c:253
#define AV_PIX_FMT_YUVA420P9
Definition: pixfmt.h:413
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:394
static AVFrame * frame
const char data[16]
Definition: mxf.c:91
#define height
uint8_t * data
Definition: avcodec.h:1445
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:219
void(* add_median_pred)(uint8_t *dst, const uint8_t *top, const uint8_t *diff, ptrdiff_t w, int *left, int *left_top)
bitstream reader API header.
int vlc_n
Definition: huffyuv.h:69
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:392
int chroma_h_shift
Definition: huffyuv.h:73
int bits_per_coded_sample
bits per sample/pixel from the demuxer (needed for huffyuv).
Definition: avcodec.h:2750
#define AV_PIX_FMT_YUV422P12
Definition: pixfmt.h:384
#define AV_PIX_FMT_YUVA420P16
Definition: pixfmt.h:419
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
Definition: avcodec.h:870
#define A(x)
Definition: vp56_arith.h:28
uint16_t * temp16[3]
identical to temp but 16bit type
Definition: huffyuv.h:81
#define av_log(a,...)
uint8_t len[4][MAX_VLC_N]
Definition: huffyuv.h:83
av_cold int ff_huffyuv_alloc_temp(HYuvContext *s)
Definition: huffyuv.c:58
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:814
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
Definition: pixfmt.h:176
#define UPDATE_CACHE(name, gb)
Definition: get_bits.h:178
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:258
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
#define R
Definition: huffyuvdsp.h:34
int chroma_v_shift
Definition: huffyuv.h:74
Definition: huffyuv.h:51
av_cold void ff_huffyuv_common_end(HYuvContext *s)
Definition: huffyuv.c:86
static const unsigned char classic_add_luma[256]
Definition: huffyuvdec.c:61
uint8_t * bitstream_buffer
Definition: huffyuv.h:87
#define AVERROR(e)
Definition: error.h:43
int flags
Definition: huffyuv.h:76
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:2474
#define B
Definition: huffyuvdsp.h:32
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
const char * r
Definition: vf_curves.c:114
#define AV_PIX_FMT_YUVA444P16
Definition: pixfmt.h:421
int bgr32
Definition: huffyuv.h:66
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:236
const char * name
Name of the codec implementation.
Definition: avcodec.h:3431
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:382
int chroma
Definition: huffyuv.h:71
VLC vlc[8]
Definition: huffyuv.h:86
static const uint8_t offset[127][2]
Definition: vf_spp.c:92
#define CLOSE_READER(name, gb)
Definition: get_bits.h:149
huffyuv codec for libavcodec.
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
Definition: avcodec.h:1024
#define classic_shift_chroma_table_size
Definition: huffyuvdec.c:52
Definition: vlc.h:26
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:70
#define ONLY_IF_THREADS_ENABLED(x)
Define a function with only the non-default version specified.
Definition: internal.h:225
#define AV_PIX_FMT_YUV422P9
Definition: pixfmt.h:377
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:282
#define AV_CODEC_CAP_DRAW_HORIZ_BAND
Decoder can use draw_horiz_band callback.
Definition: avcodec.h:962
int decorrelate
Definition: huffyuv.h:62
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:398
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:363
#define FFMIN(a, b)
Definition: common.h:96
int width
Definition: huffyuv.h:75
int last_slice_end
Definition: huffyuv.h:79
#define VLC_INTERN(dst, table, gb, name, bits, max_depth)
Subset of GET_VLC for use in hand-roller VLC code.
Definition: huffyuvdec.c:603
#define width
int width
picture width / height.
Definition: avcodec.h:1706
static const unsigned char classic_shift_chroma[classic_shift_chroma_table_size+AV_INPUT_BUFFER_PADDING_SIZE]
Definition: huffyuvdec.c:53
uint8_t w
Definition: llviddspenc.c:38
int ff_huffyuv_generate_bits_table(uint32_t *dst, const uint8_t *len_table, int n)
Definition: huffyuv.c:39
#define s(width, name)
Definition: cbs_vp9.c:257
#define LAST_SKIP_BITS(name, gb, num)
Definition: get_bits.h:199
static av_always_inline int get_vlc2(GetBitContext *s, VLC_TYPE(*table)[2], int bits, int max_depth)
Parse a vlc code.
Definition: get_bits.h:762
int n
Definition: avisynth_c.h:684
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:69
#define AV_PIX_FMT_YUVA444P10
Definition: pixfmt.h:418
uint8_t * temp[3]
Definition: huffyuv.h:80
AVCodec ff_ffvhuff_decoder
if(ret< 0)
Definition: vf_mcdeint.c:279
#define AV_PIX_FMT_YUV444P9
Definition: pixfmt.h:378
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:397
static void error(const char *err)
#define SHOW_UBITS(name, gb, num)
Definition: get_bits.h:211
int alpha
Definition: huffyuv.h:70
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:62
#define AV_PIX_FMT_YUV420P16
Definition: pixfmt.h:390
static void add_bytes(HYuvContext *s, uint8_t *dst, uint8_t *src, int w)
Definition: huffyuvdec.c:888
int(* add_left_pred_int16)(uint16_t *dst, const uint16_t *src, unsigned mask, ptrdiff_t w, unsigned left)
#define AV_PIX_FMT_YUV420P14
Definition: pixfmt.h:387
AVCodec ff_huffyuv_decoder
Definition: huffyuvdec.c:1243
Libavcodec external API header.
int yuv
Definition: huffyuv.h:72
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:257
LLVidDSPContext llviddsp
Definition: huffyuv.h:92
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
Definition: pixfmt.h:177
HuffYUVDSPContext hdsp
Definition: huffyuv.h:90
int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
Wrapper around get_buffer() for frame-multithreaded codecs.
static const int16_t alpha[]
Definition: ilbcdata.h:55
main external API structure.
Definition: avcodec.h:1533
#define AV_PIX_FMT_RGB32
Definition: pixfmt.h:352
packed YUV 4:2:2, 16bpp, Y0 Cb Y1 Cr
Definition: pixfmt.h:67
#define OPEN_READER(name, gb)
Definition: get_bits.h:138
void * buf
Definition: avisynth_c.h:690
int interlaced
Definition: huffyuv.h:61
int extradata_size
Definition: avcodec.h:1635
void ff_llviddsp_init(LLVidDSPContext *c)
#define AV_PIX_FMT_YUV420P10
Definition: pixfmt.h:379
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:72
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:460
int index
Definition: gxfenc.c:89
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:615
av_cold void ff_huffyuvdsp_init(HuffYUVDSPContext *c, enum AVPixelFormat pix_fmt)
Definition: huffyuvdsp.c:83
static void decode_gray_bitstream(HYuvContext *s, int count)
Definition: huffyuvdec.c:760
#define AV_PIX_FMT_YUV420P9
Definition: pixfmt.h:376
#define classic_shift_luma_table_size
Definition: huffyuvdec.c:44
const VDPAUPixFmtMap * map
static av_const int sign_extend(int val, unsigned bits)
Definition: mathops.h:130
#define AV_PIX_FMT_YUV422P14
Definition: pixfmt.h:388
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:396
int version
Definition: huffyuv.h:64
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:380
#define AV_PIX_FMT_YUV444P12
Definition: pixfmt.h:386
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:240
Predictor predictor
Definition: huffyuv.h:58
static av_cold int decode_init(AVCodecContext *avctx)
Definition: huffyuvdec.c:291
void(* add_hfyu_left_pred_bgr32)(uint8_t *dst, const uint8_t *src, intptr_t w, uint8_t *left)
Definition: huffyuvdsp.h:45
AVCodecContext * avctx
Definition: huffyuv.h:57
GetBitContext gb
Definition: huffyuv.h:59
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
Y , 8bpp.
Definition: pixfmt.h:74
Definition: huffyuv.h:52
#define G
Definition: huffyuvdsp.h:33
#define BITS_LEFT(name, gb)
Definition: get_bits.h:191
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:215
void(* add_hfyu_median_pred_int16)(uint16_t *dst, const uint16_t *top, const uint16_t *diff, unsigned mask, int w, int *left, int *left_top)
Definition: huffyuvdsp.h:42
#define AV_PIX_FMT_YUVA444P9
Definition: pixfmt.h:415
static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: huffyuvdec.c:905
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:73
av_cold void ff_huffyuv_common_init(AVCodecContext *avctx)
Definition: huffyuv.c:71
static av_cold int decode_end(AVCodecContext *avctx)
Definition: huffyuvdec.c:277
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
Definition: avcodec.h:782
void * priv_data
Definition: avcodec.h:1560
static av_always_inline int diff(const uint32_t a, const uint32_t b)
int len
static const unsigned char classic_add_chroma[256]
Definition: huffyuvdec.c:80
VLC_TYPE(* table)[2]
code, bits
Definition: vlc.h:28
static int left_prediction(HYuvContext *s, uint8_t *dst, const uint8_t *src, int w, int acc)
Definition: huffyuvdec.c:879
static const unsigned char classic_shift_luma[classic_shift_luma_table_size+AV_INPUT_BUFFER_PADDING_SIZE]
Definition: huffyuvdec.c:45
FILE * out
Definition: movenc.c:54
#define av_freep(p)
void INT64 INT64 count
Definition: avisynth_c.h:690
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:99
#define av_always_inline
Definition: attributes.h:39
uint32_t bits[4][MAX_VLC_N]
Definition: huffyuv.h:84
const char int length
Definition: avisynth_c.h:768
#define LEFT
Definition: cdgraphics.c:166
void(* add_bytes)(uint8_t *dst, uint8_t *src, ptrdiff_t w)
This structure stores compressed data.
Definition: avcodec.h:1422
static void decorrelate(SnowContext *s, SubBand *b, IDWTELEM *src, int stride, int inverse, int use_median)
Definition: snowenc.c:1329
BswapDSPContext bdsp
Definition: huffyuv.h:89
void ff_free_vlc(VLC *vlc)
Definition: bitstream.c:359
static int read_len_table(uint8_t *dst, GetBitContext *gb, int n)
Definition: huffyuvdec.c:99
#define OP8bits(dst0, dst1, code)
Definition: huffyuvdec.c:648
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:968
#define AV_PIX_FMT_YUV422P16
Definition: pixfmt.h:391
static int read_huffman_tables(HYuvContext *s, const uint8_t *src, int length)
Definition: huffyuvdec.c:208
for(j=16;j >0;--j)
#define AV_PIX_FMT_0RGB32
Definition: pixfmt.h:356