1 files changed, 22 insertions, 12 deletions
diff --git a/src/map/polygon_room.rs b/src/map/polygon_room.rs
index a209a7c..fd4122e 100644
--- a/src/map/polygon_room.rs
+++ b/src/map/polygon_room.rs
@@ -1,12 +1,18 @@
+//! Polygon rooms are the standard rooms in graf karto. They can take the form of anything that a
+//! [Polygon](crate::math::Polygon) can have.
+
 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 nalgebra::{Matrix3, Point2};
 use raylib::drawing::{RaylibDraw, RaylibDrawHandle};
 
+/// Data type for the Polygon room.
 pub type PolygonRoomData = Polygon<f64>;
 
+/// A polygon room, which can be placed and modified in the world.
 pub struct PolygonRoom {
     data: PolygonRoomData,
     // The polygon shape, but in triangles, so the polygon does not have to be triangulated every frame.
@@ -15,6 +21,7 @@ pub struct PolygonRoom {
 }
 
 impl PolygonRoom {
+    /// Create a room from the given polygon data.
     pub fn from_data(data: PolygonRoomData) -> Self {
         Self {
             data: data.clone(),
@@ -23,6 +30,8 @@ impl PolygonRoom {
         }
     }
 
+    // When the internal polygon changes, it must be retriangulated to be drawn on the screen
+    // properly, so this function must be called any time that happens.
     fn retriangulate(&mut self) {
         self.triangulated = math::triangulate(self.data.clone());
     }
@@ -56,20 +65,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();