#include "libavutil/avassert.h"
#include "libavutil/pixfmt.h"
#include "cbs.h"
#include "cbs_internal.h"
#include "cbs_av1.h"
#include "internal.h"
#include "cbs_av1_syntax_template.c"

Macros
#define	HEADER(name)

#define	CHECK(call)

#define	FUNC_NAME(rw, codec, name) cbs_ ## codec ## _ ## rw ## _ ## name

#define	FUNC_AV1(rw, name) FUNC_NAME(rw, av1, name)

#define	FUNC(name) FUNC_AV1(READWRITE, name)

#define	SUBSCRIPTS(subs, ...) (subs > 0 ? ((int[subs + 1]){ subs, __VA_ARGS__ }) : NULL)

#define	fb(width, name) xf(width, name, current->name, 0, MAX_UINT_BITS(width), 0)

#define	fc(width, name, range_min, range_max) xf(width, name, current->name, range_min, range_max, 0)

#define	flag(name) fb(1, name)

#define	su(width, name) xsu(width, name, current->name, 0)

#define	fbs(width, name, subs, ...) xf(width, name, current->name, 0, MAX_UINT_BITS(width), subs, __VA_ARGS__)

#define	fcs(width, name, range_min, range_max, subs, ...) xf(width, name, current->name, range_min, range_max, subs, __VA_ARGS__)

#define	flags(name, subs, ...) xf(1, name, current->name, 0, 1, subs, __VA_ARGS__)

#define	sus(width, name, subs, ...) xsu(width, name, current->name, subs, __VA_ARGS__)

#define	fixed(width, name, value)

#define	READ

#define	READWRITE read

#define	RWContext GetBitContext

#define	xf(width, name, var, range_min, range_max, subs, ...)

#define	xsu(width, name, var, subs, ...)

#define	uvlc(name, range_min, range_max)

#define	ns(max_value, name, subs, ...)

#define	increment(name, min, max)

#define	subexp(name, max, subs, ...)

#define	delta_q(name)

#define	leb128(name)

#define	infer(name, value)

#define	byte_alignment(rw) (get_bits_count(rw) % 8)

#define	WRITE

#define	READWRITE write

#define	RWContext PutBitContext

#define	xf(width, name, var, range_min, range_max, subs, ...)

#define	xsu(width, name, var, subs, ...)

#define	uvlc(name, range_min, range_max)

#define	ns(max_value, name, subs, ...)

#define	increment(name, min, max)

#define	subexp(name, max, subs, ...)

#define	delta_q(name)

#define	leb128(name)

#define	infer(name, value)

#define	byte_alignment(rw) (put_bits_count(rw) % 8)

Functions
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)

static int	cbs_av1_write_uvlc (CodedBitstreamContext ctx, PutBitContext pbc, const char *name, uint32_t value, uint32_t range_min, uint32_t range_max)

static int	cbs_av1_read_leb128 (CodedBitstreamContext ctx, GetBitContext gbc, const char name, uint64_t write_to)

static int	cbs_av1_write_leb128 (CodedBitstreamContext ctx, PutBitContext pbc, const char *name, uint64_t value)

static int	cbs_av1_read_ns (CodedBitstreamContext ctx, GetBitContext gbc, uint32_t n, const char name, const int subscripts, uint32_t *write_to)

static int	cbs_av1_write_ns (CodedBitstreamContext ctx, PutBitContext pbc, uint32_t n, const char name, const int subscripts, uint32_t value)

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)

static int	cbs_av1_write_increment (CodedBitstreamContext ctx, PutBitContext pbc, uint32_t range_min, uint32_t range_max, const char *name, uint32_t value)

static int	cbs_av1_read_subexp (CodedBitstreamContext ctx, GetBitContext gbc, uint32_t range_max, const char name, const int subscripts, uint32_t *write_to)

static int	cbs_av1_write_subexp (CodedBitstreamContext ctx, PutBitContext pbc, uint32_t range_max, const char name, const int subscripts, uint32_t value)

static int	cbs_av1_tile_log2 (int blksize, int target)

static int	cbs_av1_get_relative_dist (const AV1RawSequenceHeader *seq, unsigned int a, unsigned int b)

