Author: joern
Date: Thu May 28 11:34:43 2015
New Revision: 1682215

URL: http://svn.apache.org/r1682215
Log:
OPENNLP-784 Now recognizes all kinds of white spaces to seperate a line in the 
annotation config

Modified:
    
opennlp/trunk/opennlp-tools/src/main/java/opennlp/tools/formats/brat/AnnotationConfiguration.java

Modified: 
opennlp/trunk/opennlp-tools/src/main/java/opennlp/tools/formats/brat/AnnotationConfiguration.java
URL: 
http://svn.apache.org/viewvc/opennlp/trunk/opennlp-tools/src/main/java/opennlp/tools/formats/brat/AnnotationConfiguration.java?rev=1682215&r1=1682214&r2=1682215&view=diff
==============================================================================
--- 
opennlp/trunk/opennlp-tools/src/main/java/opennlp/tools/formats/brat/AnnotationConfiguration.java
 (original)
+++ 
opennlp/trunk/opennlp-tools/src/main/java/opennlp/tools/formats/brat/AnnotationConfiguration.java
 Thu May 28 11:34:43 2015
@@ -25,6 +25,7 @@ import java.nio.charset.Charset;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
+import opennlp.tools.tokenize.WhitespaceTokenizer;
 
 public class AnnotationConfiguration {
 
@@ -58,7 +59,7 @@ public class AnnotationConfiguration {
     while ((line = reader.readLine())!= null) {
       line = line.trim();
 
-      if (line.length() == 0) {
+      if (line.isEmpty()) {
         continue;
       } else if (line.startsWith("#")) {
         continue;
@@ -66,18 +67,19 @@ public class AnnotationConfiguration {
         sectionType = line.substring(line.indexOf('[') + 1, line.indexOf(']'));
       }
       else {
+        String typeName = WhitespaceTokenizer.INSTANCE.tokenize(line)[0];
 
         switch (sectionType) {
         case "entities":
-          typeToClassMap.put(line, AnnotationConfiguration.ENTITY_TYPE);
+          typeToClassMap.put(typeName, AnnotationConfiguration.ENTITY_TYPE);
           break;
 
         case "relations":
-          typeToClassMap.put(line.substring(0, line.indexOf(' ')), 
AnnotationConfiguration.RELATION_TYPE);
+          typeToClassMap.put(typeName, AnnotationConfiguration.RELATION_TYPE);
           break;
 
         case "attributes":
-          typeToClassMap.put(line.substring(0, line.indexOf(' ')), 
AnnotationConfiguration.ATTRIBUTE_TYPE);
+          typeToClassMap.put(typeName, AnnotationConfiguration.ATTRIBUTE_TYPE);
           break;
 
         default:


Reply via email to