[DL-Learner SVN] SF.net SVN: dl-learner:[4030] trunk/components-core/src/main/java/org/ dllearner/algorithms/isle/index

[<lor...@us...>]

Revision: 4030
          http://sourceforge.net/p/dl-learner/code/4030
Author:   lorenz_b
Date:     2013-09-03 14:28:00 +0000 (Tue, 03 Sep 2013)
Log Message:
-----------
Implemented AnnotatedDocument interface.

Modified Paths:
--------------
    trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/semantic/simple/SimpleSemanticIndex.java

Added Paths:
-----------
    trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/AnnotatedTextDocument.java

Added: trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/AnnotatedTextDocument.java
===================================================================
--- trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/AnnotatedTextDocument.java	                        (rev 0)
+++ trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/AnnotatedTextDocument.java	2013-09-03 14:28:00 UTC (rev 4030)
@@ -0,0 +1,91 @@
+/**
+ * 
+ */
+package org.dllearner.algorithms.isle.index;
+
+import java.util.HashSet;
+import java.util.Set;
+
+import org.dllearner.core.owl.Entity;
+
+/**
+ * @author Lorenz Buehmann
+ *
+ */
+public class AnnotatedTextDocument implements AnnotatedDocument{
+	
+	private TextDocument document;
+	private Set<Annotation> annotations;
+	private Set<Entity> entities;
+
+	
+	public AnnotatedTextDocument(TextDocument document, Set<Annotation> annotations) {
+		this.document = document;
+		this.annotations = annotations;
+		
+		entities = new HashSet<Entity>();
+		for (Annotation annotation : annotations) {
+			entities.add(annotation.getEntity());
+		}
+	}
+
+	/* (non-Javadoc)
+	 * @see org.dllearner.algorithms.isle.index.Document#getContent()
+	 */
+	@Override
+	public String getContent() {
+		return document.getContent();
+	}
+
+	/* (non-Javadoc)
+	 * @see org.dllearner.algorithms.isle.index.Document#getRawContent()
+	 */
+	@Override
+	public String getRawContent() {
+		return document.getRawContent();
+	}
+
+	/* (non-Javadoc)
+	 * @see org.dllearner.algorithms.isle.index.AnnotatedDocument#getContainedEntities()
+	 */
+	@Override
+	public Set<Entity> getContainedEntities() {
+		return entities;
+	}
+
+	/* (non-Javadoc)
+	 * @see org.dllearner.algorithms.isle.index.AnnotatedDocument#getAnnotations()
+	 */
+	@Override
+	public Set<Annotation> getAnnotations() {
+		return annotations;
+	}
+
+	/* (non-Javadoc)
+	 * @see org.dllearner.algorithms.isle.index.AnnotatedDocument#getAnnotation(int, int)
+	 */
+	@Override
+	public Annotation getAnnotation(int offset, int length) {
+		for (Annotation annotation : annotations) {
+			if(annotation.getOffset() == offset && annotation.getLength() == length){
+				return annotation;
+			}
+		}
+		return null;
+	}
+
+	/* (non-Javadoc)
+	 * @see org.dllearner.algorithms.isle.index.AnnotatedDocument#getEntityFrequency(org.dllearner.core.owl.Entity)
+	 */
+	@Override
+	public int getEntityFrequency(Entity entity) {
+		int cnt = 0;
+		for (Annotation annotation : annotations) {
+			if(annotation.getEntity().equals(entity)){
+				cnt++;
+			}
+		}
+		return cnt;
+	}
+
+}

Modified: trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/semantic/simple/SimpleSemanticIndex.java
===================================================================
--- trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/semantic/simple/SimpleSemanticIndex.java	2013-09-03 14:19:50 UTC (rev 4029)
+++ trunk/components-core/src/main/java/org/dllearner/algorithms/isle/index/semantic/simple/SimpleSemanticIndex.java	2013-09-03 14:28:00 UTC (rev 4030)
@@ -3,6 +3,7 @@
  */
 package org.dllearner.algorithms.isle.index.semantic.simple;
 
+import org.dllearner.algorithms.isle.index.AnnotatedDocument;
 import org.dllearner.algorithms.isle.index.Document;
 import org.dllearner.algorithms.isle.index.syntactic.SyntacticIndex;
 import org.dllearner.algorithms.isle.index.semantic.SemanticIndex;
@@ -41,7 +42,7 @@
      * @see org.dllearner.algorithms.isle.SemanticIndex#getDocuments(org.dllearner.core.owl.Entity)
      */
     @Override
-    public Set<Document> getDocuments(Entity entity) {
+    public Set<AnnotatedDocument> getDocuments(Entity entity) {
         Set<Document> documents = new HashSet<Document>();
         Map<String, Double> relevantText = labelRetriever.getRelevantText(entity);
 

This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.