Spaces:
Running
Running
File size: 2,620 Bytes
fdcbf65 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import random
from helper import run_action
class Dungeon:
def __init__(self, width, height):
self.width = width
self.height = height
self.grid = self.generate_dungeon()
self.items = []
self.npcs = []
self.story = ""
self.current_room = "entrance"
def generate_dungeon(self):
# Enhanced procedural generation
grid = [["." for _ in range(self.width)] for _ in range(self.height)]
# Create rooms
rooms = []
for _ in range(3):
room_w = random.randint(3, 5)
room_h = random.randint(3, 5)
x = random.randint(1, self.width - room_w - 1)
y = random.randint(1, self.height - room_h - 1)
rooms.append((x, y, room_w, room_h))
# Carve out room
for i in range(y, y + room_h):
for j in range(x, x + room_w):
grid[i][j] = "."
# Connect rooms with corridors
for i in range(len(rooms) - 1):
x1, y1 = rooms[i][0] + rooms[i][2] // 2, rooms[i][1] + rooms[i][3] // 2
x2, y2 = (
rooms[i + 1][0] + rooms[i + 1][2] // 2,
rooms[i + 1][1] + rooms[i + 1][3] // 2,
)
# Horizontal corridor
for x in range(min(x1, x2), max(x1, x2) + 1):
grid[y1][x] = "."
# Vertical corridor
for y in range(min(y1, y2), max(y1, y2) + 1):
grid[y2][x2] = "."
return grid
def display(self, level=0, player_pos=None):
"""Return ASCII representation of dungeon"""
grid = self.grid.copy()
# Mark player position
if player_pos:
x, y = player_pos
if 0 <= x < self.width and 0 <= y < self.height:
grid[y][x] = "@"
# Mark NPCs
for npc in self.npcs:
x, y = npc.position
if 0 <= x < self.width and 0 <= y < self.height:
grid[y][x] = "N"
# Mark items
for item in self.items:
x, y = item.position
if 0 <= x < self.width and 0 <= y < self.height:
grid[y][x] = "i"
# Convert to string
return "\n".join("".join(row) for row in grid)
def get_room_description(self, room_name, game_state):
"""Get AI-generated room description based on game state"""
prompt = f"Describe the {room_name} of the dungeon in the kingdom of {game_state['kingdom']}"
# Use helper.run_action() to get AI description
return run_action(prompt, [], game_state)
|