eu.crydee.alignment.aligner.MetricsP.java Source code

Java tutorial

Introduction

Here is the source code for eu.crydee.alignment.aligner.MetricsP.java

Source

/*
 * Copyright 2014 Hugo m09? Mougard.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package eu.crydee.alignment.aligner;

import eu.crydee.alignment.aligner.ae.CosineSimilarityAE;
import eu.crydee.alignment.aligner.ae.IsfAE;
import eu.crydee.alignment.aligner.ae.MetricsSummaryAE;
import eu.crydee.alignment.aligner.ae.TAFC;
import eu.crydee.alignment.aligner.ae.StopWordsAE;
import eu.crydee.alignment.aligner.ae.XmiSerializerC;
import eu.crydee.alignment.aligner.cr.MetricsCR;
import eu.crydee.alignment.metricslab.model.Complete;
import java.io.IOException;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.commons.cli.PosixParser;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.uima.UIMAException;
import org.apache.uima.analysis_engine.AnalysisEngineDescription;
import org.apache.uima.cas.CAS;
import org.apache.uima.collection.CollectionReader;
import org.apache.uima.fit.factory.AggregateBuilder;
import static org.apache.uima.fit.factory.CollectionReaderFactory.createReader;
import static org.apache.uima.fit.factory.AnalysisEngineFactory.createEngineDescription;
import org.apache.uima.fit.pipeline.SimplePipeline;
import org.apache.uima.resource.ResourceInitializationException;

/**
 *
 * @author Hugo m09? Mougard
 */
public class MetricsP {

    private static final Logger logger = LogManager.getLogger(MetricsP.class);

    private static String WHITELIST_FILE_PATH, CORPUS_PATH;

    public static void main(String[] args) throws ResourceInitializationException, UIMAException, IOException {
        parseArguments(args);
        CollectionReader cr = createReader(MetricsCR.class, MetricsCR.PARAM_WHITELIST_FILE_PATH,
                WHITELIST_FILE_PATH, MetricsCR.PARAM_BRITANNICA_CORPUS_PATH, CORPUS_PATH,
                MetricsCR.PARAM_VIEW_NAME_ELEMENTARY, Config.elementaryView, MetricsCR.PARAM_VIEW_NAME_NORMAL,
                Config.regularView);
        AnalysisEngineDescription stopWords = createEngineDescription(StopWordsAE.class);
        AnalysisEngineDescription isf = createEngineDescription(IsfAE.class, IsfAE.PARAM_VIEW_ELEMENTARY,
                Config.elementaryView, IsfAE.PARAM_VIEW_REGULAR, Config.regularView);
        AnalysisEngineDescription cosineSimilarity = createEngineDescription(CosineSimilarityAE.class,
                CosineSimilarityAE.PARAM_VIEW_LEFT, Config.elementaryView, CosineSimilarityAE.PARAM_VIEW_RIGHT,
                Config.regularView);
        AnalysisEngineDescription casWriter = createEngineDescription(XmiSerializerC.class,
                XmiSerializerC.PARAM_OUT_FOLDER, Config.metricsCas);
        AnalysisEngineDescription tafWriter = createEngineDescription(TAFC.class, TAFC.PARAM_OUTPUT_FOLDER,
                Config.metricsTaf, TAFC.PARAM_VIEW_LEFT, Config.elementaryView, TAFC.PARAM_VIEW_RIGHT,
                Config.regularView);
        AnalysisEngineDescription metricsSummarizer = createEngineDescription(MetricsSummaryAE.class,
                MetricsSummaryAE.PARAM_TAF_FOLDER, Config.metricsTaf, MetricsSummaryAE.PARAM_HTML_OUTPUT_FILEPATH,
                Config.metricsHtml, MetricsSummaryAE.PARAM_METRICS_TO_SUMMARIZE,
                new String[] { Complete.LEFT_ORDER_CONSERVATION_KEY, Complete.RIGHT_ORDER_CONSERVATION_KEY,
                        Complete.SYMMETRY, Complete.LEFT_COOCCURRENCE, Complete.RIGHT_COOCCURRENCE,
                        Complete.LEFT_STRICT_ORDER_CONSERVATION_KEY, Complete.RIGHT_STRICT_ORDER_CONSERVATION_KEY,
                        Complete.LEFT_ORDER_CONSERVATION_AT_R1_KEY, Complete.RIGHT_ORDER_CONSERVATION_AT_R1_KEY });
        AggregateBuilder b = new AggregateBuilder();
        b.add(stopWords, CAS.NAME_DEFAULT_SOFA, Config.elementaryView);
        b.add(stopWords, CAS.NAME_DEFAULT_SOFA, Config.regularView);
        b.add(isf);
        b.add(cosineSimilarity);
        b.add(casWriter);
        b.add(tafWriter);
        b.add(metricsSummarizer);

        SimplePipeline.runPipeline(cr, b.createAggregate());
    }

    static private void parseArguments(String[] args) {
        Options shortCircuitOptions = new Options();
        shortCircuitOptions
                .addOption(OptionBuilder.withLongOpt("help").withDescription("Print this message.").create('h'));
        shortCircuitOptions
                .addOption(OptionBuilder.withLongOpt("version").withDescription("Print the version.").create('v'));
        Options options = new Options();
        options.addOption(OptionBuilder.isRequired().withLongOpt("corpus").hasArg().withArgName("folder-path")
                .withDescription("Path to the britannica corpus.").create('c'));
        options.addOption(OptionBuilder.isRequired().withLongOpt("whitelist-path").hasArg().withArgName("file-path")
                .withDescription("Path to the whitelist of files to test. One " + "city per line.").create('w'));
        try {
            CommandLineParser parser = new PosixParser();
            CommandLine cmd = parser.parse(shortCircuitOptions, args, true);
            if (cmd.hasOption('h')) {
                HelpFormatter formatter = new HelpFormatter();
                formatter.printHelp("aligner", options, true);
                System.exit(0);
            }
            if (cmd.hasOption('v')) {
                System.out.println("aligner v1.0.0-SNAPSHOT");
                System.exit(0);
            }
            cmd = parser.parse(options, args);
            WHITELIST_FILE_PATH = cmd.getOptionValue('w');
            CORPUS_PATH = cmd.getOptionValue('c');
        } catch (ParseException ex) {
            System.err.println("The CLI args could not be parsed.");
            System.err.println("The error message was:");
            System.err.println(" " + ex.getMessage());
            System.exit(1);
        }
    }
}