1 files changed, 59 insertions, 0 deletions
diff --git a/src/editor.rs b/src/editor.rs
new file mode 100644
index 0000000..9dcbefc
--- /dev/null
+++ b/src/editor.rs
@@ -0,0 +1,59 @@
+use crate::map_data::MapData;
+use crate::tool::*;
+use crate::transform::Transform;
+use raylib::core::drawing::RaylibDrawHandle;
+use raylib::ffi::KeyboardKey;
+use raylib::RaylibHandle;
+
+pub struct Editor {
+    map_data: MapData,
+    tools: Vec<Box<dyn Tool>>,
+    active: usize,
+}
+
+impl Editor {
+    pub fn new() -> Self {
+        let mut tools: Vec<Box<dyn Tool>> = Vec::with_capacity(ToolType::NumTools as usize);
+        assert_eq!(ToolType::RoomTool as u8, 0);
+        tools.push(Box::new(RoomTool::new()));
+        assert_eq!(ToolType::WallTool as u8, 1);
+        tools.push(Box::new(WallTool::new()));
+        assert_eq!(ToolType::DeletionTool as u8, 2);
+        tools.push(Box::new(DeletionTool::new()));
+
+        assert_eq!(ToolType::NumTools as usize, tools.len());
+
+        Self {
+            map_data: MapData::new(),
+            tools,
+            active: 0,
+        }
+    }
+
+    pub fn update(&mut self, rl: &RaylibHandle, transform: &Transform) {
+        // Handle keybindings for tool change
+        self.active = if rl.is_key_pressed(KeyboardKey::KEY_R) {
+            ToolType::RoomTool as usize
+        } else if rl.is_key_pressed(KeyboardKey::KEY_W) {
+            ToolType::WallTool as usize
+        } else {
+            self.active
+        };
+
+        for tool in &mut self.tools {
+            tool.update(&self.map_data, rl, transform);
+        }
+
+        self.tools[self.active].active_update(&mut self.map_data, rl, transform);
+    }
+
+    pub fn draw_tools(&self, rld: &mut RaylibDrawHandle, transform: &Transform) {
+        for tool in &self.tools {
+            tool.draw(&self.map_data, rld, transform);
+        }
+    }
+
+    pub fn map_data(&self) -> &MapData {
+        &self.map_data
+    }
+}