namespace MapGeneration { public enum RoomType { General, Entrance, Exit, Key } } using UnityEngine; namespace MapGeneration { public class ConnectionPoint : MonoBehaviour { public bool IsConnected { get; set; } = false; public Room ParentRoom { get; private set; } private void Awake() { ParentRoom = GetComponentInParent(); if (ParentRoom == null) { Debug.LogError("ConnectionPoint must be a child of a GameObject with a Room component.", this); } } // Optional: Visualize connection points in the editor private void OnDrawGizmos() { Gizmos.color = IsConnected ? Color.red : Color.green; Gizmos.DrawSphere(transform.position, 0.3f); // Draw a small sphere Gizmos.color = Color.blue; Gizmos.DrawLine(transform.position, transform.position + transform.forward * 1.0f); // Show direction } private void OnDrawGizmosSelected() { Gizmos.color = Color.yellow; Gizmos.DrawSphere(transform.position, 0.4f); } } } using UnityEngine; using System.Collections.Generic; using System.Linq; namespace MapGeneration { [RequireComponent(typeof(BoxCollider))] public class Room : MonoBehaviour { [Tooltip("Type of the room (Entrance, Exit, Key, General).")] public RoomType Type = RoomType.General; [Tooltip("Points where other rooms can connect.")] [SerializeField] private List connectionPoints = new List(); private BoxCollider roomBoundsCollider; private bool isPlaced = false; // Flag to prevent overlap checks before initial placement public Bounds RoomBounds => roomBoundsCollider != null ? roomBoundsCollider.bounds : new Bounds(transform.position, Vector3.one); // Provide default if collider not ready private void Awake() { roomBoundsCollider = GetComponent(); if (roomBoundsCollider == null) { Debug.LogError("Room requires a BoxCollider component for bounds checking.", this); // Optionally add one dynamically if needed, but it's better to set it up in the prefab // roomBoundsCollider = gameObject.AddComponent(); // roomBoundsCollider.isTrigger = true; // Important for overlap checks } else { // Ensure the collider used for overlap checks is a trigger // Or ensure physics settings prevent collisions between rooms if using non-triggers if (!roomBoundsCollider.isTrigger) { Debug.LogWarning($"Room '{gameObject.name}' BoxCollider should ideally be a Trigger for overlap checks. Setting isTrigger=true.", this); roomBoundsCollider.isTrigger = true; } } // Automatically find ConnectionPoint children if not assigned if (connectionPoints == null || connectionPoints.Count == 0) { connectionPoints = GetComponentsInChildren().ToList(); if (connectionPoints.Count == 0) { Debug.LogWarning($"Room '{gameObject.name}' has no ConnectionPoint children assigned or found.", this); } } } public List GetAvailableConnectionPoints() { return connectionPoints.Where(cp => cp != null && !cp.IsConnected).ToList(); } public void MarkAsPlaced() { isPlaced = true; } ///

/// Checks if this room overlaps with any other rooms on the specified layer. /// Only checks if the room has been marked as 'placed'. ///

/// The layer mask containing other rooms. /// An optional margin to add to the bounds check. /// True if overlapping, false otherwise. public bool CheckOverlap(LayerMask roomLayerMask, float margin = 0.1f) { if (!isPlaced || roomBoundsCollider == null) return false; // Don't check before placement or if no collider // Slightly inflate the bounds to prevent perfectly touching colliders from being missed Vector3 checkSize = roomBoundsCollider.size + Vector3.one * margin; Vector3 checkCenter = transform.TransformPoint(roomBoundsCollider.center); // Use collider center in world space // Find colliders overlapping the box Collider[] overlaps = Physics.OverlapBox(checkCenter, checkSize / 2f, transform.rotation, roomLayerMask, QueryTriggerInteraction.Collide); foreach (Collider overlap in overlaps) { // Ignore self and potentially the room we just connected to (if necessary, though layer mask should handle this) if (overlap.gameObject != this.gameObject && overlap.GetComponent() != null) { // Debug.LogWarning($"Overlap detected between {this.name} and {overlap.name}", this); return true; // Found an overlap with another room } } return false; // No overlaps found } ///

/// Call this if an overlap is detected after placement attempt. ///

public void HandleOverlap() { // Option 1: Destroy the overlapping room immediately // Debug.LogWarning($"Room {name} is overlapping and will self-destruct.", this); // Destroy(gameObject); // Option 2: Signal the generator to retry (preferred) // The generator should handle the destruction or repositioning. } // Optional: Visualize room bounds in the editor private void OnDrawGizmosSelected() { if (roomBoundsCollider != null) { Gizmos.color = Color.cyan; // Draw wireframe cube matching the BoxCollider's bounds in world space Matrix4x4 originalMatrix = Gizmos.matrix; Gizmos.matrix = Matrix4x4.TRS(transform.TransformPoint(roomBoundsCollider.center), transform.rotation, transform.lossyScale); Gizmos.DrawWireCube(Vector3.zero, roomBoundsCollider.size); Gizmos.matrix = originalMatrix; // Restore original matrix } } } } using UnityEngine; using System.Collections.Generic; using System.Linq; namespace MapGeneration { public class MapGenerator : MonoBehaviour { [Header("Room Settings")] [SerializeField] private GameObject[] roomPrefabs; // Allow multiple prefabs [SerializeField] private int minRooms = 10; [SerializeField] private int maxRooms = 20; [SerializeField] private int numKeyRooms = 2; [Header("Generation Settings")] [SerializeField] private int maxPlacementAttemptsPerRoom = 50; // Prevents infinite loops [SerializeField] private LayerMask roomLayerMask; // Layer used for rooms (for overlap checks) [SerializeField] private float overlapCheckMargin = 0.1f; // Small margin for overlap checks [Header("Debug")] [SerializeField] private bool generateOnStart = true; [SerializeField] private Transform generationParent; // Optional parent for generated rooms private List placedRooms = new List(); private List availableConnectionPoints = new List(); private int targetRoomCount; private int currentKeyRooms = 0; private void Start() { if (generateOnStart) { GenerateMap(); } } [ContextMenu("Generate Map")] public void GenerateMap() { ClearMap(); if (roomPrefabs == null || roomPrefabs.Length == 0) { Debug.LogError("No room prefabs assigned to MapGenerator.", this); return; } if (roomLayerMask == 0) // LayerMask is an int, 0 means 'Nothing' { Debug.LogError("Room Layer Mask is not set in MapGenerator. Overlap checks will fail.", this); return; } targetRoomCount = Random.Range(minRooms, maxRooms + 1); if (numKeyRooms + 2 > targetRoomCount) // +2 for Entrance and Exit { Debug.LogWarning($"Target room count ({targetRoomCount}) is too small for the required number of key rooms ({numKeyRooms}) plus Entrance and Exit. Adjusting target count.", this); targetRoomCount = numKeyRooms + 2; } if (!CreateInitialRoom()) return; int attempts = 0; const int maxTotalAttempts = 1000; // Safety break for the entire generation while (placedRooms.Count < targetRoomCount && availableConnectionPoints.Count > 0 && attempts < maxTotalAttempts) { if (!TryPlaceRoom()) { // Optional: Could implement backtracking here if a dead end is reached // For now, we just stop if we run out of valid places from available points Debug.LogWarning($"Failed to place room {placedRooms.Count + 1} after multiple attempts. Stopping generation.", this); // break; // Decide if we should stop or let it try other points } attempts++; } if (placedRooms.Count < targetRoomCount && availableConnectionPoints.Count == 0) { Debug.LogWarning($"Generation stopped: Ran out of available connection points before reaching target room count ({placedRooms.Count}/{targetRoomCount}).", this); } else if (attempts >= maxTotalAttempts) { Debug.LogWarning($"Generation stopped: Reached maximum total attempts ({maxTotalAttempts}).", this); } AssignExitRoom(); Debug.Log($"Map generation complete. Placed {placedRooms.Count} rooms.", this); } [ContextMenu("Clear Map")] public void ClearMap() { // Clear previous generation placedRooms.Clear(); availableConnectionPoints.Clear(); currentKeyRooms = 0; // Find the parent transform or use this generator's transform Transform parent = generationParent != null ? generationParent : transform; // Destroy immediate is needed for editor context menu usage while (parent.childCount > 0) { DestroyImmediate(parent.GetChild(0).gameObject); } } private bool CreateInitialRoom() { GameObject selectedPrefab = GetRandomRoomPrefab(); if (selectedPrefab == null) return false; GameObject roomGO = Instantiate(selectedPrefab, Vector3.zero, Quaternion.identity, generationParent); Room initialRoom = roomGO.GetComponent(); if (initialRoom == null) { Debug.LogError("Initial room prefab does not have a Room component.", this); Destroy(roomGO); return false; } if (initialRoom.gameObject.layer != Mathf.RoundToInt(Mathf.Log(roomLayerMask.value, 2))) { Debug.LogWarning($"Initial Room '{initialRoom.name}' is not on the specified Room Layer. Assigning it.", initialRoom); initialRoom.gameObject.layer = Mathf.RoundToInt(Mathf.Log(roomLayerMask.value, 2)); } initialRoom.Type = RoomType.Entrance; initialRoom.name = "Room_0_Entrance"; initialRoom.MarkAsPlaced(); // Mark as placed *before* adding points placedRooms.Add(initialRoom); availableConnectionPoints.AddRange(initialRoom.GetAvailableConnectionPoints()); return true; } private bool TryPlaceRoom() { if (availableConnectionPoints.Count == 0) return false; // Try placing from a random available connection point int connectionAttempt = 0; while(connectionAttempt < maxPlacementAttemptsPerRoom && availableConnectionPoints.Count > 0) { int randomIndex = Random.Range(0, availableConnectionPoints.Count); ConnectionPoint existingConnection = availableConnectionPoints[randomIndex]; // If the parent room was somehow destroyed, remove the point if (existingConnection == null || existingConnection.ParentRoom == null) { availableConnectionPoints.RemoveAt(randomIndex); connectionAttempt++; continue; } GameObject selectedPrefab = GetRandomRoomPrefab(); if (selectedPrefab == null) return false; // Should not happen if checked earlier GameObject newRoomGO = Instantiate(selectedPrefab, generationParent); // Instantiate without position initially Room newRoom = newRoomGO.GetComponent(); if (newRoom == null) { Debug.LogError("Selected room prefab does not have a Room component.", this); Destroy(newRoomGO); connectionAttempt++; // Count as an attempt continue; // Try another connection point or prefab } if (newRoom.gameObject.layer != Mathf.RoundToInt(Mathf.Log(roomLayerMask.value, 2))) { Debug.LogWarning($"Room '{newRoom.name}' is not on the specified Room Layer. Assigning it.", newRoom); newRoom.gameObject.layer = Mathf.RoundToInt(Mathf.Log(roomLayerMask.value, 2)); } List newRoomConnections = newRoom.GetAvailableConnectionPoints(); if (newRoomConnections.Count == 0) { Debug.LogWarning($"Prefab {selectedPrefab.name} has no available connection points. Skipping.", newRoomGO); Destroy(newRoomGO); connectionAttempt++; // Count as an attempt continue; // Try another connection point or prefab } // Try connecting each point of the new room to the existing point bool placedSuccessfully = false; foreach (ConnectionPoint newConnection in newRoomConnections.OrderBy(x => Random.value)) // Shuffle points { // --- Position and Align --- // 1. Calculate the rotation needed to make the new connection face the existing one. // The target direction for the new connection is the opposite of the existing connection's forward. Quaternion targetRotation = Quaternion.LookRotation(-existingConnection.transform.forward, existingConnection.transform.up); // 2. Apply this rotation relative to the new connection's current orientation. // We want newConnection.forward to align with -existingConnection.forward. newRoom.transform.rotation = targetRotation * Quaternion.Inverse(Quaternion.LookRotation(newConnection.transform.forward, newConnection.transform.up)); // 3. Calculate the position offset needed. // We want newConnection.position to be at existingConnection.position. Vector3 positionOffset = existingConnection.transform.position - newConnection.transform.position; // 4. Apply the offset to the new room's position. newRoom.transform.position += positionOffset; // --- End Position and Align --- // --- Overlap Check --- newRoom.MarkAsPlaced(); // Temporarily mark as placed for the check bool overlaps = newRoom.CheckOverlap(roomLayerMask, overlapCheckMargin); if (!overlaps) { // Success! Finalize placement AssignRoomType(newRoom); newRoom.name = $"Room_{placedRooms.Count}_{newRoom.Type}"; placedRooms.Add(newRoom); // Mark connections as used and remove them from the available list existingConnection.IsConnected = true; newConnection.IsConnected = true; availableConnectionPoints.Remove(existingConnection); // Don't remove newConnection from its own list yet, it might be needed later // Add the *other* connections of the new room to the available list availableConnectionPoints.AddRange(newRoom.GetAvailableConnectionPoints()); // Gets only unconnected ones placedSuccessfully = true; break; // Stop trying connections for this new room } else { newRoom.MarkAsPlaced(); // Revert placed status if overlap check failed } } // End foreach newConnection if (placedSuccessfully) { return true; // Room was placed successfully } else { // Failed to place this room instance from the chosen existing connection point Destroy(newRoomGO); // Clean up the failed instance // Don't remove existingConnection from available list yet, maybe another prefab works connectionAttempt++; // Continue the while loop to try another available connection point } } // End while connectionAttempt // If we exit the while loop, we failed after max attempts or ran out of points Debug.LogWarning($"Failed to place a room after {connectionAttempt} attempts from available points.", this); return false; } private void AssignRoomType(Room room) { // Prioritize Key rooms if needed if (currentKeyRooms < numKeyRooms) { // Check if we *must* place a key room now to meet the target int remainingRoomsToPlace = targetRoomCount - placedRooms.Count; int remainingKeyRoomsNeeded = numKeyRooms - currentKeyRooms; if (remainingKeyRoomsNeeded >= remainingRoomsToPlace) { room.Type = RoomType.Key; currentKeyRooms++; } else { // Decide randomly or based on other logic if it's a key room or general // For simplicity, let's add a chance, but ensure we hit the target eventually float keyRoomChance = (float)remainingKeyRoomsNeeded / remainingRoomsToPlace; if (Random.value < keyRoomChance) // Simple chance { room.Type = RoomType.Key; currentKeyRooms++; } else { room.Type = RoomType.General; } } } else { room.Type = RoomType.General; } // Exit room is assigned at the end } private void AssignExitRoom() { if (placedRooms.Count < 2) return; // Need at least Entrance and one other // Find a room that is not the Entrance and preferably has few connections (a dead end) // Or simply the last placed room, or furthest from entrance. // Option: Furthest room from entrance Room entrance = placedRooms.FirstOrDefault(r => r.Type == RoomType.Entrance); if (entrance == null) return; // Should not happen Room furthestRoom = null; float maxDist = -1f; foreach (Room room in placedRooms) { if (room.Type == RoomType.Entrance) continue; float dist = Vector3.Distance(entrance.transform.position, room.transform.position); if (dist > maxDist) { maxDist = dist; furthestRoom = room; } } if (furthestRoom != null && furthestRoom.Type != RoomType.Key) // Don't override Key rooms if possible { furthestRoom.Type = RoomType.Exit; furthestRoom.name = $"Room_{placedRooms.IndexOf(furthestRoom)}_Exit"; } else if (furthestRoom != null && furthestRoom.Type == RoomType.Key) { // If the furthest is a key room, try the second furthest non-key/non-entrance room // (More complex logic could be added here) // For now, fall back to the last placed non-entrance/non-key room Room lastNonSpecial = placedRooms.LastOrDefault(r => r.Type == RoomType.General); if(lastNonSpecial != null) { lastNonSpecial.Type = RoomType.Exit; lastNonSpecial.name = $"Room_{placedRooms.IndexOf(lastNonSpecial)}_Exit"; } else // If only entrance and key rooms exist, force the last one (that isn't entrance) { Room lastRoom = placedRooms.LastOrDefault(r => r.Type != RoomType.Entrance); if(lastRoom != null) { lastRoom.Type = RoomType.Exit; lastRoom.name = $"Room_{placedRooms.IndexOf(lastRoom)}_Exit"; } } } else { // Fallback: If no suitable room found (e.g., only 2 rooms), make the last non-entrance room the exit Room lastRoom = placedRooms.LastOrDefault(r => r.Type != RoomType.Entrance); if (lastRoom != null) { lastRoom.Type = RoomType.Exit; lastRoom.name = $"Room_{placedRooms.IndexOf(lastRoom)}_Exit"; } } } private GameObject GetRandomRoomPrefab() { if (roomPrefabs == null || roomPrefabs.Length == 0) return null; return roomPrefabs[Random.Range(0, roomPrefabs.Length)]; } } } **Setup Instructions:** 1. **Create Scripts:** Create the four C# scripts (`RoomType.cs`, `ConnectionPoint.cs`, `Room.cs`, `MapGenerator.cs`) in your `Assets/Scripts` folder (or a subfolder like `Assets/Scripts/MapGeneration`). Copy the code provided above into each corresponding file. 2. **Create Room Layer:** * Go to `Edit -> Project Settings -> Tags and Layers`. * In the `Layers` section, add a new User Layer (e.g., "Rooms"). 3. **Create Room Prefab(s):** * Create a new 3D Object (e.g., Cube) in your scene. Name it something like `RoomPrefab_Basic`. * **Add Components:** * Add the `Room.cs` script to it. * Ensure it has a `BoxCollider`. **Crucially, set `Is Trigger` to `true` on the BoxCollider.** Adjust the collider size to roughly match the visual bounds of your room model. * Set the **Layer** of this GameObject to the "Rooms" layer you created. * **Add Connection Points:** * Create empty GameObjects as children of the `RoomPrefab_Basic`. Name them `ConnectionPoint_North`, `ConnectionPoint_East`, etc. * Position these empty GameObjects on the *outer surface* of your room model where you want connections to happen (e.g., in the middle of each face). * **Important:** Rotate each `ConnectionPoint` GameObject so its **blue Z-axis (forward)** points *outwards* from the room. This defines the connection direction. * Add the `ConnectionPoint.cs` script to each of these child GameObjects. * **Assign Connections (Optional but Recommended):** Drag the `ConnectionPoint` child GameObjects onto the `Connection Points` list in the `Room` script component in the Inspector for the `RoomPrefab_Basic`. (The script will try to find them automatically if you don't, but assigning is safer). * **Create Prefab:** Drag the configured `RoomPrefab_Basic` GameObject from the Hierarchy into your `Assets/Prefabs` folder to create a prefab. You can delete the instance from the scene now. * *(Optional)* Create more room prefabs with different shapes, sizes, and connection point layouts following the same steps. 4. **Setup Map Generator:** * Create an empty GameObject in your scene. Name it `MapGenerator`. * Add the `MapGenerator.cs` script to it. * **Configure in Inspector:** * **Room Prefabs:** Drag your `RoomPrefab_Basic` (and any others you created) from the `Assets/Prefabs` folder onto the `Room Prefabs` list. * **Min/Max Rooms:** Set the desired range for the total number of rooms. * **Num Key Rooms:** Set how many key rooms should be generated. * **Max Placement Attempts:** Leave at default (e.g., 50) unless you encounter issues. * **Room Layer Mask:** Select the "Rooms" layer from the dropdown. **This is essential for overlap checks.** * **Overlap Check Margin:** Leave at default (0.1) unless rooms are spawning too close or too far. * **Generate On Start:** Check this if you want the map to generate when you press Play. * **Generation Parent (Optional):** If you want generated rooms to be children of a specific empty GameObject for organization, create one and drag it here. Otherwise, they will be children of the `MapGenerator` GameObject itself. 5. **Run:** Press Play in the Unity Editor. If `Generate On Start` is checked, the map should generate. You can also right-click the `MapGenerator` script component in the Inspector and select "Generate Map" while the editor is not playing. This setup provides a foundation for a dungeon/map generator. You can expand it by adding different room types, corridor pieces, more sophisticated placement logic, and visual elements.