1 files changed, 78 insertions, 0 deletions
diff --git a/src/tool/rect_room_tool.rs b/src/tool/rect_room_tool.rs
new file mode 100644
index 0000000..3eb40aa
--- /dev/null
+++ b/src/tool/rect_room_tool.rs
@@ -0,0 +1,78 @@
+use super::Tool;
+use crate::map::Map;
+use crate::math::{Rect, Vec2};
+use crate::transform::Transform;
+use raylib::core::drawing::{RaylibDraw, RaylibDrawHandle};
+use raylib::ffi::Color;
+
+pub struct RectRoomTool {
+    /// The rectangle that is currently being drawn by the user. Once it is finished, it will be
+    /// pushed into the room_rects.
+    unfinished_rect: Option<(Vec2<f64>, Vec2<f64>)>,
+}
+
+impl RectRoomTool {
+    /// Create a new room tool where no rooms have been drawn yet.
+    pub fn new() -> Self {
+        Self {
+            unfinished_rect: None,
+        }
+    }
+}
+
+impl Tool for RectRoomTool {
+    fn deactivate(&mut self) {
+        self.unfinished_rect = None;
+    }
+
+    fn update(&mut self, _map: &Map, mouse_pos_m: &Vec2<f64>) {
+        if let Some((_, ref mut pos2)) = &mut self.unfinished_rect {
+            *pos2 = *mouse_pos_m;
+        }
+    }
+
+    fn draw(&self, rld: &mut RaylibDrawHandle, transform: &Transform) {
+        if let Some((pos1, pos2)) = self.unfinished_rect {
+            rld.draw_rectangle_rec(
+                transform.rect_m_to_px(&Rect::bounding_rect(pos1, pos2)),
+                Color {
+                    r: 150,
+                    g: 200,
+                    b: 150,
+                    a: 255,
+                },
+            );
+        }
+    }
+
+    fn place_single(&mut self, map: &mut Map, mouse_pos_m: &Vec2<f64>) {
+        // Try to finish the rectangle if it has been started.
+        if let Some((pos1, pos2)) = self.unfinished_rect {
+            if pos1 == pos2 {
+                warn!("Cannot place rectangle with start and endpoint being the same");
+                return;
+            }
+
+            map.push_rect_room(Rect::bounding_rect(pos1, pos2));
+            self.unfinished_rect = None;
+        } else {
+            self.unfinished_rect = Some((*mouse_pos_m, *mouse_pos_m));
+        }
+    }
+
+    fn finish(&mut self, map: &mut Map) {
+        if let Some((pos1, pos2)) = self.unfinished_rect {
+            if pos1 == pos2 {
+                warn!("Cannot place rectangle with start and endpoint being the same");
+                return;
+            }
+
+            map.push_rect_room(Rect::bounding_rect(pos1, pos2));
+            self.unfinished_rect = None;
+        }
+    }
+
+    fn abort(&mut self) {
+        self.unfinished_rect = None;
+    }
+}