svn commit: r1536904 - /ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java

[cl...@apache.org]

Author: clin
Date: Tue Oct 29 21:58:42 2013
New Revision: 1536904

URL: http://svn.apache.org/r1536904
Log:
add nearby DateAnnotation and TimeAnnotation of an event for DocTimeRel classification

Added:
    ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java   (with props)

Added: ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java
URL: http://svn.apache.org/viewvc/ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java?rev=1536904&view=auto
==============================================================================
--- ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java (added)
+++ ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java Tue Oct 29 21:58:42 2013
@@ -0,0 +1,156 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 org.apache.ctakes.temporal.ae.feature;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
+//import java.util.logging.Logger;
+
+//import org.apache.ctakes.temporal.ae.feature.treekernel.TemporalPETExtractor;
+//import org.apache.ctakes.temporal.ae.feature.treekernel.TemporalSingleTreeExtractor;
+import org.apache.ctakes.typesystem.type.syntax.TreebankNode;
+import org.apache.ctakes.typesystem.type.textsem.DateAnnotation;
+import org.apache.ctakes.typesystem.type.textsem.EventMention;
+import org.apache.ctakes.typesystem.type.textsem.IdentifiedAnnotation;
+import org.apache.ctakes.typesystem.type.textsem.TimeAnnotation;
+import org.apache.ctakes.typesystem.type.textsem.TimeMention;
+import org.apache.ctakes.typesystem.type.textspan.Sentence;
+import org.apache.uima.analysis_engine.AnalysisEngineProcessException;
+import org.apache.uima.jcas.JCas;
+import org.apache.uima.jcas.tcas.Annotation;
+import org.apache.uima.resource.ResourceInitializationException;
+import org.cleartk.classifier.Feature;
+import org.cleartk.classifier.feature.extractor.CleartkExtractorException;
+import org.cleartk.classifier.feature.extractor.simple.SimpleFeatureExtractor;
+import org.uimafit.util.JCasUtil;
+
+public class TimeXExtractor implements SimpleFeatureExtractor {
+
+  private String name;
+//  private TemporalPETExtractor path;
+  private TemporalAttributeFeatureExtractor attr;
+  private TimeWordTypeExtractor timewd;
+//  private TemporalSingleTreeExtractor treeExt;
+  
+//  private Logger logger = Logger.getLogger(this.getClass().getName());
+
+  public TimeXExtractor() throws ResourceInitializationException {
+    super();
+    this.name = "TimeXFeature";
+//    this.path = new TemporalPETExtractor();
+    this.attr = new TemporalAttributeFeatureExtractor();
+    this.timewd = new TimeWordTypeExtractor();
+//    this.treeExt = new TemporalSingleTreeExtractor();
+  }
+
+  @Override
+  public List<Feature> extract(JCas view, Annotation annotation) throws CleartkExtractorException {
+	  List<Feature> features = new ArrayList<Feature>();
+	  
+	  //1 get covering sentence:
+	  Map<EventMention, Collection<Sentence>> coveringMap =
+			  JCasUtil.indexCovering(view, EventMention.class, Sentence.class);
+	  EventMention targetTokenAnnotation = (EventMention)annotation;
+	  Collection<Sentence> sentList = coveringMap.get(targetTokenAnnotation);
+	  
+	  //2 get TimeX
+	  Map<Integer, IdentifiedAnnotation> timeDistMap = null;
+	  
+//	  List<Feature> treePath = new ArrayList<Feature>();
+	  
+	  //3 get Document Creation Time
+//	  String sofastr = view.getSofaDataString();
+//	  int start = sofastr.indexOf("meta rev_date=");
+//	  int end = sofastr.indexOf(" start_date=");
+//	  System.out.println(sofastr.substring(start, end));
+//	  Collection<SourceData> sources = JCasUtil.select(view, SourceData.class);
+//	  for(SourceData source : sources){
+//			System.out.println("original date: "+source.getSourceOriginalDate());
+//			System.out.println("revision date: "+source.getSourceRevisionDate());
+//		}
+	  
+	  if (sentList != null && !sentList.isEmpty()){
+		  timeDistMap = new TreeMap<Integer, IdentifiedAnnotation>();
+		  
+		  for(Sentence sent : sentList) {
+			  for (TimeMention time : JCasUtil.selectCovered(view, TimeMention.class, sent)) {
+				  timeDistMap.put(Math.abs(time.getBegin() - annotation.getBegin()), time);
+			  }
+			  for (TimeAnnotation time : JCasUtil.selectCovered(view, TimeAnnotation.class, sent)) {
+				  timeDistMap.put(Math.abs(time.getBegin() - annotation.getBegin()), time);
+			  }
+			  for (DateAnnotation time : JCasUtil.selectCovered(view, DateAnnotation.class, sent)) {
+				  timeDistMap.put(Math.abs(time.getBegin() - annotation.getBegin()), time);
+			  }
+		  }
+		  
+		  //get the closest Time Expression feature
+		  for (Map.Entry<Integer, IdentifiedAnnotation> entry : timeDistMap.entrySet()) {
+			  Feature feature = new Feature(this.name, entry.getValue().getCoveredText());
+			  features.add(feature);
+			  //			  logger.info("add time feature: "+ entry.getValue().getCoveredText() + entry.getValue().getTimeClass());
+			  Feature indicator = new Feature("TimeXNearby", this.name);
+			  features.add(indicator);
+			  Feature type = new Feature("TimeXType", entry.getValue().getClass());
+			  features.add(type);
+
+			  //add PP get Heading preposition
+			  for(TreebankNode treebankNode : JCasUtil.selectCovering(
+					  view, 
+					  TreebankNode.class, 
+					  entry.getValue().getBegin(), 
+					  entry.getValue().getEnd())) {
+
+				  if(treebankNode.getNodeType().equals("PP")) {
+					  Feature PPNodeType = new Feature("Timex_PPNodeType", treebankNode.getNodeType());
+					  features.add(PPNodeType);
+					  break;
+				  }
+			  }
+
+			  //add path tree, timex attributes
+			  try {
+//				  treePath=this.path.extract(view, targetTokenAnnotation, entry.getValue());//add path between timex and event
+				  features.addAll(this.attr.extract(view, targetTokenAnnotation, entry.getValue()));//add temporal attribute features
+				  features.addAll(this.timewd.extract(view, entry.getValue()));
+			  } catch (AnalysisEngineProcessException e) {
+				  throw new IllegalArgumentException(String.format("error in gererating path feature:", features));
+			  }
+			  break;
+		  }
+	  }
+
+//	  if (treePath.isEmpty()){
+//		  try {
+//			  features.addAll(this.treeExt.extract(view, targetTokenAnnotation));
+//		  } catch (AnalysisEngineProcessException e) {
+//			  throw new IllegalArgumentException(String.format("error in gererating path feature:", features));
+//		  }
+//	  }else{
+//		  features.addAll(treePath);
+//	  }
+
+
+	  return features;
+  }
+
+}

Propchange: ctakes/trunk/ctakes-temporal/src/main/java/org/apache/ctakes/temporal/ae/feature/TimeXExtractor.java
------------------------------------------------------------------------------
    svn:mime-type = text/plain