trunk/rdesktop/rdpsnd_dsp.c

/*
   rdesktop: A Remote Desktop Protocol client.
   Sound DSP routines
   Copyright (C) Michael Gernoth 2006

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   This program 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 General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "rdesktop.h"
#include "rdpsnd.h"
#include "rdpsnd_dsp.h"

#ifdef HAVE_LIBSAMPLERATE
#include <samplerate.h>

#define SRC_CONVERTER SRC_SINC_MEDIUM_QUALITY
#endif

#define MAX_VOLUME 65535

static uint16 softvol_left = MAX_VOLUME;
static uint16 softvol_right = MAX_VOLUME;
static uint32 resample_to_srate = 44100;
static uint16 resample_to_bitspersample = 16;
static uint16 resample_to_channels = 2;
#ifdef HAVE_LIBSAMPLERATE
static SRC_STATE *src_converter = NULL;
#endif

void
rdpsnd_dsp_softvol_set(uint16 left, uint16 right)
{
        softvol_left = left;
        softvol_right = right;
        DEBUG(("rdpsnd_dsp_softvol_set: left: %u, right: %u\n", left, right));
}

void
rdpsnd_dsp_softvol(unsigned char *buffer, unsigned int size, WAVEFORMATEX * format)
{
        unsigned int factor_left, factor_right;
        unsigned char *posin = buffer;
        unsigned char *posout = buffer;

        if ((softvol_left == MAX_VOLUME) && (softvol_right == MAX_VOLUME))
                return;

        factor_left = (softvol_left * 256) / MAX_VOLUME;
        factor_right = (softvol_right * 256) / MAX_VOLUME;

        if (format->nChannels == 1)
        {
                factor_left = factor_right = (factor_left + factor_right) / 2;
        }

        if (format->wBitsPerSample == 8)
        {
                sint8 val;

                while (posout < buffer + size)
                {
                        /* Left */
                        val = *posin++;
                        val = (val * factor_left) >> 8;
                        *posout++ = val;

                        /* Right */
                        val = *posin++;
                        val = (val * factor_right) >> 8;
                        *posout++ = val;
                }
        }
        else
        {
                sint16 val;

                while (posout < buffer + size)
                {
                        /* Left */
                        val = *posin++;
                        val |= *posin++ << 8;
                        val = (val * factor_left) >> 8;
                        *posout++ = val & 0xff;
                        *posout++ = val >> 8;

                        /* Right */
                        val = *posin++;
                        val |= *posin++ << 8;
                        val = (val * factor_right) >> 8;
                        *posout++ = val & 0xff;
                        *posout++ = val >> 8;
                }
        }

        DEBUG(("using softvol with factors left: %d, right: %d (%d/%d)\n", factor_left,
               factor_right, format->wBitsPerSample, format->nChannels));
}

void
rdpsnd_dsp_swapbytes(unsigned char *buffer, unsigned int size, WAVEFORMATEX * format)
{
        int i;
        uint8 swap;

        if (format->wBitsPerSample == 8)
                return;

        for (i = 0; i < size; i += 2)
        {
                swap = *(buffer + i);
                *(buffer + i) = *(buffer + i + 1);
                *(buffer + i + 1) = swap;
        }
}

BOOL
rdpsnd_dsp_resample_set(uint32 device_srate, uint16 device_bitspersample, uint16 device_channels)
{
#ifdef HAVE_LIBSAMPLERATE
        int err;
#endif

#ifdef HAVE_LIBSAMPLERATE
        if (device_bitspersample != 16)
                return False;
#else
        if (device_srate != 44100 && device_srate != 22050)
                return False;

        if (device_bitspersample != 16 && device_bitspersample != 8)
                return False;
#endif

        if (device_channels != 1 && device_channels != 2)
                return False;

        resample_to_srate = device_srate;
        resample_to_bitspersample = device_bitspersample;
        resample_to_channels = device_channels;

#ifdef HAVE_LIBSAMPLERATE
        if (src_converter != NULL)
                src_converter = src_delete(src_converter);

        if ((src_converter = src_new(SRC_CONVERTER, device_channels, &err)) == NULL)
        {
                warning("src_new failed: %d!\n", err);
                return False;
        }
#endif

        return True;
}

