+package org.simantics.xml.sax;\r
+\r
+import java.io.BufferedReader;\r
+import java.io.BufferedWriter;\r
+import java.io.File;\r
+import java.io.FileReader;\r
+import java.io.FileWriter;\r
+import java.io.IOException;\r
+import java.io.PrintWriter;\r
+import java.util.ArrayList;\r
+import java.util.Collections;\r
+import java.util.HashSet;\r
+import java.util.List;\r
+import java.util.Set;\r
+\r
+public class OntologyCombinator {\r
+ \r
+ \r
+ public static void combine(File[] inputFiles, File outputFile) throws IOException{\r
+ Set<String> resources = new HashSet<>();\r
+ List<String> ontologyUris = new ArrayList<>();\r
+ List<String> classNames = new ArrayList<>();\r
+ String packageLine = null;\r
+ \r
+ \r
+ String packageLineStart = "package";\r
+ String resourceLineStart = "public final Resource ";\r
+ String uriLineStart = "public static final String ";\r
+ \r
+ \r
+ \r
+ for (File inputFile : inputFiles) {\r
+ BufferedReader reader = new BufferedReader(new FileReader(inputFile));\r
+ try {\r
+ String line;\r
+ String uri = null;\r
+ String filename = inputFile.getName();\r
+ String classname = filename.substring(0,filename.length()-5);\r
+ classNames.add(classname);\r
+ while ((line = reader.readLine()) != null) {\r
+ line = line.trim();\r
+ if (line.startsWith(resourceLineStart)) {\r
+ String res = line.substring(resourceLineStart.length(), line.length()-1);\r
+ resources.add(res);\r
+ } else if (line.startsWith(uriLineStart)) {\r
+ String euri = line.substring(line.indexOf('"')+1, line.length()-2);\r
+ if (uri == null) {\r
+ uri = euri;\r
+ } else {\r
+ int max = Math.min(euri.length(), uri.length());\r
+ for (int i = 0; i < max; i++) {\r
+ if (euri.charAt(i) != uri.charAt(i)) {\r
+ uri = uri.substring(0,i);\r
+ break;\r
+ }\r
+ }\r
+ }\r
+ } else if (line.startsWith(packageLineStart)) {\r
+ packageLine = line;\r
+ }\r
+ }\r
+ if (uri != null && uri.length() > 6) {\r
+ if (uri.endsWith("/")) {\r
+ uri = uri.substring(0, uri.length()-1);\r
+ }\r
+ ontologyUris.add(uri);\r
+ }\r
+ else\r
+ throw new IOException("Could not resolve ontology uri for file " + inputFile.getName());\r
+ } finally {\r
+ reader.close();\r
+ }\r
+ }\r
+ List<String> sortedResourcses = new ArrayList<>();\r
+ sortedResourcses.addAll(resources);\r
+ Collections.sort(sortedResourcses);\r
+ \r
+ PrintWriter writer = new PrintWriter(new BufferedWriter(new FileWriter(outputFile)));\r
+ String className = outputFile.getName();\r
+ int dp = className.indexOf(".");\r
+ if (dp > 0)\r
+ className = className.substring(0,dp);\r
+ \r
+ try {\r
+ writer.println(packageLine);\r
+ writer.println();\r
+ writer.println("import java.lang.reflect.Field;");\r
+ writer.println("import java.util.Collection;");\r
+ writer.println("import java.util.HashMap;");\r
+ writer.println("import java.util.Map;");\r
+ writer.println("import org.simantics.Simantics;");\r
+ writer.println("import org.simantics.db.ReadGraph;");\r
+ writer.println("import org.simantics.db.Resource;");\r
+ writer.println("import org.simantics.db.exception.DatabaseException;");\r
+ writer.println("import org.simantics.db.request.Read;");\r
+ writer.println("import org.simantics.layer0.Layer0;");\r
+ writer.println();\r
+ writer.println();\r
+ writer.println("public class " + className + " {");\r
+ writer.println(" private static boolean init = false;");\r
+ writer.println(" private static Map<Resource, Object> resourceObjects = new HashMap<>();");\r
+ writer.println(" private static Map<Resource, "+className+"> loaderObjects = new HashMap<>();");\r
+ writer.println();\r
+ writer.println(" private Resource ontologyResource;");\r
+ writer.println(" private Object resourceObject;");\r
+ writer.println();\r
+ writer.println(" static void init(ReadGraph graph) throws DatabaseException{");\r
+ writer.println(" if (init)");\r
+ writer.println(" return;");\r
+ writer.println(" ");\r
+ for (int i = 0; i < inputFiles.length; i++) {\r
+ writer.println(" add(graph.getResource(\"" +ontologyUris.get(i) +"\"), " + classNames.get(i) +".getInstance(graph));");\r
+ }\r
+ writer.println(" ");\r
+ writer.println(" init = true;");\r
+ writer.println(" }");\r
+ writer.println(" ");\r
+ writer.println(" public static void add(Resource resource, Object object) {");\r
+ writer.println(" if (resource == null || object == null)");\r
+ writer.println(" throw new IllegalArgumentException();");\r
+ writer.println(" resourceObjects.put(resource, object);");\r
+ writer.println(" loaderObjects.put(resource, new "+className+"(resource, object));");\r
+ writer.println(" }");\r
+ writer.println(" ");\r
+ writer.println(" public static "+className+" getOntology(final Resource res) throws DatabaseException {");\r
+ writer.println(" return Simantics.getSession().syncRequest(new Read<"+className+">() {");\r
+ writer.println(" @Override");\r
+ writer.println(" public "+className+" perform(ReadGraph graph) throws DatabaseException {");\r
+ writer.println(" return getOntology(graph, res);");\r
+ writer.println(" }");\r
+ writer.println(" });");\r
+ writer.println(" }");\r
+ writer.println(" ");\r
+ writer.println(" public static "+className+" getOntology(ReadGraph graph, Resource res) throws DatabaseException{");\r
+ writer.println(" init(graph);");\r
+ writer.println(" Layer0 L0 = Layer0.getInstance(graph);");\r
+ writer.println(" Collection<Resource> types = graph.getTypes(res);");\r
+ writer.println(" for (Resource type : types) {");\r
+ writer.println(" Resource part = graph.getPossibleObject(type, L0.PartOf);");\r
+ writer.println(" while (part != null) {");\r
+ writer.println(" if (loaderObjects.containsKey(part))");\r
+ writer.println(" return loaderObjects.get(part);");\r
+ writer.println(" part = graph.getPossibleObject(part, L0.PartOf);");\r
+ writer.println(" }");\r
+ writer.println(" }");\r
+ writer.println(" return null;");\r
+ writer.println(" }");\r
+ writer.println(" ");\r
+ writer.println(" private "+className+"(Resource ontologyRes, Object resourceObject) {");\r
+ writer.println(" this.ontologyResource = ontologyRes;");\r
+ writer.println(" this.resourceObject = resourceObject;");\r
+ writer.println(" init();");\r
+ writer.println(" }");\r
+ writer.println(" ");\r
+ writer.println(" private Resource get(String name) {");\r
+ writer.println(" try {");\r
+ writer.println(" Field field = resourceObject.getClass().getDeclaredField(name);");\r
+ writer.println(" return (Resource)field.get(resourceObject);");\r
+ writer.println(" } catch (Exception e){");\r
+ writer.println(" return null;");\r
+ writer.println(" }");\r
+ writer.println(" }");\r
+ writer.println(" ");\r
+ for (String s : sortedResourcses) {\r
+ writer.println(" public Resource "+s+";");\r
+ }\r
+ writer.println(" ");\r
+ writer.println(" private void init() {");\r
+ for (String s : sortedResourcses) {\r
+ writer.println(" "+s+" = get(\""+s+"\");");\r
+ }\r
+ writer.println(" }");\r
+ writer.println("}");\r
+ \r
+ \r
+ } finally {\r
+ writer.close();\r
+ }\r
+ \r
+ }\r
+}\r