Add unstable polygon room tool

1 files changed, 63 insertions, 4 deletions

diff --git a/src/tool/polygon_room_tool.rs b/src/tool/polygon_room_tool.rs
index 4aab7f7..b37774b 100644
--- a/src/tool/polygon_room_tool.rs
+++ b/src/tool/polygon_room_tool.rs
@@ -4,9 +4,10 @@ use crate::config::{PolygonRoomToolKeybindings, ToolKeybindings};
 use crate::dimension_indicator::DimensionIndicator;
 use crate::grid::{snap_to_grid, SNAP_SIZE};
 use crate::map_data::MapData;
-use crate::math::{Polygon, Vec2};
+use crate::math::{self, Polygon, Vec2};
 use crate::transform::Transform;
-use raylib::core::drawing::RaylibDrawHandle;
+use raylib::core::drawing::{RaylibDraw, RaylibDrawHandle};
+use raylib::ffi::Color;
 use raylib::RaylibHandle;
 
 pub struct PolygonRoomTool {
@@ -74,7 +75,7 @@ impl Tool for PolygonRoomTool {
                     corners.push(snapped_mouse_pos_m);
                 }
             } else {
-                self.unfinished_polygon = Some(vec![snapped_mouse_pos_m]);
+                self.unfinished_polygon = Some(vec![snapped_mouse_pos_m, snapped_mouse_pos_m]);
             }
         }
 
@@ -83,7 +84,65 @@ impl Tool for PolygonRoomTool {
         }
     }
 
-    fn draw(&self, _map: &MapData, _rld: &mut RaylibDrawHandle, _transform: &Transform) {}
+    fn draw(&self, map: &MapData, rld: &mut RaylibDrawHandle, transform: &Transform) {
+        // TODO: Buffer triangles so the polygons don't always have to be retriangulated.
+        for polygon in map.polygons() {
+            let triangles = math::triangulate(polygon.clone());
+            for triangle in triangles {
+                let triangle: [Vec2<f32>; 3] = triangle.into();
+                rld.draw_triangle(
+                    transform.point_m_to_px(triangle[0]),
+                    transform.point_m_to_px(triangle[1]),
+                    transform.point_m_to_px(triangle[2]),
+                    Color {
+                        r: 180,
+                        g: 180,
+                        b: 180,
+                        a: 255,
+                    },
+                )
+            }
+        }
+
+        // Draw the current polygon
+        if let Some(corners) = &self.unfinished_polygon {
+            let mut corners = corners.clone();
+            corners.dedup();
+            match corners.len() {
+                0 | 1 => {}
+                2 => rld.draw_line_ex(
+                    transform.point_m_to_px(corners[0]),
+                    transform.point_m_to_px(corners[1]),
+                    transform.length_m_to_px(0.1),
+                    Color {
+                        r: 150,
+                        g: 200,
+                        b: 150,
+                        a: 255,
+                    },
+                ),
+                _ => {
+                    let polygon = Polygon::new(corners);
+                    let triangles = math::triangulate(polygon);
+                    for triangle in triangles {
+                        let triangle: [Vec2<f32>; 3] = triangle.into();
+                        rld.draw_triangle(
+                            transform.point_m_to_px(triangle[0]),
+                            transform.point_m_to_px(triangle[1]),
+                            transform.point_m_to_px(triangle[2]),
+                            Color {
+                                r: 150,
+                                g: 200,
+                                b: 150,
+                                a: 255,
+                            },
+                        )
+                    }
+                }
+            }
+            self.dimension_indicator.draw(rld, transform);
+        }
+    }
 
     fn activation_key(&self) -> Button {
         self.keybindings.activation_key()