static size_t	cbs_av1_get_payload_bytes_left (GetBitContext *gbc)

static int	cbs_av1_split_fragment (CodedBitstreamContext ctx, CodedBitstreamFragment frag, int header)

static void	cbs_av1_free_tile_data (AV1RawTileData *td)

static void	cbs_av1_free_padding (AV1RawPadding *pd)

static void	cbs_av1_free_metadata (AV1RawMetadata *md)

static void	cbs_av1_free_obu (void unit, uint8_t content)

static int	cbs_av1_ref_tile_data (CodedBitstreamContext ctx, CodedBitstreamUnit unit, GetBitContext gbc, AV1RawTileData td)

static int	cbs_av1_read_unit (CodedBitstreamContext ctx, CodedBitstreamUnit unit)

static int	cbs_av1_write_obu (CodedBitstreamContext ctx, CodedBitstreamUnit unit, PutBitContext *pbc)

static int	cbs_av1_assemble_fragment (CodedBitstreamContext ctx, CodedBitstreamFragment frag)

static void	cbs_av1_close (CodedBitstreamContext *ctx)

Variables
const CodedBitstreamType	ff_cbs_type_av1

Macro Definition Documentation

◆ HEADER

#define HEADER ( name )

Value:

        do { \
        ff_cbs_trace_header(ctx, name); \
    } while (0)

Definition at line 537 of file cbs_av1.c.

◆ CHECK

#define CHECK ( call )

Value:

        do { \
        err = (call); \
        if (err < 0) \
            return err; \
    } while (0)

Definition at line 541 of file cbs_av1.c.

◆ FUNC_NAME

#define FUNC_NAME	(	rw,
		codec,
		name
	)	cbs_ ## codec ## _ ## rw ## _ ## name

Definition at line 547 of file cbs_av1.c.

◆ FUNC_AV1

#define FUNC_AV1	(	rw,
		name
	)	FUNC_NAME(rw, av1, name)

Definition at line 548 of file cbs_av1.c.

◆ FUNC

#define FUNC ( name ) FUNC_AV1(READWRITE, name)

Definition at line 549 of file cbs_av1.c.

◆ SUBSCRIPTS

#define SUBSCRIPTS	(	subs,
		...
	)	(subs > 0 ? ((int[subs + 1]){ subs, __VA_ARGS__ }) : NULL)

Definition at line 551 of file cbs_av1.c.

◆ fb

#define fb	(	width,
		name
	)	xf(width, name, current->name, 0, MAX_UINT_BITS(width), 0)

Definition at line 553 of file cbs_av1.c.

◆ fc

#define fc	(	width,
		name,
		range_min,
		range_max
	)	xf(width, name, current->name, range_min, range_max, 0)

Definition at line 555 of file cbs_av1.c.

◆ flag

#define flag ( name ) fb(1, name)

Definition at line 557 of file cbs_av1.c.

◆ su

#define su	(	width,
		name
	)	xsu(width, name, current->name, 0)

Definition at line 558 of file cbs_av1.c.

◆ fbs

#define fbs	(	width,
		name,
		subs,
		...
	)	xf(width, name, current->name, 0, MAX_UINT_BITS(width), subs, __VA_ARGS__)

Definition at line 561 of file cbs_av1.c.

◆ fcs

#define fcs	(	width,
		name,
		range_min,
		range_max,
		subs,
		...
	)	xf(width, name, current->name, range_min, range_max, subs, __VA_ARGS__)

Definition at line 563 of file cbs_av1.c.

◆ flags

#define flags	(	name,
		subs,
		...
	)	xf(1, name, current->name, 0, 1, subs, __VA_ARGS__)

Examples: ffhash.c.

Definition at line 565 of file cbs_av1.c.

◆ sus

#define sus	(	width,
		name,
		subs,
		...
	)	xsu(width, name, current->name, subs, __VA_ARGS__)

Definition at line 567 of file cbs_av1.c.

◆ fixed

#define fixed	(	width,
		name,
		value
	)

Value:

        do { \
        av_unused uint32_t fixed_value = value; \
        xf(width, name, fixed_value, value, value, 0); \
    } while (0)

