marytts.vocalizations.VocalizationIntonationReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of marytts-runtime Show documentation
The newest version!
/**
 * Portions Copyright 2006 DFKI GmbH.
 * Portions Copyright 2001 Sun Microsystems, Inc.
 * Portions Copyright 1999-2001 Language Technologies Institute, 
 * Carnegie Mellon University.
 * All Rights Reserved.  Use is subject to license terms.
 * 
 * Permission is hereby granted, free of charge, to use and distribute
 * this software and its documentation without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of this work, and to
 * permit persons to whom this work is furnished to do so, subject to
 * the following conditions:
 * 
 * 1. The code must retain the above copyright notice, this list of
 *    conditions and the following disclaimer.
 * 2. Any modifications must be clearly marked as such.
 * 3. Original authors' names are not deleted.
 * 4. The authors' names are not used to endorse or promote products
 *    derived from this software without specific prior written
 *    permission.
 *
 * DFKI GMBH AND THE CONTRIBUTORS TO THIS WORK DISCLAIM ALL WARRANTIES WITH
 * REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL DFKI GMBH NOR THE
 * CONTRIBUTORS BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF
 * THIS SOFTWARE.
 */
package marytts.vocalizations;

import java.io.BufferedInputStream;
import java.io.DataInputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Arrays;

import marytts.exceptions.MaryConfigurationException;
import marytts.util.data.MaryHeader;

/**
 * Vocalization contours and their corresponding polynomial coeffs reader from a intonation timeline reader file
 * 
 * @author sathish pammi
 * 
 */
public class VocalizationIntonationReader {

	private MaryHeader hdr = null;
	private int numberOfUnits = 0;
	private double windowSize = 0.0;
	private double skipSize = 0.0;
	private double[][] contours;
	private double[][] coeffs;
	private boolean[][] voiced;

	/****************/
	/* CONSTRUCTORS */
	/****************/

	/**
	 * Create a unit file reader from the given unit file
	 * 
	 * @param fileName
	 *            the unit file to read
	 * @throws IOException
	 *             if a problem occurs while reading
	 * @throws MaryConfigurationException
	 *             MaryConfigurationException
	 */
	public VocalizationIntonationReader(String fileName) throws IOException, MaryConfigurationException {
		load(fileName);
	}

	/**
	 * Load the given unit file
	 * 
	 * @param fileName
	 *            the unit file to read
	 * @throws IOException
	 *             if a problem occurs while reading
	 * @throws MaryConfigurationException
	 *             MaryConfigurationException
	 */
	public void load(String fileName) throws IOException, MaryConfigurationException {
		/* Open the file */
		DataInputStream dis = null;

		try {
			dis = new DataInputStream(new BufferedInputStream(new FileInputStream(fileName)));
		} catch (FileNotFoundException e) {
			throw new RuntimeException("File [" + fileName + "] was not found.");
		}

		try {
			/* Load the Mary header */
			hdr = new MaryHeader(dis);
			if (hdr.getType() != MaryHeader.LISTENERFEATS) {
				throw new RuntimeException("File [" + fileName + "] is not a valid Mary Units file.");
			}

			windowSize = dis.readFloat();

			if (this.windowSize <= 0) {
				throw new RuntimeException("File [" + fileName + "] has a negative number for window size. Aborting.");
			}

			skipSize = dis.readFloat();
			if (this.skipSize <= 0) {
				throw new RuntimeException("File [" + fileName + "] has a negative number for window size. Aborting.");
			}

			/* Read the number of units */
			numberOfUnits = dis.readInt();
			if (numberOfUnits < 0) {
				throw new RuntimeException("File [" + fileName + "] has a negative number of units. Aborting.");
			}

			contours = new double[numberOfUnits][];
			coeffs = new double[numberOfUnits][];
			voiced = new boolean[numberOfUnits][];

			for (int i = 0; i < numberOfUnits; i++) {

				// read polynomial coeffs
				int polyOrder = dis.readInt();
				coeffs[i] = new double[polyOrder];
				for (int j = 0; j < polyOrder; j++) {
					coeffs[i][j] = dis.readFloat();
				}

				// read f0 contour
				int f0ContourLength = dis.readInt();
				contours[i] = new double[f0ContourLength];

				voiced[i] = new boolean[f0ContourLength];
				Arrays.fill(voiced[i], false);
				for (int j = 0; j < f0ContourLength; j++) {
					contours[i][j] = dis.readFloat();
					if (contours[i][j] > 0) {
						voiced[i][j] = true;
					}
				}
			}
		} catch (IOException e) {
			throw new RuntimeException("Reading the Mary header from file [" + fileName + "] failed.", e);
		}

	}

	/*****************/
	/* OTHER METHODS */
	/*****************/

	/**
	 * Get the number of units in the file.
	 * 
	 * @return The number of units.
	 */
	public int getNumberOfUnits() {
		return (numberOfUnits);
	}

	/**
	 * get an intonation contour
	 * 
	 * @param unitIndexNumber
	 *            unitIndexNumber
	 * @return self.contours[unitIndexNumber]
	 */
	public double[] getContour(int unitIndexNumber) {
		return this.contours[unitIndexNumber];
	}

	public boolean[] getVoicings(int unitIndexNumber) {
		return this.voiced[unitIndexNumber];
	}

	public double getWindowSizeInSeconds() {
		return this.windowSize;
	}

	public double getSkipSizeInSeconds() {
		return this.skipSize;
	}

	/**
	 * get an intonation polynomial coeffs
	 * 
	 * @param unitIndexNumber
	 *            unitIndexNumber
	 * @return self.coeffs[unitIndexNumber]
	 */
	public double[] getIntonationCoeffs(int unitIndexNumber) {
		return this.coeffs[unitIndexNumber];
	}

	public static void main(String[] args) throws Exception {
		String fileName = "/home/sathish/Work/phd/voices/listener/vocalizations/timelines/vocalization_intonation_timeline.mry";
		VocalizationIntonationReader bcUfr = new VocalizationIntonationReader(fileName);
		// bcUfr.load(fileName);
	}
}