svn commit: r1442092 - /incubator/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/EventTimeFlatTreeFeatureExtractor.java

[tm...@apache.org]

Author: tmill
Date: Mon Feb  4 12:33:06 2013
New Revision: 1442092

URL: http://svn.apache.org/viewvc?rev=1442092&view=rev
Log:
Added feature for creating flat tree structure a la Hovy et al 2012. Untested.

Added:
    incubator/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/EventTimeFlatTreeFeatureExtractor.java

Added: incubator/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/EventTimeFlatTreeFeatureExtractor.java
URL: http://svn.apache.org/viewvc/incubator/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/EventTimeFlatTreeFeatureExtractor.java?rev=1442092&view=auto
==============================================================================
--- incubator/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/EventTimeFlatTreeFeatureExtractor.java (added)
+++ incubator/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/EventTimeFlatTreeFeatureExtractor.java Mon Feb  4 12:33:06 2013
@@ -0,0 +1,66 @@
+package org.apache.ctakes.temporal.ae.feature;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.apache.ctakes.relationextractor.ae.features.RelationFeaturesExtractor;
+import org.apache.ctakes.typesystem.type.syntax.BaseToken;
+import org.apache.ctakes.typesystem.type.textsem.EventMention;
+import org.apache.ctakes.typesystem.type.textsem.IdentifiedAnnotation;
+import org.apache.ctakes.utils.tree.SimpleTree;
+import org.apache.uima.analysis_engine.AnalysisEngineProcessException;
+import org.apache.uima.jcas.JCas;
+import org.cleartk.classifier.Feature;
+import org.uimafit.util.JCasUtil;
+
+
+public class EventTimeFlatTreeFeatureExtractor implements RelationFeaturesExtractor{
+
+	@Override
+	public List<Feature> extract(JCas jcas, IdentifiedAnnotation arg1,
+			IdentifiedAnnotation arg2) throws AnalysisEngineProcessException {
+		List<Feature> feats = new ArrayList<Feature>();
+		
+		// make sure they're in the correct order
+		if(arg1.getBegin() > arg2.getBegin()){
+			IdentifiedAnnotation temp = arg1;
+			arg1 = arg2;
+			arg2 = temp;
+		}
+		
+		SimpleTree tree = new SimpleTree("BOP");
+		
+		SimpleTree arg1Tree = null;
+		SimpleTree arg2Tree = null;
+		
+		// make sure we have the right classes
+		if(arg1 instanceof EventMention){
+			arg1Tree = new SimpleTree("EVENT");
+			arg2Tree = new SimpleTree("TIMEX");
+		}else{
+			arg1Tree = new SimpleTree("TIMEX");
+			arg2Tree = new SimpleTree("EVENT");
+		}
+		
+		List<BaseToken> tokens = JCasUtil.selectCovered(jcas, BaseToken.class, arg1.getBegin(), arg2.getEnd());
+		tree.addChild(arg1Tree);
+		for(BaseToken token : tokens){
+			SimpleTree tokenTree = new SimpleTree("TOK");
+			tokenTree.addChild(new SimpleTree(token.getPartOfSpeech()));
+			
+			if(token.getEnd() <= arg1.getEnd()){
+				arg1Tree.addChild(tokenTree);
+			}else if(token.getBegin() >= arg2.getBegin()){
+				arg2Tree.addChild(tokenTree);
+			}else{
+				tree.addChild(tokenTree);
+			}
+		}
+		tree.addChild(arg2Tree);
+		
+		feats.add(new Feature("TK_BOP", tree.toString()));
+		return feats;
+	}
+
+
+}