#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "fdctdsp.h"
#include "put_bits.h"
#include "bytestream.h"
#include "internal.h"
#include "proresdata.h"

Data Structures
struct	prores_profile

struct	TrellisNode

struct	ProresThreadData

struct	ProresContext

Macros
#define	CFACTOR_Y422 2

#define	CFACTOR_Y444 3

#define	MAX_MBS_PER_SLICE 8

#define	MAX_PLANES 4

#define	NUM_MB_LIMITS 4

#define	TRELLIS_WIDTH 16

#define	SCORE_LIMIT INT_MAX / 2

#define	MAX_STORED_Q 16

#define	GET_SIGN(x) ((x) >> 31)

#define	MAKE_CODE(x) ((((x)) * 2) ^ GET_SIGN(x))

#define	OFFSET(x) offsetof(ProresContext, x)

#define	VE AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM

Enumerations
enum	{ PRORES_PROFILE_AUTO = -1, PRORES_PROFILE_PROXY = 0, PRORES_PROFILE_LT, PRORES_PROFILE_STANDARD, PRORES_PROFILE_HQ, PRORES_PROFILE_4444, PRORES_PROFILE_4444XQ }

enum	{ QUANT_MAT_PROXY = 0, QUANT_MAT_LT, QUANT_MAT_STANDARD, QUANT_MAT_HQ, QUANT_MAT_DEFAULT }

Functions
static void	get_slice_data (ProresContext ctx, const uint16_t src, ptrdiff_t linesize, int x, int y, int w, int h, int16_t blocks, uint16_t emu_buf, int mbs_per_slice, int blocks_per_mb, int is_chroma)

static void	get_alpha_data (ProresContext ctx, const uint16_t src, ptrdiff_t linesize, int x, int y, int w, int h, int16_t *blocks, int mbs_per_slice, int abits)

static void	encode_vlc_codeword (PutBitContext *pb, unsigned codebook, int val)
	Write an unsigned rice/exp golomb codeword. More...

static void	encode_dcs (PutBitContext pb, int16_t blocks, int blocks_per_slice, int scale)

static void	encode_acs (PutBitContext pb, int16_t blocks, int blocks_per_slice, int plane_size_factor, const uint8_t scan, const int16_t qmat)

static int	encode_slice_plane (ProresContext ctx, PutBitContext pb, const uint16_t src, ptrdiff_t linesize, int mbs_per_slice, int16_t blocks, int blocks_per_mb, int plane_size_factor, const int16_t *qmat)

static void	put_alpha_diff (PutBitContext *pb, int cur, int prev, int abits)

static void	put_alpha_run (PutBitContext *pb, int run)

static int	encode_alpha_plane (ProresContext ctx, PutBitContext pb, int mbs_per_slice, uint16_t *blocks, int quant)

static int	encode_slice (AVCodecContext avctx, const AVFrame pic, PutBitContext *pb, int sizes[4], int x, int y, int quant, int mbs_per_slice)

static int	estimate_vlc (unsigned codebook, int val)

static int	estimate_dcs (int error, int16_t blocks, int blocks_per_slice, int scale)

static int	estimate_acs (int error, int16_t blocks, int blocks_per_slice, int plane_size_factor, const uint8_t scan, const int16_t qmat)

static int	estimate_slice_plane (ProresContext ctx, int error, int plane, const uint16_t src, ptrdiff_t linesize, int mbs_per_slice, int blocks_per_mb, int plane_size_factor, const int16_t qmat, ProresThreadData *td)

static int	est_alpha_diff (int cur, int prev, int abits)

static int	estimate_alpha_plane (ProresContext ctx, const uint16_t src, ptrdiff_t linesize, int mbs_per_slice, int16_t *blocks)

static int	find_slice_quant (AVCodecContext avctx, int trellis_node, int x, int y, int mbs_per_slice, ProresThreadData td)

static int	find_quant_thread (AVCodecContext avctx, void arg, int jobnr, int threadnr)

static int	encode_frame (AVCodecContext avctx, AVPacket pkt, const AVFrame pic, int got_packet)

static av_cold int	encode_close (AVCodecContext *avctx)

static void	prores_fdct (FDCTDSPContext fdsp, const uint16_t src, ptrdiff_t linesize, int16_t *block)

static av_cold int	encode_init (AVCodecContext *avctx)

