1 files changed, 40 insertions, 1 deletions

diff --git a/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/xtend-gen/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/CpsMdeOptimiserMain.java b/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/xtend-gen/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/CpsMdeOptimiserMain.java
index 288505a1..7a2a7e99 100644
--- a/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/xtend-gen/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/CpsMdeOptimiserMain.java
+++ b/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/xtend-gen/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/CpsMdeOptimiserMain.java
@@ -5,13 +5,20 @@ import hu.bme.mit.inf.dslreasoner.domains.cps.CpsPackage;
 import hu.bme.mit.inf.dslreasoner.domains.cps.CyberPhysicalSystem;
 import hu.bme.mit.inf.dslreasoner.domains.cps.generator.CpsGenerator;
 import hu.bme.mit.inf.dslreasoner.domains.cps.mdeo.ExcludedRun;
+import java.io.BufferedReader;
+import java.io.BufferedWriter;
+import java.io.FileReader;
+import java.io.FileWriter;
+import java.util.Collections;
 import java.util.Map;
+import java.util.Set;
 import org.eclipse.emf.common.util.URI;
 import org.eclipse.emf.ecore.EPackage;
 import org.eclipse.emf.ecore.resource.Resource;
 import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
 import org.eclipse.xtext.xbase.lib.CollectionLiterals;
 import org.eclipse.xtext.xbase.lib.Exceptions;
+import org.eclipse.xtext.xbase.lib.Pair;
 import uk.ac.kcl.inf.mdeoptimiser.languages.MoptStandaloneSetup;
 
 @SuppressWarnings("all")
@@ -32,15 +39,47 @@ public class CpsMdeOptimiserMain {
       XMIResourceFactoryImpl _xMIResourceFactoryImpl = new XMIResourceFactoryImpl();
       _extensionToFactoryMap.put(Resource.Factory.Registry.DEFAULT_EXTENSION, _xMIResourceFactoryImpl);
       EPackage.Registry.INSTANCE.put(CpsPackage.eNS_URI, CpsPackage.eINSTANCE);
-      final CpsGenerator generator = new CpsGenerator(1, 4, 2);
+      final CpsGenerator generator = new CpsGenerator(1, 4, 1);
       final CyberPhysicalSystem problem = generator.generateCpsProblem();
       Resource _eResource = problem.eResource();
       _eResource.setURI(URI.createFileURI(CpsMdeOptimiserMain.PROBLEM_PATH));
       problem.eResource().save(CollectionLiterals.<Object, Object>emptyMap());
+      Pair<String, String> _mappedTo = Pair.<String, String>of("cps.ecore", CpsPackage.eNS_URI);
+      CpsMdeOptimiserMain.fixupHenshinModel("model/cps.henshin", "model/cps_fixup.henshin", Collections.<String, String>unmodifiableMap(CollectionLiterals.<String, String>newHashMap(_mappedTo)));
       final Injector injector = new MoptStandaloneSetup().createInjectorAndDoEMFRegistration();
       injector.<ExcludedRun>getInstance(ExcludedRun.class).run(CpsMdeOptimiserMain.PROJECT_PATH, CpsMdeOptimiserMain.MOPT_PATH);
     } catch (Throwable _e) {
       throw Exceptions.sneakyThrow(_e);
     }
   }
+  
+  private static void fixupHenshinModel(final String originalPath, final String outputPath, final Map<String, String> remapMap) {
+    try {
+      FileReader _fileReader = new FileReader(originalPath);
+      final BufferedReader reader = new BufferedReader(_fileReader);
+      try {
+        FileWriter _fileWriter = new FileWriter(outputPath);
+        final BufferedWriter writer = new BufferedWriter(_fileWriter);
+        try {
+          String line = null;
+          while (((line = reader.readLine()) != null)) {
+            {
+              Set<Map.Entry<String, String>> _entrySet = remapMap.entrySet();
+              for (final Map.Entry<String, String> entry : _entrySet) {
+                line = line.replace(entry.getKey(), entry.getValue());
+              }
+              writer.write(line);
+              writer.write("\n");
+            }
+          }
+        } finally {
+          writer.close();
+        }
+      } finally {
+        reader.close();
+      }
+    } catch (Throwable _e) {
+      throw Exceptions.sneakyThrow(_e);
+    }
+  }
 }