Add MWT fields to AnnotationLookup, which necessitates parsing Boolean in CoreLabel

Author: AngledLuffa

src/edu/stanford/nlp/ling/AnnotationLookup.java

@@ -93,8 +93,12 @@ private enum KeyLookup {
     POS_TAG_KEY(CoreAnnotations.PartOfSpeechAnnotation.class, "pos"),
     CPOS_TAG_KEY(CoreAnnotations.CoarseTagAnnotation.class, "cpos"),
     DEPREL_KEY(CoreAnnotations.CoNLLDepTypeAnnotation.class, "deprel"),
-    HEADIDX_KEY(CoreAnnotations.CoNLLDepParentIndexAnnotation.class, "headidx");
+    HEADIDX_KEY(CoreAnnotations.CoNLLDepParentIndexAnnotation.class, "headidx"),
 
+    // MWT specific annotations
+    MWT_TEXT_KEY(CoreAnnotations.MWTTokenTextAnnotation.class, "mwt_text"),
+    IS_MWT_KEY(CoreAnnotations.IsMultiWordTokenAnnotation.class, "is_mwt"),
+    IS_FIRST_MWT_KEY(CoreAnnotations.IsFirstWordOfMWTAnnotation.class, "is_first_mwt");
 
     private final Class<? extends CoreAnnotation<?>> coreKey;
     private final String oldKey;

src/edu/stanford/nlp/ling/CoreLabel.java

@@ -198,6 +198,8 @@ private void initFromStrings(String[] keys, String[] values) {
             this.set(coreKeyClass, Double.parseDouble(values[i]));
           } else if(valueClass == Long.class) {
             this.set(coreKeyClass, Long.parseLong(values[i]));
+          } else if (valueClass == Boolean.class) {
+            this.set(coreKeyClass, Boolean.parseBoolean(values[i]));
           } else if (coreKeyClass == CoreAnnotations.CoNLLUFeats.class) {
             this.set(coreKeyClass, CoNLLUUtils.parseFeatures(values[i]));
           } else {
@@ -249,6 +251,8 @@ private void initFromStrings(Class[] keys, String[] values) {
           this.set(coreKeyClass, Double.parseDouble(values[i]));
         } else if (valueClass == Long.class) {
           this.set(coreKeyClass, Long.parseLong(values[i]));
+        } else if (valueClass == Boolean.class) {
+          this.set(coreKeyClass, Boolean.parseBoolean(values[i]));
         } else if (coreKeyClass == CoreAnnotations.CoNLLUFeats.class) {
           this.set(coreKeyClass, CoNLLUUtils.parseFeatures(values[i]));
         } else {