using UnityEngine; using System.Collections.Generic; public class MapGenerator : MonoBehaviour { [Header("Map Dimensions")] [SerializeField] private int mapWidth = 50; [SerializeField] private int mapHeight = 50; [Header("Room Settings")] [SerializeField] private int numberOfRooms = 15; [SerializeField] private int minRoomWidth = 4; [SerializeField] private int maxRoomWidth = 10; [SerializeField] private int minRoomHeight = 4; [SerializeField] private int maxRoomHeight = 10; [SerializeField] private int roomPadding = 2; // Minimum space between rooms [Header("Corridor Settings")] [SerializeField] private int corridorWidth = 1; // Keep it odd for centered corridors if needed [Header("Generation Settings")] [SerializeField] private int maxPlacementAttempts = 100; // Attempts to place a room before giving up [SerializeField] private bool generateOnStart = true; [SerializeField] private int randomSeed = 0; // 0 for random seed private List rooms; private List corridors; private void Start() { if (generateOnStart) { GenerateMap(); } } [ContextMenu("Generate Map")] public void GenerateMap() { if (randomSeed != 0) { Random.InitState(randomSeed); } rooms = new List(); corridors = new List(); CreateFirstRoom(); int attempts = 0; while (rooms.Count < numberOfRooms && attempts < maxPlacementAttempts * numberOfRooms) // Safety break { if (!TryCreateRoomWithCorridor()) { attempts++; } else { attempts = 0; // Reset attempts on success } } if (rooms.Count < numberOfRooms) { Debug.LogWarning($"Map generation stopped early. Could only place {rooms.Count} rooms out of {numberOfRooms}."); } else { Debug.Log($"Successfully generated map with {rooms.Count} rooms."); } } private void CreateFirstRoom() { int roomWidth = Random.Range(minRoomWidth, maxRoomWidth + 1); int roomHeight = Random.Range(minRoomHeight, maxRoomHeight + 1); int roomX = mapWidth / 2 - roomWidth / 2; int roomY = mapHeight / 2 - roomHeight / 2; Room firstRoom = new Room(roomX, roomY, roomWidth, roomHeight); rooms.Add(firstRoom); } private bool TryCreateRoomWithCorridor() { if (rooms.Count == 0) return false; int attempts = 0; while (attempts < maxPlacementAttempts) { attempts++; // 1. Pick a random existing room Room existingRoom = rooms[Random.Range(0, rooms.Count)]; // 2. Pick a random direction (0: N, 1: E, 2: S, 3: W) int direction = Random.Range(0, 4); // 3. Determine new room dimensions int newRoomWidth = Random.Range(minRoomWidth, maxRoomWidth + 1); int newRoomHeight = Random.Range(minRoomHeight, maxRoomHeight + 1); // 4. Calculate potential new room position and corridor int corridorLength = Random.Range(roomPadding, roomPadding * 3); // Variable corridor length RectInt potentialCorridor; Vector2Int newRoomPos = CalculateNewRoomPosition(existingRoom, direction, newRoomWidth, newRoomHeight, corridorLength, out potentialCorridor); Room potentialNewRoom = new Room(newRoomPos.x, newRoomPos.y, newRoomWidth, newRoomHeight); // 5. Check for overlaps (room and corridor) if (!IsOverlapping(potentialNewRoom, roomPadding) && !IsCorridorOverlapping(potentialCorridor, potentialNewRoom, existingRoom)) { // 6. Add room and corridor rooms.Add(potentialNewRoom); corridors.Add(potentialCorridor); return true; // Success } } return false; // Failed to place after attempts } private Vector2Int CalculateNewRoomPosition(Room existingRoom, int direction, int newWidth, int newHeight, int corridorLength, out RectInt corridorRect) { Vector2Int newPos = Vector2Int.zero; int corridorX = 0, corridorY = 0, corridorW = 0, corridorH = 0; int halfCorridorWidth = corridorWidth / 2; // For centering switch (direction) { case 0: // North newPos.x = Random.Range(existingRoom.Bounds.xMin - newWidth + 1, existingRoom.Bounds.xMax); // Align horizontally somewhat newPos.y = existingRoom.Bounds.yMax + corridorLength; // Corridor starts from top-center of existing room corridorX = existingRoom.Center.x - halfCorridorWidth; corridorY = existingRoom.Bounds.yMax; corridorW = corridorWidth; corridorH = corridorLength + (newPos.y - existingRoom.Bounds.yMax); // Extend corridor to meet new room edge // Adjust corridor if new room is offset int connectPointXN = Mathf.Clamp(corridorX + halfCorridorWidth, newPos.x, newPos.x + newWidth - 1); // We might need an L-shaped corridor here, simplified for now break; case 1: // East newPos.x = existingRoom.Bounds.xMax + corridorLength; newPos.y = Random.Range(existingRoom.Bounds.yMin - newHeight + 1, existingRoom.Bounds.yMax); // Align vertically somewhat // Corridor starts from right-center of existing room corridorX = existingRoom.Bounds.xMax; corridorY = existingRoom.Center.y - halfCorridorWidth; corridorW = corridorLength + (newPos.x - existingRoom.Bounds.xMax); // Extend corridor corridorH = corridorWidth; break; case 2: // South newPos.x = Random.Range(existingRoom.Bounds.xMin - newWidth + 1, existingRoom.Bounds.xMax); // Align horizontally somewhat newPos.y = existingRoom.Bounds.yMin - newHeight - corridorLength; // Corridor starts from bottom-center of existing room corridorX = existingRoom.Center.x - halfCorridorWidth; corridorY = newPos.y + newHeight; // Start corridor from top of new room space corridorW = corridorWidth; corridorH = corridorLength + (existingRoom.Bounds.yMin - (newPos.y + newHeight)); // Extend corridor break; case 3: // West newPos.x = existingRoom.Bounds.xMin - newWidth - corridorLength; newPos.y = Random.Range(existingRoom.Bounds.yMin - newHeight + 1, existingRoom.Bounds.yMax); // Align vertically somewhat // Corridor starts from left-center of existing room corridorX = newPos.x + newWidth; // Start corridor from right of new room space corridorY = existingRoom.Center.y - halfCorridorWidth; corridorW = corridorLength + (existingRoom.Bounds.xMin - (newPos.x + newWidth)); // Extend corridor corridorH = corridorWidth; break; } // Clamp positions to map bounds (optional, but good practice) newPos.x = Mathf.Clamp(newPos.x, 0, mapWidth - newWidth); newPos.y = Mathf.Clamp(newPos.y, 0, mapHeight - newHeight); // Create corridor rect (simple straight line for now) // More complex L-shape logic would be needed for perfect alignment corridorRect = new RectInt(corridorX, corridorY, corridorW, corridorH); // --- Simplified Straight Corridor Logic --- // This part creates a simple straight corridor between room centers. // It might overlap rooms if not careful. The overlap check helps mitigate this. Vector2Int startCenter = existingRoom.Center; Vector2Int endCenter = new Vector2Int(newPos.x + newWidth / 2, newPos.y + newHeight / 2); if (Random.value > 0.5f) // Horizontal first, then vertical { corridorRect = CreateCorridorSegment(startCenter.x, endCenter.x, startCenter.y, corridorWidth); // Need a second segment for L-shape, adding only one for simplicity now // RectInt vertCorridor = CreateCorridorSegment(endCenter.y, startCenter.y, endCenter.x, corridorWidth); // Incorrect order/pos } else // Vertical first, then horizontal { corridorRect = CreateCorridorSegment(startCenter.y, endCenter.y, startCenter.x, corridorWidth); // Need a second segment for L-shape // RectInt horizCorridor = CreateCorridorSegment(endCenter.x, startCenter.x, endCenter.y, corridorWidth); // Incorrect order/pos } // For this example, we'll just use one segment of the potential L-shape corridor // A full implementation would create two RectInts for L-shapes. // Let's create a simple horizontal corridor for now for visualization. int cX = Mathf.Min(startCenter.x, endCenter.x); int cY = startCenter.y - halfCorridorWidth; // Centered on starting room's Y int cW = Mathf.Abs(startCenter.x - endCenter.x); int cH = corridorWidth; corridorRect = new RectInt(cX, cY, cW, cH); // TODO: Add vertical segment for a proper L-shape corridor return newPos; } // Helper to create a corridor segment (adjusts for negative width/height) private RectInt CreateCorridorSegment(int startCoord, int endCoord, int fixedCoord, int width) { int pos = Mathf.Min(startCoord, endCoord); int length = Mathf.Abs(startCoord - endCoord); int halfWidth = width / 2; // Assuming horizontal segment if called like CreateCorridorSegment(x1, x2, y, width) // Assuming vertical segment if called like CreateCorridorSegment(y1, y2, x, width) // This needs clearer separation based on axis // Let's assume horizontal for now: return new RectInt(pos, fixedCoord - halfWidth, length, width); // For vertical: return new RectInt(fixedCoord - halfWidth, pos, width, length); } private bool IsOverlapping(Room room, int padding) { RectInt paddedBounds = room.Bounds; paddedBounds.xMin -= padding; paddedBounds.yMin -= padding; paddedBounds.xMax += padding; paddedBounds.yMax += padding; foreach (Room existingRoom in rooms) { if (existingRoom.Bounds.Overlaps(paddedBounds)) { return true; } } return false; } private bool IsCorridorOverlapping(RectInt corridor, Room newRoom, Room connectedRoom) { foreach (Room room in rooms) { // Don't check overlap with the two rooms the corridor connects if (room == newRoom || room == connectedRoom) continue; if (corridor.Overlaps(room.Bounds)) { // Allow minor overlap at the connection point, otherwise reject // This check is basic and might need refinement for L-shapes return true; } } // Optional: Check corridor overlap with other corridors // foreach (RectInt existingCorridor in corridors) { ... } return false; } private void OnDrawGizmos() { if (rooms == null) return; Gizmos.color = Color.white; foreach (Room room in rooms) { Gizmos.DrawWireCube(new Vector3(room.Center.x, 0, room.Center.y), // Use y as z for 3D view new Vector3(room.Bounds.width, 1, room.Bounds.height)); } Gizmos.color = Color.gray; if (corridors != null) { foreach (RectInt corridor in corridors) { Gizmos.DrawCube(new Vector3(corridor.center.x, 0, corridor.center.y), // Use y as z new Vector3(corridor.width, 0.8f, corridor.height)); // Slightly lower cube } } // Draw Map Bounds Gizmos.color = Color.yellow; Gizmos.DrawWireCube(new Vector3(mapWidth / 2f, 0, mapHeight / 2f), new Vector3(mapWidth, 1, mapHeight)); } } using UnityEngine; // Using a class allows easy reference passing if needed later, // but a struct might be slightly more performant for simple data. public class Room { public RectInt Bounds { get; private set; } public Vector2Int Center => Bounds.center; public Room(int x, int y, int width, int height) { Bounds = new RectInt(x, y, width, height); } // Optional: Add methods for checking connections, distance, etc. }