[FFmpeg-cvslog] avfilter: add colorchart video filter
Paul B Mahol
git at videolan.org
Sat Apr 23 12:03:13 EEST 2022
ffmpeg | branch: master | Paul B Mahol <onemda at gmail.com> | Sun Apr 10 19:12:53 2022 +0200| [16463520d9de05481320303dd49ea5158c715c9f] | committer: Paul B Mahol
avfilter: add colorchart video filter
> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=16463520d9de05481320303dd49ea5158c715c9f
---
Changelog | 1 +
doc/filters.texi | 19 ++++++
libavfilter/Makefile | 1 +
libavfilter/allfilters.c | 1 +
libavfilter/version.h | 2 +-
libavfilter/vsrc_testsrc.c | 153 +++++++++++++++++++++++++++++++++++++++++++++
6 files changed, 176 insertions(+), 1 deletion(-)
diff --git a/Changelog b/Changelog
index e028d7dfe0..7a63e3d1ee 100644
--- a/Changelog
+++ b/Changelog
@@ -12,6 +12,7 @@ version 5.1:
- feedback video filter
- pixelize video filter
- colormap video filter
+- colorchart video source filter
version 5.0:
diff --git a/doc/filters.texi b/doc/filters.texi
index 92d0e976d8..a428191f6a 100644
--- a/doc/filters.texi
+++ b/doc/filters.texi
@@ -26287,6 +26287,7 @@ ffplay -f lavfi life=s=300x200:mold=10:r=60:ratio=0.1:death_color=#C83232:life_c
@anchor{allrgb}
@anchor{allyuv}
@anchor{color}
+ at anchor{colorchart}
@anchor{colorspectrum}
@anchor{haldclutsrc}
@anchor{nullsrc}
@@ -26306,6 +26307,8 @@ The @code{allyuv} source returns frames of size 4096x4096 of all yuv colors.
The @code{color} source provides an uniformly colored input.
+The @code{colorchart} source provides a colors checker chart.
+
The @code{colorspectrum} source provides a color spectrum input.
The @code{haldclutsrc} source provides an identity Hald CLUT. See also
@@ -26409,6 +26412,22 @@ Set the type of the color spectrum, only available in the
@item white
@item all
@end table
+
+ at item patch_size
+Set patch size of single color patch, only available in the
+ at code{colorchart} source. Default is @code{64x64}.
+
+ at item preset
+Set colorchecker colors preset, only available in the
+ at code{colorchart} source.
+
+Available values are:
+ at table @samp
+ at item reference
+ at item skintones
+ at end table
+
+Default value is @code{reference}.
@end table
@subsection Examples
diff --git a/libavfilter/Makefile b/libavfilter/Makefile
index 1af07108fb..772454ba23 100644
--- a/libavfilter/Makefile
+++ b/libavfilter/Makefile
@@ -541,6 +541,7 @@ OBJS-$(CONFIG_ALLRGB_FILTER) += vsrc_testsrc.o
OBJS-$(CONFIG_ALLYUV_FILTER) += vsrc_testsrc.o
OBJS-$(CONFIG_CELLAUTO_FILTER) += vsrc_cellauto.o
OBJS-$(CONFIG_COLOR_FILTER) += vsrc_testsrc.o
+OBJS-$(CONFIG_COLORCHART_FILTER) += vsrc_testsrc.o
OBJS-$(CONFIG_COLORSPECTRUM_FILTER) += vsrc_testsrc.o
OBJS-$(CONFIG_COREIMAGESRC_FILTER) += vf_coreimage.o
OBJS-$(CONFIG_FREI0R_SRC_FILTER) += vf_frei0r.o
diff --git a/libavfilter/allfilters.c b/libavfilter/allfilters.c
index 8c7ae17391..7b71918ed6 100644
--- a/libavfilter/allfilters.c
+++ b/libavfilter/allfilters.c
@@ -512,6 +512,7 @@ extern const AVFilter ff_vsrc_allrgb;
extern const AVFilter ff_vsrc_allyuv;
extern const AVFilter ff_vsrc_cellauto;
extern const AVFilter ff_vsrc_color;
+extern const AVFilter ff_vsrc_colorchart;
extern const AVFilter ff_vsrc_colorspectrum;
extern const AVFilter ff_vsrc_coreimagesrc;
extern const AVFilter ff_vsrc_frei0r_src;
diff --git a/libavfilter/version.h b/libavfilter/version.h
index 7b856361c8..547b3b4e7d 100644
--- a/libavfilter/version.h
+++ b/libavfilter/version.h
@@ -31,7 +31,7 @@
#include "version_major.h"
-#define LIBAVFILTER_VERSION_MINOR 34
+#define LIBAVFILTER_VERSION_MINOR 35
#define LIBAVFILTER_VERSION_MICRO 100
diff --git a/libavfilter/vsrc_testsrc.c b/libavfilter/vsrc_testsrc.c
index f859cf2f48..32f13c0c40 100644
--- a/libavfilter/vsrc_testsrc.c
+++ b/libavfilter/vsrc_testsrc.c
@@ -55,6 +55,7 @@
typedef struct TestSourceContext {
const AVClass *class;
int w, h;
+ int pw, ph;
unsigned int nb_frame;
AVRational time_base, frame_rate;
int64_t pts;
@@ -1885,3 +1886,155 @@ const AVFilter ff_vsrc_colorspectrum = {
};
#endif /* CONFIG_COLORSPECTRUM_FILTER */
+
+#if CONFIG_COLORCHART_FILTER
+
+static const AVOption colorchart_options[] = {
+ COMMON_OPTIONS_NOSIZE
+ { "patch_size", "set the single patch size", OFFSET(pw), AV_OPT_TYPE_IMAGE_SIZE, {.str="64x64"}, 0, 0, FLAGS },
+ { "preset", "set the color checker chart preset", OFFSET(type), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "preset" },
+ { "reference", "reference", 0, AV_OPT_TYPE_CONST,{.i64=0}, 0, 0, FLAGS, "preset" },
+ { "skintones", "skintones", 0, AV_OPT_TYPE_CONST,{.i64=1}, 0, 0, FLAGS, "preset" },
+ { NULL }
+};
+
+AVFILTER_DEFINE_CLASS(colorchart);
+
+static const uint8_t reference_colors[][3] = {
+ { 115, 82, 68 }, // dark skin
+ { 194, 150, 130 }, // light skin
+ { 98, 122, 157 }, // blue sky
+ { 87, 108, 67 }, // foliage
+ { 133, 128, 177 }, // blue flower
+ { 103, 189, 170 }, // bluish green
+
+ { 214, 126, 44 }, // orange
+ { 80, 91, 166 }, // purple red
+ { 193, 90, 99 }, // moderate red
+ { 94, 60, 108 }, // purple
+ { 157, 188, 64 }, // yellow green
+ { 224, 163, 46 }, // orange yellow
+
+ { 56, 61, 150 }, // blue
+ { 70, 148, 73 }, // green
+ { 175, 54, 60 }, // red
+ { 233, 199, 31 }, // yellow
+ { 187, 86, 149 }, // magenta
+ { 8, 133, 161 }, // cyan
+
+ { 243, 243, 242 }, // white
+ { 200, 200, 200 }, // neutral 8
+ { 160, 160, 160 }, // neutral 65
+ { 122, 122, 121 }, // neutral 5
+ { 85, 85, 85 }, // neutral 35
+ { 52, 52, 52 }, // black
+};
+
+static const uint8_t skintones_colors[][3] = {
+ { 54, 38, 43 },
+ { 105, 43, 42 },
+ { 147, 43, 43 },
+ { 77, 41, 42 },
+ { 134, 43, 41 },
+ { 201, 134, 118 },
+
+ { 59, 41, 41 },
+ { 192, 103, 76 },
+ { 208, 156, 141 },
+ { 152, 82, 61 },
+ { 162, 132, 118 },
+ { 212, 171, 150 },
+
+ { 205, 91, 31 },
+ { 164, 100, 55 },
+ { 204, 136, 95 },
+ { 178, 142, 116 },
+ { 210, 152, 108 },
+ { 217, 167, 131 },
+
+ { 206, 166, 126 },
+ { 208, 163, 97 },
+ { 245, 180, 0 },
+ { 212, 184, 125 },
+ { 179, 165, 150 },
+ { 196, 184, 105 },
+};
+
+typedef struct ColorChartPreset {
+ int w, h;
+ const uint8_t (*colors)[3];
+} ColorChartPreset;
+
+static const ColorChartPreset colorchart_presets[] = {
+ { 6, 4, reference_colors, },
+ { 6, 4, skintones_colors, },
+};
+
+static int colorchart_config_props(AVFilterLink *inlink)
+{
+ AVFilterContext *ctx = inlink->src;
+ TestSourceContext *s = ctx->priv;
+
+ av_assert0(ff_draw_init(&s->draw, inlink->format, 0) >= 0);
+ if (av_image_check_size(s->w, s->h, 0, ctx) < 0)
+ return AVERROR(EINVAL);
+ return config_props(inlink);
+}
+
+static void colorchart_fill_picture(AVFilterContext *ctx, AVFrame *frame)
+{
+ TestSourceContext *test = ctx->priv;
+ const int preset = test->type;
+ const int w = colorchart_presets[preset].w;
+ const int h = colorchart_presets[preset].h;
+ const int pw = test->pw;
+ const int ph = test->pw;
+
+ for (int y = 0; y < h; y++) {
+ for (int x = 0; x < w; x++) {
+ uint32_t pc = AV_RB24(colorchart_presets[preset].colors[y * w + x]);
+ FFDrawColor color;
+
+ set_color(test, &color, pc);
+ ff_fill_rectangle(&test->draw, &color, frame->data, frame->linesize,
+ x * pw, y * ph, pw, ph);
+ }
+ }
+}
+
+static av_cold int colorchart_init(AVFilterContext *ctx)
+{
+ TestSourceContext *test = ctx->priv;
+ const int preset = test->type;
+ const int w = colorchart_presets[preset].w;
+ const int h = colorchart_presets[preset].h;
+
+ test->w = w * test->pw;
+ test->h = h * test->ph;
+ test->draw_once = 1;
+ test->fill_picture_fn = colorchart_fill_picture;
+ return init(ctx);
+}
+
+static const AVFilterPad avfilter_vsrc_colorchart_outputs[] = {
+ {
+ .name = "default",
+ .type = AVMEDIA_TYPE_VIDEO,
+ .config_props = colorchart_config_props,
+ },
+};
+
+const AVFilter ff_vsrc_colorchart = {
+ .name = "colorchart",
+ .description = NULL_IF_CONFIG_SMALL("Generate color checker chart."),
+ .priv_size = sizeof(TestSourceContext),
+ .priv_class = &colorchart_class,
+ .init = colorchart_init,
+ .uninit = uninit,
+ .activate = activate,
+ .inputs = NULL,
+ FILTER_OUTPUTS(avfilter_vsrc_colorchart_outputs),
+ FILTER_SINGLE_PIXFMT(AV_PIX_FMT_GBRP),
+};
+
+#endif /* CONFIG_COLORCHART_FILTER */
More information about the ffmpeg-cvslog
mailing list