[deb_ffmpeg.git] / ffmpeg / libavfilter / trim.c

/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <float.h>
#include <math.h>
#include <stdint.h>

#include "config.h"

#include "libavutil/avassert.h"
#include "libavutil/channel_layout.h"
#include "libavutil/common.h"
#include "libavutil/log.h"
#include "libavutil/mathematics.h"
#include "libavutil/opt.h"
#include "libavutil/samplefmt.h"

#include "audio.h"
#include "avfilter.h"
#include "internal.h"

typedef struct TrimContext {
    const AVClass *class;

    /*
     * AVOptions
     */
    int64_t duration;
    int64_t start_time, end_time;
    int64_t start_frame, end_frame;

    double duration_dbl;
    double start_time_dbl, end_time_dbl;
    /*
     * in the link timebase for video,
     * in 1/samplerate for audio
     */
    int64_t start_pts, end_pts;
    int64_t start_sample, end_sample;

    /*
     * number of video frames that arrived on this filter so far
     */
    int64_t nb_frames;
    /*
     * number of audio samples that arrived on this filter so far
     */
    int64_t nb_samples;
    /*
     * timestamp of the first frame in the output, in the timebase units
     */
    int64_t first_pts;
    /*
     * duration in the timebase units
     */
    int64_t duration_tb;

    int64_t next_pts;

    int eof;
} TrimContext;

static av_cold int init(AVFilterContext *ctx)
{
    TrimContext *s = ctx->priv;

    s->first_pts = AV_NOPTS_VALUE;

    return 0;
}

static int config_input(AVFilterLink *inlink)
{
    AVFilterContext *ctx = inlink->dst;
    TrimContext       *s = ctx->priv;
    AVRational tb = (inlink->type == AVMEDIA_TYPE_VIDEO) ?
                     inlink->time_base : (AVRational){ 1, inlink->sample_rate };

    if (s->start_time_dbl != DBL_MAX)
        s->start_time = s->start_time_dbl * 1e6;
    if (s->end_time_dbl != DBL_MAX)
        s->end_time = s->end_time_dbl * 1e6;
    if (s->duration_dbl != 0)
        s->duration = s->duration_dbl * 1e6;

    if (s->start_time != INT64_MAX) {
        int64_t start_pts = av_rescale_q(s->start_time, AV_TIME_BASE_Q, tb);
        if (s->start_pts == AV_NOPTS_VALUE || start_pts < s->start_pts)
            s->start_pts = start_pts;
    }
    if (s->end_time != INT64_MAX) {
        int64_t end_pts = av_rescale_q(s->end_time, AV_TIME_BASE_Q, tb);
        if (s->end_pts == AV_NOPTS_VALUE || end_pts > s->end_pts)
            s->end_pts = end_pts;
    }
    if (s->duration)
        s->duration_tb = av_rescale_q(s->duration, AV_TIME_BASE_Q, tb);

    return 0;
}

static int config_output(AVFilterLink *outlink)
{
    outlink->flags |= FF_LINK_FLAG_REQUEST_LOOP;
    return 0;
}

#define OFFSET(x) offsetof(TrimContext, x)
#define COMMON_OPTS                                                                                                                                                         \
    { "starti",      "Timestamp of the first frame that "                                                                                                        \
        "should be passed",                                              OFFSET(start_time),  AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX },    INT64_MIN, INT64_MAX, FLAGS }, \
    { "endi",        "Timestamp of the first frame that "                                                                                                        \
        "should be dropped again",                                       OFFSET(end_time),    AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX },    INT64_MIN, INT64_MAX, FLAGS }, \
    { "start_pts",   "Timestamp of the first frame that should be "                                                                                                         \
       " passed",                                                        OFFSET(start_pts),   AV_OPT_TYPE_INT64,  { .i64 = AV_NOPTS_VALUE }, INT64_MIN, INT64_MAX, FLAGS }, \
    { "end_pts",     "Timestamp of the first frame that should be "                                                                                                         \
        "dropped again",                                                 OFFSET(end_pts),     AV_OPT_TYPE_INT64,  { .i64 = AV_NOPTS_VALUE }, INT64_MIN, INT64_MAX, FLAGS }, \
    { "durationi",   "Maximum duration of the output",                   OFFSET(duration),    AV_OPT_TYPE_DURATION, { .i64 = 0 },                    0, INT64_MAX, FLAGS },

