MStreamRecorder.java

/*
 *  Copyright (c) 2007 by Damien Di Fede <ddf@compartmental.net>
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU Library 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 Library General Public License for more details.
 *
 *   You should have received a copy of the GNU Library General Public
 *   License along with this program; if not, write to the Free Software
 *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

package ddf.minim;

import java.io.File;
import java.io.IOException;

import javax.sound.sampled.AudioFileFormat;
import javax.sound.sampled.AudioFormat;
import javax.sound.sampled.AudioSystem;

import org.tritonus.share.sampled.AudioSystemShadow;
import org.tritonus.share.sampled.FloatSampleBuffer;
import org.tritonus.share.sampled.file.AudioOutputStream;

/**
 * MStreamRecorder using the Tritonus AudioOutputStream class to stream audio
 * directly to disk. The limitation of this approach is that the file format and
 * the file name must be known before recording begins because the file must be
 * created. The advantage is that you do not incur the overhead of an in-memory
 * buffer and saving will not cause your sketch to hang because all the audio is
 * already on disk and all that must be done is closing the file. Unlike
 * MBufferedRecorder, specifying the file format upon saving will do nothing and
 * you cannot easily save your recorded audio to multiple formats. There are
 * also fewer formats available to save in, limiting you to AIFF, AU, and WAV.
 * 
 * @author Damien Di Fede
 * 
 */
final class MStreamRecorder implements SampleRecorder {
	// output stream representing the file being written to
	private AudioOutputStream aos;
	// float sample buffer used for converting float samples to bytes
	private FloatSampleBuffer fsb;
	private String name;
	private AudioFileFormat.Type type;
	private AudioFormat format;
	private boolean recording;

	/**
	 * 
	 * @param fileName
	 * @param fileType
	 * @param fileFormat
	 */
	MStreamRecorder(String fileName, AudioFileFormat.Type fileType,
			AudioFormat fileFormat, int bufferSize) {
		name = fileName;
		type = fileType;
		format = fileFormat;
		try {
			aos = AudioSystemShadow.getAudioOutputStream(type, format,
					AudioSystem.NOT_SPECIFIED, new File(name));
		} catch (IOException e) {
			Minim
					.error("AudioFileOut.setType: Error obtaining new output stream."
							+ e.getMessage());
		}
		fsb = new FloatSampleBuffer(format.getChannels(), bufferSize, format
				.getSampleRate());
		recording = false;
	}

	public String filePath() {
		return name;
	}

	public void beginRecord() {
		recording = true;
	}

	public void endRecord() {
		recording = false;
	}

	public boolean isRecording() {
		return recording;
	}

	/**
	 * Finishes the recording process by closing the file.
	 */
	public void save() {
		try {
			aos.close();
		} catch (IOException e) {
			Minim
					.error("AudioRecorder.save: An error occurred when trying to save the file:\n"
							+ e.getMessage());
		}
	}

	public void samples(float[] samp) {
		if (recording) {
			System.arraycopy(samp, 0, fsb.getChannel(0), 0, samp.length);
			byte[] raw = fsb.convertToByteArray(format);
			try {
				aos.write(raw, 0, raw.length);
			} catch (IOException e) {
				Minim
						.error("AudioRecorder: An error occurred while trying to write to the file:\n"
								+ e.getMessage());
			}
		}
	}

	public void samples(float[] sampL, float[] sampR) {
		if (recording) {
			System.arraycopy(sampL, 0, fsb.getChannel(0), 0, sampL.length);
			System.arraycopy(sampR, 0, fsb.getChannel(1), 0, sampR.length);
			byte[] raw = fsb.convertToByteArray(format);
			try {
				aos.write(raw, 0, raw.length);
			} catch (IOException e) {
				Minim
						.error("AudioRecorder: An error occurred while trying to write to the file:\n"
								+ e.getMessage());
			}
		}
	}
}