1 files changed, 72 insertions, 0 deletions

diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Main.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Main.xtend
new file mode 100644
index 00000000..2e793b23
--- /dev/null
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Main.xtend
@@ -0,0 +1,72 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.graph.EMFGraph
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.input.GraphReader
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.output.CsvFileWriter
+import hu.bme.mit.inf.dslreasoner.domains.yakindu.sgraph.yakindumm.impl.YakindummPackageImpl
+import java.util.ArrayList
+
+//import yakindumm2.impl.Yakindumm2PackageImpl
+
+class Main {
+        static class RWInformation{
+                public var String inputFolder;
+                public var String outputFolder;
+                public var int numRuns;
+                
+                new(String inputFolder, String outputFolder, int numRuns){
+                        this.inputFolder = inputFolder;
+                        this.outputFolder = outputFolder;
+                        this.numRuns = numRuns;
+                }
+        }
+        
+        def static void main(String[] args){
+                //init model
+                YakindummPackageImpl.eINSTANCE.eClass;
+//              Yakindumm2PackageImpl.eINSTANCE.eClass;
+                val infos = initData();                                                                 
+                                                                                
+                println("Start Reading Models...");
+                var reader = new GraphReader(YakindummPackageImpl.eINSTANCE);
+                for(info : infos){
+                        calculateAllModels(info.inputFolder, info.outputFolder,info.numRuns, reader);
+                }
+                
+                //human input has different package declaration
+//              reader = new GraphReader(Yakindumm2PackageImpl.eINSTANCE);
+                val human = new RWInformation("inputs/humanInput/", "../plot/statistics/humanOutput/", 1);
+                calculateAllModels(human.inputFolder, human.outputFolder,human.numRuns, reader);
+                
+                
+                println("finished");
+        }
+        
+        static def initData(){
+                val infos = new ArrayList<RWInformation>();
+                infos.add(new RWInformation("inputs/alloyInput/models/", "../plot/statistics/alloyOutput/", 20));
+                infos.add(new RWInformation("inputs/randomInput/models/", "../plot/statistics/randomOutput/", 20));
+                infos.add(new RWInformation("inputs/viatraInput30/", "../plot/statistics/viatraOutput30/", 20));
+                infos.add(new RWInformation("inputs/viatraInput100/", "../plot/statistics/viatraOutput100/", 10));
+                return infos;
+        }
+        
+        static def calculateAllModels(String inputFolder, String outputFolder, int numRuns, GraphReader reader){
+                for(var i = 1; i <= numRuns; i++){
+                        val models = new ArrayList<EMFGraph>();
+                        models.addAll(reader.readModels(inputFolder + "run" + i));
+                        for(model : models){
+                                calculateAndOutputMetrics(model, YakindummPackageImpl.eNAME, outputFolder+model.name+"_run_"+i+".csv");
+                        }
+                }
+                println("output results Ended for: " + outputFolder);
+                
+                
+        }
+        
+        static def calculateAndOutputMetrics(EMFGraph model, String metaModel, String fileName){
+                //println("evaluating for " + model.name);
+                model.metaModel = metaModel;
+                CsvFileWriter.write(model.evaluateAllMetrics(), fileName);
+        }
+}
\ No newline at end of file