sgienc: remove unused SgiContext
[ffmpeg.git] / libavfilter / trim.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include <float.h>
20 #include <math.h>
21
22 #include "config.h"
23
24 #include "libavutil/avassert.h"
25 #include "libavutil/channel_layout.h"
26 #include "libavutil/common.h"
27 #include "libavutil/log.h"
28 #include "libavutil/mathematics.h"
29 #include "libavutil/opt.h"
30 #include "libavutil/samplefmt.h"
31
32 #include "audio.h"
33 #include "avfilter.h"
34 #include "internal.h"
35
36 typedef struct TrimContext {
37     const AVClass *class;
38
39     /*
40      * AVOptions
41      */
42     int64_t duration;
43     int64_t start_time, end_time;
44     int64_t start_frame, end_frame;
45
46     double duration_dbl;
47     double start_time_dbl, end_time_dbl;
48     /*
49      * in the link timebase for video,
50      * in 1/samplerate for audio
51      */
52     int64_t start_pts, end_pts;
53     int64_t start_sample, end_sample;
54
55     /*
56      * number of video frames that arrived on this filter so far
57      */
58     int64_t nb_frames;
59     /*
60      * number of audio samples that arrived on this filter so far
61      */
62     int64_t nb_samples;
63     /*
64      * timestamp of the first frame in the output, in the timebase units
65      */
66     int64_t first_pts;
67     /*
68      * duration in the timebase units
69      */
70     int64_t duration_tb;
71
72     int64_t next_pts;
73
74     int eof;
75 } TrimContext;
76
77 static int init(AVFilterContext *ctx)
78 {
79     TrimContext *s = ctx->priv;
80
81     s->first_pts = AV_NOPTS_VALUE;
82
83     return 0;
84 }
85
86 static int config_input(AVFilterLink *inlink)
87 {
88     AVFilterContext *ctx = inlink->dst;
89     TrimContext       *s = ctx->priv;
90     AVRational tb = (inlink->type == AVMEDIA_TYPE_VIDEO) ?
91                      inlink->time_base : (AVRational){ 1, inlink->sample_rate };
92
93     if (s->start_time_dbl != DBL_MAX)
94         s->start_time = s->start_time_dbl * 1e6;
95     if (s->end_time_dbl != DBL_MAX)
96         s->end_time = s->end_time_dbl * 1e6;
97     if (s->duration_dbl != DBL_MAX)
98         s->duration = s->duration_dbl * 1e6;
99
100     if (s->start_time != INT64_MAX) {
101         int64_t start_pts = av_rescale_q(s->start_time, AV_TIME_BASE_Q, tb);
102         if (s->start_pts == AV_NOPTS_VALUE || start_pts < s->start_pts)
103             s->start_pts = start_pts;
104     }
105     if (s->end_time != INT64_MAX) {
106         int64_t end_pts = av_rescale_q(s->end_time, AV_TIME_BASE_Q, tb);
107         if (s->end_pts == AV_NOPTS_VALUE || end_pts > s->end_pts)
108             s->end_pts = end_pts;
109     }
110     if (s->duration)
111         s->duration_tb = av_rescale_q(s->duration, AV_TIME_BASE_Q, tb);
112
113     return 0;
114 }
115
116 static int config_output(AVFilterLink *outlink)
117 {
118     outlink->flags |= FF_LINK_FLAG_REQUEST_LOOP;
119     return 0;
120 }
121
122 #define OFFSET(x) offsetof(TrimContext, x)
123 #define COMMON_OPTS                                                                                                                                                         \
124     { "starti",      "Timestamp of the first frame that "                                                                                                        \
125         "should be passed",                                              OFFSET(start_time),  AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX },    INT64_MIN, INT64_MAX, FLAGS }, \
126     { "endi",        "Timestamp of the first frame that "                                                                                                        \
127         "should be dropped again",                                       OFFSET(end_time),    AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX },    INT64_MIN, INT64_MAX, FLAGS }, \
128     { "start_pts",   "Timestamp of the first frame that should be "                                                                                                         \
129        " passed",                                                        OFFSET(start_pts),   AV_OPT_TYPE_INT64,  { .i64 = AV_NOPTS_VALUE }, INT64_MIN, INT64_MAX, FLAGS }, \
130     { "end_pts",     "Timestamp of the first frame that should be "                                                                                                         \
131         "dropped again",                                                 OFFSET(end_pts),     AV_OPT_TYPE_INT64,  { .i64 = AV_NOPTS_VALUE }, INT64_MIN, INT64_MAX, FLAGS }, \
132     { "durationi",   "Maximum duration of the output",                   OFFSET(duration),    AV_OPT_TYPE_DURATION, { .i64 = 0 },                    0, INT64_MAX, FLAGS },
133
134 #define COMPAT_OPTS \
135     { "start",       "Timestamp in seconds of the first frame that "                                                                                                        \
136         "should be passed",                                              OFFSET(start_time_dbl),AV_OPT_TYPE_DOUBLE, { .dbl = DBL_MAX },       -DBL_MAX, DBL_MAX,     FLAGS }, \
137     { "end",         "Timestamp in seconds of the first frame that "                                                                                                        \
138         "should be dropped again",                                       OFFSET(end_time_dbl),  AV_OPT_TYPE_DOUBLE, { .dbl = DBL_MAX },       -DBL_MAX, DBL_MAX,     FLAGS }, \
139     { "duration",    "Maximum duration of the output in seconds",        OFFSET(duration_dbl),  AV_OPT_TYPE_DOUBLE, { .dbl = 0 },                      0,   DBL_MAX, FLAGS },
140
141
142 #if CONFIG_TRIM_FILTER
143 static int trim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
144 {
145     AVFilterContext *ctx = inlink->dst;
146     TrimContext       *s = ctx->priv;
147     int drop;
148
149     /* drop everything if EOF has already been returned */
150     if (s->eof) {
151         av_frame_free(&frame);
152         return 0;
153     }
154
155     if (s->start_frame >= 0 || s->start_pts != AV_NOPTS_VALUE) {
156         drop = 1;
157         if (s->start_frame >= 0 && s->nb_frames >= s->start_frame)
158             drop = 0;
159         if (s->start_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
160             frame->pts >= s->start_pts)
161             drop = 0;
162         if (drop)
163             goto drop;
164     }
165
166     if (s->first_pts == AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE)
167         s->first_pts = frame->pts;
168
169     if (s->end_frame != INT64_MAX || s->end_pts != AV_NOPTS_VALUE || s->duration_tb) {
170         drop = 1;
171
172         if (s->end_frame != INT64_MAX && s->nb_frames < s->end_frame)
173             drop = 0;
174         if (s->end_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
175             frame->pts < s->end_pts)
176             drop = 0;
177         if (s->duration_tb && frame->pts != AV_NOPTS_VALUE &&
178             frame->pts - s->first_pts < s->duration_tb)
179             drop = 0;
180
181         if (drop) {
182             s->eof = inlink->closed = 1;
183             goto drop;
184         }
185     }
186
187     s->nb_frames++;
188
189     return ff_filter_frame(ctx->outputs[0], frame);
190
191 drop:
192     s->nb_frames++;
193     av_frame_free(&frame);
194     return 0;
195 }
196
197 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
198 static const AVOption trim_options[] = {
199     COMMON_OPTS
200     { "start_frame", "Number of the first frame that should be passed "
201         "to the output",                                                 OFFSET(start_frame), AV_OPT_TYPE_INT64,  { .i64 = -1 },       -1, INT64_MAX, FLAGS },
202     { "end_frame",   "Number of the first frame that should be dropped "
203         "again",                                                         OFFSET(end_frame),   AV_OPT_TYPE_INT64,  { .i64 = INT64_MAX }, 0, INT64_MAX, FLAGS },
204     COMPAT_OPTS
205     { NULL },
206 };
207 #undef FLAGS
208
209 AVFILTER_DEFINE_CLASS(trim);
210
211 static const AVFilterPad trim_inputs[] = {
212     {
213         .name         = "default",
214         .type         = AVMEDIA_TYPE_VIDEO,
215         .filter_frame = trim_filter_frame,
216         .config_props = config_input,
217     },
218     { NULL }
219 };
220
221 static const AVFilterPad trim_outputs[] = {
222     {
223         .name          = "default",
224         .type          = AVMEDIA_TYPE_VIDEO,
225         .config_props  = config_output,
226     },
227     { NULL }
228 };
229
230 AVFilter avfilter_vf_trim = {
231     .name        = "trim",
232     .description = NULL_IF_CONFIG_SMALL("Pick one continuous section from the input, drop the rest."),
233
234     .init        = init,
235
236     .priv_size   = sizeof(TrimContext),
237     .priv_class  = &trim_class,
238
239     .inputs      = trim_inputs,
240     .outputs     = trim_outputs,
241 };
242 #endif // CONFIG_TRIM_FILTER
243
244 #if CONFIG_ATRIM_FILTER
245 static int atrim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
246 {
247     AVFilterContext *ctx = inlink->dst;
248     TrimContext       *s = ctx->priv;
249     int64_t start_sample, end_sample = frame->nb_samples;
250     int64_t pts;
251     int drop;
252
253     /* drop everything if EOF has already been returned */
254     if (s->eof) {
255         av_frame_free(&frame);
256         return 0;
257     }
258
259     if (frame->pts != AV_NOPTS_VALUE)
260         pts = av_rescale_q(frame->pts, inlink->time_base,
261                            (AVRational){ 1, inlink->sample_rate });
262     else
263         pts = s->next_pts;
264     s->next_pts = pts + frame->nb_samples;
265
266     /* check if at least a part of the frame is after the start time */
267     if (s->start_sample < 0 && s->start_pts == AV_NOPTS_VALUE) {
268         start_sample = 0;
269     } else {
270         drop = 1;
271         start_sample = frame->nb_samples;
272
273         if (s->start_sample >= 0 &&
274             s->nb_samples + frame->nb_samples > s->start_sample) {
275             drop         = 0;
276             start_sample = FFMIN(start_sample, s->start_sample - s->nb_samples);
277         }
278
279         if (s->start_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
280             pts + frame->nb_samples > s->start_pts) {
281             drop = 0;
282             start_sample = FFMIN(start_sample, s->start_pts - pts);
283         }
284
285         if (drop)
286             goto drop;
287     }
288
289     if (s->first_pts == AV_NOPTS_VALUE)
290         s->first_pts = pts + start_sample;
291
292     /* check if at least a part of the frame is before the end time */
293     if (s->end_sample == INT64_MAX && s->end_pts == AV_NOPTS_VALUE && !s->duration_tb) {
294         end_sample = frame->nb_samples;
295     } else {
296         drop       = 1;
297         end_sample = 0;
298
299         if (s->end_sample != INT64_MAX &&
300             s->nb_samples < s->end_sample) {
301             drop       = 0;
302             end_sample = FFMAX(end_sample, s->end_sample - s->nb_samples);
303         }
304
305         if (s->end_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
306             pts < s->end_pts) {
307             drop       = 0;
308             end_sample = FFMAX(end_sample, s->end_pts - pts);
309         }
310
311         if (s->duration_tb && pts - s->first_pts < s->duration_tb) {
312             drop       = 0;
313             end_sample = FFMAX(end_sample, s->first_pts + s->duration_tb - pts);
314         }
315
316         if (drop) {
317             s->eof = inlink->closed = 1;
318             goto drop;
319         }
320     }
321
322     s->nb_samples += frame->nb_samples;
323     start_sample   = FFMAX(0, start_sample);
324     end_sample     = FFMIN(frame->nb_samples, end_sample);
325     av_assert0(start_sample < end_sample);
326
327     if (start_sample) {
328         AVFrame *out = ff_get_audio_buffer(ctx->outputs[0], end_sample - start_sample);
329         if (!out) {
330             av_frame_free(&frame);
331             return AVERROR(ENOMEM);
332         }
333
334         av_frame_copy_props(out, frame);
335         av_samples_copy(out->extended_data, frame->extended_data, 0, start_sample,
336                         out->nb_samples, inlink->channels,
337                         frame->format);
338         if (out->pts != AV_NOPTS_VALUE)
339             out->pts += av_rescale_q(start_sample, (AVRational){ 1, out->sample_rate },
340                                      inlink->time_base);
341
342         av_frame_free(&frame);
343         frame = out;
344     } else
345         frame->nb_samples = end_sample;
346
347     return ff_filter_frame(ctx->outputs[0], frame);
348
349 drop:
350     s->nb_samples += frame->nb_samples;
351     av_frame_free(&frame);
352     return 0;
353 }
354
355 #define FLAGS AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
356 static const AVOption atrim_options[] = {
357     COMMON_OPTS
358     { "start_sample", "Number of the first audio sample that should be "
359         "passed to the output",                                          OFFSET(start_sample), AV_OPT_TYPE_INT64,  { .i64 = -1 },       -1, INT64_MAX, FLAGS },
360     { "end_sample",   "Number of the first audio sample that should be "
361         "dropped again",                                                 OFFSET(end_sample),   AV_OPT_TYPE_INT64,  { .i64 = INT64_MAX }, 0, INT64_MAX, FLAGS },
362     COMPAT_OPTS
363     { NULL },
364 };
365 #undef FLAGS
366
367 AVFILTER_DEFINE_CLASS(atrim);
368
369 static const AVFilterPad atrim_inputs[] = {
370     {
371         .name         = "default",
372         .type         = AVMEDIA_TYPE_AUDIO,
373         .filter_frame = atrim_filter_frame,
374         .config_props = config_input,
375     },
376     { NULL }
377 };
378
379 static const AVFilterPad atrim_outputs[] = {
380     {
381         .name          = "default",
382         .type          = AVMEDIA_TYPE_AUDIO,
383         .config_props  = config_output,
384     },
385     { NULL }
386 };
387
388 AVFilter avfilter_af_atrim = {
389     .name        = "atrim",
390     .description = NULL_IF_CONFIG_SMALL("Pick one continuous section from the input, drop the rest."),
391
392     .init        = init,
393
394     .priv_size   = sizeof(TrimContext),
395     .priv_class  = &atrim_class,
396
397     .inputs      = atrim_inputs,
398     .outputs     = atrim_outputs,
399 };
400 #endif // CONFIG_ATRIM_FILTER