source: mainline/uspace/lib/pcm/src/format.c@ 34e1206

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 34e1206 was 39f1c86, checked in by Ondřej Hlavatý <aearsis@…>, 8 years ago

byteorder: move also int variants
  • Property mode set to 100644
File size: 9.6 KB
Line 
1/*
2 * Copyright (c) 2012 Jan Vesely
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * - The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** @addtogroup audio
30 * @brief HelenOS sound server
31 * @{
32 */
33/** @file
34 */
35
36#include <assert.h>
37#include <byteorder.h>
38#include <errno.h>
39#include <macros.h>
40#include <stdio.h>
41
42#include "format.h"
43
44// TODO float endian?
45#define float_le2host(x) (x)
46#define float_be2host(x) (x)
47
48#define host2float_le(x) (x)
49#define host2float_be(x) (x)
50
51#define from(x, type, endian) (float)(type ## _ ## endian ## 2host(x))
52#define to(x, type, endian) (float)(host2 ## type ## _ ## endian(x))
53
54/** Default linear PCM format */
55const pcm_format_t AUDIO_FORMAT_DEFAULT = {
56 .channels = 2,
57 .sampling_rate = 44100,
58 .sample_format = PCM_SAMPLE_SINT16_LE,
59 };
60
61/** Special ANY PCM format.
62 * This format is used if the real format is no know or important.
63 */
64const pcm_format_t AUDIO_FORMAT_ANY = {
65 .channels = 0,
66 .sampling_rate = 0,
67 .sample_format = 0,
68 };
69
70static float get_normalized_sample(const void *buffer, size_t size,
71 unsigned frame, unsigned channel, const pcm_format_t *f);
72
73/**
74 * Compare PCM format attribtues.
75 * @param a Format description.
76 * @param b Format description.
77 * @return True if a and b describe the same format, false otherwise.
78 */
79bool pcm_format_same(const pcm_format_t *a, const pcm_format_t* b)
80{
81 assert(a);
82 assert(b);
83 return
84 a->sampling_rate == b->sampling_rate &&
85 a->channels == b->channels &&
86 a->sample_format == b->sample_format;
87}
88
89/**
90 * Fill audio buffer with silence in the specified format.
91 * @param dst Destination audio buffer.
92 * @param size Size of the destination audio buffer.
93 * @param f Pointer to the format description.
94 */
95void pcm_format_silence(void *dst, size_t size, const pcm_format_t *f)
96{
97#define SET_NULL(type, endian, nullv) \
98do { \
99 type *buffer = dst; \
100 const size_t sample_count = size / sizeof(type); \
101 for (unsigned i = 0; i < sample_count; ++i) { \
102 buffer[i] = to((type)nullv, type, endian); \
103 } \
104} while (0)
105
106 switch (f->sample_format) {
107 case PCM_SAMPLE_UINT8:
108 SET_NULL(uint8_t, le, INT8_MIN); break;
109 case PCM_SAMPLE_SINT8:
110 SET_NULL(int8_t, le, 0); break;
111 case PCM_SAMPLE_UINT16_LE:
112 SET_NULL(uint16_t, le, INT16_MIN); break;
113 case PCM_SAMPLE_SINT16_LE:
114 SET_NULL(int16_t, le, 0); break;
115 case PCM_SAMPLE_UINT16_BE:
116 SET_NULL(uint16_t, be, INT16_MIN); break;
117 case PCM_SAMPLE_SINT16_BE:
118 SET_NULL(int16_t, be, 0); break;
119 case PCM_SAMPLE_UINT32_LE:
120 SET_NULL(uint32_t, le, INT32_MIN); break;
121 case PCM_SAMPLE_SINT32_LE:
122 SET_NULL(int32_t, le, 0); break;
123 case PCM_SAMPLE_UINT32_BE:
124 SET_NULL(uint32_t, be, INT32_MIN); break;
125 case PCM_SAMPLE_SINT32_BE:
126 SET_NULL(int32_t, le, 0); break;
127 case PCM_SAMPLE_UINT24_32_LE:
128 case PCM_SAMPLE_SINT24_32_LE:
129 case PCM_SAMPLE_UINT24_32_BE:
130 case PCM_SAMPLE_SINT24_32_BE:
131 case PCM_SAMPLE_UINT24_LE:
132 case PCM_SAMPLE_SINT24_LE:
133 case PCM_SAMPLE_UINT24_BE:
134 case PCM_SAMPLE_SINT24_BE:
135 case PCM_SAMPLE_FLOAT32:
136 default: ;
137 }
138#undef SET_NULL
139}
140
141/**
142 * Mix audio data of the same format and size.
143 * @param dst Destination buffer
144 * @param src Source buffer
145 * @param size Size of both the destination and the source buffer
146 * @param f Pointer to the format descriptor.
147 * @return Error code.
148 */
149errno_t pcm_format_mix(void *dst, const void *src, size_t size, const pcm_format_t *f)
150{
151 return pcm_format_convert_and_mix(dst, size, src, size, f, f);
152}
153
154/**
155 * Add and mix audio data.
156 * @param dst Destination audio buffer
157 * @param dst_size Size of the destination buffer
158 * @param src Source audio buffer
159 * @param src_size Size of the source buffer.
160 * @param sf Pointer to the source format descriptor.
161 * @param df Pointer to the destination format descriptor.
162 * @return Error code.
163 *
164 * Buffers must contain entire frames. Destination buffer is always filled.
165 * If there are not enough data in the source buffer silent data is assumed.
166 */
167errno_t pcm_format_convert_and_mix(void *dst, size_t dst_size, const void *src,
168 size_t src_size, const pcm_format_t *sf, const pcm_format_t *df)
169{
170 if (!dst || !src || !sf || !df)
171 return EINVAL;
172 const size_t src_frame_size = pcm_format_frame_size(sf);
173 if ((src_size % src_frame_size) != 0)
174 return EINVAL;
175
176 const size_t dst_frame_size = pcm_format_frame_size(df);
177 if ((dst_size % dst_frame_size) != 0)
178 return EINVAL;
179
180 /* This is so ugly it eats kittens, and puppies, and ducklings,
181 * and all little fluffy things...
182 */
183#define LOOP_ADD(type, endian, low, high) \
184do { \
185 const unsigned frame_count = dst_size / dst_frame_size; \
186 for (size_t i = 0; i < frame_count; ++i) { \
187 for (unsigned j = 0; j < df->channels; ++j) { \
188 const float a = \
189 get_normalized_sample(dst, dst_size, i, j, df);\
190 const float b = \
191 get_normalized_sample(src, src_size, i, j, sf);\
192 float c = (a + b); \
193 if (c < -1.0) c = -1.0; \
194 if (c > 1.0) c = 1.0; \
195 c += 1.0; \
196 c *= ((float)(type)high - (float)(type)low) / 2; \
197 c += (float)(type)low; \
198 type *dst_buf = dst; \
199 const unsigned pos = i * df->channels + j; \
200 if (pos < (dst_size / sizeof(type))) \
201 dst_buf[pos] = to((type)c, type, endian); \
202 } \
203 } \
204} while (0)
205
206 switch (df->sample_format) {
207 case PCM_SAMPLE_UINT8:
208 LOOP_ADD(uint8_t, le, UINT8_MIN, UINT8_MAX); break;
209 case PCM_SAMPLE_SINT8:
210 LOOP_ADD(uint8_t, le, INT8_MIN, INT8_MAX); break;
211 case PCM_SAMPLE_UINT16_LE:
212 LOOP_ADD(uint16_t, le, UINT16_MIN, UINT16_MAX); break;
213 case PCM_SAMPLE_SINT16_LE:
214 LOOP_ADD(int16_t, le, INT16_MIN, INT16_MAX); break;
215 case PCM_SAMPLE_UINT16_BE:
216 LOOP_ADD(uint16_t, be, UINT16_MIN, UINT16_MAX); break;
217 case PCM_SAMPLE_SINT16_BE:
218 LOOP_ADD(int16_t, be, INT16_MIN, INT16_MAX); break;
219 case PCM_SAMPLE_UINT24_32_LE:
220 case PCM_SAMPLE_UINT32_LE: // TODO this are not right for 24bit
221 LOOP_ADD(uint32_t, le, UINT32_MIN, UINT32_MAX); break;
222 case PCM_SAMPLE_SINT24_32_LE:
223 case PCM_SAMPLE_SINT32_LE:
224 LOOP_ADD(int32_t, le, INT32_MIN, INT32_MAX); break;
225 case PCM_SAMPLE_UINT24_32_BE:
226 case PCM_SAMPLE_UINT32_BE:
227 LOOP_ADD(uint32_t, be, UINT32_MIN, UINT32_MAX); break;
228 case PCM_SAMPLE_SINT24_32_BE:
229 case PCM_SAMPLE_SINT32_BE:
230 LOOP_ADD(int32_t, be, INT32_MIN, INT32_MAX); break;
231 case PCM_SAMPLE_UINT24_LE:
232 case PCM_SAMPLE_SINT24_LE:
233 case PCM_SAMPLE_UINT24_BE:
234 case PCM_SAMPLE_SINT24_BE:
235 case PCM_SAMPLE_FLOAT32:
236 default:
237 return ENOTSUP;
238 }
239 return EOK;
240#undef LOOP_ADD
241}
242
243/**
244 * Converts all sample formats to float <-1,1>
245 * @param buffer Audio data
246 * @param size Size of the buffer
247 * @param frame Index of the frame to read
248 * @param channel Channel within the frame
249 * @param f Pointer to a format descriptor
250 * @return Normalized sample <-1,1>, 0.0 if the data could not be read
251 */
252static float get_normalized_sample(const void *buffer, size_t size,
253 unsigned frame, unsigned channel, const pcm_format_t *f)
254{
255 assert(f);
256 assert(buffer);
257 if (channel >= f->channels)
258 return 0.0f;
259#define GET(type, endian, low, high) \
260do { \
261 const type *src = buffer; \
262 const size_t sample_count = size / sizeof(type); \
263 const size_t sample_pos = frame * f->channels + channel; \
264 if (sample_pos >= sample_count) {\
265 return 0.0f; \
266 } \
267 float sample = from(src[sample_pos], type, endian); \
268 /* This makes it positive */ \
269 sample -= (float)(type)low; \
270 /* This makes it <0,2> */ \
271 sample /= (((float)(type)high - (float)(type)low) / 2.0f); \
272 return sample - 1.0f; \
273} while (0)
274
275 switch (f->sample_format) {
276 case PCM_SAMPLE_UINT8:
277 GET(uint8_t, le, UINT8_MIN, UINT8_MAX);
278 case PCM_SAMPLE_SINT8:
279 GET(int8_t, le, INT8_MIN, INT8_MAX);
280 case PCM_SAMPLE_UINT16_LE:
281 GET(uint16_t, le, UINT16_MIN, UINT16_MAX);
282 case PCM_SAMPLE_SINT16_LE:
283 GET(int16_t, le, INT16_MIN, INT16_MAX);
284 case PCM_SAMPLE_UINT16_BE:
285 GET(uint16_t, be, UINT16_MIN, UINT16_MAX);
286 case PCM_SAMPLE_SINT16_BE:
287 GET(int16_t, be, INT16_MIN, INT16_MAX);
288 case PCM_SAMPLE_UINT24_32_LE:
289 case PCM_SAMPLE_UINT32_LE:
290 GET(uint32_t, le, UINT32_MIN, UINT32_MAX);
291 case PCM_SAMPLE_SINT24_32_LE:
292 case PCM_SAMPLE_SINT32_LE:
293 GET(int32_t, le, INT32_MIN, INT32_MAX);
294 case PCM_SAMPLE_UINT24_32_BE:
295 case PCM_SAMPLE_UINT32_BE:
296 GET(uint32_t, be, UINT32_MIN, UINT32_MAX);
297 case PCM_SAMPLE_SINT24_32_BE:
298 case PCM_SAMPLE_SINT32_BE:
299 GET(int32_t, le, INT32_MIN, INT32_MAX);
300 case PCM_SAMPLE_UINT24_LE:
301 case PCM_SAMPLE_SINT24_LE:
302 case PCM_SAMPLE_UINT24_BE:
303 case PCM_SAMPLE_SINT24_BE:
304 case PCM_SAMPLE_FLOAT32:
305 default: ;
306 }
307 return 0;
308#undef GET
309}
310/**
311 * @}
312 */
Note: See TracBrowser for help on using the repository browser.