From 64138e8d91bc8d7bb54d9b042f872b43550dec16 Mon Sep 17 00:00:00 2001
From: Kristóf Marussy <kris7topher@gmail.com>
Date: Wed, 24 Jul 2019 10:59:02 +0200
Subject: Cardinality propagator WIP

---
 .../MultiplicityGoalConstraintCalculator.xtend     | 46 ++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra/src/hu/bme/mit/inf/dslreasoner/viatrasolver/logic2viatra/cardinality/MultiplicityGoalConstraintCalculator.xtend

(limited to 'Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra/src/hu/bme/mit/inf/dslreasoner/viatrasolver/logic2viatra/cardinality/MultiplicityGoalConstraintCalculator.xtend')

diff --git a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra/src/hu/bme/mit/inf/dslreasoner/viatrasolver/logic2viatra/cardinality/MultiplicityGoalConstraintCalculator.xtend b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra/src/hu/bme/mit/inf/dslreasoner/viatrasolver/logic2viatra/cardinality/MultiplicityGoalConstraintCalculator.xtend
new file mode 100644
index 00000000..86a59aa1
--- /dev/null
+++ b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra/src/hu/bme/mit/inf/dslreasoner/viatrasolver/logic2viatra/cardinality/MultiplicityGoalConstraintCalculator.xtend
@@ -0,0 +1,46 @@
+package hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.cardinality
+
+import org.eclipse.emf.common.notify.Notifier
+import org.eclipse.viatra.query.runtime.api.IQuerySpecification
+import org.eclipse.viatra.query.runtime.api.ViatraQueryEngine
+import org.eclipse.viatra.query.runtime.api.ViatraQueryMatcher
+import org.eclipse.viatra.query.runtime.emf.EMFScope
+
+class MultiplicityGoalConstraintCalculator {
+	val String targetRelationName;
+	val IQuerySpecification<?> querySpecification;
+	var ViatraQueryMatcher<?> matcher;
+		
+	new(String targetRelationName, IQuerySpecification<?> querySpecification) {
+		this.targetRelationName = targetRelationName
+		this.querySpecification = querySpecification
+		this.matcher = null
+	}
+	
+	new(MultiplicityGoalConstraintCalculator other) {
+		this.targetRelationName = other.targetRelationName
+		this.querySpecification = other.querySpecification
+		this.matcher = null
+	}
+	
+	def getName() {
+		targetRelationName
+	}
+	
+	def init(Notifier notifier) {
+		val engine = ViatraQueryEngine.on(new EMFScope(notifier))
+		matcher = querySpecification.getMatcher(engine)
+	}
+	
+	def calculateValue() {
+		var res = 0
+		val allMatches = this.matcher.allMatches
+		for(match : allMatches) {
+			//println(targetRelationName+ " missing multiplicity: "+match.get(3))
+			val missingMultiplicity = match.get(2) as Integer
+			res += missingMultiplicity
+		}
+		//println(targetRelationName+ " all missing multiplicities: "+res)
+		return res
+	}
+}
-- 
cgit v1.2.3-70-g09d2