ref: 36cc2a77dc286bd596f59166f6737ffb4a5e2e85
dir: /src/sndfile.c/
/* libSoX libsndfile formats.
*
* Copyright 2007 Reuben Thomas <rrt@sc3d.org>
* Copyright 1999-2005 Erik de Castro Lopo <eridk@mega-nerd.com>
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or (at
* your option) any later version.
*
* This library 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 Lesser
* General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "sox_i.h"
#ifdef HAVE_SNDFILE
#include <assert.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <sndfile.h>
#define LOG_MAX 2048 /* As per the SFC_GET_LOG_INFO example */
#ifndef HACKED_LSF
#define sf_stop(x)
#endif
/* Private data for sndfile files */
typedef struct {
SNDFILE *sf_file;
SF_INFO *sf_info;
char * log_buffer;
char const * log_buffer_ptr;
} priv_t;
/*
* Drain LSF's wonderful log buffer
*/
static void drain_log_buffer(sox_format_t * ft)
{
priv_t * sf = (priv_t *)ft->priv;
sf_command(sf->sf_file, SFC_GET_LOG_INFO, sf->log_buffer, LOG_MAX);
while (*sf->log_buffer_ptr) {
static char const warning_prefix[] = "*** Warning : ";
char const * end = strchr(sf->log_buffer_ptr, '\n');
if (!end)
end = strchr(sf->log_buffer_ptr, '\0');
if (!strncmp(sf->log_buffer_ptr, warning_prefix, strlen(warning_prefix))) {
sf->log_buffer_ptr += strlen(warning_prefix);
lsx_warn("`%s': %.*s",
ft->filename, (int)(end - sf->log_buffer_ptr), sf->log_buffer_ptr);
} else
lsx_debug("`%s': %.*s",
ft->filename, (int)(end - sf->log_buffer_ptr), sf->log_buffer_ptr);
sf->log_buffer_ptr = end;
if (*sf->log_buffer_ptr == '\n')
++sf->log_buffer_ptr;
}
}
/* Make libsndfile subtype from sample encoding and size */
static int ft_enc(unsigned size, sox_encoding_t e)
{
if (e == SOX_ENCODING_ULAW && size == 8) return SF_FORMAT_ULAW;
if (e == SOX_ENCODING_ALAW && size == 8) return SF_FORMAT_ALAW;
if (e == SOX_ENCODING_SIGN2 && size == 8) return SF_FORMAT_PCM_S8;
if (e == SOX_ENCODING_SIGN2 && size == 16) return SF_FORMAT_PCM_16;
if (e == SOX_ENCODING_SIGN2 && size == 24) return SF_FORMAT_PCM_24;
if (e == SOX_ENCODING_SIGN2 && size == 32) return SF_FORMAT_PCM_32;
if (e == SOX_ENCODING_UNSIGNED && size == 8) return SF_FORMAT_PCM_U8;
if (e == SOX_ENCODING_FLOAT && size == 32) return SF_FORMAT_FLOAT;
if (e == SOX_ENCODING_FLOAT && size == 64) return SF_FORMAT_DOUBLE;
if (e == SOX_ENCODING_G721 && size == 4) return SF_FORMAT_G721_32;
if (e == SOX_ENCODING_G723 && size == 3) return SF_FORMAT_G723_24;
if (e == SOX_ENCODING_G723 && size == 5) return SF_FORMAT_G723_40;
if (e == SOX_ENCODING_MS_ADPCM && size == 4) return SF_FORMAT_MS_ADPCM;
if (e == SOX_ENCODING_IMA_ADPCM && size == 4) return SF_FORMAT_IMA_ADPCM;
if (e == SOX_ENCODING_OKI_ADPCM && size == 4) return SF_FORMAT_VOX_ADPCM;
if (e == SOX_ENCODING_DPCM && size == 8) return SF_FORMAT_DPCM_8;
if (e == SOX_ENCODING_DPCM && size == 16) return SF_FORMAT_DPCM_16;
if (e == SOX_ENCODING_DWVW && size == 12) return SF_FORMAT_DWVW_12;
if (e == SOX_ENCODING_DWVW && size == 16) return SF_FORMAT_DWVW_16;
if (e == SOX_ENCODING_DWVW && size == 24) return SF_FORMAT_DWVW_24;
if (e == SOX_ENCODING_DWVWN && size == 0) return SF_FORMAT_DWVW_N;
if (e == SOX_ENCODING_GSM && size == 0) return SF_FORMAT_GSM610;
#ifdef HAVE_SNDFILE_1_0_12
if (e == SOX_ENCODING_FLAC && size == 8) return SF_FORMAT_PCM_S8;
if (e == SOX_ENCODING_FLAC && size == 16) return SF_FORMAT_PCM_16;
if (e == SOX_ENCODING_FLAC && size == 24) return SF_FORMAT_PCM_24;
if (e == SOX_ENCODING_FLAC && size == 32) return SF_FORMAT_PCM_32;
#endif
return 0; /* Bad encoding */
}
/* Convert format's encoding type to libSoX encoding type & size. */
static sox_encoding_t sox_enc(int ft_encoding, unsigned * size)
{
int sub = ft_encoding & SF_FORMAT_SUBMASK;
int type = ft_encoding & SF_FORMAT_TYPEMASK;
#ifdef HAVE_SNDFILE_1_0_12
if (type == SF_FORMAT_FLAC) switch (sub) {
case SF_FORMAT_PCM_S8 : *size = 8; return SOX_ENCODING_FLAC;
case SF_FORMAT_PCM_16 : *size = 16; return SOX_ENCODING_FLAC;
case SF_FORMAT_PCM_24 : *size = 24; return SOX_ENCODING_FLAC;
}
#endif
switch (sub) {
case SF_FORMAT_ULAW : *size = 8; return SOX_ENCODING_ULAW;
case SF_FORMAT_ALAW : *size = 8; return SOX_ENCODING_ALAW;
case SF_FORMAT_PCM_S8 : *size = 8; return SOX_ENCODING_SIGN2;
case SF_FORMAT_PCM_16 : *size = 16; return SOX_ENCODING_SIGN2;
case SF_FORMAT_PCM_24 : *size = 24; return SOX_ENCODING_SIGN2;
case SF_FORMAT_PCM_32 : *size = 32; return SOX_ENCODING_SIGN2;
case SF_FORMAT_PCM_U8 : *size = 8; return SOX_ENCODING_UNSIGNED;
case SF_FORMAT_FLOAT : *size = 32; return SOX_ENCODING_FLOAT;
case SF_FORMAT_DOUBLE : *size = 64; return SOX_ENCODING_FLOAT;
case SF_FORMAT_G721_32 : *size = 4; return SOX_ENCODING_G721;
case SF_FORMAT_G723_24 : *size = 3; return SOX_ENCODING_G723;
case SF_FORMAT_G723_40 : *size = 5; return SOX_ENCODING_G723;
case SF_FORMAT_MS_ADPCM : *size = 4; return SOX_ENCODING_MS_ADPCM;
case SF_FORMAT_IMA_ADPCM: *size = 4; return SOX_ENCODING_IMA_ADPCM;
case SF_FORMAT_VOX_ADPCM: *size = 4; return SOX_ENCODING_OKI_ADPCM;
case SF_FORMAT_DPCM_8 : *size = 8; return SOX_ENCODING_DPCM;
case SF_FORMAT_DPCM_16 : *size = 16; return SOX_ENCODING_DPCM;
case SF_FORMAT_DWVW_12 : *size = 12; return SOX_ENCODING_DWVW;
case SF_FORMAT_DWVW_16 : *size = 16; return SOX_ENCODING_DWVW;
case SF_FORMAT_DWVW_24 : *size = 24; return SOX_ENCODING_DWVW;
case SF_FORMAT_DWVW_N : *size = 0; return SOX_ENCODING_DWVWN;
case SF_FORMAT_GSM610 : *size = 0; return SOX_ENCODING_GSM;
default : *size = 0; return SOX_ENCODING_UNKNOWN;
}
}
static struct {
const char *ext;
int format;
} format_map[] =
{
{ "aif", SF_FORMAT_AIFF },
{ "aiff", SF_FORMAT_AIFF },
{ "wav", SF_FORMAT_WAV },
{ "au", SF_FORMAT_AU },
{ "snd", SF_FORMAT_AU },
#ifdef HAVE_SNDFILE_1_0_12
{ "caf", SF_FORMAT_CAF },
{ "flac", SF_FORMAT_FLAC },
#endif
#ifdef HAVE_SNDFILE_1_0_18
{ "wve", SF_FORMAT_WVE },
{ "ogg", SF_FORMAT_OGG },
#endif
{ "svx", SF_FORMAT_SVX },
{ "8svx", SF_FORMAT_SVX },
{ "paf", SF_ENDIAN_BIG | SF_FORMAT_PAF },
{ "fap", SF_ENDIAN_LITTLE | SF_FORMAT_PAF },
{ "gsm", SF_FORMAT_RAW | SF_FORMAT_GSM610 },
{ "nist", SF_FORMAT_NIST },
{ "sph", SF_FORMAT_NIST },
{ "ircam", SF_FORMAT_IRCAM },
{ "sf", SF_FORMAT_IRCAM },
{ "voc", SF_FORMAT_VOC },
{ "w64", SF_FORMAT_W64 },
{ "raw", SF_FORMAT_RAW },
{ "mat4", SF_FORMAT_MAT4 },
{ "mat5", SF_FORMAT_MAT5 },
{ "mat", SF_FORMAT_MAT4 },
{ "pvf", SF_FORMAT_PVF },
{ "sds", SF_FORMAT_SDS },
{ "sd2", SF_FORMAT_SD2 },
{ "vox", SF_FORMAT_RAW | SF_FORMAT_VOX_ADPCM },
{ "xi", SF_FORMAT_XI }
};
/* Convert file name or type to libsndfile format */
static int name_to_format(const char *name)
{
int k;
#define FILE_TYPE_BUFLEN (size_t)15
char buffer[FILE_TYPE_BUFLEN + 1], *cptr;
if ((cptr = strrchr(name, '.')) != NULL) {
strncpy(buffer, cptr + 1, FILE_TYPE_BUFLEN);
buffer[FILE_TYPE_BUFLEN] = '\0';
for (k = 0; buffer[k]; k++)
buffer[k] = tolower((buffer[k]));
} else
strncpy(buffer, name, FILE_TYPE_BUFLEN);
for (k = 0; k < (int)(sizeof(format_map) / sizeof(format_map [0])); k++) {
if (strcmp(buffer, format_map[k].ext) == 0)
return format_map[k].format;
}
return 0;
}
static void start(sox_format_t * ft)
{
priv_t * sf = (priv_t *)ft->priv;
int subtype = ft_enc(ft->encoding.bits_per_sample? ft->encoding.bits_per_sample : ft->signal.precision, ft->encoding.encoding);
sf->log_buffer_ptr = sf->log_buffer = lsx_malloc((size_t)LOG_MAX);
sf->sf_info = lsx_calloc(1, sizeof(SF_INFO));
/* Copy format info */
if (subtype) {
if (strcmp(ft->filetype, "sndfile") == 0)
sf->sf_info->format = name_to_format(ft->filename) | subtype;
else
sf->sf_info->format = name_to_format(ft->filetype) | subtype;
}
sf->sf_info->samplerate = ft->signal.rate;
sf->sf_info->channels = ft->signal.channels;
if (ft->signal.channels)
sf->sf_info->frames = ft->signal.length / ft->signal.channels;
}
static int check_read_params(sox_format_t * ft, unsigned channels,
sox_rate_t rate, sox_encoding_t encoding, unsigned bits_per_sample, off_t length)
{
ft->signal.length = length;
if (channels && ft->signal.channels && ft->signal.channels != channels)
lsx_warn("`%s': overriding number of channels", ft->filename);
else ft->signal.channels = channels;
if (rate && ft->signal.rate && ft->signal.rate != rate)
lsx_warn("`%s': overriding sample rate", ft->filename);
else ft->signal.rate = rate;
if (encoding && ft->encoding.encoding && ft->encoding.encoding != encoding)
lsx_warn("`%s': overriding encoding type", ft->filename);
else ft->encoding.encoding = encoding;
if (bits_per_sample && ft->encoding.bits_per_sample && ft->encoding.bits_per_sample != bits_per_sample)
lsx_warn("`%s': overriding encoding size", ft->filename);
ft->encoding.bits_per_sample = bits_per_sample;
if (sox_precision(ft->encoding.encoding, ft->encoding.bits_per_sample))
return SOX_SUCCESS;
lsx_fail_errno(ft, EINVAL, "invalid format for this file type");
return SOX_EOF;
}
/*
* Open file in sndfile.
*/
static int startread(sox_format_t * ft)
{
priv_t * sf = (priv_t *)ft->priv;
unsigned bits_per_sample;
sox_encoding_t encoding;
sox_rate_t rate;
start(ft);
sf->sf_file = sf_open_fd(fileno(ft->fp), SFM_READ, sf->sf_info, 1);
ft->fp = NULL; /* Transfer ownership of fp to LSF */
drain_log_buffer(ft);
if (sf->sf_file == NULL) {
memset(ft->sox_errstr, 0, sizeof(ft->sox_errstr));
strncpy(ft->sox_errstr, sf_strerror(sf->sf_file), sizeof(ft->sox_errstr)-1);
free(sf->sf_file);
return SOX_EOF;
}
if (!(encoding = sox_enc(sf->sf_info->format, &bits_per_sample))) {
lsx_fail_errno(ft, SOX_EFMT, "unsupported sndfile encoding %#x", sf->sf_info->format);
return SOX_EOF;
}
/* Don't believe LSF's rate for raw files */
if ((sf->sf_info->format & SF_FORMAT_TYPEMASK) == SF_FORMAT_RAW && !ft->signal.rate) {
lsx_warn("'%s': sample rate not specified; trying 8kHz", ft->filename);
rate = 8000;
}
else rate = sf->sf_info->samplerate;
#ifdef HAVE_SFC_SET_SCALE_FLOAT_INT_READ
if ((sf->sf_info->format & SF_FORMAT_SUBMASK) == SF_FORMAT_FLOAT)
sf_command(sf->sf_file, SFC_SET_SCALE_FLOAT_INT_READ, NULL, SF_TRUE);
#endif
#if 0 /* FIXME */
sox_append_comments(&ft->oob.comments, buf);
#endif
return check_read_params(ft, (unsigned)sf->sf_info->channels, rate,
encoding, bits_per_sample, (off_t)(sf->sf_info->frames * sf->sf_info->channels));
}
/*
* Read up to len samples of type sox_sample_t from file into buf[].
* Return number of samples read.
*/
static size_t read_samples(sox_format_t * ft, sox_sample_t *buf, size_t len)
{
priv_t * sf = (priv_t *)ft->priv;
/* FIXME: We assume int == sox_sample_t here */
return (size_t)sf_read_int(sf->sf_file, (int *)buf, (sf_count_t)len);
}
/*
* Close file for libsndfile (this doesn't close the file handle)
*/
static int stopread(sox_format_t * ft)
{
priv_t * sf = (priv_t *)ft->priv;
sf_stop(sf->sf_file);
drain_log_buffer(ft);
sf_close(sf->sf_file);
return SOX_SUCCESS;
}
static int startwrite(sox_format_t * ft)
{
priv_t * sf = (priv_t *)ft->priv;
start(ft);
/* If output format is invalid, try to find a sensible default */
if (!sf_format_check(sf->sf_info)) {
SF_FORMAT_INFO format_info;
int i, count;
sf_command(sf->sf_file, SFC_GET_SIMPLE_FORMAT_COUNT, &count, (int) sizeof(int));
for (i = 0; i < count; i++) {
format_info.format = i;
sf_command(sf->sf_file, SFC_GET_SIMPLE_FORMAT, &format_info, (int) sizeof(format_info));
if ((format_info.format & SF_FORMAT_TYPEMASK) == (sf->sf_info->format & SF_FORMAT_TYPEMASK)) {
sf->sf_info->format = format_info.format;
/* FIXME: Print out exactly what we chose, needs sndfile ->
sox encoding conversion functions */
break;
}
}
if (!sf_format_check(sf->sf_info)) {
lsx_fail("cannot find a usable output encoding");
return SOX_EOF;
}
if ((sf->sf_info->format & SF_FORMAT_TYPEMASK) != SF_FORMAT_RAW)
lsx_warn("cannot use desired output encoding, choosing default");
}
sf->sf_file = sf_open_fd(fileno(ft->fp), SFM_WRITE, sf->sf_info, 1);
ft->fp = NULL; /* Transfer ownership of fp to LSF */
drain_log_buffer(ft);
if (sf->sf_file == NULL) {
memset(ft->sox_errstr, 0, sizeof(ft->sox_errstr));
strncpy(ft->sox_errstr, sf_strerror(sf->sf_file), sizeof(ft->sox_errstr)-1);
free(sf->sf_file);
return SOX_EOF;
}
#ifdef HAVE_SFC_SET_SCALE_INT_FLOAT_WRITE
if ((sf->sf_info->format & SF_FORMAT_SUBMASK) == SF_FORMAT_FLOAT)
sf_command(sf->sf_file, SFC_SET_SCALE_INT_FLOAT_WRITE, NULL, SF_TRUE);
#endif
return SOX_SUCCESS;
}
/*
* Write len samples of type sox_sample_t from buf[] to file.
* Return number of samples written.
*/
static size_t write_samples(sox_format_t * ft, const sox_sample_t *buf, size_t len)
{
priv_t * sf = (priv_t *)ft->priv;
/* FIXME: We assume int == sox_sample_t here */
return (size_t)sf_write_int(sf->sf_file, (int *)buf, (sf_count_t)len);
}
/*
* Close file for libsndfile (this doesn't close the file handle)
*/
static int stopwrite(sox_format_t * ft)
{
priv_t * sf = (priv_t *)ft->priv;
sf_stop(sf->sf_file);
drain_log_buffer(ft);
sf_close(sf->sf_file);
return SOX_SUCCESS;
}
static int seek(sox_format_t * ft, uint64_t offset)
{
priv_t * sf = (priv_t *)ft->priv;
sf_seek(sf->sf_file, (sf_count_t)(offset / ft->signal.channels), SEEK_CUR);
return SOX_SUCCESS;
}
SOX_FORMAT_HANDLER(sndfile)
{
static char const * const names[] = {
"sndfile", /* Special type to force use of sndfile for the following: */
/* LSF implementation of formats built in to SoX: */
/* "aif", */
/* "au", */
/* "gsm", */
/* "nist", */
/* "raw", */
/* "sf", "ircam", */
/* "snd", */
/* "svx", */
/* "voc", */
/* "vox", */
/* "wav", */
/* LSF wrappers of formats already wrapped in SoX: */
/* "flac", */
"sds", /* ?? */
NULL
};
static unsigned const write_encodings[] = {
SOX_ENCODING_SIGN2, 16, 24, 32, 8, 0,
SOX_ENCODING_UNSIGNED, 8, 0,
SOX_ENCODING_FLOAT, 32, 64, 0,
SOX_ENCODING_ALAW, 8, 0,
SOX_ENCODING_ULAW, 8, 0,
SOX_ENCODING_IMA_ADPCM, 4, 0,
SOX_ENCODING_MS_ADPCM, 4, 0,
SOX_ENCODING_OKI_ADPCM, 4, 0,
SOX_ENCODING_GSM, 0,
0};
static sox_format_handler_t const format = {SOX_LIB_VERSION_CODE,
"Pseudo format to use libsndfile", names, 0,
startread, read_samples, stopread,
startwrite, write_samples, stopwrite,
seek, write_encodings, NULL, sizeof(priv_t)
};
return &format;
}
#endif