Übungstest 7

src/Octree.java

@@ -51,6 +51,12 @@ public class Octree {
         if (root == null) return;
         root.draw(cd);
     }
+
+    public int getNumberOfBalancedRegions() {
+        if (root == null || !(root instanceof OctreeNode r)) return 0;
+
+        return r.getNumberOfBalancedRegions();
+    }
 }
 
 abstract class OctreeItem {
@@ -149,6 +155,26 @@ class OctreeNode extends OctreeItem {
                 ((octNum & 4) != 0) ? -1 : 1)
                 .times(this.size / 4));
     }
+
+    private boolean isBalanced() {
+        for (OctreeItem i : children) {
+            if (!(i instanceof OctreeLeaf))
+                return false;
+        }
+        return true;
+    }
+
+    protected int getNumberOfBalancedRegions() {
+        if (isBalanced()) return 1;
+
+        int c = 0;
+        for (OctreeItem i : children) {
+            if (i instanceof OctreeNode n) {
+                c += n.getNumberOfBalancedRegions();
+            }
+        }
+        return c;
+    }
 }
 
 class OctreeLeaf extends OctreeItem {

src/Simulation.java

@@ -55,6 +55,7 @@ public class Simulation {
             for (Body body : bodies) {
                 octree.add(body);
             }
+            System.out.println(octree.getNumberOfBalancedRegions());
             octree.applyForces(bodies, 100);
 
             if ((seconds % 60) == 0) {