avfilter/f_loop: make sure duration of frame is never zero
[ffmpeg.git] / libavfilter / vf_il.c
1 /*
2  * Copyright (c) 2002 Michael Niedermayer <michaelni@gmx.at>
3  * Copyright (c) 2013 Paul B Mahol
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 /**
23  * @file
24  * (de)interleave fields filter
25  */
26
27 #include "libavutil/opt.h"
28 #include "libavutil/imgutils.h"
29 #include "libavutil/pixdesc.h"
30 #include "avfilter.h"
31 #include "internal.h"
32
33 enum FilterMode {
34     MODE_NONE,
35     MODE_INTERLEAVE,
36     MODE_DEINTERLEAVE
37 };
38
39 typedef struct IlContext {
40     const AVClass *class;
41     int luma_mode, chroma_mode, alpha_mode; ///<FilterMode
42     int luma_swap, chroma_swap, alpha_swap;
43     int nb_planes;
44     int linesize[4], chroma_height;
45     int has_alpha;
46 } IlContext;
47
48 #define OFFSET(x) offsetof(IlContext, x)
49 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
50
51 static const AVOption il_options[] = {
52     {"luma_mode",   "select luma mode", OFFSET(luma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "luma_mode"},
53     {"l",           "select luma mode", OFFSET(luma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "luma_mode"},
54     {"none",         NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE},         0, 0, FLAGS, "luma_mode"},
55     {"interleave",   NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE},   0, 0, FLAGS, "luma_mode"},
56     {"i",            NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE},   0, 0, FLAGS, "luma_mode"},
57     {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "luma_mode"},
58     {"d",            NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "luma_mode"},
59     {"chroma_mode", "select chroma mode", OFFSET(chroma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "chroma_mode"},
60     {"c",           "select chroma mode", OFFSET(chroma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "chroma_mode"},
61     {"none",         NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE},         0, 0, FLAGS, "chroma_mode"},
62     {"interleave",   NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE},   0, 0, FLAGS, "chroma_mode"},
63     {"i",            NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE},   0, 0, FLAGS, "chroma_mode"},
64     {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "chroma_mode"},
65     {"d",            NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "chroma_mode"},
66     {"alpha_mode", "select alpha mode", OFFSET(alpha_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "alpha_mode"},
67     {"a",          "select alpha mode", OFFSET(alpha_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "alpha_mode"},
68     {"none",         NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE},         0, 0, FLAGS, "alpha_mode"},
69     {"interleave",   NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE},   0, 0, FLAGS, "alpha_mode"},
70     {"i",            NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE},   0, 0, FLAGS, "alpha_mode"},
71     {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "alpha_mode"},
72     {"d",            NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "alpha_mode"},
73     {"luma_swap",   "swap luma fields",   OFFSET(luma_swap),   AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
74     {"ls",          "swap luma fields",   OFFSET(luma_swap),   AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
75     {"chroma_swap", "swap chroma fields", OFFSET(chroma_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
76     {"cs",          "swap chroma fields", OFFSET(chroma_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
77     {"alpha_swap",  "swap alpha fields",  OFFSET(alpha_swap),  AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
78     {"as",          "swap alpha fields",  OFFSET(alpha_swap),  AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
79     {NULL}
80 };
81
82 AVFILTER_DEFINE_CLASS(il);
83
84 static int query_formats(AVFilterContext *ctx)
85 {
86     AVFilterFormats *formats = NULL;
87     int fmt, ret;
88
89     for (fmt = 0; av_pix_fmt_desc_get(fmt); fmt++) {
90         const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt);
91         if (!(desc->flags & AV_PIX_FMT_FLAG_PAL) &&
92             !(desc->flags & AV_PIX_FMT_FLAG_HWACCEL) &&
93             (ret = ff_add_format(&formats, fmt)) < 0)
94             return ret;
95     }
96
97     return ff_set_common_formats(ctx, formats);
98 }
99
100 static int config_input(AVFilterLink *inlink)
101 {
102     IlContext *s = inlink->dst->priv;
103     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
104     int ret;
105
106     s->nb_planes = av_pix_fmt_count_planes(inlink->format);
107
108     s->has_alpha = !!(desc->flags & AV_PIX_FMT_FLAG_ALPHA);
109     if ((ret = av_image_fill_linesizes(s->linesize, inlink->format, inlink->w)) < 0)
110         return ret;
111
112     s->chroma_height = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
113
114     return 0;
115 }
116
117 static void interleave(uint8_t *dst, uint8_t *src, int w, int h,
118                        int dst_linesize, int src_linesize,
119                        enum FilterMode mode, int swap)
120 {
121     const int a = swap;
122     const int b = 1 - a;
123     const int m = h >> 1;
124     int y;
125
126     switch (mode) {
127     case MODE_DEINTERLEAVE:
128         for (y = 0; y < m; y++) {
129             memcpy(dst + dst_linesize *  y     , src + src_linesize * (y * 2 + a), w);
130             memcpy(dst + dst_linesize * (y + m), src + src_linesize * (y * 2 + b), w);
131         }
132         break;
133     case MODE_NONE:
134         for (y = 0; y < m; y++) {
135             memcpy(dst + dst_linesize *  y * 2     , src + src_linesize * (y * 2 + a), w);
136             memcpy(dst + dst_linesize * (y * 2 + 1), src + src_linesize * (y * 2 + b), w);
137         }
138         break;
139     case MODE_INTERLEAVE:
140         for (y = 0; y < m; y++) {
141             memcpy(dst + dst_linesize * (y * 2 + a), src + src_linesize *  y     , w);
142             memcpy(dst + dst_linesize * (y * 2 + b), src + src_linesize * (y + m), w);
143         }
144         break;
145     }
146 }
147
148 static int filter_frame(AVFilterLink *inlink, AVFrame *inpicref)
149 {
150     IlContext *s = inlink->dst->priv;
151     AVFilterLink *outlink = inlink->dst->outputs[0];
152     AVFrame *out;
153     int comp;
154
155     out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
156     if (!out) {
157         av_frame_free(&inpicref);
158         return AVERROR(ENOMEM);
159     }
160     av_frame_copy_props(out, inpicref);
161
162     interleave(out->data[0], inpicref->data[0],
163                s->linesize[0], inlink->h,
164                out->linesize[0], inpicref->linesize[0],
165                s->luma_mode, s->luma_swap);
166
167     for (comp = 1; comp < (s->nb_planes - s->has_alpha); comp++) {
168         interleave(out->data[comp], inpicref->data[comp],
169                    s->linesize[comp], s->chroma_height,
170                    out->linesize[comp], inpicref->linesize[comp],
171                    s->chroma_mode, s->chroma_swap);
172     }
173
174     if (s->has_alpha) {
175         comp = s->nb_planes - 1;
176         interleave(out->data[comp], inpicref->data[comp],
177                    s->linesize[comp], inlink->h,
178                    out->linesize[comp], inpicref->linesize[comp],
179                    s->alpha_mode, s->alpha_swap);
180     }
181
182     av_frame_free(&inpicref);
183     return ff_filter_frame(outlink, out);
184 }
185
186 static const AVFilterPad inputs[] = {
187     {
188         .name         = "default",
189         .type         = AVMEDIA_TYPE_VIDEO,
190         .filter_frame = filter_frame,
191         .config_props = config_input,
192     },
193     { NULL }
194 };
195
196 static const AVFilterPad outputs[] = {
197     {
198         .name = "default",
199         .type = AVMEDIA_TYPE_VIDEO,
200     },
201     { NULL }
202 };
203
204 AVFilter ff_vf_il = {
205     .name          = "il",
206     .description   = NULL_IF_CONFIG_SMALL("Deinterleave or interleave fields."),
207     .priv_size     = sizeof(IlContext),
208     .query_formats = query_formats,
209     .inputs        = inputs,
210     .outputs       = outputs,
211     .priv_class    = &il_class,
212     .flags         = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC,
213 };