FFmpeg
Data Structures | Macros | Enumerations | Functions | Variables
proresenc_kostya.c File Reference
#include "libavutil/mem.h"
#include "libavutil/mem_internal.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "codec_internal.h"
#include "encode.h"
#include "fdctdsp.h"
#include "put_bits.h"
#include "profiles.h"
#include "bytestream.h"
#include "proresdata.h"

Go to the source code of this file.

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_PROXY_CHROMA, QUANT_MAT_LT, QUANT_MAT_STANDARD,
  QUANT_MAT_HQ, QUANT_MAT_XQ_LUMA, 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, uint16_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, const uint8_t *scan, const int16_t *qmat)
 
static void 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, 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 void 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, 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, 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
 
const FFCodec ff_prores_ks_encoder
 

Macro Definition Documentation

◆ CFACTOR_Y422

#define CFACTOR_Y422   2

Definition at line 37 of file proresenc_kostya.c.

◆ CFACTOR_Y444

#define CFACTOR_Y444   3

Definition at line 38 of file proresenc_kostya.c.

◆ MAX_MBS_PER_SLICE

#define MAX_MBS_PER_SLICE   8

Definition at line 40 of file proresenc_kostya.c.

◆ MAX_PLANES

#define MAX_PLANES   4

Definition at line 42 of file proresenc_kostya.c.

◆ NUM_MB_LIMITS

#define NUM_MB_LIMITS   4

Definition at line 137 of file proresenc_kostya.c.

◆ TRELLIS_WIDTH

#define TRELLIS_WIDTH   16

Definition at line 210 of file proresenc_kostya.c.

◆ SCORE_LIMIT

#define SCORE_LIMIT   INT_MAX / 2

Definition at line 211 of file proresenc_kostya.c.

◆ MAX_STORED_Q

#define MAX_STORED_Q   16

Definition at line 220 of file proresenc_kostya.c.

◆ GET_SIGN

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

Definition at line 404 of file proresenc_kostya.c.

◆ MAKE_CODE

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

Definition at line 405 of file proresenc_kostya.c.

◆ OFFSET

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

Definition at line 1320 of file proresenc_kostya.c.

◆ VE

Definition at line 1321 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 44 of file proresenc_kostya.c.

◆ anonymous enum

anonymous enum
Enumerator
QUANT_MAT_PROXY 
QUANT_MAT_PROXY_CHROMA 
QUANT_MAT_LT 
QUANT_MAT_STANDARD 
QUANT_MAT_HQ 
QUANT_MAT_XQ_LUMA 
QUANT_MAT_DEFAULT 

Definition at line 54 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 273 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,
uint16_t *  blocks,
int  mbs_per_slice,
int  abits 
)
static

Definition at line 344 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 375 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 407 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,
const uint8_t *  scan,
const int16_t *  qmat 
)
static

Definition at line 431 of file proresenc_kostya.c.

Referenced by encode_slice_plane().

◆ encode_slice_plane()

static void 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,
const int16_t *  qmat 
)
static

Definition at line 462 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 474 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 493 of file proresenc_kostya.c.

Referenced by encode_alpha_plane().

◆ encode_alpha_plane()

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

Definition at line 507 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 535 of file proresenc_kostya.c.

Referenced by encode_frame().

◆ estimate_vlc()

static int estimate_vlc ( unsigned  codebook,
int  val 
)
inlinestatic

Definition at line 615 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 637 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,
const uint8_t *  scan,
const int16_t *  qmat 
)
static

Definition at line 666 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,
const int16_t *  qmat,
ProresThreadData td 
)
static

Definition at line 701 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 718 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 733 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 775 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 940 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 964 of file proresenc_kostya.c.

◆ encode_close()

static av_cold int encode_close ( AVCodecContext avctx)
static

Definition at line 1120 of file proresenc_kostya.c.

◆ prores_fdct()

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

Definition at line 1135 of file proresenc_kostya.c.

Referenced by encode_init().

◆ encode_init()

static av_cold int encode_init ( AVCodecContext avctx)
static

Definition at line 1149 of file proresenc_kostya.c.

Variable Documentation

◆ prores_quant_matrices

const uint8_t prores_quant_matrices[][64]
static

Definition at line 64 of file proresenc_kostya.c.

Referenced by encode_init().

◆ prores_mb_limits

const int prores_mb_limits[NUM_MB_LIMITS]
static
Initial value:
= {
1620,
2700,
6075,
9216,
}

Definition at line 138 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 1323 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,
}

Definition at line 1366 of file proresenc_kostya.c.

◆ ff_prores_ks_encoder

const FFCodec ff_prores_ks_encoder
Initial value:
= {
.p.name = "prores_ks",
CODEC_LONG_NAME("Apple ProRes (iCodec Pro)"),
.p.type = AVMEDIA_TYPE_VIDEO,
.priv_data_size = sizeof(ProresContext),
.close = encode_close,
.p.pix_fmts = (const enum AVPixelFormat[]) {
},
.color_ranges = AVCOL_RANGE_MPEG,
.p.priv_class = &proresenc_class,
.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
}

Definition at line 1373 of file proresenc_kostya.c.

AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:71
FF_CODEC_CAP_INIT_CLEANUP
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
Definition: codec_internal.h:43
encode_frame
static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic, int *got_packet)
Definition: proresenc_kostya.c:964
ff_prores_profiles
const AVProfile ff_prores_profiles[]
Definition: profiles.c:175
ProresContext
Definition: proresdec.h:43
FF_CODEC_ENCODE_CB
#define FF_CODEC_ENCODE_CB(func)
Definition: codec_internal.h:320
AV_PIX_FMT_YUV444P10
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:490
AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE
#define AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE
This encoder can reorder user opaque values from input AVFrames and return them with corresponding ou...
Definition: codec.h:159
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:296
AV_CODEC_CAP_FRAME_THREADS
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
Definition: codec.h:110
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
encode_init
static av_cold int encode_init(AVCodecContext *avctx)
Definition: proresenc_kostya.c:1149
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:237
AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:488
init
int(* init)(AVBSFContext *ctx)
Definition: dts2pts.c:368
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:94
AV_CODEC_CAP_SLICE_THREADS
#define AV_CODEC_CAP_SLICE_THREADS
Codec supports slice-based (or partition-based) multithreading.
Definition: codec.h:114
AV_PIX_FMT_YUVA444P10
#define AV_PIX_FMT_YUVA444P10
Definition: pixfmt.h:527
options
static const AVOption options[]
Definition: proresenc_kostya.c:1323
AVCOL_RANGE_MPEG
@ AVCOL_RANGE_MPEG
Narrow or limited range content.
Definition: pixfmt.h:679
proresenc_class
static const AVClass proresenc_class
Definition: proresenc_kostya.c:1366
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:72
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
encode_close
static av_cold int encode_close(AVCodecContext *avctx)
Definition: proresenc_kostya.c:1120
AV_CODEC_ID_PRORES
@ AV_CODEC_ID_PRORES
Definition: codec_id.h:200