proresenc_anatoliy.c File Reference

#include "libavutil/mem_internal.h"
#include "libavutil/opt.h"
#include "avcodec.h"
#include "dct.h"
#include "encode.h"
#include "internal.h"
#include "profiles.h"
#include "proresdata.h"
#include "put_bits.h"
#include "bytestream.h"
#include "fdctdsp.h"

Go to the source code of this file.

Data Structures
struct	ProresContext

Macros
#define	DEFAULT_SLICE_MB_WIDTH   8
#define	QSCALE(qmat, ind, val)   ((val) / ((qmat)[ind]))
#define	TO_GOLOMB(val)   (((val) * 2) ^ ((val) >> 31))
#define	DIFF_SIGN(val, sign)   (((val) >> 31) ^ (sign))
#define	IS_NEGATIVE(val)   ((((val) >> 31) ^ -1) + 1)
#define	TO_GOLOMB2(val, sign)   ((val)==0 ? 0 : ((val) << 1) + (sign))
#define	FIRST_DC_CB   0xB8
#define	OFFSET(x)   offsetof(ProresContext, x)
#define	VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Functions
static void	encode_codeword (PutBitContext *pb, int val, int codebook)
static av_always_inline int	get_level (int val)
static void	encode_dc_coeffs (PutBitContext *pb, int16_t *in, int blocks_per_slice, int *qmat)
static void	encode_ac_coeffs (PutBitContext *pb, int16_t *in, int blocks_per_slice, int *qmat, const uint8_t ff_prores_scan[64])
static void	get (uint8_t *pixels, int stride, int16_t *block)
static void	fdct_get (FDCTDSPContext *fdsp, uint8_t *pixels, int stride, int16_t *block)
static void	calc_plane_dct (FDCTDSPContext *fdsp, uint8_t *src, int16_t *blocks, int src_stride, int mb_count, int chroma, int is_422)
static int	encode_slice_plane (int16_t *blocks, int mb_count, uint8_t *buf, unsigned buf_size, int *qmat, int sub_sample_chroma, const uint8_t ff_prores_scan[64])
static av_always_inline unsigned	encode_slice_data (AVCodecContext *avctx, int16_t *blocks_y, int16_t *blocks_u, int16_t *blocks_v, unsigned mb_count, uint8_t *buf, unsigned data_size, unsigned *y_data_size, unsigned *u_data_size, unsigned *v_data_size, int qp)
static void	put_alpha_diff (PutBitContext *pb, int cur, int prev)
static void	put_alpha_run (PutBitContext *pb, int run)
static av_always_inline int	encode_alpha_slice_data (AVCodecContext *avctx, int8_t *src_a, unsigned mb_count, uint8_t *buf, unsigned data_size, unsigned *a_data_size)
static void	subimage_with_fill_template (uint16_t *src, unsigned x, unsigned y, unsigned stride, unsigned width, unsigned height, uint16_t *dst, unsigned dst_width, unsigned dst_height, int is_alpha_plane, int is_interlaced, int is_top_field)
static void	subimage_with_fill (uint16_t *src, unsigned x, unsigned y, unsigned stride, unsigned width, unsigned height, uint16_t *dst, unsigned dst_width, unsigned dst_height, int is_interlaced, int is_top_field)
static void	subimage_alpha_with_fill (uint16_t *src, unsigned x, unsigned y, unsigned stride, unsigned width, unsigned height, uint16_t *dst, unsigned dst_width, unsigned dst_height, int is_interlaced, int is_top_field)
static int	encode_slice (AVCodecContext *avctx, const AVFrame *pic, int mb_x, int mb_y, unsigned mb_count, uint8_t *buf, unsigned data_size, int unsafe, int *qp, int is_interlaced, int is_top_field)
static int	prores_encode_picture (AVCodecContext *avctx, const AVFrame *pic, uint8_t *buf, const int buf_size, const int picture_index, const int is_top_field)
static int	prores_encode_frame (AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pict, int *got_packet)
static void	scale_mat (const uint8_t *src, int *dst, int scale)
static av_cold int	prores_encode_init (AVCodecContext *avctx)
static av_cold int	prores_encode_close (AVCodecContext *avctx)

Variables
static const AVProfile	profiles []
static const int	qp_start_table [] = { 8, 3, 2, 1, 1, 1}
static const int	qp_end_table [] = { 13, 9, 6, 6, 5, 4}
static const int	bitrate_table [] = { 1000, 2100, 3500, 5400, 7000, 10000}
static const int	valid_primaries []
static const int	valid_trc []
static const int	valid_colorspace []
static const uint8_t	QMAT_LUMA [6][64]
static const uint8_t	QMAT_CHROMA [6][64]
static const uint8_t	dc_codebook [7] = { 0x04, 0x28, 0x28, 0x4D, 0x4D, 0x70, 0x70}
static const uint8_t	run_to_cb [16]
static const uint8_t	lev_to_cb [10]
static const AVOption	options []
static const AVClass	prores_enc_class
static enum AVPixelFormat	pix_fmts []
const AVCodec	ff_prores_aw_encoder
const AVCodec	ff_prores_encoder

