FFmpeg
Data Structures | Macros | Functions | Variables
roqvideoenc.c File Reference
#include <string.h>
#include "libavutil/attributes.h"
#include "libavutil/opt.h"
#include "roqvideo.h"
#include "bytestream.h"
#include "elbg.h"
#include "internal.h"
#include "mathops.h"

Go to the source code of this file.

Data Structures

struct  SubcelEvaluation
 
struct  CelEvaluation
 
struct  RoqCodebooks
 
struct  RoqTempData
 Temporary vars. More...
 
struct  CodingSpool
 

Macros

#define CHROMA_BIAS   1
 
#define MAX_CBS_4x4   256
 Maximum number of generated 4x4 codebooks. More...
 
#define MAX_CBS_2x2   256
 Maximum number of 2x2 codebooks. More...
 
#define ROQ_LAMBDA_SCALE   ((uint64_t) FF_LAMBDA_SCALE)
 
#define EVAL_MOTION(MOTION)
 
#define OFFSET(x)   offsetof(RoqContext, x)
 
#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
 

Functions

static void unpack_roq_cell (roq_cell *cell, uint8_t u[4 *3])
 
static void unpack_roq_qcell (uint8_t cb2[], roq_qcell *qcell, uint8_t u[4 *4 *3])
 
static void enlarge_roq_mb4 (uint8_t base[3 *16], uint8_t u[3 *64])
 
static int square (int x)
 
static int eval_sse (const uint8_t *a, const uint8_t *b, int count)
 
static int block_sse (uint8_t *const *buf1, uint8_t *const *buf2, int x1, int y1, int x2, int y2, const int *stride1, const int *stride2, int size)
 
static int eval_motion_dist (RoqContext *enc, int x, int y, motion_vect vect, int size)
 
static int squared_diff_macroblock (uint8_t a[], uint8_t b[], int size)
 
static int create_cel_evals (RoqContext *enc, RoqTempdata *tempData)
 Initialize cel evaluators and set their source coordinates. More...
 
static void get_frame_mb (const AVFrame *frame, int x, int y, uint8_t mb[], int dim)
 Get macroblocks from parts of the image. More...
 
static int index_mb (uint8_t cluster[], uint8_t cb[], int numCB, int *outIndex, int dim)
 Find the codebook with the lowest distortion from an image. More...
 
static void motion_search (RoqContext *enc, int blocksize)
 
static void gather_data_for_subcel (SubcelEvaluation *subcel, int x, int y, RoqContext *enc, RoqTempdata *tempData)
 Get distortion for all options available to a subcel. More...
 
static void gather_data_for_cel (CelEvaluation *cel, RoqContext *enc, RoqTempdata *tempData)
 Get distortion for all options available to a cel. More...
 
static void remap_codebooks (RoqContext *enc, RoqTempdata *tempData)
 
static void write_codebooks (RoqContext *enc, RoqTempdata *tempData)
 Write codebook chunk. More...
 
static uint8_t motion_arg (motion_vect mot)
 
static void write_typecode (CodingSpool *s, uint8_t type)
 
static void reconstruct_and_encode_image (RoqContext *enc, RoqTempdata *tempData, int w, int h, int numBlocks)
 
static void frame_block_to_cell (uint8_t *block, uint8_t *const *data, int top, int left, const int *stride)
 Create a single YUV cell from a 2x2 section of the image. More...
 
static void create_clusters (const AVFrame *frame, int w, int h, uint8_t *yuvClusters)
 Create YUV clusters for the entire image. More...
 
static int generate_codebook (RoqContext *enc, RoqTempdata *tempdata, int *points, int inputCount, roq_cell *results, int size, int cbsize)
 
static int generate_new_codebooks (RoqContext *enc, RoqTempdata *tempData)
 
static int roq_encode_video (RoqContext *enc)
 
static av_cold int roq_encode_end (AVCodecContext *avctx)
 
static av_cold int roq_encode_init (AVCodecContext *avctx)
 
static void roq_write_video_info_chunk (RoqContext *enc)
 
static int roq_encode_frame (AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)
 

Variables

static const AVOption options []
 
static const AVClass roq_class
 
AVCodec ff_roq_encoder
 

Detailed Description

id RoQ encoder by Vitor. Based on the Switchblade3 library and the Switchblade3 FFmpeg glue by Eric Lasota.

Definition in file roqvideoenc.c.

Macro Definition Documentation

◆ CHROMA_BIAS

#define CHROMA_BIAS   1

Definition at line 67 of file roqvideoenc.c.

◆ MAX_CBS_4x4

#define MAX_CBS_4x4   256

Maximum number of generated 4x4 codebooks.

Can't be 256 to workaround a Quake 3 bug.

Definition at line 73 of file roqvideoenc.c.

