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"

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.

#define	MAX_CBS_2x2 256
	Maximum number of 2x2 codebooks.

#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 void	create_cel_evals (RoqContext enc, RoqTempdata tempData)
	Initialize cel evaluators and set their source coordinates.

static void	get_frame_mb (const AVFrame *frame, int x, int y, uint8_t mb[], int dim)
	Get macroblocks from parts of the image.

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.

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.

static void	gather_data_for_cel (CelEvaluation cel, RoqContext enc, RoqTempdata *tempData)
	Get distortion for all options available to a cel.

static void	remap_codebooks (RoqContext enc, RoqTempdata tempData)

static void	write_codebooks (RoqContext enc, RoqTempdata tempData)
	Write codebook chunk.

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.

static void	create_clusters (const AVFrame frame, int w, int h, uint8_t yuvClusters)
	Create YUV clusters for the entire image.

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

static void	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

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

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

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

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

#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 299 of file roqvideoenc.c.

Referenced by motion_search().

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

Definition at line 1089 of file roqvideoenc.c.

#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 1090 of file roqvideoenc.c.

Function Documentation

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().

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().

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().

static int square ( int x )

inlinestatic

Definition at line 113 of file roqvideoenc.c.

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

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().

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(), gather_data_for_subcel(), and reconstruct_and_encode_image().

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().

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().

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

static

Initialize cel evaluators and set their source coordinates.

Definition at line 248 of file roqvideoenc.c.

Referenced by roq_encode_video().

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 266 of file roqvideoenc.c.

Referenced by gather_data_for_cel(), and gather_data_for_subcel().

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 281 of file roqvideoenc.c.

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

static void motion_search	(	RoqContext *	enc,
		int	blocksize
	)

static

Definition at line 309 of file roqvideoenc.c.

Referenced by roq_encode_video().

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 401 of file roqvideoenc.c.

Referenced by 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 465 of file roqvideoenc.c.

Referenced by roq_encode_video().

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

static

Definition at line 539 of file roqvideoenc.c.

Referenced by roq_encode_video().

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

static

Write codebook chunk.

Definition at line 571 of file roqvideoenc.c.

Referenced by roq_encode_video().

static uint8_t motion_arg ( motion_vect mot )

inlinestatic

Definition at line 595 of file roqvideoenc.c.

Referenced by reconstruct_and_encode_image().

static void write_typecode	(	CodingSpool *	s,
		uint8_t	type
	)

static

Definition at line 612 of file roqvideoenc.c.

Referenced by 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().

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 760 of file roqvideoenc.c.

Referenced by 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 781 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

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

static

Definition at line 795 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

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

static

Definition at line 830 of file roqvideoenc.c.

Referenced by roq_encode_video().

static int roq_encode_video ( RoqContext * enc )

static

Definition at line 884 of file roqvideoenc.c.

Referenced by roq_encode_frame().

static av_cold int roq_encode_end ( AVCodecContext * avctx )

static

Definition at line 947 of file roqvideoenc.c.

Referenced by roq_encode_init().

static av_cold int roq_encode_init ( AVCodecContext * avctx )

static

Definition at line 963 of file roqvideoenc.c.

static void roq_write_video_info_chunk ( RoqContext * enc )

static

Definition at line 1013 of file roqvideoenc.c.

Referenced by roq_encode_frame().

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

static

Definition at line 1038 of file roqvideoenc.c.

Variable Documentation

const AVOption options[]

static

Initial value:

= {
    { "quake3_compat", "Whether to respect known limitations in Quake 3 decoder", OFFSET(quake3_compat), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE },
    { NULL },
}

Definition at line 1091 of file roqvideoenc.c.

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 1096 of file roqvideoenc.c.

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 1103 of file roqvideoenc.c.

Data Structures

Macros

Functions

Variables

Detailed Description

Macro Definition Documentation

Function Documentation

Variable Documentation