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

#ifndef HAVE_LIBSAMPLERATE
        if (device_srate != 44100 && device_srate != 22050)
                return False;
#endif

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

        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;
        if ((format->wBitsPerSample != 8) && (format->wBitsPerSample != 16))
                return False;
#ifndef HAVE_LIBSAMPLERATE
        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;
        int err;
#else
        int offset;
#endif
        static BOOL warned = False;
        unsigned char *tmpdata = NULL;
        int samplewidth = format->wBitsPerSample / 8;
        int outsize = 0;
        int i;

        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

        /* Expand 8bit input-samples to 16bit */
#ifndef HAVE_LIBSAMPLERATE      /* libsamplerate needs 16bit samples */
        if (format->wBitsPerSample != resample_to_bitspersample)
#endif
        {
                /* source: 8 bit, dest: 16bit */
                if (format->wBitsPerSample == 8)
                {
                        tmpdata = xmalloc(size * 2);
                        for (i = 0; i < size; i++)
                        {
                                tmpdata[i * 2] = in[i];
                                tmpdata[(i * 2) + 1] = 0x00;
                        }
                        in = tmpdata;
                        samplewidth = 16 / 2;
                        size *= 2;
                }
        }

        if (resample_to_channels != format->nChannels)
        {
                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;
        }

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

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

        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;

        if ((err = src_process(src_converter, &resample_data)) != 0)
                error("src_process: %s", src_strerror(err));

        xfree(infloat);

        outsize = resample_data.output_frames_gen * resample_to_channels * samplewidth;
        *out = xmalloc(outsize);
        src_float_to_short_array(outfloat, (short *) *out, resample_data.output_frames_gen * resample_to_channels);
        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

        if (tmpdata != NULL)
                xfree(tmpdata);

        /* Shrink 16bit output-samples to 8bit */
#ifndef HAVE_LIBSAMPLERATE      /* libsamplerate produces 16bit samples */
        if (format->wBitsPerSample != resample_to_bitspersample)
#endif
        {
                /* source: 16 bit, dest: 8 bit */
                if (resample_to_bitspersample == 8)
                {
                        for (i = 0; i < outsize; i++)
                        {
                                *out[i] = *out[i * 2];
                        }
                        outsize /= 2;
                }
        }

