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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
|
package hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.rules
import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.DefinedElement
import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.LogiclanguageFactory
import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.Relation
import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.RelationDeclaration
import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.Type
import hu.bme.mit.inf.dslreasoner.logic.model.logicproblem.LogicProblem
import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.ModelGenerationStatistics
import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.patterns.GeneratedPatterns
import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.patterns.ObjectCreationPrecondition
import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialInterpretation
import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialRelationInterpretation
import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialTypeInterpratation
import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialinterpretationFactory
import java.util.LinkedHashMap
import org.eclipse.viatra.query.runtime.api.GenericPatternMatch
import org.eclipse.viatra.query.runtime.api.IQuerySpecification
import org.eclipse.viatra.query.runtime.api.ViatraQueryMatcher
import org.eclipse.viatra.transformation.runtime.emf.rules.batch.BatchTransformationRule
import org.eclipse.viatra.transformation.runtime.emf.rules.batch.BatchTransformationRuleFactory
class RefinementRuleProvider {
private extension BatchTransformationRuleFactory factory = new BatchTransformationRuleFactory
private extension PartialinterpretationFactory factory2 = PartialinterpretationFactory.eINSTANCE
private extension LogiclanguageFactory factory3 = LogiclanguageFactory.eINSTANCE
def canonizeName(String name) {
return name.replace(' ','_')
}
def LinkedHashMap<ObjectCreationPrecondition, BatchTransformationRule<GenericPatternMatch, ViatraQueryMatcher<GenericPatternMatch>>>
createObjectRefinementRules(
GeneratedPatterns patterns,
boolean nameNewElement,
ModelGenerationStatistics statistics
)
{
val res = new LinkedHashMap
for(LHSEntry: patterns.refineObjectQueries.entrySet) {
val containmentRelation = LHSEntry.key.containmentRelation
val inverseRelation = LHSEntry.key.inverseContainment
val type = LHSEntry.key.newType
val lhs = LHSEntry.value as IQuerySpecification<ViatraQueryMatcher<GenericPatternMatch>>
val rule = createObjectCreationRule(containmentRelation,inverseRelation,type,lhs,nameNewElement,statistics)
res.put(LHSEntry.key,rule)
}
return res
}
def private createObjectCreationRule(
Relation containmentRelation,
Relation inverseRelation,
Type type,
IQuerySpecification<ViatraQueryMatcher<GenericPatternMatch>> lhs,
boolean nameNewElement,
ModelGenerationStatistics statistics)
{
val name = '''addObject_«type.name.canonizeName»«
IF containmentRelation!=null»_by_«containmentRelation.name.canonizeName»«ENDIF»'''
//println("Rule created: " + name + "> " + lhs.fullyQualifiedName)
val ruleBuilder = factory.createRule
.name(name)
.precondition(lhs)
if(containmentRelation != null) {
if(inverseRelation!= null) {
ruleBuilder.action[match |
//println(name)
val startTime = System.nanoTime
//val problem = match.get(0) as LogicProblem
val interpretation = match.get(1) as PartialInterpretation
val relationInterpretation = match.get(2) as PartialRelationInterpretation
val inverseRelationInterpretation = match.get(3) as PartialRelationInterpretation
val typeInterpretation = match.get(4) as PartialTypeInterpratation
val container = match.get(5) as DefinedElement
val newElement = createDefinedElement
if(nameNewElement) {
newElement.name = '''new «interpretation.newElements.size»'''
}
// Existence
interpretation.newElements+=newElement
interpretation.maxNewElements=interpretation.maxNewElements-1
if(interpretation.minNewElements > 0) {
interpretation.minNewElements=interpretation.minNewElements-1
}
// Types
typeInterpretation.elements += newElement
typeInterpretation.supertypeInterpretation.forEach[it.elements += newElement]
// ContainmentRelation
val newLink1 = factory2.createBinaryElementRelationLink => [it.param1 = container it.param2 = newElement]
relationInterpretation.relationlinks+=newLink1
// Inverse Containment
val newLink2 = factory2.createBinaryElementRelationLink => [it.param1 = newElement it.param2 = container]
inverseRelationInterpretation.relationlinks+=newLink2
statistics.addExecutionTime(System.nanoTime-startTime)
]
} else {
ruleBuilder.action[match |
//println(name)
val startTime = System.nanoTime
//val problem = match.get(0) as LogicProblem
val interpretation = match.get(1) as PartialInterpretation
val relationInterpretation = match.get(2) as PartialRelationInterpretation
val typeInterpretation = match.get(3) as PartialTypeInterpratation
val container = match.get(4) as DefinedElement
val newElement = createDefinedElement
if(nameNewElement) {
newElement.name = '''new «interpretation.newElements.size»'''
}
// Existence
interpretation.newElements+=newElement
interpretation.maxNewElements=interpretation.maxNewElements-1
if(interpretation.minNewElements > 0) {
interpretation.minNewElements=interpretation.minNewElements-1
}
// Types
typeInterpretation.elements += newElement
typeInterpretation.supertypeInterpretation.forEach[it.elements += newElement]
// ContainmentRelation
val newLink = factory2.createBinaryElementRelationLink => [it.param1 = container it.param2 = newElement]
relationInterpretation.relationlinks+=newLink
statistics.addExecutionTime(System.nanoTime-startTime)
]
}
} else {
ruleBuilder.action[match |
println(name)
val startTime = System.nanoTime
//val problem = match.get(0) as LogicProblem
val interpretation = match.get(1) as PartialInterpretation
val typeInterpretation = match.get(2) as PartialTypeInterpratation
println("->"+typeInterpretation.interpretationOf.name)
val newElement = createDefinedElement
if(nameNewElement) {
newElement.name = '''new «interpretation.newElements.size»'''
}
// Existence
interpretation.newElements+=newElement
interpretation.maxNewElements=interpretation.maxNewElements-1
if(interpretation.minNewElements > 0) {
interpretation.minNewElements=interpretation.minNewElements-1
}
// Types
typeInterpretation.elements += newElement
typeInterpretation.supertypeInterpretation.forEach[it.elements += newElement]
statistics.addExecutionTime(System.nanoTime-startTime)
]
}
return ruleBuilder.build
}
def createRelationRefinementRules(GeneratedPatterns patterns, ModelGenerationStatistics statistics) {
val res = new LinkedHashMap
for(LHSEntry: patterns.refinerelationQueries.entrySet) {
val declaration = LHSEntry.key.key
val inverseReference = LHSEntry.key.value
val lhs = LHSEntry.value as IQuerySpecification<ViatraQueryMatcher<GenericPatternMatch>>
val rule = createRelationRefinementRule(declaration,inverseReference,lhs,statistics)
res.put(LHSEntry.key,rule)
}
return res
}
def private BatchTransformationRule<GenericPatternMatch, ViatraQueryMatcher<GenericPatternMatch>>
createRelationRefinementRule(RelationDeclaration declaration, Relation inverseRelation, IQuerySpecification<ViatraQueryMatcher<GenericPatternMatch>> lhs, ModelGenerationStatistics statistics)
{
val name = '''addRelation_«declaration.name.canonizeName»«IF inverseRelation != null»_and_«inverseRelation.name.canonizeName»«ENDIF»'''
val ruleBuilder = factory.createRule
.name(name)
.precondition(lhs)
if (inverseRelation == null) {
ruleBuilder.action [ match |
val startTime = System.nanoTime
//println(name)
// val problem = match.get(0) as LogicProblem
// val interpretation = match.get(1) as PartialInterpretation
val relationInterpretation = match.get(2) as PartialRelationInterpretation
val src = match.get(3) as DefinedElement
val trg = match.get(4) as DefinedElement
val link = createBinaryElementRelationLink => [it.param1 = src it.param2 = trg]
relationInterpretation.relationlinks += link
statistics.addExecutionTime(System.nanoTime-startTime)
]
} else {
ruleBuilder.action [ match |
val startTime = System.nanoTime
//println(name)
// val problem = match.get(0) as LogicProblem
// val interpretation = match.get(1) as PartialInterpretation
val relationInterpretation = match.get(2) as PartialRelationInterpretation
val inverseInterpretation = match.get(3) as PartialRelationInterpretation
val src = match.get(4) as DefinedElement
val trg = match.get(5) as DefinedElement
val link = createBinaryElementRelationLink => [it.param1 = src it.param2 = trg]
relationInterpretation.relationlinks += link
val inverseLink = createBinaryElementRelationLink => [it.param1 = trg it.param2 = src]
inverseInterpretation.relationlinks += inverseLink
statistics.addExecutionTime(System.nanoTime-startTime)
]
}
return ruleBuilder.build
}
}
|