Variables
static const uint8_t	prores_quant_matrices [][64]

static const int	prores_mb_limits [NUM_MB_LIMITS]

static const struct prores_profile	prores_profile_info [6]

static const AVOption	options []

static const AVClass	proresenc_class

AVCodec	ff_prores_ks_encoder

Macro Definition Documentation

◆ CFACTOR_Y422

#define CFACTOR_Y422 2

Definition at line 35 of file proresenc_kostya.c.

Referenced by encode_init().

◆ CFACTOR_Y444

#define CFACTOR_Y444 3

Definition at line 36 of file proresenc_kostya.c.

Referenced by encode_init(), encode_slice(), and find_slice_quant().

◆ MAX_MBS_PER_SLICE

#define MAX_MBS_PER_SLICE 8

Definition at line 38 of file proresenc_kostya.c.

◆ MAX_PLANES

#define MAX_PLANES 4

Definition at line 40 of file proresenc_kostya.c.

Referenced by find_slice_quant().

◆ NUM_MB_LIMITS

#define NUM_MB_LIMITS 4

Definition at line 113 of file proresenc_kostya.c.

Referenced by encode_init().

◆ TRELLIS_WIDTH

#define TRELLIS_WIDTH 16

Definition at line 179 of file proresenc_kostya.c.

Referenced by encode_init(), find_quant_thread(), and find_slice_quant().

◆ SCORE_LIMIT

#define SCORE_LIMIT INT_MAX / 2

Definition at line 180 of file proresenc_kostya.c.

Referenced by find_slice_quant().

◆ MAX_STORED_Q

#define MAX_STORED_Q 16

Definition at line 189 of file proresenc_kostya.c.

Referenced by encode_init(), encode_slice(), and find_slice_quant().

◆ GET_SIGN

#define GET_SIGN ( x ) ((x) >> 31)

Definition at line 369 of file proresenc_kostya.c.

Referenced by encode_acs(), encode_dcs(), and estimate_dcs().

◆ MAKE_CODE

#define MAKE_CODE ( x ) ((((x)) * 2) ^ GET_SIGN(x))

Definition at line 370 of file proresenc_kostya.c.

Referenced by encode_dcs(), and estimate_dcs().

◆ OFFSET

#define OFFSET ( x ) offsetof(ProresContext, x)

Definition at line 1299 of file proresenc_kostya.c.

◆ VE

#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 1300 of file proresenc_kostya.c.

Enumeration Type Documentation

◆ anonymous enum

anonymous enum

Enumerator
PRORES_PROFILE_AUTO
PRORES_PROFILE_PROXY
PRORES_PROFILE_LT
PRORES_PROFILE_STANDARD
PRORES_PROFILE_HQ
PRORES_PROFILE_4444
PRORES_PROFILE_4444XQ

Definition at line 42 of file proresenc_kostya.c.

◆ anonymous enum

anonymous enum

Enumerator
QUANT_MAT_PROXY
QUANT_MAT_LT
QUANT_MAT_STANDARD
QUANT_MAT_HQ
QUANT_MAT_DEFAULT

Definition at line 52 of file proresenc_kostya.c.

Function Documentation

◆ get_slice_data()

static void get_slice_data	(	ProresContext *	ctx,
		const uint16_t *	src,
		ptrdiff_t	linesize,
		int	x,
		int	y,
		int	w,
		int	h,
		int16_t *	blocks,
		uint16_t *	emu_buf,
		int	mbs_per_slice,
		int	blocks_per_mb,
		int	is_chroma
	)

static

Definition at line 238 of file proresenc_kostya.c.

Referenced by encode_slice(), and find_slice_quant().

◆ get_alpha_data()

static void get_alpha_data	(	ProresContext *	ctx,
		const uint16_t *	src,
		ptrdiff_t	linesize,
		int	x,
		int	y,
		int	w,
		int	h,
		int16_t *	blocks,
		int	mbs_per_slice,
		int	abits
	)

static

Definition at line 309 of file proresenc_kostya.c.

Referenced by encode_slice(), and find_slice_quant().

◆ encode_vlc_codeword()

static void encode_vlc_codeword	(	PutBitContext *	pb,
		unsigned	codebook,
		int	val
	)

inlinestatic

Write an unsigned rice/exp golomb codeword.

Definition at line 340 of file proresenc_kostya.c.

