From e3a5bda9ca0ba7570cc323a696e55bf600487b24 Mon Sep 17 00:00:00 2001
From: Stefan-5422 <32109571+Stefan-5422@users.noreply.github.com>
Date: Fri, 18 Mar 2022 10:45:53 +0100
Subject: [PATCH] Update Test Machines;

---
 .../yes/yes/behaviours/PlaceBehaviour.java    | 37 +++++++++++++++++++
 .../com/yes/yes/behaviours/TestBehaviour.java |  7 +++-
 .../yes/entities/machines/TestMachine.java    |  6 +--
 .../yes/entities/machines/TestMachine2.java   |  6 +--
 4 files changed, 46 insertions(+), 10 deletions(-)
 create mode 100644 src/main/java/com/yes/yes/behaviours/PlaceBehaviour.java

diff --git a/src/main/java/com/yes/yes/behaviours/PlaceBehaviour.java b/src/main/java/com/yes/yes/behaviours/PlaceBehaviour.java
new file mode 100644
index 0000000..a3bb03a
--- /dev/null
+++ b/src/main/java/com/yes/yes/behaviours/PlaceBehaviour.java
@@ -0,0 +1,37 @@
+package com.yes.yes.behaviours;
+
+import com.yes.yes.utils.BlockContainer;
+import com.yes.yes.utils.Component;
+import com.yes.yes.utils.Coordinate;
+import com.yes.yes.utils.Entity;
+
+public class PlaceBehaviour extends Component {
+
+    public PlaceBehaviour(Entity entity, BlockContainer blockContainer) {
+        super(entity, blockContainer);
+    }
+
+    @Override
+    public void initialize()
+    {
+        for (int x = -1; x < 2; x++) {
+            for (int y = -1; y < 2; y++) {
+                try {
+                    Entity entity = blockContainer.getBlock(new Coordinate(x, y));
+
+                    if (entity != null) {
+                        entity.trigger("placed", parent);
+                        System.out.println(entity.getClass().getSimpleName());
+                    }
+                } catch (IllegalAccessException ex) {
+                    ex.printStackTrace();
+                }
+            }
+        }
+    }
+
+    @Override
+    public void update() {
+
+    }
+}
diff --git a/src/main/java/com/yes/yes/behaviours/TestBehaviour.java b/src/main/java/com/yes/yes/behaviours/TestBehaviour.java
index 9686512..5d182ad 100644
--- a/src/main/java/com/yes/yes/behaviours/TestBehaviour.java
+++ b/src/main/java/com/yes/yes/behaviours/TestBehaviour.java
@@ -13,6 +13,7 @@ public class TestBehaviour extends Component {
 
     @Override
     public void initialize() {
+
     }
 
     @Override
@@ -23,10 +24,12 @@ public class TestBehaviour extends Component {
                 try {
                     System.out.println("Offset: X:" + x + " Y:" + y);
                     Entity entity = blockContainer.getBlock(new Coordinate(x, y));
-                    if (entity != null)
+
+                    if (entity != null) {
                         System.out.println(entity.getClass().getSimpleName());
+                    }
                 } catch (IllegalAccessException ex) {
-                   ex.printStackTrace();
+                    ex.printStackTrace();
                 }
             }
         }
diff --git a/src/main/java/com/yes/yes/entities/machines/TestMachine.java b/src/main/java/com/yes/yes/entities/machines/TestMachine.java
index 3ed846d..d117fd1 100644
--- a/src/main/java/com/yes/yes/entities/machines/TestMachine.java
+++ b/src/main/java/com/yes/yes/entities/machines/TestMachine.java
@@ -1,8 +1,7 @@
 package com.yes.yes.entities.machines;
 
-import com.yes.yes.behaviours.TestBehaviour;
+import com.yes.yes.behaviours.PlaceBehaviour;
 import com.yes.yes.utils.*;
-import com.yes.yes.utils.exceptions.AlreadyExistsException;
 import javafx.scene.shape.Circle;
 
 public class TestMachine extends Entity {
@@ -13,7 +12,6 @@ public class TestMachine extends Entity {
 
     public TestMachine(BlockContainer blockContainer) {
         this();
-        this.addBehaviour(new TestBehaviour(this, blockContainer));
-        update();
+        this.addBehaviour(new PlaceBehaviour(this, blockContainer));
     }
 }
diff --git a/src/main/java/com/yes/yes/entities/machines/TestMachine2.java b/src/main/java/com/yes/yes/entities/machines/TestMachine2.java
index f830de0..30bad3a 100644
--- a/src/main/java/com/yes/yes/entities/machines/TestMachine2.java
+++ b/src/main/java/com/yes/yes/entities/machines/TestMachine2.java
@@ -1,9 +1,8 @@
 package com.yes.yes.entities.machines;
 
-import com.yes.yes.behaviours.TestBehaviour;
+import com.yes.yes.behaviours.PlaceBehaviour;
 import com.yes.yes.utils.BlockContainer;
 import com.yes.yes.utils.Entity;
-import javafx.scene.shape.Circle;
 import javafx.scene.shape.Rectangle;
 
 public class TestMachine2 extends Entity {
@@ -14,7 +13,6 @@ public class TestMachine2 extends Entity {
 
     public TestMachine2(BlockContainer blockContainer) {
         this();
-        this.addBehaviour(new TestBehaviour(this, blockContainer));
-        update();
+        this.addBehaviour(new PlaceBehaviour(this, blockContainer));
     }
 }
\ No newline at end of file