1 files changed, 16 insertions, 2 deletions

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
index 2ea12581..6af0b6c7 100644
--- 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
@@ -3,13 +3,14 @@ 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 hu.bme.mit.inf.dslreasoner.domains.yakindu.sgraph.yakindumm.impl.YakindummPackageImpl
+import java.util.HashMap
 
 /**
  * Read the sample of the distribution of a metric provided the csv file of the metric
  */
 class RepMetricsReader {
         static def read(Domain d){
-                var reader = new GraphReader(YakindummPackageImpl.eINSTANCE);
+                var reader = new GraphReader(YakindummPackageImpl.eINSTANCE, '.xmi');
                 
                 
                 var domainRepPath = DataName.REP_PATH + d.name + '/';
@@ -20,7 +21,20 @@ class RepMetricsReader {
                 rep.naSamples = readMetrics(reader, domainRepPath + DataName.NA_REP).naSamples;
                 rep.typedOutDegreeSamples = out_d.typedOutDegreeSamples;
                 rep.edgeTypeSamples = out_d.edgeTypeSamples;
-                rep.nodeTypeSamples = out_d.nodeTypeSamples;
+                
+                //TODO: Parameterize the prior node distribution
+                var nodeTypeSamples = new HashMap<String, Double>();
+                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);
+                
+                
+                rep.nodeTypeSamples = nodeTypeSamples;
                 return rep;
         }