◆ MAX_CBS_2x2

#define MAX_CBS_2x2   256

Maximum number of 2x2 codebooks.

Definition at line 75 of file roqvideoenc.c.

◆ ROQ_LAMBDA_SCALE

#define ROQ_LAMBDA_SCALE   ((uint64_t) FF_LAMBDA_SCALE)

Definition at line 78 of file roqvideoenc.c.

◆ EVAL_MOTION

#define EVAL_MOTION (   MOTION)
Value:
do { \
diff = eval_motion_dist(enc, j, i, MOTION, blocksize); \
if (diff < lowestdiff) { \
lowestdiff = diff; \
bestpick = MOTION; \
} \
} while(0)

Definition at line 300 of file roqvideoenc.c.

◆ OFFSET

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

Definition at line 1112 of file roqvideoenc.c.

◆ VE

Definition at line 1113 of file roqvideoenc.c.

Function Documentation

◆ unpack_roq_cell()

static void unpack_roq_cell ( roq_cell cell,
uint8_t  u[4 *3] 
)
static

Definition at line 81 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

◆ unpack_roq_qcell()

static void unpack_roq_qcell ( uint8_t  cb2[],
roq_qcell qcell,
uint8_t  u[4 *4 *3] 
)
static

Definition at line 88 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

◆ enlarge_roq_mb4()

static void enlarge_roq_mb4 ( uint8_t  base[3 *16],
uint8_t  u[3 *64] 
)
static

Definition at line 103 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

◆ square()

static int square ( int  x)
inlinestatic

◆ eval_sse()

static int eval_sse ( const uint8_t a,
const uint8_t b,
int  count 
)
inlinestatic

Definition at line 118 of file roqvideoenc.c.

Referenced by block_sse(), and squared_diff_macroblock().

◆ block_sse()

static int block_sse ( uint8_t *const *  buf1,
uint8_t *const *  buf2,
int  x1,
int  y1,
int  x2,
int  y2,
const int stride1,
const int stride2,
int  size 
)
static

Definition at line 130 of file roqvideoenc.c.

Referenced by eval_motion_dist(), gather_data_for_cel(), and gather_data_for_subcel().

◆ eval_motion_dist()

static int eval_motion_dist ( RoqContext enc,
int  x,
int  y,
motion_vect  vect,
int  size 
)
static

Definition at line 146 of file roqvideoenc.c.

Referenced by gather_data_for_cel(), gather_data_for_subcel(), and motion_search().

◆ squared_diff_macroblock()

static int squared_diff_macroblock ( uint8_t  a[],
uint8_t  b[],
int  size 
)
inlinestatic
Returns
distortion between two macroblocks

Definition at line 173 of file roqvideoenc.c.

Referenced by gather_data_for_subcel(), and index_mb().

◆ create_cel_evals()

static int create_cel_evals ( RoqContext enc,
RoqTempdata *  tempData 
)
static

Initialize cel evaluators and set their source coordinates.

Definition at line 245 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ get_frame_mb()

static void get_frame_mb ( const AVFrame frame,
int  x,
int  y,
uint8_t  mb[],
int  dim 
)
static

Get macroblocks from parts of the image.

Definition at line 267 of file roqvideoenc.c.

Referenced by gather_data_for_cel(), and gather_data_for_subcel().

◆ index_mb()

static int index_mb ( uint8_t  cluster[],
uint8_t  cb[],
int  numCB,
int outIndex,
int  dim 
)
static

Find the codebook with the lowest distortion from an image.

Definition at line 282 of file roqvideoenc.c.

Referenced by gather_data_for_cel(), gather_data_for_subcel(), and generate_new_codebooks().

◆ motion_search()

static void motion_search ( RoqContext enc,
int  blocksize 
)
static

Definition at line 310 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ gather_data_for_subcel()

static void gather_data_for_subcel ( SubcelEvaluation subcel,
int  x,
int  y,
RoqContext enc,
RoqTempdata *  tempData 
)
static

Get distortion for all options available to a subcel.

Definition at line 402 of file roqvideoenc.c.

Referenced by gather_data_for_cel().

◆ gather_data_for_cel()

static void gather_data_for_cel ( CelEvaluation cel,
RoqContext enc,
RoqTempdata *  tempData 
)
static

Get distortion for all options available to a cel.

Definition at line 466 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ remap_codebooks()

static void remap_codebooks ( RoqContext enc,
RoqTempdata *  tempData 
)
static

Definition at line 540 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ write_codebooks()

static void write_codebooks ( RoqContext enc,
RoqTempdata *  tempData 
)
static

Write codebook chunk.

Definition at line 572 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ motion_arg()

static uint8_t motion_arg ( motion_vect  mot)
inlinestatic

Definition at line 596 of file roqvideoenc.c.