BOOL
rdpsnd_dsp_resample_supported(WAVEFORMATEX * format)
{
        if (format->wFormatTag != WAVE_FORMAT_PCM)
                return False;
        if ((format->nChannels != 1) && (format->nChannels != 2))
                return False;
#ifdef HAVE_LIBSAMPLERATE
        if (format->wBitsPerSample != 16)
                return False;
#else
        if ((format->wBitsPerSample != 8) && (format->wBitsPerSample != 16))
                return False;
        if ((format->nSamplesPerSec != 44100) && (format->nSamplesPerSec != 22050))
                return False;
#endif

        return True;
}

uint32
rdpsnd_dsp_resample(unsigned char **out, unsigned char *in, unsigned int size,
                    WAVEFORMATEX * format, BOOL stream_be)
{
#ifdef HAVE_LIBSAMPLERATE
        SRC_DATA resample_data;
        float *infloat, *outfloat;
        int innum, outnum;
#else
        int offset;
        int i;
#endif
        static BOOL warned = False;
        int samplewidth = format->wBitsPerSample / 8;
        int outsize = 0;

        if ((resample_to_bitspersample == format->wBitsPerSample) &&
            (resample_to_channels == format->nChannels) &&
            (resample_to_srate == format->nSamplesPerSec))
                return 0;

#ifdef B_ENDIAN
        if (!stream_be)
                rdpsnd_dsp_swapbytes(in, size, format)
#endif
                        if ((resample_to_channels != format->nChannels) ||
                            (resample_to_bitspersample != format->wBitsPerSample))
                {
                        warning("unsupported resample-settings (%u -> %u/%u -> %u/%u -> %u), not resampling!\n", format->nSamplesPerSec, resample_to_srate, format->wBitsPerSample, resample_to_bitspersample, format->nChannels, resample_to_channels);
                        warned = True;
                }

#ifdef HAVE_LIBSAMPLERATE
        if (src_converter == NULL)
        {
                warning("no samplerate converter available!!\n");
                return 0;
        }

        innum = size / samplewidth;
        outnum = innum * (resample_to_srate / format->nSamplesPerSec);

        infloat = xmalloc(sizeof(float) * innum);
        outfloat = xmalloc(sizeof(float) * outnum);

        src_short_to_float_array((short *) in, infloat, innum);

        bzero(&resample_data, sizeof(resample_data));
        resample_data.data_in = infloat;
        resample_data.data_out = outfloat;
        resample_data.input_frames = innum / resample_to_channels;
        resample_data.output_frames = outnum / resample_to_channels;
        resample_data.src_ratio = (double) resample_to_srate / (double) format->nSamplesPerSec;
        resample_data.end_of_input = 0;

        src_process(src_converter, &resample_data);
        xfree(infloat);

        outsize = outnum * samplewidth;
        *out = xmalloc(outsize);
        src_float_to_short_array(outfloat, (short *) *out, outnum);
        xfree(outfloat);

#else
        if (format->nSamplesPerSec != 22050)
        {
                if (!warned)
                {
                        warning("unsupported source samplerate (%u), not resampling!\n",
                                format->nSamplesPerSec);
                        warned = True;
                }
                return 0;
        }

        outsize = size * 2;
        *out = xmalloc(outsize);

        /* Resample from 22050 to 44100 */
        for (i = 0; i < (size / samplewidth); i++)
        {
                /* On a stereo-channel we must make sure that left and right
                   does not get mixed up, so we need to expand the sample-
                   data with channels in mind: 1234 -> 12123434
                   If we have a mono-channel, we can expand the data by simply
                   doubling the sample-data: 1234 -> 11223344 */
                if (resample_to_channels == 2)
                        offset = ((i * 2) - (i & 1)) * samplewidth;
                else
                        offset = (i * 2) * samplewidth;

                memcpy(*out + offset, in + (i * samplewidth), samplewidth);
                memcpy(*out + (resample_to_channels * samplewidth + offset),
                       in + (i * samplewidth), samplewidth);

        }
#endif

#ifdef B_ENDIAN
        if (!stream_be)
                rdpsnd_dsp_swapbytes(*out, outsize, format)
#endif
                        return outsize;
}

