FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavfilter/vf_derain.c
Date: 2024-04-19 07:31:02
Exec Total Coverage
Lines: 0 29 0.0%
Functions: 0 3 0.0%
Branches: 0 8 0.0%

Line Branch Exec Source
1 /*
2 * Copyright (c) 2019 Xuewei Meng
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 /**
22 * @file
23 * Filter implementing image derain filter using deep convolutional networks.
24 * http://openaccess.thecvf.com/content_ECCV_2018/html/Xia_Li_Recurrent_Squeeze-and-Excitation_Context_ECCV_2018_paper.html
25 */
26
27 #include "libavutil/opt.h"
28 #include "avfilter.h"
29 #include "dnn_filter_common.h"
30 #include "internal.h"
31 #include "video.h"
32
33 typedef struct DRContext {
34 const AVClass *class;
35 DnnContext dnnctx;
36 int filter_type;
37 } DRContext;
38
39 #define OFFSET(x) offsetof(DRContext, x)
40 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM
41 static const AVOption derain_options[] = {
42 { "filter_type", "filter type(derain/dehaze)", OFFSET(filter_type), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, FLAGS, .unit = "type" },
43 { "derain", "derain filter flag", 0, AV_OPT_TYPE_CONST, { .i64 = 0 }, 0, 0, FLAGS, .unit = "type" },
44 { "dehaze", "dehaze filter flag", 0, AV_OPT_TYPE_CONST, { .i64 = 1 }, 0, 0, FLAGS, .unit = "type" },
45 { "dnn_backend", "DNN backend", OFFSET(dnnctx.backend_type), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, FLAGS, .unit = "backend" },
46 #if (CONFIG_LIBTENSORFLOW == 1)
47 { "tensorflow", "tensorflow backend flag", 0, AV_OPT_TYPE_CONST, { .i64 = 1 }, 0, 0, FLAGS, .unit = "backend" },
48 #endif
49 { "model", "path to model file", OFFSET(dnnctx.model_filename), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },
50 { "input", "input name of the model", OFFSET(dnnctx.model_inputname), AV_OPT_TYPE_STRING, { .str = "x" }, 0, 0, FLAGS },
51 { "output", "output name of the model", OFFSET(dnnctx.model_outputnames_string), AV_OPT_TYPE_STRING, { .str = "y" }, 0, 0, FLAGS },
52 { NULL }
53 };
54
55 AVFILTER_DEFINE_CLASS(derain);
56
57 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
58 {
59 DNNAsyncStatusType async_state = 0;
60 AVFilterContext *ctx = inlink->dst;
61 AVFilterLink *outlink = ctx->outputs[0];
62 DRContext *dr_context = ctx->priv;
63 int dnn_result;
64 AVFrame *out;
65
66 out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
67 if (!out) {
68 av_log(ctx, AV_LOG_ERROR, "could not allocate memory for output frame\n");
69 av_frame_free(&in);
70 return AVERROR(ENOMEM);
71 }
72 av_frame_copy_props(out, in);
73
74 dnn_result = ff_dnn_execute_model(&dr_context->dnnctx, in, out);
75 if (dnn_result != 0){
76 av_log(ctx, AV_LOG_ERROR, "failed to execute model\n");
77 av_frame_free(&in);
78 return dnn_result;
79 }
80 do {
81 async_state = ff_dnn_get_result(&dr_context->dnnctx, &in, &out);
82 } while (async_state == DAST_NOT_READY);
83
84 if (async_state != DAST_SUCCESS)
85 return AVERROR(EINVAL);
86
87 av_frame_free(&in);
88
89 return ff_filter_frame(outlink, out);
90 }
91
92 static av_cold int init(AVFilterContext *ctx)
93 {
94 DRContext *dr_context = ctx->priv;
95 return ff_dnn_init(&dr_context->dnnctx, DFT_PROCESS_FRAME, ctx);
96 }
97
98 static av_cold void uninit(AVFilterContext *ctx)
99 {
100 DRContext *dr_context = ctx->priv;
101 ff_dnn_uninit(&dr_context->dnnctx);
102 }
103
104 static const AVFilterPad derain_inputs[] = {
105 {
106 .name = "default",
107 .type = AVMEDIA_TYPE_VIDEO,
108 .filter_frame = filter_frame,
109 },
110 };
111
112 const AVFilter ff_vf_derain = {
113 .name = "derain",
114 .description = NULL_IF_CONFIG_SMALL("Apply derain filter to the input."),
115 .priv_size = sizeof(DRContext),
116 .init = init,
117 .uninit = uninit,
118 FILTER_INPUTS(derain_inputs),
119 FILTER_OUTPUTS(ff_video_default_filterpad),
120 FILTER_SINGLE_PIXFMT(AV_PIX_FMT_RGB24),
121 .priv_class = &derain_class,
122 .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC,
123 };
124