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
|
/*
* SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.store.reasoning.translator.typehierarchy;
import tools.refinery.store.model.Model;
import tools.refinery.store.reasoning.refinement.PartialModelInitializer;
import tools.refinery.store.reasoning.representation.PartialRelation;
import tools.refinery.store.reasoning.seed.ModelSeed;
import tools.refinery.store.representation.Symbol;
import tools.refinery.store.representation.TruthValue;
import tools.refinery.store.tuple.Tuple;
import java.util.Arrays;
import java.util.HashMap;
import java.util.function.Function;
public class TypeHierarchyInitializer implements PartialModelInitializer {
private final TypeHierarchy typeHierarchy;
private final Symbol<InferredType> typeSymbol;
public TypeHierarchyInitializer(TypeHierarchy typeHierarchy, Symbol<InferredType> typeSymbol) {
this.typeHierarchy = typeHierarchy;
this.typeSymbol = typeSymbol;
}
@Override
public void initialize(Model model, ModelSeed modelSeed) {
var inferredTypes = new InferredType[modelSeed.getNodeCount()];
Arrays.fill(inferredTypes, typeHierarchy.getUnknownType());
for (var type : typeHierarchy.getAllTypes()) {
model.checkCancelled();
initializeType(type, inferredTypes, model, modelSeed);
}
var typeInterpretation = model.getInterpretation(typeSymbol);
var uniqueTable = new HashMap<InferredType, InferredType>();
for (int i = 0; i < inferredTypes.length; i++) {
model.checkCancelled();
var uniqueType = uniqueTable.computeIfAbsent(inferredTypes[i], Function.identity());
typeInterpretation.put(Tuple.of(i), uniqueType);
}
}
private void initializeType(PartialRelation type, InferredType[] inferredTypes, Model model, ModelSeed modelSeed) {
var cursor = modelSeed.getCursor(type, TruthValue.UNKNOWN);
var analysisResult = typeHierarchy.getAnalysisResult(type);
while (cursor.move()) {
model.checkCancelled();
var i = cursor.getKey().get(0);
checkNodeId(inferredTypes, i);
var value = cursor.getValue();
inferredTypes[i] = analysisResult.merge(inferredTypes[i], value);
}
}
private void checkNodeId(InferredType[] inferredTypes, int nodeId) {
if (nodeId < 0 || nodeId >= inferredTypes.length) {
throw new IllegalArgumentException("Expected node id %d to be lower than model size %d"
.formatted(nodeId, inferredTypes.length));
}
}
}
|
