1 files changed, 73 insertions, 0 deletions

diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/EuclideanDistance.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/EuclideanDistance.xtend
new file mode 100644
index 00000000..b945d97b
--- /dev/null
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/EuclideanDistance.xtend
@@ -0,0 +1,73 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.distance
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app.Domain
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.io.RepMetricsReader
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.metrics.MetricSampleGroup
+import com.google.common.collect.Sets
+import java.text.DecimalFormat
+import java.util.ArrayList
+import java.util.HashMap
+import java.util.List
+import java.util.Map
+import java.util.Set
+
+class EuclideanDistance extends CostDistance{
+        var MetricSampleGroup g;
+        var HashMap<String, Double> mpcPMF;
+        var HashMap<String, Double> naPMF;
+        var HashMap<String, Double> outDegreePMF;
+        var DecimalFormat formatter;
+        
+        new(Domain d){
+                var metrics = RepMetricsReader.read(d);
+                this.g = metrics;
+                
+                var mpcSamples = metrics.mpcSamples;
+                var naSamples = metrics.naSamples.stream.mapToDouble([it]).toArray();
+                var outDegreeSamples = metrics.outDegreeSamples.stream.mapToDouble([it]).toArray();
+                
+                //needs to format the number to string avoid precision issue
+                formatter = new DecimalFormat("#0.00000");  
+                
+                mpcPMF = pmfFromSamples(mpcSamples, formatter);
+                naPMF = pmfFromSamples(naSamples, formatter);
+                outDegreePMF = pmfFromSamples(outDegreeSamples, formatter);     
+        }
+        
+        override naDistance(List<Double> samples) {
+                var pmfMap = pmfFromSamples(samples, formatter);
+                return euclideanDistance(pmfMap, naPMF);
+        }
+        
+        override mpcDistance(List<Double> samples) {
+                var pmfMap = pmfFromSamples(samples, formatter);
+                return euclideanDistance(pmfMap, mpcPMF);
+        }
+        
+        override outDegreeDistance(List<Double> samples) {
+                var pmfMap = pmfFromSamples(samples, formatter);
+                return euclideanDistance(pmfMap, outDegreePMF);
+        }
+        
+        
+        def private euclideanDistance(HashMap<String, Double> pmf1, HashMap<String, Double> pmf2){
+                var keys = Sets.union(pmf1.keySet(), pmf2.keySet());
+                var pmfList1 = pmfMapToList(pmf1, keys);
+                var pmfList2 = pmfMapToList(pmf2, keys);
+                var distance = 0.0;
+                for(var i = 0; i < pmfList1.size(); i++){
+                        distance += Math.pow(pmfList1.get(i) + pmfList2.get(i), 2);
+                }
+                
+                return Math.sqrt(distance);
+        }       
+        
+        def private pmfMapToList(Map<String, Double> map, Set<String> keys){
+                var list = new ArrayList<Double>();
+                for(key : keys){
+                        var value = map.getOrDefault(key, 0.0);
+                        list.add(value);
+                }
+                return list;
+        }
+}
\ No newline at end of file