2 * This file is part of FFmpeg.
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.
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.
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
21 * Bauer stereo-to-binaural filter
26 #include "libavutil/channel_layout.h"
27 #include "libavutil/common.h"
28 #include "libavutil/opt.h"
35 typedef struct Bs2bContext
{
44 void (*filter
)(t_bs2bdp bs2bdp
, uint8_t *sample
, int n
);
47 #define OFFSET(x) offsetof(Bs2bContext, x)
48 #define A AV_OPT_FLAG_AUDIO_PARAM
50 static const AVOption options
[] = {
51 { "profile", "Apply a pre-defined crossfeed level",
52 OFFSET(profile
), AV_OPT_TYPE_INT
, { .i64
= BS2B_DEFAULT_CLEVEL
}, 0, INT_MAX
, A
, "profile" },
53 { "default", "default profile", 0, AV_OPT_TYPE_CONST
, { .i64
= BS2B_DEFAULT_CLEVEL
}, 0, 0, A
, "profile" },
54 { "cmoy", "Chu Moy circuit", 0, AV_OPT_TYPE_CONST
, { .i64
= BS2B_CMOY_CLEVEL
}, 0, 0, A
, "profile" },
55 { "jmeier", "Jan Meier circuit", 0, AV_OPT_TYPE_CONST
, { .i64
= BS2B_JMEIER_CLEVEL
}, 0, 0, A
, "profile" },
56 { "fcut", "Set cut frequency (in Hz)",
57 OFFSET(fcut
), AV_OPT_TYPE_INT
, { .i64
= 0 }, 0, BS2B_MAXFCUT
, A
},
58 { "feed", "Set feed level (in Hz)",
59 OFFSET(feed
), AV_OPT_TYPE_INT
, { .i64
= 0 }, 0, BS2B_MAXFEED
, A
},
63 static const AVClass bs2b_class
= {
64 .class_name
= "bs2b filter",
65 .item_name
= av_default_item_name
,
67 .version
= LIBAVUTIL_VERSION_INT
,
70 static av_cold
int init(AVFilterContext
*ctx
)
72 Bs2bContext
*bs2b
= ctx
->priv
;
74 if (!(bs2b
->bs2bp
= bs2b_open()))
75 return AVERROR(ENOMEM
);
77 bs2b_set_level(bs2b
->bs2bp
, bs2b
->profile
);
80 bs2b_set_level_fcut(bs2b
->bs2bp
, bs2b
->fcut
);
83 bs2b_set_level_feed(bs2b
->bs2bp
, bs2b
->feed
);
88 static av_cold
void uninit(AVFilterContext
*ctx
)
90 Bs2bContext
*bs2b
= ctx
->priv
;
93 bs2b_close(bs2b
->bs2bp
);
96 static int query_formats(AVFilterContext
*ctx
)
98 AVFilterFormats
*formats
= NULL
;
99 AVFilterChannelLayouts
*layouts
= NULL
;
101 static const enum AVSampleFormat sample_fmts
[] = {
110 if (ff_add_channel_layout(&layouts
, AV_CH_LAYOUT_STEREO
) != 0)
111 return AVERROR(ENOMEM
);
112 ff_set_common_channel_layouts(ctx
, layouts
);
114 formats
= ff_make_format_list(sample_fmts
);
116 return AVERROR(ENOMEM
);
117 ff_set_common_formats(ctx
, formats
);
119 formats
= ff_all_samplerates();
121 return AVERROR(ENOMEM
);
122 ff_set_common_samplerates(ctx
, formats
);
127 static int filter_frame(AVFilterLink
*inlink
, AVFrame
*frame
)
132 Bs2bContext
*bs2b
= inlink
->dst
->priv
;
133 AVFilterLink
*outlink
= inlink
->dst
->outputs
[0];
135 if (av_frame_is_writable(frame
)) {
138 out_frame
= ff_get_audio_buffer(inlink
, frame
->nb_samples
);
140 return AVERROR(ENOMEM
);
141 av_frame_copy(out_frame
, frame
);
142 ret
= av_frame_copy_props(out_frame
, frame
);
144 av_frame_free(&out_frame
);
145 av_frame_free(&frame
);
150 bs2b
->filter(bs2b
->bs2bp
, out_frame
->extended_data
[0], out_frame
->nb_samples
);
152 if (frame
!= out_frame
)
153 av_frame_free(&frame
);
155 return ff_filter_frame(outlink
, out_frame
);
158 static int config_output(AVFilterLink
*outlink
)
160 AVFilterContext
*ctx
= outlink
->src
;
161 Bs2bContext
*bs2b
= ctx
->priv
;
162 AVFilterLink
*inlink
= ctx
->inputs
[0];
164 int srate
= inlink
->sample_rate
;
166 switch (inlink
->format
) {
167 case AV_SAMPLE_FMT_U8
:
168 bs2b
->filter
= bs2b_cross_feed_u8
;
170 case AV_SAMPLE_FMT_S16
:
171 bs2b
->filter
= (void*)bs2b_cross_feed_s16
;
173 case AV_SAMPLE_FMT_S32
:
174 bs2b
->filter
= (void*)bs2b_cross_feed_s32
;
176 case AV_SAMPLE_FMT_FLT
:
177 bs2b
->filter
= (void*)bs2b_cross_feed_f
;
179 case AV_SAMPLE_FMT_DBL
:
180 bs2b
->filter
= (void*)bs2b_cross_feed_d
;
186 if ((srate
< BS2B_MINSRATE
) || (srate
> BS2B_MAXSRATE
))
187 return AVERROR(ENOSYS
);
189 bs2b_set_srate(bs2b
->bs2bp
, srate
);
194 static const AVFilterPad bs2b_inputs
[] = {
197 .type
= AVMEDIA_TYPE_AUDIO
,
198 .filter_frame
= filter_frame
,
203 static const AVFilterPad bs2b_outputs
[] = {
206 .type
= AVMEDIA_TYPE_AUDIO
,
207 .config_props
= config_output
,
212 AVFilter ff_af_bs2b
= {
214 .description
= NULL_IF_CONFIG_SMALL("Bauer stereo-to-binaural filter."),
215 .query_formats
= query_formats
,
216 .priv_size
= sizeof(Bs2bContext
),
217 .priv_class
= &bs2b_class
,
220 .inputs
= bs2b_inputs
,
221 .outputs
= bs2b_outputs
,