org.openimaj.audio.analysis.benchmarking.dataset.MusicSpeechDataset Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) 2011, The University of Southampton and the individual contributors.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 *   * 	Redistributions of source code must retain the above copyright notice,
 * 	this list of conditions and the following disclaimer.
 *
 *   *	Redistributions in binary form must reproduce the above copyright notice,
 * 	this list of conditions and the following disclaimer in the documentation
 * 	and/or other materials provided with the distribution.
 *
 *   *	Neither the name of the University of Southampton nor the names of its
 * 	contributors may be used to endorse or promote products derived from this
 * 	software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package org.openimaj.audio.analysis.benchmarking.dataset;

import java.io.File;
import java.io.FileFilter;
import java.io.FilenameFilter;

import org.openimaj.citation.annotation.Reference;
import org.openimaj.citation.annotation.ReferenceType;
import org.openimaj.data.dataset.Dataset;
import org.openimaj.data.dataset.ListBackedDataset;
import org.openimaj.data.dataset.MapBackedDataset;
import org.openimaj.experiment.annotations.DatasetDescription;

import cern.colt.Arrays;

/**
 *	OpenIMAJ Dataset for the MusicSpeech Database
 *
 *	@author David Dupplaw ([email protected])
 *  @created 13 Mar 2013
 */
@DatasetDescription(
		name = "Music-Speech Dataset",
		description = "The 'music-speech' corpus is a small collection of some 240 " +
				"15-second extracts collected 'at random' from the radio by Eric Scheirer " +
				"during his internship at Interval Research Corporation in the summer of 1996 " +
				"under the supervision of Malcolm Slaney",
		url = "http://labrosa.ee.columbia.edu/sounds/musp/scheislan.html")
@Reference(
		type = ReferenceType.Inproceedings,
		author = { "Scheirer E.", "Slaney, M." },
		title = "Construction And Evaluation Of A Robust Multifeature Speech/music Discriminator",
		year = "1997",
		booktitle = "Proc. ICASSP-97, Munich.")
public class MusicSpeechDataset extends MapBackedDataset, File>
{
	/** Directory of the sounds */
	private File soundsDir;

	/**
	 * 	Create the Music-Speech Dataset from the given directory.
	 *
	 *	@param baseDir The base directory of the music-speech dataset.
	 * 	@param testOrTrain TRUE for testing set, FALSE for training set
	 */
	public MusicSpeechDataset( final File baseDir, final boolean testOrTrain )
	{
		// construct the directory of which set to get
		String soundsDir = "wavfile" + File.separator;
		soundsDir += testOrTrain? "test" : "train";

		this.processDir( this.soundsDir = new File( baseDir, soundsDir ) );

		System.out.println( this );
	}

	private void processDir( final File dir )
	{
		final File[] groups = dir.listFiles( new FileFilter()
		{
			@Override
			public boolean accept( final File pathname )
			{
				return 	pathname.isDirectory() &&
						!pathname.getName().equals( "." ) &&
						!pathname.getName().equals("..");
			}
		} );

		System.out.println( Arrays.toString( groups ));

		if( groups.length == 0 )
		{
			System.out.println( "Processing "+dir );

			final File[] files = dir.listFiles( new FilenameFilter()
			{
				@Override
				public boolean accept( final File dir, final String name )
				{
					return name.endsWith( ".wav" );
				}
			} );

			final ListBackedDataset list = new ListBackedDataset();
			this.map.put( dir.getAbsolutePath().substring( this.soundsDir.getAbsolutePath().length()+1 ), list );

			for( final File file : files )
			{
				list.add( file );
			}
		}
		else
		{
			for( final File group: groups )
				this.processDir( group );
		}
	}
}