vf_colormatrix.c File Reference

ColorMatrix 2.0 is based on the original ColorMatrix filter by Wilbert Dijkhof. More...

#include <float.h>
#include "avfilter.h"
#include "formats.h"
#include "internal.h"
#include "video.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/avstring.h"

Go to the source code of this file.

Data Structures
struct	ColorMatrixContext

Macros
#define	NS(n)   ((n) < 0 ? (int)((n)*65536.0-0.5+DBL_EPSILON) : (int)((n)*65536.0+0.5))
#define	CB(n)   av_clip_uint8(n)
#define	OFFSET(x)   offsetof(ColorMatrixContext, x)
#define	FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
#define	ma   m[0][0]
#define	mb   m[0][1]
#define	mc   m[0][2]
#define	md   m[1][0]
#define	me   m[1][1]
#define	mf   m[1][2]
#define	mg   m[2][0]
#define	mh   m[2][1]
#define	mi   m[2][2]
#define	ima   im[0][0]
#define	imb   im[0][1]
#define	imc   im[0][2]
#define	imd   im[1][0]
#define	ime   im[1][1]
#define	imf   im[1][2]
#define	img   im[2][0]
#define	imh   im[2][1]
#define	imi   im[2][2]

Enumerations
enum	ColorMode {   CHANNEL, INTENSITY, NB_CLMODES, COLOR_MODE_NONE = -1,   COLOR_MODE_BT709, COLOR_MODE_FCC, COLOR_MODE_BT601, COLOR_MODE_SMPTE240M,   COLOR_MODE_COUNT }

Functions
	AVFILTER_DEFINE_CLASS (colormatrix)
static void	inverse3x3 (double im[3][3], const double m[3][3])
static void	solve_coefficients (double cm[3][3], double rgb[3][3], const double yuv[3][3])
static void	calc_coefficients (AVFilterContext *ctx)
static av_cold int	init (AVFilterContext *ctx)
static void	process_frame_uyvy422 (ColorMatrixContext *color, AVFrame *dst, AVFrame *src)
static void	process_frame_yuv422p (ColorMatrixContext *color, AVFrame *dst, AVFrame *src)
static void	process_frame_yuv420p (ColorMatrixContext *color, AVFrame *dst, AVFrame *src)
static int	config_input (AVFilterLink *inlink)
static int	query_formats (AVFilterContext *ctx)
static int	filter_frame (AVFilterLink *link, AVFrame *in)

Variables
static const double	yuv_coeff [4][3][3]
static const AVOption	colormatrix_options []
static const char *const	color_modes [] = {"bt709", "fcc", "bt601", "smpte240m"}
static const AVFilterPad	colormatrix_inputs []
static const AVFilterPad	colormatrix_outputs []
AVFilter	ff_vf_colormatrix

Detailed Description

ColorMatrix 2.0 is based on the original ColorMatrix filter by Wilbert Dijkhof.

It adds the ability to convert between any of: Rec.709, FCC, Rec.601, and SMPTE 240M. It also makes pre and post clipping optional, adds an option to use scaled or non-scaled coefficients, and more...

Definition in file vf_colormatrix.c.

Macro Definition Documentation

#define NS

(

n

)

((n) < 0 ? (int)((n)*65536.0-0.5+DBL_EPSILON) : (int)((n)*65536.0+0.5))

Definition at line 40 of file vf_colormatrix.c.

Referenced by calc_coefficients().

#define CB

(

n

)

av_clip_uint8(n)

Definition at line 41 of file vf_colormatrix.c.

Referenced by process_frame_uyvy422(), process_frame_yuv420p(), and process_frame_yuv422p().

#define OFFSET

(

x

)

offsetof(ColorMatrixContext, x)

Definition at line 76 of file vf_colormatrix.c.

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 77 of file vf_colormatrix.c.

#define ma   m[0][0]

Definition at line 93 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define mb   m[0][1]

Definition at line 94 of file vf_colormatrix.c.

Referenced by aic_decode_coeffs(), aic_decode_slice(), calculate_mode_score(), config_props(), decode_mb_info(), decode_mb_row_no_filter(), dnxhd_calc_bits_thread(), dnxhd_encode_fast(), dnxhd_encode_rdo(), dnxhd_encode_thread(), dnxhd_mb_var_thread(), dnxhd_setup_threads_slices(), draw_mandelbrot(), dv_decode_video_segment(), dv_encode_video_segment(), encode_frame(), encode_mode(), escape124_decode_frame(), ff_h264_decode_mb_cabac(), ff_h264_decode_mb_cavlc(), fill_from_cache(), filter_mb_row(), find_quant_thread(), h264_find_frame_end(), init(), inverse3x3(), ivi_decode_blocks(), ivi_process_empty_tile(), main(), mpc7_decode_frame(), parse_value(), quantize(), request_frame(), tm2_read_deltas(), uninit(), and vp78_decode_mv_mb_modes().

