[foliaspec] fix for rust code generation

proycon · Sep 2, 2020 · 1ed0a35 · 1ed0a35
1 parent 5086ec3
commit 1ed0a35
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/foliatools/foliaspec.py b/foliatools/foliaspec.py
@@ -691,15 +691,15 @@ def outputblock(block, target, varname, args, indent = ""):
                 if element['class'].endswith('Layer'):
                     if 'properties' in element and 'xmltag' in element['properties'] and element['properties']['xmltag'] and 'annotationtype' in element['properties']:
                         s += indent + "    AnnotationType." + element['properties']['annotationtype'] + ':  ' + element['class'] + ' ,\n'
-            s += indent + "    AnnotationType.PREDICATE:  SemanticRolesLayer\n"
+            s += indent + "    AnnotationType.PREDICATE:  ElementType::SemanticRolesLayer\n"
             s += indent + "}"
         elif target == 'rust':
             s += indent + "match " + args[0] + " {\n"
             for element in elements:
                 if element['class'].endswith('Layer'):
                     if 'properties' in element and 'xmltag' in element['properties'] and element['properties']['xmltag'] and 'annotationtype' in element['properties']:
                         s += indent + "    AnnotationType::" + element['properties']['annotationtype'] + ' => Some(ElementType::' + element['class'] + '),\n'
-            s += indent + "    AnnotationType::PREDICATE => Some(SemanticRolesLayer),\n"
+            s += indent + "    AnnotationType::PREDICATE => Some(ElementType::SemanticRolesLayer),\n"
             s += indent + "    _ => None\n"
             s += indent + "}"
         else: