modules: use vlc_tick_from_samples()
[vlc.git] / modules / audio_filter / resampler / src.c
1 /*****************************************************************************
2  * src.c : Secret Rabbit Code (a.k.a. libsamplerate) resampler
3  *****************************************************************************
4  * Copyright (C) 2011-2012 RĂ©mi Denis-Courmont
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU Lesser General Public License as published by
8  * the Free Software Foundation; either version 2.1 of the License, or
9  * (at your option) any later version.
10  *
11  * This program 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
14  * GNU Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
19  *****************************************************************************/
20
21 /*****************************************************************************
22  * NOTA BENE: this module requires the linking against a library which is
23  * known to require licensing under the GNU General Public License version 2
24  * (or later). Therefore, the result of compiling this module will normally
25  * be subject to the terms of that later license.
26  *****************************************************************************/
27
28
29 #ifdef HAVE_CONFIG_H
30 # include "config.h"
31 #endif
32
33 #include <vlc_common.h>
34 #include <vlc_plugin.h>
35 #include <vlc_aout.h>
36 #include <vlc_filter.h>
37 #include <samplerate.h>
38 #include <math.h>
39
40 #define SRC_CONV_TYPE_TEXT N_("Sample rate converter type")
41 #define SRC_CONV_TYPE_LONGTEXT N_( \
42     "Different resampling algorithms are supported. " \
43     "The best one is slower, while the fast one exhibits low quality.")
44 static const int conv_type_values[] = {
45     SRC_SINC_BEST_QUALITY, SRC_SINC_MEDIUM_QUALITY, SRC_SINC_FASTEST,
46     SRC_ZERO_ORDER_HOLD, SRC_LINEAR,
47 };
48 static const char *const conv_type_texts[] = {
49     N_("Sinc function (best quality)"), N_("Sinc function (medium quality)"),
50     N_("Sinc function (fast)"), N_("Zero Order Hold (fastest)"), N_("Linear (fastest)"),
51 };
52
53 static int Open (vlc_object_t *);
54 static int OpenResampler (vlc_object_t *);
55 static void Close (vlc_object_t *);
56
57 vlc_module_begin ()
58     set_shortname (N_("SRC resampler"))
59     set_description (N_("Secret Rabbit Code (libsamplerate) resampler") )
60     set_category (CAT_AUDIO)
61     set_subcategory (SUBCAT_AUDIO_RESAMPLER)
62     add_integer ("src-converter-type", SRC_SINC_FASTEST,
63                  SRC_CONV_TYPE_TEXT, SRC_CONV_TYPE_LONGTEXT, true)
64         change_integer_list (conv_type_values, conv_type_texts)
65     set_capability ("audio converter", 50)
66     set_callbacks (Open, Close)
67
68     add_submodule ()
69     set_capability ("audio resampler", 50)
70     set_callbacks (OpenResampler, Close)
71 vlc_module_end ()
72
73 static block_t *Resample (filter_t *, block_t *);
74
75 static int Open (vlc_object_t *obj)
76 {
77     filter_t *filter = (filter_t *)obj;
78
79     /* Will change rate */
80     if (filter->fmt_in.audio.i_rate == filter->fmt_out.audio.i_rate)
81         return VLC_EGENERIC;
82     return OpenResampler (obj);
83 }
84
85 static int OpenResampler (vlc_object_t *obj)
86 {
87     filter_t *filter = (filter_t *)obj;
88
89     /* Only float->float */
90     if (filter->fmt_in.audio.i_format != VLC_CODEC_FL32
91      || filter->fmt_out.audio.i_format != VLC_CODEC_FL32
92     /* No channels remapping */
93      || filter->fmt_in.audio.i_channels != filter->fmt_out.audio.i_channels )
94         return VLC_EGENERIC;
95
96     int type = var_InheritInteger (obj, "src-converter-type");
97     int err;
98
99     SRC_STATE *s = src_new (type, filter->fmt_in.audio.i_channels, &err);
100     if (s == NULL)
101     {
102         msg_Err (obj, "cannot initialize resampler: %s", src_strerror (err));
103         return VLC_EGENERIC;
104     }
105
106     filter->p_sys = s;
107     filter->pf_audio_filter = Resample;
108     return VLC_SUCCESS;
109 }
110
111 static void Close (vlc_object_t *obj)
112 {
113     filter_t *filter = (filter_t *)obj;
114     SRC_STATE *s = (SRC_STATE *)filter->p_sys;
115
116     src_delete (s);
117 }
118
119 static block_t *Resample (filter_t *filter, block_t *in)
120 {
121     block_t *out = NULL;
122     const size_t framesize = filter->fmt_out.audio.i_bytes_per_frame;
123
124     SRC_STATE *s = (SRC_STATE *)filter->p_sys;
125     SRC_DATA src;
126
127     src.src_ratio = (double)filter->fmt_out.audio.i_rate
128                   / (double)filter->fmt_in.audio.i_rate;
129
130     int err = src_set_ratio (s, src.src_ratio);
131     if (err != 0)
132     {
133         msg_Err (filter, "cannot update resampling ratio: %s",
134                  src_strerror (err));
135         goto error;
136     }
137
138     src.input_frames = in->i_nb_samples;
139     src.output_frames = ceil (src.src_ratio * src.input_frames);
140     src.end_of_input = 0;
141
142     out = block_Alloc (src.output_frames * framesize);
143     if (unlikely(out == NULL))
144         goto error;
145
146     src.data_in = (float *)in->p_buffer;
147     src.data_out = (float *)out->p_buffer;
148
149     err = src_process (s, &src);
150     if (err != 0)
151     {
152         msg_Err (filter, "cannot resample: %s", src_strerror (err));
153         block_Release (out);
154         out = NULL;
155         goto error;
156     }
157
158     if (src.input_frames_used < src.input_frames)
159         msg_Err (filter, "lost %ld of %ld input frames",
160                  src.input_frames - src.input_frames_used, src.input_frames);
161
162     out->i_buffer = src.output_frames_gen * framesize;
163     out->i_nb_samples = src.output_frames_gen;
164     out->i_pts = in->i_pts;
165     out->i_length = vlc_tick_from_samples(src.output_frames_gen, filter->fmt_out.audio.i_rate);
166 error:
167     block_Release (in);
168     return out;
169 }