roqvideoenc.c File Reference

id RoQ encoder by Vitor. More...

#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.

Referenced by block_sse(), generate_codebook(), generate_new_codebooks(), and squared_diff_macroblock().

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.

Referenced by generate_new_codebooks(), and remap_codebooks().

MAX_CBS_2x2

#define MAX_CBS_2x2   256

Maximum number of 2x2 codebooks.

Definition at line 75 of file roqvideoenc.c.

Referenced by generate_new_codebooks(), and remap_codebooks().

ROQ_LAMBDA_SCALE

#define ROQ_LAMBDA_SCALE   ((uint64_t) FF_LAMBDA_SCALE)

Definition at line 78 of file roqvideoenc.c.

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

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.

Referenced by motion_search().

OFFSET

#define OFFSET

(

x

)

offsetof(RoqContext, x)

Definition at line 1112 of file roqvideoenc.c.

VE

#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

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

Definition at line 113 of file roqvideoenc.c.

Referenced by dpcm_decode_init(), eval_sse(), fixed_sqrt(), iterative_me(), and xan_wc3_decode_frame().

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,
    .version    = LIBAVUTIL_VERSION_INT,
}

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"),
    .type                 = AVMEDIA_TYPE_VIDEO,
    .id                   = AV_CODEC_ID_ROQ,
    .priv_data_size       = sizeof(RoqContext),
    .init                 = roq_encode_init,
    .encode2              = roq_encode_frame,
    .close                = roq_encode_end,
    .pix_fmts             = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUVJ444P,
                                                        AV_PIX_FMT_NONE },
    .priv_class     = &roq_class,
}

Definition at line 1126 of file roqvideoenc.c.