Definition at line 570 of file cbs_av1.c.

◆ READ

#define READ

Definition at line 576 of file cbs_av1.c.

◆ READWRITE [1/2]

#define READWRITE read

Definition at line 665 of file cbs_av1.c.

◆ RWContext [1/2]

#define RWContext GetBitContext

Definition at line 666 of file cbs_av1.c.

◆ xf [1/2]

#define xf	(	width,
		name,
		var,
		range_min,
		range_max,
		subs,
		...
	)

Value:

        do { \
        uint32_t value; \
        CHECK(ff_cbs_read_unsigned(ctx, rw, width, #name, \
                                   SUBSCRIPTS(subs, __VA_ARGS__), \
                                   &value, range_min, range_max)); \
        var = value; \
    } while (0)

Definition at line 668 of file cbs_av1.c.

◆ xsu [1/2]

#define xsu	(	width,
		name,
		var,
		subs,
		...
	)

Value:

        do { \
        int32_t value; \
        CHECK(ff_cbs_read_signed(ctx, rw, width, #name, \
                                 SUBSCRIPTS(subs, __VA_ARGS__), &value, \
                                 MIN_INT_BITS(width), \
                                 MAX_INT_BITS(width))); \
        var = value; \
    } while (0)

Definition at line 674 of file cbs_av1.c.

◆ uvlc [1/2]

#define uvlc	(	name,
		range_min,
		range_max
	)

Value:

        do { \
        uint32_t value; \
        CHECK(cbs_av1_read_uvlc(ctx, rw, #name, \
                                &value, range_min, range_max)); \
        current->name = value; \
    } while (0)

Definition at line 681 of file cbs_av1.c.

◆ ns [1/2]

#define ns	(	max_value,
		name,
		subs,
		...
	)

Value:

        do { \
        uint32_t value; \
        CHECK(cbs_av1_read_ns(ctx, rw, max_value, #name, \
                              SUBSCRIPTS(subs, __VA_ARGS__), &value)); \
        current->name = value; \
    } while (0)

Definition at line 686 of file cbs_av1.c.

◆ increment [1/2]

#define increment	(	name,
		min,
		max
	)

Value:

        do { \
        uint32_t value; \
        CHECK(cbs_av1_read_increment(ctx, rw, min, max, #name, &value)); \
        current->name = value; \
    } while (0)

Definition at line 692 of file cbs_av1.c.

◆ subexp [1/2]

#define subexp	(	name,
		max,
		subs,
		...
	)

Value:

        do { \
        uint32_t value; \
        CHECK(cbs_av1_read_subexp(ctx, rw, max, #name, \
                                  SUBSCRIPTS(subs, __VA_ARGS__), &value)); \
        current->name = value; \
    } while (0)

Definition at line 697 of file cbs_av1.c.

◆ delta_q [1/2]

#define delta_q ( name )

Value:

        do { \
        uint8_t delta_coded; \
        int8_t delta_q; \
        xf(1, name.delta_coded, delta_coded, 0, 1, 0); \
        if (delta_coded) \
            xsu(1 + 6, name.delta_q, delta_q, 0); \
        else \
            delta_q = 0; \
        current->name = delta_q; \
    } while (0)

Definition at line 703 of file cbs_av1.c.

◆ leb128 [1/2]

#define leb128 ( name )

Value:

        do { \
        uint64_t value; \
        CHECK(cbs_av1_read_leb128(ctx, rw, #name, &value)); \
        current->name = value; \
    } while (0)

Definition at line 709 of file cbs_av1.c.

◆ infer [1/2]

#define infer	(	name,
		value
	)

Value:

        do { \
        current->name = value; \
    } while (0)

Definition at line 713 of file cbs_av1.c.

◆ byte_alignment [1/2]

#define byte_alignment ( rw ) (get_bits_count(rw) % 8)

Definition at line 722 of file cbs_av1.c.

◆ WRITE

#define WRITE

Definition at line 664 of file cbs_av1.c.

◆ READWRITE [2/2]

#define READWRITE write

Definition at line 665 of file cbs_av1.c.

◆ RWContext [2/2]

#define RWContext PutBitContext

Definition at line 666 of file cbs_av1.c.

◆ xf [2/2]

#define xf	(	width,
		name,
		var,
		range_min,
		range_max,
		subs,
		...
	)

Value:

        do { \
        CHECK(ff_cbs_write_unsigned(ctx, rw, width, #name, \
                                    SUBSCRIPTS(subs, __VA_ARGS__), \
                                    var, range_min, range_max)); \
    } while (0)

Definition at line 668 of file cbs_av1.c.

◆ xsu [2/2]

#define xsu	(	width,
		name,
		var,
		subs,
		...
	)