Referenced by reconstruct_and_encode_image().

◆ write_typecode()

static void write_typecode ( CodingSpool s,
uint8_t  type 
)
static

Definition at line 612 of file roqvideoenc.c.

Referenced by reconstruct_and_encode_image().

◆ reconstruct_and_encode_image()

static void reconstruct_and_encode_image ( RoqContext enc,
RoqTempdata *  tempData,
int  w,
int  h,
int  numBlocks 
)
static

Definition at line 626 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ frame_block_to_cell()

static void frame_block_to_cell ( uint8_t block,
uint8_t *const *  data,
int  top,
int  left,
const int stride 
)
inlinestatic

Create a single YUV cell from a 2x2 section of the image.

Definition at line 745 of file roqvideoenc.c.

Referenced by create_clusters().

◆ create_clusters()

static void create_clusters ( const AVFrame frame,
int  w,
int  h,
uint8_t yuvClusters 
)
static

Create YUV clusters for the entire image.

Definition at line 766 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

◆ generate_codebook()

static int generate_codebook ( RoqContext enc,
RoqTempdata *  tempdata,
int points,
int  inputCount,
roq_cell results,
int  size,
int  cbsize 
)
static

Definition at line 780 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

◆ generate_new_codebooks()

static int generate_new_codebooks ( RoqContext enc,
RoqTempdata *  tempData 
)
static

Definition at line 828 of file roqvideoenc.c.

Referenced by roq_encode_video().

◆ roq_encode_video()

static int roq_encode_video ( RoqContext enc)
static

Definition at line 896 of file roqvideoenc.c.

Referenced by roq_encode_frame().

◆ roq_encode_end()

static av_cold int roq_encode_end ( AVCodecContext avctx)
static

Definition at line 961 of file roqvideoenc.c.

Referenced by roq_encode_init().

◆ roq_encode_init()

static av_cold int roq_encode_init ( AVCodecContext avctx)
static

Definition at line 977 of file roqvideoenc.c.

◆ roq_write_video_info_chunk()

static void roq_write_video_info_chunk ( RoqContext enc)
static

Definition at line 1035 of file roqvideoenc.c.

Referenced by roq_encode_frame().

◆ roq_encode_frame()

static int roq_encode_frame ( AVCodecContext avctx,
AVPacket pkt,
const AVFrame frame,
int got_packet 
)
static

Definition at line 1060 of file roqvideoenc.c.

Variable Documentation

◆ options

const AVOption options[]
static
Initial value:
= {
{ "quake3_compat", "Whether to respect known limitations in Quake 3 decoder", OFFSET(quake3_compat), AV_OPT_TYPE_BOOL, { .i64 = 1 }, 0, 1, VE },
{ NULL },
}

Definition at line 1114 of file roqvideoenc.c.

◆ roq_class

const AVClass roq_class
static
Initial value:
= {
.class_name = "RoQ",
.item_name = av_default_item_name,
.option = options,
}

Definition at line 1119 of file roqvideoenc.c.

◆ ff_roq_encoder

AVCodec ff_roq_encoder
Initial value:
= {
.name = "roqvideo",
.long_name = NULL_IF_CONFIG_SMALL("id RoQ video"),
.priv_data_size = sizeof(RoqContext),
.encode2 = roq_encode_frame,
.close = roq_encode_end,
.priv_class = &roq_class,
}

Definition at line 1126 of file roqvideoenc.c.

AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
init
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
VE
#define VE
Definition: roqvideoenc.c:1113
RoqContext
Definition: roqvideo.h:44
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:275
AV_PIX_FMT_YUVJ444P
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:80
if
if(ret)
Definition: filter_design.txt:179
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
NULL
#define NULL
Definition: coverity.c:32
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:191
roq_encode_init
static av_cold int roq_encode_init(AVCodecContext *avctx)
Definition: roqvideoenc.c:977
AV_CODEC_ID_ROQ
@ AV_CODEC_ID_ROQ
Definition: avcodec.h:256
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:188
roq_encode_end
static av_cold int roq_encode_end(AVCodecContext *avctx)
Definition: roqvideoenc.c:961
options
static const AVOption options[]
Definition: roqvideoenc.c:1114
eval_motion_dist
static int eval_motion_dist(RoqContext *enc, int x, int y, motion_vect vect, int size)
Definition: roqvideoenc.c:146
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:259
roq_encode_frame
static int roq_encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)
Definition: roqvideoenc.c:1060
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
diff
static av_always_inline int diff(const uint32_t a, const uint32_t b)
Definition: vf_palettegen.c:136
OFFSET
#define OFFSET(x)
Definition: roqvideoenc.c:1112
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Definition: opt.h:240
roq_class
static const AVClass roq_class
Definition: roqvideoenc.c:1119