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