1 files changed, 201 insertions, 0 deletions

diff --git a/Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/CloseTrainsObjectiveHint.xtend b/Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/CloseTrainsObjectiveHint.xtend
new file mode 100644
index 00000000..519a228a
--- /dev/null
+++ b/Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/CloseTrainsObjectiveHint.xtend
@@ -0,0 +1,201 @@
+package modes3.run
+
+import hu.bme.mit.inf.dslreasoner.ecore2logic.Ecore2Logic
+import hu.bme.mit.inf.dslreasoner.ecore2logic.Ecore2Logic_Trace
+import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.Type
+import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.cardinality.ExtendedLinearExpressionBuilderFactory
+import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.optimization.CostElementMatch
+import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.optimization.CostElementMatchers
+import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.optimization.CostObjectiveHint
+import java.util.Collection
+import java.util.Map
+import modes3.Modes3Package
+import modes3.queries.CloseTrains_step_2
+import modes3.queries.CloseTrains_step_3
+import modes3.queries.CloseTrains_step_4
+import modes3.queries.CloseTrains_step_5
+import modes3.queries.CloseTrains_step_6
+import modes3.queries.CloseTrains_step_7
+
+class CloseTrainsObjectiveHint extends CostObjectiveHint {
+        val Type segmentType
+        val Type trainType
+
+        new(extension Ecore2Logic ecore2Logic, Ecore2Logic_Trace ecore2LogicTrace) {
+                extension val Modes3Package = Modes3Package.eINSTANCE
+                segmentType = ecore2LogicTrace.TypeofEClass(segment)
+                trainType = ecore2LogicTrace.TypeofEClass(train)
+        }
+
+        override isExact() {
+                true
+        }
+
+        override createPolyhedronExtensionOperator(Map<String, CostElementMatchers> costElementMatchers) {
+                val step2 = costElementMatchers.get(CloseTrains_step_2.instance.fullyQualifiedName)
+                val step3 = costElementMatchers.get(CloseTrains_step_3.instance.fullyQualifiedName)
+                val step4 = costElementMatchers.get(CloseTrains_step_4.instance.fullyQualifiedName)
+                val step5 = costElementMatchers.get(CloseTrains_step_5.instance.fullyQualifiedName)
+                val step6 = costElementMatchers.get(CloseTrains_step_6.instance.fullyQualifiedName)
+                val step7 = costElementMatchers.get(CloseTrains_step_7.instance.fullyQualifiedName);
+
+                [
+                        val objectiveBuilder = createBuilder
+
+                        for (m : step2.matches) {
+                                val dimension = getDimension(m.match)
+                                objectiveBuilder.add(step2.weight, dimension)
+                                dimension.tightenLowerBound(0)
+                                if (m.multi) {
+                                        createBuilder.add(1, dimension).add(-1, trainType).build.assertEqualsTo(0)
+                                } else {
+                                        dimension.tightenUpperBound(1)
+                                        if (m.must) {
+                                                dimension.tightenLowerBound(1)
+                                        }
+                                }
+                        }
+
+                        val step3Matches = step3.matches
+                        for (m : step3Matches) {
+                                val dimension = getDimension(m.match)
+                                objectiveBuilder.add(step3.weight, dimension)
+                                dimension.tightenLowerBound(0)
+                                if (!m.multi) {
+                                        dimension.tightenUpperBound(1)
+                                        if (m.must) {
+                                                dimension.tightenLowerBound(1)
+                                        }
+                                }
+                        }
+                        for (pair : step3Matches.groupBy[step2.projectMayMatch(match, 2)].entrySet) {
+                                val multiplicityBuilder = createBuilder
+                                for (m : pair.value) {
+                                        multiplicityBuilder.add(1, m.match)
+                                }
+                                multiplicityBuilder.add(-1, pair.key)
+                                multiplicityBuilder.build.assertEqualsTo(0)
+                        }
+                        boundLimit(step3Matches, 2, trainType, 1)
+                        boundLimit(step3Matches, 3, segmentType, 1)
+
+                        val step4Matches = step4.matches
+                        for (m : step4Matches) {
+                                val dimension = getDimension(m.match)
+                                objectiveBuilder.add(step4.weight, dimension)
+                                dimension.tightenLowerBound(0)
+                                if (!m.multi) {
+                                        dimension.tightenUpperBound(1)
+                                        if (m.must) {
+                                                dimension.tightenLowerBound(1)
+                                        }
+                                }
+                        }
+                        for (pair : step4Matches.groupBy[step3.projectMayMatch(match, 2, 3)].entrySet) {
+                                val multiplicityBuilder = createBuilder
+                                for (m : pair.value) {
+                                        multiplicityBuilder.add(1, m.match)
+                                }
+                                multiplicityBuilder.add(-2, pair.key)
+                                multiplicityBuilder.build.tightenUpperBound(0)
+                        }
+                        boundLimit(step4Matches, 2, trainType, 2)
+                        boundLimit(step4Matches, 3, segmentType, 2)
+                        boundLimit(step4Matches, 4, segmentType, 2)
+
+                        val step5Matches = step5.matches
+                        for (m : step5Matches) {
+                                val dimension = getDimension(m.match)
+                                objectiveBuilder.add(step5.weight, dimension)
+                                dimension.tightenLowerBound(0)
+                                if (!m.multi) {
+                                        dimension.tightenUpperBound(1)
+                                        if (m.must) {
+                                                dimension.tightenLowerBound(1)
+                                        }
+                                }
+                        }
+                        for (pair : step5Matches.groupBy[step4.projectMayMatch(match, 2, 3, 4)].entrySet) {
+                                val multiplicityBuilder = createBuilder
+                                for (m : pair.value) {
+                                        multiplicityBuilder.add(1, m.match)
+                                }
+                                multiplicityBuilder.add(-2, pair.key)
+                                multiplicityBuilder.build.tightenUpperBound(0)
+                        }
+                        boundLimit(step5Matches, 2, trainType, 4)
+                        boundLimit(step5Matches, 3, segmentType, 4)
+                        boundLimit(step5Matches, 4, segmentType, 4)
+                        boundLimit(step5Matches, 5, segmentType, 4)
+
+                        val step6Matches = step6.matches
+                        for (m : step6Matches) {
+                                val dimension = getDimension(m.match)
+                                objectiveBuilder.add(step6.weight, dimension)
+                                dimension.tightenLowerBound(0)
+                                if (m.multi) {
+                                        if (m.match.get(3) == m.match.get(5)) {
+                                                createBuilder.add(2, m.match).add(-1, step5.projectMayMatch(m.match, 2, 3, 4, 5)).build.
+                                                        assertEqualsTo(0)
+                                        } else {
+                                                createBuilder.add(1, m.match).add(-1, step5.projectMayMatch(m.match, 2, 3, 4, 5)).build.
+                                                        assertEqualsTo(0)
+                                        }
+                                } else {
+                                        dimension.tightenUpperBound(1)
+                                        if (m.must) {
+                                                dimension.tightenLowerBound(1)
+                                        }
+                                }
+                        }
+                        boundLimit(step6Matches, 2, trainType, 2)
+                        boundLimit(step6Matches, 3, segmentType, 2)
+                        boundLimit(step6Matches, 4, segmentType, 2)
+                        boundLimit(step6Matches, 5, segmentType, 2)
+
+                        val step7Matches = step7.matches
+                        for (m : step7Matches) {
+                                val dimension = getDimension(m.match)
+                                objectiveBuilder.add(step7.weight, dimension)
+                                dimension.tightenLowerBound(0)
+                                if (!m.multi) {
+                                        dimension.tightenUpperBound(1)
+                                        if (m.must) {
+                                                dimension.tightenLowerBound(1)
+                                        }
+                                }
+                        }
+                        for (pair : step7Matches.groupBy[step6.projectMayMatch(match, 2, 3, 4, 5)].entrySet) {
+                                val multiplicityBuilder = createBuilder
+                                for (m : pair.value) {
+                                        multiplicityBuilder.add(1, m.match)
+                                }
+                                multiplicityBuilder.add(-1, pair.key)
+                                multiplicityBuilder.build.tightenUpperBound(0)
+                        }
+                        boundLimit(step7Matches, 2, trainType, 2)
+                        boundLimit(step7Matches, 3, segmentType, 2)
+                        boundLimit(step7Matches, 4, segmentType, 2)
+                        boundLimit(step7Matches, 5, segmentType, 2)
+                        boundLimit(step7Matches, 6, trainType, 2)
+
+                        objectiveBuilder.buildWithBounds
+                ]
+        }
+
+        private static def boundLimit(extension ExtendedLinearExpressionBuilderFactory factory,
+                Collection<CostElementMatch> matches, int index, Type type, int count) {
+                for (pair : matches.groupBy[match.get(index)].entrySet) {
+                        val multiplicityBuilder = createBuilder
+                        for (m : pair.value) {
+                                multiplicityBuilder.add(1, m.match)
+                        }
+                        if (CostElementMatchers.isMulti(pair.key)) {
+                                multiplicityBuilder.add(-count, type)
+                                multiplicityBuilder.build.tightenUpperBound(0)
+                        } else {
+                                multiplicityBuilder.build.tightenUpperBound(count)
+                        }
+                }
+        }
+}