|
5 | 5 | import com.google.inject.Provider; |
6 | 6 | import com.regnosys.rosetta.RosettaRuntimeModule; |
7 | 7 | import com.regnosys.rosetta.RosettaStandaloneSetup; |
| 8 | +import com.regnosys.rosetta.builtin.RosettaBuiltinsService; |
8 | 9 | import com.regnosys.rosetta.common.util.ClassPathUtils; |
9 | 10 | import com.regnosys.rosetta.common.util.UrlUtils; |
10 | 11 | import com.regnosys.rosetta.generator.external.ExternalGenerator; |
|
15 | 16 | import org.eclipse.emf.ecore.resource.Resource; |
16 | 17 | import org.eclipse.emf.ecore.resource.ResourceSet; |
17 | 18 | import org.eclipse.xtext.resource.XtextResourceSet; |
| 19 | +import org.eclipse.emf.common.util.URI; |
18 | 20 | import org.jetbrains.annotations.NotNull; |
19 | 21 | import org.slf4j.Logger; |
20 | 22 | import org.slf4j.LoggerFactory; |
@@ -90,6 +92,7 @@ public static void main(String[] args) { |
90 | 92 | if (cmd.hasOption("s")) { |
91 | 93 | String srcDir = cmd.getOptionValue("s"); |
92 | 94 | translateFromSourceDir(srcDir, tgtDir); |
| 95 | + translateFromSourceDir2(srcDir, tgtDir + "-2"); |
93 | 96 | } else if (cmd.hasOption("f")) { |
94 | 97 | String srcFile = cmd.getOptionValue("f"); |
95 | 98 | translateFromSourceFile(srcFile, tgtDir); |
@@ -132,6 +135,63 @@ private static void translateFromSourceDir(String srcDir, String tgtDir) { |
132 | 135 | translateRosetta(rosettaFiles, tgtDir); |
133 | 136 | } |
134 | 137 |
|
| 138 | + private static void translateFromSourceDir2(String srcDir, String tgtDir) { |
| 139 | + LOGGER.info("translateFromSourceDir2 ... Reading from directory: {} and writing to: {}", srcDir, tgtDir); |
| 140 | + Path srcDirPath = Paths.get(srcDir); |
| 141 | + if (!Files.exists(srcDirPath)) { |
| 142 | + System.err.println("Source directory does not exist: " + srcDir); |
| 143 | + return; |
| 144 | + } |
| 145 | + if (!Files.isDirectory(srcDirPath)) { |
| 146 | + System.err.println("Source directory is not a directory: " + srcDir); |
| 147 | + return; |
| 148 | + } |
| 149 | + Injector injector = new PythonRosettaStandaloneSetup().createInjectorAndDoEMFRegistration(); |
| 150 | + PythonCodeGenerator pythonCodeGenerator = injector.getInstance(PythonCodeGenerator.class); |
| 151 | + PythonModelLoader modelLoader = injector.getInstance(PythonModelLoader.class); |
| 152 | + ResourceSet resourceSet = injector.getInstance(ResourceSet.class); |
| 153 | + |
| 154 | + List<Resource> resources = new LinkedList<>(); |
| 155 | + |
| 156 | + RosettaBuiltinsService builtins = injector.getInstance(RosettaBuiltinsService.class); |
| 157 | + resources.add (resourceSet.getResource(builtins.basicTypesURI, true)); |
| 158 | + resources.add (resourceSet.getResource(builtins.annotationsURI, true)); |
| 159 | + try { |
| 160 | + // Find all .rosetta files in the directory and load them from disk |
| 161 | + Files.walk(srcDirPath) |
| 162 | + .filter(path -> path.toString().endsWith(".rosetta")) |
| 163 | + .map(file -> resourceSet.getResource(URI.createFileURI(file.toString()), true)) |
| 164 | + .forEach(resources::add); |
| 165 | + } catch (IOException e) { |
| 166 | + System.err.println("Error reading source directory: " + e.getMessage()); |
| 167 | + return; |
| 168 | + } |
| 169 | + LOGGER.info("translateFromSourceDir2 ... Found {} .rosetta files in directory {}", resources.size(), srcDir); |
| 170 | + |
| 171 | + |
| 172 | + List<RosettaModel> models = modelLoader.rosettaModels2(resources); |
| 173 | + if (models.isEmpty()) { |
| 174 | + System.err.println("No valid Rosetta models found."); |
| 175 | + return; |
| 176 | + } |
| 177 | + String version = models.get(0).getVersion(); |
| 178 | + |
| 179 | + LOGGER.info("translateFromSourceDir2 ... Processing {} models, version: {}", models.size(), version); |
| 180 | + |
| 181 | + Map<String, CharSequence> generatedPython = new HashMap<>(); |
| 182 | + pythonCodeGenerator.beforeAllGenerate(resourceSet, models, version); |
| 183 | + for (RosettaModel model : models) { |
| 184 | + System.out.println("translateFromSourceDir2 ... processing: " + model.getName()); |
| 185 | + generatedPython.putAll(pythonCodeGenerator.beforeGenerate(model.eResource(), model, version)); |
| 186 | + generatedPython.putAll(pythonCodeGenerator.generate(model.eResource(), model, version)); |
| 187 | + generatedPython.putAll(pythonCodeGenerator.afterGenerate(model.eResource(), model, version)); |
| 188 | + } |
| 189 | + generatedPython.putAll(pythonCodeGenerator.afterAllGenerate(resourceSet, models, version)); |
| 190 | + |
| 191 | + writePythonFiles(generatedPython, tgtDir); |
| 192 | + } |
| 193 | + |
| 194 | + |
135 | 195 | private static void translateFromSourceFile(String srcFile, String tgtDir) { |
136 | 196 | LOGGER.info("Reading from file: {} and writing to: {}", srcFile, tgtDir); |
137 | 197 | Path srcFilePath = Paths.get(srcFile); |
@@ -169,13 +229,14 @@ private static void translateRosetta(List<Path> rosettaFiles, String tgtDir) { |
169 | 229 | return; |
170 | 230 | } |
171 | 231 | XtextResourceSet resourceSet = modelLoader.getResourceSet(); |
172 | | - String version = models.getFirst().getVersion(); |
| 232 | + String version = models.get(0).getVersion(); |
173 | 233 |
|
174 | 234 | LOGGER.info("Processing {} models, version: {}", models.size(), version); |
175 | 235 |
|
176 | 236 | Map<String, CharSequence> generatedPython = new HashMap<>(); |
177 | 237 | pythonCodeGenerator.beforeAllGenerate(resourceSet, models, version); |
178 | 238 | for (RosettaModel model : models) { |
| 239 | + System.out.println("translateRosetta ... processing: " + model.getName()); |
179 | 240 | generatedPython.putAll(pythonCodeGenerator.beforeGenerate(model.eResource(), model, version)); |
180 | 241 | generatedPython.putAll(pythonCodeGenerator.generate(model.eResource(), model, version)); |
181 | 242 | generatedPython.putAll(pythonCodeGenerator.afterGenerate(model.eResource(), model, version)); |
@@ -220,6 +281,15 @@ public List<RosettaModel> rosettaModels(List<Path> statics, List<Path> sourceFil |
220 | 281 | .collect(Collectors.toList()); |
221 | 282 | } |
222 | 283 |
|
| 284 | + public List<RosettaModel> rosettaModels2(List<Resource> resources) { |
| 285 | + return resources.stream() |
| 286 | + .filter(Objects::nonNull) |
| 287 | + .map(Resource::getContents) |
| 288 | + .flatMap(Collection::stream) |
| 289 | + .filter(r -> r instanceof RosettaModel) |
| 290 | + .map(r -> (RosettaModel) r) |
| 291 | + .collect(Collectors.toList()); |
| 292 | + } |
223 | 293 | public XtextResourceSet getResourceSet() { |
224 | 294 | return resourceSetProvider.get(); |
225 | 295 | } |
|
0 commit comments