slime-blood-and-pain/Assets/Scripts/Map.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

[System.Serializable]
public class ImpossibleCorridorError : System.Exception
{
    public ImpossibleCorridorError() { }
    public ImpossibleCorridorError(string message) : base(message) { }
    public ImpossibleCorridorError(string message, System.Exception inner) : base(message, inner) { }
    protected ImpossibleCorridorError(
        System.Runtime.Serialization.SerializationInfo info,
        System.Runtime.Serialization.StreamingContext context) : base(info, context) { }
}

public class MapRoom {
    public readonly Vector2Int start;
    public readonly Vector2Int end;
    public readonly int mapSize;

    public MapRoom(int mapSize, int maxRoomSize, int minRoomSize) {
        this.mapSize = mapSize;
        start = new Vector2Int(Random.Range(1, mapSize-1), Random.Range(1, mapSize-1));
        end = new Vector2Int(Random.Range(1, mapSize-1), Random.Range(1, mapSize-1));
        if(start.x > end.x) {
            int swap = start.x;
            start.x = end.x;
            end.x = swap;
        }
        if(start.y > end.y) {
            int swap = start.y;
            start.y = end.y;
            end.y = swap;
        }
        while(end.x - start.x > maxRoomSize) {
            end.x--;
            start.x++;
        }
        while(end.y - start.y > maxRoomSize) {
            end.y--;
            start.y++;
        }
        while(end.x - start.x < minRoomSize) {
            end.x++;
            start.x--;
        }
        while(end.y - start.y < minRoomSize) {
            end.y++;
            start.y--;
        }
        start.Clamp(new Vector2Int(1, 1), new Vector2Int(mapSize-1, mapSize-1));
        end.Clamp(new Vector2Int(1, 1), new Vector2Int(mapSize-1, mapSize-1));
    }

    public Vector2Int RandomPoint() {
        return new Vector2Int(Random.Range(start.x, end.x+1), Random.Range(start.y, end.y+1));
    }
}

public class MapCorridor {
    public readonly Vector2Int start;
    public readonly Vector2Int end;
    public readonly bool horizontal_priority;

    public MapCorridor(MapRoom from, MapRoom to, int mapSize) {
        start = from.RandomPoint();
        end = to.RandomPoint();
        //50%
        horizontal_priority = Random.Range(0f, 1f) >= 0.5f;
    }
}

public class Map : MonoBehaviour
{
    [BeforeStartAttribute]
    public int mapSize = 30;

    [BeforeStartAttribute]
    public int roomsToGenerate = 5;

    [BeforeStartAttribute]
    public int minRoomSize = 2;

    [BeforeStartAttribute]
    public int maxRoomSize = 6;

    [BeforeStartAttribute]
    public int maxRoomIterations = 100;

    [BeforeStartAttribute]
    public List<Sprite> floorSprites;

    [BeforeStartAttribute]
    public List<Sprite> botWallSprites;

    [BeforeStartAttribute]
    public GameObject tilePrefab;

    [BeforeStartAttribute]
    public GameObject playerPrefab;

    [BeforeStartAttribute]
    public List<GameObject> enemyPrefabs;

    [BeforeStartAttribute]
    public int enemiesToSpawn = 10;

    [BeforeStartAttribute]
    public List<GameObject> curiositiesPrefabs;

    [BeforeStartAttribute]
    public int curiositiesToSpawn = 35;

    [BeforeStartAttribute]
    public GameObject stairsPrefab;

    [BeforeStartAttribute]
    public GameObject shopCorePrefab;

    [BeforeStartAttribute]
    public GameObject shopEdgePrefab;

    [BeforeStartAttribute]
    public GameObject[] shopItems; //3

    private GameObject[,] tiles;
    private List<MapRoom> rooms;
    private TurnHandler turnHandler;

    public MapTile GetTile(Vector2Int position) {
        try {
            GameObject tileObject = tiles[position.x, position.y];
            return tileObject.GetComponent<MapTile>();
        }
        catch(System.IndexOutOfRangeException) {
            return null;
        }
    }

    public bool CanMoveTo(Vector2Int position)
    {
        try {
            MapTile tile = GetTile(position);
            if(tile == null) return false;
            bool walkable = tile.walkable;
            List<Entity> entities = turnHandler.GetEntitiesAtPosition(position);
            bool free = true;
            foreach(Entity entity in entities) {
                free &= entity.overlappable;
            }
            return walkable && free;
        }
        catch(System.IndexOutOfRangeException) {
            return false;
        }
    }

    private void EditTile(Vector2Int position, bool walkable, bool roomPart) {
        MapTile tile = GetTile(position);
        tile.walkable = walkable;
        tile.roomPart |= roomPart;
    }

