marytts.modules.InternalModule 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!
/**
 * Copyright 2000-2006 DFKI GmbH.
 * All Rights Reserved.  Use is subject to license terms.
 *
 * This file is part of MARY TTS.
 *
 * MARY TTS is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, version 3 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 *
 */
package marytts.modules;

// Log4j Logging classes
import java.io.StringReader;
import java.util.Locale;

import javax.sound.sampled.AudioFileFormat;
import javax.sound.sampled.AudioSystem;

import marytts.datatypes.MaryData;
import marytts.datatypes.MaryDataType;
import marytts.modules.synthesis.Voice;
import marytts.util.MaryUtils;

import org.apache.log4j.Logger;

/**
 * A stub implementation of the MaryModule interface as a basis for internal modules.
 * 
 * Any internal module extending this class will need to implement a constructor calling this class's constructor, and override
 * process() in a meaningful way. Care must be taken to make sure the process() method is thread-seafe.
 * 

 * Example for a subclass:
 * 
 * 
 * public class Postlex extends InternalModule {
 * 	public Postlex() {
 * 		super("Postlex", MaryDataType.PHONEMISED, MaryDataType.POSTPROCESSED);
 * 	}
 * 
 * 	public MaryData process(MaryData d) throws Exception {
 * 		Document doc = d.getDocument();
 * 		mtuPostlex(doc);
 * 		phonologicalRules(doc);
 * 		MaryData result = new MaryData(outputType());
 * 		result.setDocument(doc);
 * 		return result;
 * 	}
 * 
 * 	private void mtuPostlex(Document doc) {...}
 * 
 * 	private void phonologicalRules(Document doc) {...}
 * }
 * 
 *
 * @author Marc Schröder
 */

public class InternalModule implements MaryModule {
	private String name = null;
	private MaryDataType inputType = null;
	private MaryDataType outputType = null;
	private Locale locale = null;
	protected int state;
	/**
	 * The logger instance to be used by this module. It will identify the origin of the log message in the log file.
	 */
	protected Logger logger;

	protected InternalModule(String name, MaryDataType inputType, MaryDataType outputType, Locale locale) {
		this.name = name;
		this.inputType = inputType;
		this.outputType = outputType;
		this.locale = locale;
		logger = MaryUtils.getLogger(name());
		this.state = MODULE_OFFLINE;
	}

	// Interface MaryModule implementation:
	public String name() {
		return name;
	}

	@Deprecated
	public MaryDataType inputType() {
		return getInputType();
	}

	public MaryDataType getInputType() {
		return inputType;
	}

	@Deprecated
	public MaryDataType outputType() {
		return getOutputType();
	}

	public MaryDataType getOutputType() {
		return outputType;
	}

	public Locale getLocale() {
		return locale;
	}

	public int getState() {
		return state;
	}

	public void startup() throws Exception {
		assert state == MODULE_OFFLINE;
		logger.info("Module started (" + inputType() + "->" + outputType() + ", locale " + getLocale() + ").");

		state = MODULE_RUNNING;
	}

	/**
	 * Perform a power-on self test by processing some example input data.
	 * 
	 * @throws Error
	 *             if the module does not work properly.
	 */
	public void powerOnSelfTest() throws Error {
		assert state == MODULE_RUNNING;
		logger.info("Starting power-on self test.");
		try {
			MaryData in = new MaryData(inputType, getLocale());
			String example = inputType.exampleText(getLocale());
			if (example != null) {
				in.readFrom(new StringReader(example));
				if (outputType.equals(MaryDataType.get("AUDIO")))
					in.setAudioFileFormat(new AudioFileFormat(AudioFileFormat.Type.WAVE, Voice.AF22050, AudioSystem.NOT_SPECIFIED));
				process(in);
			} else {
				logger.debug("No example text -- no power-on self test!");
			}
		} catch (Throwable t) {
			throw new Error("Module " + name + ": Power-on self test failed.", t);
		}
		logger.info("Power-on self test complete.");
	}

	public void shutdown() {
		logger = MaryUtils.getLogger(name());
		logger.info("Module shut down.");
		state = MODULE_OFFLINE;
	}

	/**
	 * Perform this module's processing on abstract "MaryData" input d. Subclasses need to make sure that the
	 * process() method is thread-safe, because in server-mode, it will be called from different threads at the same
	 * time. A sensible way to do this seems to be not to use any global or static variables, or to use them read-only.
	 * 
	 * 
	 * @return A MaryData object of type outputType() encapsulating the processing result.
	 *         
	 *         This method just returns its input. Subclasses should override this.
	 */
	public MaryData process(MaryData d) throws Exception {
		return d; // just return input.
	}

}