using UnityEngine; using System.Collections.Generic; public class LevelGenerator : MonoBehaviour { [Header("Generation Settings")] [SerializeField] private int gridWidth = 10; [SerializeField] private int gridHeight = 10; [SerializeField] private float roomSize = 20f; [SerializeField] private int maxRooms = 15; [SerializeField] private int seed = 0; [Header("Room Prefabs")] [SerializeField] private GameObject[] startRoomPrefabs; [SerializeField] private GameObject[] normalRoomPrefabs; [SerializeField] private GameObject[] endRoomPrefabs; [SerializeField] private GameObject[] hallwayPrefabs; private Dictionary grid = new Dictionary(); private List openPositions = new List(); private System.Random random; private void Start() { GenerateLevel(); } public void GenerateLevel() { ClearLevel(); random = new System.Random(seed != 0 ? seed : Random.Range(1, int.MaxValue)); // Create start room at center Vector2Int startPos = new Vector2Int(gridWidth / 2, gridHeight / 2); CreateRoom(startPos, RoomType.Start); // Generate rooms while (grid.Count < maxRooms && openPositions.Count > 0) { int randomIndex = random.Next(0, openPositions.Count); Vector2Int pos = openPositions[randomIndex]; openPositions.RemoveAt(randomIndex); CreateRoom(pos, RoomType.Normal); } // Create end room if (grid.Count > 1) { Vector2Int endPos = FindFurthestRoom(startPos); grid[endPos].ChangeType(RoomType.End); } // Connect rooms with hallways ConnectRooms(); } private void CreateRoom(Vector2Int pos, RoomType type) { if (grid.ContainsKey(pos)) return; GameObject[] prefabs = GetPrefabsForType(type); if (prefabs.Length == 0) return; GameObject prefab = prefabs[random.Next(0, prefabs.Length)]; Vector3 worldPos = new Vector3(pos.x * roomSize, 0, pos.y * roomSize); GameObject roomObj = Instantiate(prefab, worldPos, Quaternion.identity, transform); Room room = roomObj.GetComponent() ?? roomObj.AddComponent(); room.Initialize(pos, type); grid.Add(pos, room); AddOpenPositions(pos); } private GameObject[] GetPrefabsForType(RoomType type) { return type switch { RoomType.Start => startRoomPrefabs, RoomType.End => endRoomPrefabs, _ => normalRoomPrefabs }; } private void AddOpenPositions(Vector2Int pos) { Vector2Int[] directions = new Vector2Int[] { new Vector2Int(1, 0), new Vector2Int(-1, 0), new Vector2Int(0, 1), new Vector2Int(0, -1) }; foreach (Vector2Int dir in directions) { Vector2Int newPos = pos + dir; if (!grid.ContainsKey(newPos) && newPos.x >= 0 && newPos.x < gridWidth && newPos.y >= 0 && newPos.y < gridHeight) { if (!openPositions.Contains(newPos)) { openPositions.Add(newPos); } } } } private Vector2Int FindFurthestRoom(Vector2Int startPos) { Vector2Int furthestPos = startPos; float maxDistance = 0; foreach (var kvp in grid) { float distance = Vector2Int.Distance(startPos, kvp.Key); if (distance > maxDistance) { maxDistance = distance; furthestPos = kvp.Key; } } return furthestPos; } private void ConnectRooms() { foreach (var kvp in grid) { Room room = kvp.Value; Vector2Int[] directions = new Vector2Int[] { new Vector2Int(1, 0), new Vector2Int(0, 1) }; foreach (Vector2Int dir in directions) { Vector2Int neighborPos = room.GridPosition + dir; if (grid.TryGetValue(neighborPos, out Room neighbor)) { CreateHallway(room, neighbor, dir); } } } } private void CreateHallway(Room roomA, Room roomB, Vector2Int direction) { if (hallwayPrefabs.Length == 0) return; Vector3 midpoint = (roomA.transform.position + roomB.transform.position) / 2f; Quaternion rotation = direction.x != 0 ? Quaternion.Euler(0, 90, 0) : Quaternion.identity; GameObject hallway = Instantiate( hallwayPrefabs[random.Next(0, hallwayPrefabs.Length)], midpoint, rotation, transform ); roomA.AddConnection(hallway); roomB.AddConnection(hallway); } private void ClearLevel() { foreach (Transform child in transform) { Destroy(child.gameObject); } grid.Clear(); openPositions.Clear(); } } using UnityEngine; public enum RoomType { Start, Normal, End } public class Room : MonoBehaviour { public Vector2Int GridPosition { get; private set; } public RoomType Type { get; private set; } public void Initialize(Vector2Int gridPosition, RoomType type) { GridPosition = gridPosition; Type = type; name = $"{type} Room ({gridPosition.x},{gridPosition.y})"; } public void ChangeType(RoomType newType) { Type = newType; name = $"{newType} Room ({GridPosition.x},{GridPosition.y})"; } public void AddConnection(GameObject hallway) { hallway.transform.SetParent(transform); } } #if UNITY_EDITOR using UnityEditor; using UnityEngine; [CustomEditor(typeof(LevelGenerator))] public class LevelGeneratorEditor : Editor { public override void OnInspectorGUI() { base.OnInspectorGUI(); LevelGenerator generator = (LevelGenerator)target; GUILayout.Space(10); if (GUILayout.Button("Generate Level")) { generator.GenerateLevel(); } if (GUILayout.Button("Clear Level")) { generator.ClearLevel(); } } } #endif