#define COMPAT_OPTS \
    { "start",       "Timestamp in seconds of the first frame that "                                                                                                        \
        "should be passed",                                              OFFSET(start_time_dbl),AV_OPT_TYPE_DOUBLE, { .dbl = DBL_MAX },       -DBL_MAX, DBL_MAX,     FLAGS }, \
    { "end",         "Timestamp in seconds of the first frame that "                                                                                                        \
        "should be dropped again",                                       OFFSET(end_time_dbl),  AV_OPT_TYPE_DOUBLE, { .dbl = DBL_MAX },       -DBL_MAX, DBL_MAX,     FLAGS }, \
    { "duration",    "Maximum duration of the output in seconds",        OFFSET(duration_dbl),  AV_OPT_TYPE_DOUBLE, { .dbl = 0 },                      0,   DBL_MAX, FLAGS },


#if CONFIG_TRIM_FILTER
static int trim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
    AVFilterContext *ctx = inlink->dst;
    TrimContext       *s = ctx->priv;
    int drop;

    /* drop everything if EOF has already been returned */
    if (s->eof) {
        av_frame_free(&frame);
        return 0;
    }

    if (s->start_frame >= 0 || s->start_pts != AV_NOPTS_VALUE) {
        drop = 1;
        if (s->start_frame >= 0 && s->nb_frames >= s->start_frame)
            drop = 0;
        if (s->start_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
            frame->pts >= s->start_pts)
            drop = 0;
        if (drop)
            goto drop;
    }

    if (s->first_pts == AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE)
        s->first_pts = frame->pts;

    if (s->end_frame != INT64_MAX || s->end_pts != AV_NOPTS_VALUE || s->duration_tb) {
        drop = 1;

        if (s->end_frame != INT64_MAX && s->nb_frames < s->end_frame)
            drop = 0;
        if (s->end_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
            frame->pts < s->end_pts)
            drop = 0;
        if (s->duration_tb && frame->pts != AV_NOPTS_VALUE &&
            frame->pts - s->first_pts < s->duration_tb)
            drop = 0;

        if (drop) {
            s->eof = inlink->closed = 1;
            goto drop;
        }
    }

    s->nb_frames++;

    return ff_filter_frame(ctx->outputs[0], frame);

drop:
    s->nb_frames++;
    av_frame_free(&frame);
    return 0;
}

#define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
static const AVOption trim_options[] = {
    COMMON_OPTS
    { "start_frame", "Number of the first frame that should be passed "
        "to the output",                                                 OFFSET(start_frame), AV_OPT_TYPE_INT64,  { .i64 = -1 },       -1, INT64_MAX, FLAGS },
    { "end_frame",   "Number of the first frame that should be dropped "
        "again",                                                         OFFSET(end_frame),   AV_OPT_TYPE_INT64,  { .i64 = INT64_MAX }, 0, INT64_MAX, FLAGS },
    COMPAT_OPTS
    { NULL }
};
#undef FLAGS

AVFILTER_DEFINE_CLASS(trim);

static const AVFilterPad trim_inputs[] = {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
        .filter_frame = trim_filter_frame,
        .config_props = config_input,
    },
    { NULL }
};

static const AVFilterPad trim_outputs[] = {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
        .config_props = config_output,
    },
    { NULL }
};

AVFilter ff_vf_trim = {
    .name        = "trim",
    .description = NULL_IF_CONFIG_SMALL("Pick one continuous section from the input, drop the rest."),
    .init        = init,
    .priv_size   = sizeof(TrimContext),
    .priv_class  = &trim_class,
    .inputs      = trim_inputs,
    .outputs     = trim_outputs,
};
#endif // CONFIG_TRIM_FILTER

