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"

Include dependency graph for vf_colormatrix.c:

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 *	color_modes [] = {"bt709", "fcc", "bt601", "smpte240m"}
static const AVFilterPad	colormatrix_inputs []
static const AVFilterPad	colormatrix_outputs []
AVFilter	avfilter_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 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 FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 77 of file vf_colormatrix.c.

#define ima   im[0][0]

Definition at line 101 of file vf_colormatrix.c.

Referenced by adpcm_compress_trellis(), and inverse3x3().

#define imb   im[0][1]

Definition at line 102 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define imc   im[0][2]

Definition at line 103 of file vf_colormatrix.c.

Referenced by avcodec_register_all(), and inverse3x3().

#define imd   im[1][0]

Definition at line 104 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define ime   im[1][1]

Definition at line 105 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define imf   im[1][2]

Definition at line 106 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define img   im[2][0]

Definition at line 107 of file vf_colormatrix.c.

Referenced by altivec_uyvy_rgb32(), inverse3x3(), mj2_create_image(), vp8_decode(), write_header(), and write_packet().

#define imh   im[2][1]

Definition at line 108 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define imi   im[2][2]

Definition at line 109 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define ma   m[0][0]

Definition at line 91 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define mb   m[0][1]

Definition at line 92 of file vf_colormatrix.c.

Referenced by config_props(), decode_mb_info(), 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(), encode_frame(), escape124_decode_frame(), ff_h264_decode_mb_cabac(), ff_h264_decode_mb_cavlc(), ff_h264_find_frame_end(), fill_from_cache(), find_quant_thread(), init(), inverse3x3(), ivi_decode_blocks(), ivi_process_empty_tile(), main(), mpc7_decode_frame(), request_frame(), tm2_read_deltas(), uninit(), vp8_decode_mb_row_no_filter(), vp8_decode_mv_mb_modes(), and vp8_filter_mb_row().

#define mc   m[0][2]

Definition at line 93 of file vf_colormatrix.c.

Referenced by create_vorbis_context(), inverse3x3(), put_main_header(), and qt_rtp_parse_packet().

#define md   m[1][0]

Definition at line 94 of file vf_colormatrix.c.

Referenced by ac3_sum_square_butterfly_float_c(), ac3_sum_square_butterfly_int32_c(), ff_brktimegm(), and inverse3x3().

#define me   m[1][1]

Definition at line 95 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 96 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define mg   m[2][0]

Definition at line 97 of file vf_colormatrix.c.

Referenced by inverse3x3().

#define mh   m[2][1]

Definition at line 98 of file vf_colormatrix.c.

Referenced by av_file_map(), dwt_decode53(), dwt_decode97(), dwt_encode53(), dwt_encode97(), inverse3x3(), mlp_parse(), and read_major_sync().

#define mi   m[2][2]

Definition at line 99 of file vf_colormatrix.c.

Referenced by inverse3x3().

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

(

x

)

offsetof(ColorMatrixContext, x)

Definition at line 76 of file vf_colormatrix.c.

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 calc_coefficients ( AVFilterContext *  ctx )

static

Definition at line 134 of file vf_colormatrix.c.

Referenced by init().

static int config_input ( AVFilterLink *  inlink )

static

Definition at line 306 of file vf_colormatrix.c.

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

static

Definition at line 335 of file vf_colormatrix.c.

static av_cold int init ( AVFilterContext *  ctx )

static

Definition at line 163 of file vf_colormatrix.c.

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

static

Definition at line 111 of file vf_colormatrix.c.

Referenced by calc_coefficients().

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

static

Definition at line 184 of file vf_colormatrix.c.

Referenced by filter_frame().

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

static

Definition at line 258 of file vf_colormatrix.c.

Referenced by filter_frame().

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

static

Definition at line 216 of file vf_colormatrix.c.

Referenced by filter_frame().

static int query_formats ( AVFilterContext *  ctx )

static

Definition at line 321 of file vf_colormatrix.c.

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

static

Definition at line 126 of file vf_colormatrix.c.

Referenced by calc_coefficients().

Variable Documentation

AVFilter avfilter_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,
}

Definition at line 378 of file vf_colormatrix.c.

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

static

Definition at line 161 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 360 of file vf_colormatrix.c.

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" },
    { "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 AVFilterPad colormatrix_outputs[]

static

Initial value:

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

Definition at line 370 of file vf_colormatrix.c.

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