Go to the documentation of this file.
64 "red",
"yellow",
"green",
"cyan",
"blue",
"magenta",
"white",
"neutral",
"black"
92 #define OFFSET(x) offsetof(SelectiveColorContext, x)
93 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
94 #define RANGE_OPTION(color_name, range) \
95 { color_name"s", "adjust "color_name" regions", OFFSET(opt_cmyk_adjust[range]), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS }
126 #define DECLARE_RANGE_SCALE_FUNCS(nbits) \
127 static int get_neutrals_scale##nbits(int r, int g, int b, int min_val, int max_val) \
130 return (((1<<nbits)-1)*2 - ( abs((max_val<<1) - ((1<<nbits)-1)) \
131 + abs((min_val<<1) - ((1<<nbits)-1))) + 1) >> 1; \
134 static int get_whites_scale##nbits(int r, int g, int b, int min_val, int max_val) \
137 return (min_val<<1) - ((1<<nbits)-1); \
140 static int get_blacks_scale##nbits(int r, int g, int b, int min_val, int max_val) \
143 return ((1<<nbits)-1) - (max_val<<1); \
151 const float *cmyk =
s->cmyk_adjust[
range_id];
155 if (cmyk[0] || cmyk[1] || cmyk[2] || cmyk[3]) {
158 if (cmyk[0] < -1.0 || cmyk[0] > 1.0 ||
159 cmyk[1] < -1.0 || cmyk[1] > 1.0 ||
160 cmyk[2] < -1.0 || cmyk[2] > 1.0 ||
161 cmyk[3] < -1.0 || cmyk[3] > 1.0) {
163 "Settings must be set in [-1;1] range\n",
196 #define READ16(dst) do { \
198 ret = AVERROR_INVALIDDATA; \
201 dst = AV_RB16(buf); \
209 "the settings might not be loaded properly\n",
version);
218 "but %d\n",
"CMYK"[
i],
val);
225 s->cmyk_adjust[
i][k] =
val / 100.f;
244 s->is_16bit =
desc->comp[0].depth > 8;
262 const char *opt_cmyk_adjust =
s->opt_cmyk_adjust[
i];
264 if (opt_cmyk_adjust) {
265 float *cmyk =
s->cmyk_adjust[
i];
267 sscanf(
s->opt_cmyk_adjust[
i],
"%f %f %f %f", cmyk, cmyk+1, cmyk+2, cmyk+3);
276 for (
i = 0;
i <
s->nb_process_ranges;
i++) {
278 const float *cmyk =
s->cmyk_adjust[pr->
range_id];
308 #define DECLARE_SELECTIVE_COLOR_FUNC(nbits) \
309 static inline int selective_color_##nbits(AVFilterContext *ctx, ThreadData *td, \
310 int jobnr, int nb_jobs, int direct, int correction_method) \
313 const AVFrame *in = td->in; \
314 AVFrame *out = td->out; \
315 const SelectiveColorContext *s = ctx->priv; \
316 const int height = in->height; \
317 const int width = in->width; \
318 const int slice_start = (height * jobnr ) / nb_jobs; \
319 const int slice_end = (height * (jobnr+1)) / nb_jobs; \
320 const int dst_linesize = out->linesize[0] / ((nbits + 7) / 8); \
321 const int src_linesize = in->linesize[0] / ((nbits + 7) / 8); \
322 const uint8_t roffset = s->rgba_map[R]; \
323 const uint8_t goffset = s->rgba_map[G]; \
324 const uint8_t boffset = s->rgba_map[B]; \
325 const uint8_t aoffset = s->rgba_map[A]; \
326 uint##nbits##_t *dst = (uint##nbits##_t *)out->data[0] + slice_start * dst_linesize; \
327 const uint##nbits##_t *src = (const uint##nbits##_t *)in->data[0] + slice_start * src_linesize; \
328 const uint##nbits##_t *src_r = (const uint##nbits##_t *)src + roffset; \
329 const uint##nbits##_t *src_g = (const uint##nbits##_t *)src + goffset; \
330 const uint##nbits##_t *src_b = (const uint##nbits##_t *)src + boffset; \
331 const uint##nbits##_t *src_a = (const uint##nbits##_t *)src + aoffset; \
332 uint##nbits##_t *dst_r = (uint##nbits##_t *)dst + roffset; \
333 uint##nbits##_t *dst_g = (uint##nbits##_t *)dst + goffset; \
334 uint##nbits##_t *dst_b = (uint##nbits##_t *)dst + boffset; \
335 uint##nbits##_t *dst_a = (uint##nbits##_t *)dst + aoffset; \
336 const int mid = 1<<(nbits-1); \
337 const int max = (1<<nbits)-1; \
338 const float scale = 1.f / max; \
340 for (y = slice_start; y < slice_end; y++) { \
341 for (x = 0; x < width * s->step; x += s->step) { \
342 const int r = src_r[x]; \
343 const int g = src_g[x]; \
344 const int b = src_b[x]; \
345 const int min_color = FFMIN3(r, g, b); \
346 const int max_color = FFMAX3(r, g, b); \
347 const int is_white = (r > mid && g > mid && b > mid); \
348 const int is_neutral = (r || g || b) && \
349 (r != max || g != max || b != max); \
350 const int is_black = (r < mid && g < mid && b < mid); \
351 const uint32_t range_flag = (r == max_color) << RANGE_REDS \
352 | (r == min_color) << RANGE_CYANS \
353 | (g == max_color) << RANGE_GREENS \
354 | (g == min_color) << RANGE_MAGENTAS \
355 | (b == max_color) << RANGE_BLUES \
356 | (b == min_color) << RANGE_YELLOWS \
357 | is_white << RANGE_WHITES \
358 | is_neutral << RANGE_NEUTRALS \
359 | is_black << RANGE_BLACKS; \
361 const float rnorm = r * scale; \
362 const float gnorm = g * scale; \
363 const float bnorm = b * scale; \
364 int adjust_r = 0, adjust_g = 0, adjust_b = 0; \
366 for (i = 0; i < s->nb_process_ranges; i++) { \
367 const struct process_range *pr = &s->process_ranges[i]; \
369 if (range_flag & pr->mask) { \
370 const int scale = pr->get_scale(r, g, b, min_color, max_color); \
373 const float *cmyk_adjust = s->cmyk_adjust[pr->range_id]; \
374 const float adj_c = cmyk_adjust[0]; \
375 const float adj_m = cmyk_adjust[1]; \
376 const float adj_y = cmyk_adjust[2]; \
377 const float k = cmyk_adjust[3]; \
379 adjust_r += comp_adjust(scale, rnorm, adj_c, k, correction_method); \
380 adjust_g += comp_adjust(scale, gnorm, adj_m, k, correction_method); \
381 adjust_b += comp_adjust(scale, bnorm, adj_y, k, correction_method); \
386 if (!direct || adjust_r || adjust_g || adjust_b) { \
387 dst_r[x] = av_clip_uint##nbits(r + adjust_r); \
388 dst_g[x] = av_clip_uint##nbits(g + adjust_g); \
389 dst_b[x] = av_clip_uint##nbits(b + adjust_b); \
390 if (!direct && s->step == 4) \
391 dst_a[x] = src_a[x]; \
395 src_r += src_linesize; \
396 src_g += src_linesize; \
397 src_b += src_linesize; \
398 src_a += src_linesize; \
400 dst_r += dst_linesize; \
401 dst_g += dst_linesize; \
402 dst_b += dst_linesize; \
403 dst_a += dst_linesize; \
408 #define DEF_SELECTIVE_COLOR_FUNC(name, direct, correction_method, nbits) \
409 static int selective_color_##name##_##nbits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) \
411 return selective_color_##nbits(ctx, arg, jobnr, nb_jobs, direct, correction_method); \
414 #define DEF_SELECTIVE_COLOR_FUNCS(nbits) \
415 DECLARE_SELECTIVE_COLOR_FUNC(nbits) \
416 DEF_SELECTIVE_COLOR_FUNC(indirect_absolute, 0, CORRECTION_METHOD_ABSOLUTE, nbits) \
417 DEF_SELECTIVE_COLOR_FUNC(indirect_relative, 0, CORRECTION_METHOD_RELATIVE, nbits) \
418 DEF_SELECTIVE_COLOR_FUNC( direct_absolute, 1, CORRECTION_METHOD_ABSOLUTE, nbits) \
419 DEF_SELECTIVE_COLOR_FUNC( direct_relative, 1, CORRECTION_METHOD_RELATIVE, nbits)
436 {selective_color_indirect_absolute_8, selective_color_indirect_relative_8},
437 {selective_color_direct_absolute_8, selective_color_direct_relative_8},
439 {selective_color_indirect_absolute_16, selective_color_indirect_relative_16},
440 {selective_color_direct_absolute_16, selective_color_direct_relative_16},
477 .
name =
"selectivecolor",
483 .priv_class = &selectivecolor_class,
static int register_range(SelectiveColorContext *s, int range_id)
int(* selective_color_func_type)(AVFilterContext *ctx, void *td, int jobnr, int nb_jobs)
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
#define AV_LOG_WARNING
Something somehow does not look correct.
AVPixelFormat
Pixel format.
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
#define FILTER_PIXFMTS_ARRAY(array)
The exact code depends on how similar the blocks are and how related they are to the and needs to apply these operations to the correct inlink or outlink if there are several Macros are available to factor that when no extra processing is inlink
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
This structure describes decoded (raw) audio or video data.
AVFILTER_DEFINE_CLASS(selectivecolor)
#define AV_LOG_VERBOSE
Detailed information.
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
@ AV_PIX_FMT_BGRA
packed BGRA 8:8:8:8, 32bpp, BGRABGRA...
const char * name
Filter name.
A link between two filters.
#define DECLARE_RANGE_SCALE_FUNCS(nbits)
#define DEF_SELECTIVE_COLOR_FUNCS(nbits)
int av_file_map(const char *filename, uint8_t **bufptr, size_t *size, int log_offset, void *log_ctx)
Read the file with name filename, and put its content in a newly allocated buffer or map it with mmap...
static double val(void *priv, double ch)
struct process_range process_ranges[NB_RANGES]
A filter pad used for either input or output.
static const char *const color_names[NB_RANGES]
static int get_cmy_scale(int r, int g, int b, int min_val, int max_val)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
#define FF_ARRAY_ELEMS(a)
const AVFilterPad ff_video_default_filterpad[1]
An AVFilterPad array whose only entry has name "default" and is of type AVMEDIA_TYPE_VIDEO.
char * opt_cmyk_adjust[NB_RANGES]
static int adjust(int x, int size)
#define av_assert0(cond)
assert() equivalent, that is always enabled.
static int get_rgb_scale(int r, int g, int b, int min_val, int max_val)
#define FILTER_INPUTS(array)
void av_file_unmap(uint8_t *bufptr, size_t size)
Unmap or free the buffer bufptr created by av_file_map().
@ AV_PIX_FMT_RGBA
packed RGBA 8:8:8:8, 32bpp, RGBARGBA...
#define AV_PIX_FMT_RGBA64
static const AVOption selectivecolor_options[]
Describe the class of an AVClass context structure.
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
static int parse_psfile(AVFilterContext *ctx, const char *fname)
@ AV_PIX_FMT_BGR0
packed BGR 8:8:8, 32bpp, BGRXBGRX... X=unused/undefined
const AVFilter ff_vf_selectivecolor
@ AV_PIX_FMT_ABGR
packed ABGR 8:8:8:8, 32bpp, ABGRABGR...
@ CORRECTION_METHOD_ABSOLUTE
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
static void scale(int *out, const int *in, const int w, const int h, const int shift)
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
int av_get_padded_bits_per_pixel(const AVPixFmtDescriptor *pixdesc)
Return the number of bits per pixel for the pixel format described by pixdesc, including any padding ...
static int comp_adjust(int scale, float value, float adjust, float k, int correction_method)
get_range_scale_func get_scale
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
@ AV_PIX_FMT_RGB0
packed RGB 8:8:8, 32bpp, RGBXRGBX... X=unused/undefined
static int config_input(AVFilterLink *inlink)
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
@ AV_PIX_FMT_ARGB
packed ARGB 8:8:8:8, 32bpp, ARGBARGB...
#define AV_PIX_FMT_BGRA64
#define i(width, name, range_min, range_max)
int w
agreed upon image width
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Used for passing data between threads.
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default value
const char * name
Pad name.
@ AV_PIX_FMT_0BGR
packed BGR 8:8:8, 32bpp, XBGRXBGR... X=unused/undefined
float cmyk_adjust[NB_RANGES][4]
#define RANGE_OPTION(color_name, range)
int h
agreed upon image height
int(* get_range_scale_func)(int r, int g, int b, int min_val, int max_val)
static const AVFilterPad selectivecolor_inputs[]
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
static enum AVPixelFormat pix_fmts[]
#define FILTER_OUTPUTS(array)
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
@ AV_PIX_FMT_0RGB
packed RGB 8:8:8, 32bpp, XRGBXRGB... X=unused/undefined
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
static av_always_inline int ff_filter_execute(AVFilterContext *ctx, avfilter_action_func *func, void *arg, int *ret, int nb_jobs)
@ CORRECTION_METHOD_RELATIVE