FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
Data Structures | Macros | Enumerations | Functions | Variables
proresenc_kostya.c File Reference
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "dsputil.h"
#include "put_bits.h"
#include "bytestream.h"
#include "internal.h"
#include "proresdsp.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) << 1) ^ 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_PROXY = 0, PRORES_PROFILE_LT, PRORES_PROFILE_STANDARD, PRORES_PROFILE_HQ,
  PRORES_PROFILE_4444
}
 
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, int 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, int 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.
 
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, int 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, const uint16_t *src, int linesize, 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, int 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, int *error, const uint16_t *src, int linesize, int mbs_per_slice, int quant, int16_t *blocks)
 
static int find_slice_quant (AVCodecContext *avctx, const AVFrame *pic, 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 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 struct prores_profile prores_profile_info [5]
 
static const AVOption options []
 
static const AVClass proresenc_class
 
AVCodec ff_prores_ks_encoder
 

Macro Definition Documentation

#define CFACTOR_Y422   2

Definition at line 36 of file proresenc_kostya.c.

Referenced by encode_init().

#define CFACTOR_Y444   3

Definition at line 37 of file proresenc_kostya.c.

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

#define MAX_MBS_PER_SLICE   8

Definition at line 39 of file proresenc_kostya.c.

#define MAX_PLANES   4

Definition at line 41 of file proresenc_kostya.c.

Referenced by find_slice_quant().

#define NUM_MB_LIMITS   4

Definition at line 112 of file proresenc_kostya.c.

Referenced by encode_init().

#define TRELLIS_WIDTH   16

Definition at line 170 of file proresenc_kostya.c.

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

#define SCORE_LIMIT   INT_MAX / 2

Definition at line 171 of file proresenc_kostya.c.

Referenced by find_slice_quant().

#define MAX_STORED_Q   16

Definition at line 180 of file proresenc_kostya.c.

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

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

Definition at line 356 of file proresenc_kostya.c.

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

#define MAKE_CODE (   x)    (((x) << 1) ^ GET_SIGN(x))

Definition at line 357 of file proresenc_kostya.c.

Referenced by encode_dcs(), and estimate_dcs().

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

Definition at line 1212 of file proresenc_kostya.c.

Definition at line 1213 of file proresenc_kostya.c.

Enumeration Type Documentation

anonymous enum
Enumerator:
PRORES_PROFILE_PROXY 
PRORES_PROFILE_LT 
PRORES_PROFILE_STANDARD 
PRORES_PROFILE_HQ 
PRORES_PROFILE_4444 

Definition at line 43 of file proresenc_kostya.c.

anonymous enum
Enumerator:
QUANT_MAT_PROXY 
QUANT_MAT_LT 
QUANT_MAT_STANDARD 
QUANT_MAT_HQ 
QUANT_MAT_DEFAULT 

Definition at line 51 of file proresenc_kostya.c.

Function Documentation

static void get_slice_data ( ProresContext ctx,
const uint16_t *  src,
int  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 225 of file proresenc_kostya.c.

Referenced by encode_slice(), and find_slice_quant().

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

Definition at line 296 of file proresenc_kostya.c.

Referenced by encode_slice(), and find_slice_quant().

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

Write an unsigned rice/exp golomb codeword.

Definition at line 327 of file proresenc_kostya.c.

Referenced by encode_acs(), and encode_dcs().

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

Definition at line 359 of file proresenc_kostya.c.

Referenced by encode_slice_plane().

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 385 of file proresenc_kostya.c.

Referenced by encode_slice_plane().

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

Definition at line 419 of file proresenc_kostya.c.

Referenced by encode_slice().

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

Definition at line 438 of file proresenc_kostya.c.

Referenced by encode_alpha_plane().

static void put_alpha_run ( PutBitContext pb,
int  run 
)
static

Definition at line 458 of file proresenc_kostya.c.

Referenced by encode_alpha_plane().

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

Definition at line 472 of file proresenc_kostya.c.

Referenced by 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 505 of file proresenc_kostya.c.

Referenced by encode_frame().

static int estimate_vlc ( unsigned  codebook,
int  val 
)
inlinestatic

Definition at line 577 of file proresenc_kostya.c.

Referenced by estimate_acs(), and estimate_dcs().

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

Definition at line 599 of file proresenc_kostya.c.

Referenced by estimate_slice_plane().

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 630 of file proresenc_kostya.c.

Referenced by estimate_slice_plane().

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

Definition at line 666 of file proresenc_kostya.c.

Referenced by find_slice_quant().

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

Definition at line 684 of file proresenc_kostya.c.

Referenced by estimate_alpha_plane().

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

Definition at line 700 of file proresenc_kostya.c.

Referenced by find_slice_quant().

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

Definition at line 744 of file proresenc_kostya.c.

Referenced by find_quant_thread().

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

Definition at line 901 of file proresenc_kostya.c.

Referenced by encode_frame().

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

Definition at line 925 of file proresenc_kostya.c.

static av_cold int encode_close ( AVCodecContext avctx)
static

Definition at line 1050 of file proresenc_kostya.c.

Referenced by encode_init().

static av_cold int encode_init ( AVCodecContext avctx)
static

Definition at line 1067 of file proresenc_kostya.c.

Variable Documentation

const uint8_t prores_quant_matrices[][64]
static

Definition at line 59 of file proresenc_kostya.c.

Referenced by encode_init().

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

Definition at line 113 of file proresenc_kostya.c.

Referenced by encode_init().

struct prores_profile prores_profile_info[5]
static

Referenced by encode_init().

const AVOption options[]
static

Definition at line 1215 of file proresenc_kostya.c.

const AVClass proresenc_class
static
Initial value:
= {
.class_name = "ProRes encoder",
.item_name = av_default_item_name,
.option = options,
}

Definition at line 1254 of file proresenc_kostya.c.

AVCodec ff_prores_ks_encoder
Initial value:
= {
.name = "prores_ks",
.priv_data_size = sizeof(ProresContext),
.encode2 = encode_frame,
.capabilities = CODEC_CAP_SLICE_THREADS,
.long_name = NULL_IF_CONFIG_SMALL("Apple ProRes (iCodec Pro)"),
.pix_fmts = (const enum AVPixelFormat[]) {
},
.priv_class = &proresenc_class,
}

Definition at line 1261 of file proresenc_kostya.c.