Use u128 variable for pieces

martinnovaak · web-flow · commit 8c9c8c7057f8 · 2024-02-02T15:02:05.000+01:00
diff --git a/dataparser/src/position.rs b/dataparser/src/position.rs
@@ -3,7 +3,7 @@
 #[derive(Debug, Clone, Copy, PartialEq, Default)]
 pub struct Position {
     occupancy: u64,
-    pieces: [u8; 16],
+    pieces: u128,
     score: i16,
     result: u8,
     stm_king: u8,
@@ -24,16 +24,15 @@ impl Position {
             F: FnMut(i16, i16),
     {
         let mut occupancy = self.occupancy;
-        let mut index = 0;
+        let mut pieces = self.pieces;
         const PIECE_MIRROR: [i16; 12] = [384, 448, 512, 576, 640, 704, 0, 64, 128, 192, 256, 320];
 
         while occupancy != 0 {
             let square = occupancy.trailing_zeros() as i16;
-            let shifts = [1, 16];
-            let piece = ((self.pieces[index / 2] / shifts[index % 2]) & 0b1111) as i16;
+            let piece = (pieces & 0b1111) as i16;
 
             occupancy &= occupancy - 1;
-            index += 1;
+            pieces >>= 4;
 
             let stm_feature = piece * 64 + square;
             let nstm_feature = PIECE_MIRROR[piece as usize] + square ^ 56;
