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
|
package rba.tool.editor.generator
import com.google.inject.Inject
import java.util.Set
import org.eclipse.emf.ecore.resource.Resource
import org.eclipse.emf.ecore.resource.ResourceSet
import org.eclipse.xtext.generator.GeneratorContext
import org.eclipse.xtext.generator.IFileSystemAccess
import org.eclipse.xtext.generator.IFileSystemAccess2
import org.eclipse.xtext.generator.IGenerator
import org.eclipse.xtext.generator.IGenerator2
import org.eclipse.xtext.generator.IGeneratorContext
import org.eclipse.xtext.generator.OutputConfiguration
import org.eclipse.xtext.util.CancelIndicator
import rba.tool.editor.generator.json.IJSONGenerator
import rba.tool.editor.generator.rtmodel.IRTModelUpdater
import rba.tool.editor.generator.z3.IConstraintGenerator
import rba.tool.editor.generator.z3.ISortValueGenerator
class RBAModelGeneratorDelegate implements IGenerator, IGenerator2 {
@Inject(optional=true)
private ISortValueGenerator sortValueGenerator;
private ThreadLocal<Boolean> sortValueGeneratorInBuilding = new ThreadLocal<Boolean>() {
override protected initialValue() {
return false;
}
};
@Inject(optional=true)
private IRTModelUpdater rtModelUpdater;
private ThreadLocal<Boolean> rtModelUpdaterInBuilding = new ThreadLocal<Boolean>() {
override protected initialValue() {
return false;
}
};
@Inject(optional=true)
private IConstraintGenerator constraintGenerator;
private ThreadLocal<Boolean> constraintGeneratorInBuilding = new ThreadLocal<Boolean>() {
override protected initialValue() {
return false;
}
};
@Inject(optional=true)
private IJSONGenerator jsonGenerator;
private ThreadLocal<Boolean> jsonGeneratorInBuilding = new ThreadLocal<Boolean>() {
override protected initialValue() {
return false;
}
};
def public void generate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
generate(input.resourceSet, fsa, context);
}
def public void generate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
if (sortValueGenerator !== null && !sortValueGeneratorInBuilding.get.booleanValue) {
sortValueGeneratorInBuilding.set(true);
}
if (rtModelUpdater !== null && !rtModelUpdaterInBuilding.get.booleanValue) {
rtModelUpdaterInBuilding.set(true);
}
if (constraintGenerator !== null && !constraintGeneratorInBuilding.get.booleanValue) {
constraintGeneratorInBuilding.set(true);
}
if (jsonGenerator !== null && !jsonGeneratorInBuilding.get.booleanValue) {
jsonGeneratorInBuilding.set(true);
}
try {
beforeGenerate(input, fsa, context);
doGenerate(input, fsa, context);
} finally {
afterGenerate(input, fsa, context);
}
}
override doGenerate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
doGenerate(input.resourceSet, fsa, context);
}
def void doGenerate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
if (sortValueGenerator !== null && sortValueGeneratorInBuilding.get.booleanValue) {
sortValueGenerator.doGenerate(input, fsa, context);
}
if (rtModelUpdater !== null && rtModelUpdaterInBuilding.get.booleanValue) {
rtModelUpdater.doGenerate(input, fsa, context);
}
if (constraintGenerator !== null && constraintGeneratorInBuilding.get.booleanValue) {
constraintGenerator.doGenerate(input, fsa, context);
}
if (jsonGenerator !== null && jsonGeneratorInBuilding.get.booleanValue) {
jsonGenerator.doGenerate(input, fsa, context);
}
}
override beforeGenerate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
beforeGenerate(input.resourceSet, fsa, context);
}
def void beforeGenerate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
if (sortValueGenerator !== null && sortValueGeneratorInBuilding.get.booleanValue) {
sortValueGenerator.beforeGenerate(input, fsa, context);
}
if (rtModelUpdater !== null && rtModelUpdaterInBuilding.get.booleanValue) {
rtModelUpdater.beforeGenerate(input, fsa, context);
}
if (constraintGenerator !== null && constraintGeneratorInBuilding.get.booleanValue) {
constraintGenerator.beforeGenerate(input, fsa, context);
}
if (jsonGenerator !== null && jsonGeneratorInBuilding.get.booleanValue) {
jsonGenerator.beforeGenerate(input, fsa, context);
}
}
override afterGenerate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
afterGenerate(input.resourceSet, fsa, context);
}
def void afterGenerate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
if (sortValueGenerator !== null && sortValueGeneratorInBuilding.get.booleanValue) {
sortValueGenerator.afterGenerate(input, fsa, context);
}
if (rtModelUpdater !== null && rtModelUpdaterInBuilding.get.booleanValue) {
rtModelUpdater.afterGenerate(input, fsa, context);
}
if (constraintGenerator !== null && constraintGeneratorInBuilding.get.booleanValue) {
constraintGenerator.afterGenerate(input, fsa, context);
}
if (jsonGenerator !== null && jsonGeneratorInBuilding.get.booleanValue) {
jsonGenerator.afterGenerate(input, fsa, context);
}
}
override doGenerate(Resource input, IFileSystemAccess fsa) {
doGenerate(input.resourceSet, fsa);
}
def void doGenerate(ResourceSet input, IFileSystemAccess fsa) {
val casted = fsa as IFileSystemAccess2;
val context = new GeneratorContext();
context.setCancelIndicator(CancelIndicator.NullImpl);
try {
beforeGenerate(input, casted, context);
doGenerate(input, casted, context);
} finally {
afterGenerate(input, casted, context);
}
}
def public void setRBAModelInBuilding(boolean value) {
sortValueGeneratorInBuilding.set(value);
rtModelUpdaterInBuilding.set(value);
constraintGeneratorInBuilding.set(value);
jsonGeneratorInBuilding.set(value);
}
def public boolean isRBAModelInBuilding() {
if (sortValueGeneratorInBuilding.get().booleanValue) {
return true;
}
if (rtModelUpdaterInBuilding.get().booleanValue) {
return true;
}
if (constraintGeneratorInBuilding.get().booleanValue) {
return true;
}
if (jsonGeneratorInBuilding.get().booleanValue) {
return true;
}
return false;
}
def public Set<OutputConfiguration> getRBAModelGeneratorOutputConfigurations() {
return newHashSet(IJSONGenerator.OUTPUT_CONFIGURATION);
}
}
|
