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 "fdctdsp.h"
#include "put_bits.h"
#include "bytestream.h"
#include "internal.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, 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

#define CFACTOR_Y422   2

Definition at line 35 of file proresenc_kostya.c.

Referenced by encode_init().

#define CFACTOR_Y444   3

Definition at line 36 of file proresenc_kostya.c.

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

#define MAX_MBS_PER_SLICE   8

Definition at line 38 of file proresenc_kostya.c.

#define MAX_PLANES   4

Definition at line 40 of file proresenc_kostya.c.

Referenced by find_slice_quant().

#define NUM_MB_LIMITS   4

Definition at line 135 of file proresenc_kostya.c.

Referenced by encode_init().

#define TRELLIS_WIDTH   16

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

Referenced by find_slice_quant().

#define MAX_STORED_Q   16

Definition at line 218 of file proresenc_kostya.c.

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

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

Definition at line 400 of file proresenc_kostya.c.

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

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

Definition at line 401 of file proresenc_kostya.c.

Referenced by encode_dcs(), and estimate_dcs().

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

Definition at line 1366 of file proresenc_kostya.c.

#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 1367 of file proresenc_kostya.c.

Enumeration Type Documentation

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

Function Documentation

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

Referenced by encode_slice(), and find_slice_quant().

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 340 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 371 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 403 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 429 of file proresenc_kostya.c.

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

Referenced by encode_alpha_plane().

static void put_alpha_run ( PutBitContext *  pb,
int  run 
)
static

Definition at line 501 of file proresenc_kostya.c.

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

Referenced by encode_frame().

static int estimate_vlc ( unsigned  codebook,
int  val 
)
inlinestatic

Definition at line 637 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 659 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 690 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,
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 726 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 744 of file proresenc_kostya.c.

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

Referenced by 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 801 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 970 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 994 of file proresenc_kostya.c.

static av_cold int encode_close ( AVCodecContext *  avctx)
static

Definition at line 1155 of file proresenc_kostya.c.

Referenced by encode_init().

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

Definition at line 1170 of file proresenc_kostya.c.

Referenced by encode_init().

static av_cold int encode_init ( AVCodecContext *  avctx)
static

Definition at line 1184 of file proresenc_kostya.c.

Variable Documentation

const uint8_t prores_quant_matrices[][64]
static

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

Referenced by encode_init().

const struct prores_profile prores_profile_info[6]
static

Referenced by encode_init().

const AVOption options[]
static

Definition at line 1369 of file proresenc_kostya.c.

const AVClass proresenc_class
static
Initial value:
= {
.class_name = "ProRes encoder",
.item_name = av_default_item_name,
.option = options,
.version = LIBAVUTIL_VERSION_INT,
}
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:191
options
static const AVOption options[]
Definition: proresenc_kostya.c:1369

Definition at line 1412 of file proresenc_kostya.c.

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,
}
proresenc_class
static const AVClass proresenc_class
Definition: proresenc_kostya.c:1412
encode_frame
static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic, int *got_packet)
Definition: proresenc_kostya.c:994
encode_init
static av_cold int encode_init(AVCodecContext *avctx)
Definition: proresenc_kostya.c:1184
init
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
AV_CODEC_ID_PRORES
Definition: avcodec.h:366
encode_close
static av_cold int encode_close(AVCodecContext *avctx)
Definition: proresenc_kostya.c:1155
AV_CODEC_CAP_INTRA_ONLY
#define AV_CODEC_CAP_INTRA_ONLY
Codec is intra only.
Definition: avcodec.h:1054
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:186
AV_PIX_FMT_YUV444P10
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:382
AV_CODEC_CAP_FRAME_THREADS
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
Definition: avcodec.h:1024
AV_PIX_FMT_YUVA444P10
#define AV_PIX_FMT_YUVA444P10
Definition: pixfmt.h:418
AV_CODEC_CAP_SLICE_THREADS
#define AV_CODEC_CAP_SLICE_THREADS
Codec supports slice-based (or partition-based) multithreading.
Definition: avcodec.h:1028
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:266
AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:380
AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64

Definition at line 1419 of file proresenc_kostya.c.

Generated on Tue Nov 6 2018 18:11:43 for FFmpeg by   doxygen 1.8.6