Trying to simplify Satellite case study

1 files changed, 31 insertions, 0 deletions

diff --git a/Domains/hu.bme.mit.inf.dslreasoner.domains.satellite/src/hu/bme/mit/inf/dslreasoner/domains/satellite/mdeo/LocalSearchEngineManager.xtend b/Domains/hu.bme.mit.inf.dslreasoner.domains.satellite/src/hu/bme/mit/inf/dslreasoner/domains/satellite/mdeo/LocalSearchEngineManager.xtend
new file mode 100644
index 00000000..ee7f0060
--- /dev/null
+++ b/Domains/hu.bme.mit.inf.dslreasoner.domains.satellite/src/hu/bme/mit/inf/dslreasoner/domains/satellite/mdeo/LocalSearchEngineManager.xtend
@@ -0,0 +1,31 @@
+package hu.bme.mit.inf.dslreasoner.domains.satellite.mdeo
+
+import hu.bme.mit.inf.dslreasoner.domains.satellite.queries.SatelliteQueries
+import java.util.WeakHashMap
+import org.eclipse.emf.ecore.EObject
+import org.eclipse.viatra.query.runtime.api.AdvancedViatraQueryEngine
+import org.eclipse.viatra.query.runtime.api.ViatraQueryEngine
+import org.eclipse.viatra.query.runtime.api.ViatraQueryEngineOptions
+import org.eclipse.viatra.query.runtime.emf.EMFScope
+import org.eclipse.viatra.query.runtime.localsearch.matcher.integration.LocalSearchHints
+
+class LocalSearchEngineManager {
+        public static val INSTANCE = new LocalSearchEngineManager
+
+        val WeakHashMap<EObject, ViatraQueryEngine> engineMap = new WeakHashMap
+
+        private new() {
+        }
+
+        def getEngine(EObject eObject) {
+                engineMap.computeIfAbsent(eObject) [
+                        val scope = new EMFScope(it)
+                        val localSearchHints = LocalSearchHints.^default.build
+                        val options = ViatraQueryEngineOptions.defineOptions.withDefaultHint(localSearchHints).withDefaultBackend(
+                                localSearchHints.queryBackendFactory).build
+                        val engine = AdvancedViatraQueryEngine.on(scope, options)
+                        SatelliteQueries.instance.prepare(engine)
+                        engine
+                ]
+        }
+}