Detailed Description

Apple ProRes encoder (Anatoliy Wasserman version) Known FOURCCs: 'ap4h' (444), 'apch' (HQ), 'apcn' (422), 'apcs' (LT), 'acpo' (Proxy)

Definition in file proresenc_anatoliy.c.

Macro Definition Documentation

DEFAULT_SLICE_MB_WIDTH

#define DEFAULT_SLICE_MB_WIDTH   8

Definition at line 42 of file proresenc_anatoliy.c.

QSCALE

#define QSCALE	(		qmat,
			ind,
			val
	)		((val) / ((qmat)[ind]))

Definition at line 229 of file proresenc_anatoliy.c.

TO_GOLOMB

#define TO_GOLOMB

(

val

)

(((val) * 2) ^ ((val) >> 31))

Definition at line 230 of file proresenc_anatoliy.c.

DIFF_SIGN

#define DIFF_SIGN	(		val,
			sign
	)		(((val) >> 31) ^ (sign))

Definition at line 231 of file proresenc_anatoliy.c.

IS_NEGATIVE

#define IS_NEGATIVE

(

val

)

((((val) >> 31) ^ -1) + 1)

Definition at line 232 of file proresenc_anatoliy.c.

TO_GOLOMB2

#define TO_GOLOMB2	(		val,
			sign
	)		((val)==0 ? 0 : ((val) << 1) + (sign))

Definition at line 233 of file proresenc_anatoliy.c.

FIRST_DC_CB

#define FIRST_DC_CB   0xB8

Definition at line 241 of file proresenc_anatoliy.c.

OFFSET

#define OFFSET

(

x

)

offsetof(ProresContext, x)

Definition at line 922 of file proresenc_anatoliy.c.

VE

#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 923 of file proresenc_anatoliy.c.

Function Documentation

encode_codeword()

static void encode_codeword ( PutBitContext *  pb, int  val, int  codebook  )

static

Definition at line 201 of file proresenc_anatoliy.c.

Referenced by encode_ac_coeffs(), and encode_dc_coeffs().

get_level()

static av_always_inline int get_level ( int  val )

static

Definition at line 235 of file proresenc_anatoliy.c.

Referenced by encode_ac_coeffs(), and encode_dc_coeffs().

encode_dc_coeffs()

static void encode_dc_coeffs ( PutBitContext *  pb, int16_t *  in, int  blocks_per_slice, int *  qmat  )

static

Definition at line 245 of file proresenc_anatoliy.c.

Referenced by encode_slice_plane().

encode_ac_coeffs()

static void encode_ac_coeffs ( PutBitContext *  pb, int16_t *  in, int  blocks_per_slice, int *  qmat, const uint8_t  ff_prores_scan[64]  )

static

Definition at line 276 of file proresenc_anatoliy.c.

Referenced by encode_slice_plane().

get()

static void get ( uint8_t *  pixels, int  stride, int16_t *  block  )

static

Definition at line 307 of file proresenc_anatoliy.c.

Referenced by decode_str(), fdct_get(), ff_framesync_get_frame(), and convert_from_tensorflow.TFConverter::get_dense_params().

fdct_get()

static void fdct_get ( FDCTDSPContext *  fdsp, uint8_t *  pixels, int  stride, int16_t *  block  )

static

Definition at line 319 of file proresenc_anatoliy.c.

Referenced by calc_plane_dct().

calc_plane_dct()

static void calc_plane_dct ( FDCTDSPContext *  fdsp, uint8_t *  src, int16_t *  blocks, int  src_stride, int  mb_count, int  chroma, int  is_422  )

static

Definition at line 325 of file proresenc_anatoliy.c.

Referenced by encode_slice().

encode_slice_plane()

static int encode_slice_plane ( int16_t *  blocks, int  mb_count, uint8_t *  buf, unsigned  buf_size, int *  qmat, int  sub_sample_chroma, const uint8_t  ff_prores_scan[64]  )

static

Definition at line 362 of file proresenc_anatoliy.c.

Referenced by encode_slice_data().

encode_slice_data()

static av_always_inline unsigned encode_slice_data ( AVCodecContext *  avctx, int16_t *  blocks_y, int16_t *  blocks_u, int16_t *  blocks_v, unsigned  mb_count, uint8_t *  buf, unsigned  data_size, unsigned *  y_data_size, unsigned *  u_data_size, unsigned *  v_data_size, int  qp  )

static

Definition at line 378 of file proresenc_anatoliy.c.

