65 "red",
"yellow",
"green",
"cyan",
"blue",
"magenta",
"white",
"neutral",
"black"
93 #define OFFSET(x) offsetof(SelectiveColorContext, x)
94 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
95 #define RANGE_OPTION(color_name, range) \
96 { color_name"s", "adjust "color_name" regions", OFFSET(opt_cmyk_adjust[range]), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS }
127 #define DECLARE_RANGE_SCALE_FUNCS(nbits) \
128 static int get_neutrals_scale##nbits(int r, int g, int b, int min_val, int max_val) \
131 return (((1<<nbits)-1)*2 - ( abs((max_val<<1) - ((1<<nbits)-1)) \
132 + abs((min_val<<1) - ((1<<nbits)-1))) + 1) >> 1; \
135 static int get_whites_scale##nbits(int r, int g, int b, int min_val, int max_val) \
138 return (min_val<<1) - ((1<<nbits)-1); \
141 static int get_blacks_scale##nbits(int r, int g, int b, int min_val, int max_val) \
144 return ((1<<nbits)-1) - (max_val<<1); \
152 const float *cmyk =
s->cmyk_adjust[
range_id];
156 if (cmyk[0] || cmyk[1] || cmyk[2] || cmyk[3]) {
159 if (cmyk[0] < -1.0 || cmyk[0] > 1.0 ||
160 cmyk[1] < -1.0 || cmyk[1] > 1.0 ||
161 cmyk[2] < -1.0 || cmyk[2] > 1.0 ||
162 cmyk[3] < -1.0 || cmyk[3] > 1.0) {
164 "Settings must be set in [-1;1] range\n",
197 #define READ16(dst) do { \
199 ret = AVERROR_INVALIDDATA; \
202 dst = AV_RB16(buf); \
210 "the settings might not be loaded properly\n",
version);
219 "but %d\n",
"CMYK"[
i],
val);
226 s->cmyk_adjust[
i][k] =
val / 100.;
245 s->is_16bit =
desc->comp[0].depth > 8;
263 const char *opt_cmyk_adjust =
s->opt_cmyk_adjust[
i];
265 if (opt_cmyk_adjust) {
266 float *cmyk =
s->cmyk_adjust[
i];
268 sscanf(
s->opt_cmyk_adjust[
i],
"%f %f %f %f", cmyk, cmyk+1, cmyk+2, cmyk+3);
277 for (
i = 0;
i <
s->nb_process_ranges;
i++) {
279 const float *cmyk =
s->cmyk_adjust[pr->
range_id];
316 #define DECLARE_SELECTIVE_COLOR_FUNC(nbits) \
317 static inline int selective_color_##nbits(AVFilterContext *ctx, ThreadData *td, \
318 int jobnr, int nb_jobs, int direct, int correction_method) \
321 const AVFrame *in = td->in; \
322 AVFrame *out = td->out; \
323 const SelectiveColorContext *s = ctx->priv; \
324 const int height = in->height; \
325 const int width = in->width; \
326 const int slice_start = (height * jobnr ) / nb_jobs; \
327 const int slice_end = (height * (jobnr+1)) / nb_jobs; \
328 const int dst_linesize = out->linesize[0]; \
329 const int src_linesize = in->linesize[0]; \
330 const uint8_t roffset = s->rgba_map[R]; \
331 const uint8_t goffset = s->rgba_map[G]; \
332 const uint8_t boffset = s->rgba_map[B]; \
333 const uint8_t aoffset = s->rgba_map[A]; \
335 for (y = slice_start; y < slice_end; y++) { \
336 uint##nbits##_t *dst = ( uint##nbits##_t *)(out->data[0] + y * dst_linesize); \
337 const uint##nbits##_t *src = (const uint##nbits##_t *)( in->data[0] + y * src_linesize); \
339 for (x = 0; x < width * s->step; x += s->step) { \
340 const int r = src[x + roffset]; \
341 const int g = src[x + goffset]; \
342 const int b = src[x + boffset]; \
343 const int min_color = FFMIN3(r, g, b); \
344 const int max_color = FFMAX3(r, g, b); \
345 const int is_white = (r > 1<<(nbits-1) && g > 1<<(nbits-1) && b > 1<<(nbits-1)); \
346 const int is_neutral = (r || g || b) && \
347 (r != (1<<nbits)-1 || g != (1<<nbits)-1 || b != (1<<nbits)-1); \
348 const int is_black = (r < 1<<(nbits-1) && g < 1<<(nbits-1) && b < 1<<(nbits-1)); \
349 const uint32_t range_flag = (r == max_color) << RANGE_REDS \
350 | (r == min_color) << RANGE_CYANS \
351 | (g == max_color) << RANGE_GREENS \
352 | (g == min_color) << RANGE_MAGENTAS \
353 | (b == max_color) << RANGE_BLUES \
354 | (b == min_color) << RANGE_YELLOWS \
355 | is_white << RANGE_WHITES \
356 | is_neutral << RANGE_NEUTRALS \
357 | is_black << RANGE_BLACKS; \
359 const float rnorm = r * (1.f / ((1<<nbits)-1)); \
360 const float gnorm = g * (1.f / ((1<<nbits)-1)); \
361 const float bnorm = b * (1.f / ((1<<nbits)-1)); \
362 int adjust_r = 0, adjust_g = 0, adjust_b = 0; \
364 for (i = 0; i < s->nb_process_ranges; i++) { \
365 const struct process_range *pr = &s->process_ranges[i]; \
367 if (range_flag & pr->mask) { \
368 const int scale = pr->get_scale(r, g, b, min_color, max_color); \
371 const float *cmyk_adjust = s->cmyk_adjust[pr->range_id]; \
372 const float adj_c = cmyk_adjust[0]; \
373 const float adj_m = cmyk_adjust[1]; \
374 const float adj_y = cmyk_adjust[2]; \
375 const float k = cmyk_adjust[3]; \
377 adjust_r += comp_adjust(scale, rnorm, adj_c, k, correction_method); \
378 adjust_g += comp_adjust(scale, gnorm, adj_m, k, correction_method); \
379 adjust_b += comp_adjust(scale, bnorm, adj_y, k, correction_method); \
384 if (!direct || adjust_r || adjust_g || adjust_b) { \
385 dst[x + roffset] = av_clip_uint##nbits(r + adjust_r); \
386 dst[x + goffset] = av_clip_uint##nbits(g + adjust_g); \
387 dst[x + boffset] = av_clip_uint##nbits(b + adjust_b); \
388 if (!direct && s->step == 4) \
389 dst[x + aoffset] = src[x + aoffset]; \
396 #define DEF_SELECTIVE_COLOR_FUNC(name, direct, correction_method, nbits) \
397 static int selective_color_##name##_##nbits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) \
399 return selective_color_##nbits(ctx, arg, jobnr, nb_jobs, direct, correction_method); \
402 #define DEF_SELECTIVE_COLOR_FUNCS(nbits) \
403 DECLARE_SELECTIVE_COLOR_FUNC(nbits) \
404 DEF_SELECTIVE_COLOR_FUNC(indirect_absolute, 0, CORRECTION_METHOD_ABSOLUTE, nbits) \
405 DEF_SELECTIVE_COLOR_FUNC(indirect_relative, 0, CORRECTION_METHOD_RELATIVE, nbits) \
406 DEF_SELECTIVE_COLOR_FUNC( direct_absolute, 1, CORRECTION_METHOD_ABSOLUTE, nbits) \
407 DEF_SELECTIVE_COLOR_FUNC( direct_relative, 1, CORRECTION_METHOD_RELATIVE, nbits)
424 {selective_color_indirect_absolute_8, selective_color_indirect_relative_8},
425 {selective_color_direct_absolute_8, selective_color_direct_relative_8},
427 {selective_color_indirect_absolute_16, selective_color_indirect_relative_16},
428 {selective_color_direct_absolute_16, selective_color_direct_relative_16},
474 .
name =
"selectivecolor",
480 .priv_class = &selectivecolor_class,
static double val(void *priv, double ch)
static const AVFilterPad inputs[]
static const AVFilterPad outputs[]
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert0(cond)
assert() equivalent, that is always enabled.
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Main libavfilter public API header.
#define flags(name, subs,...)
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
#define AV_LOG_WARNING
Something somehow does not look correct.
#define AV_LOG_VERBOSE
Detailed information.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static void direct(const float *in, const FFTComplex *ir, int len, float *out)
void av_file_unmap(uint8_t *bufptr, size_t size)
Unmap or free the buffer bufptr created by av_file_map().
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...
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
static enum AVPixelFormat pix_fmts[]
static int adjust(int x, int size)
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 ...
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
#define AV_PIX_FMT_RGBA64
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
@ AV_PIX_FMT_BGR0
packed BGR 8:8:8, 32bpp, BGRXBGRX... X=unused/undefined
@ AV_PIX_FMT_ARGB
packed ARGB 8:8:8:8, 32bpp, ARGBARGB...
@ AV_PIX_FMT_BGRA
packed BGRA 8:8:8:8, 32bpp, BGRABGRA...
@ AV_PIX_FMT_ABGR
packed ABGR 8:8:8:8, 32bpp, ABGRABGR...
@ AV_PIX_FMT_0BGR
packed BGR 8:8:8, 32bpp, XBGRXBGR... X=unused/undefined
@ AV_PIX_FMT_RGBA
packed RGBA 8:8:8:8, 32bpp, RGBARGBA...
@ AV_PIX_FMT_RGB0
packed RGB 8:8:8, 32bpp, RGBXRGBX... X=unused/undefined
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
@ AV_PIX_FMT_0RGB
packed RGB 8:8:8, 32bpp, XRGBXRGB... X=unused/undefined
#define AV_PIX_FMT_BGRA64
#define FF_ARRAY_ELEMS(a)
Describe the class of an AVClass context structure.
A link between two filters.
int w
agreed upon image width
int h
agreed upon image height
AVFilterContext * dst
dest filter
int format
agreed upon media format
A filter pad used for either input or output.
const char * name
Pad name.
const char * name
Filter name.
AVFormatInternal * internal
An opaque field for libavformat internal usage.
This structure describes decoded (raw) audio or video data.
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
float cmyk_adjust[NB_RANGES][4]
struct process_range process_ranges[NB_RANGES]
char * opt_cmyk_adjust[NB_RANGES]
Used for passing data between threads.
get_range_scale_func get_scale
#define DEF_SELECTIVE_COLOR_FUNCS(nbits)
static int get_cmy_scale(int r, int g, int b, int min_val, int max_val)
static const AVFilterPad selectivecolor_outputs[]
static const char *const color_names[NB_RANGES]
static int get_rgb_scale(int r, int g, int b, int min_val, int max_val)
static int query_formats(AVFilterContext *ctx)
static const AVOption selectivecolor_options[]
static int config_input(AVFilterLink *inlink)
AVFILTER_DEFINE_CLASS(selectivecolor)
#define DECLARE_RANGE_SCALE_FUNCS(nbits)
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
static int parse_psfile(AVFilterContext *ctx, const char *fname)
AVFilter ff_vf_selectivecolor
int(* selective_color_func_type)(AVFilterContext *ctx, void *td, int jobnr, int nb_jobs)
static int comp_adjust(int scale, float value, float adjust, float k, int correction_method)
static const AVFilterPad selectivecolor_inputs[]
static int register_range(SelectiveColorContext *s, int range_id)
#define RANGE_OPTION(color_name, range)
int(* get_range_scale_func)(int r, int g, int b, int min_val, int max_val)
@ CORRECTION_METHOD_ABSOLUTE
@ CORRECTION_METHOD_RELATIVE
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.