3 files changed, 264 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..00b38d90
--- /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,52 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.io;
+
+import java.io.File
+import java.io.FileNotFoundException
+import java.io.FileOutputStream
+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;
+                }
+                val PrintWriter writer = new PrintWriter(new File(uri));
+                output(writer, datas, uri);
+        }
+        
+        def static void append(ArrayList<ArrayList<String>> datas, String uri) {
+                if(datas.size() <= 0) {
+                        return;
+                }
+                val PrintWriter writer = new PrintWriter(new FileOutputStream(new File(uri), true));
+                output(writer, datas, uri);
+        }
+        
+        
+        def private static void output(PrintWriter writer, ArrayList<ArrayList<String>> datas, String uri) {
+                //println("Output csv for " + uri);
+                try  {
+                        
+                        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..053e0da3
--- /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,112 @@
+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.EdgeTypeMetric
+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.NodeTypeMetric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.OutDegreeMetric
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.TypedOutDegree
+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
+import org.eclipse.emf.ecore.EGenericType
+import org.eclipse.emf.ecore.EStructuralFeature
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.TypedClusteringCoefficientMetric
+
+class GraphReader{
+        val ResourceSet resSet = new ResourceSetImpl();
+        val referenceTypes = new ArrayList<EReference>();
+        var String suffix;
+        
+        new(EPackage metaModel, String suffix) {
+                Resource.Factory.Registry.INSTANCE.extensionToFactoryMap.put("*",new XMIResourceFactoryImpl)
+                this.suffix = suffix;
+                
+                //find all reference types in the meta model
+                metaModel.eAllContents.forEach[
+                        if(it instanceof EReference){
+                                referenceTypes.add(it);
+                        }
+                ]
+        }
+        
+        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());
+                metrics.add(new TypedOutDegree());              
+                metrics.add(new NodeTypeMetric());
+                metrics.add(new EdgeTypeMetric());
+                var count = 1
+                //check all files in the directory with suffix
+                for(String name : dir.list.filter[it| it.endsWith(suffix)]){
+                        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(suffix, ""), referenceTypes);
+                        }
+
+                        count ++;
+                        graphs.add(g);
+                }
+                
+                return graphs;
+        }
+        
+        def EMFGraph readModel(String path, String filename){
+                val metrics = new ArrayList<Metric>();
+                metrics.add(new OutDegreeMetric());
+                metrics.add(new NodeActivityMetric());
+                metrics.add(new MultiplexParticipationCoefficientMetric());
+                metrics.add(new TypedOutDegree());              
+                metrics.add(new NodeTypeMetric());
+                metrics.add(new EdgeTypeMetric());
+                
+                val file = new File(filename);
+                val roots = readModel(EObject, path,  file.name);                       
+                //add a list of metrics
+                val g = new EMFGraph();
+                for(root : roots){
+                        g.init(root, metrics, filename.replaceFirst(suffix, ""), referenceTypes);
+                }
+                return g
+        }
+        
+        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 Exception(getURI(path, name).toString());
+                }
+        }
+        
+        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..2e7be586
--- /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,100 @@
+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.MetricSampleGroup
+import github.impl.GithubPackageImpl
+import hu.bme.mit.inf.dslreasoner.domains.yakindu.sgraph.yakindumm.impl.YakindummPackageImpl
+import java.util.HashMap
+import org.eclipse.emf.ecore.EReference
+import org.eclipse.emf.ecore.impl.EcorePackageImpl
+
+/**
+ * Read the sample of the distribution of a metric provided the csv file of the metric
+ */
+class RepMetricsReader {
+        static var Domain domain;
+        static def read(Domain d){
+                var GraphReader reader;
+                if(d == Domain.Yakindumm){
+                        reader = new GraphReader(YakindummPackageImpl.eINSTANCE, '.xmi');
+                }else if (d == Domain.Ecore){
+                        reader = new GraphReader(EcorePackageImpl.eINSTANCE, '.ecore');
+                }else if (d == Domain.Github){
+                        reader = new GraphReader(GithubPackageImpl.eINSTANCE, '.githubmodel')
+                }
+                
+                domain = d;
+                
+                var domainRepPath = DataName.REP_PATH + d.name + '/';
+                var rep = new MetricSampleGroup()
+                var out_d = readMetrics(reader, domainRepPath + DataName.OUT_D_REP);
+                var mpc = readMetrics(reader, domainRepPath + DataName.MPC_REP);
+                rep.mpcSamples = mpc.mpcSamples;
+                rep.outDegreeSamples = out_d.outDegreeSamples;
+                rep.naSamples = readMetrics(reader, domainRepPath + DataName.NA_REP).naSamples;
+                rep.typedOutDegreeSamples = out_d.typedOutDegreeSamples;
+                rep.edgeTypeSamples = mpc.edgeTypeSamples;
+                
+                //TODO: Parameterize the prior node distribution
+                var nodeTypeSamples = new HashMap<String, Double>();
+                if(d == Domain.Yakindumm){
+                        nodeTypeSamples.put('Entry', 0.04257802080554814);
+                        nodeTypeSamples.put('Choice', 0.1267671379034409);
+                        nodeTypeSamples.put('State', 0.1596092291277674);
+                        nodeTypeSamples.put('Transition', 0.6138636969858629);
+                        nodeTypeSamples.put('Statechart', 0.010136036276340358);
+                        nodeTypeSamples.put('Region', 0.04467858095492131);
+                        nodeTypeSamples.put('Exit', 0.0018338223526273673);
+                        nodeTypeSamples.put('FinalState', 0.0005334755934915977);
+                }else if(d ==Domain.Ecore){
+                        nodeTypeSamples.put('EAttribute', 0.23539778449144008);
+                        nodeTypeSamples.put('EClass', 0.33081570996978854);
+                        nodeTypeSamples.put('EReference', 0.30996978851963747);
+                        nodeTypeSamples.put('EPackage', 0.012789526686807653);
+                        nodeTypeSamples.put('EAnnotation', 0.002517623363544813);
+                        nodeTypeSamples.put('EEnumLiteral', 0.07275931520644502);
+                        nodeTypeSamples.put('EEnum', 0.013645518630412891);
+                        nodeTypeSamples.put('EDataType', 0.004028197381671702);
+                        nodeTypeSamples.put('EParameter', 0.005941591137965764);
+                        nodeTypeSamples.put('EGenericType', 0.002014098690835851);
+                        nodeTypeSamples.put('EOperation', 0.009415911379657605);
+                        nodeTypeSamples.put('ETypeParameter', 0.0007049345417925478);   
+                }else if (d == Domain.Github){
+                        nodeTypeSamples.put('Project', 0.012636538873420432);
+                        nodeTypeSamples.put('Commit', 0.5525808524309276);
+                        nodeTypeSamples.put('User', 0.05847076461769116);
+                        nodeTypeSamples.put('Issue', 0.12743628185907047);
+                        nodeTypeSamples.put('PullRequest', 0.07560505461554937);
+                        nodeTypeSamples.put('IssueEvent', 0.17327050760334123);
+                }
+                
+                
+                
+                rep.nodeTypeSamples = nodeTypeSamples;
+                return rep;
+        }
+        
+        /**
+         * Read representative model
+         */
+        private static def readMetrics(GraphReader r, String path){
+                val model = r.readModels(path).head;
+                if(domain == Domain.Ecore){
+                        var refsToRemove = EcorePackageImpl.eINSTANCE.eAllContents.filter(EReference).filter[
+                                it.name.equals('eGenericType') || it.name.equals('eGenericSuperTypes') || it.name.equals('eFactoryInstance') ||
+                                it.name.equals('eGenericExceptions') || it.name.equals('references') || it.name.equals('contents');
+                        ];
+                        refsToRemove.forEach[model.removeReference(it)];
+                }
+                return model.evaluateAllMetricsToSamples();
+        }
+        
+}
+
+class DataName{
+        public static val REP_PATH = 'data/';
+        public static val MPC_REP = 'mpc_rep';
+        public static val NA_REP = 'na_rep';
+        public static val OUT_D_REP = 'out_d_rep';
+}
+