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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of de.unistuttgart.ims.drama.main Show documentation

The newest version!

package de.unistuttgart.ims.drama.main.annotation;

import org.apache.uima.analysis_engine.AnalysisEngineProcessException;
import org.apache.uima.fit.component.JCasAnnotator_ImplBase;
import org.apache.uima.fit.descriptor.ConfigurationParameter;
import org.apache.uima.fit.factory.AnnotationFactory;
import org.apache.uima.fit.util.JCasUtil;
import org.apache.uima.jcas.JCas;

import de.tudarmstadt.ukp.dkpro.core.api.lexmorph.type.pos.POS;
import de.tudarmstadt.ukp.dkpro.core.api.segmentation.type.Token;
import de.unistuttgart.ims.drama.api.Scene;
import de.unistuttgart.ims.drama.util.DramaUtil;

public class PreparePosAnnotation extends JCasAnnotator_ImplBase {

	public static final String PARAM_SCENE_BEGIN = "Scene Begin";
	public static final String PARAM_SCENE_END = "Scene End";

	@ConfigurationParameter(name = PARAM_SCENE_BEGIN, defaultValue = "8")
	int sceneBegin = 8;
	@ConfigurationParameter(name = PARAM_SCENE_END, defaultValue = "11")
	int sceneEnd = 11;

	@Override
	public void process(JCas aJCas) throws AnalysisEngineProcessException {
		String did = DramaUtil.getDrama(aJCas).getDocumentId();
		// if (did.equalsIgnoreCase("rksp.0"))
		prepareRksp0(aJCas);
	}

	public void prepareRksp0(JCas jcas) {

		int sceneNum = 1;
		for (Scene scene : JCasUtil.select(jcas, Scene.class)) {
			if (sceneNum >= sceneBegin && sceneNum < sceneEnd)
				// if (scene.getBegin() == 36351 || scene.getBegin() == 39097 ||
				// scene.getBegin() == 39975) {
				for (Token token : JCasUtil.selectCovered(Token.class, scene)) {
					AnnotationFactory.createAnnotation(jcas, token.getBegin(), token.getEnd(), POS.class);
				}

			sceneNum++;
		}
	}

}