view src/samer/audio/StreamSource.java @ 1:5df24c91468d

Oh my what a mess.

/*
 *	StreamSource.java
 *
 *	Copyright (c) 2000, Samer Abdallah, King's College London.
 *	All rights reserved.
 *
 *	This software is provided AS iS and WITHOUT ANY WARRANTY;
 *	without even the implied warranty of MERCHANTABILITY or
 *	FITNESS FOR A PARTICULAR PURPOSE.
 */

package samer.audio;

import  samer.core.*;
import  samer.core.types.*;
import  samer.core.util.*;
import  samer.tools.*;
import  javax.sound.sampled.*;
import  javax.swing.*;
import  java.io.*;
import  java.net.URL;
import  java.util.*;

/**
	An AudioSource that read from an input stream. Can read any
	format for which the appropriate JavaSound plug-in is installed on
	your system, eg WAV, AU, MP3, OGG. The implementation of 
	AudioSource.reader() returns a Task which gets samples from the audio 
	files, going through the play list one by one. If the loop flag
	is true, then samples are returned indefinitely by looping through the
	playlist; otherwise, an EOFException is thrown if an attempt is made
	to read beyond the end of the last file.
	
	Properties read from current environment:
	<dl>
		<dt>current<dd>Current file (String)
		<dt>loop<dd>Loop playlist? (Boolean) [true]
	</dl>
					
*/

public class StreamSource implements AudioSource
{
	int				channelsToMix;
	AudioInputStream source;
	InputStream		in=null;
	AudioFormat		format=null;
	byte[]  			byte_buf=null;
	int				chunk=0;

	/**
	 * Construct a StreamSource initialised with current file and loop initialised
	 * from the current Environment. No exception is thrown if the current file
	 */
	public StreamSource(AudioInputStream s) throws Exception { source=s; }
	public StreamSource(InputStream s) throws Exception { source=AudioSystem.getAudioInputStream(s); }
	public StreamSource(File file) throws Exception { source=AudioSystem.getAudioInputStream(file); }
	public StreamSource(URL url) throws Exception { source=AudioSystem.getAudioInputStream(url); }

	public void dispose() { close(); }

	/** Returns current file */
	public void setTargetFormat(AudioFormat f) { format=f; }

	public AudioFormat getFormat() { return format; }

	public boolean isOpen() { return in!=null; }

	/** Closes current input stream */
	public synchronized void close() {
		try { 
			if (in!=null) {
				Shell.trace("Closing audio stream...");
				in.close(); in=null; byte_buf=null; 
			}
		}
		catch (IOException ex) {}
	}

	/** Opens the playlist starting with the first file.  */
	public synchronized void open() throws Exception { if (!isOpen()) openCurrent(); }

	/** Opens the current file. Next read will returns samples
	 *  from head of given file. If setFormat() was called previously, then
	 *  we will attempt to do format conversion. */
	private synchronized void openCurrent() throws Exception
	{
		AudioInputStream     s=source;
		AudioFormat				fmt1, af=s.getFormat();

		Shell.trace("  format: "+af);

		// convert to target format if required
		if (format!=null) {
			if (!format.equals(af)) {
				Shell.trace("  converting to "+format);
				if (af.getChannels()>format.getChannels()) {
					Shell.trace("  channels mix down required.");
					fmt1 = new AudioFormat( format.getEncoding(), format.getSampleRate(), 
								format.getSampleSizeInBits(), 
								af.getChannels(), format.getFrameSize(), format.getFrameRate(), format.isBigEndian());

					channelsToMix = af.getChannels();
				} else {
					channelsToMix = 0;
					fmt1=format;
				}
				s=convertFormat(s,af,fmt1);
			}
		}

		// If we have a reader task, then update the byte buffer
		if (chunk>0) setByteBuffer();

		// Shell.trace("stream format: "+s.getFormat());
		in = new BufferedInputStream(s,64*1024);
	}

	/** Returns number of bytes available in current file */
	public int available() throws Exception { return in.available(); }

	private void setChunkSize(int s) { chunk=s; }
	private void setByteBuffer() {
		byte_buf = new byte[2*chunk*(channelsToMix>0 ? channelsToMix : 1)];
	}

	/** Returns a Task which copies samples as doubles into the given
	 *  buffer between the given positions. */
	public Task reader(final double [] dbuf, final int off, final int len) {
		setChunkSize(len);
		if (in!=null) setByteBuffer();

		return new AnonymousTask() {
			public void run() throws Exception {
				int n = 0;

				synchronized (StreamSource.this) {
					int blen = byte_buf.length;
					while (n < blen) {
						int count = in.read(byte_buf, n, blen - n);
						if (count > 0) n+=count;
						else throw new EOFException();
					}
				}
				if (channelsToMix>0) Util.shortToDoubleMixDown(byte_buf,dbuf,off,len,channelsToMix);
				else Util.shortToDouble(byte_buf,dbuf,off,len);
			}
		};
	}

	/** Returns a Task which copies samples as floats into the given
	 *  buffer between the given positions. */
	public Task reader(final float [] dbuf, final int off, final int len) {
		setChunkSize(len);
		if (in!=null) setByteBuffer();

		return new AnonymousTask() {
			public synchronized void run() throws Exception {
				int n = 0;

				synchronized (StreamSource.this) {
					int blen = byte_buf.length;
					while (n < blen) {
						int count = in.read(byte_buf, n, blen - n);
						if (count > 0) n+=count;
						else throw new EOFException();
					}
				}
				if (channelsToMix>0) Util.shortToFloatMixDown(byte_buf,dbuf,off,len,channelsToMix);
				else Util.shortToFloat(byte_buf,dbuf,off,len);
			}
		};
	}

	private static AudioInputStream convertFormat(AudioInputStream sin, AudioFormat fin, AudioFormat fout) throws Exception
	{
		if (fin.equals(fout)) return sin;
		else if (fin.getEncoding()==AudioFormat.Encoding.PCM_SIGNED) {
			try { return AudioSystem.getAudioInputStream(fout,sin); }
			catch (IllegalArgumentException ex) { Shell.trace("Direct conversion failed"); }

			AudioFormat fint = new AudioFormat(  // PCM
				fout.getSampleRate(), fout.getSampleSizeInBits(),
				fin.getChannels(), true, fout.isBigEndian());
			Shell.trace("Trying PCM conversion via "+fint.toString());
			return AudioSystem.getAudioInputStream(fout,AudioSystem.getAudioInputStream(fint,sin));
		} else {
		// First, check for MP3 - if so, cannot convert number of channels
			if (fin.getChannels()==fout.getChannels() && fin.getSampleRate()==fout.getSampleRate()) {
				Shell.trace("Trying direct decoding from "+fin.getEncoding().toString());
				return AudioSystem.getAudioInputStream(fout,sin);
			} else {
				AudioFormat fint = new AudioFormat( 
					fin.getSampleRate(), fout.getSampleSizeInBits(),
					fin.getChannels(), true, fout.isBigEndian());
				Shell.trace("Trying conversion via "+fint.toString());
				return convertFormat(AudioSystem.getAudioInputStream(fint,sin),fint,fout);
			}
		}
	}
}