Value:

        do { \
        CHECK(ff_cbs_write_signed(ctx, rw, width, #name, \
                                  SUBSCRIPTS(subs, __VA_ARGS__), var, \
                                  MIN_INT_BITS(width), \
                                  MAX_INT_BITS(width))); \
    } while (0)

Definition at line 674 of file cbs_av1.c.

◆ uvlc [2/2]

#define uvlc	(	name,
		range_min,
		range_max
	)

Value:

        do { \
        CHECK(cbs_av1_write_uvlc(ctx, rw, #name, current->name, \
                                 range_min, range_max)); \
    } while (0)

Definition at line 681 of file cbs_av1.c.

◆ ns [2/2]

#define ns	(	max_value,
		name,
		subs,
		...
	)

Value:

        do { \
        CHECK(cbs_av1_write_ns(ctx, rw, max_value, #name, \
                               SUBSCRIPTS(subs, __VA_ARGS__), \
                               current->name)); \
    } while (0)

Definition at line 686 of file cbs_av1.c.

◆ increment [2/2]

#define increment	(	name,
		min,
		max
	)

Value:

        do { \
        CHECK(cbs_av1_write_increment(ctx, rw, min, max, #name, \
                                      current->name)); \
    } while (0)

Definition at line 692 of file cbs_av1.c.

◆ subexp [2/2]

#define subexp	(	name,
		max,
		subs,
		...
	)

Value:

        do { \
        CHECK(cbs_av1_write_subexp(ctx, rw, max, #name, \
                                   SUBSCRIPTS(subs, __VA_ARGS__), \
                                   current->name)); \
    } while (0)

Definition at line 697 of file cbs_av1.c.

◆ delta_q [2/2]

#define delta_q ( name )

Value:

        do { \
        xf(1, name.delta_coded, current->name != 0, 0, 1, 0); \
        if (current->name) \
            xsu(1 + 6, name.delta_q, current->name, 0); \
    } while (0)

Definition at line 703 of file cbs_av1.c.

◆ leb128 [2/2]

#define leb128 ( name )

Value:

        do { \
        CHECK(cbs_av1_write_leb128(ctx, rw, #name, current->name)); \
    } while (0)

Definition at line 709 of file cbs_av1.c.

◆ infer [2/2]

#define infer	(	name,
		value
	)

Value:

        do { \
        if (current->name != (value)) { \
            av_log(ctx->log_ctx, AV_LOG_WARNING, "Warning: " \
                   "%s does not match inferred value: " \
                   "%"PRId64", but should be %"PRId64".\n", \
                   #name, (int64_t)current->name, (int64_t)(value)); \
        } \
    } while (0)

Definition at line 713 of file cbs_av1.c.

◆ byte_alignment [2/2]

#define byte_alignment ( rw ) (put_bits_count(rw) % 8)

Definition at line 722 of file cbs_av1.c.

Function Documentation

◆ cbs_av1_read_uvlc()

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
	)

static

Definition at line 28 of file cbs_av1.c.

◆ cbs_av1_write_uvlc()

static int cbs_av1_write_uvlc	(	CodedBitstreamContext *	ctx,
		PutBitContext *	pbc,
		const char *	name,
		uint32_t	value,
		uint32_t	range_min,
		uint32_t	range_max
	)

static

Definition at line 106 of file cbs_av1.c.

◆ cbs_av1_read_leb128()

static int cbs_av1_read_leb128	(	CodedBitstreamContext *	ctx,
		GetBitContext *	gbc,
		const char *	name,
		uint64_t *	write_to
	)

static

Definition at line 151 of file cbs_av1.c.

Referenced by cbs_av1_read_unit(), and cbs_av1_split_fragment().

◆ cbs_av1_write_leb128()

static int cbs_av1_write_leb128	(	CodedBitstreamContext *	ctx,
		PutBitContext *	pbc,
		const char *	name,
		uint64_t	value
	)

static

Definition at line 184 of file cbs_av1.c.

Referenced by cbs_av1_write_obu().

◆ cbs_av1_read_ns()

static int cbs_av1_read_ns	(	CodedBitstreamContext *	ctx,
		GetBitContext *	gbc,
		uint32_t	n,
		const char *	name,
		const int *	subscripts,
		uint32_t *	write_to
	)

static

Definition at line 214 of file cbs_av1.c.

Referenced by cbs_av1_read_subexp().

◆ cbs_av1_write_ns()

static int cbs_av1_write_ns	(	CodedBitstreamContext *	ctx,
		PutBitContext *	pbc,
		uint32_t	n,
		const char *	name,
		const int *	subscripts,
		uint32_t	value
	)

static

Definition at line 264 of file cbs_av1.c.

Referenced by cbs_av1_write_subexp().

◆ cbs_av1_read_increment()

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
	)