Referenced by encode_slice().

put_alpha_diff()

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

static

Definition at line 401 of file proresenc_anatoliy.c.

Referenced by encode_alpha_slice_data().

put_alpha_run()

static void put_alpha_run ( PutBitContext *  pb, int  run  )

inlinestatic

Definition at line 421 of file proresenc_anatoliy.c.

Referenced by encode_alpha_slice_data().

encode_alpha_slice_data()

static av_always_inline int encode_alpha_slice_data ( AVCodecContext *  avctx, int8_t *  src_a, unsigned  mb_count, uint8_t *  buf, unsigned  data_size, unsigned *  a_data_size  )

static

Definition at line 434 of file proresenc_anatoliy.c.

Referenced by encode_slice().

subimage_with_fill_template()

static void subimage_with_fill_template ( uint16_t *  src, unsigned  x, unsigned  y, unsigned  stride, unsigned  width, unsigned  height, uint16_t *  dst, unsigned  dst_width, unsigned  dst_height, int  is_alpha_plane, int  is_interlaced, int  is_top_field  )

inlinestatic

Definition at line 475 of file proresenc_anatoliy.c.

Referenced by subimage_alpha_with_fill(), and subimage_with_fill().

subimage_with_fill()

static void subimage_with_fill ( uint16_t *  src, unsigned  x, unsigned  y, unsigned  stride, unsigned  width, unsigned  height, uint16_t *  dst, unsigned  dst_width, unsigned  dst_height, int  is_interlaced, int  is_top_field  )

static

Definition at line 523 of file proresenc_anatoliy.c.

Referenced by encode_slice().

subimage_alpha_with_fill()

static void subimage_alpha_with_fill ( uint16_t *  src, unsigned  x, unsigned  y, unsigned  stride, unsigned  width, unsigned  height, uint16_t *  dst, unsigned  dst_width, unsigned  dst_height, int  is_interlaced, int  is_top_field  )

static

Definition at line 531 of file proresenc_anatoliy.c.

Referenced by encode_slice().

encode_slice()

static int encode_slice ( AVCodecContext *  avctx, const AVFrame *  pic, int  mb_x, int  mb_y, unsigned  mb_count, uint8_t *  buf, unsigned  data_size, int  unsafe, int *  qp, int  is_interlaced, int  is_top_field  )

static

Definition at line 538 of file proresenc_anatoliy.c.

Referenced by prores_encode_picture().

prores_encode_picture()

static int prores_encode_picture ( AVCodecContext *  avctx, const AVFrame *  pic, uint8_t *  buf, const int  buf_size, const int  picture_index, const int  is_top_field  )

static

Definition at line 656 of file proresenc_anatoliy.c.

Referenced by prores_encode_frame().

prores_encode_frame()

static int prores_encode_frame ( AVCodecContext *  avctx, AVPacket *  pkt, const AVFrame *  pict, int *  got_packet  )

static

Definition at line 720 of file proresenc_anatoliy.c.

scale_mat()

static void scale_mat ( const uint8_t *  src, int *  dst, int  scale  )

static

Definition at line 804 of file proresenc_anatoliy.c.

Referenced by prores_encode_init().

prores_encode_init()

static av_cold int prores_encode_init ( AVCodecContext *  avctx )

static

Definition at line 811 of file proresenc_anatoliy.c.

prores_encode_close()

static av_cold int prores_encode_close ( AVCodecContext *  avctx )

static

Definition at line 913 of file proresenc_anatoliy.c.

Variable Documentation

profiles

const AVProfile profiles[]

static

Initial value:

= {
    { FF_PROFILE_PRORES_PROXY,    "apco"},
    { FF_PROFILE_PRORES_LT,       "apcs"},
    { FF_PROFILE_PRORES_STANDARD, "apcn"},
    { FF_PROFILE_PRORES_HQ,       "apch"},
    { FF_PROFILE_PRORES_4444,     "ap4h"},
    { FF_PROFILE_PRORES_XQ,       "ap4x"},
    { FF_PROFILE_UNKNOWN }
}

Definition at line 44 of file proresenc_anatoliy.c.

Referenced by prores_encode_init().

qp_start_table

const int qp_start_table[] = { 8, 3, 2, 1, 1, 1}

static

Definition at line 54 of file proresenc_anatoliy.c.

Referenced by encode_slice(), and prores_encode_picture().

qp_end_table

const int qp_end_table[] = { 13, 9, 6, 6, 5, 4}

static

Definition at line 55 of file proresenc_anatoliy.c.

Referenced by encode_slice().

bitrate_table

const int bitrate_table[] = { 1000, 2100, 3500, 5400, 7000, 10000}

static

Definition at line 56 of file proresenc_anatoliy.c.

Referenced by encode_slice().

valid_primaries

const int valid_primaries[]

static

Initial value:

