rust/landgen/src/wavefront_collapse/tile_image.rs
branchtransitional_engine
changeset 15948 9bd828451d77
parent 15947 60b5639cc3a5
child 15949 659c92124c26
--- a/rust/landgen/src/wavefront_collapse/tile_image.rs	Fri Feb 03 15:59:18 2023 +0100
+++ b/rust/landgen/src/wavefront_collapse/tile_image.rs	Sun Feb 12 14:19:02 2023 +0100
@@ -1,4 +1,5 @@
-use super::transform::RotationTransform;
+use super::transform::Transform;
+use integral_geometry::Size;
 use std::rc::Rc;
 use vec2d::Vec2D;
 
@@ -25,12 +26,16 @@
             reverse: !self.symmetrical && !self.reverse,
         }
     }
+
+    pub fn is_compatible(&self, other: &Self) -> bool {
+        self.id == other.id && (self.reverse != other.reverse || self.symmetrical)
+    }
 }
 
 #[derive(Clone)]
 pub struct TileImage<T, I: PartialEq + Clone> {
     image: Rc<Vec2D<T>>,
-    transform: RotationTransform,
+    transform: Transform,
     top: Edge<I>,
     right: Edge<I>,
     bottom: Edge<I>,
@@ -47,7 +52,7 @@
     ) -> Self {
         Self {
             image: Rc::new(image),
-            transform: RotationTransform::default(),
+            transform: Transform::default(),
             top,
             right,
             bottom,
@@ -91,7 +96,7 @@
     pub fn rotated180(&self) -> Self {
         Self {
             image: self.image.clone(),
-            transform: self.transform.rotate90(),
+            transform: self.transform.rotate180(),
             top: self.bottom.clone(),
             right: self.left.clone(),
             bottom: self.top.clone(),
@@ -102,7 +107,7 @@
     pub fn rotated270(&self) -> Self {
         Self {
             image: self.image.clone(),
-            transform: self.transform.rotate90(),
+            transform: self.transform.rotate270(),
             top: self.left.clone(),
             right: self.top.clone(),
             bottom: self.right.clone(),
@@ -125,6 +130,48 @@
     pub fn top_edge(&self) -> &Edge<I> {
         &self.top
     }
+
+    pub fn size(&self) -> Size {
+        match self.transform {
+            Transform::Rotate0(_) => self.image.size(),
+            Transform::Rotate90(_) => Size::new(self.image.size().height, self.image.size().width),
+        }
+    }
+
+    pub fn get(&self, row: usize, column: usize) -> Option<&T> {
+        match self.transform {
+            Transform::Rotate0(_) => {
+                let image_row = if self.transform.is_flipped() {
+                    self.image.height().wrapping_sub(1).wrapping_sub(row)
+                } else {
+                    row
+                };
+
+                let image_column = if self.transform.is_mirrored() {
+                    self.image.width().wrapping_sub(1).wrapping_sub(column)
+                } else {
+                    column
+                };
+
+                self.image.get(image_row, image_column)
+            },
+            Transform::Rotate90(_) => {
+                let image_row = if self.transform.is_flipped() {
+                    column
+                } else {
+                    self.image.height().wrapping_sub(1).wrapping_sub(column)
+                };
+
+                let image_column = if self.transform.is_mirrored() {
+                    self.image.width().wrapping_sub(1).wrapping_sub(row)
+                } else {
+                    row
+                };
+
+                self.image.get(image_row, image_column)
+            },
+        }
+    }
 }
 
 #[cfg(test)]