1 files changed, 79 insertions, 0 deletions

diff --git a/Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/ModelManager.java b/Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/ModelManager.java
new file mode 100644
index 00000000..85fd208c
--- /dev/null
+++ b/Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/ModelManager.java
@@ -0,0 +1,79 @@
+package hu.bme.mit.inf.yakinduModelExtractor;
+
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.OutputStreamWriter;
+import java.io.Writer;
+import java.util.LinkedList;
+import java.util.List;
+
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.emf.ecore.resource.Resource;
+import org.eclipse.emf.ecore.resource.ResourceSet;
+import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
+import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
+import org.eclipse.gmf.runtime.notation.NotationPackage;
+import org.yakindu.sct.model.sgraph.SGraphPackage;

+

+public class ModelManager {

+        ResourceSet resourceSet;

+        

+        public ModelManager() {

+                init();

+        }

+        

+        public void init() {

+                SGraphPackage.eINSTANCE.eClass();

+                NotationPackage.eINSTANCE.eClass();

+                resourceSet = new ResourceSetImpl();
+                Resource.Factory.Registry.INSTANCE.getExtensionToFactoryMap().put("*", new XMIResourceFactoryImpl());

+        }

+        
+        public List<String> loadAllModelPathsInDirectory(String path) {
+                File directory = new File(path);
+                if(directory.exists() && directory.isDirectory()) {
+                        List<String> filePaths = new LinkedList<>();
+                        for(File f  : directory.listFiles()) {
+                                if(f.isFile()) {
+                                        String filePath =  f.getPath();
+                                        if(filePath.endsWith("sct")) {
+                                                filePaths.add(filePath);
+                                        }
+                                }
+                        }
+                        return filePaths;
+                } else {
+                        throw new IllegalArgumentException("invalid path");
+                }
+        }
+        

+        public EObject loadModel(String path) {

+                Resource resource = this.resourceSet.getResource(URI.createFileURI(path), true);

+                return resource.getContents().get(0);

+        }

+        

+        public boolean saveModel(EObject root, String path) {

+                Resource resource = this.resourceSet.createResource(URI.createURI(path));

+                resource.getContents().add(root);

+                try {

+                        resource.save(null);

+                        return true;

+                } catch (IOException e) {

+                        System.err.println("Unable to save file: "+path);

+                        return false;

+                }

+        }

+        

+        public boolean saveFile(String path, String content) {

+                try (Writer writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(path), "utf-8"))) {

+                   writer.write(content.toString());

+                   return true;

+                } catch (IOException ex) {

+                        System.err.println("Unable to save file: "+path);

+                        return false;

+                }

+        }

+}