Macros | Functions | Variables
mpegvideo_enc.c File Reference

The simplest mpeg encoder (well, it was the simplest!). More...

#include "libavutil/intmath.h"
#include "libavutil/mathematics.h"
#include "libavutil/pixdesc.h"
#include "libavutil/opt.h"
#include "avcodec.h"
#include "dsputil.h"
#include "mpegvideo.h"
#include "h263.h"
#include "mathops.h"
#include "mjpegenc.h"
#include "msmpeg4.h"
#include "faandct.h"
#include "thread.h"
#include "aandcttab.h"
#include "flv.h"
#include "mpeg4video.h"
#include "internal.h"
#include "bytestream.h"
#include <limits.h>

Go to the source code of this file.

Macros

#define COPY(a)   dst->a= src->a
#define MERGE(field)   dst->field += src->field; src->field=0
#define OFFSET(x)   offsetof(MpegEncContext, x)
#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Functions

static int encode_picture (MpegEncContext *s, int picture_number)
static int dct_quantize_refine (MpegEncContext *s, DCTELEM *block, int16_t *weight, DCTELEM *orig, int n, int qscale)
static int sse_mb (MpegEncContext *s)
static void denoise_dct_c (MpegEncContext *s, DCTELEM *block)
static int dct_quantize_trellis_c (MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow)
void ff_convert_matrix (DSPContext *dsp, int(*qmat)[64], uint16_t(*qmat16)[2][64], const uint16_t *quant_matrix, int bias, int qmin, int qmax, int intra)
static void update_qscale (MpegEncContext *s)
void ff_write_quant_matrix (PutBitContext *pb, uint16_t *matrix)
void ff_init_qscale_tab (MpegEncContext *s)
 init s->current_picture.qscale_table from s->lambda_table
static void copy_picture_attributes (MpegEncContext *s, AVFrame *dst, AVFrame *src)
static void update_duplicate_context_after_me (MpegEncContext *dst, MpegEncContext *src)
static void MPV_encode_defaults (MpegEncContext *s)
 Set the given MpegEncContext to defaults for encoding.
av_cold int ff_MPV_encode_init (AVCodecContext *avctx)
av_cold int ff_MPV_encode_end (AVCodecContext *avctx)
static int get_sae (uint8_t *src, int ref, int stride)
static int get_intra_count (MpegEncContext *s, uint8_t *src, uint8_t *ref, int stride)
static int load_input_picture (MpegEncContext *s, AVFrame *pic_arg)
static int skip_check (MpegEncContext *s, Picture *p, Picture *ref)
static int encode_frame (AVCodecContext *c, AVFrame *frame)
static int estimate_best_b_count (MpegEncContext *s)
static int select_input_picture (MpegEncContext *s)
int ff_MPV_encode_picture (AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic_arg, int *got_packet)
static void dct_single_coeff_elimination (MpegEncContext *s, int n, int threshold)
static void clip_coeffs (MpegEncContext *s, DCTELEM *block, int last_index)
static void get_visual_weight (int16_t *weight, uint8_t *ptr, int stride)
static av_always_inline void encode_mb_internal (MpegEncContext *s, int motion_x, int motion_y, int mb_block_height, int mb_block_count)
static av_always_inline void encode_mb (MpegEncContext *s, int motion_x, int motion_y)
static void copy_context_before_encode (MpegEncContext *d, MpegEncContext *s, int type)
static void copy_context_after_encode (MpegEncContext *d, MpegEncContext *s, int type)
static void encode_mb_hq (MpegEncContext *s, MpegEncContext *backup, MpegEncContext *best, int type, PutBitContext pb[2], PutBitContext pb2[2], PutBitContext tex_pb[2], int *dmin, int *next_block, int motion_x, int motion_y)
static int sse (MpegEncContext *s, uint8_t *src1, uint8_t *src2, int w, int h, int stride)
static int pre_estimate_motion_thread (AVCodecContext *c, void *arg)
static int estimate_motion_thread (AVCodecContext *c, void *arg)
static int mb_var_thread (AVCodecContext *c, void *arg)
static void write_slice_end (MpegEncContext *s)
static void write_mb_info (MpegEncContext *s)
static void update_mb_info (MpegEncContext *s, int startcode)
static int encode_thread (AVCodecContext *c, void *arg)
static void merge_context_after_me (MpegEncContext *dst, MpegEncContext *src)
static void merge_context_after_encode (MpegEncContext *dst, MpegEncContext *src)
static int estimate_qp (MpegEncContext *s, int dry_run)
static void set_frame_distances (MpegEncContext *s)
static void build_basis (uint8_t *perm)
int ff_dct_quantize_c (MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow)

