added movable rooms

This commit is contained in:
grimsace
2026-03-06 09:56:26 -06:00
parent 70eb567cf4
commit edb5807dee
3 changed files with 201 additions and 9 deletions
+195 -9
View File
@@ -1,6 +1,6 @@
mod layout; mod layout;
mod persistence;
mod seed; mod seed;
mod settings;
mod ui; mod ui;
use std::collections::HashSet; use std::collections::HashSet;
@@ -23,11 +23,12 @@ fn main() -> eframe::Result<()> {
struct DungeonApp { struct DungeonApp {
settings: UiSettings, settings: UiSettings,
layout: DungeonLayout, layout: DungeonLayout,
room_drag: Option<RoomDragState>,
} }
impl Default for DungeonApp { impl Default for DungeonApp {
fn default() -> Self { fn default() -> Self {
let settings = persistence::load_settings().unwrap_or_default(); let settings = settings::load_settings().unwrap_or_default();
let layout = generate_layout( let layout = generate_layout(
settings.cols, settings.cols,
settings.rows, settings.rows,
@@ -39,14 +40,17 @@ impl Default for DungeonApp {
settings.corridor_randomness, settings.corridor_randomness,
settings.dead_end_rooms_percent, settings.dead_end_rooms_percent,
); );
Self {
Self { settings, layout } settings,
layout,
room_drag: None,
}
} }
} }
impl Drop for DungeonApp { impl Drop for DungeonApp {
fn drop(&mut self) { fn drop(&mut self) {
if let Err(err) = persistence::save_settings(&self.settings) { if let Err(err) = settings::save_settings(&self.settings) {
eprintln!("Failed to save settings: {err}"); eprintln!("Failed to save settings: {err}");
} }
} }
@@ -60,15 +64,20 @@ impl eframe::App for DungeonApp {
self.settings.max_room_size = self.settings.min_room_size; self.settings.max_room_size = self.settings.min_room_size;
} }
if panel_result.reset_clicked {
self.regenerate_layout();
}
if panel_result.settings_changed { if panel_result.settings_changed {
self.regenerate_layout(); self.regenerate_layout();
} }
egui::CentralPanel::default().show(ctx, |ui| { egui::CentralPanel::default().show(ctx, |ui| {
let available = ui.available_size_before_wrap(); let available = ui.available_size_before_wrap();
let (response, painter) = ui.allocate_painter(available, egui::Sense::hover()); let (response, painter) = ui.allocate_painter(available, egui::Sense::click_and_drag());
let canvas = response.rect.shrink(12.0); let canvas = response.rect.shrink(12.0);
let geometry = draw_grid(&painter, canvas, self.settings.cols, self.settings.rows); let geometry = draw_grid(&painter, canvas, self.settings.cols, self.settings.rows);
self.handle_room_drag(&response, &geometry);
draw_layout(&painter, &geometry, &self.layout); draw_layout(&painter, &geometry, &self.layout);
}); });
} }
@@ -76,6 +85,7 @@ impl eframe::App for DungeonApp {
impl DungeonApp { impl DungeonApp {
fn regenerate_layout(&mut self) { fn regenerate_layout(&mut self) {
self.room_drag = None;
self.layout = generate_layout( self.layout = generate_layout(
self.settings.cols, self.settings.cols,
self.settings.rows, self.settings.rows,
@@ -88,6 +98,148 @@ impl DungeonApp {
self.settings.dead_end_rooms_percent, self.settings.dead_end_rooms_percent,
); );
} }
fn handle_room_drag(&mut self, response: &egui::Response, geometry: &GridGeometry) {
if response.drag_started()
&& let Some(pointer_pos) = response.interact_pointer_pos()
&& let Some((room_idx, offset_x, offset_y)) =
self.room_at_pointer(pointer_pos, geometry)
{
self.room_drag = Some(RoomDragState {
room_idx,
offset_x,
offset_y,
});
}
if response.dragged()
&& let Some(pointer_pos) = response.interact_pointer_pos()
&& let Some(drag) = self.room_drag
{
self.drag_room_to_pointer(drag, pointer_pos, geometry);
}
if response.drag_stopped() || !response.ctx.input(|i| i.pointer.primary_down()) {
self.room_drag = None;
}
}
fn room_at_pointer(
&self,
pointer_pos: egui::Pos2,
geometry: &GridGeometry,
) -> Option<(usize, f32, f32)> {
let (grid_x, grid_y) = pointer_to_grid(pointer_pos, geometry)?;
for (room_idx, room) in self.layout.rooms.iter().enumerate().rev() {
let room_x = room.x as f32;
let room_y = room.y as f32;
let room_right = room_x + room.width as f32;
let room_bottom = room_y + room.height as f32;
if grid_x >= room_x && grid_x < room_right && grid_y >= room_y && grid_y < room_bottom {
return Some((room_idx, grid_x - room_x, grid_y - room_y));
}
}
None
}
fn drag_room_to_pointer(
&mut self,
drag: RoomDragState,
pointer_pos: egui::Pos2,
geometry: &GridGeometry,
) {
let Some((grid_x, grid_y)) = pointer_to_grid(pointer_pos, geometry) else {
return;
};
if drag.room_idx >= self.layout.rooms.len() {
self.room_drag = None;
return;
}
let (old_x, old_y, width, height) = {
let room = &self.layout.rooms[drag.room_idx];
(room.x, room.y, room.width, room.height)
};
let max_x = self.settings.cols.saturating_sub(width);
let max_y = self.settings.rows.saturating_sub(height);
let target_x = (grid_x - drag.offset_x).floor().max(0.0) as usize;
let target_y = (grid_y - drag.offset_y).floor().max(0.0) as usize;
let new_x = target_x.min(max_x);
let new_y = target_y.min(max_y);
if new_x == old_x && new_y == old_y {
return;
}
self.layout.rooms[drag.room_idx].x = new_x;
self.layout.rooms[drag.room_idx].y = new_y;
self.adjust_corridors_for_room_move(old_x, old_y, width, height, new_x, new_y);
}
fn adjust_corridors_for_room_move(
&mut self,
old_x: usize,
old_y: usize,
width: usize,
height: usize,
new_x: usize,
new_y: usize,
) {
let dx = new_x as isize - old_x as isize;
let dy = new_y as isize - old_y as isize;
if dx == 0 && dy == 0 {
return;
}
let room_cells = RoomCells {
x: old_x,
y: old_y,
width,
height,
};
for corridor in &mut self.layout.corridors {
if room_cells.contains(corridor.from) {
corridor.from = shift_cell(
corridor.from,
dx,
dy,
self.settings.cols,
self.settings.rows,
);
}
if room_cells.contains(corridor.to) {
corridor.to =
shift_cell(corridor.to, dx, dy, self.settings.cols, self.settings.rows);
}
}
}
}
#[derive(Debug, Clone, Copy)]
struct RoomDragState {
room_idx: usize,
offset_x: f32,
offset_y: f32,
}
struct RoomCells {
x: usize,
y: usize,
width: usize,
height: usize,
}
impl RoomCells {
fn contains(&self, cell: (usize, usize)) -> bool {
let (cx, cy) = cell;
cx >= self.x && cx < self.x + self.width && cy >= self.y && cy < self.y + self.height
}
} }
struct GridGeometry { struct GridGeometry {
@@ -150,9 +302,12 @@ fn draw_layout(painter: &egui::Painter, geometry: &GridGeometry, layout: &Dungeo
let mut corridor_edges_set = HashSet::new(); let mut corridor_edges_set = HashSet::new();
for corridor in &layout.corridors { for corridor in &layout.corridors {
corridor_edges_set.insert(normalized_edge(corridor.from, corridor.to)); let cells = corridor_cells(corridor.from, corridor.to);
for (col, row) in corridor_cells(corridor.from, corridor.to) { for cell in &cells {
corridor_cells_set.insert((col, row)); corridor_cells_set.insert(*cell);
}
for pair in cells.windows(2) {
corridor_edges_set.insert(normalized_edge(pair[0], pair[1]));
} }
} }
@@ -235,6 +390,15 @@ fn corridor_cells(from: (usize, usize), to: (usize, usize)) -> Vec<(usize, usize
for x in start..=end { for x in start..=end {
cells.push((x, y)); cells.push((x, y));
} }
} else {
// If an endpoint drift makes a segment diagonal, render it as an L path.
let corner = (to.0, from.1);
cells.extend(corridor_cells(from, corner));
for cell in corridor_cells(corner, to) {
if cells.last().copied() != Some(cell) {
cells.push(cell);
}
}
} }
cells cells
} }
@@ -251,3 +415,25 @@ fn cell_rect(geometry: &GridGeometry, col: usize, row: usize) -> egui::Rect {
fn normalized_edge(a: (usize, usize), b: (usize, usize)) -> ((usize, usize), (usize, usize)) { fn normalized_edge(a: (usize, usize), b: (usize, usize)) -> ((usize, usize), (usize, usize)) {
if a <= b { (a, b) } else { (b, a) } if a <= b { (a, b) } else { (b, a) }
} }
fn pointer_to_grid(pointer_pos: egui::Pos2, geometry: &GridGeometry) -> Option<(f32, f32)> {
if !geometry.rect.contains(pointer_pos) {
return None;
}
let x = (pointer_pos.x - geometry.rect.left()) / geometry.cell_size;
let y = (pointer_pos.y - geometry.rect.top()) / geometry.cell_size;
Some((x, y))
}
fn shift_cell(
cell: (usize, usize),
dx: isize,
dy: isize,
cols: usize,
rows: usize,
) -> (usize, usize) {
let x = (cell.0 as isize + dx).clamp(0, cols.saturating_sub(1) as isize) as usize;
let y = (cell.1 as isize + dy).clamp(0, rows.saturating_sub(1) as isize) as usize;
(x, y)
}
+6
View File
@@ -50,6 +50,7 @@ impl Default for UiSettings {
#[derive(Debug, Clone, Copy, Default)] #[derive(Debug, Clone, Copy, Default)]
pub struct SidePanelResult { pub struct SidePanelResult {
pub settings_changed: bool, pub settings_changed: bool,
pub reset_clicked: bool,
} }
pub fn draw_side_panel(ctx: &egui::Context, settings: &mut UiSettings) -> SidePanelResult { pub fn draw_side_panel(ctx: &egui::Context, settings: &mut UiSettings) -> SidePanelResult {
@@ -132,6 +133,11 @@ fn draw_generate_tab(ui: &mut egui::Ui, settings: &mut UiSettings, result: &mut
result.settings_changed = true; result.settings_changed = true;
} }
}); });
ui.add_space(8.0);
if ui.button("Reset").clicked() {
result.reset_clicked = true;
}
} }
fn draw_layout_tab(ui: &mut egui::Ui, settings: &mut UiSettings, result: &mut SidePanelResult) { fn draw_layout_tab(ui: &mut egui::Ui, settings: &mut UiSettings, result: &mut SidePanelResult) {