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
|
/*
* SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.generator.cli.commands;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters;
import com.google.inject.Inject;
import org.eclipse.emf.ecore.resource.Resource;
import tools.refinery.generator.ModelGeneratorFactory;
import tools.refinery.generator.ProblemLoader;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
@Parameters(commandDescription = "Generate a model from a partial model")
public class GenerateCommand {
@Inject
private ProblemLoader loader;
@Inject
private ModelGeneratorFactory generatorFactory;
private String inputPath;
private String outputPath = "-";
private List<String> scopes = new ArrayList<>();
private List<String> overrideScopes = new ArrayList<>();
private long randomSeed = 1;
@Parameter(description = "input path", required = true)
public void setInputPath(String inputPath) {
this.inputPath = inputPath;
}
@Parameter(names = {"-output", "-o"}, description = "Output path")
public void setOutputPath(String outputPath) {
this.outputPath = outputPath;
}
@Parameter(names = {"-scope", "-s"}, description = "Extra scope constraints")
public void setScopes(List<String> scopes) {
this.scopes = scopes;
}
@Parameter(names = {"-scope-override", "-S"}, description = "Override scope constraints")
public void setOverrideScopes(List<String> overrideScopes) {
this.overrideScopes = overrideScopes;
}
@Parameter(names = {"-random-seed", "-r"}, description = "Random seed")
public void setRandomSeed(long randomSeed) {
this.randomSeed = randomSeed;
}
public void run() throws IOException {
var problem = isStandardStream(inputPath) ? loader.loadStream(System.in) : loader.loadFile(inputPath);
problem = loader.loadScopeConstraints(problem, scopes, overrideScopes);
var generator = generatorFactory.createGenerator(problem);
generator.setRandomSeed(randomSeed);
generator.generate();
var solution = generator.serializeSolution();
var solutionResource = solution.eResource();
var saveOptions = Map.of();
if (isStandardStream(outputPath)) {
printSolution(solutionResource, saveOptions);
} else {
try (var outputStream = new FileOutputStream(outputPath)) {
solutionResource.save(outputStream, saveOptions);
}
}
}
private boolean isStandardStream(String path) {
return path == null || path.equals("-");
}
// We deliberately write to the standard output if no output path is specified.
@SuppressWarnings("squid:S106")
private void printSolution(Resource solutionResource, Map<?, ?> saveOptions) throws IOException {
solutionResource.save(System.out, saveOptions);
}
}
|
