Add BitBoard::to_matrix_and_offset Python method

nnmm · Jul 26, 2024 · edd0f64 · edd0f64
1 parent 8babd2c
commit edd0f64
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/gomori/src/board/bitboard.rs b/gomori/src/board/bitboard.rs
@@ -457,6 +457,17 @@ mod python {
         fn __bool__(&self) -> bool {
             !self.is_empty()
         }
+
+        // Python exclusive
+        fn to_matrix_and_offset(&self) -> ([[bool; 7]; 7], (i8, i8)) {
+            let arr = std::array::from_fn(|i_local| {
+                std::array::from_fn(|j_local| {
+                    let idx = i_local * 7 + j_local;
+                    self.bits & (1u64 << idx) != 0
+                })
+            });
+            (arr, self.offset())
+        }
     }
 }