STREAM
rdpsnd_dsp_process(STREAM s, struct audio_driver * current_driver, WAVEFORMATEX * format)
{
        static struct stream out;
        BOOL stream_be = False;

        /* softvol and byteswap do not change the amount of data they
           return, so they can operate on the input-stream */
        if (current_driver->wave_out_volume == rdpsnd_dsp_softvol_set)
                rdpsnd_dsp_softvol(s->data, s->size, format);

#ifdef B_ENDIAN
        if (current_driver->need_byteswap_on_be)
        {
                rdpsnd_dsp_swapbytes(s->data, s->size, format);
                stream_be = True;
        }
#endif

        out.data = NULL;

        if (current_driver->need_resampling)
                out.size = rdpsnd_dsp_resample(&out.data, s->data, s->size, format, stream_be);

        if (out.data == NULL)
        {
                out.data = xmalloc(s->size);
                memcpy(out.data, s->data, s->size);
                out.size = s->size;
        }

        out.p = out.data;
        out.end = out.p + out.size;

        return &out;
}
1	stargo	1258	/*
2			rdesktop: A Remote Desktop Protocol client.
3			Sound DSP routines
4			Copyright (C) Michael Gernoth 2006
5
6			This program is free software; you can redistribute it and/or modify
7			it under the terms of the GNU General Public License as published by
8			the Free Software Foundation; either version 2 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 General Public License for more details.
15
16			You should have received a copy of the GNU General Public License
17			along with this program; if not, write to the Free Software
18			Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19			*/
20
21			#include "rdesktop.h"
22			#include "rdpsnd.h"
23			#include "rdpsnd_dsp.h"
24
25	stargo	1278	#ifdef HAVE_LIBSAMPLERATE
26			#include <samplerate.h>
27
28			#define SRC_CONVERTER SRC_SINC_MEDIUM_QUALITY
29			#endif
30
31	stargo	1258	#define MAX_VOLUME 65535
32
33			static uint16 softvol_left = MAX_VOLUME;
34			static uint16 softvol_right = MAX_VOLUME;
35	stargo	1276	static uint32 resample_to_srate = 44100;
36			static uint16 resample_to_bitspersample = 16;
37			static uint16 resample_to_channels = 2;
38	stargo	1278	#ifdef HAVE_LIBSAMPLERATE
39			static SRC_STATE *src_converter = NULL;
40			#endif
41	stargo	1258
42			void
43			rdpsnd_dsp_softvol_set(uint16 left, uint16 right)
44			{
45			softvol_left = left;
46			softvol_right = right;
47			DEBUG(("rdpsnd_dsp_softvol_set: left: %u, right: %u\n", left, right));
48			}
49
50	stargo	1260	void
51	stargo	1259	rdpsnd_dsp_softvol(unsigned char buffer, unsigned int size, WAVEFORMATEX format)
52	stargo	1258	{
53			unsigned int factor_left, factor_right;
54	stargo	1259	unsigned char *posin = buffer;
55			unsigned char *posout = buffer;
56	stargo	1258
57	stargo	1259	if ((softvol_left == MAX_VOLUME) && (softvol_right == MAX_VOLUME))
58			return;
59
60	stargo	1270	factor_left = (softvol_left * 256) / MAX_VOLUME;
61			factor_right = (softvol_right * 256) / MAX_VOLUME;
62	stargo	1258
63			if (format->nChannels == 1)
64			{
65			factor_left = factor_right = (factor_left + factor_right) / 2;
66			}
67
68			if (format->wBitsPerSample == 8)
69			{
70	stargo	1262	sint8 val;
71	stargo	1258
72	stargo	1259	while (posout < buffer + size)
73	stargo	1258	{
74			/* Left */
75			val = *posin++;
76			val = (val * factor_left) >> 8;
77			*posout++ = val;
78
79			/* Right */
80			val = *posin++;
81			val = (val * factor_right) >> 8;
82			*posout++ = val;
83			}
84			}
85			else
86			{
87	stargo	1262	sint16 val;
88	stargo	1258
89	stargo	1259	while (posout < buffer + size)
90	stargo	1258	{
91			/* Left */
92			val = *posin++;
93			val \|= *posin++ << 8;
94			val = (val * factor_left) >> 8;
95			*posout++ = val & 0xff;
96			*posout++ = val >> 8;
97
98			/* Right */
99			val = *posin++;
100			val \|= *posin++ << 8;
101			val = (val * factor_right) >> 8;
102			*posout++ = val & 0xff;
103			*posout++ = val >> 8;
104			}
105			}
106
107	stargo	1263	DEBUG(("using softvol with factors left: %d, right: %d (%d/%d)\n", factor_left,
108			factor_right, format->wBitsPerSample, format->nChannels));
109	stargo	1258	}
110
111	stargo	1260	void
112			rdpsnd_dsp_swapbytes(unsigned char buffer, unsigned int size, WAVEFORMATEX format)
113			{
114			int i;
115			uint8 swap;
116
117			if (format->wBitsPerSample == 8)
118			return;
119
120			for (i = 0; i < size; i += 2)
121			{
122			swap = *(buffer + i);
123			(buffer + i) = (buffer + i + 1);
124			*(buffer + i + 1) = swap;
125			}
126			}
127
128	stargo	1276	BOOL
129			rdpsnd_dsp_resample_set(uint32 device_srate, uint16 device_bitspersample, uint16 device_channels)
130			{
131	stargo	1278	#ifdef HAVE_LIBSAMPLERATE
132			int err;
133			#endif
134
135			#ifdef HAVE_LIBSAMPLERATE
136			if (device_bitspersample != 16)
137			return False;
138			#else
139	stargo	1276	if (device_srate != 44100 && device_srate != 22050)
140			return False;
141	stargo	1263
142	stargo	1276	if (device_bitspersample != 16 && device_bitspersample != 8)
143			return False;
144	stargo	1278	#endif
145	stargo	1276
146			if (device_channels != 1 && device_channels != 2)
147			return False;
148
149			resample_to_srate = device_srate;
150			resample_to_bitspersample = device_bitspersample;
151			resample_to_channels = device_channels;
152
153	stargo	1278	#ifdef HAVE_LIBSAMPLERATE
154			if (src_converter != NULL)
155			src_converter = src_delete(src_converter);
156
157			if ((src_converter = src_new(SRC_CONVERTER, device_channels, &err)) == NULL)
158			{
159			warning("src_new failed: %d!\n", err);
160			return False;
161			}
162			#endif
163
164	stargo	1276	return True;
165			}
166
167			BOOL
168			rdpsnd_dsp_resample_supported(WAVEFORMATEX * format)
169			{
170			if (format->wFormatTag != WAVE_FORMAT_PCM)
171			return False;
172			if ((format->nChannels != 1) && (format->nChannels != 2))
173			return False;
174	stargo	1278	#ifdef HAVE_LIBSAMPLERATE
175			if (format->wBitsPerSample != 16)
176			return False;
177			#else
178	stargo	1276	if ((format->wBitsPerSample != 8) && (format->wBitsPerSample != 16))
179			return False;
180			if ((format->nSamplesPerSec != 44100) && (format->nSamplesPerSec != 22050))
181			return False;
182	stargo	1278	#endif
183	stargo	1276
184			return True;
185			}
186
187			uint32
188			rdpsnd_dsp_resample(unsigned char *out, unsigned char in, unsigned int size,
189	stargo	1278	WAVEFORMATEX * format, BOOL stream_be)
190	stargo	1276	{
191	stargo	1278	#ifdef HAVE_LIBSAMPLERATE
192			SRC_DATA resample_data;
193			float infloat, outfloat;
194			int innum, outnum;
195			#else
196			int offset;
197			int i;
198			#endif
199	stargo	1276	static BOOL warned = False;
200			int samplewidth = format->wBitsPerSample / 8;
201	stargo	1278	int outsize = 0;
202	stargo	1276
203	stargo	1277	if ((resample_to_bitspersample == format->wBitsPerSample) &&
204			(resample_to_channels == format->nChannels) &&
205			(resample_to_srate == format->nSamplesPerSec))
206			return 0;
207
208	stargo	1278	#ifdef B_ENDIAN
209			if (!stream_be)
210			rdpsnd_dsp_swapbytes(in, size, format)
211			#endif
212			if ((resample_to_channels != format->nChannels) \|\|
213			(resample_to_bitspersample != format->wBitsPerSample))
214			{
215			warning("unsupported resample-settings (%u -> %u/%u -> %u/%u -> %u), not resampling!\n", format->nSamplesPerSec, resample_to_srate, format->wBitsPerSample, resample_to_bitspersample, format->nChannels, resample_to_channels);
216			warned = True;
217			}
218
219			#ifdef HAVE_LIBSAMPLERATE
220			if (src_converter == NULL)
221	stargo	1276	{
222	stargo	1278	warning("no samplerate converter available!!\n");
223			return 0;
224			}
225
226			innum = size / samplewidth;
227			outnum = innum * (resample_to_srate / format->nSamplesPerSec);
228
229			infloat = xmalloc(sizeof(float) * innum);
230			outfloat = xmalloc(sizeof(float) * outnum);
231
232			src_short_to_float_array((short *) in, infloat, innum);
233
234			bzero(&resample_data, sizeof(resample_data));
235			resample_data.data_in = infloat;
236			resample_data.data_out = outfloat;
237			resample_data.input_frames = innum / resample_to_channels;
238			resample_data.output_frames = outnum / resample_to_channels;
239			resample_data.src_ratio = (double) resample_to_srate / (double) format->nSamplesPerSec;
240			resample_data.end_of_input = 0;
241
242			src_process(src_converter, &resample_data);
243			xfree(infloat);
244
245			outsize = outnum * samplewidth;
246			*out = xmalloc(outsize);
247			src_float_to_short_array(outfloat, (short ) out, outnum);
248			xfree(outfloat);
249
250			#else
251			if (format->nSamplesPerSec != 22050)
252			{
253	stargo	1276	if (!warned)
254			{
255	stargo	1278	warning("unsupported source samplerate (%u), not resampling!\n",
256			format->nSamplesPerSec);
257	stargo	1276	warned = True;
258			}
259			return 0;
260			}
261
262	stargo	1277	outsize = size * 2;
263			*out = xmalloc(outsize);
264
265			/* Resample from 22050 to 44100 */
266			for (i = 0; i < (size / samplewidth); i++)
267	stargo	1276	{
268	stargo	1277	/* On a stereo-channel we must make sure that left and right
269			does not get mixed up, so we need to expand the sample-
270			data with channels in mind: 1234 -> 12123434
271			If we have a mono-channel, we can expand the data by simply
272			doubling the sample-data: 1234 -> 11223344 */
273			if (resample_to_channels == 2)
274			offset = ((i * 2) - (i & 1)) * samplewidth;
275			else
276			offset = (i * 2) * samplewidth;
277	stargo	1276
278	stargo	1277	memcpy(out + offset, in + (i samplewidth), samplewidth);
279			memcpy(out + (resample_to_channels samplewidth + offset),
280			in + (i * samplewidth), samplewidth);
281	stargo	1276
282			}
283	stargo	1278	#endif
284	stargo	1276
285	stargo	1278	#ifdef B_ENDIAN
286			if (!stream_be)
287			rdpsnd_dsp_swapbytes(*out, outsize, format)
288			#endif
289			return outsize;
290	stargo	1276	}
291
292	stargo	1263	STREAM
293	stargo	1276	rdpsnd_dsp_process(STREAM s, struct audio_driver * current_driver, WAVEFORMATEX * format)
294	stargo	1258	{
295	stargo	1263	static struct stream out;
296	stargo	1278	BOOL stream_be = False;
297	stargo	1258
298	stargo	1263	/* softvol and byteswap do not change the amount of data they
299			return, so they can operate on the input-stream */
300	stargo	1258	if (current_driver->wave_out_volume == rdpsnd_dsp_softvol_set)
301	stargo	1263	rdpsnd_dsp_softvol(s->data, s->size, format);
302	stargo	1258
303	stargo	1260	#ifdef B_ENDIAN
304			if (current_driver->need_byteswap_on_be)
305	stargo	1278	{
306	stargo	1263	rdpsnd_dsp_swapbytes(s->data, s->size, format);
307	stargo	1278	stream_be = True;
308			}
309	stargo	1260	#endif
310
311	stargo	1276	out.data = NULL;
312	stargo	1263
313	stargo	1279	if (current_driver->need_resampling)
314	stargo	1278	out.size = rdpsnd_dsp_resample(&out.data, s->data, s->size, format, stream_be);
315	stargo	1263
316	stargo	1276	if (out.data == NULL)
317			{
318			out.data = xmalloc(s->size);
319			memcpy(out.data, s->data, s->size);
320			out.size = s->size;
321			}
322
323	stargo	1263	out.p = out.data;
324			out.end = out.p + out.size;
325
326			return &out;
327	stargo	1258	}