1 files changed, 69 insertions, 0 deletions

diff --git a/Metrics/ca.mcgill.ecse.dslreasoner.realistic.metrics/bin/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculations/CalcSQROSZ2.xtend_ b/Metrics/ca.mcgill.ecse.dslreasoner.realistic.metrics/bin/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculations/CalcSQROSZ2.xtend_
new file mode 100644
index 00000000..e999d62f
--- /dev/null
+++ b/Metrics/ca.mcgill.ecse.dslreasoner.realistic.metrics/bin/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculations/CalcSQROSZ2.xtend_
@@ -0,0 +1,69 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculations
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.examples.Util
+import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.neighbourhood.PartialInterpretation2ImmutableTypeLattice
+import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialInterpretation
+import java.util.HashMap
+import java.util.Map
+import java.util.Set
+import org.eclipse.emf.ecore.EObject
+
+import static extension hu.bme.mit.inf.dslreasoner.util.CollectionsUtil.*
+
+class CalcSQROSZ2 extends CalcMetric {
+
+        // ///////////////////
+        // SQROSZ2(v) = # depth-4 paths that lead to v / total # depth-4 paths
+        // ///////////////////
+        static val neighbourhoodComputer = new PartialInterpretation2ImmutableTypeLattice
+
+        override calcFromModel(EObject model) {
+                val nodes = model.eResource.allContents.toList
+
+                // fill HashSet
+                var Map<EObject, Set<EObject>> node2Neighbours = new HashMap
+                Util.fillWithNodes(nodes, node2Neighbours)
+
+                // iterate over nodes and add connected nodes
+                Util.getNeighboursList(nodes, node2Neighbours)
+
+                // Measurements
+                var totalSQR = 0.0
+                var totalDenom = 0.0
+                var num4thNeighbours = 0.0
+                for (node : nodes) {
+                        val neighbours = node.lookup(node2Neighbours)
+
+                        for (neighbour1 : neighbours) {
+                                val neighbours1 = neighbour1.lookup(node2Neighbours)
+
+                                for (neighbour2 : neighbours1) {
+                                        val neighbours2 = neighbour2.lookup(node2Neighbours)
+
+                                        for (neighbour3 : neighbours2) {
+                                                val neighbours3 = neighbour3.lookup(node2Neighbours)
+                                                num4thNeighbours = neighbours3.size
+
+                                                if (neighbours3.contains(node)) {
+                                                        totalSQR += 1
+                                                }
+                                                totalDenom += num4thNeighbours
+                                        }
+                                }
+                        }
+                        
+                }
+                val avgSQR = totalSQR / totalDenom
+
+                return avgSQR
+        }
+
+        override calcFromNHLattice(PartialInterpretation pm) {
+                return 0.0
+//              return getCfromNHLattice(pm, 2, v)
+        }
+
+        override calcFromNHLattice(PartialInterpretation pm, Integer depth) {
+                return 0.0
+        }
+}
\ No newline at end of file