    private void FillWithWalls() {
        for(int x = 0; x < mapSize; x++) {
            for(int y = 0; y < mapSize; y++) {
                GameObject tileObject = Instantiate(tilePrefab, transform);
                tileObject.transform.position = new Vector3(x, y, 0);
                tiles[x, y] = tileObject;
                tileObject.name = "Tile [" + x.ToString() + ", " + y.ToString() + "]";
                MapTile tile = tileObject.GetComponent<MapTile>();
                tile.walkable = false;
            }
        }
    }

    private void PlaceRoom(MapRoom mr) {
        for(int x = mr.start.x; x <= mr.end.x; x++) {
            for(int y = mr.start.y; y <= mr.end.y; y++) {
                EditTile(new Vector2Int(x, y), true, true);
            }
        }
    }

    private bool ScanRoom(MapRoom mr) {
        //Returns true if the room can be safely placed
        for(int x = Mathf.Clamp(mr.start.x-1, 0, mapSize-1); x <= Mathf.Clamp(mr.end.x+1, 0, mapSize-1); x++) {
            for(int y = Mathf.Clamp(mr.start.y-1, 0, mapSize-1); y <= Mathf.Clamp(mr.end.y+1, 0, mapSize-1); y++) {
                if(GetTile(new Vector2Int(x, y)).roomPart) {
                    return false;
                }
            }
        }
        return true;
    }

    private void PlaceCorridor(MapCorridor mc) {
        Vector2Int cursor = new Vector2Int(mc.start.x, mc.start.y);
        EditTile(cursor, true, false);
        if(mc.horizontal_priority) {
            while(cursor.x != mc.end.x) {
                if(cursor.x > mc.end.x) cursor.x--;
                else cursor.x++;
                EditTile(cursor, true, false);
            }
            while(cursor.y != mc.end.y) {
                if(cursor.y > mc.end.y) cursor.y--;
                else cursor.y++;
                EditTile(cursor, true, false);
            }
        }
        else
        {
            while(cursor.y != mc.end.y) {
                if(cursor.y > mc.end.y) cursor.y--;
                else cursor.y++;
                EditTile(cursor, true, false);
            } 
            while(cursor.x != mc.end.x) {
                if(cursor.x > mc.end.x) cursor.x--;
                else cursor.x++;
                EditTile(cursor, true, false);
            }
        }
    }
    
    private void GenerateMap() {
        FillWithWalls();
        int roomIterations = 0;
        while(rooms.Count < roomsToGenerate && roomIterations < maxRoomIterations) {
            roomIterations++;
            MapRoom room = new MapRoom(mapSize, maxRoomSize, minRoomSize);
            if(ScanRoom(room)) {
                //Fill with the room
                PlaceRoom(room);
                rooms.Add(room);
            }
            //Place a corridor
            if(rooms.Count > 1) {
                MapRoom from = rooms[rooms.Count-2];
                MapRoom to = rooms[rooms.Count-1];
                try {
                    MapCorridor corridor = new MapCorridor(from, to, mapSize);
                    PlaceCorridor(corridor);
                }
                catch (ImpossibleCorridorError) {
                }
            }
        }
    }

    private void PlacePlayer() {
        //Check for an existing player
        MapRoom room = rooms[Random.Range(0, rooms.Count)];
        Vector2Int point = room.RandomPoint();
        GameObject playerObject = GameObject.FindGameObjectWithTag("Player");
        if(playerObject == null) {
            playerObject = Instantiate(playerPrefab, turnHandler.transform);
        }
        else {
            playerObject.transform.parent = turnHandler.transform;
        }
        playerObject.name = "Player";
        playerObject.transform.position = new Vector3(point.x, point.y, 0);
    }

    private void PlaceEnemies() {
        for(int i = 0; i < enemiesToSpawn; i++) {
            MapRoom room = rooms[Random.Range(0, rooms.Count)];
            Vector2Int point;
            while(true) {
                point = room.RandomPoint();
                if(turnHandler.GetEntitiesAtPosition(point).Count > 0) {
                    continue;
                }
                break;
            }
            GameObject enemyPrefab = enemyPrefabs[Random.Range(0, enemyPrefabs.Count)];
            GameObject enemyObject = Instantiate(enemyPrefab, turnHandler.transform);
            enemyObject.name = "Enemy " + i.ToString();
            enemyObject.transform.position = new Vector3(point.x, point.y, 0);
        }
    }

    private void PlaceCuriosities() {
        for(int i = 0; i < curiositiesToSpawn; i++) {
            MapRoom room = rooms[Random.Range(0, rooms.Count)];
            Vector2Int point;
            while(true) {
                point = room.RandomPoint();
                if(turnHandler.GetEntitiesAtPosition(point).Count > 0) {
                    continue;
                }
                break;
            }
            GameObject curiosityPrefab = curiositiesPrefabs[Random.Range(0, curiositiesPrefabs.Count)];
            GameObject curiosityObject = Instantiate(curiosityPrefab, turnHandler.transform);
            curiosityObject.name = "Curiosity " + i.ToString();
            curiosityObject.transform.position = new Vector3(point.x, point.y, 0);
        }
    }

