From e2934b95cbad65bc89cbc285c3ece2124bf44a55 Mon Sep 17 00:00:00 2001
From: eug-vs <eugene@eug-vs.xyz>
Date: Sun, 3 Sep 2023 19:29:54 +0300
Subject: feat: improved scoring for (semi)dead positions

---
 src/board/piece.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'src/board/piece.rs')

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() {
-- 
cgit v1.2.3