Move contact edge stream creation

Signed-off-by: pacien <pacien.trangirard@pacien.net>

3 files changed, 46 insertions, 38 deletions

diff --git a/src/docs/class.puml b/src/docs/class.puml
index 21f09b6..7bfccca 100644
--- a/src/docs/class.puml
+++ b/src/docs/class.puml
@@ -191,6 +191,8 @@ package block {
     JBoxBlock(BlockType, BodyType, Shape, Vec2)
     Vec2 getPos()
     void link(World)
+    void unlink(World)
+    Stream<ContactEdge> streamContactEdges()
   }
   
   class RobotBlock extends Block {
diff --git a/src/main/java/fr/umlv/java/wallj/block/JBoxBlock.java b/src/main/java/fr/umlv/java/wallj/block/JBoxBlock.java
index e01f4b0..1ef5503 100644
--- a/src/main/java/fr/umlv/java/wallj/block/JBoxBlock.java
+++ b/src/main/java/fr/umlv/java/wallj/block/JBoxBlock.java
@@ -3,10 +3,17 @@ package fr.umlv.java.wallj.block;
 import org.jbox2d.collision.shapes.Shape;
 import org.jbox2d.common.Vec2;
 import org.jbox2d.dynamics.*;
+import org.jbox2d.dynamics.contacts.ContactEdge;
 
 import java.util.Objects;
+import java.util.Spliterator;
+import java.util.function.Consumer;
+import java.util.stream.Stream;
+import java.util.stream.StreamSupport;
 
 /**
+ * A block linked to a physical body.
+ *
  * @author Pacien TRAN-GIRARD
  */
 public abstract class JBoxBlock extends Block {
@@ -27,6 +34,43 @@ public abstract class JBoxBlock extends Block {
     return body;
   }
 
+  /**
+   * @return a stream of contact edges of this body
+   */
+  public Stream<ContactEdge> streamContactEdges() {
+    // If only we had (a working) Java 9...
+    // return Stream.iterate(getBody().getContactList(), c -> c.next)
+    //        .takeWhile(Objects::nonNull);
+
+    return StreamSupport.stream(new Spliterator<ContactEdge>() {
+      private ContactEdge contactEdge = getBody().getContactList();
+
+      @Override
+      public boolean tryAdvance(Consumer<? super ContactEdge> consumer) {
+        if (contactEdge == null) return false;
+
+        consumer.accept(contactEdge);
+        contactEdge = contactEdge.next;
+        return true;
+      }
+
+      @Override
+      public Spliterator<ContactEdge> trySplit() {
+        return null;
+      }
+
+      @Override
+      public long estimateSize() {
+        return Long.MAX_VALUE;
+      }
+
+      @Override
+      public int characteristics() {
+        return NONNULL;
+      }
+    }, false);
+  }
+
   @Override
   public Vec2 getPos() {
     if (body == null) throw new IllegalStateException("Uninitialised block.");
diff --git a/src/main/java/fr/umlv/java/wallj/block/TrashBlock.java b/src/main/java/fr/umlv/java/wallj/block/TrashBlock.java
index 1f2dfa6..cc76917 100644
--- a/src/main/java/fr/umlv/java/wallj/block/TrashBlock.java
+++ b/src/main/java/fr/umlv/java/wallj/block/TrashBlock.java
@@ -7,13 +7,9 @@ import fr.umlv.java.wallj.event.BlockDestroyEvent;
 import fr.umlv.java.wallj.event.Event;
 import org.jbox2d.common.Vec2;
 import org.jbox2d.dynamics.BodyType;
-import org.jbox2d.dynamics.contacts.ContactEdge;
 
 import java.awt.*;
-import java.util.Spliterator;
-import java.util.function.Consumer;
 import java.util.stream.Stream;
-import java.util.stream.StreamSupport;
 
 /**
  * A trash block.
@@ -43,38 +39,4 @@ public class TrashBlock extends JBoxBlock {
     context.getGraphicsContext().paintRectangle(Color.RED, getPos(), TileVec2.TILE_DIM, TileVec2.TILE_DIM);
     return Stream.empty();
   }
-
-  private Stream<ContactEdge> streamContactEdges() {
-    // If only we had (a working) Java 9...
-    // return Stream.iterate(getBody().getContactList(), c -> c.next)
-    //        .takeWhile(Objects::nonNull);
-
-    return StreamSupport.stream(new Spliterator<ContactEdge>() {
-      private ContactEdge contactEdge = getBody().getContactList();
-
-      @Override
-      public boolean tryAdvance(Consumer<? super ContactEdge> consumer) {
-        if (contactEdge == null) return false;
-
-        consumer.accept(contactEdge);
-        contactEdge = contactEdge.next;
-        return true;
-      }
-
-      @Override
-      public Spliterator<ContactEdge> trySplit() {
-        return null;
-      }
-
-      @Override
-      public long estimateSize() {
-        return Long.MAX_VALUE;
-      }
-
-      @Override
-      public int characteristics() {
-        return NONNULL;
-      }
-    }, false);
-  }
 }