org.simantics.xml.sax/src/org/simantics/xml/sax/ImporterGenerator.java

   1 package org.simantics.xml.sax;\r
   2 \r
   3 import java.io.File;\r
   4 import java.io.IOException;\r
   5 import java.io.PrintWriter;\r
   6 import java.io.StringWriter;\r
   7 import java.util.ArrayList;\r
   8 import java.util.List;\r
   9 import java.util.Map;\r
  10 \r
  11 import javax.xml.namespace.QName;\r
  12 \r
  13 import org.simantics.utils.datastructures.BijectionMap;\r
  14 import org.simantics.xml.sax.configuration.AttributeComposition;\r
  15 import org.simantics.xml.sax.configuration.Configuration;\r
  16 import org.simantics.xml.sax.configuration.IDProvider;\r
  17 import org.simantics.xml.sax.configuration.IDReference;\r
  18 import org.simantics.xml.sax.configuration.UnrecognizedChildElement;\r
  19 import org.w3._2001.xmlschema.Annotated;\r
  20 import org.w3._2001.xmlschema.Attribute;\r
  21 import org.w3._2001.xmlschema.AttributeGroup;\r
  22 import org.w3._2001.xmlschema.ComplexType;\r
  23 import org.w3._2001.xmlschema.Element;\r
  24 import org.w3._2001.xmlschema.LocalComplexType;\r
  25 import org.w3._2001.xmlschema.LocalSimpleType;\r
  26 import org.w3._2001.xmlschema.NamedAttributeGroup;\r
  27 import org.w3._2001.xmlschema.Restriction;\r
  28 import org.w3._2001.xmlschema.Schema;\r
  29 import org.w3._2001.xmlschema.SimpleType;\r
  30 import org.w3._2001.xmlschema.TopLevelAttribute;\r
  31 \r
  32 public class ImporterGenerator extends JavaGenerator{\r
  33         \r
  34         public ImporterGenerator(Configuration configuration) {\r
  35                 super(configuration);\r
  36         }\r
  37         \r
  38         public void createParser(Schema schema,String ontologyUri, String className, SchemaConverter converter) throws IOException {\r
  39                 this.schema = schema;\r
  40                 this.ontologyClassName = className;\r
  41                 this.converter = converter;\r
  42                 \r
  43                 \r
  44                 String packageParts[] = className.split("\\.");\r
  45                 String name = packageParts[packageParts.length-1];\r
  46         \r
  47                 \r
  48                 ontShort = name.substring(0, 3).toUpperCase();\r
  49                 ontShort +=".";\r
  50                 String parserPackagePostfix = "_elem";\r
  51                 String importerClassPostfix = "Importer";\r
  52                 String parserClassPostfix = "Parser";\r
  53                 elementPackageName = name+parserPackagePostfix;\r
  54                 \r
  55                 importParserDir= new File(converter.getParserDir().getAbsolutePath()+"/"+elementPackageName);\r
  56                 if (!importParserDir.exists())\r
  57                         importParserDir.mkdirs();\r
  58                 \r
  59                 handle(schema);\r
  60                 // Create Importer class\r
  61                 File importerFile = new File(converter.getParserDir().getAbsolutePath()+"/"+name+importerClassPostfix+".java");\r
  62                 PrintWriter mainWriter = createFile(importerFile);\r
  63                 mainWriter.println("package " + converter.getPluginName() +";");\r
  64                 mainWriter.println();\r
  65                 mainWriter.println("import java.io.File;");\r
  66                 mainWriter.println("import org.simantics.db.Session;");\r
  67                 mainWriter.println("import org.simantics.xml.sax.base.AbstractImporter;");\r
  68                 mainWriter.println();\r
  69                 mainWriter.println("public class " + name + importerClassPostfix+" extends AbstractImporter {");\r
  70                 mainWriter.println();\r
  71                 mainWriter.println("   public " + name + importerClassPostfix+"(Session session, File file)  {");\r
  72                 mainWriter.println("      super(session, file, new "+name + parserClassPostfix+"());");\r
  73                 mainWriter.println("   }");\r
  74                 mainWriter.println();\r
  75                 mainWriter.println("}");\r
  76                 \r
  77                 mainWriter.println();\r
  78                 mainWriter.flush();\r
  79                 mainWriter.close();\r
  80                 \r
  81                 // Create Parser class\r
  82                 File parserFile = new File(converter.getParserDir().getAbsolutePath()+"/"+name+parserClassPostfix+".java");\r
  83                 mainWriter = createFile(parserFile);\r
  84                 mainWriter.println("package " + converter.getPluginName() +";");\r
  85                 mainWriter.println();\r
  86                 mainWriter.println("import org.simantics.xml.sax.base.XMLParser;");\r
  87                 mainWriter.println();\r
  88                 mainWriter.println("public class " + name + parserClassPostfix+" extends XMLParser {");\r
  89                 mainWriter.println();\r
  90                 mainWriter.println("   public " + name + parserClassPostfix+"()  {");\r
  91                 if (schema.getTargetNamespace() != null)\r
  92                 mainWriter.println("      setSchemaURI(\""+schema.getTargetNamespace()+"\");");\r
  93                 for (String s : ruleClassNames) {\r
  94                 mainWriter.println("      add(new "+s+"());");\r
  95                 }\r
  96                 mainWriter.println("   }");\r
  97                 mainWriter.println("}");\r
  98                 \r
  99                 mainWriter.println();\r
 100                 mainWriter.flush();\r
 101                 mainWriter.close();\r
 102         }\r
 103 \r
 104         protected void handle(TopLevelAttribute topLevelAttribute) {\r
 105         }\r
 106         \r
 107         @Override\r
 108         protected void handleSimpleType(SchemaObject parent, SchemaObject simpleTypeObj) {\r
 109         }\r
 110         \r
 111         @Override\r
 112         protected void handleComplexType(SchemaObject complexTypeObj) {\r
 113                 ComplexType topLevelComplexType = complexTypeObj.getComplexType();\r
 114                 \r
 115                 String name = getName(complexTypeObj);\r
 116                 \r
 117 //              if (topLevelComplexType.getName().equals("Reference"))\r
 118 //                      System.out.println();\r
 119                 \r
 120                 String className = name;//"_"+name;\r
 121                 \r
 122                 FileWriter fw = new FileWriter();\r
 123                 try {\r
 124                         fw.writer = createFile(new File(importParserDir.getAbsolutePath()+"/"+className+".java"));\r
 125                 } catch (IOException e) {\r
 126                         throw new RuntimeException(e);\r
 127                 }\r
 128                 writers.put(complexTypeObj, fw);\r
 129 \r
 130                 String baseClass = getBaseClass();\r
 131                 \r
 132                 boolean inherited = false;\r
 133                 \r
 134                 QName type = getComplexTypeBase(topLevelComplexType);\r
 135                 if (type != null && !type.getNamespaceURI().equals("http://www.w3.org/2001/XMLSchema")) {\r
 136                         SchemaObject obj = complexTypeName.get(type.getLocalPart());\r
 137                         if (obj != null) {\r
 138                                 baseClass = getName(obj);\r
 139                                 inherited = true;\r
 140                         }\r
 141                 }\r
 142                 \r
 143                 provider = getIDProvider(topLevelComplexType);\r
 144                 List<IDReference> references = getIDReferences(topLevelComplexType);\r
 145                 UnrecognizedChildElement unknownChildElement = getUnknown(topLevelComplexType);\r
 146 \r
 147                 List<String> intrerfaces = new ArrayList<String>();\r
 148                 if (references.size() > 0)\r
 149                         intrerfaces.add("org.simantics.xml.sax.base.IDReferenceParser");\r
 150                 if (unknownChildElement != null)\r
 151                         intrerfaces.add("org.simantics.xml.sax.base.UnrecognizedElementParser");\r
 152                 \r
 153                 createClassHeader(fw.writer, false);\r
 154                 writeClass(fw.writer,false, null, className, baseClass, intrerfaces);\r
 155                         \r
 156                 writeIDProvider(fw.writer);\r
 157         \r
 158                 fw.writer.println("   @Override");\r
 159                 fw.writer.println("   public Resource create(WriteGraph graph, ParserElement element) throws DatabaseException{");\r
 160                 fw.writer.println("      Layer0 L0 = Layer0.getInstance(graph);");\r
 161                 fw.writer.println("      "+getOntologyImport());\r
 162                 fw.writer.println("      Resource res = graph.newResource();");\r
 163                 fw.writer.println("      graph.claim(res, L0.InstanceOf, "+ontShort+name+");");\r
 164                 fw.writer.println("      return res;");\r
 165                 fw.writer.println("   }");\r
 166                 fw.writer.println();\r
 167                 \r
 168                 fw.writer.println("   @Override");\r
 169                 fw.writer.println("   public boolean connectParent(WriteGraph graph, ParserElement parent, ParserElement element) throws DatabaseException{");\r
 170                 fw.writer.println("      "+getOntologyImport());\r
 171                 fw.writer.println("      graph.claim(parent.getData(), "+this.ontShort+getName(complexTypeObj,"has")+", element.getData());");\r
 172                 fw.writer.println("      return true;");\r
 173                 fw.writer.println("   }");\r
 174                 fw.writer.println();\r
 175                                 \r
 176                 StringWriter stringWriter = new StringWriter();\r
 177                 fw.delayedWriter = new PrintWriter(stringWriter);\r
 178                 StringWriter stringWriter2 = new StringWriter();\r
 179                 fw.delayedWriter2 = new PrintWriter(stringWriter2);\r
 180                 \r
 181                 fw.writer.println("   public " + className + "() {");\r
 182                 fw.writer.println("      super();");\r
 183                 \r
 184                 handleComplexTypeExtension(complexTypeObj);\r
 185                 \r
 186                 fw.writer.println("   }");\r
 187                 \r
 188                 fw.writer.println("   @Override");\r
 189                 fw.writer.println("   public boolean connectChild(WriteGraph graph, ParserElement element, ParserElement child) throws DatabaseException{");\r
 190                 fw.writer.println("      "+getOntologyImport());\r
 191                 \r
 192                 if (stringWriter.getBuffer().length() > 0) {\r
 193                         fw.writer.write(stringWriter.toString());\r
 194                 }\r
 195                 \r
 196                 \r
 197                 fw.writer.println("      return false;");\r
 198                 fw.writer.println("   }");\r
 199                 fw.writer.println();\r
 200                 \r
 201                 if (stringWriter2.getBuffer().length() > 0) {\r
 202                         fw.writer.write(stringWriter2.toString());\r
 203                 }\r
 204                 \r
 205                 stringWriter = null;\r
 206                 fw.delayedWriter.close();\r
 207                 fw.delayedWriter=null;\r
 208                 stringWriter2 = null;\r
 209                 fw.delayedWriter2.close();\r
 210                 fw.delayedWriter2 = null;\r
 211                 \r
 212                 fw.writer.println("   @Override");\r
 213                 fw.writer.println("   public void configure(WriteGraph graph, Deque<ParserElement> parents, ParserElement element) throws DatabaseException {");\r
 214                 if (inherited) {\r
 215                 fw.writer.println("             super.configure(graph,parents,element);");\r
 216                 }\r
 217                 fw.writer.println("        "+getOntologyImport());\r
 218                 \r
 219                 handleComplexTypeAttributes(complexTypeObj);\r
 220                 handleExtensionAttributes(complexTypeObj);\r
 221                 \r
 222                 fw.writer.println("   }");\r
 223                 \r
 224                 writeIDReferences(fw.writer,name, references);\r
 225                 writeUnknownChild(fw.writer,name,unknownChildElement);\r
 226                 \r
 227                 fw.writer.println("}");\r
 228                 fw.writer.println();\r
 229                 fw.writer.flush();\r
 230                 fw.writer.close();\r
 231                 fw.writer = null;\r
 232                 writers.remove(complexTypeObj);\r
 233                 provider = null;\r
 234         }\r
 235         \r
 236         @Override\r
 237         //protected void handleIndicator(SchemaObject parent, SchemaElement indicator, SchemaElement element,  boolean reference, String ref, QName refType) {\r
 238         protected void handleIndicator(SchemaObject parent, SchemaElement indicator, SchemaElement element, String refName, RefType referenceType) {\r
 239                 FileWriter fw = getWriter(parent);\r
 240                 if (referenceType != RefType.Element) {\r
 241                         QName refType;\r
 242                         if (referenceType == RefType.Type) {\r
 243                                 refType = element.getElement().getType();\r
 244                                 if (refName == null)\r
 245                                         refName = element.getElement().getName();\r
 246                         } else {\r
 247                                 refType = element.getElement().getRef();\r
 248                                 if (refName == null)\r
 249                                         refName = refType.getLocalPart();\r
 250                         }\r
 251                         \r
 252                         String binding = getBindingFromPrimitiveType(refType);\r
 253                         if (binding == null) {\r
 254                                 SchemaObject refElement = elementName.get(refType.getLocalPart());\r
 255                                 SchemaObject refComplexType = complexTypeName.get(refType.getLocalPart());\r
 256         \r
 257                                 String className = null;\r
 258                                 String refClassName = null;\r
 259                                 if (refElement != null)\r
 260                                         className = getName(refElement);\r
 261                                 else\r
 262                                         className = getName(refComplexType);\r
 263                                 \r
 264                                 if (refComplexType != null) {\r
 265                                         refClassName = getName(refComplexType);\r
 266                                 } else {\r
 267                                         refClassName = getName(refElement);\r
 268                                 }\r
 269                                 \r
 270                                 if (referenceType != RefType.Reference)\r
 271                                 fw.writer.println("        addParser(\""+ refName +"\", "+className+".class);");\r
 272                                 else\r
 273                                 fw.writer.println("        addParser("+className+".class);");\r
 274                                 if (!className.equals(refClassName))\r
 275                                 fw.writer.println("        addParser("+refClassName+".class);");\r
 276                                 \r
 277                                 fw.delayedWriter.println("         if (child.getElementParser() instanceof "+refClassName+"){");\r
 278                                 fw.delayedWriter.println("            graph.claim(element.getData(), "+ontShort+getName(parent)+"_has"+refName + ", child.getData());");\r
 279                                 if (useElementList(parent, indicator,element, referenceType == RefType.Reference, refName, refType)) {\r
 280                 \r
 281                                 // element type specific list\r
 282                                 fw.delayedWriter.println("            {");\r
 283                                 fw.delayedWriter.println("               Resource list = graph.getPossibleObject(element.getData(),"+ontShort+getName(parent)+"_has"+refName + "List);");\r
 284                                 fw.delayedWriter.println("               if (list == null) {");\r
 285                                 fw.delayedWriter.println("                  list = org.simantics.db.common.utils.ListUtils.create(graph, java.util.Collections.singletonList(child.getData()));");\r
 286                                 fw.delayedWriter.println("                  graph.claim(element.getData(),"+ontShort+getName(parent)+"_has"+refName + "List,list);");\r
 287                                 fw.delayedWriter.println("               } else {");\r
 288                                 fw.delayedWriter.println("                  org.simantics.db.common.utils.ListUtils.insertBack(graph, list, java.util.Collections.singletonList(child.getData()));");\r
 289                                 fw.delayedWriter.println("               }");\r
 290                                 fw.delayedWriter.println("            }");\r
 291                                 }\r
 292                                 if (useOriginalList(parent, indicator,element, referenceType == RefType.Reference, refName, refType)) {\r
 293                                 // generic list\r
 294                                 fw.delayedWriter.println("            {");\r
 295                                 fw.delayedWriter.println("               XMLResource XML = XMLResource.getInstance(graph);");\r
 296                                 fw.delayedWriter.println("               Resource list = graph.getPossibleObject(element.getData(), XML.hasOriginalElementList);");\r
 297                                 fw.delayedWriter.println("               if (list == null) {");\r
 298                                 fw.delayedWriter.println("                  list = org.simantics.db.common.utils.ListUtils.create(graph, java.util.Collections.singletonList(child.getData()));");\r
 299                                 fw.delayedWriter.println("                  graph.claim(element.getData(), XML.hasOriginalElementList,list);");\r
 300                                 fw.delayedWriter.println("               } else {");\r
 301                                 fw.delayedWriter.println("                  org.simantics.db.common.utils.ListUtils.insertBack(graph, list, java.util.Collections.singletonList(child.getData()));");\r
 302                                 fw.delayedWriter.println("               }");\r
 303                                 fw.delayedWriter.println("            }");\r
 304                                 }\r
 305                                 \r
 306                                 fw.delayedWriter.println("            return true;");\r
 307                                 fw.delayedWriter.println("         }");\r
 308                         } else {\r
 309                                 //writer.println(commentTag+ontShort+"."+parent+".has"+ref + " <R " +  ontShort+".hasElement");\r
 310                                 fw.writer.println("        addParser(\""+ refName +"\", "+getName(parent) +"_" +refName+".class);");\r
 311                                 \r
 312                                 fw.delayedWriter2.println("   public static class " + getName(parent) +"_" +refName+" extends org.simantics.xml.sax.base.ValueElementParser {");\r
 313                                 fw.delayedWriter2.println("      public "+ getName(parent) +"_" +refName +"(){");\r
 314                                 fw.delayedWriter2.println("           super(\""+refName+"\"," +this.ontologyClassName+".URIs."+getName(parent) + "_has"+refName+", "+binding+");");\r
 315                                 fw.delayedWriter2.println("       }");\r
 316                                 fw.delayedWriter2.println("   }");\r
 317                         }\r
 318                 } else {\r
 319                         Element attrs= element.getElement();\r
 320                         SchemaObject obj = getWithObj(parent, attrs);\r
 321                         \r
 322                         String className = getName(obj);\r
 323                         if (refName == null)\r
 324                                 refName = attrs.getName();\r
 325                         \r
 326                         \r
 327                         //if (!reference)\r
 328                         fw.writer.println("        addParser(\""+ refName +"\", "+className+".class);");\r
 329 //                      else\r
 330 //                      fw.writer.println("        addParser("+className+".class);");\r
 331                         \r
 332                         fw.delayedWriter.println("         if (child.getElementParser() instanceof "+className+"){");\r
 333                         fw.delayedWriter.println("            graph.claim(element.getData(), "+ontShort+getName(parent)+"_has"+refName + ", child.getData());");\r
 334                         if (useElementList(parent, indicator,element, false, refName, new QName(obj.getName()))) {\r
 335 \r
 336                         // element type specific list\r
 337                         fw.delayedWriter.println("            {");\r
 338                         fw.delayedWriter.println("               Resource list = graph.getPossibleObject(element.getData(),"+ontShort+getName(parent)+"_has"+refName + "List);");\r
 339                         fw.delayedWriter.println("               if (list == null) {");\r
 340                         fw.delayedWriter.println("                  list = org.simantics.db.common.utils.ListUtils.create(graph, java.util.Collections.singletonList(child.getData()));");\r
 341                         fw.delayedWriter.println("                  graph.claim(element.getData(),"+ontShort+getName(parent)+"_has"+refName + "List,list);");\r
 342                         fw.delayedWriter.println("               } else {");\r
 343                         fw.delayedWriter.println("                  org.simantics.db.common.utils.ListUtils.insertBack(graph, list, java.util.Collections.singletonList(child.getData()));");\r
 344                         fw.delayedWriter.println("               }");\r
 345                         fw.delayedWriter.println("            }");\r
 346                         }\r
 347                         if (useOriginalList(parent, indicator,element, false, refName, new QName(obj.getName()))) {\r
 348                         // generic list\r
 349                         fw.delayedWriter.println("            {");\r
 350                         fw.delayedWriter.println("               XMLResource XML = XMLResource.getInstance(graph);");\r
 351                         fw.delayedWriter.println("               Resource list = graph.getPossibleObject(element.getData(), XML.hasOriginalElementList);");\r
 352                         fw.delayedWriter.println("               if (list == null) {");\r
 353                         fw.delayedWriter.println("                  list = org.simantics.db.common.utils.ListUtils.create(graph, java.util.Collections.singletonList(child.getData()));");\r
 354                         fw.delayedWriter.println("                  graph.claim(element.getData(), XML.hasOriginalElementList,list);");\r
 355                         fw.delayedWriter.println("               } else {");\r
 356                         fw.delayedWriter.println("                  org.simantics.db.common.utils.ListUtils.insertBack(graph, list, java.util.Collections.singletonList(child.getData()));");\r
 357                         fw.delayedWriter.println("               }");\r
 358                         fw.delayedWriter.println("            }");\r
 359                         }\r
 360                         \r
 361                         fw.delayedWriter.println("            return true;");\r
 362                         fw.delayedWriter.println("         }");\r
 363                 }\r
 364         }\r
 365         \r
 366         @Override\r
 367         protected void handleIndicator(SchemaObject parent, SchemaElement indicator, SchemaElement any) {               \r
 368                 // generates overriding method that allows parsing any element\r
 369                 FileWriter fw = getWriter(parent);\r
 370                 fw.delayedWriter2.println("   @Override");\r
 371                 fw.delayedWriter2.println("   public Class<? extends org.simantics.xml.sax.base.XMLElementParser> getParser(java.util.Map<java.lang.String,org.simantics.xml.sax.base.XMLElementParser> parsers, ParserElement element, ParserElement child) {");\r
 372                 fw.delayedWriter2.println("      Class<? extends org.simantics.xml.sax.base.XMLElementParser> parserClass = super.getParser(parsers, element, child);");\r
 373                 fw.delayedWriter2.println("      if (parserClass != null) return parserClass;");\r
 374                 fw.delayedWriter2.println("      org.simantics.xml.sax.base.XMLElementParser parser = parsers.get(child.getQName());");\r
 375                 fw.delayedWriter2.println("      if (parser != null) return parser.getClass();");\r
 376                 fw.delayedWriter2.println("      return null;");\r
 377                 fw.delayedWriter2.println("   }");\r
 378         }\r
 379 \r
 380         @Override       \r
 381         protected void handle(SchemaObject parent, Attribute attribute) {\r
 382                 String name = attribute.getName();\r
 383                 QName primitiveType = attribute.getType();\r
 384                 SimpleType simpleType = attribute.getSimpleType();\r
 385                 QName ref = attribute.getRef();\r
 386                 \r
 387                 String relationName;\r
 388                 String attrName;\r
 389                 if (name != null) {\r
 390                         attrName = name;\r
 391                         relationName = ontShort+"has"+name;\r
 392                         if (parent != null)\r
 393                                 relationName = ontShort+getName(parent)+"_has"+name;\r
 394                 }\r
 395                 else if (ref != null && parent != null) {\r
 396                         attrName = ref.getLocalPart();\r
 397                         relationName = ontShort+getName(parent)+"_has"+ref.getLocalPart();\r
 398                         \r
 399                         Attribute referred = getRefAttribute(ref);\r
 400                         if (referred != null) {\r
 401                                 primitiveType = referred.getType();\r
 402                                 simpleType = referred.getSimpleType();\r
 403                         }\r
 404                         \r
 405                 } else {\r
 406                         throw new RuntimeException();\r
 407                 }\r
 408                 boolean isReference = false;\r
 409                 if (provider!= null && provider.getAttribute().getName().equals(attrName))\r
 410                         isReference = true;\r
 411                 \r
 412                 FileWriter fw = getWriter(parent);\r
 413                 if (primitiveType != null) {\r
 414                         \r
 415                         //String binding = getBindingFromPrimitiveType(primitiveType);\r
 416                         TypeEntry binding = getTypeEntry(primitiveType);\r
 417                         \r
 418                         if (binding != null) {\r
 419                                 writeAttribute(fw, attrName, relationName, binding, isReference);\r
 420                                 return;\r
 421                     } else {\r
 422                         if (simpleType == null) {\r
 423                                 SchemaObject simpleTypeObj = simpleTypeName.get(primitiveType.getLocalPart());\r
 424                                 if (simpleTypeObj != null)\r
 425                                         simpleType = simpleTypeObj.getSimpleType();\r
 426                         }       \r
 427                     }\r
 428                 } \r
 429                 if (simpleType != null) {\r
 430                         org.w3._2001.xmlschema.List list = simpleType.getList();\r
 431                         if (list != null) {\r
 432                                 TypeEntry binding = getTypeEntry(new QName(SCHEMA_NS, "string"));\r
 433                                 writeAttribute(fw, attrName, relationName, binding, isReference);\r
 434                         } else {\r
 435                                 Restriction restriction = simpleType.getRestriction();\r
 436                                 if (restriction == null)\r
 437                                         throw new RuntimeException("Cannot resolve type for Attribute " + attrName + " -> " + primitiveType.getLocalPart()+ ", SimpleType restriction is unset");\r
 438                                 QName base = restriction.getBase();\r
 439                                 \r
 440                                 \r
 441                                 //String binding = getBindingFromPrimitiveType(base);\r
 442                                 TypeEntry binding = getTypeEntry(base);\r
 443                                 writeAttribute(fw, attrName, relationName, binding, isReference);\r
 444                         }\r
 445                 } else {\r
 446                         // TODO : using default String attribute should be configured with rules.\r
 447                         //throw new RuntimeException("Cannot resolve type for Attribute " + attrName + " -> " + primitiveType.getLocalPart());\r
 448                         fw.writer.println("    //FIXME: Cannot resolve type for Attribute " + attrName + " Using default type String");\r
 449                         //writeAttribute(fw, attrName, relationName, "STRING", isReference);\r
 450                         writeAttribute(fw, attrName, relationName, getTypeEntry("string"), isReference);\r
 451                 }\r
 452         }\r
 453         \r
 454         //private void writeAttribute(FileWriter fw, String attrName, String relationName, String binding, boolean isReference) {\r
 455         private void writeAttribute(FileWriter fw, String attrName, String relationName, TypeEntry binding, boolean isReference) {\r
 456                 fw.writer.println("      {");\r
 457                 fw.writer.println("         Attribute a = element.getAttribute(\"" +attrName+"\");");\r
 458                 fw.writer.println("         if (a != null) {");\r
 459                 fw.writer.println("            graph.claimLiteral(element.getData(),"+relationName+","+getValueGetterMethod(binding,"a")+", "+binding.binding+");");\r
 460                 if (isReference)\r
 461                 fw.writer.println("            idProviderValue = a.getValue();");       \r
 462                 fw.writer.println("         }");\r
 463                 fw.writer.println("      }");\r
 464         }\r
 465         \r
 466         @Override\r
 467         protected void handleAttributes(SchemaObject simpleTypeObj) {\r
 468                 SchemaObject parent = simpleTypeObj.getParent();\r
 469                 FileWriter fw = getWriter(parent);\r
 470                 \r
 471                 SimpleType simpleType = simpleTypeObj.getSimpleType();\r
 472                 Restriction restriction = simpleType.getRestriction();\r
 473                 if (restriction == null)\r
 474                         throw new RuntimeException("Cannot resolve type for Element " + getName(parent));\r
 475                 QName base = restriction.getBase();\r
 476                 \r
 477                 \r
 478                 //String binding = getBindingFromPrimitiveType(base);\r
 479                 TypeEntry binding = getTypeEntry(base);\r
 480                 fw.writer.println("   @Override");\r
 481                 fw.writer.println("   public void configure(WriteGraph graph, ParserElement element, java.lang.String value) throws DatabaseException {");\r
 482                 //fw.writer.println("      graph.claimValue(element.getData(),"+getValueGetter(binding)+", Bindings."+binding+");");\r
 483                 fw.writer.println("      graph.claimValue(element.getData(),"+getValueGetter(binding)+", "+binding.binding +");");\r
 484                 fw.writer.println("   }");\r
 485                 \r
 486         }\r
 487         \r
 488         @Override\r
 489         protected void handle(SchemaObject parent, AttributeGroup attribute) {\r
 490                 if (parent != null) {\r
 491                         FileWriter fw = getWriter(parent);\r
 492                         NamedAttributeGroup group = getAttributeGroup(attribute.getRef().getLocalPart());\r
 493                         fw.writer.println(commentTag+"    AttributeGroup " + group.getName());\r
 494                         SchemaObject obj = new SchemaObject(parent,attribute);\r
 495                         for (Annotated annotated : group.getAttributeOrAttributeGroup()) {\r
 496                                 if (annotated instanceof Attribute) {\r
 497                                         //handle("AttributeGroups_"+group.getName(),(Attribute)annotated);\r
 498                                         handle(obj,(Attribute)annotated);\r
 499                                 } else if (annotated instanceof AttributeGroup) {\r
 500                                         //handle("AttributeGroups_"+group.getName(),(AttributeGroup)annotated);\r
 501                                         handle(obj,(AttributeGroup)annotated);\r
 502                                 }\r
 503                         }\r
 504                         fw.writer.println(commentTag+"    End of AttributeGroup " + group.getName());\r
 505                 }\r
 506                 \r
 507         }\r
 508         \r
 509         @Override\r
 510         protected void handleAttributeComposition(SchemaObject parent, AttributeComposition composition, BijectionMap<org.simantics.xml.sax.configuration.Attribute, Annotated> attributes) {\r
 511                 FileWriter fw = getWriter(parent);\r
 512                 QName type = new QName(CONVERSION_NS, composition.getType());\r
 513                 String arrayBinding = getBindingFromPrimitiveType(type);\r
 514                 String javaType = getJavaTypeFromPrimitiveType(type);\r
 515                 String name = composition.getName();\r
 516                 \r
 517                 String relationName;\r
 518                 if (name != null) {\r
 519                         relationName = ontShort+"has"+name;\r
 520                         if (parent != null)\r
 521                                 relationName = ontShort+getName(parent)+"_has"+name;\r
 522                 } else {\r
 523                         throw new RuntimeException();\r
 524                 }\r
 525                 \r
 526                 fw.writer.println("      {");\r
 527                 int i = 0;\r
 528                 for (org.simantics.xml.sax.configuration.Attribute a : composition.getAttribute())\r
 529                 fw.writer.println("         Attribute a"+(i++)+" = element.getAttribute(\"" +a.getName()+"\");");\r
 530                 fw.writer.print  ("         "+javaType + " value = new "+javaType+"{");\r
 531                 i = 0;\r
 532                 // FIXME : handle optional attributes properly.\r
 533                 for (org.simantics.xml.sax.configuration.Attribute a : composition.getAttribute()) {\r
 534                         Attribute attribute = ((Attribute)attributes.getRight(a));\r
 535                         //QName atype = getBaseType(attribute);\r
 536                         QName atype = getPrimitiveType(attribute);\r
 537                         String defaultValue = attribute.getDefault();\r
 538                         if (defaultValue == null)\r
 539                             defaultValue = getDefaultValue(atype);\r
 540                                 \r
 541                         //String binding = getBindingFromPrimitiveType(atype);\r
 542                         TypeEntry binding = getTypeEntry(atype);\r
 543                         if (i > 0)\r
 544                                 fw.writer.print(",");\r
 545                         if (defaultValue != null)\r
 546                                 fw.writer.print("a"+(i)+"!= null ? "+ getValueGetterMethod(binding,"a"+(i++)) + " : " +defaultValue);\r
 547                         else\r
 548                                 fw.writer.print(getValueGetterMethod(binding,"a"+(i++)));\r
 549                 }\r
 550                 fw.writer.println("};");\r
 551                 fw.writer.println("         graph.claimLiteral(element.getData(),"+relationName+", value, "+arrayBinding+");");\r
 552                 fw.writer.println("      }");\r
 553                 \r
 554         }\r
 555         \r
 556         protected String getDefaultValue(QName atype) {\r
 557                 Map<String,TypeEntry> types = typeMap.get(atype.getNamespaceURI());\r
 558                 if (types == null)\r
 559                         return null;\r
 560                 TypeEntry entry =  types.get(atype.getLocalPart());\r
 561                 if (entry == null)\r
 562                         return null;\r
 563                 return entry.defaultValue;\r
 564         }\r
 565         \r
 566         \r
 567         \r
 568 \r
 569         \r
 570         IDProvider provider;\r
 571         @Override\r
 572         protected void handleElement(SchemaObject elementObj)  {\r
 573                 Element element = elementObj.getElement();\r
 574                 \r
 575                 String name = getName(elementObj);//topLevelElement.getName();\r
 576                 String className = name;\r
 577                 \r
 578                 FileWriter fw = new FileWriter();\r
 579                 try {\r
 580                         fw.writer = createFile(new File(importParserDir.getAbsolutePath()+"/"+className+".java"));\r
 581                 } catch (IOException e) {\r
 582                         throw new RuntimeException(e);\r
 583                 }\r
 584                 writers.put(elementObj, fw);\r
 585                 boolean isList = false;\r
 586                 \r
 587                 Inheritance inhertiance = getElementInheritance(elementObj);    \r
 588                 \r
 589                 provider = getIDProvider(element);\r
 590                 List<IDReference> references = getIDReferences(element);\r
 591                 UnrecognizedChildElement unknownChildElement = getUnknown(element);\r
 592 \r
 593                 List<String> intrerfaces = new ArrayList<String>();\r
 594                 if (references.size() > 0)\r
 595                         intrerfaces.add("org.simantics.xml.sax.base.IDReferenceParser");\r
 596                 if (unknownChildElement != null)\r
 597                         intrerfaces.add("org.simantics.xml.sax.base.UnrecognizedElementParser");\r
 598 \r
 599                 createClassHeader(fw.writer, isList);\r
 600                 writeClass(fw.writer,false, element.getName(), className, inhertiance.baseClass, intrerfaces);\r
 601                 writeIDProvider(fw.writer);\r
 602                 fw.writer.println("   @Override");\r
 603                 fw.writer.println("   public Resource create(WriteGraph graph, ParserElement element) throws DatabaseException{");\r
 604                 fw.writer.println("      Layer0 L0 = Layer0.getInstance(graph);");\r
 605                 fw.writer.println("      "+getOntologyImport());\r
 606                 if (!isList) {\r
 607                 fw.writer.println("      Resource res = graph.newResource();");\r
 608                 fw.writer.println("      graph.claim(res, L0.InstanceOf, "+ontShort+name+");");\r
 609                 } else {\r
 610                 fw.writer.println("      Resource res = ListUtils.create(graph, "+ontShort+name+", Collections.EMPTY_LIST);");\r
 611                 }\r
 612                 fw.writer.println("      return res;");\r
 613                 fw.writer.println("   }");\r
 614                 fw.writer.println();\r
 615                 \r
 616                 fw.writer.println("   @Override");\r
 617                 fw.writer.println("   public boolean connectParent(WriteGraph graph, ParserElement parent, ParserElement element) throws DatabaseException{");\r
 618                 fw.writer.println("      "+getOntologyImport());\r
 619                 fw.writer.println("      graph.claim(parent.getData(), "+this.ontShort+getName(elementObj,"has")+", element.getData());");\r
 620                 fw.writer.println("      return true;");\r
 621                 fw.writer.println("   }");\r
 622                 fw.writer.println();\r
 623                 \r
 624                 fw.writer.println("   @Override");\r
 625                 fw.writer.println("   public void configure(WriteGraph graph, Deque<ParserElement> parents, ParserElement element) throws DatabaseException {");\r
 626                 if (inhertiance.type == InheritanceType.ComplexType) {\r
 627                 fw.writer.println("             super.configure(graph,parents,element);");\r
 628                 }\r
 629                 fw.writer.println("        "+getOntologyImport());\r
 630                 \r
 631                 LocalComplexType complexType = element.getComplexType();\r
 632                 LocalSimpleType simpleType = element.getSimpleType();\r
 633                 \r
 634                 if (complexType != null) {\r
 635                         SchemaObject obj = complexTypes.get(complexType);\r
 636                         handleElementComplexTypeAttributes(obj);\r
 637                 } \r
 638                 fw.writer.println("   }");\r
 639                 \r
 640                 if (inhertiance.type == InheritanceType.AtomicType) {\r
 641                 fw.writer.println();\r
 642                 fw.writer.println("   @Override");\r
 643                 fw.writer.println("   public void configure(WriteGraph graph, ParserElement element, java.lang.String string) throws DatabaseException {");\r
 644                 fw.writer.println("      graph.claimValue(element.getData(), "+getValueGetter(inhertiance.atomicType,"string")+", "+inhertiance.atomicType.binding+");");\r
 645                 fw.writer.println("   }");\r
 646                 }\r
 647 //              if (simpleType != null) {\r
 648 //                      SchemaObject obj = simpleTypes.get(simpleType);\r
 649 //                      handleElementSimpleTypeAttributes(obj);\r
 650 //              }\r
 651                 \r
 652                 \r
 653                 StringWriter stringWriter = new StringWriter();\r
 654                 fw.delayedWriter = new PrintWriter(stringWriter);\r
 655                 StringWriter stringWriter2 = new StringWriter();\r
 656                 fw.delayedWriter2 = new PrintWriter(stringWriter2);\r
 657                 \r
 658                 fw.writer.println("   public " + className + "() {");\r
 659                 fw.writer.println("      super();");\r
 660                 \r
 661                 if (complexType != null) {\r
 662                         SchemaObject obj = complexTypes.get(complexType);\r
 663                         handleComplexTypeExtension(obj);\r
 664                 } else if (simpleType != null) {\r
 665                         \r
 666                 }\r
 667                 \r
 668                 fw.writer.println("   }");\r
 669                 \r
 670                 fw.writer.println("   @Override");\r
 671                 fw.writer.println("   public boolean connectChild(WriteGraph graph, ParserElement element, ParserElement child) throws DatabaseException{");\r
 672                 fw.writer.println("      "+getOntologyImport());\r
 673                 \r
 674                 if (stringWriter.getBuffer().length() > 0) {\r
 675                         fw.writer.write(stringWriter.toString());\r
 676                 }\r
 677                 if (inhertiance.type == InheritanceType.ComplexType) {\r
 678                 fw.writer.println("      return super.connectChild(graph,element,child);");     \r
 679                 } else {\r
 680                 fw.writer.println("      return false;");\r
 681                 }\r
 682                 fw.writer.println("   }");\r
 683                 fw.writer.println();\r
 684                 \r
 685                 if (stringWriter2.getBuffer().length() > 0) {\r
 686                         fw.writer.write(stringWriter2.toString());\r
 687                 }\r
 688                 \r
 689                 stringWriter = null;\r
 690                 fw.delayedWriter.close();\r
 691                 fw.delayedWriter = null;\r
 692                 stringWriter2 = null;\r
 693                 fw.delayedWriter2.close();\r
 694                 fw.delayedWriter2 = null;\r
 695                 \r
 696                 writeIDReferences(fw.writer,name, references);\r
 697                 writeUnknownChild(fw.writer,name,unknownChildElement);\r
 698 \r
 699                 fw.writer.println("}");\r
 700                 fw.writer.println();\r
 701                 fw.writer.flush();\r
 702                 fw.writer.close();\r
 703                 fw.writer = null;\r
 704                 writers.remove(elementObj);\r
 705                 provider = null;\r
 706                 \r
 707                 ruleClassNames.add(converter.getPluginName()+"."+elementPackageName+"."+name);\r
 708         }\r
 709 \r
 710         \r
 711         private void writeIDProvider(PrintWriter writer) {\r
 712                 if (provider != null) {\r
 713                 writer.println("   java.lang.String idProviderValue = null;");\r
 714                 writer.println();\r
 715                 writer.println("   @Override");\r
 716                 writer.println("   public java.lang.String getID() {");\r
 717                 writer.println("        java.lang.String s = idProviderValue;");\r
 718                 writer.println("        idProviderValue = null;");\r
 719                 writer.println("        return s;");\r
 720                 writer.println("   }");\r
 721                 writer.println();\r
 722                 if (provider.getPriority() != null) {\r
 723                 writer.println("   @Override");\r
 724                 writer.println("   public int idPriority() {");\r
 725                 writer.println("        return " + provider.getPriority().intValue()+";");\r
 726                 writer.println("   }"); \r
 727                 }\r
 728                 }\r
 729         }\r
 730         \r
 731         private void writeIDReferences(PrintWriter writer,String name, List<IDReference> references) {\r
 732                 if (references.size() > 0) {\r
 733                 writer.println("   @Override");\r
 734                 writer.println("   public boolean connectReferences(WriteGraph graph, ParserElement element, java.util.Map<java.lang.String, ParserElement> map) throws DatabaseException {");\r
 735                 writer.println("      "+getOntologyImport());\r
 736                 writer.println("      boolean result = true;");\r
 737                 for (IDReference ref : references) {\r
 738                 writer.println("      {");\r
 739                 writer.println("         Attribute a = element.getAttribute(\"" + ref.getIDSource().getName() +"\");");\r
 740                 writer.println("         if (a != null) {");\r
 741                 writer.println("            ParserElement refEle = map.get(a.value);");\r
 742                 writer.println("            if (refEle != null) {");\r
 743                 writer.println("               Resource ref = refEle.getData();");\r
 744                 writer.println("               graph.claim(element.getData(), "+ontShort+name+"_"+ref.getReference().getName()+", ref);");\r
 745                 writer.println("            } else {");\r
 746                 writer.println("               result = false;");\r
 747                 writer.println("            }");\r
 748                 writer.println("         }");\r
 749                 writer.println("      }");\r
 750                 }\r
 751                 writer.println("      return result;");\r
 752                 writer.println("   }");\r
 753                 writer.println();\r
 754                 }\r
 755         }\r
 756         \r
 757         private void writeUnknownChild(PrintWriter writer,String name, UnrecognizedChildElement unknownChildElement) {\r
 758                 if (unknownChildElement == null)\r
 759                         return;\r
 760                 \r
 761                 writer.println("   @Override");\r
 762                 writer.println("   public void configureChild(WriteGraph graph, Deque<ParserElement> parents, ParserElement element, ParserElement child) throws DatabaseException {");\r
 763                 writer.println("    " + unknownChildElement.getJavaMethod());\r
 764                 writer.println("   }");\r
 765                 writer.println();\r
 766                 \r
 767         }\r
 768         \r
 769         @Override\r
 770         protected String getBaseClass() {\r
 771                 return "org.simantics.xml.sax.base.XMLElementNamedChildParserBase";\r
 772         }\r
 773         \r
 774         protected void createClassHeader(PrintWriter writer, boolean isList) {\r
 775                 writer.println("package " + converter.getPluginName() +"."+elementPackageName+";");\r
 776                 writer.println();\r
 777                 writer.println("import java.util.Deque;");\r
 778                 writer.println("import org.simantics.databoard.Bindings;");\r
 779                 writer.println("import org.simantics.db.Resource;");\r
 780                 writer.println("import org.simantics.db.WriteGraph;");\r
 781                 writer.println("import org.simantics.db.exception.DatabaseException;");\r
 782                 writer.println("import org.simantics.xml.sax.base.Attribute;");\r
 783                 writer.println("import org.simantics.xml.sax.base.ParserElement;");\r
 784                 writer.println("import org.simantics.xml.sax.ontology.XMLResource;");\r
 785                 if (!isList) {\r
 786                 writer.println("import org.simantics.layer0.Layer0;");\r
 787             } else {\r
 788             writer.println("import java.util.Collections;");\r
 789                 writer.println("import org.simantics.db.common.utils.ListUtils;");\r
 790                 }\r
 791                 writer.println();\r
 792         }\r
 793         \r
 794 }\r