#define mc   m[0][2]

Definition at line 95 of file vf_colormatrix.c.

Referenced by create_vorbis_context(), inverse3x3(), ivi_mc(), mc_chroma_dir(), mc_luma_dir(), put_main_header(), and qt_rtp_parse_packet().

#define md   m[1][0]

Definition at line 96 of file vf_colormatrix.c.

Referenced by ac3_sum_square_butterfly_float_c(), ac3_sum_square_butterfly_int32_c(), config_input(), filter_frame(), inverse3x3(), and uninit().

#define me   m[1][1]

Definition at line 97 of file vf_colormatrix.c.

Referenced by backup_duplicate_context(), inverse3x3(), merge_context_after_me(), and read_ts().

#define mf   m[1][2]

Definition at line 98 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define mg   m[2][0]

Definition at line 99 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define mh   m[2][1]

Definition at line 100 of file vf_colormatrix.c.

Referenced by av_file_map(), dwt_decode53(), dwt_decode97_float(), dwt_decode97_int(), dwt_encode53(), dwt_encode97_float(), dwt_encode97_int(), inverse3x3(), mlp_parse(), and read_major_sync().

#define mi   m[2][2]

Definition at line 101 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define ima   im[0][0]

Definition at line 103 of file vf_colormatrix.c.

Referenced by adpcm_compress_trellis(), and inverse3x3().

#define imb   im[0][1]

Definition at line 104 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define imc   im[0][2]

Definition at line 105 of file vf_colormatrix.c.

Referenced by avcodec_register_all(), and inverse3x3().

#define imd   im[1][0]

Definition at line 106 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define ime   im[1][1]

Definition at line 107 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define imf   im[1][2]

Definition at line 108 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define img   im[2][0]

Examples:

qsvdec.c.

Definition at line 109 of file vf_colormatrix.c.

Referenced by apply_color_indexing_transform(), apply_color_transform(), apply_predictor_transform(), apply_subtract_green_transform(), decode_entropy_coded_image(), decode_entropy_image(), decode_packet(), inverse3x3(), mj2_create_image(), parse_transform_color_indexing(), setup_shm(), vp8_decode(), write_header(), write_packet(), write_picture(), and xcbgrab_frame().

#define imh   im[2][1]

Definition at line 110 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define imi   im[2][2]

Definition at line 111 of file vf_colormatrix.c.

Referenced by inverse3x3().

Enumeration Type Documentation

enum ColorMode

Enumerator:

CHANNEL
INTENSITY
NB_CLMODES
COLOR_MODE_NONE
COLOR_MODE_BT709
COLOR_MODE_FCC
COLOR_MODE_BT601
COLOR_MODE_SMPTE240M
COLOR_MODE_COUNT

Definition at line 58 of file vf_colormatrix.c.

Function Documentation

AVFILTER_DEFINE_CLASS

(

colormatrix

)

static void inverse3x3 ( double  im[3][3], const double  m[3][3]  )

static

Definition at line 113 of file vf_colormatrix.c.

Referenced by calc_coefficients().

static void solve_coefficients ( double  cm[3][3], double  rgb[3][3], const double  yuv[3][3]  )

static

Definition at line 128 of file vf_colormatrix.c.

Referenced by calc_coefficients().

static void calc_coefficients ( AVFilterContext *  ctx )

static

Definition at line 136 of file vf_colormatrix.c.

Referenced by filter_frame().

static av_cold int init ( AVFilterContext *  ctx )

static

Definition at line 165 of file vf_colormatrix.c.

static void process_frame_uyvy422 ( ColorMatrixContext *  color, AVFrame *  dst, AVFrame *  src  )

static

Definition at line 182 of file vf_colormatrix.c.

Referenced by filter_frame().

static void process_frame_yuv422p ( ColorMatrixContext *  color, AVFrame *  dst, AVFrame *  src  )

static

Definition at line 214 of file vf_colormatrix.c.

Referenced by filter_frame().

static void process_frame_yuv420p ( ColorMatrixContext *  color, AVFrame *  dst, AVFrame *  src  )

static

Definition at line 256 of file vf_colormatrix.c.

Referenced by filter_frame().

static int config_input ( AVFilterLink *  inlink )

static

Definition at line 304 of file vf_colormatrix.c.

static int query_formats ( AVFilterContext *  ctx )

static

Definition at line 319 of file vf_colormatrix.c.

static int filter_frame ( AVFilterLink *  link, AVFrame *  in  )

static

Definition at line 333 of file vf_colormatrix.c.

Variable Documentation

const double yuv_coeff[4][3][3]

static

Initial value:

= {
    { { +0.7152, +0.0722, +0.2126 }, 
      { -0.3850, +0.5000, -0.1150 },
      { -0.4540, -0.0460, +0.5000 } },
    { { +0.5900, +0.1100, +0.3000 }, 
      { -0.3310, +0.5000, -0.1690 },
      { -0.4210, -0.0790, +0.5000 } },
    { { +0.5870, +0.1140, +0.2990 }, 
      { -0.3313, +0.5000, -0.1687 },
      { -0.4187, -0.0813, +0.5000 } },
    { { +0.7010, +0.0870, +0.2120 }, 
      { -0.3840, +0.5000, -0.1160 },
      { -0.4450, -0.0550, +0.5000 } },
}

Definition at line 43 of file vf_colormatrix.c.

Referenced by calc_coefficients().

const AVOption colormatrix_options[]

static

Initial value:

= {
    { "src", "set source color matrix",      OFFSET(source), AV_OPT_TYPE_INT, {.i64=COLOR_MODE_NONE}, COLOR_MODE_NONE, COLOR_MODE_COUNT-1, .flags=FLAGS, .unit="color_mode" },
    { "dst", "set destination color matrix", OFFSET(dest),   AV_OPT_TYPE_INT, {.i64=COLOR_MODE_NONE}, COLOR_MODE_NONE, COLOR_MODE_COUNT-1, .flags=FLAGS, .unit="color_mode" },
    { "bt709",     "set BT.709 colorspace",      0, AV_OPT_TYPE_CONST, {.i64=COLOR_MODE_BT709},       .flags=FLAGS, .unit="color_mode" },
    { "fcc",       "set FCC colorspace   ",      0, AV_OPT_TYPE_CONST, {.i64=COLOR_MODE_FCC},         .flags=FLAGS, .unit="color_mode" },
    { "bt601",     "set BT.601 colorspace",      0, AV_OPT_TYPE_CONST, {.i64=COLOR_MODE_BT601},       .flags=FLAGS, .unit="color_mode" },
    { "bt470",     "set BT.470 colorspace",      0, AV_OPT_TYPE_CONST, {.i64=COLOR_MODE_BT601},       .flags=FLAGS, .unit="color_mode" },
    { "smpte170m", "set SMTPE-170M colorspace",  0, AV_OPT_TYPE_CONST, {.i64=COLOR_MODE_BT601},       .flags=FLAGS, .unit="color_mode" },
    { "smpte240m", "set SMPTE-240M colorspace",  0, AV_OPT_TYPE_CONST, {.i64=COLOR_MODE_SMPTE240M},   .flags=FLAGS, .unit="color_mode" },
    { NULL }
}

Definition at line 79 of file vf_colormatrix.c.

const char* const color_modes[] = {"bt709", "fcc", "bt601", "smpte240m"}

static

Definition at line 163 of file vf_colormatrix.c.

Referenced by config_input().

const AVFilterPad colormatrix_inputs[]

static

Initial value:

= {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
        .config_props = config_input,
        .filter_frame = filter_frame,
    },
    { NULL }
}

Definition at line 386 of file vf_colormatrix.c.

const AVFilterPad colormatrix_outputs[]

static

Initial value:

= {
    {
        .name = "default",
        .type = AVMEDIA_TYPE_VIDEO,
    },
    { NULL }
}

Definition at line 396 of file vf_colormatrix.c.

AVFilter ff_vf_colormatrix

Initial value:

= {
    .name          = "colormatrix",
    .description   = NULL_IF_CONFIG_SMALL("Convert color matrix."),
    .priv_size     = sizeof(ColorMatrixContext),
    .init          = init,
    .query_formats = query_formats,
    .inputs        = colormatrix_inputs,
    .outputs       = colormatrix_outputs,
    .priv_class    = &colormatrix_class,
    .flags         = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC,
}

Definition at line 404 of file vf_colormatrix.c.