wave.c

/* Sonic library
   Copyright 2010
   Bill Cox
   This file is part of the Sonic Library.

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., 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.  */

/*
This file supports read/write wave files.
*/
#include <stdlib.h>
#include <limits.h>
#include <string.h>
#include <sndfile.h>
#include "wave.h"

struct waveFileStruct {
    SNDFILE *soundFile;
    short *values;
    int numValues;
    int numChannels;
};

/* Open the file for reading.  Also determine it's sample rate. */
waveFile openInputWaveFile(
    char *fileName,
    int *sampleRate)
{
    SF_INFO info;
    SNDFILE *soundFile;
    waveFile file;

    info.format = 0;
    soundFile = sf_open(fileName, SFM_READ, &info);
    if(soundFile == NULL) {
	fprintf(stderr, "Unable to open wave file %s: %s\n", fileName, sf_strerror(NULL));
	return NULL;
    }
    file = (waveFile)calloc(1, sizeof(struct waveFileStruct));
    file->soundFile = soundFile;
    file->numChannels = info.channels;
    file->numValues = 42;
    file->values = (short *)calloc(file->numValues, sizeof(short));
    *sampleRate = info.samplerate;
    printf("Frames = %ld, sample rate = %d, channels = %d, format = %d\n",
        info.frames, info.samplerate, info.channels, info.format);
    return file;
}

/* Open the file for reading. */
waveFile openOutputWaveFile(
    char *fileName,
    int sampleRate)
{
    SF_INFO info;
    SNDFILE *soundFile;
    waveFile file;

    info.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
    info.samplerate = sampleRate;
    info.channels = 1;
    soundFile = sf_open(fileName, SFM_WRITE, &info);
    if(soundFile == NULL) {
	fprintf(stderr, "Unable to open wave file %s: %s\n", fileName, sf_strerror(NULL));
	return NULL;
    }
    file = (waveFile)calloc(1, sizeof(struct waveFileStruct));
    file->soundFile = soundFile;
    file->numValues = 42;
    file->values = (short *)calloc(file->numValues, sizeof(short));
    return file;
}

/* Close the sound file. */
void closeWaveFile(
    waveFile file)
{
    SNDFILE *soundFile = file->soundFile;

    sf_close(soundFile);
}

/* Read from the wave file. */
int readFromWaveFile(
    waveFile file,
    short *buffer,
    int maxSamples)
{
    SNDFILE *soundFile = file->soundFile;
    int value;
    short *values;
    int samplesRead;
    int numChannels = file->numChannels;
    int i, j;

    if(maxSamples*numChannels > file->numValues) {
	file->numValues = maxSamples*numChannels;
	file->values = (short *)realloc(file->values, file->numValues*sizeof(short));
    }
    values = file->values;
    samplesRead = sf_read_short(soundFile, values, maxSamples*numChannels);
    if(samplesRead <= 0) {
	return 0;
    }
    samplesRead /= numChannels;
    if(numChannels > 1) {
	for(i = 0; i < samplesRead; i++) {
	    value = 0;
	    for(j = 0; j < numChannels; j++) {
		value += values[i*numChannels + j];
	    }
	    if(value >= 0) {
		buffer[i] = value/numChannels;
	    } else {
		/* On some OSes, dividing a negative number rounds the wrong way */
		buffer[i] = -(-value/numChannels);
	    }
	}
    } else {
	memcpy(buffer, values, samplesRead*sizeof(short));
    }
    return samplesRead;
}

/* Write to the wave file. */
int writeToWaveFile(
    waveFile file,
    short *buffer,
    int numSamples)
{
    SNDFILE *soundFile = file->soundFile;
    int numWritten;

    if(numSamples > file->numValues) {
	file->numValues = numSamples*3/2;
	file->values = (short *)realloc(file->values, file->numValues*sizeof(short));
    }
    numWritten = sf_write_short(soundFile, buffer, numSamples);
    if(numWritten != numSamples) {
	fprintf(stderr, "Unable to write wave file.\n");
	return 0;
    }
    return 1;
}