= { AVCOL_PRI_RESERVED0, AVCOL_PRI_BT709, AVCOL_PRI_UNSPECIFIED, AVCOL_PRI_BT470BG,
                                        AVCOL_PRI_SMPTE170M, AVCOL_PRI_BT2020, AVCOL_PRI_SMPTE431, AVCOL_PRI_SMPTE432, INT_MAX }

Definition at line 58 of file proresenc_anatoliy.c.

Referenced by prores_encode_frame().

valid_trc

const int valid_trc[]

static

Initial value:

= { AVCOL_TRC_RESERVED0, AVCOL_TRC_BT709, AVCOL_TRC_UNSPECIFIED, AVCOL_TRC_SMPTE2084,
                                        AVCOL_TRC_ARIB_STD_B67, INT_MAX }

Definition at line 60 of file proresenc_anatoliy.c.

Referenced by prores_encode_frame().

valid_colorspace

const int valid_colorspace[]

static

Initial value:

= { AVCOL_SPC_BT709, AVCOL_SPC_UNSPECIFIED, AVCOL_SPC_SMPTE170M,
                                        AVCOL_SPC_BT2020_NCL, INT_MAX }

Definition at line 62 of file proresenc_anatoliy.c.

Referenced by prores_encode_frame().

QMAT_LUMA

const uint8_t QMAT_LUMA[6][64]

static

Definition at line 65 of file proresenc_anatoliy.c.

Referenced by prores_encode_frame(), and prores_encode_init().

QMAT_CHROMA

const uint8_t QMAT_CHROMA[6][64]

static

Definition at line 123 of file proresenc_anatoliy.c.

Referenced by prores_encode_frame(), and prores_encode_init().

dc_codebook

const uint8_t dc_codebook[7] = { 0x04, 0x28, 0x28, 0x4D, 0x4D, 0x70, 0x70}

static

Definition at line 243 of file proresenc_anatoliy.c.

Referenced by encode_dc_coeffs().

run_to_cb

const uint8_t run_to_cb[16]

static

Initial value:

= { 0x06, 0x06, 0x05, 0x05, 0x04, 0x29,
        0x29, 0x29, 0x29, 0x28, 0x28, 0x28, 0x28, 0x28, 0x28, 0x4C }

Definition at line 271 of file proresenc_anatoliy.c.

Referenced by encode_ac_coeffs().

lev_to_cb

const uint8_t lev_to_cb[10]

static

Initial value:

= { 0x04, 0x0A, 0x05, 0x06, 0x04, 0x28,
        0x28, 0x28, 0x28, 0x4C }

Definition at line 273 of file proresenc_anatoliy.c.

Referenced by encode_ac_coeffs().

options

const AVOption options[]

static

Initial value:

= {
    { "vendor", "vendor ID", OFFSET(vendor), AV_OPT_TYPE_STRING, { .str = "fmpg" }, 0, 0, VE },
    { NULL }
}

Definition at line 925 of file proresenc_anatoliy.c.

prores_enc_class

const AVClass prores_enc_class

static

Initial value:

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

Definition at line 930 of file proresenc_anatoliy.c.

pix_fmts

enum AVPixelFormat pix_fmts[]

static

Initial value:

= {
    AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10,
    AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_NONE
}

Definition at line 937 of file proresenc_anatoliy.c.

ff_prores_aw_encoder

const AVCodec ff_prores_aw_encoder

Initial value:

= {
    .name           = "prores_aw",
    .long_name      = NULL_IF_CONFIG_SMALL("Apple ProRes"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_PRORES,
    .pix_fmts       = pix_fmts,
    .priv_data_size = sizeof(ProresContext),
    .init           = prores_encode_init,
    .close          = prores_encode_close,
    .encode2        = prores_encode_frame,
    .capabilities   = AV_CODEC_CAP_FRAME_THREADS,
    .priv_class     = &prores_enc_class,
    .profiles       = NULL_IF_CONFIG_SMALL(ff_prores_profiles),
    .caps_internal  = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP,
}

Definition at line 942 of file proresenc_anatoliy.c.

ff_prores_encoder

const AVCodec ff_prores_encoder

Initial value:

= {
    .name           = "prores",
    .long_name      = NULL_IF_CONFIG_SMALL("Apple ProRes"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_PRORES,
    .pix_fmts       = pix_fmts,
    .priv_data_size = sizeof(ProresContext),
    .init           = prores_encode_init,
    .close          = prores_encode_close,
    .encode2        = prores_encode_frame,
    .capabilities   = AV_CODEC_CAP_FRAME_THREADS,
    .priv_class     = &prores_enc_class,
    .profiles       = NULL_IF_CONFIG_SMALL(ff_prores_profiles),
    .caps_internal  = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP,
}

Definition at line 958 of file proresenc_anatoliy.c.