ref: 986131d13a247a071d4450fd0411aa43a9e67141
dir: /src/io/source_wavread.c/
/*
Copyright (C) 2014 Paul Brossier <piem@aubio.org>
This file is part of aubio.
aubio 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 3 of the License, or
(at your option) any later version.
aubio 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 aubio. If not, see <http://www.gnu.org/licenses/>.
*/
#include "aubio_priv.h"
#ifdef HAVE_WAVREAD
#include "fvec.h"
#include "fmat.h"
#include "source_wavread.h"
#include <errno.h>
#define AUBIO_WAVREAD_BUFSIZE 1024
#define SHORT_TO_FLOAT(x) (smpl_t)(x * 3.0517578125e-05)
struct _aubio_source_wavread_t {
uint_t hop_size;
uint_t samplerate;
uint_t channels;
// some data about the file
char_t *path;
uint_t input_samplerate;
uint_t input_channels;
// internal stuff
FILE *fid;
uint_t read_samples;
uint_t blockalign;
uint_t bitspersample;
uint_t read_index;
uint_t eof;
uint_t duration;
size_t seek_start;
unsigned char *short_output;
fmat_t *output;
};
static unsigned int read_little_endian (unsigned char *buf,
unsigned int length);
static unsigned int read_little_endian (unsigned char *buf,
unsigned int length)
{
uint_t i, ret = 0;
for (i = 0; i < length; i++) {
ret += buf[i] << (i * 8);
}
return ret;
}
aubio_source_wavread_t * new_aubio_source_wavread(const char_t * path, uint_t samplerate, uint_t hop_size) {
aubio_source_wavread_t * s = AUBIO_NEW(aubio_source_wavread_t);
size_t bytes_read = 0, bytes_junk = 0, bytes_expected = 44;
unsigned char buf[5] = "\0";
unsigned int format, channels, sr, byterate, blockalign, duration, bitspersample;//, data_size;
if (path == NULL) {
AUBIO_ERR("source_wavread: Aborted opening null path\n");
goto beach;
}
if ((sint_t)samplerate < 0) {
AUBIO_ERR("source_wavread: Can not open %s with samplerate %d\n", path, samplerate);
goto beach;
}
if ((sint_t)hop_size <= 0) {
AUBIO_ERR("source_wavread: Can not open %s with hop_size %d\n", path, hop_size);
goto beach;
}
if (s->path) AUBIO_FREE(s->path);
s->path = AUBIO_ARRAY(char_t, strnlen(path, PATH_MAX) + 1);
strncpy(s->path, path, strnlen(path, PATH_MAX) + 1);
s->samplerate = samplerate;
s->hop_size = hop_size;
s->fid = fopen((const char *)path, "rb");
if (!s->fid) {
AUBIO_ERR("source_wavread: Failed opening %s (System error: %s)\n", s->path, strerror(errno));
goto beach;
}
// ChunkID
bytes_read += fread(buf, 1, 4, s->fid);
buf[4] = '\0';
if ( strcmp((const char *)buf, "RIFF") != 0 ) {
AUBIO_ERR("source_wavread: Failed opening %s (could not find RIFF header)\n", s->path);
goto beach;
}
// ChunkSize
bytes_read += fread(buf, 1, 4, s->fid);
// Format
bytes_read += fread(buf, 1, 4, s->fid);
buf[4] = '\0';
if ( strcmp((const char *)buf, "WAVE") != 0 ) {
AUBIO_ERR("source_wavread: Failed opening %s (wrong format in RIFF header)\n", s->path);
goto beach;
}
// Subchunk1ID
bytes_read += fread(buf, 1, 4, s->fid);
buf[4] = '\0';
// check if we have a JUNK Chunk
if ( strcmp((const char *)buf, "JUNK") == 0 ) {
bytes_junk = fread(buf, 1, 4, s->fid);
buf[4] = '\0';
bytes_junk += read_little_endian(buf, 4);
if (fseek(s->fid, bytes_read + bytes_junk, SEEK_SET) != 0) {
AUBIO_ERR("source_wavread: Failed opening %s (could not seek past JUNK Chunk: %s)\n",
s->path, strerror(errno));
goto beach;
}
bytes_read += bytes_junk;
bytes_expected += bytes_junk + 4;
// now really read the fmt chunk
bytes_read += fread(buf, 1, 4, s->fid);
buf[4] = '\0';
}
// get the fmt chunk
if ( strcmp((const char *)buf, "fmt ") != 0 ) {
AUBIO_ERR("source_wavread: Failed opening %s (could not find 'fmt ' in RIFF header)\n", s->path);
goto beach;
}
// Subchunk1Size
bytes_read += fread(buf, 1, 4, s->fid);
format = read_little_endian(buf, 4);
if ( format != 16 ) {
// TODO accept format 18
AUBIO_ERR("source_wavread: Failed opening %s (not encoded with PCM)\n", s->path);
goto beach;
}
if ( buf[1] || buf[2] | buf[3] ) {
AUBIO_ERR("source_wavread: Failed opening %s (Subchunk1Size should be 0)\n", s->path);
goto beach;
}
// AudioFormat
bytes_read += fread(buf, 1, 2, s->fid);
if ( buf[0] != 1 || buf[1] != 0) {
AUBIO_ERR("source_wavread: Failed opening %s (AudioFormat should be PCM)\n", s->path);
goto beach;
}
// NumChannels
bytes_read += fread(buf, 1, 2, s->fid);
channels = read_little_endian(buf, 2);
// SampleRate
bytes_read += fread(buf, 1, 4, s->fid);
sr = read_little_endian(buf, 4);
// ByteRate
bytes_read += fread(buf, 1, 4, s->fid);
byterate = read_little_endian(buf, 4);
// BlockAlign
bytes_read += fread(buf, 1, 2, s->fid);
blockalign = read_little_endian(buf, 2);
// BitsPerSample
bytes_read += fread(buf, 1, 2, s->fid);
bitspersample = read_little_endian(buf, 2);
#if 0
if ( bitspersample != 16 ) {
AUBIO_ERR("source_wavread: can not process %dbit file %s\n",
bitspersample, s->path);
goto beach;
}
#endif
if ( byterate * 8 != sr * channels * bitspersample ) {
AUBIO_ERR("source_wavread: Failed opening %s (wrong byterate)\n", s->path);
goto beach;
}
if ( blockalign * 8 != channels * bitspersample ) {
AUBIO_ERR("source_wavread: Failed opening %s (wrong blockalign)\n", s->path);
goto beach;
}
s->input_samplerate = sr;
s->input_channels = channels;
#if 0
AUBIO_DBG("channels %d\n", channels);
AUBIO_DBG("sr %d\n", sr);
AUBIO_DBG("byterate %d\n", byterate);
AUBIO_DBG("blockalign %d\n", blockalign);
AUBIO_DBG("bitspersample %d\n", bitspersample);
AUBIO_DBG("found %d channels in %s\n", s->input_channels, s->path);
AUBIO_DBG("found %d samplerate in %s\n", s->input_samplerate, s->path);
#endif
if (samplerate == 0) {
s->samplerate = s->input_samplerate;
} else if (samplerate != s->input_samplerate) {
AUBIO_ERR("source_wavread: can not resample %s from %d to %dHz\n",
s->path, s->input_samplerate, samplerate);
goto beach;
}
// Subchunk2ID
bytes_read += fread(buf, 1, 4, s->fid);
buf[4] = '\0';
while ( strcmp((const char *)buf, "data") != 0 ) {
if (feof(s->fid) || ferror(s->fid)) {
AUBIO_ERR("source_wavread: no data RIFF header found in %s\n", s->path);
goto beach;
}
bytes_junk = fread(buf, 1, 4, s->fid);
buf[4] = '\0';
bytes_junk += read_little_endian(buf, 4);
if (fseek(s->fid, bytes_read + bytes_junk, SEEK_SET) != 0) {
AUBIO_ERR("source_wavread: could not seek past unknown chunk in %s (%s)\n",
s->path, strerror(errno));
goto beach;
}
bytes_read += bytes_junk;
bytes_expected += bytes_junk+ 4;
bytes_read += fread(buf, 1, 4, s->fid);
buf[4] = '\0';
}
// Subchunk2Size
bytes_read += fread(buf, 1, 4, s->fid);
duration = read_little_endian(buf, 4) / blockalign;
//data_size = buf[0] + (buf[1] << 8) + (buf[2] << 16) + (buf[3] << 24);
//AUBIO_MSG("found %d frames in %s\n", 8 * data_size / bitspersample / channels, s->path);
// check the total number of bytes read is correct
if ( bytes_read != bytes_expected ) {
#ifndef HAVE_WIN_HACKS
AUBIO_ERR("source_wavread: short read (%zd instead of %zd) in %s\n",
bytes_read, bytes_expected, s->path);
#else // mingw does not know about %zd...
AUBIO_ERR("source_wavread: short read (%d instead of %d) in %s\n",
(int)bytes_read, (int)bytes_expected, s->path);
#endif
goto beach;
}
s->seek_start = bytes_read;
s->output = new_fmat(s->input_channels, AUBIO_WAVREAD_BUFSIZE);
s->blockalign= blockalign;
s->bitspersample = bitspersample;
s->duration = duration;
s->short_output = (unsigned char *)calloc(s->blockalign, AUBIO_WAVREAD_BUFSIZE);
s->read_index = 0;
s->read_samples = 0;
s->eof = 0;
return s;
beach:
//AUBIO_ERR("source_wavread: can not read %s at samplerate %dHz with a hop_size of %d\n",
// s->path, s->samplerate, s->hop_size);
del_aubio_source_wavread(s);
return NULL;
}
void aubio_source_wavread_readframe(aubio_source_wavread_t *s, uint_t *wavread_read);
void aubio_source_wavread_readframe(aubio_source_wavread_t *s, uint_t *wavread_read) {
unsigned char *short_ptr = s->short_output;
size_t read = fread(short_ptr, s->blockalign, AUBIO_WAVREAD_BUFSIZE, s->fid);
uint_t i, j, b, bitspersample = s->bitspersample;
uint_t wrap_at = (1 << ( bitspersample - 1 ) );
uint_t wrap_with = (1 << bitspersample);
smpl_t scaler = 1. / wrap_at;
int signed_val = 0;
unsigned int unsigned_val = 0;
for (j = 0; j < read; j++) {
for (i = 0; i < s->input_channels; i++) {
unsigned_val = 0;
for (b = 0; b < bitspersample; b+=8 ) {
unsigned_val += *(short_ptr) << b;
short_ptr++;
}
signed_val = unsigned_val;
// FIXME why does 8 bit conversion maps [0;255] to [-128;127]
// instead of [0;127] to [0;127] and [128;255] to [-128;-1]
if (bitspersample == 8) signed_val -= wrap_at;
else if (unsigned_val >= wrap_at) signed_val = unsigned_val - wrap_with;
s->output->data[i][j] = signed_val * scaler;
}
}
*wavread_read = read;
if (read == 0) s->eof = 1;
}
void aubio_source_wavread_do(aubio_source_wavread_t * s, fvec_t * read_data, uint_t * read){
uint_t i, j;
uint_t end = 0;
uint_t total_wrote = 0;
if (s->fid == NULL) {
AUBIO_ERR("source_wavread: could not read from %s (file not opened)\n",
s->path);
return;
}
while (total_wrote < s->hop_size) {
end = MIN(s->read_samples - s->read_index, s->hop_size - total_wrote);
for (i = 0; i < end; i++) {
read_data->data[i + total_wrote] = 0;
for (j = 0; j < s->input_channels; j++ ) {
read_data->data[i + total_wrote] += s->output->data[j][i + s->read_index];
}
read_data->data[i + total_wrote] /= (smpl_t)(s->input_channels);
}
total_wrote += end;
if (total_wrote < s->hop_size) {
uint_t wavread_read = 0;
aubio_source_wavread_readframe(s, &wavread_read);
s->read_samples = wavread_read;
s->read_index = 0;
if (s->eof) {
break;
}
} else {
s->read_index += end;
}
}
if (total_wrote < s->hop_size) {
for (i = end; i < s->hop_size; i++) {
read_data->data[i] = 0.;
}
}
*read = total_wrote;
}
void aubio_source_wavread_do_multi(aubio_source_wavread_t * s, fmat_t * read_data, uint_t * read){
uint_t i,j;
uint_t end = 0;
uint_t total_wrote = 0;
if (s->fid == NULL) {
AUBIO_ERR("source_wavread: could not read from %s (file not opened)\n",
s->path);
return;
}
while (total_wrote < s->hop_size) {
end = MIN(s->read_samples - s->read_index, s->hop_size - total_wrote);
for (j = 0; j < read_data->height; j++) {
for (i = 0; i < end; i++) {
read_data->data[j][i + total_wrote] = s->output->data[j][i];
}
}
total_wrote += end;
if (total_wrote < s->hop_size) {
uint_t wavread_read = 0;
aubio_source_wavread_readframe(s, &wavread_read);
s->read_samples = wavread_read;
s->read_index = 0;
if (s->eof) {
break;
}
} else {
s->read_index += end;
}
}
if (total_wrote < s->hop_size) {
for (j = 0; j < read_data->height; j++) {
for (i = end; i < s->hop_size; i++) {
read_data->data[j][i] = 0.;
}
}
}
*read = total_wrote;
}
uint_t aubio_source_wavread_get_samplerate(aubio_source_wavread_t * s) {
return s->samplerate;
}
uint_t aubio_source_wavread_get_channels(aubio_source_wavread_t * s) {
return s->input_channels;
}
uint_t aubio_source_wavread_seek (aubio_source_wavread_t * s, uint_t pos) {
uint_t ret = 0;
if (s->fid == NULL) {
AUBIO_ERR("source_wavread: could not seek %s (file not opened?)\n", s->path, pos);
return AUBIO_FAIL;
}
if ((sint_t)pos < 0) {
AUBIO_ERR("source_wavread: could not seek %s at %d (seeking position should be >= 0)\n", s->path, pos);
return AUBIO_FAIL;
}
ret = fseek(s->fid, s->seek_start + pos * s->blockalign, SEEK_SET);
if (ret != 0) {
AUBIO_ERR("source_wavread: could not seek %s at %d (%s)\n", s->path, pos, strerror(errno));
return AUBIO_FAIL;
}
// reset some values
s->eof = 0;
s->read_index = 0;
return AUBIO_OK;
}
uint_t aubio_source_wavread_get_duration (const aubio_source_wavread_t * s) {
if (s && s->duration) {
return s->duration;
}
return 0;
}
uint_t aubio_source_wavread_close (aubio_source_wavread_t * s) {
if (s->fid == NULL) {
return AUBIO_OK;
}
if (fclose(s->fid)) {
AUBIO_ERR("source_wavread: could not close %s (%s)\n", s->path, strerror(errno));
return AUBIO_FAIL;
}
s->fid = NULL;
return AUBIO_OK;
}
void del_aubio_source_wavread(aubio_source_wavread_t * s) {
if (!s) return;
aubio_source_wavread_close(s);
if (s->short_output) AUBIO_FREE(s->short_output);
if (s->output) del_fmat(s->output);
if (s->path) AUBIO_FREE(s->path);
AUBIO_FREE(s);
}
#endif /* HAVE_WAVREAD */