Variables

static uint8_t default_mv_penalty [MAX_FCODE+1][MAX_MV *2+1]
static uint8_t default_fcode_tab [MAX_MV *2+1]
const AVOption ff_mpv_generic_options []
static int16_t basis [64][64]
static const AVOption h263_options []
static const AVClass h263_class
AVCodec ff_h263_encoder
static const AVOption h263p_options []
static const AVClass h263p_class
AVCodec ff_h263p_encoder
AVCodec ff_msmpeg4v2_encoder
AVCodec ff_msmpeg4v3_encoder
AVCodec ff_wmv1_encoder

Detailed Description

The simplest mpeg encoder (well, it was the simplest!).

Definition in file mpegvideo_enc.c.

Macro Definition Documentation

#define COPY (   a)    dst->a= src->a
#define MERGE (   field)    dst->field += src->field; src->field=0

Definition at line 3018 of file mpegvideo_enc.c.

Referenced by merge_context_after_encode(), and merge_context_after_me().

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

Definition at line 4121 of file mpegvideo_enc.c.

#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 4122 of file mpegvideo_enc.c.

Function Documentation

static void build_basis ( uint8_t *  perm)
static

Definition at line 3649 of file mpegvideo_enc.c.

Referenced by dct_quantize_refine().

static void clip_coeffs ( MpegEncContext *  s,
DCTELEM *  block,
int  last_index 
)
inlinestatic

Definition at line 1668 of file mpegvideo_enc.c.

Referenced by encode_mb_internal().

static void copy_context_after_encode ( MpegEncContext *  d,
MpegEncContext *  s,
int  type 
)
inlinestatic

Definition at line 2124 of file mpegvideo_enc.c.

Referenced by encode_mb_hq(), and encode_thread().

static void copy_context_before_encode ( MpegEncContext *  d,
MpegEncContext *  s,
int  type 
)
inlinestatic

Definition at line 2096 of file mpegvideo_enc.c.

Referenced by encode_mb_hq(), and encode_thread().

static void copy_picture_attributes ( MpegEncContext *  s,
AVFrame *  dst,
AVFrame *  src 
)
static

Definition at line 189 of file mpegvideo_enc.c.

Referenced by load_input_picture(), and select_input_picture().

static int dct_quantize_refine ( MpegEncContext *  s,
DCTELEM *  block,
int16_t *  weight,
DCTELEM *  orig,
int  n,
int  qscale 
)
static

Definition at line 3668 of file mpegvideo_enc.c.

Referenced by encode_mb_internal().

static int dct_quantize_trellis_c ( MpegEncContext *  s,
DCTELEM *  block,
int  n,
int  qscale,
int *  overflow 
)
static

Definition at line 3354 of file mpegvideo_enc.c.

Referenced by ff_MPV_encode_init().

static void dct_single_coeff_elimination ( MpegEncContext *  s,
int  n,
int  threshold 
)
inlinestatic

Definition at line 1612 of file mpegvideo_enc.c.

Referenced by encode_mb_internal().

static void denoise_dct_c ( MpegEncContext *  s,
DCTELEM *  block 
)
static

Definition at line 3330 of file mpegvideo_enc.c.

Referenced by ff_MPV_encode_init().

static int encode_frame ( AVCodecContext *  c,
AVFrame *  frame 
)
static

Definition at line 1077 of file mpegvideo_enc.c.

static av_always_inline void encode_mb ( MpegEncContext *  s,
int  motion_x,
int  motion_y 
)
static

Definition at line 2090 of file mpegvideo_enc.c.

Referenced by encode_mb_hq(), and encode_thread().

