roqvideoenc.c File Reference

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
#define EVAL_MOTION (   MOTION)
Value:
do { \
diff = eval_motion_dist(enc, j, i, MOTION, blocksize); \
if (diff < lowestdiff) { \
lowestdiff = diff; \
bestpick = MOTION; \
} \
} while(0)
static int eval_motion_dist(RoqContext *enc, int x, int y, motion_vect vect, int size)
Definition: roqvideoenc.c:144
initialize output if(nPeaks >3)%at least 3 peaks in spectrum for trying to find f0 nf0peaks
#define diff(a, as, b, bs)
Definition: vf_phase.c:80
synthesis window for stochastic i

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
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",
.priv_data_size = sizeof(RoqContext),
.encode2 = roq_encode_frame,
.supported_framerates = (const AVRational[]){ {30,1}, {0,0} },
.pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV444P,
.long_name = NULL_IF_CONFIG_SMALL("id RoQ video"),
}
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:73
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
static int roq_encode_init(AVCodecContext *avctx)
Definition: roqvideoenc.c:955
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
struct RoqContext RoqContext
static int roq_encode_end(AVCodecContext *avctx)
Definition: roqvideoenc.c:939
static void close(AVCodecParserContext *s)
Definition: h264_parser.c:375
rational number numerator/denominator
Definition: rational.h:43
static int roq_encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)
Definition: roqvideoenc.c:1025
AVPixelFormat
Pixel format.
Definition: pixfmt.h:66

Definition at line 1075 of file roqvideoenc.c.