Referenced by encode_acs(), and encode_dcs().

◆ encode_dcs()

static void encode_dcs	(	PutBitContext *	pb,
		int16_t *	blocks,
		int	blocks_per_slice,
		int	scale
	)

static

Definition at line 372 of file proresenc_kostya.c.

Referenced by encode_slice_plane().

◆ encode_acs()

static void encode_acs	(	PutBitContext *	pb,
		int16_t *	blocks,
		int	blocks_per_slice,
		int	plane_size_factor,
		const uint8_t *	scan,
		const int16_t *	qmat
	)

static

Definition at line 398 of file proresenc_kostya.c.

Referenced by encode_slice_plane().

◆ encode_slice_plane()

static int encode_slice_plane	(	ProresContext *	ctx,
		PutBitContext *	pb,
		const uint16_t *	src,
		ptrdiff_t	linesize,
		int	mbs_per_slice,
		int16_t *	blocks,
		int	blocks_per_mb,
		int	plane_size_factor,
		const int16_t *	qmat
	)

static

Definition at line 432 of file proresenc_kostya.c.

Referenced by encode_slice().

◆ put_alpha_diff()

static void put_alpha_diff	(	PutBitContext *	pb,
		int	cur,
		int	prev,
		int	abits
	)

static

Definition at line 451 of file proresenc_kostya.c.

Referenced by encode_alpha_plane().

◆ put_alpha_run()

static void put_alpha_run	(	PutBitContext *	pb,
		int	run
	)

static

Definition at line 470 of file proresenc_kostya.c.

Referenced by encode_alpha_plane().

◆ encode_alpha_plane()

static int encode_alpha_plane	(	ProresContext *	ctx,
		PutBitContext *	pb,
		int	mbs_per_slice,
		uint16_t *	blocks,
		int	quant
	)

static

Definition at line 484 of file proresenc_kostya.c.

Referenced by encode_slice().

◆ encode_slice()

static int encode_slice	(	AVCodecContext *	avctx,
		const AVFrame *	pic,
		PutBitContext *	pb,
		int	sizes[4],
		int	x,
		int	y,
		int	quant,
		int	mbs_per_slice
	)

static

Definition at line 516 of file proresenc_kostya.c.

Referenced by encode_frame().

◆ estimate_vlc()

static int estimate_vlc	(	unsigned	codebook,
		int	val
	)

inlinestatic

Definition at line 593 of file proresenc_kostya.c.

Referenced by estimate_acs(), and estimate_dcs().

◆ estimate_dcs()

static int estimate_dcs	(	int *	error,
		int16_t *	blocks,
		int	blocks_per_slice,
		int	scale
	)

static

Definition at line 615 of file proresenc_kostya.c.

Referenced by estimate_slice_plane().

◆ estimate_acs()

static int estimate_acs	(	int *	error,
		int16_t *	blocks,
		int	blocks_per_slice,
		int	plane_size_factor,
		const uint8_t *	scan,
		const int16_t *	qmat
	)

static

Definition at line 646 of file proresenc_kostya.c.

Referenced by estimate_slice_plane().

◆ estimate_slice_plane()

static int estimate_slice_plane	(	ProresContext *	ctx,
		int *	error,
		int	plane,
		const uint16_t *	src,
		ptrdiff_t	linesize,
		int	mbs_per_slice,
		int	blocks_per_mb,
		int	plane_size_factor,
		const int16_t *	qmat,
		ProresThreadData *	td
	)

static

Definition at line 682 of file proresenc_kostya.c.

Referenced by find_slice_quant().

◆ est_alpha_diff()

static int est_alpha_diff	(	int	cur,
		int	prev,
		int	abits
	)

static

Definition at line 700 of file proresenc_kostya.c.

Referenced by estimate_alpha_plane().

◆ estimate_alpha_plane()

static int estimate_alpha_plane	(	ProresContext *	ctx,
		const uint16_t *	src,
		ptrdiff_t	linesize,
		int	mbs_per_slice,
		int16_t *	blocks
	)

static

Definition at line 715 of file proresenc_kostya.c.

Referenced by find_slice_quant().

◆ find_slice_quant()

static int find_slice_quant	(	AVCodecContext *	avctx,
		int	trellis_node,
		int	x,
		int	y,
		int	mbs_per_slice,
		ProresThreadData *	td
	)

