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