package com.aurellem.capture.audio;

 

 import java.io.File;

 import java.io.FileNotFoundException;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.nio.ByteBuffer;

 

 import javax.sound.sampled.AudioFormat;

 import javax.sound.sampled.AudioSystem;

 

 import org.tritonus.sampled.file.WaveAudioOutputStream;

 import org.tritonus.share.sampled.file.TDataOutputStream;

 import org.tritonus.share.sampled.file.TNonSeekableDataOutputStream;

 

 /** 

  * A SoundProcessor that sends all sound data it receives to a wav file.

  * 

  * @author Robert McIntyre

  */

 

 public class WaveFileWriter implements SoundProcessor {

 

     public File targetFile;

     private WaveAudioOutputStream wao;

     private TDataOutputStream tos;

     private boolean initialized = false;

 	

     public WaveFileWriter(File targetFile) 

 	throws FileNotFoundException{

 	tos = new TNonSeekableDataOutputStream

 	    (new FileOutputStream(targetFile));

     }

 

     public void init(AudioFormat format){

 	wao = new WaveAudioOutputStream(format,AudioSystem.NOT_SPECIFIED, tos);

     }

 		

     public void process(ByteBuffer audioSamples, int numSamples, AudioFormat format) {

 	audioSamples.clear();

 	if (!initialized){

 	    init(format);

 	    initialized = true;

 	}

 		

 	byte[] data = new byte[numSamples];

 	audioSamples.get(data, 0, numSamples);

 	try {wao.write(data, 0, numSamples);}

 	catch (IOException e) {e.printStackTrace();}

     }

 

     public void cleanup() {

 	try {wao.close();}

 	catch (IOException e) {e.printStackTrace();}

     }

 

 

 }