comparison src/samer/audio/MultiFileAudioStream.java @ 1:5df24c91468d

Oh my what a mess.

+/*
+ *	FileSource.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.util.*;
+import  java.nio.ByteBuffer;
+
+/**
+*/
+
+
+public abstract class MultiFileAudioStream extends InputStream 
+{
+	List<File>				list=null;
+	ListIterator<File>	it=null;
+	int						channelsToMix;
+	File						curFile=null;
+	InputStream				in=null;
+	AudioFormat				format=null, inFormat=null;
+	boolean              loop=false;
+
+	/**
+	 * Construct a FileSource initialised with current file and loop initialised
+	 * from the current Environment. No exception is thrown if the current file
+	 * cannot be opened, however, you must be sure to set it to a valid file
+	 * (or to set a playlist) before trying to read any samples.
+	 */
+	public MultiFileAudioStream(AudioFormat target, List<File> files)
+	{
+		list=files;
+		format=target;
+	}
+
+	public AudioFormat getFormat() { return format; }
+
+	/** Set playlist to all WAV files in given directory */
+	public List<File> directory(File dir, String ext) {
+		return Arrays.asList(dir.listFiles(getFileFilter(ext))); 
+	}
+
+	private static java.io.FileFilter getFileFilter(final String ext) { 
+		return new java.io.FileFilter() {
+			public boolean accept(File file) {
+				return file.getName().toLowerCase().endsWith(ext);
+			}
+		};
+	}
+
+	public synchronized List<File> getPlaylist() { return list; }
+
+	/** Go back to start of playlist. Next block of samples will be from head
+	 *  of first file in playlist. Will throw an exception if there is no playlist */
+	public synchronized void rewind() throws IOException {
+		it=list.listIterator(); next();
+	}
+
+	/** Move to head of next file in playlist */
+	public synchronized void next() throws IOException
+	{
+		boolean wasOpen=isOpen();
+
+		if (wasOpen) close();
+		if (!it.hasNext()) {
+			if (!loop) throw new EOFException();
+			it=list.listIterator();
+			if (!it.hasNext()) throw new IOException("no files in playlist");
+		}
+		curFile=it.next();
+		if (wasOpen) openCurrent();
+	}
+
+	/** Move to head of previous file in playlist */
+	public synchronized void prev() throws Exception
+	{
+		boolean wasOpen=isOpen();
+		if (wasOpen) close();
+		if (!it.hasPrevious()) {
+			if (!loop) throw new EOFException();
+			it=list.listIterator(list.size());
+			if (!it.hasPrevious()) throw new Exception("no files in playlist");
+		}
+		curFile=it.previous();
+		if (wasOpen) openCurrent();
+	}
+
+	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; 
+			}
+		}
+		catch (IOException ex) {}
+	}
+
+	/** Opens the playlist starting with the first file.  */
+	public synchronized void open() throws Exception { rewind(); 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 IOException
+	{
+		try {
+			File file=curFile;
+			Shell.trace("\nFileSource:Opening audio file "+file);
+
+			AudioInputStream		s=AudioSystem.getAudioInputStream(file);
+			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()) {
+						int frameSize = af.getChannels()*format.getSampleSizeInBits()/8;
+						Shell.trace("  channels mix down required.");
+						fmt1 = new AudioFormat( format.getEncoding(), format.getSampleRate(), 
+									format.getSampleSizeInBits(), 
+									af.getChannels(), frameSize, format.getFrameRate(), format.isBigEndian());
+
+						channelsToMix = af.getChannels();
+					} else {
+						channelsToMix = 0;
+						fmt1=format;
+					}
+					Shell.trace("  converting via "+fmt1);
+					s=convertFormat(s,af,fmt1);
+					inFormat = fmt1;
+				} else {
+					Shell.trace("  no formation conversion required");
+					channelsToMix = 0;
+					inFormat = af;
+				}
+			} else {
+				Shell.trace("  using stream native format");
+				channelsToMix = 0;
+				inFormat = af;
+			}
+			in=s;
+		}
+		catch (Exception ex) { throw new IOException("Failed to open audio file"); }
+	}
+
+	/** Returns number of bytes available in current file */
+	public int available() { 
+		try {
+			return in.available(); 
+		}
+		catch (Exception ex) { return 0; }
+	}
+
+	/** Returns a Task which copies samples as doubles into the given
+	 *  buffer between the given positions. */
+	public int read(byte [] buf, int off, int len) throws IOException {
+
+		// loop until len samples copied into dbuf
+		int rem=len, pos=off;
+		while (rem>0) {
+			int chunk = in.read(buf, pos, rem);
+			if (chunk > 0) { // append this chunk to output
+				pos+=chunk; rem-=chunk;
+			} else if (it!=null) next(); // next file if there is one
+			else if (!loop) throw new EOFException(); // not looping and no more files
+			else rewind(); // back to first file
+		}
+		return len;
+	}
+
+
+	private static AudioInputStream convertFormat(AudioInputStream sin, AudioFormat fin, AudioFormat fout) throws Exception
+	{
+		Shell.trace("\nconvertFormat:");
+		Shell.trace("  | source: "+fin.toString());
+		Shell.trace("  | target: "+fout.toString());
+
+		if (fin.equals(fout)) return sin;
+		else if (fin.getEncoding()==AudioFormat.Encoding.PCM_SIGNED) {
+			try { 
+				Shell.trace("  | Trying direct (PCM) from "+fin.getEncoding().toString());
+				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 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 recursive via "+fint.toString());
+				return convertFormat(AudioSystem.getAudioInputStream(fint,sin),fint,fout);
+			}
+		}
+	}
+}
+