#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	/*
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	#ifdef HAVE_LIBSAMPLERATE
26	#include <samplerate.h>
27
28	#define SRC_CONVERTER SRC_SINC_MEDIUM_QUALITY
29	#endif
30
31	#define MAX_VOLUME 65535
32
33	static uint16 softvol_left = MAX_VOLUME;
34	static uint16 softvol_right = MAX_VOLUME;
35	static uint32 resample_to_srate = 44100;
36	static uint16 resample_to_bitspersample = 16;
37	static uint16 resample_to_channels = 2;
38	#ifdef HAVE_LIBSAMPLERATE
39	static SRC_STATE *src_converter = NULL;
40	#endif
41
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	void
51	rdpsnd_dsp_softvol(unsigned char buffer, unsigned int size, WAVEFORMATEX format)
52	{
53	unsigned int factor_left, factor_right;
54	unsigned char *posin = buffer;
55	unsigned char *posout = buffer;
56
57	if ((softvol_left == MAX_VOLUME) && (softvol_right == MAX_VOLUME))
58	return;
59
60	factor_left = (softvol_left * 256) / MAX_VOLUME;
61	factor_right = (softvol_right * 256) / MAX_VOLUME;
62
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	sint8 val;
71
72	while (posout < buffer + size)
73	{
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	sint16 val;
88
89	while (posout < buffer + size)
90	{
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	DEBUG(("using softvol with factors left: %d, right: %d (%d/%d)\n", factor_left,
108	factor_right, format->wBitsPerSample, format->nChannels));
109	}
110
111	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	BOOL
129	rdpsnd_dsp_resample_set(uint32 device_srate, uint16 device_bitspersample, uint16 device_channels)
130	{
131	#ifdef HAVE_LIBSAMPLERATE
132	int err;
133	#endif
134
135	#ifndef HAVE_LIBSAMPLERATE
136	if (device_srate != 44100 && device_srate != 22050)
137	return False;
138	#endif
139
140	if (device_bitspersample != 16 && device_bitspersample != 8)
141	return False;
142
143	if (device_channels != 1 && device_channels != 2)
144	return False;
145
146	resample_to_srate = device_srate;
147	resample_to_bitspersample = device_bitspersample;
148	resample_to_channels = device_channels;
149
150	#ifdef HAVE_LIBSAMPLERATE
151	if (src_converter != NULL)
152	src_converter = src_delete(src_converter);
153
154	if ((src_converter = src_new(SRC_CONVERTER, device_channels, &err)) == NULL)
155	{
156	warning("src_new failed: %d!\n", err);
157	return False;
158	}
159	#endif
160
161	return True;
162	}
163
164	BOOL
165	rdpsnd_dsp_resample_supported(WAVEFORMATEX * format)
166	{
167	if (format->wFormatTag != WAVE_FORMAT_PCM)
168	return False;
169	if ((format->nChannels != 1) && (format->nChannels != 2))
170	return False;
171	if ((format->wBitsPerSample != 8) && (format->wBitsPerSample != 16))
172	return False;
173	#ifndef HAVE_LIBSAMPLERATE
174	if ((format->nSamplesPerSec != 44100) && (format->nSamplesPerSec != 22050))
175	return False;
176	#endif
177
178	return True;
179	}
180
181	uint32
182	rdpsnd_dsp_resample(unsigned char *out, unsigned char in, unsigned int size,
183	WAVEFORMATEX * format, BOOL stream_be)
184	{
185	#ifdef HAVE_LIBSAMPLERATE
186	SRC_DATA resample_data;
187	float infloat, outfloat;
188	int innum, outnum;
189	int err;
190	#else
191	int offset;
192	#endif
193	static BOOL warned = False;
194	unsigned char *tmpdata = NULL;
195	int samplewidth = format->wBitsPerSample / 8;
196	int outsize = 0;
197	int i;
198
199	if ((resample_to_bitspersample == format->wBitsPerSample) &&
200	(resample_to_channels == format->nChannels) &&
201	(resample_to_srate == format->nSamplesPerSec))
202	return 0;
203
204	#ifdef B_ENDIAN
205	if (!stream_be)
206	rdpsnd_dsp_swapbytes(in, size, format);
207	#endif
208
209	/* Expand 8bit input-samples to 16bit */
210	#ifndef HAVE_LIBSAMPLERATE /* libsamplerate needs 16bit samples */
211	if (format->wBitsPerSample != resample_to_bitspersample)
212	#endif
213	{
214	/* source: 8 bit, dest: 16bit */
215	if (format->wBitsPerSample == 8)
216	{
217	tmpdata = xmalloc(size * 2);
218	for (i = 0; i < size; i++)
219	{
220	tmpdata[i * 2] = in[i];
221	tmpdata[(i * 2) + 1] = 0x00;
222	}
223	in = tmpdata;
224	samplewidth = 16 / 2;
225	size *= 2;
226	}
227	}
228
229	if (resample_to_channels != format->nChannels)
230	{
231	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);
232	warned = True;
233	}
234
235	/* Do the resampling */
236	#ifdef HAVE_LIBSAMPLERATE
237	if (src_converter == NULL)
238	{
239	warning("no samplerate converter available!!\n");
240	return 0;
241	}
242
243	innum = size / samplewidth;
244	outnum = ((float)innum * ((float)resample_to_srate / (float)format->nSamplesPerSec)) + 1;
245
246	infloat = xmalloc(sizeof(float) * innum);
247	outfloat = xmalloc(sizeof(float) * outnum);
248
249	src_short_to_float_array((short *) in, infloat, innum);
250
251	bzero(&resample_data, sizeof(resample_data));
252	resample_data.data_in = infloat;
253	resample_data.data_out = outfloat;
254	resample_data.input_frames = innum / resample_to_channels;
255	resample_data.output_frames = outnum / resample_to_channels;
256	resample_data.src_ratio = (double) resample_to_srate / (double) format->nSamplesPerSec;
257	resample_data.end_of_input = 0;
258
259	if ((err = src_process(src_converter, &resample_data)) != 0)
260	error("src_process: %s", src_strerror(err));
261
262	xfree(infloat);
263
264	outsize = resample_data.output_frames_gen * resample_to_channels * samplewidth;
265	*out = xmalloc(outsize);
266	src_float_to_short_array(outfloat, (short ) out, resample_data.output_frames_gen * resample_to_channels);
267	xfree(outfloat);
268
269	#else
270	if (format->nSamplesPerSec != 22050)
271	{
272	if (!warned)
273	{
274	warning("unsupported source samplerate (%u), not resampling!\n",
275	format->nSamplesPerSec);
276	warned = True;
277	}
278	return 0;
279	}
280
281	outsize = size * 2;
282	*out = xmalloc(outsize);
283
284	/* Resample from 22050 to 44100 */
285	for (i = 0; i < (size / samplewidth); i++)
286	{
287	/* On a stereo-channel we must make sure that left and right
288	does not get mixed up, so we need to expand the sample-
289	data with channels in mind: 1234 -> 12123434
290	If we have a mono-channel, we can expand the data by simply
291	doubling the sample-data: 1234 -> 11223344 */
292	if (resample_to_channels == 2)
293	offset = ((i * 2) - (i & 1)) * samplewidth;
294	else
295	offset = (i * 2) * samplewidth;
296
297	memcpy(out + offset, in + (i samplewidth), samplewidth);
298	memcpy(out + (resample_to_channels samplewidth + offset),
299	in + (i * samplewidth), samplewidth);
300
301	}
302	#endif
303
304	if (tmpdata != NULL)
305	xfree(tmpdata);
306
307	/* Shrink 16bit output-samples to 8bit */
308	#ifndef HAVE_LIBSAMPLERATE /* libsamplerate produces 16bit samples */
309	if (format->wBitsPerSample != resample_to_bitspersample)
310	#endif
311	{
312	/* source: 16 bit, dest: 8 bit */
313	if (resample_to_bitspersample == 8)
314	{
315	for (i = 0; i < outsize; i++)
316	{
317	out[i] = out[i * 2];
318	}
319	outsize /= 2;
320	}
321	}
322
323	#ifdef B_ENDIAN
324	if (!stream_be)
325	rdpsnd_dsp_swapbytes(*out, outsize, format);
326	#endif
327	return outsize;
328	}
329
330	STREAM
331	rdpsnd_dsp_process(STREAM s, struct audio_driver * current_driver, WAVEFORMATEX * format)
332	{
333	static struct stream out;
334	BOOL stream_be = False;
335
336	/* softvol and byteswap do not change the amount of data they
337	return, so they can operate on the input-stream */
338	if (current_driver->wave_out_volume == rdpsnd_dsp_softvol_set)
339	rdpsnd_dsp_softvol(s->data, s->size, format);
340
341	#ifdef B_ENDIAN
342	if (current_driver->need_byteswap_on_be)
343	{
344	rdpsnd_dsp_swapbytes(s->data, s->size, format);
345	stream_be = True;
346	}
347	#endif
348
349	out.data = NULL;
350
351	if (current_driver->need_resampling)
352	out.size = rdpsnd_dsp_resample(&out.data, s->data, s->size, format, stream_be);
353
354	if (out.data == NULL)
355	{
356	out.data = xmalloc(s->size);
357	memcpy(out.data, s->data, s->size);
358	out.size = s->size;
359	}
360
361	out.p = out.data;
362	out.end = out.p + out.size;
363
364	return &out;
365	}