static void encode_mb_hq ( MpegEncContext *  s,
MpegEncContext *  backup,
MpegEncContext *  best,
int  type,
PutBitContext  pb[2],
PutBitContext  pb2[2],
PutBitContext  tex_pb[2],
int *  dmin,
int *  next_block,
int  motion_x,
int  motion_y 
)
inlinestatic

Definition at line 2163 of file mpegvideo_enc.c.

Referenced by encode_thread().

static av_always_inline void encode_mb_internal ( MpegEncContext *  s,
int  motion_x,
int  motion_y,
int  mb_block_height,
int  mb_block_count 
)
static

Definition at line 1726 of file mpegvideo_enc.c.

Referenced by encode_mb().

static int encode_picture ( MpegEncContext *  s,
int  picture_number 
)
static

Definition at line 3108 of file mpegvideo_enc.c.

Referenced by ff_MPV_encode_picture().

static int encode_thread ( AVCodecContext *  c,
void *  arg 
)
static

Definition at line 2389 of file mpegvideo_enc.c.

Referenced by encode_picture().

static int estimate_best_b_count ( MpegEncContext *  s)
static

Definition at line 1092 of file mpegvideo_enc.c.

Referenced by select_input_picture().

static int estimate_motion_thread ( AVCodecContext *  c,
void *  arg 
)
static

Definition at line 2277 of file mpegvideo_enc.c.

Referenced by encode_picture().

static int estimate_qp ( MpegEncContext *  s,
int  dry_run 
)
static

Definition at line 3057 of file mpegvideo_enc.c.

Referenced by encode_picture().

void ff_convert_matrix ( DSPContext *  dsp,
int(*)  qmat[64],
uint16_t(*)  qmat16[2][64],
const uint16_t *  quant_matrix,
int  bias,
int  qmin,
int  qmax,
int  intra 
)

Definition at line 72 of file mpegvideo_enc.c.

Referenced by dnxhd_init_qmat(), encode_picture(), and ff_MPV_encode_init().

int ff_dct_quantize_c ( MpegEncContext *  s,
DCTELEM *  block,
int  n,
int  qscale,
int *  overflow 
)

Definition at line 4041 of file mpegvideo_enc.c.

Referenced by dnxhd_encode_init(), encode_mb_internal(), and ff_MPV_encode_init().

void ff_init_qscale_tab ( MpegEncContext *  s)

init s->current_picture.qscale_table from s->lambda_table

Definition at line 176 of file mpegvideo_enc.c.

Referenced by estimate_qp(), and ff_clean_h263_qscales().

av_cold int ff_MPV_encode_end ( AVCodecContext *  avctx)

Definition at line 856 of file mpegvideo_enc.c.

Referenced by mpeg12_class().

av_cold int ff_MPV_encode_init ( AVCodecContext *  avctx)

Definition at line 273 of file mpegvideo_enc.c.

Referenced by encode_init(), and wmv2_encode_init().

int ff_MPV_encode_picture ( AVCodecContext *  avctx,
AVPacket *  pkt,
const AVFrame *  pic_arg,
int *  got_packet 
)

Definition at line 1409 of file mpegvideo_enc.c.

Referenced by mpeg12_class().

void ff_write_quant_matrix ( PutBitContext *  pb,
uint16_t *  matrix 
)

Definition at line 160 of file mpegvideo_enc.c.

Referenced by mpeg1_encode_sequence_header(), and mpeg4_encode_vol_header().

static int get_intra_count ( MpegEncContext *  s,
uint8_t *  src,
uint8_t *  ref,
int  stride 
)
static

Definition at line 886 of file mpegvideo_enc.c.

Referenced by select_input_picture().

static int get_sae ( uint8_t *  src,
int  ref,
int  stride 
)
static

Definition at line 872 of file mpegvideo_enc.c.

Referenced by get_intra_count().

static void get_visual_weight ( int16_t *  weight,
uint8_t *  ptr,
int  stride 
)
static

Definition at line 1702 of file mpegvideo_enc.c.

Referenced by encode_mb_internal().

static int load_input_picture ( MpegEncContext *  s,
AVFrame *  pic_arg 
)
static

Definition at line 910 of file mpegvideo_enc.c.

Referenced by ff_MPV_encode_picture().

static int mb_var_thread ( AVCodecContext *  c,
void *  arg 
)
static

