Add dimensional indicator with scaling

1 files changed, 14 insertions, 12 deletions

diff --git a/src/map/polygon_room.rs b/src/map/polygon_room.rs
index a209a7c..65eeaab 100644
--- a/src/map/polygon_room.rs
+++ b/src/map/polygon_room.rs
@@ -1,9 +1,10 @@
 use super::Mappable;
 use crate::colours::DEFAULT_COLOURS;
-use crate::math::{self, Polygon, Rect, Triangle, Vec2};
-use crate::scaleable::Scaleable;
+use crate::math::{self, Polygon, Rect, Triangle};
 use crate::transform::Transform;
+use crate::transformable::NonRigidTransformable;
 use raylib::drawing::{RaylibDraw, RaylibDrawHandle};
+use nalgebra::{Matrix3, Point2};
 
 pub type PolygonRoomData = Polygon<f64>;
 
@@ -56,20 +57,21 @@ impl Mappable for PolygonRoom {
         Rect::bounding_rect_n(&self.data.corners())
     }
 
-    fn as_scaleable(&self) -> Option<&dyn Scaleable> {
-        Some(self as &dyn Scaleable)
+    fn as_non_rigid(&self) -> Option<&dyn NonRigidTransformable> {
+        Some(self as &dyn NonRigidTransformable)
     }
-}
 
-impl Scaleable for PolygonRoom {
-    fn scale(&mut self, by: &Vec2<f64>) {
-        if by.x < 0. || by.y < 0. {
-            panic!("Cannot set dimensions with negative size");
-        }
+    fn as_non_rigid_mut(&mut self) -> Option<&mut dyn NonRigidTransformable> {
+        Some(self as &mut dyn NonRigidTransformable)
+    }
+}
 
+impl NonRigidTransformable for PolygonRoom {
+    fn apply_matrix(&mut self, matrix: &Matrix3<f64>) {
         for corner in self.data.corners_mut() {
-            corner.x *= by.x;
-            corner.y *= by.y;
+            *corner = matrix
+                .transform_point(&Point2::new(corner.x, corner.y))
+                .into();
         }
 
         self.retriangulate();