blob: 3159f7071cb53ef7ab15fc0b5ed8a7c0a4bc774f (
plain) (
blame)
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
|
/*******************************************************************************
* Copyright (c) 2010-2014, Zoltan Ujhelyi, Istvan Rath and Daniel Varro
* 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.plan;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.stream.Collectors;
import tools.refinery.viatra.runtime.localsearch.operations.ISearchOperation;
import tools.refinery.viatra.runtime.matchers.psystem.PBody;
import tools.refinery.viatra.runtime.matchers.psystem.PVariable;
import tools.refinery.viatra.runtime.matchers.tuple.TupleMask;
/**
* A SearchPlan stores a collection of SearchPlanOperations for a fixed order of variables.
*
* @author Zoltan Ujhelyi
*
*/
public class SearchPlan {
private final List<ISearchOperation> operations;
private final Map<Integer, PVariable> variableMapping;
private final TupleMask parameterMask;
private final PBody body;
/**
* @since 2.0
*/
public SearchPlan(PBody body, List<ISearchOperation> operations, TupleMask parameterMask, Map<PVariable, Integer> variableMapping) {
this.body = body;
this.operations = Collections.unmodifiableList(new ArrayList<>(operations));
this.parameterMask = parameterMask;
this.variableMapping = Collections.unmodifiableMap(variableMapping.entrySet().stream()
.collect(Collectors.toMap(Entry::getValue, Entry::getKey)));
}
/**
* Returns an immutable list of operations stored in the plan.
* @return the operations
*/
public List<ISearchOperation> getOperations() {
return operations;
}
/**
* Returns an immutable map of variable mappings for the plan
* @since 2.0
*/
public Map<Integer, PVariable> getVariableMapping() {
return variableMapping;
}
/**
* Returns the index of a given operation in the plan
* @since 2.0
*/
public int getOperationIndex(ISearchOperation operation) {
return operations.indexOf(operation);
}
/**
* @since 2.0
*/
public TupleMask getParameterMask() {
return parameterMask;
}
/**
* @since 2.0
*/
public PBody getSourceBody() {
return body;
}
@Override
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append("{\n");
for(ISearchOperation operation : this.getOperations()){
sb.append("\t");
sb.append(operation);
sb.append("\n");
}
sb.append("}");
return sb.toString();
}
}
|