rust/land2d/src/lib.rs
changeset 14071 4b40bdd214df
parent 13972 03e41712eef8
child 14073 9c817b2eedae
--- a/rust/land2d/src/lib.rs	Mon Oct 29 23:40:17 2018 +0300
+++ b/rust/land2d/src/lib.rs	Mon Oct 29 23:06:18 2018 +0100
@@ -2,25 +2,31 @@
 extern crate vec2d;
 
 use std::cmp;
-use std::ops;
 
 use integral_geometry::{ArcPoints, EquidistantPoints, LinePoints, Point};
 
 pub struct Land2D<T> {
     pixels: vec2d::Vec2D<T>,
+    play_width: usize,
+    play_height: usize,
     width_mask: usize,
     height_mask: usize,
 }
 
 impl<T: Copy + PartialEq> Land2D<T> {
-    pub fn new(width: usize, height: usize, fill_value: T) -> Self {
-        assert!(width.is_power_of_two());
-        assert!(height.is_power_of_two());
+    pub fn new(play_width: usize, play_height: usize, fill_value: T) -> Self {
+        let real_width = play_width.next_power_of_two();
+        let real_height = play_height.next_power_of_two();
+
+        assert!(real_width > 0);
+        assert!(real_height > 0);
 
         Self {
-            pixels: vec2d::Vec2D::new(width, height, fill_value),
-            width_mask: !(width - 1),
-            height_mask: !(height - 1),
+            pixels: vec2d::Vec2D::new(real_width, real_height, fill_value),
+            play_width,
+            play_height,
+            width_mask: !(real_width - 1),
+            height_mask: !(real_height - 1),
         }
     }
 
@@ -35,6 +41,16 @@
     }
 
     #[inline]
+    pub fn play_width(&self) -> usize {
+        self.play_width
+    }
+
+    #[inline]
+    pub fn play_height(&self) -> usize {
+        self.play_height
+    }
+
+    #[inline]
     pub fn is_valid_x(&self, x: i32) -> bool {
         (x as usize & self.width_mask) == 0
     }
@@ -240,7 +256,12 @@
 
     #[test]
     fn basics() {
-        let l: Land2D<u8> = Land2D::new(32, 64, 0);
+        let l: Land2D<u8> = Land2D::new(30, 50, 0);
+
+        assert_eq!(l.play_width(), 30);
+        assert_eq!(l.play_height(), 50);
+        assert_eq!(l.width(), 32);
+        assert_eq!(l.height(), 64);
 
         assert!(l.is_valid_coordinate(0, 0));
         assert!(!l.is_valid_coordinate(-1, -1));