1 files changed, 46 insertions, 0 deletions

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

+        }

+}