ArduChess/Board.h

165 lines
3.5 KiB
C++

#ifndef __BOARD_H_INC
#define __BOARD_H_INC
#include "Types.h"
#include "Panic.h"
#include "Move.h"
struct Unmove {
byte sq_from;
byte sq_to;
byte captured;
byte enpassant;
};
class Board {
public:
Board();
void make(Move m);
void unmake();
void print();
bool black_moving() {
return field[PTR_SIDE_AND_CASTLERIGHT] & 0x1;
}
unsigned long get_zobrist() {
long* addr = (long*) &field[PTR_ZOBRIST];
return *addr;
}
byte field[128] = {
W_ROOK, W_KNGT, W_BSHP, W_QUEN, W_KING, W_BSHP, W_KNGT, W_ROOK, 0, 0, 0, 0, 0, 0, 0, 0,
W_PAWN, W_PAWN, W_PAWN, W_PAWN, W_PAWN, W_PAWN, W_PAWN, W_PAWN, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
B_PAWN, B_PAWN, B_PAWN, B_PAWN, B_PAWN, B_PAWN, B_PAWN, B_PAWN, 0, 0, 0, 0, 0, 0, 0, 0,
B_ROOK, B_KNGT, B_BSHP, B_QUEN, B_KING, B_BSHP, B_KNGT, B_ROOK, 0, 0, 0, 0, 0, 0, 0, 0,
};
private:
// Private function defs
void next_unmove();
void prev_unmove();
void store_unmove(Unmove u);
Unmove read_unmove();
// 0x88-fill definitions
static const byte PTR_SIDE_AND_CASTLERIGHT = 0x08; //byte (1=side, 2,4=white castle, 8,16=black)
// CAN FILL 0x09
static const byte PTR_W_KING = 0x0A; // byte (points to index or maybe 64-arr index)
// const byte PTR_B_KING = 0x0B; (PTR_W_KING | COLOR or PTR_W_KING + COLOR)
static const byte PTR_ZOBRIST = 0x0C; // 4 bytes
// 0x0D
// 0x0E
// 0x0F
static const byte PTR_ENPASSANT = 0x18;
static const byte PTR_REVMOV = 0x19;
// free space
static const byte PTR_UNMOVE_START = 0x28;
static const byte PTR_UNMOVE_LAST = 0x7F;
byte PTR_UNMOVE = PTR_UNMOVE_START;
};
Board::Board() {
// Then, other data we need to store is assigned 0-fields.
field[PTR_ZOBRIST+1] = 1;
}
void Board::print() {
for(char i = 7; i >= 0; i--) {
for(byte j = 0; j < 16; j++) {
Serial.print(field[i*16 + j], HEX);
Serial.print(" ");
}
Serial.println();
}
}
void Board::make(Move m) {
Unmove u;
u.captured = field[m.sq_to];
u.sq_from = m.sq_from;
u.sq_to = m.sq_to;
byte new_val = m.pc_prom & 0b1111;
if(new_val != P_EMPTY) {
u.sq_to |= 0x80;
} else {
new_val = field[m.sq_from];
}
field[m.sq_to] = field[m.sq_from];
field[m.sq_from] = P_EMPTY;
// TODO handle castling
// TODO handle castling rights
// TODO handle revmov clock
// TODO handle enpassant
field[PTR_SIDE_AND_CASTLERIGHT] ^= 0x01;
store_unmove(u);
}
void Board::unmake() {
field[PTR_SIDE_AND_CASTLERIGHT] ^= 0x01;
Unmove u = read_unmove();
if((u.sq_to & 0x80) == 0) {
// regular move
field[u.sq_from] = field[u.sq_to];
} else {
// piece was promoted
// so the source is a pawn
field[u.sq_from] = W_PAWN | black_moving() << 3;
}
field[u.sq_to] = u.captured & 0b1111;
}
void Board::next_unmove() {
PTR_UNMOVE++;
if(PTR_UNMOVE > PTR_UNMOVE_LAST) {
panic(F("Unmove stack overflow"));
}
if(!(PTR_UNMOVE & 0x8)) {
PTR_UNMOVE += 0x8;
}
}
void Board::prev_unmove() {
PTR_UNMOVE--;
if(PTR_UNMOVE < PTR_UNMOVE_START) {
panic(F("Unmaking from empty stack"));
}
if(!(PTR_UNMOVE & 0x8)) {
PTR_UNMOVE -= 0x8;
}
}
void Board::store_unmove(Unmove u) {
byte *ub = (byte*) &u;
for(size_t i = 0; i < sizeof(u); i++) {
next_unmove();
field[PTR_UNMOVE] = ub[i];
}
}
Unmove Board::read_unmove() {
Unmove u;
byte* ptr = (byte*) &u;
for(int i = sizeof(u) - 1; i >= 0; i--) {
ptr[i] = field[PTR_UNMOVE];
prev_unmove();
}
return u;
}
#endif