static

Definition at line 313 of file cbs_av1.c.

Referenced by cbs_av1_read_subexp().

◆ cbs_av1_write_increment()

static int cbs_av1_write_increment	(	CodedBitstreamContext *	ctx,
		PutBitContext *	pbc,
		uint32_t	range_min,
		uint32_t	range_max,
		const char *	name,
		uint32_t	value
	)

static

Definition at line 350 of file cbs_av1.c.

Referenced by cbs_av1_write_subexp().

◆ cbs_av1_read_subexp()

static int cbs_av1_read_subexp	(	CodedBitstreamContext *	ctx,
		GetBitContext *	gbc,
		uint32_t	range_max,
		const char *	name,
		const int *	subscripts,
		uint32_t *	write_to
	)

static

Definition at line 391 of file cbs_av1.c.

◆ cbs_av1_write_subexp()

static int cbs_av1_write_subexp	(	CodedBitstreamContext *	ctx,
		PutBitContext *	pbc,
		uint32_t	range_max,
		const char *	name,
		const int *	subscripts,
		uint32_t	value
	)

static

Definition at line 441 of file cbs_av1.c.

◆ cbs_av1_tile_log2()

static int cbs_av1_tile_log2	(	int	blksize,
		int	target
	)

static

Definition at line 506 of file cbs_av1.c.

Referenced by tile_group_obu(), and tile_info().

◆ cbs_av1_get_relative_dist()

static int cbs_av1_get_relative_dist	(	const AV1RawSequenceHeader *	seq,
		unsigned int	a,
		unsigned int	b
	)

static

Definition at line 513 of file cbs_av1.c.

Referenced by skip_mode_params().

◆ cbs_av1_get_payload_bytes_left()

static size_t cbs_av1_get_payload_bytes_left ( GetBitContext * gbc )

static

Definition at line 525 of file cbs_av1.c.

Referenced by metadata_itut_t35(), and padding_obu().

◆ cbs_av1_split_fragment()

static int cbs_av1_split_fragment	(	CodedBitstreamContext *	ctx,
		CodedBitstreamFragment *	frag,
		int	header
	)

static

Definition at line 741 of file cbs_av1.c.

◆ cbs_av1_free_tile_data()

static void cbs_av1_free_tile_data ( AV1RawTileData * td )

static

Definition at line 817 of file cbs_av1.c.

Referenced by cbs_av1_free_obu().

◆ cbs_av1_free_padding()

static void cbs_av1_free_padding ( AV1RawPadding * pd )

static

Definition at line 822 of file cbs_av1.c.

Referenced by cbs_av1_free_obu().

◆ cbs_av1_free_metadata()

static void cbs_av1_free_metadata ( AV1RawMetadata * md )

static

Definition at line 827 of file cbs_av1.c.

Referenced by cbs_av1_free_obu().

◆ cbs_av1_free_obu()

static void cbs_av1_free_obu	(	void *	unit,
		uint8_t *	content
	)

static

Definition at line 836 of file cbs_av1.c.

