feat: improved scoring for (semi)dead positions

1 files changed, 11 insertions, 0 deletions

diff --git a/src/board/piece.rs b/src/board/piece.rs
index 1889e94..3de27ca 100644
--- a/src/board/piece.rs
+++ b/src/board/piece.rs
@@ -1,3 +1,5 @@
+use super::color::Color;
+
 #[derive(Debug, Clone, Copy, PartialEq, Eq, num_enum::FromPrimitive)]
 #[repr(usize)]
 pub enum Piece {
@@ -21,6 +23,15 @@ impl Piece {
         let index = *self as usize;
         Self::from(index % 6)
     }
+
+    /// Note: assumes self is without color!
+    pub fn colored(&self, color: Color) -> Self {
+        match color {
+            Color::White => *self,
+            Color::Black => Self::from(*self as usize + 6)
+        }
+    }
+
     // Return the price of the peice
     pub fn static_eval(&self) -> f32 {
         match self.without_color() {