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
|
/*******************************************************************************
* Copyright (c) 2010-2017, Zoltan Ujhelyi, IncQuery Labs Ltd.
* Copyright (c) 2023 The Refinery Authors <https://refinery.tools/>
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0 which is available at
* http://www.eclipse.org/legal/epl-v20.html.
*
* SPDX-License-Identifier: EPL-2.0
*******************************************************************************/
package tools.refinery.viatra.runtime.localsearch.planner.compiler;
import tools.refinery.viatra.runtime.localsearch.operations.generic.GenericTypeCheck;
import tools.refinery.viatra.runtime.localsearch.operations.generic.GenericTypeExtend;
import tools.refinery.viatra.runtime.localsearch.operations.generic.GenericTypeExtendSingleValue;
import tools.refinery.viatra.runtime.matchers.context.IInputKey;
import tools.refinery.viatra.runtime.matchers.context.IQueryRuntimeContext;
import tools.refinery.viatra.runtime.matchers.psystem.PVariable;
import tools.refinery.viatra.runtime.matchers.psystem.basicdeferred.TypeFilterConstraint;
import tools.refinery.viatra.runtime.matchers.psystem.basicenumerables.TypeConstraint;
import tools.refinery.viatra.runtime.matchers.tuple.Tuple;
import tools.refinery.viatra.runtime.matchers.tuple.TupleMask;
import java.util.*;
/**
* @author Zoltan Ujhelyi
* @since 1.7
*
*/
public class GenericOperationCompiler extends AbstractOperationCompiler {
public GenericOperationCompiler(IQueryRuntimeContext runtimeContext) {
super(runtimeContext);
}
@Override
protected void createCheck(TypeFilterConstraint typeConstraint, Map<PVariable, Integer> variableMapping) {
IInputKey inputKey = typeConstraint.getInputKey();
Tuple tuple = typeConstraint.getVariablesTuple();
int[] positions = new int[tuple.getSize()];
for (int i = 0; i < tuple.getSize(); i++) {
PVariable variable = (PVariable) tuple.get(i);
positions[i] = variableMapping.get(variable);
}
operations.add(new GenericTypeCheck(inputKey, positions, TupleMask.fromSelectedIndices(variableMapping.size(), positions)));
}
@Override
protected void createCheck(TypeConstraint typeConstraint, Map<PVariable, Integer> variableMapping) {
IInputKey inputKey = typeConstraint.getSupplierKey();
Tuple tuple = typeConstraint.getVariablesTuple();
int[] positions = new int[tuple.getSize()];
for (int i = 0; i < tuple.getSize(); i++) {
PVariable variable = (PVariable) tuple.get(i);
positions[i] = variableMapping.get(variable);
}
operations.add(new GenericTypeCheck(inputKey, positions, TupleMask.fromSelectedIndices(variableMapping.size(), positions)));
}
@Override
protected void createUnaryTypeCheck(IInputKey inputKey, int position) {
int[] positions = new int[] {position};
operations.add(new GenericTypeCheck(inputKey, positions, TupleMask.fromSelectedIndices(1, positions)));
}
@Override
protected void createExtend(TypeConstraint typeConstraint, Map<PVariable, Integer> variableMapping) {
IInputKey inputKey = typeConstraint.getSupplierKey();
Tuple tuple = typeConstraint.getVariablesTuple();
int[] positions = new int[tuple.getSize()];
List<Integer> boundVariableIndices = new ArrayList<>();
List<Integer> boundVariables = new ArrayList<>();
Set<Integer> unboundVariables = new HashSet<>();
for (int i = 0; i < tuple.getSize(); i++) {
PVariable variable = (PVariable) tuple.get(i);
Integer position = variableMapping.get(variable);
positions[i] = position;
if (variableBindings.get(typeConstraint).contains(position)) {
boundVariableIndices.add(i);
boundVariables.add(position);
} else {
unboundVariables.add(position);
}
}
TupleMask indexerMask = TupleMask.fromSelectedIndices(inputKey.getArity(), boundVariableIndices);
TupleMask callMask = TupleMask.fromSelectedIndices(variableMapping.size(), boundVariables);
// If multiple tuple elements from the indexer should be bound to the same variable, we must use a
// {@link GenericTypeExtend} check whether the tuple elements have the same value.
if (unboundVariables.size() == 1 && indexerMask.getSize() + 1 == indexerMask.getSourceWidth()) {
operations.add(new GenericTypeExtendSingleValue(inputKey, positions, callMask, indexerMask, unboundVariables.iterator().next()));
} else {
operations.add(new GenericTypeExtend(inputKey, positions, callMask, indexerMask, unboundVariables));
}
}
}
|
