1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
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.TrainLocations_step_2
import modes3.queries.TrainLocations_step_3
class TrainLocationsObjectiveHint 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(TrainLocations_step_2.instance.fullyQualifiedName)
val step3 = costElementMatchers.get(TrainLocations_step_3.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)
}
}
}
boundLimit(step3Matches, 2, trainType, 1)
boundLimit(step3Matches, 3, segmentType, 1)
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)
}
}
}
}
|