FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
ffv1dec.c
Go to the documentation of this file.
1 /*
2  * FFV1 decoder
3  *
4  * Copyright (c) 2003-2013 Michael Niedermayer <michaelni@gmx.at>
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 /**
24  * @file
25  * FF Video Codec 1 (a lossless codec) decoder
26  */
27 
28 #include "libavutil/avassert.h"
29 #include "libavutil/crc.h"
30 #include "libavutil/opt.h"
31 #include "libavutil/imgutils.h"
32 #include "libavutil/pixdesc.h"
33 #include "libavutil/timer.h"
34 #include "avcodec.h"
35 #include "internal.h"
36 #include "get_bits.h"
37 #include "rangecoder.h"
38 #include "golomb.h"
39 #include "mathops.h"
40 #include "ffv1.h"
41 
43  int is_signed)
44 {
45  if (get_rac(c, state + 0))
46  return 0;
47  else {
48  int i, e, a;
49  e = 0;
50  while (get_rac(c, state + 1 + FFMIN(e, 9))) // 1..10
51  e++;
52 
53  a = 1;
54  for (i = e - 1; i >= 0; i--)
55  a += a + get_rac(c, state + 22 + FFMIN(i, 9)); // 22..31
56 
57  e = -(is_signed && get_rac(c, state + 11 + FFMIN(e, 10))); // 11..21
58  return (a ^ e) - e;
59  }
60 }
61 
62 static av_noinline int get_symbol(RangeCoder *c, uint8_t *state, int is_signed)
63 {
64  return get_symbol_inline(c, state, is_signed);
65 }
66 
67 static inline int get_vlc_symbol(GetBitContext *gb, VlcState *const state,
68  int bits)
69 {
70  int k, i, v, ret;
71 
72  i = state->count;
73  k = 0;
74  while (i < state->error_sum) { // FIXME: optimize
75  k++;
76  i += i;
77  }
78 
79  v = get_sr_golomb(gb, k, 12, bits);
80  ff_dlog(NULL, "v:%d bias:%d error:%d drift:%d count:%d k:%d",
81  v, state->bias, state->error_sum, state->drift, state->count, k);
82 
83 #if 0 // JPEG LS
84  if (k == 0 && 2 * state->drift <= -state->count)
85  v ^= (-1);
86 #else
87  v ^= ((2 * state->drift + state->count) >> 31);
88 #endif
89 
90  ret = fold(v + state->bias, bits);
91 
92  update_vlc_state(state, v);
93 
94  return ret;
95 }
96 
98  int16_t *sample[2],
99  int plane_index, int bits)
100 {
101  PlaneContext *const p = &s->plane[plane_index];
102  RangeCoder *const c = &s->c;
103  int x;
104  int run_count = 0;
105  int run_mode = 0;
106  int run_index = s->run_index;
107 
108  if (s->slice_coding_mode == 1) {
109  int i;
110  for (x = 0; x < w; x++) {
111  int v = 0;
112  for (i=0; i<bits; i++) {
113  uint8_t state = 128;
114  v += v + get_rac(c, &state);
115  }
116  sample[1][x] = v;
117  }
118  return;
119  }
120 
121  for (x = 0; x < w; x++) {
122  int diff, context, sign;
123 
124  context = get_context(p, sample[1] + x, sample[0] + x, sample[1] + x);
125  if (context < 0) {
126  context = -context;
127  sign = 1;
128  } else
129  sign = 0;
130 
131  av_assert2(context < p->context_count);
132 
133  if (s->ac) {
134  diff = get_symbol_inline(c, p->state[context], 1);
135  } else {
136  if (context == 0 && run_mode == 0)
137  run_mode = 1;
138 
139  if (run_mode) {
140  if (run_count == 0 && run_mode == 1) {
141  if (get_bits1(&s->gb)) {
142  run_count = 1 << ff_log2_run[run_index];
143  if (x + run_count <= w)
144  run_index++;
145  } else {
146  if (ff_log2_run[run_index])
147  run_count = get_bits(&s->gb, ff_log2_run[run_index]);
148  else
149  run_count = 0;
150  if (run_index)
151  run_index--;
152  run_mode = 2;
153  }
154  }
155  run_count--;
156  if (run_count < 0) {
157  run_mode = 0;
158  run_count = 0;
159  diff = get_vlc_symbol(&s->gb, &p->vlc_state[context],
160  bits);
161  if (diff >= 0)
162  diff++;
163  } else
164  diff = 0;
165  } else
166  diff = get_vlc_symbol(&s->gb, &p->vlc_state[context], bits);
167 
168  ff_dlog(s->avctx, "count:%d index:%d, mode:%d, x:%d pos:%d\n",
169  run_count, run_index, run_mode, x, get_bits_count(&s->gb));
170  }
171 
172  if (sign)
173  diff = -diff;
174 
175  sample[1][x] = av_mod_uintp2(predict(sample[1] + x, sample[0] + x) + diff, bits);
176  }
177  s->run_index = run_index;
178 }
179 
181  int w, int h, int stride, int plane_index)
182 {
183  int x, y;
184  int16_t *sample[2];
185  sample[0] = s->sample_buffer + 3;
186  sample[1] = s->sample_buffer + w + 6 + 3;
187 
188  s->run_index = 0;
189 
190  memset(s->sample_buffer, 0, 2 * (w + 6) * sizeof(*s->sample_buffer));
191 
192  for (y = 0; y < h; y++) {
193  int16_t *temp = sample[0]; // FIXME: try a normal buffer
194 
195  sample[0] = sample[1];
196  sample[1] = temp;
197 
198  sample[1][-1] = sample[0][0];
199  sample[0][w] = sample[0][w - 1];
200 
201 // { START_TIMER
202  if (s->avctx->bits_per_raw_sample <= 8) {
203  decode_line(s, w, sample, plane_index, 8);
204  for (x = 0; x < w; x++)
205  src[x + stride * y] = sample[1][x];
206  } else {
207  decode_line(s, w, sample, plane_index, s->avctx->bits_per_raw_sample);
208  if (s->packed_at_lsb) {
209  for (x = 0; x < w; x++) {
210  ((uint16_t*)(src + stride*y))[x] = sample[1][x];
211  }
212  } else {
213  for (x = 0; x < w; x++) {
214  ((uint16_t*)(src + stride*y))[x] = sample[1][x] << (16 - s->avctx->bits_per_raw_sample);
215  }
216  }
217  }
218 // STOP_TIMER("decode-line") }
219  }
220 }
221 
222 static void decode_rgb_frame(FFV1Context *s, uint8_t *src[3], int w, int h, int stride[3])
223 {
224  int x, y, p;
225  int16_t *sample[4][2];
226  int lbd = s->avctx->bits_per_raw_sample <= 8;
227  int bits = s->avctx->bits_per_raw_sample > 0 ? s->avctx->bits_per_raw_sample : 8;
228  int offset = 1 << bits;
229 
230  for (x = 0; x < 4; x++) {
231  sample[x][0] = s->sample_buffer + x * 2 * (w + 6) + 3;
232  sample[x][1] = s->sample_buffer + (x * 2 + 1) * (w + 6) + 3;
233  }
234 
235  s->run_index = 0;
236 
237  memset(s->sample_buffer, 0, 8 * (w + 6) * sizeof(*s->sample_buffer));
238 
239  for (y = 0; y < h; y++) {
240  for (p = 0; p < 3 + s->transparency; p++) {
241  int16_t *temp = sample[p][0]; // FIXME: try a normal buffer
242 
243  sample[p][0] = sample[p][1];
244  sample[p][1] = temp;
245 
246  sample[p][1][-1]= sample[p][0][0 ];
247  sample[p][0][ w]= sample[p][0][w-1];
248  if (lbd && s->slice_coding_mode == 0)
249  decode_line(s, w, sample[p], (p + 1)/2, 9);
250  else
251  decode_line(s, w, sample[p], (p + 1)/2, bits + (s->slice_coding_mode != 1));
252  }
253  for (x = 0; x < w; x++) {
254  int g = sample[0][1][x];
255  int b = sample[1][1][x];
256  int r = sample[2][1][x];
257  int a = sample[3][1][x];
258 
259  if (s->slice_coding_mode != 1) {
260  b -= offset;
261  r -= offset;
262  g -= (b * s->slice_rct_by_coef + r * s->slice_rct_ry_coef) >> 2;
263  b += g;
264  r += g;
265  }
266 
267  if (lbd)
268  *((uint32_t*)(src[0] + x*4 + stride[0]*y)) = b + (g<<8) + (r<<16) + (a<<24);
269  else {
270  *((uint16_t*)(src[0] + x*2 + stride[0]*y)) = b;
271  *((uint16_t*)(src[1] + x*2 + stride[1]*y)) = g;
272  *((uint16_t*)(src[2] + x*2 + stride[2]*y)) = r;
273  }
274  }
275  }
276 }
277 
279 {
280  RangeCoder *c = &fs->c;
282  unsigned ps, i, context_count;
283  memset(state, 128, sizeof(state));
284 
285  av_assert0(f->version > 2);
286 
287  fs->slice_x = get_symbol(c, state, 0) * f->width ;
288  fs->slice_y = get_symbol(c, state, 0) * f->height;
289  fs->slice_width = (get_symbol(c, state, 0) + 1) * f->width + fs->slice_x;
290  fs->slice_height = (get_symbol(c, state, 0) + 1) * f->height + fs->slice_y;
291 
292  fs->slice_x /= f->num_h_slices;
293  fs->slice_y /= f->num_v_slices;
294  fs->slice_width = fs->slice_width /f->num_h_slices - fs->slice_x;
295  fs->slice_height = fs->slice_height/f->num_v_slices - fs->slice_y;
296  if ((unsigned)fs->slice_width > f->width || (unsigned)fs->slice_height > f->height)
297  return -1;
298  if ( (unsigned)fs->slice_x + (uint64_t)fs->slice_width > f->width
299  || (unsigned)fs->slice_y + (uint64_t)fs->slice_height > f->height)
300  return -1;
301 
302  for (i = 0; i < f->plane_count; i++) {
303  PlaneContext * const p = &fs->plane[i];
304  int idx = get_symbol(c, state, 0);
305  if (idx > (unsigned)f->quant_table_count) {
306  av_log(f->avctx, AV_LOG_ERROR, "quant_table_index out of range\n");
307  return -1;
308  }
309  p->quant_table_index = idx;
310  memcpy(p->quant_table, f->quant_tables[idx], sizeof(p->quant_table));
311  context_count = f->context_count[idx];
312 
313  if (p->context_count < context_count) {
314  av_freep(&p->state);
315  av_freep(&p->vlc_state);
316  }
318  }
319 
320  ps = get_symbol(c, state, 0);
321  if (ps == 1) {
322  f->cur->interlaced_frame = 1;
323  f->cur->top_field_first = 1;
324  } else if (ps == 2) {
325  f->cur->interlaced_frame = 1;
326  f->cur->top_field_first = 0;
327  } else if (ps == 3) {
328  f->cur->interlaced_frame = 0;
329  }
330  f->cur->sample_aspect_ratio.num = get_symbol(c, state, 0);
331  f->cur->sample_aspect_ratio.den = get_symbol(c, state, 0);
332 
333  if (av_image_check_sar(f->width, f->height,
334  f->cur->sample_aspect_ratio) < 0) {
335  av_log(f->avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
338  f->cur->sample_aspect_ratio = (AVRational){ 0, 1 };
339  }
340 
341  if (fs->version > 3) {
342  fs->slice_reset_contexts = get_rac(c, state);
343  fs->slice_coding_mode = get_symbol(c, state, 0);
344  if (fs->slice_coding_mode != 1) {
345  fs->slice_rct_by_coef = get_symbol(c, state, 0);
346  fs->slice_rct_ry_coef = get_symbol(c, state, 0);
347  if ((uint64_t)fs->slice_rct_by_coef + (uint64_t)fs->slice_rct_ry_coef > 4) {
348  av_log(f->avctx, AV_LOG_ERROR, "slice_rct_y_coef out of range\n");
349  return AVERROR_INVALIDDATA;
350  }
351  }
352  }
353 
354  return 0;
355 }
356 
357 static int decode_slice(AVCodecContext *c, void *arg)
358 {
359  FFV1Context *fs = *(void **)arg;
360  FFV1Context *f = fs->avctx->priv_data;
361  int width, height, x, y, ret;
362  const int ps = av_pix_fmt_desc_get(c->pix_fmt)->comp[0].step_minus1 + 1;
363  AVFrame * const p = f->cur;
364  int i, si;
365 
366  for( si=0; fs != f->slice_context[si]; si ++)
367  ;
368 
369  if(f->fsrc && !p->key_frame)
371 
372  if(f->fsrc && !p->key_frame) {
373  FFV1Context *fssrc = f->fsrc->slice_context[si];
374  FFV1Context *fsdst = f->slice_context[si];
375  av_assert1(fsdst->plane_count == fssrc->plane_count);
376  av_assert1(fsdst == fs);
377 
378  if (!p->key_frame)
379  fsdst->slice_damaged |= fssrc->slice_damaged;
380 
381  for (i = 0; i < f->plane_count; i++) {
382  PlaneContext *psrc = &fssrc->plane[i];
383  PlaneContext *pdst = &fsdst->plane[i];
384 
385  av_free(pdst->state);
386  av_free(pdst->vlc_state);
387  memcpy(pdst, psrc, sizeof(*pdst));
388  pdst->state = NULL;
389  pdst->vlc_state = NULL;
390 
391  if (fssrc->ac) {
393  memcpy(pdst->state, psrc->state, CONTEXT_SIZE * psrc->context_count);
394  } else {
395  pdst->vlc_state = av_malloc_array(sizeof(*pdst->vlc_state), psrc->context_count);
396  memcpy(pdst->vlc_state, psrc->vlc_state, sizeof(*pdst->vlc_state) * psrc->context_count);
397  }
398  }
399  }
400 
401  fs->slice_rct_by_coef = 1;
402  fs->slice_rct_ry_coef = 1;
403 
404  if (f->version > 2) {
405  if (ffv1_init_slice_state(f, fs) < 0)
406  return AVERROR(ENOMEM);
407  if (decode_slice_header(f, fs) < 0) {
408  fs->slice_damaged = 1;
409  return AVERROR_INVALIDDATA;
410  }
411  }
412  if ((ret = ffv1_init_slice_state(f, fs)) < 0)
413  return ret;
414  if (f->cur->key_frame || fs->slice_reset_contexts)
415  ffv1_clear_slice_state(f, fs);
416 
417  width = fs->slice_width;
418  height = fs->slice_height;
419  x = fs->slice_x;
420  y = fs->slice_y;
421 
422  if (!fs->ac) {
423  if (f->version == 3 && f->micro_version > 1 || f->version > 3)
424  get_rac(&fs->c, (uint8_t[]) { 129 });
425  fs->ac_byte_count = f->version > 2 || (!x && !y) ? fs->c.bytestream - fs->c.bytestream_start - 1 : 0;
426  init_get_bits(&fs->gb,
427  fs->c.bytestream_start + fs->ac_byte_count,
428  (fs->c.bytestream_end - fs->c.bytestream_start - fs->ac_byte_count) * 8);
429  }
430 
431  av_assert1(width && height);
432  if (f->colorspace == 0) {
433  const int chroma_width = FF_CEIL_RSHIFT(width, f->chroma_h_shift);
434  const int chroma_height = FF_CEIL_RSHIFT(height, f->chroma_v_shift);
435  const int cx = x >> f->chroma_h_shift;
436  const int cy = y >> f->chroma_v_shift;
437  decode_plane(fs, p->data[0] + ps*x + y*p->linesize[0], width, height, p->linesize[0], 0);
438 
439  if (f->chroma_planes) {
440  decode_plane(fs, p->data[1] + ps*cx+cy*p->linesize[1], chroma_width, chroma_height, p->linesize[1], 1);
441  decode_plane(fs, p->data[2] + ps*cx+cy*p->linesize[2], chroma_width, chroma_height, p->linesize[2], 1);
442  }
443  if (fs->transparency)
444  decode_plane(fs, p->data[3] + ps*x + y*p->linesize[3], width, height, p->linesize[3], (f->version >= 4 && !f->chroma_planes) ? 1 : 2);
445  } else {
446  uint8_t *planes[3] = { p->data[0] + ps * x + y * p->linesize[0],
447  p->data[1] + ps * x + y * p->linesize[1],
448  p->data[2] + ps * x + y * p->linesize[2] };
449  decode_rgb_frame(fs, planes, width, height, p->linesize);
450  }
451  if (fs->ac && f->version > 2) {
452  int v;
453  get_rac(&fs->c, (uint8_t[]) { 129 });
454  v = fs->c.bytestream_end - fs->c.bytestream - 2 - 5*f->ec;
455  if (v) {
456  av_log(f->avctx, AV_LOG_ERROR, "bytestream end mismatching by %d\n", v);
457  fs->slice_damaged = 1;
458  }
459  }
460 
461  emms_c();
462 
463  ff_thread_report_progress(&f->picture, si, 0);
464 
465  return 0;
466 }
467 
468 static int read_quant_table(RangeCoder *c, int16_t *quant_table, int scale)
469 {
470  int v;
471  int i = 0;
473 
474  memset(state, 128, sizeof(state));
475 
476  for (v = 0; i < 128; v++) {
477  unsigned len = get_symbol(c, state, 0) + 1;
478 
479  if (len > 128 - i || !len)
480  return AVERROR_INVALIDDATA;
481 
482  while (len--) {
483  quant_table[i] = scale * v;
484  i++;
485  }
486  }
487 
488  for (i = 1; i < 128; i++)
489  quant_table[256 - i] = -quant_table[i];
490  quant_table[128] = -quant_table[127];
491 
492  return 2 * v - 1;
493 }
494 
496  int16_t quant_table[MAX_CONTEXT_INPUTS][256])
497 {
498  int i;
499  int context_count = 1;
500 
501  for (i = 0; i < 5; i++) {
502  context_count *= read_quant_table(c, quant_table[i], context_count);
503  if (context_count > 32768U) {
504  return AVERROR_INVALIDDATA;
505  }
506  }
507  return (context_count + 1) / 2;
508 }
509 
511 {
512  RangeCoder *const c = &f->c;
514  int i, j, k, ret;
515  uint8_t state2[32][CONTEXT_SIZE];
516 
517  memset(state2, 128, sizeof(state2));
518  memset(state, 128, sizeof(state));
519 
521  ff_build_rac_states(c, 0.05 * (1LL << 32), 256 - 8);
522 
523  f->version = get_symbol(c, state, 0);
524  if (f->version < 2) {
525  av_log(f->avctx, AV_LOG_ERROR, "Invalid version in global header\n");
526  return AVERROR_INVALIDDATA;
527  }
528  if (f->version > 2) {
529  c->bytestream_end -= 4;
530  f->micro_version = get_symbol(c, state, 0);
531  if (f->micro_version < 0)
532  return AVERROR_INVALIDDATA;
533  }
534  f->ac = f->avctx->coder_type = get_symbol(c, state, 0);
535  if (f->ac > 1) {
536  for (i = 1; i < 256; i++)
537  f->state_transition[i] = get_symbol(c, state, 1) + c->one_state[i];
538  }
539 
540  f->colorspace = get_symbol(c, state, 0); //YUV cs type
541  f->avctx->bits_per_raw_sample = get_symbol(c, state, 0);
542  f->chroma_planes = get_rac(c, state);
543  f->chroma_h_shift = get_symbol(c, state, 0);
544  f->chroma_v_shift = get_symbol(c, state, 0);
545  f->transparency = get_rac(c, state);
546  f->plane_count = 1 + (f->chroma_planes || f->version<4) + f->transparency;
547  f->num_h_slices = 1 + get_symbol(c, state, 0);
548  f->num_v_slices = 1 + get_symbol(c, state, 0);
549 
550  if (f->chroma_h_shift > 4U || f->chroma_v_shift > 4U) {
551  av_log(f->avctx, AV_LOG_ERROR, "chroma shift parameters %d %d are invalid\n",
553  return AVERROR_INVALIDDATA;
554  }
555 
556  if (f->num_h_slices > (unsigned)f->width || !f->num_h_slices ||
557  f->num_v_slices > (unsigned)f->height || !f->num_v_slices
558  ) {
559  av_log(f->avctx, AV_LOG_ERROR, "slice count invalid\n");
560  return AVERROR_INVALIDDATA;
561  }
562 
563  f->quant_table_count = get_symbol(c, state, 0);
564  if (f->quant_table_count > (unsigned)MAX_QUANT_TABLES)
565  return AVERROR_INVALIDDATA;
566 
567  for (i = 0; i < f->quant_table_count; i++) {
568  f->context_count[i] = read_quant_tables(c, f->quant_tables[i]);
569  if (f->context_count[i] < 0) {
570  av_log(f->avctx, AV_LOG_ERROR, "read_quant_table error\n");
571  return AVERROR_INVALIDDATA;
572  }
573  }
574  if ((ret = ffv1_allocate_initial_states(f)) < 0)
575  return ret;
576 
577  for (i = 0; i < f->quant_table_count; i++)
578  if (get_rac(c, state)) {
579  for (j = 0; j < f->context_count[i]; j++)
580  for (k = 0; k < CONTEXT_SIZE; k++) {
581  int pred = j ? f->initial_states[i][j - 1][k] : 128;
582  f->initial_states[i][j][k] =
583  (pred + get_symbol(c, state2[k], 1)) & 0xFF;
584  }
585  }
586 
587  if (f->version > 2) {
588  f->ec = get_symbol(c, state, 0);
589  if (f->micro_version > 2)
590  f->intra = get_symbol(c, state, 0);
591  }
592 
593  if (f->version > 2) {
594  unsigned v;
597  if (v) {
598  av_log(f->avctx, AV_LOG_ERROR, "CRC mismatch %X!\n", v);
599  return AVERROR_INVALIDDATA;
600  }
601  }
602 
603  if (f->avctx->debug & FF_DEBUG_PICT_INFO)
605  "global: ver:%d.%d, coder:%d, colorspace: %d bpr:%d chroma:%d(%d:%d), alpha:%d slices:%dx%d qtabs:%d ec:%d intra:%d\n",
606  f->version, f->micro_version,
607  f->ac,
608  f->colorspace,
611  f->transparency,
612  f->num_h_slices, f->num_v_slices,
614  f->ec,
615  f->intra
616  );
617  return 0;
618 }
619 
620 static int read_header(FFV1Context *f)
621 {
623  int i, j, context_count = -1; //-1 to avoid warning
624  RangeCoder *const c = &f->slice_context[0]->c;
625 
626  memset(state, 128, sizeof(state));
627 
628  if (f->version < 2) {
630  unsigned v= get_symbol(c, state, 0);
631  if (v >= 2) {
632  av_log(f->avctx, AV_LOG_ERROR, "invalid version %d in ver01 header\n", v);
633  return AVERROR_INVALIDDATA;
634  }
635  f->version = v;
636  f->ac = f->avctx->coder_type = get_symbol(c, state, 0);
637  if (f->ac > 1) {
638  for (i = 1; i < 256; i++)
639  f->state_transition[i] = get_symbol(c, state, 1) + c->one_state[i];
640  }
641 
642  colorspace = get_symbol(c, state, 0); //YUV cs type
643  bits_per_raw_sample = f->version > 0 ? get_symbol(c, state, 0) : f->avctx->bits_per_raw_sample;
644  chroma_planes = get_rac(c, state);
645  chroma_h_shift = get_symbol(c, state, 0);
646  chroma_v_shift = get_symbol(c, state, 0);
647  transparency = get_rac(c, state);
648 
649  if (f->plane_count) {
650  if (colorspace != f->colorspace ||
651  bits_per_raw_sample != f->avctx->bits_per_raw_sample ||
652  chroma_planes != f->chroma_planes ||
653  chroma_h_shift != f->chroma_h_shift ||
654  chroma_v_shift != f->chroma_v_shift ||
655  transparency != f->transparency) {
656  av_log(f->avctx, AV_LOG_ERROR, "Invalid change of global parameters\n");
657  return AVERROR_INVALIDDATA;
658  }
659  }
660 
661  if (chroma_h_shift > 4U || chroma_v_shift > 4U) {
662  av_log(f->avctx, AV_LOG_ERROR, "chroma shift parameters %d %d are invalid\n",
663  chroma_h_shift, chroma_v_shift);
664  return AVERROR_INVALIDDATA;
665  }
666 
667  f->colorspace = colorspace;
673 
674  f->plane_count = 2 + f->transparency;
675  }
676 
677  if (f->colorspace == 0) {
678  if (f->avctx->skip_alpha) f->transparency = 0;
679  if (!f->transparency && !f->chroma_planes) {
680  if (f->avctx->bits_per_raw_sample <= 8)
682  else
684  } else if (f->avctx->bits_per_raw_sample<=8 && !f->transparency) {
685  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
686  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUV444P; break;
687  case 0x01: f->avctx->pix_fmt = AV_PIX_FMT_YUV440P; break;
688  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUV422P; break;
689  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUV420P; break;
690  case 0x20: f->avctx->pix_fmt = AV_PIX_FMT_YUV411P; break;
691  case 0x22: f->avctx->pix_fmt = AV_PIX_FMT_YUV410P; break;
692  }
693  } else if (f->avctx->bits_per_raw_sample <= 8 && f->transparency) {
694  switch(16*f->chroma_h_shift + f->chroma_v_shift) {
695  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUVA444P; break;
696  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUVA422P; break;
697  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUVA420P; break;
698  }
699  } else if (f->avctx->bits_per_raw_sample == 9 && !f->transparency) {
700  f->packed_at_lsb = 1;
701  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
702  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUV444P9; break;
703  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUV422P9; break;
704  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUV420P9; break;
705  }
706  } else if (f->avctx->bits_per_raw_sample == 9 && f->transparency) {
707  f->packed_at_lsb = 1;
708  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
709  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUVA444P9; break;
710  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUVA422P9; break;
711  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUVA420P9; break;
712  }
713  } else if (f->avctx->bits_per_raw_sample == 10 && !f->transparency) {
714  f->packed_at_lsb = 1;
715  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
716  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUV444P10; break;
717  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUV422P10; break;
718  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUV420P10; break;
719  }
720  } else if (f->avctx->bits_per_raw_sample == 10 && f->transparency) {
721  f->packed_at_lsb = 1;
722  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
723  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUVA444P10; break;
724  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUVA422P10; break;
725  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUVA420P10; break;
726  }
727  } else if (f->avctx->bits_per_raw_sample == 16 && !f->transparency){
728  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
729  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUV444P16; break;
730  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUV422P16; break;
731  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUV420P16; break;
732  }
733  } else if (f->avctx->bits_per_raw_sample == 16 && f->transparency){
734  switch(16 * f->chroma_h_shift + f->chroma_v_shift) {
735  case 0x00: f->avctx->pix_fmt = AV_PIX_FMT_YUVA444P16; break;
736  case 0x10: f->avctx->pix_fmt = AV_PIX_FMT_YUVA422P16; break;
737  case 0x11: f->avctx->pix_fmt = AV_PIX_FMT_YUVA420P16; break;
738  }
739  }
740  } else if (f->colorspace == 1) {
741  if (f->chroma_h_shift || f->chroma_v_shift) {
743  "chroma subsampling not supported in this colorspace\n");
744  return AVERROR(ENOSYS);
745  }
746  if ( f->avctx->bits_per_raw_sample == 9)
748  else if (f->avctx->bits_per_raw_sample == 10)
750  else if (f->avctx->bits_per_raw_sample == 12)
752  else if (f->avctx->bits_per_raw_sample == 14)
754  else
756  else f->avctx->pix_fmt = AV_PIX_FMT_0RGB32;
757  } else {
758  av_log(f->avctx, AV_LOG_ERROR, "colorspace not supported\n");
759  return AVERROR(ENOSYS);
760  }
761  if (f->avctx->pix_fmt == AV_PIX_FMT_NONE) {
762  av_log(f->avctx, AV_LOG_ERROR, "format not supported\n");
763  return AVERROR(ENOSYS);
764  }
765 
766  ff_dlog(f->avctx, "%d %d %d\n",
768  if (f->version < 2) {
769  context_count = read_quant_tables(c, f->quant_table);
770  if (context_count < 0) {
771  av_log(f->avctx, AV_LOG_ERROR, "read_quant_table error\n");
772  return AVERROR_INVALIDDATA;
773  }
774  } else if (f->version < 3) {
775  f->slice_count = get_symbol(c, state, 0);
776  } else {
777  const uint8_t *p = c->bytestream_end;
778  for (f->slice_count = 0;
779  f->slice_count < MAX_SLICES && 3 < p - c->bytestream_start;
780  f->slice_count++) {
781  int trailer = 3 + 5*!!f->ec;
782  int size = AV_RB24(p-trailer);
783  if (size + trailer > p - c->bytestream_start)
784  break;
785  p -= size + trailer;
786  }
787  }
788  if (f->slice_count > (unsigned)MAX_SLICES || f->slice_count <= 0) {
789  av_log(f->avctx, AV_LOG_ERROR, "slice count %d is invalid\n", f->slice_count);
790  return AVERROR_INVALIDDATA;
791  }
792 
793  for (j = 0; j < f->slice_count; j++) {
794  FFV1Context *fs = f->slice_context[j];
795  fs->ac = f->ac;
796  fs->packed_at_lsb = f->packed_at_lsb;
797 
798  fs->slice_damaged = 0;
799 
800  if (f->version == 2) {
801  fs->slice_x = get_symbol(c, state, 0) * f->width ;
802  fs->slice_y = get_symbol(c, state, 0) * f->height;
803  fs->slice_width = (get_symbol(c, state, 0) + 1) * f->width + fs->slice_x;
804  fs->slice_height = (get_symbol(c, state, 0) + 1) * f->height + fs->slice_y;
805 
806  fs->slice_x /= f->num_h_slices;
807  fs->slice_y /= f->num_v_slices;
808  fs->slice_width = fs->slice_width / f->num_h_slices - fs->slice_x;
809  fs->slice_height = fs->slice_height / f->num_v_slices - fs->slice_y;
810  if ((unsigned)fs->slice_width > f->width ||
811  (unsigned)fs->slice_height > f->height)
812  return AVERROR_INVALIDDATA;
813  if ( (unsigned)fs->slice_x + (uint64_t)fs->slice_width > f->width
814  || (unsigned)fs->slice_y + (uint64_t)fs->slice_height > f->height)
815  return AVERROR_INVALIDDATA;
816  }
817 
818  for (i = 0; i < f->plane_count; i++) {
819  PlaneContext *const p = &fs->plane[i];
820 
821  if (f->version == 2) {
822  int idx = get_symbol(c, state, 0);
823  if (idx > (unsigned)f->quant_table_count) {
825  "quant_table_index out of range\n");
826  return AVERROR_INVALIDDATA;
827  }
828  p->quant_table_index = idx;
829  memcpy(p->quant_table, f->quant_tables[idx],
830  sizeof(p->quant_table));
831  context_count = f->context_count[idx];
832  } else {
833  memcpy(p->quant_table, f->quant_table, sizeof(p->quant_table));
834  }
835 
836  if (f->version <= 2) {
837  av_assert0(context_count >= 0);
838  if (p->context_count < context_count) {
839  av_freep(&p->state);
840  av_freep(&p->vlc_state);
841  }
843  }
844  }
845  }
846  return 0;
847 }
848 
850 {
851  FFV1Context *f = avctx->priv_data;
852  int ret;
853 
854  if ((ret = ffv1_common_init(avctx)) < 0)
855  return ret;
856 
857  if (avctx->extradata && (ret = read_extra_header(f)) < 0)
858  return ret;
859 
860  if ((ret = ffv1_init_slice_contexts(f)) < 0)
861  return ret;
862 
863  avctx->internal->allocate_progress = 1;
864 
865  return 0;
866 }
867 
868 static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
869 {
870  uint8_t *buf = avpkt->data;
871  int buf_size = avpkt->size;
872  FFV1Context *f = avctx->priv_data;
873  RangeCoder *const c = &f->slice_context[0]->c;
874  int i, ret;
875  uint8_t keystate = 128;
876  uint8_t *buf_p;
877  AVFrame *p;
878 
879  if (f->last_picture.f)
882 
883  f->cur = p = f->picture.f;
884 
885  if (f->version < 3 && avctx->field_order > AV_FIELD_PROGRESSIVE) {
886  /* we have interlaced material flagged in container */
887  p->interlaced_frame = 1;
888  if (avctx->field_order == AV_FIELD_TT || avctx->field_order == AV_FIELD_TB)
889  p->top_field_first = 1;
890  }
891 
892  f->avctx = avctx;
893  ff_init_range_decoder(c, buf, buf_size);
894  ff_build_rac_states(c, 0.05 * (1LL << 32), 256 - 8);
895 
896  p->pict_type = AV_PICTURE_TYPE_I; //FIXME I vs. P
897  if (get_rac(c, &keystate)) {
898  p->key_frame = 1;
899  f->key_frame_ok = 0;
900  if ((ret = read_header(f)) < 0)
901  return ret;
902  f->key_frame_ok = 1;
903  } else {
904  if (!f->key_frame_ok) {
905  av_log(avctx, AV_LOG_ERROR,
906  "Cannot decode non-keyframe without valid keyframe\n");
907  return AVERROR_INVALIDDATA;
908  }
909  p->key_frame = 0;
910  }
911 
912  if ((ret = ff_thread_get_buffer(avctx, &f->picture, AV_GET_BUFFER_FLAG_REF)) < 0)
913  return ret;
914 
915  if (avctx->debug & FF_DEBUG_PICT_INFO)
916  av_log(avctx, AV_LOG_DEBUG, "ver:%d keyframe:%d coder:%d ec:%d slices:%d bps:%d\n",
917  f->version, p->key_frame, f->ac, f->ec, f->slice_count, f->avctx->bits_per_raw_sample);
918 
919  ff_thread_finish_setup(avctx);
920 
921  buf_p = buf + buf_size;
922  for (i = f->slice_count - 1; i >= 0; i--) {
923  FFV1Context *fs = f->slice_context[i];
924  int trailer = 3 + 5*!!f->ec;
925  int v;
926 
927  if (i || f->version > 2) v = AV_RB24(buf_p-trailer) + trailer;
928  else v = buf_p - c->bytestream_start;
929  if (buf_p - c->bytestream_start < v) {
930  av_log(avctx, AV_LOG_ERROR, "Slice pointer chain broken\n");
931  return AVERROR_INVALIDDATA;
932  }
933  buf_p -= v;
934 
935  if (f->ec) {
936  unsigned crc = av_crc(av_crc_get_table(AV_CRC_32_IEEE), 0, buf_p, v);
937  if (crc) {
938  int64_t ts = avpkt->pts != AV_NOPTS_VALUE ? avpkt->pts : avpkt->dts;
939  av_log(f->avctx, AV_LOG_ERROR, "CRC mismatch %X!", crc);
940  if (ts != AV_NOPTS_VALUE && avctx->pkt_timebase.num) {
941  av_log(f->avctx, AV_LOG_ERROR, "at %f seconds\n", ts*av_q2d(avctx->pkt_timebase));
942  } else if (ts != AV_NOPTS_VALUE) {
943  av_log(f->avctx, AV_LOG_ERROR, "at %"PRId64"\n", ts);
944  } else {
945  av_log(f->avctx, AV_LOG_ERROR, "\n");
946  }
947  fs->slice_damaged = 1;
948  }
949  }
950 
951  if (i) {
952  ff_init_range_decoder(&fs->c, buf_p, v);
953  } else
954  fs->c.bytestream_end = buf_p + v;
955 
956  fs->avctx = avctx;
957  fs->cur = p;
958  }
959 
960  avctx->execute(avctx,
961  decode_slice,
962  &f->slice_context[0],
963  NULL,
964  f->slice_count,
965  sizeof(void*));
966 
967  for (i = f->slice_count - 1; i >= 0; i--) {
968  FFV1Context *fs = f->slice_context[i];
969  int j;
970  if (fs->slice_damaged && f->last_picture.f->data[0]) {
971  const uint8_t *src[4];
972  uint8_t *dst[4];
973  ff_thread_await_progress(&f->last_picture, INT_MAX, 0);
974  for (j = 0; j < 4; j++) {
975  int sh = (j == 1 || j == 2) ? f->chroma_h_shift : 0;
976  int sv = (j == 1 || j == 2) ? f->chroma_v_shift : 0;
977  dst[j] = p->data[j] + p->linesize[j] *
978  (fs->slice_y >> sv) + (fs->slice_x >> sh);
979  src[j] = f->last_picture.f->data[j] + f->last_picture.f->linesize[j] *
980  (fs->slice_y >> sv) + (fs->slice_x >> sh);
981  }
982  av_image_copy(dst, p->linesize, src,
983  f->last_picture.f->linesize,
984  avctx->pix_fmt,
985  fs->slice_width,
986  fs->slice_height);
987  }
988  }
989  ff_thread_report_progress(&f->picture, INT_MAX, 0);
990 
991  f->picture_number++;
992 
993  if (f->last_picture.f)
995  f->cur = NULL;
996  if ((ret = av_frame_ref(data, f->picture.f)) < 0)
997  return ret;
998 
999  *got_frame = 1;
1000 
1001  return buf_size;
1002 }
1003 
1005 {
1006  FFV1Context *f = avctx->priv_data;
1007  int i, ret;
1008 
1009  f->picture.f = NULL;
1010  f->last_picture.f = NULL;
1011  f->sample_buffer = NULL;
1012  f->slice_count = 0;
1013 
1014  for (i = 0; i < f->quant_table_count; i++) {
1015  av_assert0(f->version > 1);
1016  f->initial_states[i] = av_memdup(f->initial_states[i],
1017  f->context_count[i] * sizeof(*f->initial_states[i]));
1018  }
1019 
1020  f->picture.f = av_frame_alloc();
1021  f->last_picture.f = av_frame_alloc();
1022 
1023  if ((ret = ffv1_init_slice_contexts(f)) < 0)
1024  return ret;
1025 
1026  return 0;
1027 }
1028 
1029 static void copy_fields(FFV1Context *fsdst, FFV1Context *fssrc, FFV1Context *fsrc)
1030 {
1031  fsdst->version = fsrc->version;
1032  fsdst->micro_version = fsrc->micro_version;
1033  fsdst->chroma_planes = fsrc->chroma_planes;
1034  fsdst->chroma_h_shift = fsrc->chroma_h_shift;
1035  fsdst->chroma_v_shift = fsrc->chroma_v_shift;
1036  fsdst->transparency = fsrc->transparency;
1037  fsdst->plane_count = fsrc->plane_count;
1038  fsdst->ac = fsrc->ac;
1039  fsdst->colorspace = fsrc->colorspace;
1040 
1041  fsdst->ec = fsrc->ec;
1042  fsdst->intra = fsrc->intra;
1043  fsdst->slice_damaged = fssrc->slice_damaged;
1044  fsdst->key_frame_ok = fsrc->key_frame_ok;
1045 
1047  fsdst->packed_at_lsb = fsrc->packed_at_lsb;
1048  fsdst->slice_count = fsrc->slice_count;
1049  if (fsrc->version<3){
1050  fsdst->slice_x = fssrc->slice_x;
1051  fsdst->slice_y = fssrc->slice_y;
1052  fsdst->slice_width = fssrc->slice_width;
1053  fsdst->slice_height = fssrc->slice_height;
1054  }
1055 }
1056 
1058 {
1059  FFV1Context *fsrc = src->priv_data;
1060  FFV1Context *fdst = dst->priv_data;
1061  int i, ret;
1062 
1063  if (dst == src)
1064  return 0;
1065 
1066  {
1070  memcpy(initial_states, fdst->initial_states, sizeof(fdst->initial_states));
1071  memcpy(slice_context, fdst->slice_context , sizeof(fdst->slice_context));
1072 
1073  memcpy(fdst, fsrc, sizeof(*fdst));
1074  memcpy(fdst->initial_states, initial_states, sizeof(fdst->initial_states));
1075  memcpy(fdst->slice_context, slice_context , sizeof(fdst->slice_context));
1076  fdst->picture = picture;
1077  fdst->last_picture = last_picture;
1078  for (i = 0; i<fdst->num_h_slices * fdst->num_v_slices; i++) {
1079  FFV1Context *fssrc = fsrc->slice_context[i];
1080  FFV1Context *fsdst = fdst->slice_context[i];
1081  copy_fields(fsdst, fssrc, fsrc);
1082  }
1083  av_assert0(!fdst->plane[0].state);
1084  av_assert0(!fdst->sample_buffer);
1085  }
1086 
1087  av_assert1(fdst->slice_count == fsrc->slice_count);
1088 
1089 
1090  ff_thread_release_buffer(dst, &fdst->picture);
1091  if (fsrc->picture.f->data[0]) {
1092  if ((ret = ff_thread_ref_frame(&fdst->picture, &fsrc->picture)) < 0)
1093  return ret;
1094  }
1095 
1096  fdst->fsrc = fsrc;
1097 
1098  return 0;
1099 }
1100 
1102  .name = "ffv1",
1103  .long_name = NULL_IF_CONFIG_SMALL("FFmpeg video codec #1"),
1104  .type = AVMEDIA_TYPE_VIDEO,
1105  .id = AV_CODEC_ID_FFV1,
1106  .priv_data_size = sizeof(FFV1Context),
1107  .init = decode_init,
1108  .close = ffv1_close,
1109  .decode = decode_frame,
1111  .update_thread_context = ONLY_IF_THREADS_ENABLED(update_thread_context),
1112  .capabilities = CODEC_CAP_DR1 /*| CODEC_CAP_DRAW_HORIZ_BAND*/ |
1114 };