From a39405f8e26bdd4db304fe1ea13dccf9400d0445 Mon Sep 17 00:00:00 2001
From: Dennis Kobert <dennis@kobert.dev>
Date: Fri, 20 Dec 2019 06:47:20 +0100
Subject: Add bit operation wrapper funftions

---
 src/solvers.rs | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/solvers.rs b/src/solvers.rs
index 9bf0ad6..7e17b38 100644
--- a/src/solvers.rs
+++ b/src/solvers.rs
@@ -74,6 +74,21 @@ impl<T: num::PrimInt> Solver<T> {
     }
 
     pub fn solve(&mut self) -> Wall {
-        Wall::create_empty(self.w)
+        let wall = Wall::create_empty(self.w);
+        wall.heights
+            .iter()
+            .take(2)
+            .zip(0..1)
+            .for_each(|(x, i)| self.set_stone(*x as usize, i));
+        wall
+    }
+
+    fn set_stone(&mut self, row: usize, stone: u32) {
+        self.solve_stack[row] =
+            self.solve_stack[row] | T::from(1 << stone).expect("Stone placing index out of bounds");
+    }
+
+    fn get_stone(&self, row: usize, stone: u32) {
+        self.solve_stack[row] & T::from(1 << stone).expect("Requested stone index out of bounds");
     }
 }
-- 
cgit v1.2.3-54-g00ecf