FFmpeg  4.1.11
cbs_av1.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include "libavutil/avassert.h"
20 #include "libavutil/pixfmt.h"
21 
22 #include "cbs.h"
23 #include "cbs_internal.h"
24 #include "cbs_av1.h"
25 #include "internal.h"
26 
27 
29  const char *name, uint32_t *write_to,
30  uint32_t range_min, uint32_t range_max)
31 {
32  uint32_t zeroes, bits_value, value;
33  int position;
34 
35  if (ctx->trace_enable)
36  position = get_bits_count(gbc);
37 
38  zeroes = 0;
39  while (1) {
40  if (get_bits_left(gbc) < 1) {
41  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid uvlc code at "
42  "%s: bitstream ended.\n", name);
43  return AVERROR_INVALIDDATA;
44  }
45 
46  if (get_bits1(gbc))
47  break;
48  ++zeroes;
49  }
50 
51  if (zeroes >= 32) {
52  value = MAX_UINT_BITS(32);
53  } else {
54  if (get_bits_left(gbc) < zeroes) {
55  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid uvlc code at "
56  "%s: bitstream ended.\n", name);
57  return AVERROR_INVALIDDATA;
58  }
59 
60  bits_value = get_bits_long(gbc, zeroes);
61  value = bits_value + (UINT32_C(1) << zeroes) - 1;
62  }
63 
64  if (ctx->trace_enable) {
65  char bits[65];
66  int i, j, k;
67 
68  if (zeroes >= 32) {
69  while (zeroes > 32) {
70  k = FFMIN(zeroes - 32, 32);
71  for (i = 0; i < k; i++)
72  bits[i] = '0';
73  bits[i] = 0;
74  ff_cbs_trace_syntax_element(ctx, position, name,
75  NULL, bits, 0);
76  zeroes -= k;
77  position += k;
78  }
79  }
80 
81  for (i = 0; i < zeroes; i++)
82  bits[i] = '0';
83  bits[i++] = '1';
84 
85  if (zeroes < 32) {
86  for (j = 0; j < zeroes; j++)
87  bits[i++] = (bits_value >> (zeroes - j - 1) & 1) ? '1' : '0';
88  }
89 
90  bits[i] = 0;
91  ff_cbs_trace_syntax_element(ctx, position, name,
92  NULL, bits, value);
93  }
94 
95  if (value < range_min || value > range_max) {
96  av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
97  "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
98  name, value, range_min, range_max);
99  return AVERROR_INVALIDDATA;
100  }
101 
102  *write_to = value;
103  return 0;
104 }
105 
107  const char *name, uint32_t value,
108  uint32_t range_min, uint32_t range_max)
109 {
110  uint32_t v;
111  int position, zeroes;
112 
113  if (value < range_min || value > range_max) {
114  av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
115  "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
116  name, value, range_min, range_max);
117  return AVERROR_INVALIDDATA;
118  }
119 
120  if (ctx->trace_enable)
121  position = put_bits_count(pbc);
122 
123  if (value == 0) {
124  zeroes = 0;
125  put_bits(pbc, 1, 1);
126  } else {
127  zeroes = av_log2(value + 1);
128  v = value - (1U << zeroes) + 1;
129  put_bits(pbc, zeroes, 0);
130  put_bits(pbc, 1, 1);
131  put_bits(pbc, zeroes, v);
132  }
133 
134  if (ctx->trace_enable) {
135  char bits[65];
136  int i, j;
137  i = 0;
138  for (j = 0; j < zeroes; j++)
139  bits[i++] = '0';
140  bits[i++] = '1';
141  for (j = 0; j < zeroes; j++)
142  bits[i++] = (v >> (zeroes - j - 1) & 1) ? '1' : '0';
143  bits[i++] = 0;
144  ff_cbs_trace_syntax_element(ctx, position, name, NULL,
145  bits, value);
146  }
147 
148  return 0;
149 }
150 
152  const char *name, uint64_t *write_to)
153 {
154  uint64_t value;
155  int position, err, i;
156 
157  if (ctx->trace_enable)
158  position = get_bits_count(gbc);
159 
160  value = 0;
161  for (i = 0; i < 8; i++) {
162  int subscript[2] = { 1, i };
163  uint32_t byte;
164  err = ff_cbs_read_unsigned(ctx, gbc, 8, "leb128_byte[i]", subscript,
165  &byte, 0x00, 0xff);
166  if (err < 0)
167  return err;
168 
169  value |= (uint64_t)(byte & 0x7f) << (i * 7);
170  if (!(byte & 0x80))
171  break;
172  }
173 
174  if (value > UINT32_MAX)
175  return AVERROR_INVALIDDATA;
176 
177  if (ctx->trace_enable)
178  ff_cbs_trace_syntax_element(ctx, position, name, NULL, "", value);
179 
180  *write_to = value;
181  return 0;
182 }
183 
185  const char *name, uint64_t value)
186 {
187  int position, err, len, i;
188  uint8_t byte;
189 
190  len = (av_log2(value) + 7) / 7;
191 
192  if (ctx->trace_enable)
193  position = put_bits_count(pbc);
194 
195  for (i = 0; i < len; i++) {
196  int subscript[2] = { 1, i };
197 
198  byte = value >> (7 * i) & 0x7f;
199  if (i < len - 1)
200  byte |= 0x80;
201 
202  err = ff_cbs_write_unsigned(ctx, pbc, 8, "leb128_byte[i]", subscript,
203  byte, 0x00, 0xff);
204  if (err < 0)
205  return err;
206  }
207 
208  if (ctx->trace_enable)
209  ff_cbs_trace_syntax_element(ctx, position, name, NULL, "", value);
210 
211  return 0;
212 }
213 
215  int width, const char *name,
216  const int *subscripts, int32_t *write_to)
217 {
218  int position;
219  int32_t value;
220 
221  if (ctx->trace_enable)
222  position = get_bits_count(gbc);
223 
224  if (get_bits_left(gbc) < width) {
225  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid signed value at "
226  "%s: bitstream ended.\n", name);
227  return AVERROR_INVALIDDATA;
228  }
229 
230  value = get_sbits(gbc, width);
231 
232  if (ctx->trace_enable) {
233  char bits[33];
234  int i;
235  for (i = 0; i < width; i++)
236  bits[i] = value & (1 << (width - i - 1)) ? '1' : '0';
237  bits[i] = 0;
238 
239  ff_cbs_trace_syntax_element(ctx, position,
240  name, subscripts, bits, value);
241  }
242 
243  *write_to = value;
244  return 0;
245 }
246 
248  int width, const char *name,
249  const int *subscripts, int32_t value)
250 {
251  if (put_bits_left(pbc) < width)
252  return AVERROR(ENOSPC);
253 
254  if (ctx->trace_enable) {
255  char bits[33];
256  int i;
257  for (i = 0; i < width; i++)
258  bits[i] = value & (1 << (width - i - 1)) ? '1' : '0';
259  bits[i] = 0;
260 
262  name, subscripts, bits, value);
263  }
264 
265  put_sbits(pbc, width, value);
266 
267  return 0;
268 }
269 
271  uint32_t n, const char *name,
272  const int *subscripts, uint32_t *write_to)
273 {
274  uint32_t w, m, v, extra_bit, value;
275  int position;
276 
277  av_assert0(n > 0);
278 
279  if (ctx->trace_enable)
280  position = get_bits_count(gbc);
281 
282  w = av_log2(n) + 1;
283  m = (1 << w) - n;
284 
285  if (get_bits_left(gbc) < w) {
286  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid non-symmetric value at "
287  "%s: bitstream ended.\n", name);
288  return AVERROR_INVALIDDATA;
289  }
290 
291  if (w - 1 > 0)
292  v = get_bits(gbc, w - 1);
293  else
294  v = 0;
295 
296  if (v < m) {
297  value = v;
298  } else {
299  extra_bit = get_bits1(gbc);
300  value = (v << 1) - m + extra_bit;
301  }
302 
303  if (ctx->trace_enable) {
304  char bits[33];
305  int i;
306  for (i = 0; i < w - 1; i++)
307  bits[i] = (v >> i & 1) ? '1' : '0';
308  if (v >= m)
309  bits[i++] = extra_bit ? '1' : '0';
310  bits[i] = 0;
311 
312  ff_cbs_trace_syntax_element(ctx, position,
313  name, subscripts, bits, value);
314  }
315 
316  *write_to = value;
317  return 0;
318 }
319 
321  uint32_t n, const char *name,
322  const int *subscripts, uint32_t value)
323 {
324  uint32_t w, m, v, extra_bit;
325  int position;
326 
327  if (value > n) {
328  av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
329  "%"PRIu32", but must be in [0,%"PRIu32"].\n",
330  name, value, n);
331  return AVERROR_INVALIDDATA;
332  }
333 
334  if (ctx->trace_enable)
335  position = put_bits_count(pbc);
336 
337  w = av_log2(n) + 1;
338  m = (1 << w) - n;
339 
340  if (put_bits_left(pbc) < w)
341  return AVERROR(ENOSPC);
342 
343  if (value < m) {
344  v = value;
345  put_bits(pbc, w - 1, v);
346  } else {
347  v = m + ((value - m) >> 1);
348  extra_bit = (value - m) & 1;
349  put_bits(pbc, w - 1, v);
350  put_bits(pbc, 1, extra_bit);
351  }
352 
353  if (ctx->trace_enable) {
354  char bits[33];
355  int i;
356  for (i = 0; i < w - 1; i++)
357  bits[i] = (v >> i & 1) ? '1' : '0';
358  if (value >= m)
359  bits[i++] = extra_bit ? '1' : '0';
360  bits[i] = 0;
361 
362  ff_cbs_trace_syntax_element(ctx, position,
363  name, subscripts, bits, value);
364  }
365 
366  return 0;
367 }
368 
370  uint32_t range_min, uint32_t range_max,
371  const char *name, uint32_t *write_to)
372 {
373  uint32_t value;
374  int position, i;
375  char bits[33];
376 
377  av_assert0(range_min <= range_max && range_max - range_min < sizeof(bits) - 1);
378  if (ctx->trace_enable)
379  position = get_bits_count(gbc);
380 
381  for (i = 0, value = range_min; value < range_max;) {
382  if (get_bits_left(gbc) < 1) {
383  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid increment value at "
384  "%s: bitstream ended.\n", name);
385  return AVERROR_INVALIDDATA;
386  }
387  if (get_bits1(gbc)) {
388  bits[i++] = '1';
389  ++value;
390  } else {
391  bits[i++] = '0';
392  break;
393  }
394  }
395 
396  if (ctx->trace_enable) {
397  bits[i] = 0;
398  ff_cbs_trace_syntax_element(ctx, position,
399  name, NULL, bits, value);
400  }
401 
402  *write_to = value;
403  return 0;
404 }
405 
407  uint32_t range_min, uint32_t range_max,
408  const char *name, uint32_t value)
409 {
410  int len;
411 
412  av_assert0(range_min <= range_max && range_max - range_min < 32);
413  if (value < range_min || value > range_max) {
414  av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
415  "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
416  name, value, range_min, range_max);
417  return AVERROR_INVALIDDATA;
418  }
419 
420  if (value == range_max)
421  len = range_max - range_min;
422  else
423  len = value - range_min + 1;
424  if (put_bits_left(pbc) < len)
425  return AVERROR(ENOSPC);
426 
427  if (ctx->trace_enable) {
428  char bits[33];
429  int i;
430  for (i = 0; i < len; i++) {
431  if (range_min + i == value)
432  bits[i] = '0';
433  else
434  bits[i] = '1';
435  }
436  bits[i] = 0;
438  name, NULL, bits, value);
439  }
440 
441  if (len > 0)
442  put_bits(pbc, len, (1 << len) - 1 - (value != range_max));
443 
444  return 0;
445 }
446 
448  uint32_t range_max, const char *name,
449  const int *subscripts, uint32_t *write_to)
450 {
451  uint32_t value;
452  int position, err;
453  uint32_t max_len, len, range_offset, range_bits;
454 
455  if (ctx->trace_enable)
456  position = get_bits_count(gbc);
457 
458  av_assert0(range_max > 0);
459  max_len = av_log2(range_max - 1) - 3;
460 
461  err = cbs_av1_read_increment(ctx, gbc, 0, max_len,
462  "subexp_more_bits", &len);
463  if (err < 0)
464  return err;
465 
466  if (len) {
467  range_bits = 2 + len;
468  range_offset = 1 << range_bits;
469  } else {
470  range_bits = 3;
471  range_offset = 0;
472  }
473 
474  if (len < max_len) {
475  err = ff_cbs_read_unsigned(ctx, gbc, range_bits,
476  "subexp_bits", NULL, &value,
477  0, MAX_UINT_BITS(range_bits));
478  if (err < 0)
479  return err;
480 
481  } else {
482  err = cbs_av1_read_ns(ctx, gbc, range_max - range_offset,
483  "subexp_final_bits", NULL, &value);
484  if (err < 0)
485  return err;
486  }
487  value += range_offset;
488 
489  if (ctx->trace_enable)
490  ff_cbs_trace_syntax_element(ctx, position,
491  name, subscripts, "", value);
492 
493  *write_to = value;
494  return err;
495 }
496 
498  uint32_t range_max, const char *name,
499  const int *subscripts, uint32_t value)
500 {
501  int position, err;
502  uint32_t max_len, len, range_offset, range_bits;
503 
504  if (value > range_max) {
505  av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
506  "%"PRIu32", but must be in [0,%"PRIu32"].\n",
507  name, value, range_max);
508  return AVERROR_INVALIDDATA;
509  }
510 
511  if (ctx->trace_enable)
512  position = put_bits_count(pbc);
513 
514  av_assert0(range_max > 0);
515  max_len = av_log2(range_max - 1) - 3;
516 
517  if (value < 8) {
518  range_bits = 3;
519  range_offset = 0;
520  len = 0;
521  } else {
522  range_bits = av_log2(value);
523  len = range_bits - 2;
524  if (len > max_len) {
525  // The top bin is combined with the one below it.
526  av_assert0(len == max_len + 1);
527  --range_bits;
528  len = max_len;
529  }
530  range_offset = 1 << range_bits;
531  }
532 
533  err = cbs_av1_write_increment(ctx, pbc, 0, max_len,
534  "subexp_more_bits", len);
535  if (err < 0)
536  return err;
537 
538  if (len < max_len) {
539  err = ff_cbs_write_unsigned(ctx, pbc, range_bits,
540  "subexp_bits", NULL,
541  value - range_offset,
542  0, MAX_UINT_BITS(range_bits));
543  if (err < 0)
544  return err;
545 
546  } else {
547  err = cbs_av1_write_ns(ctx, pbc, range_max - range_offset,
548  "subexp_final_bits", NULL,
549  value - range_offset);
550  if (err < 0)
551  return err;
552  }
553 
554  if (ctx->trace_enable)
555  ff_cbs_trace_syntax_element(ctx, position,
556  name, subscripts, "", value);
557 
558  return err;
559 }
560 
561 
562 static int cbs_av1_tile_log2(int blksize, int target)
563 {
564  int k;
565  for (k = 0; (blksize << k) < target; k++);
566  return k;
567 }
568 
570  unsigned int a, unsigned int b)
571 {
572  unsigned int diff, m;
573  if (!seq->enable_order_hint)
574  return 0;
575  diff = a - b;
576  m = 1 << seq->order_hint_bits_minus_1;
577  diff = (diff & (m - 1)) - (diff & m);
578  return diff;
579 }
580 
581 
582 #define HEADER(name) do { \
583  ff_cbs_trace_header(ctx, name); \
584  } while (0)
585 
586 #define CHECK(call) do { \
587  err = (call); \
588  if (err < 0) \
589  return err; \
590  } while (0)
591 
592 #define FUNC_NAME(rw, codec, name) cbs_ ## codec ## _ ## rw ## _ ## name
593 #define FUNC_AV1(rw, name) FUNC_NAME(rw, av1, name)
594 #define FUNC(name) FUNC_AV1(READWRITE, name)
595 
596 #define SUBSCRIPTS(subs, ...) (subs > 0 ? ((int[subs + 1]){ subs, __VA_ARGS__ }) : NULL)
597 
598 #define fb(width, name) \
599  xf(width, name, current->name, 0, MAX_UINT_BITS(width), 0)
600 #define fc(width, name, range_min, range_max) \
601  xf(width, name, current->name, range_min, range_max, 0)
602 #define flag(name) fb(1, name)
603 #define su(width, name) \
604  xsu(width, name, current->name, 0)
605 
606 #define fbs(width, name, subs, ...) \
607  xf(width, name, current->name, 0, MAX_UINT_BITS(width), subs, __VA_ARGS__)
608 #define fcs(width, name, range_min, range_max, subs, ...) \
609  xf(width, name, current->name, range_min, range_max, subs, __VA_ARGS__)
610 #define flags(name, subs, ...) \
611  xf(1, name, current->name, 0, 1, subs, __VA_ARGS__)
612 #define sus(width, name, subs, ...) \
613  xsu(width, name, current->name, subs, __VA_ARGS__)
614 
615 #define fixed(width, name, value) do { \
616  av_unused uint32_t fixed_value = value; \
617  xf(width, name, fixed_value, value, value, 0); \
618  } while (0)
619 
620 
621 #define READ
622 #define READWRITE read
623 #define RWContext GetBitContext
624 
625 #define xf(width, name, var, range_min, range_max, subs, ...) do { \
626  uint32_t value = range_min; \
627  CHECK(ff_cbs_read_unsigned(ctx, rw, width, #name, \
628  SUBSCRIPTS(subs, __VA_ARGS__), \
629  &value, range_min, range_max)); \
630  var = value; \
631  } while (0)
632 
633 #define xsu(width, name, var, subs, ...) do { \
634  int32_t value = 0; \
635  CHECK(cbs_av1_read_su(ctx, rw, width, #name, \
636  SUBSCRIPTS(subs, __VA_ARGS__), &value)); \
637  var = value; \
638  } while (0)
639 
640 #define uvlc(name, range_min, range_max) do { \
641  uint32_t value = range_min; \
642  CHECK(cbs_av1_read_uvlc(ctx, rw, #name, \
643  &value, range_min, range_max)); \
644  current->name = value; \
645  } while (0)
646 
647 #define ns(max_value, name, subs, ...) do { \
648  uint32_t value = 0; \
649  CHECK(cbs_av1_read_ns(ctx, rw, max_value, #name, \
650  SUBSCRIPTS(subs, __VA_ARGS__), &value)); \
651  current->name = value; \
652  } while (0)
653 
654 #define increment(name, min, max) do { \
655  uint32_t value = 0; \
656  CHECK(cbs_av1_read_increment(ctx, rw, min, max, #name, &value)); \
657  current->name = value; \
658  } while (0)
659 
660 #define subexp(name, max, subs, ...) do { \
661  uint32_t value = 0; \
662  CHECK(cbs_av1_read_subexp(ctx, rw, max, #name, \
663  SUBSCRIPTS(subs, __VA_ARGS__), &value)); \
664  current->name = value; \
665  } while (0)
666 
667 #define delta_q(name) do { \
668  uint8_t delta_coded; \
669  int8_t delta_q; \
670  xf(1, name.delta_coded, delta_coded, 0, 1, 0); \
671  if (delta_coded) \
672  xsu(1 + 6, name.delta_q, delta_q, 0); \
673  else \
674  delta_q = 0; \
675  current->name = delta_q; \
676  } while (0)
677 
678 #define leb128(name) do { \
679  uint64_t value = 0; \
680  CHECK(cbs_av1_read_leb128(ctx, rw, #name, &value)); \
681  current->name = value; \
682  } while (0)
683 
684 #define infer(name, value) do { \
685  current->name = value; \
686  } while (0)
687 
688 #define byte_alignment(rw) (get_bits_count(rw) % 8)
689 
690 #include "cbs_av1_syntax_template.c"
691 
692 #undef READ
693 #undef READWRITE
694 #undef RWContext
695 #undef xf
696 #undef xsu
697 #undef uvlc
698 #undef leb128
699 #undef ns
700 #undef increment
701 #undef subexp
702 #undef delta_q
703 #undef leb128
704 #undef infer
705 #undef byte_alignment
706 
707 
708 #define WRITE
709 #define READWRITE write
710 #define RWContext PutBitContext
711 
712 #define xf(width, name, var, range_min, range_max, subs, ...) do { \
713  CHECK(ff_cbs_write_unsigned(ctx, rw, width, #name, \
714  SUBSCRIPTS(subs, __VA_ARGS__), \
715  var, range_min, range_max)); \
716  } while (0)
717 
718 #define xsu(width, name, var, subs, ...) do { \
719  CHECK(cbs_av1_write_su(ctx, rw, width, #name, \
720  SUBSCRIPTS(subs, __VA_ARGS__), var)); \
721  } while (0)
722 
723 #define uvlc(name, range_min, range_max) do { \
724  CHECK(cbs_av1_write_uvlc(ctx, rw, #name, current->name, \
725  range_min, range_max)); \
726  } while (0)
727 
728 #define ns(max_value, name, subs, ...) do { \
729  CHECK(cbs_av1_write_ns(ctx, rw, max_value, #name, \
730  SUBSCRIPTS(subs, __VA_ARGS__), \
731  current->name)); \
732  } while (0)
733 
734 #define increment(name, min, max) do { \
735  CHECK(cbs_av1_write_increment(ctx, rw, min, max, #name, \
736  current->name)); \
737  } while (0)
738 
739 #define subexp(name, max, subs, ...) do { \
740  CHECK(cbs_av1_write_subexp(ctx, rw, max, #name, \
741  SUBSCRIPTS(subs, __VA_ARGS__), \
742  current->name)); \
743  } while (0)
744 
745 #define delta_q(name) do { \
746  xf(1, name.delta_coded, current->name != 0, 0, 1, 0); \
747  if (current->name) \
748  xsu(1 + 6, name.delta_q, current->name, 0); \
749  } while (0)
750 
751 #define leb128(name) do { \
752  CHECK(cbs_av1_write_leb128(ctx, rw, #name, current->name)); \
753  } while (0)
754 
755 #define infer(name, value) do { \
756  if (current->name != (value)) { \
757  av_log(ctx->log_ctx, AV_LOG_WARNING, "Warning: " \
758  "%s does not match inferred value: " \
759  "%"PRId64", but should be %"PRId64".\n", \
760  #name, (int64_t)current->name, (int64_t)(value)); \
761  } \
762  } while (0)
763 
764 #define byte_alignment(rw) (put_bits_count(rw) % 8)
765 
766 #include "cbs_av1_syntax_template.c"
767 
768 #undef READ
769 #undef READWRITE
770 #undef RWContext
771 #undef xf
772 #undef xsu
773 #undef uvlc
774 #undef leb128
775 #undef ns
776 #undef increment
777 #undef subexp
778 #undef delta_q
779 #undef infer
780 #undef byte_alignment
781 
782 
785  int header)
786 {
787  GetBitContext gbc;
788  uint8_t *data;
789  size_t size;
790  uint64_t obu_length;
791  int pos, err, trace;
792 
793  // Don't include this parsing in trace output.
794  trace = ctx->trace_enable;
795  ctx->trace_enable = 0;
796 
797  data = frag->data;
798  size = frag->data_size;
799 
800  if (INT_MAX / 8 < size) {
801  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid fragment: "
802  "too large (%"SIZE_SPECIFIER" bytes).\n", size);
803  err = AVERROR_INVALIDDATA;
804  goto fail;
805  }
806 
807  while (size > 0) {
809  uint64_t obu_size;
810 
811  init_get_bits(&gbc, data, 8 * size);
812 
813  err = cbs_av1_read_obu_header(ctx, &gbc, &header);
814  if (err < 0)
815  goto fail;
816 
817  if (!header.obu_has_size_field) {
818  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid OBU for raw "
819  "stream: size field must be present.\n");
820  err = AVERROR_INVALIDDATA;
821  goto fail;
822  }
823 
824  if (get_bits_left(&gbc) < 8) {
825  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid OBU: fragment "
826  "too short (%"SIZE_SPECIFIER" bytes).\n", size);
827  err = AVERROR_INVALIDDATA;
828  goto fail;
829  }
830 
831  err = cbs_av1_read_leb128(ctx, &gbc, "obu_size", &obu_size);
832  if (err < 0)
833  goto fail;
834 
835  pos = get_bits_count(&gbc);
836  av_assert0(pos % 8 == 0 && pos / 8 <= size);
837 
838  obu_length = pos / 8 + obu_size;
839 
840  if (size < obu_length) {
841  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid OBU length: "
842  "%"PRIu64", but only %"SIZE_SPECIFIER" bytes remaining in fragment.\n",
843  obu_length, size);
844  err = AVERROR_INVALIDDATA;
845  goto fail;
846  }
847 
848  err = ff_cbs_insert_unit_data(ctx, frag, -1, header.obu_type,
849  data, obu_length, frag->data_ref);
850  if (err < 0)
851  goto fail;
852 
853  data += obu_length;
854  size -= obu_length;
855  }
856 
857  err = 0;
858 fail:
859  ctx->trace_enable = trace;
860  return err;
861 }
862 
864 {
866 }
867 
869 {
870  switch (md->metadata_type) {
873  break;
874  }
875 }
876 
877 static void cbs_av1_free_obu(void *unit, uint8_t *content)
878 {
879  AV1RawOBU *obu = (AV1RawOBU*)content;
880 
881  switch (obu->header.obu_type) {
882  case AV1_OBU_TILE_GROUP:
884  break;
885  case AV1_OBU_FRAME:
887  break;
888  case AV1_OBU_TILE_LIST:
890  break;
891  case AV1_OBU_METADATA:
893  break;
894  }
895 
896  av_freep(&obu);
897 }
898 
900  CodedBitstreamUnit *unit,
901  GetBitContext *gbc,
903 {
904  int pos;
905 
906  pos = get_bits_count(gbc);
907  if (pos >= 8 * unit->data_size) {
908  av_log(ctx->log_ctx, AV_LOG_ERROR, "Bitstream ended before "
909  "any data in tile group (%d bits read).\n", pos);
910  return AVERROR_INVALIDDATA;
911  }
912  // Must be byte-aligned at this point.
913  av_assert0(pos % 8 == 0);
914 
915  td->data_ref = av_buffer_ref(unit->data_ref);
916  if (!td->data_ref)
917  return AVERROR(ENOMEM);
918 
919  td->data = unit->data + pos / 8;
920  td->data_size = unit->data_size - pos / 8;
921 
922  return 0;
923 }
924 
926  CodedBitstreamUnit *unit)
927 {
928  CodedBitstreamAV1Context *priv = ctx->priv_data;
929  AV1RawOBU *obu;
930  GetBitContext gbc;
931  int err, start_pos, end_pos;
932 
933  err = ff_cbs_alloc_unit_content(ctx, unit, sizeof(*obu),
935  if (err < 0)
936  return err;
937  obu = unit->content;
938 
939  err = init_get_bits(&gbc, unit->data, 8 * unit->data_size);
940  if (err < 0)
941  return err;
942 
943  err = cbs_av1_read_obu_header(ctx, &gbc, &obu->header);
944  if (err < 0)
945  return err;
946  av_assert0(obu->header.obu_type == unit->type);
947 
948  if (obu->header.obu_has_size_field) {
949  uint64_t obu_size;
950  err = cbs_av1_read_leb128(ctx, &gbc, "obu_size", &obu_size);
951  if (err < 0)
952  return err;
953  obu->obu_size = obu_size;
954  } else {
955  if (unit->data_size < 1 + obu->header.obu_extension_flag) {
956  av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid OBU length: "
957  "unit too short (%"SIZE_SPECIFIER").\n", unit->data_size);
958  return AVERROR_INVALIDDATA;
959  }
960  obu->obu_size = unit->data_size - 1 - obu->header.obu_extension_flag;
961  }
962 
963  start_pos = get_bits_count(&gbc);
964 
965  if (obu->header.obu_extension_flag) {
966  priv->temporal_id = obu->header.temporal_id;
967  priv->spatial_id = obu->header.spatial_id;
968 
971  priv->operating_point_idc) {
972  int in_temporal_layer =
973  (priv->operating_point_idc >> priv->temporal_id ) & 1;
974  int in_spatial_layer =
975  (priv->operating_point_idc >> (priv->spatial_id + 8)) & 1;
976  if (!in_temporal_layer || !in_spatial_layer) {
977  // Decoding will drop this OBU at this operating point.
978  }
979  }
980  } else {
981  priv->temporal_id = 0;
982  priv->spatial_id = 0;
983  }
984 
985  priv->ref = (AV1ReferenceFrameState *)&priv->read_ref;
986 
987  switch (obu->header.obu_type) {
989  {
990  err = cbs_av1_read_sequence_header_obu(ctx, &gbc,
991  &obu->obu.sequence_header);
992  if (err < 0)
993  return err;
994 
996  priv->sequence_header = NULL;
997 
999  if (!priv->sequence_header_ref)
1000  return AVERROR(ENOMEM);
1001  priv->sequence_header = &obu->obu.sequence_header;
1002  }
1003  break;
1005  {
1006  err = cbs_av1_read_temporal_delimiter_obu(ctx, &gbc);
1007  if (err < 0)
1008  return err;
1009  }
1010  break;
1011  case AV1_OBU_FRAME_HEADER:
1013  {
1014  err = cbs_av1_read_frame_header_obu(ctx, &gbc,
1015  &obu->obu.frame_header,
1016  obu->header.obu_type ==
1018  unit->data_ref);
1019  if (err < 0)
1020  return err;
1021  }
1022  break;
1023  case AV1_OBU_TILE_GROUP:
1024  {
1025  err = cbs_av1_read_tile_group_obu(ctx, &gbc,
1026  &obu->obu.tile_group);
1027  if (err < 0)
1028  return err;
1029 
1030  err = cbs_av1_ref_tile_data(ctx, unit, &gbc,
1031  &obu->obu.tile_group.tile_data);
1032  if (err < 0)
1033  return err;
1034  }
1035  break;
1036  case AV1_OBU_FRAME:
1037  {
1038  err = cbs_av1_read_frame_obu(ctx, &gbc, &obu->obu.frame,
1039  unit->data_ref);
1040  if (err < 0)
1041  return err;
1042 
1043  err = cbs_av1_ref_tile_data(ctx, unit, &gbc,
1044  &obu->obu.frame.tile_group.tile_data);
1045  if (err < 0)
1046  return err;
1047  }
1048  break;
1049  case AV1_OBU_TILE_LIST:
1050  {
1051  err = cbs_av1_read_tile_list_obu(ctx, &gbc,
1052  &obu->obu.tile_list);
1053  if (err < 0)
1054  return err;
1055 
1056  err = cbs_av1_ref_tile_data(ctx, unit, &gbc,
1057  &obu->obu.tile_list.tile_data);
1058  if (err < 0)
1059  return err;
1060  }
1061  break;
1062  case AV1_OBU_METADATA:
1063  {
1064  err = cbs_av1_read_metadata_obu(ctx, &gbc, &obu->obu.metadata);
1065  if (err < 0)
1066  return err;
1067  }
1068  break;
1069  case AV1_OBU_PADDING:
1070  default:
1071  return AVERROR(ENOSYS);
1072  }
1073 
1074  end_pos = get_bits_count(&gbc);
1075  av_assert0(end_pos <= unit->data_size * 8);
1076 
1077  if (obu->obu_size > 0 &&
1079  obu->header.obu_type != AV1_OBU_TILE_LIST &&
1080  obu->header.obu_type != AV1_OBU_FRAME) {
1081  int nb_bits = obu->obu_size * 8 + start_pos - end_pos;
1082 
1083  if (nb_bits <= 0)
1084  return AVERROR_INVALIDDATA;
1085 
1086  err = cbs_av1_read_trailing_bits(ctx, &gbc, nb_bits);
1087  if (err < 0)
1088  return err;
1089  }
1090 
1091  return 0;
1092 }
1093 
1095  CodedBitstreamUnit *unit,
1096  PutBitContext *pbc)
1097 {
1098  CodedBitstreamAV1Context *priv = ctx->priv_data;
1099  AV1RawOBU *obu = unit->content;
1100  PutBitContext pbc_tmp;
1101  AV1RawTileData *td;
1102  size_t header_size;
1103  int err, start_pos, end_pos, data_pos;
1104 
1105  // OBUs in the normal bitstream format must contain a size field
1106  // in every OBU (in annex B it is optional, but we don't support
1107  // writing that).
1108  obu->header.obu_has_size_field = 1;
1109 
1110  err = cbs_av1_write_obu_header(ctx, pbc, &obu->header);
1111  if (err < 0)
1112  return err;
1113 
1114  if (obu->header.obu_has_size_field) {
1115  pbc_tmp = *pbc;
1116  // Add space for the size field to fill later.
1117  put_bits32(pbc, 0);
1118  put_bits32(pbc, 0);
1119  }
1120 
1121  td = NULL;
1122  start_pos = put_bits_count(pbc);
1123 
1124  priv->ref = (AV1ReferenceFrameState *)&priv->write_ref;
1125 
1126  switch (obu->header.obu_type) {
1128  {
1129  err = cbs_av1_write_sequence_header_obu(ctx, pbc,
1130  &obu->obu.sequence_header);
1131  if (err < 0)
1132  return err;
1133 
1135  priv->sequence_header = NULL;
1136 
1138  if (!priv->sequence_header_ref)
1139  return AVERROR(ENOMEM);
1140  priv->sequence_header = &obu->obu.sequence_header;
1141  }
1142  break;
1144  {
1145  err = cbs_av1_write_temporal_delimiter_obu(ctx, pbc);
1146  if (err < 0)
1147  return err;
1148  }
1149  break;
1150  case AV1_OBU_FRAME_HEADER:
1152  {
1153  err = cbs_av1_write_frame_header_obu(ctx, pbc,
1154  &obu->obu.frame_header,
1155  obu->header.obu_type ==
1157  NULL);
1158  if (err < 0)
1159  return err;
1160  }
1161  break;
1162  case AV1_OBU_TILE_GROUP:
1163  {
1164  err = cbs_av1_write_tile_group_obu(ctx, pbc,
1165  &obu->obu.tile_group);
1166  if (err < 0)
1167  return err;
1168 
1169  td = &obu->obu.tile_group.tile_data;
1170  }
1171  break;
1172  case AV1_OBU_FRAME:
1173  {
1174  err = cbs_av1_write_frame_obu(ctx, pbc, &obu->obu.frame, NULL);
1175  if (err < 0)
1176  return err;
1177 
1178  td = &obu->obu.frame.tile_group.tile_data;
1179  }
1180  break;
1181  case AV1_OBU_TILE_LIST:
1182  {
1183  err = cbs_av1_write_tile_list_obu(ctx, pbc, &obu->obu.tile_list);
1184  if (err < 0)
1185  return err;
1186 
1187  td = &obu->obu.tile_list.tile_data;
1188  }
1189  break;
1190  case AV1_OBU_METADATA:
1191  {
1192  err = cbs_av1_write_metadata_obu(ctx, pbc, &obu->obu.metadata);
1193  if (err < 0)
1194  return err;
1195  }
1196  break;
1197  case AV1_OBU_PADDING:
1198  default:
1199  return AVERROR(ENOSYS);
1200  }
1201 
1202  end_pos = put_bits_count(pbc);
1203  header_size = (end_pos - start_pos + 7) / 8;
1204  if (td) {
1205  obu->obu_size = header_size + td->data_size;
1206  } else if (header_size > 0) {
1207  // Add trailing bits and recalculate.
1208  err = cbs_av1_write_trailing_bits(ctx, pbc, 8 - end_pos % 8);
1209  if (err < 0)
1210  return err;
1211  end_pos = put_bits_count(pbc);
1212  obu->obu_size = header_size = (end_pos - start_pos + 7) / 8;
1213  } else {
1214  // Empty OBU.
1215  obu->obu_size = 0;
1216  }
1217 
1218  end_pos = put_bits_count(pbc);
1219  // Must now be byte-aligned.
1220  av_assert0(end_pos % 8 == 0);
1221  flush_put_bits(pbc);
1222  start_pos /= 8;
1223  end_pos /= 8;
1224 
1225  *pbc = pbc_tmp;
1226  err = cbs_av1_write_leb128(ctx, pbc, "obu_size", obu->obu_size);
1227  if (err < 0)
1228  return err;
1229 
1230  data_pos = put_bits_count(pbc) / 8;
1231  flush_put_bits(pbc);
1232  av_assert0(data_pos <= start_pos);
1233 
1234  if (8 * obu->obu_size > put_bits_left(pbc))
1235  return AVERROR(ENOSPC);
1236 
1237  if (obu->obu_size > 0) {
1238  memmove(priv->write_buffer + data_pos,
1239  priv->write_buffer + start_pos, header_size);
1240  skip_put_bytes(pbc, header_size);
1241 
1242  if (td) {
1243  memcpy(priv->write_buffer + data_pos + header_size,
1244  td->data, td->data_size);
1245  skip_put_bytes(pbc, td->data_size);
1246  }
1247  }
1248 
1249  return 0;
1250 }
1251 
1253  CodedBitstreamUnit *unit)
1254 {
1255  CodedBitstreamAV1Context *priv = ctx->priv_data;
1256  PutBitContext pbc;
1257  int err;
1258 
1259  if (!priv->write_buffer) {
1260  // Initial write buffer size is 1MB.
1261  priv->write_buffer_size = 1024 * 1024;
1262 
1263  reallocate_and_try_again:
1264  err = av_reallocp(&priv->write_buffer, priv->write_buffer_size);
1265  if (err < 0) {
1266  av_log(ctx->log_ctx, AV_LOG_ERROR, "Unable to allocate a "
1267  "sufficiently large write buffer (last attempt "
1268  "%"SIZE_SPECIFIER" bytes).\n", priv->write_buffer_size);
1269  return err;
1270  }
1271  }
1272 
1273  init_put_bits(&pbc, priv->write_buffer, priv->write_buffer_size);
1274 
1275  err = cbs_av1_write_obu(ctx, unit, &pbc);
1276  if (err == AVERROR(ENOSPC)) {
1277  // Overflow.
1278  priv->write_buffer_size *= 2;
1279  goto reallocate_and_try_again;
1280  }
1281  if (err < 0)
1282  return err;
1283 
1284  // Overflow but we didn't notice.
1285  av_assert0(put_bits_count(&pbc) <= 8 * priv->write_buffer_size);
1286 
1287  // OBU data must be byte-aligned.
1288  av_assert0(put_bits_count(&pbc) % 8 == 0);
1289 
1290  unit->data_size = put_bits_count(&pbc) / 8;
1291  flush_put_bits(&pbc);
1292 
1293  err = ff_cbs_alloc_unit_data(ctx, unit, unit->data_size);
1294  if (err < 0)
1295  return err;
1296 
1297  memcpy(unit->data, priv->write_buffer, unit->data_size);
1298 
1299  return 0;
1300 }
1301 
1303  CodedBitstreamFragment *frag)
1304 {
1305  size_t size, pos;
1306  int i;
1307 
1308  size = 0;
1309  for (i = 0; i < frag->nb_units; i++)
1310  size += frag->units[i].data_size;
1311 
1313  if (!frag->data_ref)
1314  return AVERROR(ENOMEM);
1315  frag->data = frag->data_ref->data;
1316  memset(frag->data + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
1317 
1318  pos = 0;
1319  for (i = 0; i < frag->nb_units; i++) {
1320  memcpy(frag->data + pos, frag->units[i].data,
1321  frag->units[i].data_size);
1322  pos += frag->units[i].data_size;
1323  }
1324  av_assert0(pos == size);
1325  frag->data_size = size;
1326 
1327  return 0;
1328 }
1329 
1331 {
1332  CodedBitstreamAV1Context *priv = ctx->priv_data;
1333 
1336 
1337  av_freep(&priv->write_buffer);
1338 }
1339 
1342 
1343  .priv_data_size = sizeof(CodedBitstreamAV1Context),
1344 
1349 
1350  .close = &cbs_av1_close,
1351 };
const char * name
Definition: avisynth_c.h:775
AV1RawMetadata metadata
Definition: cbs_av1.h:379
static void av_unused put_bits32(PutBitContext *s, uint32_t value)
Write exactly 32 bits into a bitstream.
Definition: put_bits.h:250
#define NULL
Definition: coverity.c:32
int nb_units
Number of units in this fragment.
Definition: cbs.h:147
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
static void cbs_av1_free_obu(void *unit, uint8_t *content)
Definition: cbs_av1.c:877
AV1ReferenceFrameState write_ref[AV1_NUM_REF_FRAMES]
Definition: cbs_av1.h:426
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it...
Definition: buffer.c:125
int size
size_t obu_size
Definition: cbs_av1.h:371
AV1RawTileList tile_list
Definition: cbs_av1.h:378
static void put_sbits(PutBitContext *pb, int n, int32_t value)
Definition: put_bits.h:240
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:208
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:381
AVBufferRef * payload_ref
Definition: cbs_av1.h:337
static void cbs_av1_free_metadata(AV1RawMetadata *md)
Definition: cbs_av1.c:868
static int cbs_av1_write_su(CodedBitstreamContext *ctx, PutBitContext *pbc, int width, const char *name, const int *subscripts, int32_t value)
Definition: cbs_av1.c:247
AVBufferRef * sequence_header_ref
Definition: cbs_av1.h:400
CodedBitstreamUnitType type
Codec-specific type of this unit.
Definition: cbs.h:68
static int cbs_av1_read_uvlc(CodedBitstreamContext *ctx, GetBitContext *gbc, const char *name, uint32_t *write_to, uint32_t range_min, uint32_t range_max)
Definition: cbs_av1.c:28
static void cbs_av1_close(CodedBitstreamContext *ctx)
Definition: cbs_av1.c:1330
const char * b
Definition: vf_curves.c:116
int ff_cbs_alloc_unit_content(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit, size_t size, void(*free)(void *opaque, uint8_t *data))
Definition: cbs.c:585
uint8_t order_hint_bits_minus_1
Definition: cbs_av1.h:122
static int get_sbits(GetBitContext *s, int n)
Definition: get_bits.h:361
int ff_cbs_write_unsigned(CodedBitstreamContext *ctx, PutBitContext *pbc, int width, const char *name, const int *subscripts, uint32_t value, uint32_t range_min, uint32_t range_max)
Definition: cbs.c:469
static int cbs_av1_write_increment(CodedBitstreamContext *ctx, PutBitContext *pbc, uint32_t range_min, uint32_t range_max, const char *name, uint32_t value)
Definition: cbs_av1.c:406
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
int trace_enable
Enable trace output during read/write operations.
Definition: cbs.h:197
AV1RawTileData tile_data
Definition: cbs_av1.h:309
static int cbs_av1_write_uvlc(CodedBitstreamContext *ctx, PutBitContext *pbc, const char *name, uint32_t value, uint32_t range_min, uint32_t range_max)
Definition: cbs_av1.c:106
int(* assemble_fragment)(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag)
Definition: cbs_internal.h:53
uint8_t
uint8_t * data
Definition: cbs_av1.h:286
static int cbs_av1_read_unit(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit)
Definition: cbs_av1.c:925
uint8_t * write_buffer
Definition: cbs_av1.h:429
union AV1RawMetadata::@46 metadata
#define MAX_UINT_BITS(length)
Definition: cbs_internal.h:96
int ff_cbs_read_unsigned(CodedBitstreamContext *ctx, GetBitContext *gbc, int width, const char *name, const int *subscripts, uint32_t *write_to, uint32_t range_min, uint32_t range_max)
Definition: cbs.c:426
const char data[16]
Definition: mxf.c:91
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:219
int ff_cbs_alloc_unit_data(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit, size_t size)
Allocate a new internal data buffer of the given size in the unit.
Definition: cbs.c:606
Coded bitstream unit structure.
Definition: cbs.h:64
void * content
Pointer to the decomposed form of this unit.
Definition: cbs.h:101
static const uint8_t header[24]
Definition: sdr2.c:67
CodedBitstreamUnit * units
Pointer to an array of units of length nb_units.
Definition: cbs.h:153
uint8_t * data
Pointer to the directly-parsable bitstream form of this unit.
Definition: cbs.h:75
int ff_cbs_insert_unit_data(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag, int position, CodedBitstreamUnitType type, uint8_t *data, size_t data_size, AVBufferRef *data_buf)
Insert a new unit into a fragment with the given data bitstream.
Definition: cbs.c:686
#define av_log(a,...)
size_t data_size
The number of bytes in the bitstream.
Definition: cbs.h:129
static int cbs_av1_read_ns(CodedBitstreamContext *ctx, GetBitContext *gbc, uint32_t n, const char *name, const int *subscripts, uint32_t *write_to)
Definition: cbs_av1.c:270
#define U(x)
Definition: vp56_arith.h:37
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:814
static void cbs_av1_free_tile_data(AV1RawTileData *td)
Definition: cbs_av1.c:863
#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 td
Definition: regdef.h:70
void(* close)(CodedBitstreamContext *ctx)
Definition: cbs_internal.h:57
static int put_bits_left(PutBitContext *s)
Definition: put_bits.h:93
AV1RawTileData tile_data
Definition: cbs_av1.h:296
uint8_t temporal_id
Definition: cbs_av1.h:36
AVBufferRef * frame_header_ref
Definition: cbs_av1.h:403
#define AVERROR(e)
Definition: error.h:43
AV1RawFrame frame
Definition: cbs_av1.h:376
static int cbs_av1_read_subexp(CodedBitstreamContext *ctx, GetBitContext *gbc, uint32_t range_max, const char *name, const int *subscripts, uint32_t *write_to)
Definition: cbs_av1.c:447
simple assert() macros that are a bit more flexible than ISO C assert().
static int cbs_av1_read_su(CodedBitstreamContext *ctx, GetBitContext *gbc, int width, const char *name, const int *subscripts, int32_t *write_to)
Definition: cbs_av1.c:214
#define fail()
Definition: checkasm.h:117
static int cbs_av1_tile_log2(int blksize, int target)
Definition: cbs_av1.c:562
static int put_bits_count(PutBitContext *s)
Definition: put_bits.h:85
void * log_ctx
Logging context to be passed to all av_log() calls associated with this context.
Definition: cbs.h:164
const CodedBitstreamType ff_cbs_type_av1
Definition: cbs_av1.c:1340
uint8_t obu_extension_flag
Definition: cbs_av1.h:32
int(* write_unit)(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit)
Definition: cbs_internal.h:48
static void skip_put_bytes(PutBitContext *s, int n)
Skip the given number of bytes.
Definition: put_bits.h:333
#define md
#define FFMIN(a, b)
Definition: common.h:96
static int cbs_av1_write_obu(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit, PutBitContext *pbc)
Definition: cbs_av1.c:1094
static int cbs_av1_write_leb128(CodedBitstreamContext *ctx, PutBitContext *pbc, const char *name, uint64_t value)
Definition: cbs_av1.c:184
static int cbs_av1_assemble_fragment(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag)
Definition: cbs_av1.c:1302
#define width
AV1ReferenceFrameState * ref
Definition: cbs_av1.h:424
uint8_t w
Definition: llviddspenc.c:38
AV1RawOBUHeader header
Definition: cbs_av1.h:369
int32_t
AVFormatContext * ctx
Definition: movenc.c:48
static int cbs_av1_split_fragment(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag, int header)
Definition: cbs_av1.c:783
int n
Definition: avisynth_c.h:684
static int cbs_av1_ref_tile_data(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit, GetBitContext *gbc, AV1RawTileData *td)
Definition: cbs_av1.c:899
uint8_t * data
Pointer to the bitstream form of this fragment.
Definition: cbs.h:122
size_t data_size
Definition: cbs_av1.h:287
#define av_log2
Definition: intmath.h:83
int av_reallocp(void *ptr, size_t size)
Allocate, reallocate, or free a block of memory through a pointer to a pointer.
Definition: mem.c:163
AVBufferRef * av_buffer_alloc(int size)
Allocate an AVBuffer of the given size using av_malloc().
Definition: buffer.c:67
AV1ReferenceFrameState read_ref[AV1_NUM_REF_FRAMES]
Definition: cbs_av1.h:425
Coded bitstream fragment structure, combining one or more units.
Definition: cbs.h:116
uint8_t * data
The data buffer.
Definition: buffer.h:89
uint8_t enable_order_hint
Definition: cbs_av1.h:113
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:487
double value
Definition: eval.c:98
Context structure for coded bitstream operations.
Definition: cbs.h:159
static int cbs_av1_write_subexp(CodedBitstreamContext *ctx, PutBitContext *pbc, uint32_t range_max, const char *name, const int *subscripts, uint32_t value)
Definition: cbs_av1.c:497
AVBufferRef * content_ref
If content is reference counted, a reference to the buffer containing content.
Definition: cbs.h:106
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:615
AV1RawTileGroup tile_group
Definition: cbs_av1.h:301
uint8_t obu_type
Definition: cbs_av1.h:31
enum AVCodecID codec_id
Definition: cbs_internal.h:29
AV1RawSequenceHeader * sequence_header
Definition: cbs_av1.h:399
uint8_t spatial_id
Definition: cbs_av1.h:37
int(* read_unit)(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit)
Definition: cbs_internal.h:44
static unsigned int get_bits_long(GetBitContext *s, int n)
Read 0-32 bits.
Definition: get_bits.h:531
static int cbs_av1_write_unit(CodedBitstreamContext *ctx, CodedBitstreamUnit *unit)
Definition: cbs_av1.c:1252
static int cbs_av1_write_ns(CodedBitstreamContext *ctx, PutBitContext *pbc, uint32_t n, const char *name, const int *subscripts, uint32_t value)
Definition: cbs_av1.c:320
AV1RawSequenceHeader sequence_header
Definition: cbs_av1.h:374
#define SIZE_SPECIFIER
Definition: internal.h:264
AVBufferRef * data_ref
Definition: cbs_av1.h:288
void * priv_data
Internal codec-specific data.
Definition: cbs.h:180
AV1RawMetadataITUTT35 itut_t35
Definition: cbs_av1.h:362
common internal api header.
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:101
int(* split_fragment)(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag, int header)
Definition: cbs_internal.h:38
AVBufferRef * data_ref
A reference to the buffer containing data.
Definition: cbs.h:92
AVBufferRef * av_buffer_ref(AVBufferRef *buf)
Create a new reference to an AVBuffer.
Definition: buffer.c:93
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:48
#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
static av_always_inline int diff(const uint32_t a, const uint32_t b)
pixel format definitions
uint64_t metadata_type
Definition: cbs_av1.h:357
int len
uint8_t obu_has_size_field
Definition: cbs_av1.h:33
AV1RawTileGroup tile_group
Definition: cbs_av1.h:377
static int cbs_av1_read_increment(CodedBitstreamContext *ctx, GetBitContext *gbc, uint32_t range_min, uint32_t range_max, const char *name, uint32_t *write_to)
Definition: cbs_av1.c:369
AV1RawFrameHeader frame_header
Definition: cbs_av1.h:375
#define av_freep(p)
union AV1RawOBU::@47 obu
static int cbs_av1_read_leb128(CodedBitstreamContext *ctx, GetBitContext *gbc, const char *name, uint64_t *write_to)
Definition: cbs_av1.c:151
static int cbs_av1_get_relative_dist(const AV1RawSequenceHeader *seq, unsigned int a, unsigned int b)
Definition: cbs_av1.c:569
AVBufferRef * data_ref
A reference to the buffer containing data.
Definition: cbs.h:139
void ff_cbs_trace_syntax_element(CodedBitstreamContext *ctx, int position, const char *str, const int *subscripts, const char *bits, int64_t value)
Definition: cbs.c:376
size_t data_size
The number of bytes in the bitstream (including any padding bits in the final byte).
Definition: cbs.h:80