Commit d122c8b1 authored by Paul B Mahol's avatar Paul B Mahol

avfilter/vf_edgedetect: add canny mode

Signed-off-by: 's avatarPaul B Mahol <onemda@gmail.com>
parent aba39cc1
......@@ -8276,6 +8276,9 @@ Draw white/gray wires on black background.
@item colormix
Mix the colors to create a paint/cartoon effect.
@item canny
Apply Canny edge detector on all selected planes.
@end table
Default value is @var{wires}.
......
......@@ -35,6 +35,7 @@
enum FilterMode {
MODE_WIRES,
MODE_COLORMIX,
MODE_CANNY,
NB_MODE
};
......@@ -61,6 +62,7 @@ static const AVOption edgedetect_options[] = {
{ "mode", "set mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64=MODE_WIRES}, 0, NB_MODE-1, FLAGS, "mode" },
{ "wires", "white/gray wires on black", 0, AV_OPT_TYPE_CONST, {.i64=MODE_WIRES}, INT_MIN, INT_MAX, FLAGS, "mode" },
{ "colormix", "mix colors", 0, AV_OPT_TYPE_CONST, {.i64=MODE_COLORMIX}, INT_MIN, INT_MAX, FLAGS, "mode" },
{ "canny", "detect edges on planes", 0, AV_OPT_TYPE_CONST, {.i64=MODE_CANNY}, INT_MIN, INT_MAX, FLAGS, "mode" },
{ NULL }
};
......@@ -79,6 +81,7 @@ static int query_formats(AVFilterContext *ctx)
{
const EdgeDetectContext *edgedetect = ctx->priv;
static const enum AVPixelFormat wires_pix_fmts[] = {AV_PIX_FMT_GRAY8, AV_PIX_FMT_NONE};
static const enum AVPixelFormat canny_pix_fmts[] = {AV_PIX_FMT_YUV444P, AV_PIX_FMT_GBRP, AV_PIX_FMT_GRAY8, AV_PIX_FMT_NONE};
static const enum AVPixelFormat colormix_pix_fmts[] = {AV_PIX_FMT_GBRP, AV_PIX_FMT_GRAY8, AV_PIX_FMT_NONE};
AVFilterFormats *fmts_list;
const enum AVPixelFormat *pix_fmts = NULL;
......@@ -87,6 +90,8 @@ static int query_formats(AVFilterContext *ctx)
pix_fmts = wires_pix_fmts;
} else if (edgedetect->mode == MODE_COLORMIX) {
pix_fmts = colormix_pix_fmts;
} else if (edgedetect->mode == MODE_CANNY) {
pix_fmts = canny_pix_fmts;
} else {
av_assert0(0);
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment