lavfi: add asetnsamples audio filter
[ffmpeg.git] / libavfilter / asrc_anullsrc.c
1 /*
2  * Copyright 2010 S.N. Hemanth Meenakshisundaram <smeenaks ucsd edu>
3  * Copyright 2010 Stefano Sabatini <stefano.sabatini-lala poste it>
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  * null audio source
25  */
26
27 #include "internal.h"
28 #include "libavutil/audioconvert.h"
29 #include "libavutil/opt.h"
30
31 #include "audio.h"
32 #include "avfilter.h"
33 #include "internal.h"
34
35 typedef struct {
36     const AVClass *class;
37     char   *channel_layout_str;
38     uint64_t channel_layout;
39     char   *sample_rate_str;
40     int     sample_rate;
41     int nb_samples;             ///< number of samples per requested frame
42     int64_t pts;
43 } ANullContext;
44
45 #define OFFSET(x) offsetof(ANullContext, x)
46
47 static const AVOption anullsrc_options[]= {
48     { "channel_layout", "set channel_layout", OFFSET(channel_layout_str), AV_OPT_TYPE_STRING, {.str = "stereo"}, 0, 0 },
49     { "cl",             "set channel_layout", OFFSET(channel_layout_str), AV_OPT_TYPE_STRING, {.str = "stereo"}, 0, 0 },
50     { "sample_rate",    "set sample rate",    OFFSET(sample_rate_str)   , AV_OPT_TYPE_STRING, {.str = "44100"}, 0, 0 },
51     { "r",              "set sample rate",    OFFSET(sample_rate_str)   , AV_OPT_TYPE_STRING, {.str = "44100"}, 0, 0 },
52     { "nb_samples",     "set the number of samples per requested frame", OFFSET(nb_samples), AV_OPT_TYPE_INT, {.dbl = 1024}, 0, INT_MAX },
53     { "n",              "set the number of samples per requested frame", OFFSET(nb_samples), AV_OPT_TYPE_INT, {.dbl = 1024}, 0, INT_MAX },
54     { NULL },
55 };
56
57 static const AVClass anullsrc_class = {
58     "ANullSrcContext",
59     av_default_item_name,
60     anullsrc_options
61 };
62
63 static int init(AVFilterContext *ctx, const char *args, void *opaque)
64 {
65     ANullContext *null = ctx->priv;
66     int ret;
67
68     null->class = &anullsrc_class;
69     av_opt_set_defaults(null);
70
71     if ((ret = (av_set_options_string(null, args, "=", ":"))) < 0) {
72         av_log(ctx, AV_LOG_ERROR, "Error parsing options string: '%s'\n", args);
73         return ret;
74     }
75
76     if ((ret = ff_parse_sample_rate(&null->sample_rate,
77                                      null->sample_rate_str, ctx)) < 0)
78         return ret;
79
80     if ((ret = ff_parse_channel_layout(&null->channel_layout,
81                                         null->channel_layout_str, ctx)) < 0)
82         return ret;
83
84     return 0;
85 }
86
87 static int config_props(AVFilterLink *outlink)
88 {
89     ANullContext *null = outlink->src->priv;
90     char buf[128];
91     int chans_nb;
92
93     outlink->sample_rate = null->sample_rate;
94     outlink->channel_layout = null->channel_layout;
95
96     chans_nb = av_get_channel_layout_nb_channels(null->channel_layout);
97     av_get_channel_layout_string(buf, sizeof(buf), chans_nb, null->channel_layout);
98     av_log(outlink->src, AV_LOG_INFO,
99            "sample_rate:%d channel_layout:'%s' nb_samples:%d\n",
100            null->sample_rate, buf, null->nb_samples);
101
102     return 0;
103 }
104
105 static int request_frame(AVFilterLink *outlink)
106 {
107     ANullContext *null = outlink->src->priv;
108     AVFilterBufferRef *samplesref;
109
110     samplesref =
111         ff_get_audio_buffer(outlink, AV_PERM_WRITE, null->nb_samples);
112     samplesref->pts = null->pts;
113     samplesref->pos = -1;
114     samplesref->audio->channel_layout = null->channel_layout;
115     samplesref->audio->sample_rate = outlink->sample_rate;
116
117     ff_filter_samples(outlink, avfilter_ref_buffer(samplesref, ~0));
118     avfilter_unref_buffer(samplesref);
119
120     null->pts += null->nb_samples;
121     return 0;
122 }
123
124 AVFilter avfilter_asrc_anullsrc = {
125     .name        = "anullsrc",
126     .description = NULL_IF_CONFIG_SMALL("Null audio source, return empty audio frames."),
127
128     .init        = init,
129     .priv_size   = sizeof(ANullContext),
130
131     .inputs      = (const AVFilterPad[]) {{ .name = NULL}},
132
133     .outputs     = (const AVFilterPad[]) {{ .name = "default",
134                                       .type = AVMEDIA_TYPE_AUDIO,
135                                       .config_props = config_props,
136                                       .request_frame = request_frame, },
137                                     { .name = NULL}},
138 };