3 files changed, 197 insertions, 0 deletions

diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/CsvFileWriter.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/CsvFileWriter.xtend
new file mode 100644
index 00000000..bed356e9
--- /dev/null
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/CsvFileWriter.xtend
@@ -0,0 +1,37 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.io;
+
+import java.io.File
+import java.io.FileNotFoundException
+import java.io.PrintWriter
+import java.util.ArrayList
+import java.util.List
+
+class CsvFileWriter {
+        def static void write(ArrayList<ArrayList<String>> datas, String uri) {
+                if(datas.size() <= 0) {
+                        return;
+                }
+                
+                //println("Output csv for " + uri);
+                try  {
+                        val PrintWriter writer = new PrintWriter(new File(uri));
+                        val output = new StringBuilder;
+                        for(List<String> datarow : datas){
+                                for(var i = 0; i < datarow.size() - 1; i++){
+                                        output.append(datarow.get(i) + ',');
+                                }
+                                
+                                if(datarow.size > 1){
+                                        output.append(datarow.get(datarow.size() - 1));
+                                        output.append('\n');
+                                }
+                        }
+                        
+                        writer.write(output.toString());
+                        writer.close();
+                        //println("Output csv finished");
+                }catch(FileNotFoundException e) {
+                        e.printStackTrace();
+                }
+        }
+}
diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/GraphReader.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/GraphReader.xtend
new file mode 100644
index 00000000..fc56e142
--- /dev/null
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/GraphReader.xtend
@@ -0,0 +1,81 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.io;
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.graph.EMFGraph
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.Metric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.MultiplexParticipationCoefficientMetric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.NodeActivityMetric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.OutDegreeMetric
+import java.io.File
+import java.io.FileNotFoundException
+import java.util.ArrayList
+import java.util.List
+import org.eclipse.emf.common.util.URI
+import org.eclipse.emf.ecore.EObject
+import org.eclipse.emf.ecore.EPackage
+import org.eclipse.emf.ecore.EReference
+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
+
+class GraphReader{
+        val ResourceSet resSet = new ResourceSetImpl();
+        val referenceTypes = new ArrayList<String>();
+        
+        new(EPackage metaModel) {
+                Resource.Factory.Registry.INSTANCE.extensionToFactoryMap.put("*",new XMIResourceFactoryImpl)
+                
+                //find all reference types in the meta model
+                metaModel.eAllContents.forEach[
+                        if(it instanceof EReference){
+                                referenceTypes.add(it.name);
+                        }
+                ]
+        }
+        
+        def List<EMFGraph> readModels(String path){
+                val dir = new File(path);
+                if(!dir.isDirectory){
+                        throw new Exception("expecting a directory");
+                }
+                
+                val graphs = new ArrayList<EMFGraph>();
+                
+                val metrics = new ArrayList<Metric>();
+                metrics.add(new OutDegreeMetric());
+                metrics.add(new NodeActivityMetric());
+                metrics.add(new MultiplexParticipationCoefficientMetric());
+                                
+                //check all files in the directory with xmi
+                for(String name : dir.list.filter[it| it.endsWith(".xmi")]){
+                        val file = new File(name);
+                        val roots = readModel(EObject, path,  file.name);                       
+                        //add a list of metrics
+                        val g = new EMFGraph();
+                        for(root : roots){
+                                g.init(root, metrics, name.replaceFirst(".xmi", ""), referenceTypes);
+                        }
+                        
+                        graphs.add(g);
+                }
+                
+                return graphs;
+        }
+        
+        def <RootType extends EObject> List<RootType> readModel(Class<RootType> type, String path, String name) {
+                try {
+                        val resource = resSet.getResource(getURI(path, name),true);
+                        if(resource === null) throw new FileNotFoundException(getURI(path, name).toString)
+                        else {          
+                                return resource.contents as List<RootType>
+                        }
+                } catch(Exception e) {
+                        e.printStackTrace();
+                        throw new FileNotFoundException(getURI(path, name).toString + "reason: " + e.message)
+                }
+        }
+        
+        def static getURI(String path, String name) {
+                URI.createFileURI(path + "/"  + name)
+        }
+}
\ No newline at end of file
diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/RepMetricsReader.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/RepMetricsReader.xtend
new file mode 100644
index 00000000..867ddd1a
--- /dev/null
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/io/RepMetricsReader.xtend
@@ -0,0 +1,79 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.io
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app.Domain
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.MultiplexParticipationCoefficientMetric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.NodeActivityMetric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.OutDegreeMetric
+import java.io.File
+import java.util.List
+import java.util.Scanner
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.MetricSampleGroup
+
+/**
+ * Read the sample of the distribution of a metric provided the csv file of the metric
+ */
+class RepMetricsReader {
+        static def read(Domain d){
+                var domainRepPath = CsvDataName.REP_PATH + d.name + '/';
+                var rep = new MetricSampleGroup()
+                rep.mpcSamples = readFile(domainRepPath + CsvDataName.MPC_REP, MultiplexParticipationCoefficientMetric.valueName,
+                                                          MultiplexParticipationCoefficientMetric.countName).map[Double.parseDouble(it)];
+                rep.naSamples = readFile(domainRepPath+CsvDataName.NA_REP, NodeActivityMetric.valueName, NodeActivityMetric.countName
+                                                        ).map[Double.parseDouble(it)];
+                rep.outDegreeSamples = readFile(domainRepPath+CsvDataName.OUT_D_REP, OutDegreeMetric.valueName, OutDegreeMetric.countName
+                                                        ).map[Double.parseDouble(it)];
+                return rep;
+        }
+        
+        /**
+         * read metric data and parse it to samples
+         */
+        private static def List<String> readFile(String filename, String valueDataName, String countDataName){
+                var s = new Scanner(new File(filename));
+                val counts = newArrayList();
+                val values = newArrayList();
+                //read data from csv
+                while(s.hasNext()){
+                        var data = s.nextLine().split(',');     
+                        
+                        if(data.size >= 1){
+                                if(data.get(0).equals(countDataName)){
+                                        //add all data with parsing them as integers
+                                        counts.addAll(data.subList(1, data.size()).map[Integer.parseInt(it)]);
+                                }else if(data.get(0).equals(valueDataName)){
+                                        //add all data without parsing (there can be either double or string, to be parsed later)
+                                        values.addAll(data.subList(1, data.size()));
+                                }
+                        }
+                }
+                
+                return createSamples(counts, values);
+        }
+        
+        // create samples from values and counts
+        private static def List<String> createSamples(List<Integer> counts, List<String> values){
+                val samples = newArrayList();
+                
+                if(counts.size() != values.size()){
+                        throw new RuntimeException("counts and values should have the same size!");
+                }
+                
+                for(var i = 0; i < counts.size(); i++){
+                        for(var j = 0; j < counts.get(i); j++){
+                                samples.add(values.get(i));
+                        }
+                }
+                
+                return samples;
+        }
+        
+        
+}
+
+class CsvDataName{
+        public static val REP_PATH = 'data/';
+        public static val MPC_REP = 'mpc_rep.csv';
+        public static val NA_REP = 'na_rep.csv';
+        public static val OUT_D_REP = 'out_d_rep.csv';
+}
+