#if CONFIG_ATRIM_FILTER
static int atrim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
    AVFilterContext *ctx = inlink->dst;
    TrimContext       *s = ctx->priv;
    int64_t start_sample, end_sample;
    int64_t pts;
    int drop;

    /* drop everything if EOF has already been returned */
    if (s->eof) {
        av_frame_free(&frame);
        return 0;
    }

    if (frame->pts != AV_NOPTS_VALUE)
        pts = av_rescale_q(frame->pts, inlink->time_base,
                           (AVRational){ 1, inlink->sample_rate });
    else
        pts = s->next_pts;
    s->next_pts = pts + frame->nb_samples;

    /* check if at least a part of the frame is after the start time */
    if (s->start_sample < 0 && s->start_pts == AV_NOPTS_VALUE) {
        start_sample = 0;
    } else {
        drop = 1;
        start_sample = frame->nb_samples;

        if (s->start_sample >= 0 &&
            s->nb_samples + frame->nb_samples > s->start_sample) {
            drop         = 0;
            start_sample = FFMIN(start_sample, s->start_sample - s->nb_samples);
        }

        if (s->start_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
            pts + frame->nb_samples > s->start_pts) {
            drop = 0;
            start_sample = FFMIN(start_sample, s->start_pts - pts);
        }

        if (drop)
            goto drop;
    }

    if (s->first_pts == AV_NOPTS_VALUE)
        s->first_pts = pts + start_sample;

    /* check if at least a part of the frame is before the end time */
    if (s->end_sample == INT64_MAX && s->end_pts == AV_NOPTS_VALUE && !s->duration_tb) {
        end_sample = frame->nb_samples;
    } else {
        drop       = 1;
        end_sample = 0;

        if (s->end_sample != INT64_MAX &&
            s->nb_samples < s->end_sample) {
            drop       = 0;
            end_sample = FFMAX(end_sample, s->end_sample - s->nb_samples);
        }

        if (s->end_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
            pts < s->end_pts) {
            drop       = 0;
            end_sample = FFMAX(end_sample, s->end_pts - pts);
        }

        if (s->duration_tb && pts - s->first_pts < s->duration_tb) {
            drop       = 0;
            end_sample = FFMAX(end_sample, s->first_pts + s->duration_tb - pts);
        }

        if (drop) {
            s->eof = inlink->closed = 1;
            goto drop;
        }
    }

    s->nb_samples += frame->nb_samples;
    start_sample   = FFMAX(0, start_sample);
    end_sample     = FFMIN(frame->nb_samples, end_sample);
    av_assert0(start_sample < end_sample || (start_sample == end_sample && !frame->nb_samples));

    if (start_sample) {
        AVFrame *out = ff_get_audio_buffer(ctx->outputs[0], end_sample - start_sample);
        if (!out) {
            av_frame_free(&frame);
            return AVERROR(ENOMEM);
        }

        av_frame_copy_props(out, frame);
        av_samples_copy(out->extended_data, frame->extended_data, 0, start_sample,
                        out->nb_samples, inlink->channels,
                        frame->format);
        if (out->pts != AV_NOPTS_VALUE)
            out->pts += av_rescale_q(start_sample, (AVRational){ 1, out->sample_rate },
                                     inlink->time_base);

        av_frame_free(&frame);
        frame = out;
    } else
        frame->nb_samples = end_sample;

    return ff_filter_frame(ctx->outputs[0], frame);

drop:
    s->nb_samples += frame->nb_samples;
    av_frame_free(&frame);
    return 0;
}

#define FLAGS AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
static const AVOption atrim_options[] = {
    COMMON_OPTS
    { "start_sample", "Number of the first audio sample that should be "
        "passed to the output",                                          OFFSET(start_sample), AV_OPT_TYPE_INT64,  { .i64 = -1 },       -1, INT64_MAX, FLAGS },
    { "end_sample",   "Number of the first audio sample that should be "
        "dropped again",                                                 OFFSET(end_sample),   AV_OPT_TYPE_INT64,  { .i64 = INT64_MAX }, 0, INT64_MAX, FLAGS },
    COMPAT_OPTS
    { NULL }
};
#undef FLAGS

AVFILTER_DEFINE_CLASS(atrim);

static const AVFilterPad atrim_inputs[] = {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_AUDIO,
        .filter_frame = atrim_filter_frame,
        .config_props = config_input,
    },
    { NULL }
};

static const AVFilterPad atrim_outputs[] = {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_AUDIO,
        .config_props = config_output,
    },
    { NULL }
};

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