Handling ComplexTypes defined with SimpleContent

[simantics/interop.git] / org.simantics.xml.sax.base / src / org / simantics / xml / sax / base / XMLWriter.java

diff --git a/org.simantics.xml.sax.base/src/org/simantics/xml/sax/base/XMLWriter.java b/org.simantics.xml.sax.base/src/org/simantics/xml/sax/base/XMLWriter.java
index 0881a5d109d3b3b56ba7e287e594da67c282078e..f37e4d8e2ce438094aa224f1099a2e46f8163830 100644 (file)
--- a/org.simantics.xml.sax.base/src/org/simantics/xml/sax/base/XMLWriter.java
+++ b/org.simantics.xml.sax.base/src/org/simantics/xml/sax/base/XMLWriter.java
@@ -2,7 +2,6 @@ package org.simantics.xml.sax.base;
 \r
 import java.lang.reflect.Constructor;\r
 import java.lang.reflect.InvocationTargetException;\r
-import java.util.Collection;\r
 import java.util.HashMap;\r
 import java.util.HashSet;\r
 import java.util.LinkedHashSet;\r
@@ -21,16 +20,19 @@ import org.simantics.db.Statement;
 import org.simantics.db.common.utils.ListUtils;\r
 import org.simantics.db.common.utils.NameUtils;\r
 import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.layer0.Layer0;\r
 import org.simantics.message.ILogger;\r
 import org.simantics.xml.sax.ontology.XMLResource;\r
 \r
 public class XMLWriter {\r
        \r
        private ReadGraph graph;\r
-       private Map<String, XMLWriter> subWriters = new HashMap<String, XMLWriter>();\r
+       private Map<Resource, XMLWriter> subWriters = new HashMap<Resource, XMLWriter>();\r
        private Map<Class<? extends XMLElementWriter>, XMLElementWriter> namedWriters = new HashMap<Class<? extends XMLElementWriter>, XMLElementWriter>();\r
        private Map<Resource,XMLElementWriter> writers = new HashMap<>();\r
        private String schemaURI;\r
+       private String ontologyURI;\r
+       private Resource ontology;\r
        \r
        private ILogger logger;\r
        \r
@@ -38,10 +40,12 @@ public class XMLWriter {
                return graph;\r
        }\r
        \r
-       public void setGraph(ReadGraph graph) {\r
+       public void setGraph(ReadGraph graph) throws DatabaseException {\r
                this.graph = graph;\r
                for (XMLWriter p : subWriters.values())\r
                        p.setGraph(graph);\r
+               if (ontologyURI != null)\r
+                       this.ontology = graph.getResource(ontologyURI);\r
        }\r
        \r
        public String getSchemaURI() {\r
@@ -52,6 +56,18 @@ public class XMLWriter {
                this.schemaURI = schemaURI;\r
        }\r
        \r
+       public String getOntologyURI() {\r
+               return ontologyURI;\r
+       }\r
+       \r
+       public void setOntologyURI(String ontologyURI) {\r
+               this.ontologyURI = ontologyURI;\r
+       }\r
+       \r
+       public Resource getOntology() {\r
+               return ontology;\r
+       }\r
+       \r
        public void add(XMLElementWriter writer) throws DatabaseException {\r
                Resource type = writer.getType(graph);\r
                if (type != null)\r
@@ -60,7 +76,7 @@ public class XMLWriter {
        }\r
        \r
        public void add(XMLWriter writer) {\r
-               subWriters.put(writer.getSchemaURI(), writer);\r
+               subWriters.put(writer.getOntology(), writer);\r
        }\r
        \r
        public void write(Resource root, XMLStreamWriter writer) throws DatabaseException, XMLStreamException {\r
@@ -84,7 +100,9 @@ public class XMLWriter {
                if (graph.hasValue(instance.instance))\r
                        elementWriter.characters(graph, instance, writer);\r
                // get all child elements\r
-               Collection<Statement> childElements = graph.getStatements(instance.instance, XML.hasElement);\r
+               Set<Statement> childElements = new HashSet<>();\r
+               childElements.addAll(graph.getStatements(instance.instance, XML.hasElement));\r
+               childElements.addAll(graph.getStatements(instance.instance, XML.hasComplexType));\r
                // load elements, assign writers\r
                Map<Resource,WriterElement> elementMap = new HashMap<>();\r
                for (Statement s : childElements) {\r
@@ -104,6 +122,8 @@ public class XMLWriter {
                        if (processed.contains(r)) // badly generated writer could report elements several times. \r
                                continue;\r
                        WriterElement child = elementMap.get(r);\r
+                       if (child == null)\r
+                               throw new DatabaseException("Trying to export unregonized resource " +NameUtils.getSafeName(graph, r) + " " + r);\r
                        write(child, writer);\r
                        processed.add(r);\r
                }\r
@@ -112,6 +132,8 @@ public class XMLWriter {
                        if (processed.contains(stm.getObject()))\r
                                continue;\r
                        WriterElement child = elementMap.get(stm.getObject());\r
+                       if (child == null)\r
+                               throw new DatabaseException("Trying to export unregonized resource " +NameUtils.getSafeName(graph, stm.getObject()) + " " + stm.getObject());\r
                        write(child, writer);\r
                }\r
        \r
@@ -128,13 +150,20 @@ public class XMLWriter {
                                childWriter = this.writers.get(childWriterClass);\r
                                if (childWriter == null) {\r
                                        try {\r
-                                               Constructor<? extends XMLElementWriter> c = childWriterClass.getConstructor(ReadGraph.class);\r
+                                               Constructor<? extends XMLElementWriter> c = null;\r
+                                               try {\r
+                                                       c = childWriterClass.getConstructor(ReadGraph.class);\r
+                                                       childWriter = c.newInstance(graph);\r
+                                               } catch (NoSuchMethodException e) {\r
+                                                       c = childWriterClass.getConstructor();\r
+                                                       childWriter = c.newInstance();\r
+                                               }\r
                                                //childWriter = childWriterClass.newInstance();\r
-                                               childWriter = c.newInstance(graph);\r
+                                               \r
                                                namedWriters.put(childWriterClass, childWriter);\r
                                        } catch (IllegalAccessException|InstantiationException|NoSuchMethodException|SecurityException|InvocationTargetException e) {\r
                                                String err = "Error processing " + childWriterClass.getName() + " : element writers must have accessible constructor with ReadGraph parameter";\r
-                                               logger.log(new Status(IStatus.ERROR, XMLParser.PLUGIN_ID, err));\r
+                                               logger.log(new Status(IStatus.ERROR, XMLParser.PLUGIN_ID, err, e));\r
                                        } \r
                                }\r
                        }\r
@@ -143,10 +172,30 @@ public class XMLWriter {
                        childWriter = writers.get(type);\r
                }\r
                if (childWriter == null) {\r
-                       throw new DatabaseException("Cannot locate writer for " + NameUtils.getSafeName(graph, child.instance) + ", " + child.instance);\r
+                       Resource type = graph.getSingleType(child.instance);\r
+                       Resource ontology = getOntology(type);\r
+                       if (ontology != null) {\r
+                               XMLWriter xmlWriter = subWriters.get(ontology);\r
+                               if (xmlWriter != null) {\r
+                                       childWriter = xmlWriter.writers.get(type);\r
+                               }\r
+                       }\r
+                       if (childWriter == null)\r
+                               throw new DatabaseException("Cannot locate writer for " + NameUtils.getSafeName(graph, child.instance) + ", " + child.instance);\r
                }\r
                child.writer = childWriter;\r
                \r
        }\r
+       \r
+       private Resource getOntology(Resource type) throws DatabaseException {\r
+               Layer0 L0 = Layer0.getInstance(graph);\r
+               Resource r = type;\r
+               while (true) {\r
+                       r = graph.getPossibleObject(r, L0.PartOf);\r
+                       if (r != null && graph.isInstanceOf(r, L0.Ontology))\r
+                               break;\r
+               }\r
+               return r;\r
+       }\r
 \r
 }\r