add gruid buildingsystem

Assets/Scripts/GridBuildingSystem.cs

@@ -5,45 +5,135 @@ using UnityEngine.Tilemaps;
 
 public class GridBuildingSystem : MonoBehaviour
 {
-  public PlaceableObject placeableObject;
-
 
+  public PlaceableObject TESTPO; //! DEBUG
   public Tilemap collisionTm;
+  public TileBase occupiedTile;
 
-  GameObject newBuilding;
+  private PlaceableObject placeableObject;
+  private GameObject newBuildingObject;
+  private bool isDemolishing = false;
 
-
-  private void Start()
-  {
-    newBuilding = Instantiate(placeableObject.prefab, Vector3.zero, Quaternion.identity);
-  }
   private void Update()
   {
-    if (newBuilding)
+    Vector3 mousePosition = Camera.main.ScreenToWorldPoint(Input.mousePosition);
+    int snappedX = Mathf.RoundToInt(mousePosition.x);
+    int snappedY = Mathf.RoundToInt(mousePosition.y);
+    Vector3Int snappedMousePosition = new Vector3Int(snappedX, snappedY, 0);
+
+    if (Input.GetKeyDown(KeyCode.N))
     {
-      Vector3 newPosition = Camera.main.ScreenToWorldPoint(Input.mousePosition);
-      //* Cursor.visible = false;
-      int snappedX = Mathf.RoundToInt(newPosition.x);
-      int snappedY = Mathf.RoundToInt(newPosition.y);
-      Vector3Int snappedPosition = new Vector3Int(snappedX, snappedY, 0);
+      isDemolishing = !isDemolishing;
+      Debug.Log(isDemolishing);
+    }
 
-      newBuilding.transform.position = snappedPosition;
-
-      // TODO change to red if occupied
-
-      if (Input.GetMouseButtonDown(0) && !IsOccupied(snappedPosition, placeableObject))
+    if (!isDemolishing)
+    {
+      //! DEBUG
+      if (Input.GetKeyDown(KeyCode.B))
       {
-        placeableObject.Place();
-        // TODO mark tiles as occupied
-        // TODO unselect
+        SelectBuilding(TESTPO);
+      }
+
+      if (Input.GetMouseButtonDown(1))
+      {
+        UnselectBuilding();
+      }
+
+      if (newBuildingObject)
+      {
+        Cursor.visible = false;
+
+        newBuildingObject.transform.position = snappedMousePosition;
+        bool isOccupied = IsOccupied(snappedMousePosition, placeableObject.GetSizeVector());
+
+        if (isOccupied)
+        {
+          newBuildingObject.GetComponent<SpriteRenderer>().color = Color.red;
+        }
+        else
+        {
+          newBuildingObject.GetComponent<SpriteRenderer>().color = Color.white;
+        }
+
+        if (Input.GetMouseButtonDown(0) && !isOccupied)
+        {
+          newBuildingObject.GetComponent<Building>().Place();
+          newBuildingObject.GetComponent<Building>().placeableObject = placeableObject;
+          OccupyTiles(snappedMousePosition, placeableObject.GetSizeVector());
+          newBuildingObject = null;
+        }
+      }
+      else
+      {
+        Cursor.visible = true;
+      }
+    }
+    else
+    {
+      if (Input.GetMouseButtonDown(0))
+      {
+        Building demoBuilding = BuildingManager.GetBuildingByGridPosition(snappedMousePosition);
+
+        if (demoBuilding)
+        {
+          demoBuilding.Demolish();
+          UnOccupyTiles(demoBuilding.GetGridPosition(), demoBuilding.placeableObject.GetSizeVector());
+        }
       }
     }
 
   }
 
-  private bool IsOccupied(Vector3Int startPos, PlaceableObject po)
+  public void SelectBuilding(PlaceableObject newPo)
   {
-    Vector3Int poSize = new Vector3Int(po.width, po.height, 1);
+    if (newBuildingObject)
+    {
+      Destroy(newBuildingObject);
+    }
+
+    placeableObject = newPo;
+    newBuildingObject = Instantiate(placeableObject.prefab, Vector3.zero, Quaternion.identity);
+  }
+
+  public void UnselectBuilding()
+  {
+    if (newBuildingObject)
+    {
+      Destroy(newBuildingObject);
+    }
+    newBuildingObject = null;
+    placeableObject = null;
+  }
+
+  private void OccupyTiles(Vector3Int startPos, Vector3Int poSize)
+  {
+    BoundsInt area = new BoundsInt(startPos, poSize);
+    TileBase[] tileArray = new TileBase[area.size.x * area.size.y * area.size.z];
+    for (int index = 0; index < tileArray.Length; index++)
+    {
+      tileArray[index] = occupiedTile;
+    }
+
+    collisionTm.SetTilesBlock(area, tileArray);
+
+  }
+
+  private void UnOccupyTiles(Vector3Int startPos, Vector3Int poSize)
+  {
+    BoundsInt area = new BoundsInt(startPos, poSize);
+    TileBase[] tileArray = new TileBase[area.size.x * area.size.y * area.size.z];
+    for (int index = 0; index < tileArray.Length; index++)
+    {
+      tileArray[index] = null;
+    }
+
+    collisionTm.SetTilesBlock(area, tileArray);
+  }
+
+  private bool IsOccupied(Vector3Int startPos, Vector3Int poSize)
+  {
+
 
     BoundsInt area = new BoundsInt(startPos, poSize);
     TileBase[] tileArray = collisionTm.GetTilesBlock(area);