de.unistuttgart.ims.drama.main.annotation.Export Maven / Gradle / Ivy

Go to download
package de.unistuttgart.ims.drama.main.annotation;

import java.io.IOException;

import org.apache.uima.UIMAException;
import org.apache.uima.collection.CollectionReaderDescription;
import org.apache.uima.fit.factory.AggregateBuilder;
import org.apache.uima.fit.factory.AnalysisEngineFactory;
import org.apache.uima.fit.factory.CollectionReaderFactory;
import org.apache.uima.fit.pipeline.SimplePipeline;
import org.apache.uima.resource.ResourceInitializationException;

import com.lexicalscope.jewel.cli.CliFactory;

import de.tudarmstadt.ukp.dkpro.core.io.xmi.XmiWriter;
import de.tudarmstadt.ukp.dkpro.core.tokit.BreakIteratorSegmenter;
import de.unistuttgart.ims.drama.main.Options;
import de.unistuttgart.quadrama.core.convert.QdCoreference2DkproCoreference;
import de.unistuttgart.quadrama.io.tei.GerDraCorReader;

public class Export {

	public static void main(String[] args) throws ResourceInitializationException, UIMAException, IOException {

		// runMain("AnnikaHaag",
		// new String[] { "--input",
		// "/Users/reiterns/Documents/QuaDramA/Coreference
		// Annotations/annotations_last_tei/AH.xml",
		// "--output", "/Users/reiterns/Documents/QuaDramA/Coreference
		// Annotations/WebAnno-Start/AH" });
		runMain("AnjaSchmelzle",
				new String[] { "--input",
						"/Users/reiterns/Documents/QuaDramA/Coreference Annotations/annotations_last_tei/AS.xml",
						"--output", "/Users/reiterns/Documents/QuaDramA/Coreference Annotations/WebAnno-S2/AS" });
		runMain("SonjaEberhardt",
				new String[] { "--input",
						"/Users/reiterns/Documents/QuaDramA/Coreference Annotations/annotations_last_tei/SE.xml",
						"--output", "/Users/reiterns/Documents/QuaDramA/Coreference Annotations/WebAnno-S2/SE" });
		runMain("rjmw.0",
				new String[] { "--input", "/Users/reiterns/Documents/QuaDramA/gerdracor/data-tgids/rjmw.0.xml",
						"--output", "/Users/reiterns/Documents/QuaDramA/Coreference Annotations/WebAnno-S2" });

	}

	public static void runMain(String documentId, String[] args) throws UIMAException, IOException {
		Options options = CliFactory.parseArguments(Options.class, args);

		CollectionReaderDescription reader = CollectionReaderFactory.createReaderDescription(GerDraCorReader.class,
				GerDraCorReader.PARAM_INPUT, options.getInput(), GerDraCorReader.PARAM_TEI_COMPAT, true,
				GerDraCorReader.PARAM_REMOVE_XML_ANNOTATIONS, true);

		AggregateBuilder builder = new AggregateBuilder();
		builder.add(AnalysisEngineFactory.createEngineDescription(BreakIteratorSegmenter.class,
				BreakIteratorSegmenter.PARAM_WRITE_SENTENCE, false));
		builder.add(AnalysisEngineFactory.createEngineDescription(QdCoreference2DkproCoreference.class,
				QdCoreference2DkproCoreference.PARAM_INCLUDE_SPEAKERS, true));
		builder.add(AnalysisEngineFactory.createEngineDescription(XmiWriter.class, XmiWriter.PARAM_TARGET_LOCATION,
				options.getOutput()));

		SimplePipeline.runPipeline(reader, builder.createAggregateDescription());
	}

}