1 package org.simantics.db.impl.query;
4 import java.io.IOException;
5 import java.io.UnsupportedEncodingException;
6 import java.net.MalformedURLException;
8 import java.net.URLDecoder;
10 import org.simantics.db.ObjectResourceIdMap;
11 import org.simantics.utils.FileUtils;
13 public class CodeGen {
19 String[] signatureR1RelationInfo = { "int r", "r", "keyR", "long", "InternalProcedure<RelationInfo>", "entry.id" };
20 String[] signatureR1Bytes = { "int r", "r", "keyR", "long", "InternalProcedure<byte[]>", "entry.id" };
21 String[] signatureR1IntSet = { "int r", "r", "keyR", "long", "InternalProcedure<IntSet>", "entry.id" };
22 String[] signatureR1IP = { "int r", "r", "keyR", "long", "IntProcedure", "entry.id" };
23 String[] signatureR2IP = { "int r1, int r2", "r1,r2", "keyR2", "long", "IntProcedure", "entry.id" };
24 String[] signatureR2TIP = { "int r1, int r2", "r1,r2", "keyR2", "long", "TripleIntProcedure", "entry.id" };
25 String[] signatureID1 = { "String id", "id", "keyID", "String", "InternalProcedure<Integer>", "entry.id" };
26 String[] signatureID2 = { "String id", "id", "keyID", "String", "InternalProcedure<TObjectIntHashMap<String>>", "entry.id" };
27 String[] signatureChildMap = { "int r", "r", "keyR", "long", "InternalProcedure<ObjectResourceIdMap<String>>", "entry.id" };
28 String[] signatureRead = { "Read<?> r", "r", "id", "long", "AsyncProcedure", "entry.request" };
29 String[] signatureAsyncRead = { "AsyncRead<?> r", "r", "id", "long", "AsyncProcedure", "entry.request" };
30 String[] signatureMultiRead = { "MultiRead<?> r", "r", "id", "long", "AsyncMultiProcedure", "entry.request" };
31 String[] signatureAsyncMultiRead = { "AsyncMultiRead<?> r", "r", "id", "long", "AsyncMultiProcedure", "entry.request" };
32 String[] signatureExternalRead = { "ExternalRead<?> r", "r", "id", "long", "AsyncProcedure", "entry.request" };
34 private void line(StringBuilder content, String line) {
35 for(int i=0;i<indent;i++)
41 public void generateQuery(StringBuilder content, String clazz, String[] signature, boolean runnerShortcut) {
42 generateGetOrCreate(content, clazz, signature);
43 generateRemove(content, clazz, signature);
44 generateRunner(content, clazz, signature, runnerShortcut);
47 public void generateRunner(StringBuilder content, String clazz, String[] signature, boolean shortcut) {
49 line(content, "public static void runner" + clazz + "(ReadGraphImpl graph, " + signature[0] + ", CacheEntry parent, ListenerBase listener, " + signature[4] + " procedure) throws DatabaseException {");
50 line(content, " QueryCache cache = graph.processor.cache;");
52 line(content, " if(parent == null && listener == null && !cache.shouldCache(graph.processor, " + signature[1] + ")) {");
53 line(content, " " + clazz + ".computeForEach(graph, " + signature[1] + ", null, procedure);");
54 line(content, " return;");
57 line(content, " if(procedure == null) procedure = emptyProcedure" + clazz + ";");
58 line(content, " " + clazz + " entry = (" + clazz + ")cache.getOrCreate" + clazz + "(" + signature[1] + ");");
59 line(content, " ListenerEntry listenerEntry = cache.registerDependencies(graph, entry, parent, listener, procedure, false);");
60 line(content, " if(entry.isReady()) entry.performFromCache(graph, procedure);");
61 line(content, " else {");
62 if(shortcut) line(content, " " + clazz + ".computeForEach(graph, " + signature[1] + ", entry, procedure);");
63 else line(content, " entry.compute(graph, procedure);");
64 line(content, " if(listenerEntry != null) cache.primeListenerEntry(listenerEntry, entry.getResult());");
71 public void generateRemove(StringBuilder content, String clazz, String[] signature) {
73 String lower = Character.toLowerCase(clazz.charAt(0)) + clazz.substring(1);
75 line(content, "void remove(" + clazz + " entry) {");
76 line(content, " synchronized(" + lower + "Map) {");
77 line(content, " " + lower + "Map.remove(" + signature[5] + ");");
84 public void generateGetOrCreate(StringBuilder content, String clazz, String[] signature) {
86 String lower = Character.toLowerCase(clazz.charAt(0)) + clazz.substring(1);
88 line(content, "" + clazz + " getOrCreate" + clazz + "(" + signature[0] + ") throws DatabaseException {");
89 line(content, " " + clazz + " existing = null;");
90 line(content, " synchronized(" + lower + "Map) {");
91 line(content, " existing = (" + clazz + ")" + lower + "Map.get(" + signature[1] + ");");
92 line(content, " if(existing == null) {");
93 line(content, " existing = new " + clazz + "(" + signature[1] + ");");
94 line(content, " existing.clearResult(querySupport);");
95 line(content, " existing.setPending();");
96 line(content, " " + lower + "Map.put(" + signature[2] + "(" + signature[1] + "), existing);");
97 line(content, " size++;");
98 line(content, " return existing;");
100 line(content, " if(existing.requiresComputation()) {");
101 line(content, " existing.setPending();");
102 line(content, " return existing;");
105 line(content, " if(existing.isPending()) waitPending(existing);");
106 line(content, " return existing;");
112 public void generate() {
114 URL classLocation = CodeGen.class.getResource(".");
115 if (classLocation != null) {
116 if (classLocation.getProtocol().equals("file")) {
118 URL resource = new URL(classLocation, ".");
119 File path = new File(URLDecoder.decode(resource.getPath(), "UTF-8"));
120 String target = path.getAbsolutePath().replace("\\", "/");
121 target = target.replace("/bin/", "/src/") + "/QueryCache.java";
122 System.err.println("target=" + target);
123 File source = new File(target);
124 StringBuilder content = new StringBuilder();
125 content.append("package org.simantics.db.impl.query;\n");
126 content.append("\n");
128 content.append("import org.simantics.db.ObjectResourceIdMap;\n");
129 content.append("import org.simantics.db.RelationInfo;\n");
130 content.append("import org.simantics.db.exception.DatabaseException;\n");
131 content.append("import org.simantics.db.impl.graph.ReadGraphImpl;\n");
132 content.append("import org.simantics.db.impl.procedure.InternalProcedure;\n");
133 content.append("import org.simantics.db.procedure.AsyncMultiProcedure;\n");
134 content.append("import org.simantics.db.procedure.AsyncProcedure;\n");
135 content.append("import org.simantics.db.procedure.ListenerBase;\n");
136 content.append("import org.simantics.db.request.AsyncMultiRead;\n");
137 content.append("import org.simantics.db.request.AsyncRead;\n");
138 content.append("import org.simantics.db.request.ExternalRead;\n");
139 content.append("import org.simantics.db.request.MultiRead;\n");
140 content.append("import org.simantics.db.request.Read;\n");
141 content.append("\n");
142 content.append("import gnu.trove.map.hash.TObjectIntHashMap;\n");
143 content.append("\n");
145 content.append("public class QueryCache extends QueryCacheBase {\n");
146 content.append("\n");
147 line(content,"public QueryCache(QuerySupport querySupport, int threads) {");
148 line(content," super(querySupport, threads);");
150 content.append("\n");
152 generateQuery(content, "Objects", signatureR2IP, true);
153 generateQuery(content, "Statements", signatureR2TIP, true);
154 generateQuery(content, "DirectObjects", signatureR2IP, true);
155 generateQuery(content, "RelationInfoQuery", signatureR1RelationInfo, true);
156 generateQuery(content, "URIToResource", signatureID1, true);
157 generateQuery(content, "ValueQuery", signatureR1Bytes, true);
158 generateQuery(content, "OrderedSet", signatureR1IP, true);
159 generateQuery(content, "PrincipalTypes", signatureR1IP, true);
160 generateQuery(content, "DirectPredicates", signatureR1IntSet, true);
161 generateQuery(content, "Predicates", signatureR1IntSet, true);
162 generateQuery(content, "ReadEntry", signatureRead, true);
163 generateQuery(content, "AsyncReadEntry", signatureAsyncRead, true);
164 generateQuery(content, "Types", signatureR1IntSet, true);
165 //generateQuery(content, "NamespaceIndex", signatureID2, true);
166 generateQuery(content, "ChildMap", signatureChildMap, true);
168 generateQuery(content, "AssertedStatements", signatureR2TIP, false);
169 generateQuery(content, "AssertedPredicates", signatureR1IP, false);
170 generateQuery(content, "DirectSuperRelations", signatureR1IP, false);
171 generateQuery(content, "SuperTypes", signatureR1IntSet, false);
172 generateQuery(content, "TypeHierarchy", signatureR1IntSet, false);
173 generateQuery(content, "SuperRelations", signatureR1IntSet, false);
174 generateQuery(content, "MultiReadEntry", signatureMultiRead, false);
175 generateQuery(content, "AsyncMultiReadEntry", signatureAsyncMultiRead, false);
176 generateQuery(content, "ExternalReadEntry", signatureExternalRead, false);
177 content.append("}\n");
178 FileUtils.writeFile(source, content.toString().getBytes());
179 } catch (MalformedURLException e) {
181 } catch (UnsupportedEncodingException e) {
183 } catch (IOException e) {
191 public static void main(String[] args) {
192 new CodeGen().generate();