hu.ppke.itk.nlpg.purepos.cli.configuration.ConfigurationReader.java Source code

Introduction

Here is the source code for hu.ppke.itk.nlpg.purepos.cli.configuration.ConfigurationReader.java

Source

/*******************************************************************************
 * Copyright (c) 2012 Gyrgy Orosz, Attila Novk.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v3
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/
 * 
 * This file is part of PurePos.
 * 
 * PurePos is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * PurePos 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 Public License for more details.
 * 
 * Contributors:
 *     Gyrgy Orosz - initial API and implementation
 ******************************************************************************/
package hu.ppke.itk.nlpg.purepos.cli.configuration;

import hu.ppke.itk.nlpg.purepos.model.internal.StringMapping;

import java.io.File;
import java.util.LinkedList;
import java.util.List;

import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.XMLConfiguration;
import org.apache.commons.configuration.tree.ConfigurationNode;

public class ConfigurationReader {
    private static final String TAG = "to";
    private static final String PATTERN = "pattern";
    private static final String TAG_MAPPING = "tag_mapping";
    private static final String LEMMA_MAPPING = "lemma_mapping";
    private static final String GUESSED_MARKER = "guessed_marker";
    private static final String SUFFIX_MODEL_PARAMETERS = "suffix_model_weight";

    public Configuration read(File f) throws ConfigurationException {
        XMLConfiguration xconf = new XMLConfiguration(f);

        List<ConfigurationNode> tagMappings = xconf.getRootNode().getChildren(TAG_MAPPING);
        LinkedList<StringMapping> tag_ret = new LinkedList<StringMapping>();
        for (ConfigurationNode m : tagMappings) {
            String spat = (String) m.getAttributes(PATTERN).get(0).getValue();
            String stag = (String) m.getAttributes(TAG).get(0).getValue();
            tag_ret.add(new StringMapping(spat, stag));
        }

        List<ConfigurationNode> lemmaMappings = xconf.getRootNode().getChildren(LEMMA_MAPPING);
        LinkedList<StringMapping> lemma_ret = new LinkedList<StringMapping>();
        for (ConfigurationNode m : lemmaMappings) {
            String spat = (String) m.getAttributes(PATTERN).get(0).getValue();
            String stag = (String) m.getAttributes(TAG).get(0).getValue();
            lemma_ret.add(new StringMapping(spat, stag));
        }

        List<ConfigurationNode> markers = xconf.getRootNode().getChildren(GUESSED_MARKER);

        String guessedMarker = "";
        if (markers.size() > 0) {
            guessedMarker = (String) markers.get(0).getValue();
        }

        List<ConfigurationNode> params = xconf.getRootNode().getChildren(SUFFIX_MODEL_PARAMETERS);

        Double weight = null;
        if (params.size() > 0) {
            weight = Double.valueOf((String) params.get(0).getValue());
        }

        return new Configuration(tag_ret, lemma_ret, guessedMarker, weight);
    }
}