    private void PlaceStairs() {
        MapRoom room = rooms[Random.Range(0, rooms.Count)];
        Vector2Int point;
        while(true) {
            point = room.RandomPoint();
            if(turnHandler.GetEntitiesAtPosition(point).Count > 0) {
                continue;
            }
            break;
        }
        GameObject curiosityObject = Instantiate(stairsPrefab, turnHandler.transform);
        curiosityObject.name = "Stairs";
        curiosityObject.transform.position = new Vector3(point.x, point.y, 0);
    }

    public static Sprite SampleSprite(List<Sprite> list) {
        return list[Random.Range(0, list.Count)];
    }

    public void GenerateTileSprites() {
        for(int x = 0; x < mapSize; x++) {
            for(int y = 0; y < mapSize; y++) {
                MapTile tile = GetTile(new Vector2Int(x, y));
                MapTile otherTile;
                if(tile.walkable) tile.sprite = SampleSprite(floorSprites);
                else if((bool)(otherTile = GetTile(new Vector2Int(x, y+1))) && otherTile.walkable) tile.sprite = SampleSprite(botWallSprites);
                //TODO: corners
            }
        }
    }

    private void PlaceShop() {
        MapRoom room = rooms[Random.Range(0, rooms.Count)];
        Vector2Int point;
        while(true) {
            bool done = true;
            point = room.RandomPoint();
            for(int x = -2; x <= 2; x++) {
                for(int y = -2; y <= 2; y++) {
                    if(turnHandler.GetEntitiesAtPosition(new Vector2Int(x, y)).Count > 0) {
                        done = false;
                    }
                    if(!done) break;
                }
                if(!done) break;
            }
            if(done) break;
        }
        GameObject shopCore = Instantiate(shopCorePrefab, turnHandler.transform);
        shopCore.name = "Shop (Core)";
        shopCore.transform.position = new Vector3(point.x, point.y, 0);
        GameObject shopEdge = Instantiate(shopEdgePrefab, turnHandler.transform);
        shopEdge.name = "Shop (Edge)";
        shopEdge.transform.position = new Vector3(point.x+1, point.y, 0);
        shopEdge = Instantiate(shopEdgePrefab, turnHandler.transform);
        shopEdge.name = "Shop (Edge)";
        shopEdge.transform.position = new Vector3(point.x-1, point.y, 0);
        shopEdge = Instantiate(shopEdgePrefab, turnHandler.transform);
        shopEdge.name = "Shop (Edge)";
        shopEdge.transform.position = new Vector3(point.x+1, point.y+1, 0);
        shopEdge = Instantiate(shopEdgePrefab, turnHandler.transform);
        shopEdge.name = "Shop (Edge)";
        shopEdge.transform.position = new Vector3(point.x-1, point.y+1, 0);
        shopEdge = Instantiate(shopEdgePrefab, turnHandler.transform);
        shopEdge.name = "Shop (Edge)";
        shopEdge.transform.position = new Vector3(point.x, point.y+1, 0);
        shopEdge = Instantiate(shopEdgePrefab, turnHandler.transform);
        shopEdge.name = "Shop (Edge)";
        GameObject shopItem1 = Instantiate(shopItems[0], turnHandler.transform);
        shopItem1.transform.position = new Vector3(point.x, point.y-1, 0);
        shopEdge.name = "Shop Item 1";
        GameObject shopItem2 = Instantiate(shopItems[1], turnHandler.transform);
        shopItem2.transform.position = new Vector3(point.x+1, point.y-1, 0);
        shopEdge.name = "Shop Item 2";
        GameObject shopItem3 = Instantiate(shopItems[2], turnHandler.transform);
        shopItem3.transform.position = new Vector3(point.x-1, point.y-1, 0);
        shopEdge.name = "Shop Item 3";
    }

    public void NewLevel() {
        //Cleanup everything.
        transform.parent.BroadcastMessage("OnNewLevel");
        //Initialize everything
        tiles = new GameObject[mapSize, mapSize];
        rooms = new List<MapRoom>();
        turnHandler = GameObject.FindGameObjectWithTag("GameController").GetComponentInChildren<TurnHandler>();

        GenerateMap();
        GenerateTileSprites();
        PlaceShop();
        PlacePlayer();
        PlaceEnemies();
        PlaceCuriosities();
        PlaceStairs();
    }

    public void OnNewLevel() {
        //Suppress the error
    }

    private void Start()
    {
        NewLevel();
    }
}