Definition at line 2304 of file mpegvideo_enc.c.

Referenced by encode_picture().

static void merge_context_after_encode ( MpegEncContext *  dst,
MpegEncContext *  src 
)
static

Definition at line 3025 of file mpegvideo_enc.c.

Referenced by encode_picture().

static void merge_context_after_me ( MpegEncContext *  dst,
MpegEncContext *  src 
)
static

Definition at line 3019 of file mpegvideo_enc.c.

Referenced by encode_picture().

static void MPV_encode_defaults ( MpegEncContext *  s)
static

Set the given MpegEncContext to defaults for encoding.

the changed fields will not depend upon the prior state of the MpegEncContext.

Definition at line 260 of file mpegvideo_enc.c.

Referenced by ff_MPV_encode_init().

static int pre_estimate_motion_thread ( AVCodecContext *  c,
void *  arg 
)
static

Definition at line 2258 of file mpegvideo_enc.c.

Referenced by encode_picture().

static int select_input_picture ( MpegEncContext *  s)
static

Definition at line 1216 of file mpegvideo_enc.c.

Referenced by ff_MPV_encode_picture().

static void set_frame_distances ( MpegEncContext *  s)
static

Definition at line 3094 of file mpegvideo_enc.c.

Referenced by encode_picture().

static int skip_check ( MpegEncContext *  s,
Picture *  p,
Picture *  ref 
)
static

Definition at line 1040 of file mpegvideo_enc.c.

Referenced by select_input_picture().

static int sse ( MpegEncContext *  s,
uint8_t *  src1,
uint8_t *  src2,
int  w,
int  h,
int  stride 
)
static

Definition at line 2214 of file mpegvideo_enc.c.

Referenced by block_sse(), encode_thread(), main(), and sse_mb().

static int sse_mb ( MpegEncContext *  s)
static

Definition at line 2235 of file mpegvideo_enc.c.

Referenced by encode_mb_hq().

static void update_duplicate_context_after_me ( MpegEncContext *  dst,
MpegEncContext *  src 
)
static

Definition at line 237 of file mpegvideo_enc.c.

Referenced by encode_picture().

static void update_mb_info ( MpegEncContext *  s,
int  startcode 
)
static

Definition at line 2366 of file mpegvideo_enc.c.

Referenced by encode_thread().

static void update_qscale ( MpegEncContext *  s)
inlinestatic

Definition at line 150 of file mpegvideo_enc.c.

Referenced by encode_mb_internal(), encode_picture(), and estimate_qp().

static void write_mb_info ( MpegEncContext *  s)
static

Definition at line 2346 of file mpegvideo_enc.c.

Referenced by update_mb_info().

static void write_slice_end ( MpegEncContext *  s)
static

Definition at line 2328 of file mpegvideo_enc.c.

Referenced by encode_thread().

Variable Documentation

int16_t basis[64][64]
static

Definition at line 3647 of file mpegvideo_enc.c.

Referenced by add_8x8basis(), and try_8x8basis().

uint8_t default_fcode_tab[MAX_MV *2+1]
static

Definition at line 65 of file mpegvideo_enc.c.

Referenced by MPV_encode_defaults().

uint8_t default_mv_penalty[MAX_FCODE+1][MAX_MV *2+1]
static

Definition at line 64 of file mpegvideo_enc.c.

Referenced by MPV_encode_defaults().

AVCodec ff_h263_encoder
Initial value:
{
.name = "h263",
.type = AVMEDIA_TYPE_VIDEO,
.id = AV_CODEC_ID_H263,
.priv_data_size = sizeof(MpegEncContext),
.init = ff_MPV_encode_init,
.encode2 = ff_MPV_encode_picture,
.close = ff_MPV_encode_end,
.pix_fmts= (const enum AVPixelFormat[]){AV_PIX_FMT_YUV420P, AV_PIX_FMT_NONE},
.long_name= NULL_IF_CONFIG_SMALL("H.263 / H.263-1996"),
.priv_class = &h263_class,
}

Definition at line 4138 of file mpegvideo_enc.c.

