2 Copyright (C) 2014 Paul Brossier <piem@aubio.org>
4 This file is part of aubio.
6 aubio 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 3 of the License, or
9 (at your option) any later version.
11 aubio 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.
16 You should have received a copy of the GNU General Public License
17 along with aubio. If not, see <http://www.gnu.org/licenses/>.
25 #include "aubio_priv.h"
28 #include "source_wavread.h"
32 #define AUBIO_WAVREAD_BUFSIZE 1024
34 #define SHORT_TO_FLOAT(x) (smpl_t)(x * 3.0517578125e-05)
36 struct _aubio_source_wavread_t {
41 // some data about the file
43 uint_t input_samplerate;
44 uint_t input_channels;
57 unsigned char *short_output;
61 unsigned int read_little_endian (unsigned char *buf, unsigned int length);
62 unsigned int read_little_endian (unsigned char *buf, unsigned int length) {
64 for (i = 0; i < length; i++) {
65 ret += buf[i] << (i * 8);
70 aubio_source_wavread_t * new_aubio_source_wavread(const char_t * path, uint_t samplerate, uint_t hop_size) {
71 aubio_source_wavread_t * s = AUBIO_NEW(aubio_source_wavread_t);
72 size_t bytes_read = 0, bytes_expected = 44;
74 unsigned int format, channels, sr, byterate, blockalign, bitspersample;//, data_size;
77 AUBIO_ERR("source_wavread: Aborted opening null path\n");
80 if ((sint_t)samplerate < 0) {
81 AUBIO_ERR("source_wavread: Can not open %s with samplerate %d\n", path, samplerate);
84 if ((sint_t)hop_size <= 0) {
85 AUBIO_ERR("source_wavread: Can not open %s with hop_size %d\n", path, hop_size);
89 if (s->path) AUBIO_FREE(s->path);
90 s->path = AUBIO_ARRAY(char_t, strnlen(path, PATH_MAX));
91 strncpy(s->path, path, strnlen(path, PATH_MAX));
93 s->samplerate = samplerate;
94 s->hop_size = hop_size;
96 s->fid = fopen((const char *)path, "rb");
98 AUBIO_ERR("source_wavread: Failed opening %s (System error: %s)\n", s->path, strerror(errno));
103 bytes_read += fread(buf, 1, 4, s->fid);
105 if ( strcmp((const char *)buf, "RIFF") != 0 ) {
106 AUBIO_ERR("source_wavread: could not find RIFF header in %s\n", s->path);
111 bytes_read += fread(buf, 1, 4, s->fid);
114 bytes_read += fread(buf, 1, 4, s->fid);
116 if ( strcmp((const char *)buf, "WAVE") != 0 ) {
117 AUBIO_ERR("source_wavread: wrong format in RIFF header in %s\n", s->path);
122 bytes_read += fread(buf, 1, 4, s->fid);
124 if ( strcmp((const char *)buf, "fmt ") != 0 ) {
125 AUBIO_ERR("source_wavread: fmt RIFF header in %s\n", s->path);
130 bytes_read += fread(buf, 1, 4, s->fid);
131 format = read_little_endian(buf, 4);
132 if ( format != 16 ) {
133 // TODO accept format 18
134 AUBIO_ERR("source_wavread: file %s is not encoded with PCM\n", s->path);
137 if ( buf[1] || buf[2] | buf[3] ) {
138 AUBIO_ERR("source_wavread: Subchunk1Size should be 0, in %s\n", s->path);
143 bytes_read += fread(buf, 1, 2, s->fid);
144 if ( buf[0] != 1 || buf[1] != 0) {
145 AUBIO_ERR("source_wavread: AudioFormat should be PCM, in %s\n", s->path);
150 bytes_read += fread(buf, 1, 2, s->fid);
151 channels = read_little_endian(buf, 2);
154 bytes_read += fread(buf, 1, 4, s->fid);
155 sr = read_little_endian(buf, 4);
158 bytes_read += fread(buf, 1, 4, s->fid);
159 byterate = read_little_endian(buf, 4);
162 bytes_read += fread(buf, 1, 2, s->fid);
163 blockalign = read_little_endian(buf, 2);
166 bytes_read += fread(buf, 1, 2, s->fid);
167 bitspersample = read_little_endian(buf, 2);
169 if ( bitspersample != 16 ) {
170 AUBIO_ERR("source_wavread: can not process %dbit file %s\n",
171 bitspersample, s->path);
176 if ( byterate * 8 != sr * channels * bitspersample ) {
177 AUBIO_ERR("source_wavread: wrong byterate in %s\n", s->path);
181 if ( blockalign * 8 != channels * bitspersample ) {
182 AUBIO_ERR("source_wavread: wrong blockalign in %s\n", s->path);
186 s->input_samplerate = sr;
187 s->input_channels = channels;
190 AUBIO_DBG("channels %d\n", channels);
191 AUBIO_DBG("sr %d\n", sr);
192 AUBIO_DBG("byterate %d\n", byterate);
193 AUBIO_DBG("blockalign %d\n", blockalign);
194 AUBIO_DBG("bitspersample %d\n", bitspersample);
196 AUBIO_DBG("found %d channels in %s\n", s->input_channels, s->path);
197 AUBIO_DBG("found %d samplerate in %s\n", s->input_samplerate, s->path);
200 if (samplerate == 0) {
201 s->samplerate = s->input_samplerate;
202 } else if (samplerate != s->input_samplerate) {
203 AUBIO_ERR("source_wavread: can not resample %s from %d to %dHz\n",
204 s->path, s->input_samplerate, samplerate);
209 bytes_read += fread(buf, 1, 4, s->fid);
211 if ( strcmp((const char *)buf, "data") != 0 ) {
212 AUBIO_ERR("source_wavread: data RIFF header not found in %s\n", s->path);
217 bytes_read += fread(buf, 1, 4, s->fid);
218 //data_size = buf[0] + (buf[1] << 8) + (buf[2] << 16) + (buf[3] << 24);
219 //AUBIO_MSG("found %d frames in %s\n", 8 * data_size / bitspersample / channels, s->path);
221 // check the total number of bytes read is correct
222 if ( bytes_read != bytes_expected ) {
223 #ifndef HAVE_WIN_HACKS
224 AUBIO_ERR("source_wavread: short read (%zd instead of %zd) in %s\n",
225 bytes_read, bytes_expected, s->path);
226 #else // mingw does not know about %zd...
227 AUBIO_ERR("source_wavread: short read (%d instead of %d) in %s\n",
228 (int)bytes_read, (int)bytes_expected, s->path);
232 s->seek_start = bytes_read;
234 s->output = new_fmat(s->input_channels, AUBIO_WAVREAD_BUFSIZE);
235 s->blockalign= blockalign;
236 s->bitspersample = bitspersample;
238 s->short_output = (unsigned char *)calloc(s->blockalign, AUBIO_WAVREAD_BUFSIZE);
246 //AUBIO_ERR("source_wavread: can not read %s at samplerate %dHz with a hop_size of %d\n",
247 // s->path, s->samplerate, s->hop_size);
248 del_aubio_source_wavread(s);
252 void aubio_source_wavread_readframe(aubio_source_wavread_t *s, uint_t *wavread_read);
254 void aubio_source_wavread_readframe(aubio_source_wavread_t *s, uint_t *wavread_read) {
255 unsigned char *short_ptr = s->short_output;
256 size_t read = fread(short_ptr, s->blockalign, AUBIO_WAVREAD_BUFSIZE, s->fid);
257 uint_t i, j, b, bitspersample = s->bitspersample;
258 uint_t wrap_at = (1 << ( bitspersample - 1 ) );
259 uint_t wrap_with = (1 << bitspersample);
260 smpl_t scaler = 1. / wrap_at;
262 unsigned int unsigned_val = 0;
264 for (j = 0; j < read; j++) {
265 for (i = 0; i < s->input_channels; i++) {
267 for (b = 0; b < bitspersample; b+=8 ) {
268 unsigned_val += *(short_ptr) << b;
271 signed_val = unsigned_val;
272 // FIXME why does 8 bit conversion maps [0;255] to [-128;127]
273 // instead of [0;127] to [0;127] and [128;255] to [-128;-1]
274 if (bitspersample == 8) signed_val -= wrap_at;
275 else if (unsigned_val >= wrap_at) signed_val = unsigned_val - wrap_with;
276 s->output->data[i][j] = signed_val * scaler;
280 *wavread_read = read;
282 if (read == 0) s->eof = 1;
285 void aubio_source_wavread_do(aubio_source_wavread_t * s, fvec_t * read_data, uint_t * read){
288 uint_t total_wrote = 0;
289 while (total_wrote < s->hop_size) {
290 end = MIN(s->read_samples - s->read_index, s->hop_size - total_wrote);
291 for (i = 0; i < end; i++) {
292 read_data->data[i + total_wrote] = 0;
293 for (j = 0; j < s->input_channels; j++ ) {
294 read_data->data[i + total_wrote] += s->output->data[j][i + s->read_index];
296 read_data->data[i + total_wrote] /= (smpl_t)(s->input_channels);
299 if (total_wrote < s->hop_size) {
300 uint_t wavread_read = 0;
301 aubio_source_wavread_readframe(s, &wavread_read);
302 s->read_samples = wavread_read;
308 s->read_index += end;
311 if (total_wrote < s->hop_size) {
312 for (i = end; i < s->hop_size; i++) {
313 read_data->data[i] = 0.;
319 void aubio_source_wavread_do_multi(aubio_source_wavread_t * s, fmat_t * read_data, uint_t * read){
322 uint_t total_wrote = 0;
323 while (total_wrote < s->hop_size) {
324 end = MIN(s->read_samples - s->read_index, s->hop_size - total_wrote);
325 for (j = 0; j < read_data->height; j++) {
326 for (i = 0; i < end; i++) {
327 read_data->data[j][i + total_wrote] = s->output->data[j][i];
331 if (total_wrote < s->hop_size) {
332 uint_t wavread_read = 0;
333 aubio_source_wavread_readframe(s, &wavread_read);
334 s->read_samples = wavread_read;
340 s->read_index += end;
343 if (total_wrote < s->hop_size) {
344 for (j = 0; j < read_data->height; j++) {
345 for (i = end; i < s->hop_size; i++) {
346 read_data->data[j][i] = 0.;
353 uint_t aubio_source_wavread_get_samplerate(aubio_source_wavread_t * s) {
354 return s->samplerate;
357 uint_t aubio_source_wavread_get_channels(aubio_source_wavread_t * s) {
358 return s->input_channels;
361 uint_t aubio_source_wavread_seek (aubio_source_wavread_t * s, uint_t pos) {
363 if ((sint_t)pos < 0) {
366 ret = fseek(s->fid, s->seek_start + pos * s->blockalign, SEEK_SET);
368 AUBIO_ERR("source_wavread: could not seek %s at %d (%s)\n", s->path, pos, strerror(errno));
377 uint_t aubio_source_wavread_close (aubio_source_wavread_t * s) {
381 if (fclose(s->fid)) {
382 AUBIO_ERR("source_wavread: could not close %s (%s)\n", s->path, strerror(errno));
389 void del_aubio_source_wavread(aubio_source_wavread_t * s) {
391 aubio_source_wavread_close(s);
392 if (s->short_output) AUBIO_FREE(s->short_output);
393 if (s->output) del_fmat(s->output);
394 if (s->path) AUBIO_FREE(s->path);
398 #endif /* HAVE_WAVREAD */