static

Definition at line 757 of file proresenc_kostya.c.

Referenced by find_quant_thread().

◆ find_quant_thread()

static int find_quant_thread	(	AVCodecContext *	avctx,
		void *	arg,
		int	jobnr,
		int	threadnr
	)

static

Definition at line 911 of file proresenc_kostya.c.

Referenced by encode_frame().

◆ encode_frame()

static int encode_frame	(	AVCodecContext *	avctx,
		AVPacket *	pkt,
		const AVFrame *	pic,
		int *	got_packet
	)

static

Definition at line 935 of file proresenc_kostya.c.

◆ encode_close()

static av_cold int encode_close ( AVCodecContext * avctx )

static

Definition at line 1096 of file proresenc_kostya.c.

Referenced by encode_init().

◆ prores_fdct()

static void prores_fdct	(	FDCTDSPContext *	fdsp,
		const uint16_t *	src,
		ptrdiff_t	linesize,
		int16_t *	block
	)

static

Definition at line 1111 of file proresenc_kostya.c.

Referenced by encode_init().

◆ encode_init()

static av_cold int encode_init ( AVCodecContext * avctx )

static

Definition at line 1125 of file proresenc_kostya.c.

Variable Documentation

◆ prores_quant_matrices

const uint8_t prores_quant_matrices[][64]

static

Definition at line 60 of file proresenc_kostya.c.

Referenced by encode_init().

◆ prores_mb_limits

const int prores_mb_limits[NUM_MB_LIMITS]

static

Initial value:

Definition at line 114 of file proresenc_kostya.c.

Referenced by encode_init().

◆ prores_profile_info

const struct prores_profile prores_profile_info[6]

static

Referenced by encode_init().

◆ options

const AVOption options[]

static

Definition at line 1302 of file proresenc_kostya.c.

◆ proresenc_class

const AVClass proresenc_class

static

Initial value:

= {
    .class_name = "ProRes encoder",
    .item_name  = av_default_item_name,
    .option     = options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 1345 of file proresenc_kostya.c.

◆ ff_prores_ks_encoder

AVCodec ff_prores_ks_encoder

Initial value:

= {
    .name           = "prores_ks",
    .long_name      = NULL_IF_CONFIG_SMALL("Apple ProRes (iCodec Pro)"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_PRORES,
    .priv_data_size = sizeof(ProresContext),
    .init           = encode_init,
    .close          = encode_close,
    .encode2        = encode_frame,
    .capabilities   = AV_CODEC_CAP_SLICE_THREADS | AV_CODEC_CAP_FRAME_THREADS | AV_CODEC_CAP_INTRA_ONLY,
    .pix_fmts       = (const enum AVPixelFormat[]) {
                          AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10,
                          AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_NONE
                      },
    .priv_class     = &proresenc_class,
}

Definition at line 1352 of file proresenc_kostya.c.

Data Structures

Macros

Enumerations

Functions

Variables

Macro Definition Documentation

◆ CFACTOR_Y422

◆ CFACTOR_Y444

◆ MAX_MBS_PER_SLICE

◆ MAX_PLANES

◆ NUM_MB_LIMITS

◆ TRELLIS_WIDTH

◆ SCORE_LIMIT

◆ MAX_STORED_Q

◆ GET_SIGN

◆ MAKE_CODE

◆ OFFSET

◆ VE

Enumeration Type Documentation

◆ anonymous enum

◆ anonymous enum

Function Documentation

◆ get_slice_data()

◆ get_alpha_data()

◆ encode_vlc_codeword()

◆ encode_dcs()

◆ encode_acs()

◆ encode_slice_plane()

◆ put_alpha_diff()

◆ put_alpha_run()

◆ encode_alpha_plane()

◆ encode_slice()

◆ estimate_vlc()

◆ estimate_dcs()

◆ estimate_acs()

◆ estimate_slice_plane()

◆ est_alpha_diff()

◆ estimate_alpha_plane()

◆ find_slice_quant()

◆ find_quant_thread()

◆ encode_frame()

◆ encode_close()

◆ prores_fdct()

◆ encode_init()

Variable Documentation

◆ prores_quant_matrices

◆ prores_mb_limits

◆ prores_profile_info

◆ options

◆ proresenc_class

◆ ff_prores_ks_encoder