1 files changed, 61 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/PartialInterpretationMetric.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/PartialInterpretationMetric.xtend
new file mode 100644
index 00000000..5f958cfa
--- /dev/null
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/PartialInterpretationMetric.xtend
@@ -0,0 +1,61 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.graph.PartialInterpretationGraph
+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 ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.output.CsvFileWriter
+import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialInterpretation
+import java.io.File
+import java.io.FileNotFoundException
+import java.io.PrintWriter
+import java.util.ArrayList
+import java.util.List
+import org.eclipse.emf.ecore.util.EcoreUtil
+import org.eclipse.viatra.dse.api.Solution
+
+class PartialInterpretationMetric {
+        var static state = 0;
+        def static void calculateMetric(PartialInterpretation partial, String path, int id, String currentStateId){
+                val metrics = new ArrayList<Metric>();
+                metrics.add(new OutDegreeMetric());
+                metrics.add(new NodeActivityMetric());
+                metrics.add(new MultiplexParticipationCoefficientMetric());
+                
+                new File(path).mkdir();
+                val filename = path + "/state_"+state+".csv";
+                state++;
+                val metricCalculator = new PartialInterpretationGraph(partial, metrics, currentStateId);
+                
+                CsvFileWriter.write(metricCalculator.evaluateAllMetrics(), filename);
+        }
+        
+        def static void outputTrajectories(PartialInterpretation empty, List<Solution>  solutions){
+                
+                for(solution : solutions){
+                        val emptySolutionCopy = EcoreUtil.copy(empty)
+                        val trajectory = solution.shortestTrajectory;
+                        trajectory.modelWithEditingDomain = emptySolutionCopy
+                        val stateCodes = newArrayList()
+                        
+                        while(trajectory.doNextTransformation){
+                                println(trajectory.stateCoder.createStateCode)
+                                stateCodes.add(trajectory.stateCoder.createStateCode.toString)
+                        }
+                        
+                        try{
+                                val path = "debug/metric/trajectories/trajectory"+trajectory.stateCoder.createStateCode.toString+".csv"
+                                val PrintWriter writer = new PrintWriter(new File(path))
+                                val output = new StringBuilder
+                                for(stateCode : stateCodes){
+                                        output.append(stateCode+'\n')
+                                }
+                                writer.write(output.toString())
+                                writer.close()
+                        }catch(FileNotFoundException e) {
+                                e.printStackTrace()
+                        }
+                }                       
+        }
+}
\ No newline at end of file