virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

lavfi/curves: make use of options to store the preset names.

Browse Source
This commit is contained in:
Clément Bœsch 2013-04-01 20:37:45 +02:00
parent 2208a46f8f
commit 6278bc8a6c
1 changed files with 56 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
libavfilter/vf_curves.c
View File

@ -33,9 +33,24 @@ struct keypoint {
#define NB_COMP 3 #define NB_COMP 3
enum preset {
PRESET_NONE,
PRESET_COLOR_NEGATIVE,
PRESET_CROSS_PROCESS,
PRESET_DARKER,
PRESET_INCREASE_CONTRAST,
PRESET_LIGHTER,
PRESET_LINEAR_CONTRAST,
PRESET_MEDIUM_CONTRAST,
PRESET_NEGATIVE,
PRESET_STRONG_CONTRAST,
PRESET_VINTAGE,
NB_PRESETS,
};
typedef struct { typedef struct {
const AVClass *class; const AVClass *class;
char *preset; enum preset preset;
char *comp_points_str[NB_COMP]; char *comp_points_str[NB_COMP];
uint8_t graph[NB_COMP][256]; uint8_t graph[NB_COMP][256];
} CurvesContext; } CurvesContext;
@ -49,55 +64,67 @@ static const AVOption curves_options[] = {
{ "g", "set green points coordinates", OFFSET(comp_points_str[1]), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS }, { "g", "set green points coordinates", OFFSET(comp_points_str[1]), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS },
{ "blue", "set blue points coordinates", OFFSET(comp_points_str[2]), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS }, { "blue", "set blue points coordinates", OFFSET(comp_points_str[2]), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS },
{ "b", "set blue points coordinates", OFFSET(comp_points_str[2]), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS }, { "b", "set blue points coordinates", OFFSET(comp_points_str[2]), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS },
{ "preset", "select a color curves preset", OFFSET(preset), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS }, { "preset", "select a color curves preset", OFFSET(preset), AV_OPT_TYPE_INT, {.i64=PRESET_NONE}, PRESET_NONE, NB_PRESETS-1, FLAGS, "preset_name" },
{ "color_negative", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_COLOR_NEGATIVE}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "cross_process", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_CROSS_PROCESS}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "darker", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_DARKER}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "increase_contrast", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_INCREASE_CONTRAST}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "lighter", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_LIGHTER}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "linear_contrast", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_LINEAR_CONTRAST}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "medium_contrast", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_MEDIUM_CONTRAST}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "negative", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_NEGATIVE}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "strong_contrast", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_STRONG_CONTRAST}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ "vintage", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PRESET_VINTAGE}, INT_MIN, INT_MAX, FLAGS, "preset_name" },
{ NULL } { NULL }
}; };
AVFILTER_DEFINE_CLASS(curves); AVFILTER_DEFINE_CLASS(curves);
static const struct { static const struct {
const char *name;
const char *r; const char *r;
const char *g; const char *g;
const char *b; const char *b;
} curves_presets[] = { { } curves_presets[] = {
"color_negative", [PRESET_COLOR_NEGATIVE] = {
"0/1 0.129/1 0.466/0.498 0.725/0 1/0", "0/1 0.129/1 0.466/0.498 0.725/0 1/0",
"0/1 0.109/1 0.301/0.498 0.517/0 1/0", "0/1 0.109/1 0.301/0.498 0.517/0 1/0",
"0/1 0.098/1 0.235/0.498 0.423/0 1/0", "0/1 0.098/1 0.235/0.498 0.423/0 1/0",
},{ },
"cross_process", [PRESET_CROSS_PROCESS] = {
"0.25/0.156 0.501/0.501 0.686/0.745", "0.25/0.156 0.501/0.501 0.686/0.745",
"0.25/0.188 0.38/0.501 0.745/0.815 1/0.815", "0.25/0.188 0.38/0.501 0.745/0.815 1/0.815",
"0.231/0.094 0.709/0.874", "0.231/0.094 0.709/0.874",
},{ },
"darker", "0.5/0.4", "0.5/0.4", "0.5/0.4", [PRESET_DARKER] = {
},{ "0.5/0.4", "0.5/0.4", "0.5/0.4",
"increase_contrast", },
[PRESET_INCREASE_CONTRAST] = {
"0.149/0.066 0.831/0.905 0.905/0.98", "0.149/0.066 0.831/0.905 0.905/0.98",
"0.149/0.066 0.831/0.905 0.905/0.98", "0.149/0.066 0.831/0.905 0.905/0.98",
"0.149/0.066 0.831/0.905 0.905/0.98", "0.149/0.066 0.831/0.905 0.905/0.98",
},{ },
"lighter", "0.4/0.5", "0.4/0.5", "0.4/0.5", [PRESET_LIGHTER] = {
},{ "0.4/0.5", "0.4/0.5", "0.4/0.5",
"linear_contrast", },
[PRESET_LINEAR_CONTRAST] = {
"0.305/0.286 0.694/0.713", "0.305/0.286 0.694/0.713",
"0.305/0.286 0.694/0.713", "0.305/0.286 0.694/0.713",
"0.305/0.286 0.694/0.713", "0.305/0.286 0.694/0.713",
},{ },
"medium_contrast", [PRESET_MEDIUM_CONTRAST] = {
"0.286/0.219 0.639/0.643", "0.286/0.219 0.639/0.643",
"0.286/0.219 0.639/0.643", "0.286/0.219 0.639/0.643",
"0.286/0.219 0.639/0.643", "0.286/0.219 0.639/0.643",
},{ },
"negative", "0/1 1/0", "0/1 1/0", "0/1 1/0", [PRESET_NEGATIVE] = {
},{ "0/1 1/0", "0/1 1/0", "0/1 1/0",
"strong_contrast", },
[PRESET_STRONG_CONTRAST] = {
"0.301/0.196 0.592/0.6 0.686/0.737", "0.301/0.196 0.592/0.6 0.686/0.737",
"0.301/0.196 0.592/0.6 0.686/0.737", "0.301/0.196 0.592/0.6 0.686/0.737",
"0.301/0.196 0.592/0.6 0.686/0.737", "0.301/0.196 0.592/0.6 0.686/0.737",
},{ },
"vintage", [PRESET_VINTAGE] = {
"0/0.11 0.42/0.51 1/0.95", "0/0.11 0.42/0.51 1/0.95",
"0.50/0.48", "0.50/0.48",
"0/0.22 0.49/0.44 1/0.8", "0/0.22 0.49/0.44 1/0.8",
@ -292,31 +319,18 @@ static av_cold int init(AVFilterContext *ctx, const char *args)
CurvesContext *curves = ctx->priv; CurvesContext *curves = ctx->priv;
struct keypoint *comp_points[NB_COMP] = {0}; struct keypoint *comp_points[NB_COMP] = {0};
if (curves->preset) { if (curves->preset != PRESET_NONE) {
char **pts = curves->comp_points_str; char **pts = curves->comp_points_str;
if (pts[0] || pts[1] || pts[2]) { if (pts[0] || pts[1] || pts[2]) {
av_log(ctx, AV_LOG_ERROR, "It is not possible to mix a preset " av_log(ctx, AV_LOG_ERROR, "It is not possible to mix a preset "
"with explicit points placements\n"); "with explicit points placements\n");
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
for (i = 0; i < FF_ARRAY_ELEMS(curves_presets); i++) { pts[0] = av_strdup(curves_presets[curves->preset].r);
if (!strcmp(curves->preset, curves_presets[i].name)) { pts[1] = av_strdup(curves_presets[curves->preset].g);
pts[0] = av_strdup(curves_presets[i].r); pts[2] = av_strdup(curves_presets[curves->preset].b);
pts[1] = av_strdup(curves_presets[i].g); if (!pts[0] || !pts[1] || !pts[2])
pts[2] = av_strdup(curves_presets[i].b); return AVERROR(ENOMEM);
if (!pts[0] || !pts[1] || !pts[2])
return AVERROR(ENOMEM);
break;
}
}
if (i == FF_ARRAY_ELEMS(curves_presets)) {
av_log(ctx, AV_LOG_ERROR, "Preset '%s' not found. Available presets:",
curves->preset);
for (i = 0; i < FF_ARRAY_ELEMS(curves_presets); i++)
av_log(ctx, AV_LOG_ERROR, " %s", curves_presets[i].name);
av_log(ctx, AV_LOG_ERROR, ".\n");
return AVERROR(EINVAL);
}
} }
for (i = 0; i < NB_COMP; i++) { for (i = 0; i < NB_COMP; i++) {