FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
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 0x170:
422  avctx->pix_fmt = AV_PIX_FMT_GRAY8A;
423  break;
424  case 0x470:
425  avctx->pix_fmt = AV_PIX_FMT_GBRP;
426  break;
427  case 0x480:
428  avctx->pix_fmt = AV_PIX_FMT_GBRP9;
429  break;
430  case 0x490:
431  avctx->pix_fmt = AV_PIX_FMT_GBRP10;
432  break;
433  case 0x4B0:
434  avctx->pix_fmt = AV_PIX_FMT_GBRP12;
435  break;
436  case 0x4D0:
437  avctx->pix_fmt = AV_PIX_FMT_GBRP14;
438  break;
439  case 0x4F0:
440  avctx->pix_fmt = AV_PIX_FMT_GBRP16;
441  break;
442  case 0x570:
443  avctx->pix_fmt = AV_PIX_FMT_GBRAP;
444  break;
445  case 0x670:
446  avctx->pix_fmt = AV_PIX_FMT_YUV444P;
447  break;
448  case 0x680:
449  avctx->pix_fmt = AV_PIX_FMT_YUV444P9;
450  break;
451  case 0x690:
452  avctx->pix_fmt = AV_PIX_FMT_YUV444P10;
453  break;
454  case 0x6B0:
455  avctx->pix_fmt = AV_PIX_FMT_YUV444P12;
456  break;
457  case 0x6D0:
458  avctx->pix_fmt = AV_PIX_FMT_YUV444P14;
459  break;
460  case 0x6F0:
461  avctx->pix_fmt = AV_PIX_FMT_YUV444P16;
462  break;
463  case 0x671:
464  avctx->pix_fmt = AV_PIX_FMT_YUV422P;
465  break;
466  case 0x681:
467  avctx->pix_fmt = AV_PIX_FMT_YUV422P9;
468  break;
469  case 0x691:
470  avctx->pix_fmt = AV_PIX_FMT_YUV422P10;
471  break;
472  case 0x6B1:
473  avctx->pix_fmt = AV_PIX_FMT_YUV422P12;
474  break;
475  case 0x6D1:
476  avctx->pix_fmt = AV_PIX_FMT_YUV422P14;
477  break;
478  case 0x6F1:
479  avctx->pix_fmt = AV_PIX_FMT_YUV422P16;
480  break;
481  case 0x672:
482  avctx->pix_fmt = AV_PIX_FMT_YUV411P;
483  break;
484  case 0x674:
485  avctx->pix_fmt = AV_PIX_FMT_YUV440P;
486  break;
487  case 0x675:
488  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
489  break;
490  case 0x685:
491  avctx->pix_fmt = AV_PIX_FMT_YUV420P9;
492  break;
493  case 0x695:
494  avctx->pix_fmt = AV_PIX_FMT_YUV420P10;
495  break;
496  case 0x6B5:
497  avctx->pix_fmt = AV_PIX_FMT_YUV420P12;
498  break;
499  case 0x6D5:
500  avctx->pix_fmt = AV_PIX_FMT_YUV420P14;
501  break;
502  case 0x6F5:
503  avctx->pix_fmt = AV_PIX_FMT_YUV420P16;
504  break;
505  case 0x67A:
506  avctx->pix_fmt = AV_PIX_FMT_YUV410P;
507  break;
508  case 0x770:
509  avctx->pix_fmt = AV_PIX_FMT_YUVA444P;
510  break;
511  case 0x780:
512  avctx->pix_fmt = AV_PIX_FMT_YUVA444P9;
513  break;
514  case 0x790:
516  break;
517  case 0x7F0:
519  break;
520  case 0x771:
521  avctx->pix_fmt = AV_PIX_FMT_YUVA422P;
522  break;
523  case 0x781:
524  avctx->pix_fmt = AV_PIX_FMT_YUVA422P9;
525  break;
526  case 0x791:
528  break;
529  case 0x7F1:
531  break;
532  case 0x775:
533  avctx->pix_fmt = AV_PIX_FMT_YUVA420P;
534  break;
535  case 0x785:
536  avctx->pix_fmt = AV_PIX_FMT_YUVA420P9;
537  break;
538  case 0x795:
540  break;
541  case 0x7F5:
543  break;
544  default:
545  ret = AVERROR_INVALIDDATA;
546  goto error;
547  }
548  }
549 
550  ff_huffyuv_common_init(avctx);
551 
552  if ((avctx->pix_fmt == AV_PIX_FMT_YUV422P || avctx->pix_fmt == AV_PIX_FMT_YUV420P) && avctx->width & 1) {
553  av_log(avctx, AV_LOG_ERROR, "width must be even for this colorspace\n");
554  ret = AVERROR_INVALIDDATA;
555  goto error;
556  }
557  if (s->predictor == MEDIAN && avctx->pix_fmt == AV_PIX_FMT_YUV422P &&
558  avctx->width % 4) {
559  av_log(avctx, AV_LOG_ERROR, "width must be a multiple of 4 "
560  "for this combination of colorspace and predictor type.\n");
561  ret = AVERROR_INVALIDDATA;
562  goto error;
563  }
564 
565  if ((ret = ff_huffyuv_alloc_temp(s)) < 0) {
567  goto error;
568  }
569 
570  return 0;
571  error:
572  decode_end(avctx);
573  return ret;
574 }
575 
576 #if HAVE_THREADS
577 static av_cold int decode_init_thread_copy(AVCodecContext *avctx)
578 {
579  HYuvContext *s = avctx->priv_data;
580  int i, ret;
581 
582  s->avctx = avctx;
583 
584  if ((ret = ff_huffyuv_alloc_temp(s)) < 0) {
586  return ret;
587  }
588 
589  for (i = 0; i < 8; i++)
590  s->vlc[i].table = NULL;
591 
592  if (s->version >= 2) {
593  if ((ret = read_huffman_tables(s, avctx->extradata + 4,
594  avctx->extradata_size)) < 0)
595  return ret;
596  } else {
597  if ((ret = read_old_huffman_tables(s)) < 0)
598  return ret;
599  }
600 
601  return 0;
602 }
603 #endif
604 
605 /** Subset of GET_VLC for use in hand-roller VLC code */
606 #define VLC_INTERN(dst, table, gb, name, bits, max_depth) \
607  code = table[index][0]; \
608  n = table[index][1]; \
609  if (max_depth > 1 && n < 0) { \
610  LAST_SKIP_BITS(name, gb, bits); \
611  UPDATE_CACHE(name, gb); \
612  \
613  nb_bits = -n; \
614  index = SHOW_UBITS(name, gb, nb_bits) + code; \
615  code = table[index][0]; \
616  n = table[index][1]; \
617  if (max_depth > 2 && n < 0) { \
618  LAST_SKIP_BITS(name, gb, nb_bits); \
619  UPDATE_CACHE(name, gb); \
620  \
621  nb_bits = -n; \
622  index = SHOW_UBITS(name, gb, nb_bits) + code; \
623  code = table[index][0]; \
624  n = table[index][1]; \
625  } \
626  } \
627  dst = code; \
628  LAST_SKIP_BITS(name, gb, n)
629 
630 
631 #define GET_VLC_DUAL(dst0, dst1, name, gb, dtable, table1, table2, \
632  bits, max_depth, OP) \
633  do { \
634  unsigned int index = SHOW_UBITS(name, gb, bits); \
635  int code, n = dtable[index][1]; \
636  \
637  if (n<=0) { \
638  int nb_bits; \
639  VLC_INTERN(dst0, table1, gb, name, bits, max_depth); \
640  \
641  UPDATE_CACHE(re, gb); \
642  index = SHOW_UBITS(name, gb, bits); \
643  VLC_INTERN(dst1, table2, gb, name, bits, max_depth); \
644  } else { \
645  code = dtable[index][0]; \
646  OP(dst0, dst1, code); \
647  LAST_SKIP_BITS(name, gb, n); \
648  } \
649  } while (0)
650 
651 #define OP8bits(dst0, dst1, code) dst0 = code>>8; dst1 = code
652 
653 #define READ_2PIX(dst0, dst1, plane1) \
654  UPDATE_CACHE(re, &s->gb); \
655  GET_VLC_DUAL(dst0, dst1, re, &s->gb, s->vlc[4+plane1].table, \
656  s->vlc[0].table, s->vlc[plane1].table, VLC_BITS, 3, OP8bits)
657 
659 {
660  int i, icount;
661  OPEN_READER(re, &s->gb);
662  count /= 2;
663 
664  icount = get_bits_left(&s->gb) / (32 * 4);
665  if (count >= icount) {
666  for (i = 0; i < icount; i++) {
667  READ_2PIX(s->temp[0][2 * i], s->temp[1][i], 1);
668  READ_2PIX(s->temp[0][2 * i + 1], s->temp[2][i], 2);
669  }
670  for (; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
671  READ_2PIX(s->temp[0][2 * i ], s->temp[1][i], 1);
672  if (BITS_LEFT(re, &s->gb) <= 0) break;
673  READ_2PIX(s->temp[0][2 * i + 1], s->temp[2][i], 2);
674  }
675  for (; i < count; i++)
676  s->temp[0][2 * i ] = s->temp[1][i] =
677  s->temp[0][2 * i + 1] = s->temp[2][i] = 0;
678  } else {
679  for (i = 0; i < count; i++) {
680  READ_2PIX(s->temp[0][2 * i], s->temp[1][i], 1);
681  READ_2PIX(s->temp[0][2 * i + 1], s->temp[2][i], 2);
682  }
683  }
684  CLOSE_READER(re, &s->gb);
685 }
686 
687 #define READ_2PIX_PLANE(dst0, dst1, plane, OP) \
688  UPDATE_CACHE(re, &s->gb); \
689  GET_VLC_DUAL(dst0, dst1, re, &s->gb, s->vlc[4+plane].table, \
690  s->vlc[plane].table, s->vlc[plane].table, VLC_BITS, 3, OP)
691 
692 #define OP14bits(dst0, dst1, code) dst0 = code>>8; dst1 = sign_extend(code, 8)
693 
694 /* TODO instead of restarting the read when the code isn't in the first level
695  * of the joint table, jump into the 2nd level of the individual table. */
696 #define READ_2PIX_PLANE16(dst0, dst1, plane){\
697  dst0 = get_vlc2(&s->gb, s->vlc[plane].table, VLC_BITS, 3)<<2;\
698  dst0 += get_bits(&s->gb, 2);\
699  dst1 = get_vlc2(&s->gb, s->vlc[plane].table, VLC_BITS, 3)<<2;\
700  dst1 += get_bits(&s->gb, 2);\
701 }
703 {
704  int i, count = width/2;
705 
706  if (s->bps <= 8) {
707  OPEN_READER(re, &s->gb);
708  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
709  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
710  READ_2PIX_PLANE(s->temp[0][2 * i], s->temp[0][2 * i + 1], plane, OP8bits);
711  }
712  } else {
713  for(i=0; i<count; i++){
714  READ_2PIX_PLANE(s->temp[0][2 * i], s->temp[0][2 * i + 1], plane, OP8bits);
715  }
716  }
717  if( width&1 && BITS_LEFT(re, &s->gb)>0 ) {
718  unsigned int index;
719  int nb_bits, code, n;
720  UPDATE_CACHE(re, &s->gb);
721  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
722  VLC_INTERN(s->temp[0][width-1], s->vlc[plane].table,
723  &s->gb, re, VLC_BITS, 3);
724  }
725  CLOSE_READER(re, &s->gb);
726  } else if (s->bps <= 14) {
727  OPEN_READER(re, &s->gb);
728  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
729  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
730  READ_2PIX_PLANE(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane, OP14bits);
731  }
732  } else {
733  for(i=0; i<count; i++){
734  READ_2PIX_PLANE(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane, OP14bits);
735  }
736  }
737  if( width&1 && BITS_LEFT(re, &s->gb)>0 ) {
738  unsigned int index;
739  int nb_bits, code, n;
740  UPDATE_CACHE(re, &s->gb);
741  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
742  VLC_INTERN(s->temp16[0][width-1], s->vlc[plane].table,
743  &s->gb, re, VLC_BITS, 3);
744  }
745  CLOSE_READER(re, &s->gb);
746  } else {
747  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
748  for (i = 0; i < count && get_bits_left(&s->gb) > 0; i++) {
749  READ_2PIX_PLANE16(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane);
750  }
751  } else {
752  for(i=0; i<count; i++){
753  READ_2PIX_PLANE16(s->temp16[0][2 * i], s->temp16[0][2 * i + 1], plane);
754  }
755  }
756  if( width&1 && get_bits_left(&s->gb)>0 ) {
757  int dst = get_vlc2(&s->gb, s->vlc[plane].table, VLC_BITS, 3)<<2;
758  s->temp16[0][width-1] = dst + get_bits(&s->gb, 2);
759  }
760  }
761 }
762 
764 {
765  int i;
766  OPEN_READER(re, &s->gb);
767  count /= 2;
768 
769  if (count >= (get_bits_left(&s->gb)) / (32 * 2)) {
770  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
771  READ_2PIX(s->temp[0][2 * i], s->temp[0][2 * i + 1], 0);
772  }
773  } else {
774  for (i = 0; i < count; i++) {
775  READ_2PIX(s->temp[0][2 * i], s->temp[0][2 * i + 1], 0);
776  }
777  }
778  CLOSE_READER(re, &s->gb);
779 }
780 
782  int decorrelate, int alpha)
783 {
784  int i;
785  OPEN_READER(re, &s->gb);
786 
787  for (i = 0; i < count && BITS_LEFT(re, &s->gb) > 0; i++) {
788  unsigned int index;
789  int code, n, nb_bits;
790 
791  UPDATE_CACHE(re, &s->gb);
792  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
793  n = s->vlc[4].table[index][1];
794 
795  if (n>0) {
796  code = s->vlc[4].table[index][0];
797  *(uint32_t *) &s->temp[0][4 * i] = s->pix_bgr_map[code];
798  LAST_SKIP_BITS(re, &s->gb, n);
799  } else {
800  if (decorrelate) {
801  VLC_INTERN(s->temp[0][4 * i + G], s->vlc[1].table,
802  &s->gb, re, VLC_BITS, 3);
803 
804  UPDATE_CACHE(re, &s->gb);
805  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
806  VLC_INTERN(code, s->vlc[0].table, &s->gb, re, VLC_BITS, 3);
807  s->temp[0][4 * i + B] = code + s->temp[0][4 * i + G];
808 
809  UPDATE_CACHE(re, &s->gb);
810  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
811  VLC_INTERN(code, s->vlc[2].table, &s->gb, re, VLC_BITS, 3);
812  s->temp[0][4 * i + R] = code + s->temp[0][4 * i + G];
813  } else {
814  VLC_INTERN(s->temp[0][4 * i + B], s->vlc[0].table,
815  &s->gb, re, VLC_BITS, 3);
816 
817  UPDATE_CACHE(re, &s->gb);
818  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
819  VLC_INTERN(s->temp[0][4 * i + G], s->vlc[1].table,
820  &s->gb, re, VLC_BITS, 3);
821 
822  UPDATE_CACHE(re, &s->gb);
823  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
824  VLC_INTERN(s->temp[0][4 * i + R], s->vlc[2].table,
825  &s->gb, re, VLC_BITS, 3);
826  }
827  }
828  if (alpha) {
829  UPDATE_CACHE(re, &s->gb);
830  index = SHOW_UBITS(re, &s->gb, VLC_BITS);
831  VLC_INTERN(s->temp[0][4 * i + A], s->vlc[2].table,
832  &s->gb, re, VLC_BITS, 3);
833  } else
834  s->temp[0][4 * i + A] = 0;
835  }
836  CLOSE_READER(re, &s->gb);
837 }
838 
840 {
841  if (s->decorrelate) {
842  if (s->bitstream_bpp == 24)
843  decode_bgr_1(s, count, 1, 0);
844  else
845  decode_bgr_1(s, count, 1, 1);
846  } else {
847  if (s->bitstream_bpp == 24)
848  decode_bgr_1(s, count, 0, 0);
849  else
850  decode_bgr_1(s, count, 0, 1);
851  }
852 }
853 
854 static void draw_slice(HYuvContext *s, AVFrame *frame, int y)
855 {
856  int h, cy, i;
858 
859  if (!s->avctx->draw_horiz_band)
860  return;
861 
862  h = y - s->last_slice_end;
863  y -= h;
864 
865  if (s->bitstream_bpp == 12)
866  cy = y >> 1;
867  else
868  cy = y;
869 
870  offset[0] = frame->linesize[0] * y;
871  offset[1] = frame->linesize[1] * cy;
872  offset[2] = frame->linesize[2] * cy;
873  for (i = 3; i < AV_NUM_DATA_POINTERS; i++)
874  offset[i] = 0;
875  emms_c();
876 
877  s->avctx->draw_horiz_band(s->avctx, frame, offset, y, 3, h);
878 
879  s->last_slice_end = y + h;
880 }
881 
882 static int left_prediction(HYuvContext *s, uint8_t *dst, const uint8_t *src, int w, int acc)
883 {
884  if (s->bps <= 8) {
885  return s->llviddsp.add_left_pred(dst, src, w, acc);
886  } else {
887  return s->llviddsp.add_left_pred_int16(( uint16_t *)dst, (const uint16_t *)src, s->n-1, w, acc);
888  }
889 }
890 
891 static void add_bytes(HYuvContext *s, uint8_t *dst, uint8_t *src, int w)
892 {
893  if (s->bps <= 8) {
894  s->llviddsp.add_bytes(dst, src, w);
895  } else {
896  s->hdsp.add_int16((uint16_t*)dst, (const uint16_t*)src, s->n - 1, w);
897  }
898 }
899 
900 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)
901 {
902  if (s->bps <= 8) {
903  s->llviddsp.add_median_pred(dst, src, diff, w, left, left_top);
904  } else {
905  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);
906  }
907 }
908 static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
909  AVPacket *avpkt)
910 {
911  const uint8_t *buf = avpkt->data;
912  int buf_size = avpkt->size;
913  HYuvContext *s = avctx->priv_data;
914  const int width = s->width;
915  const int width2 = s->width >> 1;
916  const int height = s->height;
917  int fake_ystride, fake_ustride, fake_vstride;
918  ThreadFrame frame = { .f = data };
919  AVFrame *const p = data;
920  int table_size = 0, ret;
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
1258 AVCodec ff_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(* add_left_pred_int16)(uint16_t *dst, const uint16_t *src, unsigned mask, ptrdiff_t w, unsigned left)
void(* bswap_buf)(uint32_t *dst, const uint32_t *src, int w)
Definition: bswapdsp.h:25
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:900
#define OP14bits(dst0, dst1, code)
Definition: huffyuvdec.c:692
#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:418
const char * s
Definition: avisynth_c.h:768
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
#define AV_NUM_DATA_POINTERS
Definition: frame.h:202
static void decode_plane_bitstream(HYuvContext *s, int width, int plane)
Definition: huffyuvdec.c:702
#define AV_PIX_FMT_YUVA422P9
Definition: pixfmt.h:412
static float alpha(float a)
#define READ_2PIX_PLANE(dst0, dst1, plane, OP)
Definition: huffyuvdec.c:687
This structure describes decoded (raw) audio or video data.
Definition: frame.h:201
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:101
#define AV_PIX_FMT_YUVA420P10
Definition: pixfmt.h:414
#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:415
int bitstream_bpp
Definition: huffyuv.h:63
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:67
misc image utilities
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:261
static int init_thread_copy(AVCodecContext *avctx)
Definition: tta.c:392
const char * g
Definition: vf_curves.c:112
Definition: vf_geq.c:47
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
int acc
Definition: yuv2rgb.c:546
static void decode_422_bitstream(HYuvContext *s, int count)
Definition: huffyuvdec.c:658
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:268
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:180
int size
Definition: avcodec.h:1680
const char * b
Definition: vf_curves.c:113
#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:781
#define VLC_BITS
Definition: asvdec.c:37
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1989
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:121
#define AV_PIX_FMT_YUV420P12
Definition: pixfmt.h:383
#define MAX_VLC_N
Definition: huffyuv.h:47
int context
Definition: huffyuv.h:77
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:222
#define READ_2PIX_PLANE16(dst0, dst1, plane)
Definition: huffyuvdec.c:696
unsigned int bitstream_buffer_size
Definition: huffyuv.h:88
#define src
Definition: vp8dsp.c:254
static int generate_joint_tables(HYuvContext *s)
Definition: huffyuvdec.c:118
AVCodec.
Definition: avcodec.h:3739
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:42
int height
Definition: huffyuv.h:75
#define READ_2PIX(dst0, dst1, plane1)
Definition: huffyuvdec.c:653
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:854
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
void(* add_bytes)(uint8_t *dst, uint8_t *src, ptrdiff_t w)
#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:102
uint8_t bits
Definition: crc.c:296
uint8_t
#define av_cold
Definition: attributes.h:82
static void decode_bgr_bitstream(HYuvContext *s, int count)
Definition: huffyuvdec.c:839
int bps
Definition: huffyuv.h:67
Multithreading support functions.
#define emms_c()
Definition: internal.h:54
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1876
#define AV_PIX_FMT_YUVA420P9
Definition: pixfmt.h:411
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:394
static AVFrame * frame
#define height
uint8_t * data
Definition: avcodec.h:1679
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:199
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:3157
#define AV_PIX_FMT_YUV422P12
Definition: pixfmt.h:384
#define AV_PIX_FMT_YUVA420P16
Definition: pixfmt.h:417
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
Definition: avcodec.h:904
#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:587
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
Definition: pixfmt.h:188
#define UPDATE_CACHE(name, gb)
Definition: get_bits.h:160
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
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
#define init_vlc(vlc, nb_bits, nb_codes,bits, bits_wrap, bits_size,codes, codes_wrap, codes_size,flags)
Definition: vlc.h:38
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:2447
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:179
const char * r
Definition: vf_curves.c:111
#define AV_PIX_FMT_YUVA444P16
Definition: pixfmt.h:419
int bgr32
Definition: huffyuv.h:66
uint16_t width
Definition: gdv.c:47
GLsizei GLsizei * length
Definition: opengl_enc.c:115
const char * name
Name of the codec implementation.
Definition: avcodec.h:3746
#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
GLsizei count
Definition: opengl_enc.c:109
#define CLOSE_READER(name, gb)
Definition: get_bits.h:131
huffyuv codec for libavcodec.
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
Definition: avcodec.h:1065
#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:66
#define ONLY_IF_THREADS_ENABLED(x)
Define a function with only the non-default version specified.
Definition: internal.h:218
#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:281
#define AV_CODEC_CAP_DRAW_HORIZ_BAND
Decoder can use draw_horiz_band callback.
Definition: avcodec.h:996
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, 'draw_horiz_band' is called by the libavcodec decoder to draw a horizontal band...
Definition: avcodec.h:2022
int decorrelate
Definition: huffyuv.h:62
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:398
alias for AV_PIX_FMT_YA8
Definition: pixfmt.h:157
#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:606
int width
picture width / height.
Definition: avcodec.h:1948
static const unsigned char classic_shift_chroma[classic_shift_chroma_table_size+AV_INPUT_BUFFER_PADDING_SIZE]
Definition: huffyuvdec.c:53
int ff_huffyuv_generate_bits_table(uint32_t *dst, const uint8_t *len_table, int n)
Definition: huffyuv.c:39
#define LAST_SKIP_BITS(name, gb, num)
Definition: get_bits.h:181
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:554
int n
Definition: avisynth_c.h:684
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:65
#define AV_PIX_FMT_YUVA444P10
Definition: pixfmt.h:416
uint8_t * temp[3]
Definition: huffyuv.h:80
#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:193
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:891
#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:232
LLVidDSPContext llviddsp
Definition: huffyuv.h:92
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
Definition: pixfmt.h:189
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.
main external API structure.
Definition: avcodec.h:1761
Definition: vf_geq.c:47
#define AV_PIX_FMT_RGB32
Definition: pixfmt.h:353
packed YUV 4:2:2, 16bpp, Y0 Cb Y1 Cr
Definition: pixfmt.h:63
#define OPEN_READER(name, gb)
Definition: get_bits.h:120
void * buf
Definition: avisynth_c.h:690
int interlaced
Definition: huffyuv.h:61
int extradata_size
Definition: avcodec.h:1877
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:68
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:306
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:425
Definition: vf_geq.c:47
static void decode_gray_bitstream(HYuvContext *s, int count)
Definition: huffyuvdec.c:763
#define AV_PIX_FMT_YUV420P9
Definition: pixfmt.h:376
#define classic_shift_luma_table_size
Definition: huffyuvdec.c:44
const VDPAUPixFmtMap * map
#define u(width,...)
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:215
Predictor predictor
Definition: huffyuv.h:58
static av_cold int decode_init(AVCodecContext *avctx)
Definition: huffyuvdec.c:291
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:62
Y , 8bpp.
Definition: pixfmt.h:70
Definition: huffyuv.h:52
if(ret< 0)
Definition: vf_mcdeint.c:279
#define BITS_LEFT(name, gb)
Definition: get_bits.h:173
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:229
#define AV_PIX_FMT_YUVA444P9
Definition: pixfmt.h:413
static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: huffyuvdec.c:908
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:69
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:777
void * priv_data
Definition: avcodec.h:1803
static av_always_inline int diff(const uint32_t a, const uint32_t b)
int(* add_left_pred)(uint8_t *dst, const uint8_t *src, ptrdiff_t w, int left)
void(* add_hfyu_left_pred_bgr32)(uint8_t *dst, const uint8_t *src, intptr_t w, uint8_t *left)
Definition: huffyuvdsp.h:45
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:882
void(* add_int16)(uint16_t *dst, const uint16_t *src, unsigned mask, int w)
Definition: huffyuvdsp.h:39
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)
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:100
#define av_always_inline
Definition: attributes.h:39
uint32_t bits[4][MAX_VLC_N]
Definition: huffyuv.h:84
#define LEFT
Definition: cdgraphics.c:168
This structure stores compressed data.
Definition: avcodec.h:1656
static void decorrelate(SnowContext *s, SubBand *b, IDWTELEM *src, int stride, int inverse, int use_median)
Definition: snowenc.c:1327
BswapDSPContext bdsp
Definition: huffyuv.h:89
void ff_free_vlc(VLC *vlc)
Definition: bitstream.c:354
static int read_len_table(uint8_t *dst, GetBitContext *gb, int n)
Definition: huffyuvdec.c:99
#define OP8bits(dst0, dst1, code)
Definition: huffyuvdec.c:651
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:1002
#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)
void(* add_median_pred)(uint8_t *dst, const uint8_t *top, const uint8_t *diff, ptrdiff_t w, int *left, int *left_top)
#define AV_PIX_FMT_0RGB32
Definition: pixfmt.h:357
av_cold void ff_huffyuvdsp_init(HuffYUVDSPContext *c, enum AVPixelFormat pix_fmt)
Definition: huffyuvdsp.c:83