id RoQ encoder by Vitor. More...

#include <string.h>
#include "roqvideo.h"
#include "bytestream.h"
#include "elbg.h"
#include "internal.h"
#include "mathops.h"

Include dependency graph for roqvideoenc.c:

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 255
	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)

Typedefs
typedef struct RoqTempData	RoqTempdata
	Temporary vars. More...

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. 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 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 void	roq_encode_video (RoqContext *enc)

static int	roq_encode_end (AVCodecContext *avctx)

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

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

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

Referenced by motion_search().

#define MAX_CBS_2x2 256

Maximum number of 2x2 codebooks.

Definition at line 73 of file roqvideoenc.c.

Referenced by generate_new_codebooks(), and remap_codebooks().

#define MAX_CBS_4x4 255

Maximum number of generated 4x4 codebooks.

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

Definition at line 71 of file roqvideoenc.c.

Referenced by generate_new_codebooks(), and remap_codebooks().

#define ROQ_LAMBDA_SCALE ((uint64_t) FF_LAMBDA_SCALE)

Definition at line 76 of file roqvideoenc.c.

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

Typedef Documentation

typedef struct RoqTempData RoqTempdata

Temporary vars.

Function Documentation

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

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

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

static

Initialize cel evaluators and set their source coordinates.

Definition at line 246 of file roqvideoenc.c.

Referenced by roq_encode_video().

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

Referenced by generate_new_codebooks().

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

static

Definition at line 144 of file roqvideoenc.c.

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

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

inlinestatic

Definition at line 116 of file roqvideoenc.c.

Referenced by block_sse(), and squared_diff_macroblock().

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

Referenced by create_clusters().

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

Referenced by gather_data_for_cel().

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

static

Definition at line 793 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

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

static

Definition at line 828 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 264 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 279 of file roqvideoenc.c.

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

static uint8_t motion_arg ( motion_vect mot )

inlinestatic

Definition at line 593 of file roqvideoenc.c.

Referenced by reconstruct_and_encode_image().

static void motion_search	(	RoqContext *	enc,
		int	blocksize
	)

static

Definition at line 307 of file roqvideoenc.c.

Referenced by roq_encode_video().

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

static

Definition at line 624 of file roqvideoenc.c.

Referenced by roq_encode_video().

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

static

Definition at line 537 of file roqvideoenc.c.

Referenced by roq_encode_video().

static int roq_encode_end ( AVCodecContext * avctx )

static

Definition at line 939 of file roqvideoenc.c.

Referenced by roq_encode_init().

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

static

Definition at line 1025 of file roqvideoenc.c.

static int roq_encode_init ( AVCodecContext * avctx )

static

Definition at line 955 of file roqvideoenc.c.

static void roq_encode_video ( RoqContext * enc )

static

Definition at line 882 of file roqvideoenc.c.

Referenced by roq_encode_frame().

static void roq_write_video_info_chunk ( RoqContext * enc )

static

Definition at line 1000 of file roqvideoenc.c.

Referenced by roq_encode_frame().

static int square ( int x )

inlinestatic

Definition at line 111 of file roqvideoenc.c.

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

static int squared_diff_macroblock	(	uint8_t	a[],
		uint8_t	b[],
		int	size
	)

inlinestatic

Returns: distortion between two macroblocks

Definition at line 171 of file roqvideoenc.c.

Referenced by gather_data_for_subcel(), and index_mb().

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

static

Definition at line 79 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 86 of file roqvideoenc.c.

Referenced by generate_new_codebooks().

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

static

Write codebook chunk.

Definition at line 569 of file roqvideoenc.c.

Referenced by roq_encode_video().

static void write_typecode	(	CodingSpool *	s,
		uint8_t	type
	)

static

Definition at line 610 of file roqvideoenc.c.

Referenced by reconstruct_and_encode_image().

Variable Documentation

AVCodec ff_roq_encoder

Initial value:

= {
    .name                 = "roqvideo",
    .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,
    .supported_framerates = (const AVRational[]){ {30,1}, {0,0} },
    .pix_fmts             = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV444P,
                                                        AV_PIX_FMT_NONE },
    .long_name            = NULL_IF_CONFIG_SMALL("id RoQ video"),
}

Definition at line 1075 of file roqvideoenc.c.

Paris Hackday Code

Data Structures

Macros

Typedefs

Functions

Variables

Detailed Description

Macro Definition Documentation

Typedef Documentation

Function Documentation

Variable Documentation