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