50f3cbd0028abcab9ccefeb083e450c75acd8063
[ffmpeg.git] / libavfilter / af_volume.c
1 /*
2  * Copyright (c) 2011 Stefano Sabatini
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  * audio volume filter
24  * based on ffmpeg.c code
25  */
26
27 #include "libavutil/audioconvert.h"
28 #include "libavutil/eval.h"
29 #include "audio.h"
30 #include "avfilter.h"
31 #include "formats.h"
32
33 typedef struct {
34     double volume;
35     int    volume_i;
36 } VolumeContext;
37
38 static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
39 {
40     VolumeContext *vol = ctx->priv;
41     char *tail;
42     int ret = 0;
43
44     vol->volume = 1.0;
45
46     if (args) {
47         /* parse the number as a decimal number */
48         double d = strtod(args, &tail);
49
50         if (*tail) {
51             if (!strcmp(tail, "dB")) {
52                 /* consider the argument an adjustement in decibels */
53                 d = pow(10, d/20);
54             } else {
55                 /* parse the argument as an expression */
56                 ret = av_expr_parse_and_eval(&d, args, NULL, NULL,
57                                              NULL, NULL, NULL, NULL,
58                                              NULL, 0, ctx);
59             }
60         }
61
62         if (ret < 0) {
63             av_log(ctx, AV_LOG_ERROR,
64                    "Invalid volume argument '%s'\n", args);
65             return AVERROR(EINVAL);
66         }
67
68         if (d < 0 || d > 65536) { /* 65536 = INT_MIN / (128 * 256) */
69             av_log(ctx, AV_LOG_ERROR,
70                    "Negative or too big volume value %f\n", d);
71             return AVERROR(EINVAL);
72         }
73
74         vol->volume = d;
75     }
76
77     vol->volume_i = (int)(vol->volume * 256 + 0.5);
78     av_log(ctx, AV_LOG_INFO, "volume=%f\n", vol->volume);
79     return 0;
80 }
81
82 static int query_formats(AVFilterContext *ctx)
83 {
84     AVFilterFormats *formats = NULL;
85     AVFilterChannelLayouts *layouts;
86     enum AVSampleFormat sample_fmts[] = {
87         AV_SAMPLE_FMT_U8,
88         AV_SAMPLE_FMT_S16,
89         AV_SAMPLE_FMT_S32,
90         AV_SAMPLE_FMT_FLT,
91         AV_SAMPLE_FMT_DBL,
92         AV_SAMPLE_FMT_NONE
93     };
94     int packing_fmts[] = { AVFILTER_PACKED, -1 };
95
96     layouts = ff_all_channel_layouts();
97     if (!layouts)
98         return AVERROR(ENOMEM);
99     ff_set_common_channel_layouts(ctx, layouts);
100
101     formats = avfilter_make_format_list(sample_fmts);
102     if (!formats)
103         return AVERROR(ENOMEM);
104     avfilter_set_common_sample_formats(ctx, formats);
105
106     formats = avfilter_make_format_list(packing_fmts);
107     if (!formats)
108         return AVERROR(ENOMEM);
109     avfilter_set_common_packing_formats(ctx, formats);
110
111     formats = ff_all_samplerates();
112     if (!formats)
113         return AVERROR(ENOMEM);
114     ff_set_common_samplerates(ctx, formats);
115
116     return 0;
117 }
118
119 static void filter_samples(AVFilterLink *inlink, AVFilterBufferRef *insamples)
120 {
121     VolumeContext *vol = inlink->dst->priv;
122     AVFilterLink *outlink = inlink->dst->outputs[0];
123     const int nb_samples = insamples->audio->nb_samples *
124         av_get_channel_layout_nb_channels(insamples->audio->channel_layout);
125     const double volume   = vol->volume;
126     const int    volume_i = vol->volume_i;
127     int i;
128
129     if (volume_i != 256) {
130         switch (insamples->format) {
131         case AV_SAMPLE_FMT_U8:
132         {
133             uint8_t *p = (void *)insamples->data[0];
134             for (i = 0; i < nb_samples; i++) {
135                 int v = (((*p - 128) * volume_i + 128) >> 8) + 128;
136                 *p++ = av_clip_uint8(v);
137             }
138             break;
139         }
140         case AV_SAMPLE_FMT_S16:
141         {
142             int16_t *p = (void *)insamples->data[0];
143             for (i = 0; i < nb_samples; i++) {
144                 int v = ((int64_t)*p * volume_i + 128) >> 8;
145                 *p++ = av_clip_int16(v);
146             }
147             break;
148         }
149         case AV_SAMPLE_FMT_S32:
150         {
151             int32_t *p = (void *)insamples->data[0];
152             for (i = 0; i < nb_samples; i++) {
153                 int64_t v = (((int64_t)*p * volume_i + 128) >> 8);
154                 *p++ = av_clipl_int32(v);
155             }
156             break;
157         }
158         case AV_SAMPLE_FMT_FLT:
159         {
160             float *p = (void *)insamples->data[0];
161             float scale = (float)volume;
162             for (i = 0; i < nb_samples; i++) {
163                 *p++ *= scale;
164             }
165             break;
166         }
167         case AV_SAMPLE_FMT_DBL:
168         {
169             double *p = (void *)insamples->data[0];
170             for (i = 0; i < nb_samples; i++) {
171                 *p *= volume;
172                 p++;
173             }
174             break;
175         }
176         }
177     }
178     ff_filter_samples(outlink, insamples);
179 }
180
181 AVFilter avfilter_af_volume = {
182     .name           = "volume",
183     .description    = NULL_IF_CONFIG_SMALL("Change input volume."),
184     .query_formats  = query_formats,
185     .priv_size      = sizeof(VolumeContext),
186     .init           = init,
187
188     .inputs  = (const AVFilterPad[])  {{ .name     = "default",
189                                    .type           = AVMEDIA_TYPE_AUDIO,
190                                    .filter_samples = filter_samples,
191                                    .min_perms      = AV_PERM_READ|AV_PERM_WRITE},
192                                  { .name = NULL}},
193
194     .outputs = (const AVFilterPad[])  {{ .name     = "default",
195                                    .type           = AVMEDIA_TYPE_AUDIO, },
196                                  { .name = NULL}},
197 };