Referenced by cbs_av1_read_unit().

◆ cbs_av1_ref_tile_data()

static int cbs_av1_ref_tile_data	(	CodedBitstreamContext *	ctx,
		CodedBitstreamUnit *	unit,
		GetBitContext *	gbc,
		AV1RawTileData *	td
	)

static

Definition at line 861 of file cbs_av1.c.

Referenced by cbs_av1_read_unit().

◆ cbs_av1_read_unit()

static int cbs_av1_read_unit	(	CodedBitstreamContext *	ctx,
		CodedBitstreamUnit *	unit
	)

static

Definition at line 887 of file cbs_av1.c.

◆ cbs_av1_write_obu()

static int cbs_av1_write_obu	(	CodedBitstreamContext *	ctx,
		CodedBitstreamUnit *	unit,
		PutBitContext *	pbc
	)

static

Definition at line 1062 of file cbs_av1.c.

◆ cbs_av1_assemble_fragment()

static int cbs_av1_assemble_fragment	(	CodedBitstreamContext *	ctx,
		CodedBitstreamFragment *	frag
	)

static

Definition at line 1229 of file cbs_av1.c.

◆ cbs_av1_close()

static void cbs_av1_close ( CodedBitstreamContext * ctx )

static

Definition at line 1257 of file cbs_av1.c.

Variable Documentation

◆ ff_cbs_type_av1

const CodedBitstreamType ff_cbs_type_av1

Initial value:

= {
    .codec_id          = AV_CODEC_ID_AV1,
 
    .priv_data_size    = sizeof(CodedBitstreamAV1Context),
 
    .split_fragment    = &cbs_av1_split_fragment,
    .read_unit         = &cbs_av1_read_unit,
    .write_unit        = &cbs_av1_write_obu,
    .assemble_fragment = &cbs_av1_assemble_fragment,
 
    .close             = &cbs_av1_close,
}

Definition at line 1265 of file cbs_av1.c.

Macros

Functions

Variables

Macro Definition Documentation

◆ HEADER

◆ CHECK

◆ FUNC_NAME

◆ FUNC_AV1

◆ FUNC

◆ SUBSCRIPTS

◆ fb

◆ fc

◆ flag

◆ su

◆ fbs

◆ fcs

◆ flags

◆ sus

◆ fixed

◆ READ

◆ READWRITE [1/2]

◆ RWContext [1/2]

◆ xf [1/2]

◆ xsu [1/2]

◆ uvlc [1/2]

◆ ns [1/2]

◆ increment [1/2]

◆ subexp [1/2]

◆ delta_q [1/2]

◆ leb128 [1/2]

◆ infer [1/2]

◆ byte_alignment [1/2]

◆ WRITE

◆ READWRITE [2/2]

◆ RWContext [2/2]

◆ xf [2/2]

◆ xsu [2/2]

◆ uvlc [2/2]

◆ ns [2/2]

◆ increment [2/2]

◆ subexp [2/2]

◆ delta_q [2/2]

◆ leb128 [2/2]

◆ infer [2/2]

◆ byte_alignment [2/2]

Function Documentation

◆ cbs_av1_read_uvlc()

◆ cbs_av1_write_uvlc()

◆ cbs_av1_read_leb128()

◆ cbs_av1_write_leb128()

◆ cbs_av1_read_ns()

◆ cbs_av1_write_ns()

◆ cbs_av1_read_increment()

◆ cbs_av1_write_increment()

◆ cbs_av1_read_subexp()

◆ cbs_av1_write_subexp()

◆ cbs_av1_tile_log2()

◆ cbs_av1_get_relative_dist()

◆ cbs_av1_get_payload_bytes_left()

◆ cbs_av1_split_fragment()

◆ cbs_av1_free_tile_data()

◆ cbs_av1_free_padding()

◆ cbs_av1_free_metadata()

◆ cbs_av1_free_obu()

◆ cbs_av1_ref_tile_data()

◆ cbs_av1_read_unit()

◆ cbs_av1_write_obu()

◆ cbs_av1_assemble_fragment()

◆ cbs_av1_close()

Variable Documentation

◆ ff_cbs_type_av1