DeLex.java

/**
 * Copyright 2005 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 <http://www.gnu.org/licenses/>.
 *
 */

package marytts.language.de.phonemiser;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Hashtable;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

public class DeLex {

	private String modelPath = "DE-LEX.xml";

	protected Hashtable modelTable;

	public DeLex(String knowledgeBasePath) {

		this.modelTable = new Hashtable(40);
		this.modelPath = knowledgeBasePath + modelPath;
		loadInputModel();

	}

	public void loadInputModel() {

		try {

			// System.out.println("############################################################");
			// System.out.println("Processing File : '"+modelPath+"'");

			DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
			DocumentBuilder parser = factory.newDocumentBuilder();
			Document doc = parser.parse(modelPath);
			Element rootNode = doc.getDocumentElement();

			if (rootNode.getNodeName().equals("entries")) {
				NodeList entriesList = rootNode.getChildNodes();

				for (int i = 0; i < entriesList.getLength(); i++) {
					if (entriesList.item(i).getNodeType() == org.w3c.dom.Node.ELEMENT_NODE) {
						Element entry = (Element) entriesList.item(i);

						NodeList keyList = entry.getElementsByTagName("key");

						NodeList valueList = entry.getElementsByTagName("value");

						if (keyList.getLength() == 1 && valueList.getLength() == 1) {
							NodeList keyTextNodeList = ((Element) keyList.item(0)).getChildNodes();
							NodeList valueTextNodeList = ((Element) valueList.item(0)).getChildNodes();

							String keyString = "";
							String valueString = "";

							for (int a = 0; a < keyTextNodeList.getLength(); a++) {
								if (keyTextNodeList.item(a).getNodeType() == org.w3c.dom.Node.TEXT_NODE) {
									org.w3c.dom.Text textNode = (org.w3c.dom.Text) keyTextNodeList.item(a);
									keyString += textNode.getData().trim();
								}
							}

							for (int a = 0; a < valueTextNodeList.getLength(); a++) {
								if (valueTextNodeList.item(a).getNodeType() == org.w3c.dom.Node.TEXT_NODE) {
									org.w3c.dom.Text textNode = (org.w3c.dom.Text) valueTextNodeList.item(a);
									valueString += textNode.getData().trim();

									// Element valueDOM = (Element) valueElementNodeList.item(a);
									// valueString = document2String(valueDOM);
								}
							}
							// System.out.println("<key> : "+keyString);
							// System.out.println("<value> : "+valueString);
							this.modelTable.put(keyString, valueString);

						} else {
							System.out.println("ERROR: The " + i + "th entry contains wrong number of subElements !!");
						}
					}
				}

			} else
				System.out.println("Document is not wellformed. Top-level element should be <entries>.");

		} catch (IOException e) {
			System.out.println("Can't find file '" + modelPath + "'.");
		} catch (ParserConfigurationException e) {
			System.out.println("Parser Configuration exception (Use trace mode for details)");
		} catch (SAXException e) {
			System.out.println("Error in XML file:\n" + e.getMessage());
		} catch (Exception e) {
			System.out.println("Unknown Error\n" + e.getMessage());
			e.printStackTrace();
		}

	}

	/**
	 * Prints a textual representation of a DOM object into a text string..
	 * 
	 * @param element
	 *            DOM object to parse.
	 * @return String representation of <i>document</i>.
	 */
	public static String document2String(Element element) {
		String result = null;

		StringWriter strWtr = new StringWriter();
		StreamResult strResult = new StreamResult(strWtr);
		TransformerFactory tfac = TransformerFactory.newInstance();
		try {
			Transformer trans = tfac.newTransformer();
			trans.setOutputProperty("omit-xml-declaration", "yes");

			trans.transform(new DOMSource(element), strResult);
		} catch (Exception e) {
			System.err.println("XMLUtils.document2String(): " + e);
		}
		result = strResult.getWriter().toString();

		return result;
	}// document2String()

}