AVCodec ff_h263p_encoder
Initial value:
{
.name = "h263p",
.type = AVMEDIA_TYPE_VIDEO,
.id = AV_CODEC_ID_H263P,
.priv_data_size = sizeof(MpegEncContext),
.init = ff_MPV_encode_init,
.encode2 = ff_MPV_encode_picture,
.close = ff_MPV_encode_end,
.capabilities = CODEC_CAP_SLICE_THREADS,
.pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV420P, AV_PIX_FMT_NONE },
.long_name = NULL_IF_CONFIG_SMALL("H.263+ / H.263-1998 / H.263 version 2"),
.priv_class = &h263p_class,
}

Definition at line 4166 of file mpegvideo_enc.c.

const AVOption ff_mpv_generic_options[]
Initial value:
{
FF_MPV_COMMON_OPTS
{ NULL },
}

Definition at line 67 of file mpegvideo_enc.c.

AVCodec ff_msmpeg4v2_encoder
Initial value:
{
.name = "msmpeg4v2",
.type = AVMEDIA_TYPE_VIDEO,
.id = AV_CODEC_ID_MSMPEG4V2,
.priv_data_size = sizeof(MpegEncContext),
.init = ff_MPV_encode_init,
.encode2 = ff_MPV_encode_picture,
.close = ff_MPV_encode_end,
.pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV420P, AV_PIX_FMT_NONE },
.long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 2"),
.priv_class = &msmpeg4v2_class,
}

Definition at line 4182 of file mpegvideo_enc.c.

AVCodec ff_msmpeg4v3_encoder
Initial value:
{
.name = "msmpeg4",
.type = AVMEDIA_TYPE_VIDEO,
.id = AV_CODEC_ID_MSMPEG4V3,
.priv_data_size = sizeof(MpegEncContext),
.init = ff_MPV_encode_init,
.encode2 = ff_MPV_encode_picture,
.close = ff_MPV_encode_end,
.pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV420P, AV_PIX_FMT_NONE },
.long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 3"),
.priv_class = &msmpeg4v3_class,
}

Definition at line 4197 of file mpegvideo_enc.c.

AVCodec ff_wmv1_encoder
Initial value:
{
.name = "wmv1",
.type = AVMEDIA_TYPE_VIDEO,
.id = AV_CODEC_ID_WMV1,
.priv_data_size = sizeof(MpegEncContext),
.init = ff_MPV_encode_init,
.encode2 = ff_MPV_encode_picture,
.close = ff_MPV_encode_end,
.pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV420P, AV_PIX_FMT_NONE },
.long_name = NULL_IF_CONFIG_SMALL("Windows Media Video 7"),
.priv_class = &wmv1_class,
}

Definition at line 4212 of file mpegvideo_enc.c.

const AVClass h263_class
static
Initial value:
{
.class_name = "H.263 encoder",
.item_name = av_default_item_name,
.option = h263_options,
.version = LIBAVUTIL_VERSION_INT,
}

Definition at line 4131 of file mpegvideo_enc.c.

const AVOption h263_options[]
static
Initial value:
{
{ "obmc", "use overlapped block motion compensation.", OFFSET(obmc), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
{ "structured_slices","Write slice start position at every GOB header instead of just GOB number.", OFFSET(h263_slice_structured), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE},
{ "mb_info", "emit macroblock info for RFC 2190 packetization, the parameter value is the maximum payload size", OFFSET(mb_info), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, VE },
FF_MPV_COMMON_OPTS
{ NULL },
}

Definition at line 4123 of file mpegvideo_enc.c.

const AVClass h263p_class
static
Initial value:
{
.class_name = "H.263p encoder",
.item_name = av_default_item_name,
.option = h263p_options,
.version = LIBAVUTIL_VERSION_INT,
}

Definition at line 4159 of file mpegvideo_enc.c.

const AVOption h263p_options[]
static
Initial value:
{
{ "umv", "Use unlimited motion vectors.", OFFSET(umvplus), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
{ "aiv", "Use alternative inter VLC.", OFFSET(alt_inter_vlc), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
{ "obmc", "use overlapped block motion compensation.", OFFSET(obmc), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
{ "structured_slices", "Write slice start position at every GOB header instead of just GOB number.", OFFSET(h263_slice_structured), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE},
FF_MPV_COMMON_OPTS
{ NULL },
}

Definition at line 4151 of file mpegvideo_enc.c.