Compare commits
33 Commits
Server/web
...
v113
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2aaa2aca70 | ||
|
|
456d386551 | ||
|
|
e663d61b9e | ||
|
|
c3ed181b07 | ||
|
|
5b3f2488d4 | ||
|
|
659413ca31 | ||
|
|
4de505bc21 | ||
|
|
7c58f0d508 | ||
|
|
c9246d1342 | ||
|
|
eab795b6df | ||
|
|
f8ab14a026 | ||
|
|
274897a070 | ||
|
|
8ecbeb9c41 | ||
|
|
a1482d11f2 | ||
|
|
f9a302c9a0 | ||
|
|
ad530e9155 | ||
|
|
182aa59ee1 | ||
|
|
296f1f6c3a | ||
|
|
f7355f8e74 | ||
|
|
cd58a7a321 | ||
|
|
2beb8ab303 | ||
|
|
5425ccc5cd | ||
|
|
1104c8e6c5 | ||
|
|
274ffcf5ec | ||
|
|
5854dbc20b | ||
|
|
f36a196b2f | ||
|
|
11f26809df | ||
|
|
c88fbe68e3 | ||
|
|
7f4c53ddb7 | ||
|
|
38b38845d6 | ||
|
|
66dd2877b2 | ||
|
|
3cd53c7d70 | ||
|
|
5b3bdc750f |
@@ -1,2 +1,7 @@
|
|||||||
mod attackmaps;
|
mod attackmaps;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
mod legality;
|
||||||
|
mod checkinfo;
|
||||||
|
mod attacks;
|
||||||
|
|
||||||
|
pub mod board;
|
||||||
89
engine/src/bitboard/attacks.rs
Normal file
89
engine/src/bitboard/attacks.rs
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
use super::board::Board;
|
||||||
|
use super::attackmaps::*;
|
||||||
|
|
||||||
|
impl Board {
|
||||||
|
|
||||||
|
const RANK_2: u64 = 0x0000_0000_0000_FF00;
|
||||||
|
const RANK_7: u64 = 0x00FF_0000_0000_0000;
|
||||||
|
|
||||||
|
pub fn get_pseudo_pawn_moves(&self, sq: u32) -> u64 {
|
||||||
|
let pawn: u64 = 1 << sq;
|
||||||
|
let mut move_mask: u64 = 0u64;
|
||||||
|
let move_offset: i8 = 8 - 16 * self.side_to_move as i8;
|
||||||
|
|
||||||
|
let next_sq: u64 = if move_offset > 0 {pawn << move_offset} else {pawn >> -move_offset};
|
||||||
|
if (self.occupancy[2] & next_sq) == 0 {
|
||||||
|
move_mask |= next_sq;
|
||||||
|
|
||||||
|
if (self.side_to_move == 0 && pawn & Self::RANK_2 != 0)
|
||||||
|
|| (self.side_to_move == 1 && pawn & Self::RANK_7 != 0) {
|
||||||
|
|
||||||
|
let next_sq: u64 = if move_offset > 0 {next_sq << move_offset} else {next_sq >> -move_offset};
|
||||||
|
if (self.occupancy[2] & next_sq) == 0 {
|
||||||
|
move_mask |= next_sq;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return move_mask;
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn get_pseudo_knight_moves(&self, sq: u32) -> u64 {
|
||||||
|
return KNIGHT_ATTACK_MAP[sq as usize];
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn get_pseudo_king_moves(&self, sq: u32) -> u64 {
|
||||||
|
return KING_ATTACK_MAP[sq as usize];
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn get_pseudo_pawn_captures(&self, sq: u32) -> u64 {
|
||||||
|
return PAWN_ATTACK_MAP[sq as usize][self.side_to_move as usize];
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn get_pseudo_opponent_pawn_captures(&self, sq: u32) -> u64 {
|
||||||
|
return PAWN_ATTACK_MAP[sq as usize][1 - self.side_to_move as usize];
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn get_pseudo_bishop_moves(&self, sq: u32) -> u64 {
|
||||||
|
let mut moves = 0u64;
|
||||||
|
let sq = sq as usize;
|
||||||
|
let occupancy = self.occupancy[2];
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 1);
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 3);
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 5);
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 7);
|
||||||
|
|
||||||
|
return moves;
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn get_pseudo_rook_moves(&self, sq: u32) -> u64 {
|
||||||
|
let mut moves: u64 = 0u64;
|
||||||
|
let occupancy = self.occupancy[2];
|
||||||
|
let sq = sq as usize;
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 0);
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 2);
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 4);
|
||||||
|
moves |= get_raycast_from_square_in_direction(occupancy, sq, 6);
|
||||||
|
|
||||||
|
return moves;
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn get_pseudo_queen_moves(&self, sq: u32) -> u64 {
|
||||||
|
return self.get_pseudo_bishop_moves(sq) | self.get_pseudo_rook_moves(sq);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn get_raycast_from_square_in_direction(occupancy: u64, sq: usize, dir: usize) -> u64 {
|
||||||
|
let is_up: bool = dir / 4 == 0;
|
||||||
|
let mut ray: u64 = RAY_TABLE[sq][dir];
|
||||||
|
let blockers: u64 = occupancy & ray;
|
||||||
|
|
||||||
|
if blockers != 0 {
|
||||||
|
let first_blocker: u32 = if is_up { blockers.trailing_zeros() } else { 63 - blockers.leading_zeros() };
|
||||||
|
|
||||||
|
ray &= !RAY_TABLE[first_blocker as usize][dir];
|
||||||
|
}
|
||||||
|
|
||||||
|
return ray;
|
||||||
|
}
|
||||||
182
engine/src/bitboard/board.rs
Normal file
182
engine/src/bitboard/board.rs
Normal file
@@ -0,0 +1,182 @@
|
|||||||
|
use super::utils::try_get_square_number_from_notation;
|
||||||
|
|
||||||
|
pub struct Board {
|
||||||
|
pub(in super) bitboards: [u64; 12], // 0-5 -> white pieces (P, N, B, R, Q, K), 6-11 -> black pieces (p, n, b, r, q, k)
|
||||||
|
pub(in super) piece_board: [u8; 64], // same as board indexes, 12 -> empty square
|
||||||
|
pub(in super) occupancy: [u64; 3], // 0 -> white, 1 -> black, 2 -> combined
|
||||||
|
pub(in super) castling_rights: u8, // 0b0000_KQkq
|
||||||
|
pub(in super) pinned_squares: [u8; 64], // 0 -> E-W, 1 -> NE-SW, 2 -> N-S, 3 -> SE-NW, 4 -> no pin
|
||||||
|
pub(in super) pin_mask: u64, // 1 -> pin, 0 -> no pin
|
||||||
|
pub(in super) en_passant_square: u64, // 1 -> ep square, 0 -> no ep square
|
||||||
|
pub(in super) side_to_move: u8 // 0 -> white to play, 1 -> black to play
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Board {
|
||||||
|
|
||||||
|
pub fn new_clear() -> Self {
|
||||||
|
let mut bit_board: Self = Self {
|
||||||
|
bitboards: [0x0000_0000_0000_0000; 12],
|
||||||
|
piece_board: [12; 64],
|
||||||
|
occupancy: [0x0000_0000_0000_0000; 3],
|
||||||
|
castling_rights: 0b0000_0000,
|
||||||
|
pinned_squares: [4; 64],
|
||||||
|
pin_mask: 0u64,
|
||||||
|
en_passant_square: 0x0000_0000_0000_0000,
|
||||||
|
side_to_move: 0
|
||||||
|
};
|
||||||
|
|
||||||
|
return bit_board;
|
||||||
|
}
|
||||||
|
pub fn new() -> Self {
|
||||||
|
let mut bit_board: Board = Self {
|
||||||
|
bitboards: [0x0000_0000_0000_FF00,
|
||||||
|
0x0000_0000_0000_0042,
|
||||||
|
0x0000_0000_0000_0024,
|
||||||
|
0x0000_0000_0000_0081,
|
||||||
|
0x0000_0000_0000_0008,
|
||||||
|
0x0000_0000_0000_0010,
|
||||||
|
0x00FF_0000_0000_0000,
|
||||||
|
0x4200_0000_0000_0000,
|
||||||
|
0x2400_0000_0000_0000,
|
||||||
|
0x8100_0000_0000_0000,
|
||||||
|
0x0800_0000_0000_0000,
|
||||||
|
0x1000_0000_0000_0000],
|
||||||
|
piece_board: [12; 64],
|
||||||
|
occupancy: [0; 3],
|
||||||
|
castling_rights: 0b0000_1111,
|
||||||
|
pinned_squares: [4; 64],
|
||||||
|
pin_mask: 0u64,
|
||||||
|
en_passant_square: 0x0000_0000_0000_0000,
|
||||||
|
side_to_move: 0
|
||||||
|
};
|
||||||
|
bit_board.calc_occupancy();
|
||||||
|
bit_board.calc_piece_board();
|
||||||
|
|
||||||
|
return bit_board;
|
||||||
|
}
|
||||||
|
pub fn build(fen: &str) -> Self {
|
||||||
|
let mut board: Board = Board::new_clear();
|
||||||
|
|
||||||
|
let mut col: i32 = 0;
|
||||||
|
let mut row: i32 = 7;
|
||||||
|
let pieces: [char; 12] = ['p', 'n', 'b', 'r', 'q', 'k', 'P', 'N', 'B', 'R', 'Q', 'K'];
|
||||||
|
let mut coming_up: &str = fen;
|
||||||
|
|
||||||
|
for (i, c) in coming_up.chars().enumerate() {
|
||||||
|
if pieces.contains(&c) {
|
||||||
|
// board.place_piece(row*8 + col, c);
|
||||||
|
col += 1;
|
||||||
|
}
|
||||||
|
else if ('1'..='8').contains(&c) {
|
||||||
|
col += c.to_string().parse::<i32>().unwrap();
|
||||||
|
}
|
||||||
|
else if c == '/' {
|
||||||
|
row -= 1;
|
||||||
|
col = 0;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
coming_up = &coming_up[i+1..];
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
board.calc_occupancy();
|
||||||
|
|
||||||
|
match coming_up.chars().next().unwrap() {
|
||||||
|
'w' => board.side_to_move = 0,
|
||||||
|
'b' => board.side_to_move = 1,
|
||||||
|
_ => panic!("invalid fen notation / to be handled later")
|
||||||
|
}
|
||||||
|
coming_up = &coming_up[2..];
|
||||||
|
|
||||||
|
for (i, c) in coming_up.chars().enumerate() {
|
||||||
|
match c {
|
||||||
|
'K' => board.castling_rights |= 1 << 3,
|
||||||
|
'Q' => board.castling_rights |= 1 << 2,
|
||||||
|
'k' => board.castling_rights |= 1 << 1,
|
||||||
|
'q' => board.castling_rights |= 1,
|
||||||
|
'-' => {
|
||||||
|
coming_up = &coming_up[i+2..];
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
coming_up = &coming_up[i+1..];
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
match coming_up.chars().next().unwrap() {
|
||||||
|
'-' => {
|
||||||
|
coming_up = &coming_up[1..];
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
let notation = coming_up.split(' ').next().unwrap();
|
||||||
|
if let Ok(epsq_index) = try_get_square_number_from_notation(notation) {
|
||||||
|
board.en_passant_square = 1 << epsq_index;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
board.calc_pinned_squares();
|
||||||
|
board.calc_piece_board();
|
||||||
|
|
||||||
|
return board;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn bitboards(&self, index: usize) -> u64 {
|
||||||
|
return self.bitboards[index];
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn piece_board(&self, sq: u8) -> u8 {
|
||||||
|
return self.piece_board[sq as usize];
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn occupancy(&self, side: usize) -> u64 {
|
||||||
|
return self.occupancy[side];
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn castling_rights(&self) -> u8 {
|
||||||
|
return self.castling_rights;
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn pinned_squares(&self, sq: usize) -> u8 {
|
||||||
|
return self.pinned_squares[sq];
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn pin_mask(&self) -> u64 {
|
||||||
|
return self.pin_mask;
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn en_passant_square(&self) -> u64 {
|
||||||
|
return self.en_passant_square;
|
||||||
|
}
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn side_to_move(&self) -> u8 {
|
||||||
|
return self.side_to_move;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn current_king_square(&self) -> u32 {
|
||||||
|
return if self.side_to_move == 0 { self.bitboards[5].trailing_zeros() } else { self.bitboards[11].trailing_zeros() };
|
||||||
|
}
|
||||||
|
|
||||||
|
fn calc_occupancy(&mut self) {
|
||||||
|
self.occupancy = [0u64; 3];
|
||||||
|
for b in 0..6 {
|
||||||
|
self.occupancy[0] |= self.bitboards[b];
|
||||||
|
}
|
||||||
|
for b in 6..12 {
|
||||||
|
self.occupancy[1] |= self.bitboards[b];
|
||||||
|
}
|
||||||
|
self.occupancy[2] = self.occupancy[0] | self.occupancy[1];
|
||||||
|
}
|
||||||
|
fn calc_piece_board(&mut self) {
|
||||||
|
for sq in 0..64 {
|
||||||
|
for b in 0..12 {
|
||||||
|
if (self.bitboards[b as usize] & 1 << sq) != 0 {
|
||||||
|
self.piece_board[sq] = b;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
21
engine/src/bitboard/checkinfo.rs
Normal file
21
engine/src/bitboard/checkinfo.rs
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
|
||||||
|
pub struct CheckInfo {
|
||||||
|
pub check_count: u8,
|
||||||
|
pub move_mask: u64
|
||||||
|
}
|
||||||
|
|
||||||
|
impl CheckInfo {
|
||||||
|
|
||||||
|
pub fn new() -> Self {
|
||||||
|
return Self {
|
||||||
|
check_count: 0,
|
||||||
|
move_mask: 0xFFFF_FFFF_FFFF_FFFF
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline(always)]
|
||||||
|
pub fn add_checker(&mut self, move_mask: u64) {
|
||||||
|
self.move_mask &= move_mask;
|
||||||
|
self.check_count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
48
engine/src/bitboard/legality.rs
Normal file
48
engine/src/bitboard/legality.rs
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
use super::board::Board;
|
||||||
|
use super::attackmaps::RAY_TABLE;
|
||||||
|
|
||||||
|
impl Board {
|
||||||
|
|
||||||
|
pub(in super) fn calc_pinned_squares(&mut self) {
|
||||||
|
self.pinned_squares = [4; 64];
|
||||||
|
self.pin_mask = 0u64;
|
||||||
|
|
||||||
|
let friendly_pieces: u64 = self.occupancy[self.side_to_move as usize];
|
||||||
|
let offset: usize = 6 * self.side_to_move as usize;
|
||||||
|
let king_board: u64 = self.bitboards[5 + offset];
|
||||||
|
let king_sq: u32 = king_board.trailing_zeros();
|
||||||
|
let opponent_queen_bishop_mask: u64 = self.bitboards[8 - offset] | self.bitboards[10 - offset];
|
||||||
|
let opponent_queen_rook_mask: u64 = self.bitboards[9 - offset] | self.bitboards[10 - offset];
|
||||||
|
|
||||||
|
// Queen-Rook directions
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_rook_mask, 0);
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_rook_mask, 2);
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_rook_mask, 4);
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_rook_mask, 6);
|
||||||
|
|
||||||
|
// Queen-Bishop directions
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_bishop_mask, 1);
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_bishop_mask, 3);
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_bishop_mask, 5);
|
||||||
|
self.set_pinned_in_ray_direction(king_sq, friendly_pieces, opponent_queen_bishop_mask, 7);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(in super) fn set_pinned_in_ray_direction(&mut self, king_sq: u32, friendly_pieces: u64, attackers: u64, dir: u8) {
|
||||||
|
let is_up: bool = dir / 4 == 0;
|
||||||
|
let mask: u64 = RAY_TABLE[king_sq as usize][dir as usize];
|
||||||
|
let blockers: u64 = self.occupancy[2] & mask;
|
||||||
|
if blockers == 0 { return; }
|
||||||
|
let first_blocker_sq: u32 = if is_up { blockers.trailing_zeros() } else { 63 - blockers.leading_zeros() };
|
||||||
|
if (friendly_pieces & 1 << first_blocker_sq) != 0 {
|
||||||
|
let blockers: u64 = blockers & !(1 << first_blocker_sq);
|
||||||
|
if blockers == 0 { return; }
|
||||||
|
let second_blocker_sq: u32 = if is_up { blockers.trailing_zeros() } else { 63 - blockers.leading_zeros() };
|
||||||
|
|
||||||
|
if (attackers & 1 << second_blocker_sq) != 0 {
|
||||||
|
self.pinned_squares[first_blocker_sq as usize] = dir % 4;
|
||||||
|
self.pin_mask |= 1 << first_blocker_sq;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@@ -27,6 +27,27 @@ pub fn notation_from_square_number(sq: u8) -> String {
|
|||||||
return notation;
|
return notation;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn try_get_square_number_from_notation(notation: &str) -> Result<u8, ()> {
|
||||||
|
|
||||||
|
let file = match notation.chars().nth(0).unwrap() {
|
||||||
|
'a' => 0,
|
||||||
|
'b' => 1,
|
||||||
|
'c' => 2,
|
||||||
|
'd' => 3,
|
||||||
|
'e' => 4,
|
||||||
|
'f' => 5,
|
||||||
|
'g' => 6,
|
||||||
|
'h' => 7,
|
||||||
|
_ => { return Result::Err(()); }
|
||||||
|
};
|
||||||
|
if let Some(rank) = notation.chars().nth(1) {
|
||||||
|
return Result::Ok(file + 8 * (rank.to_digit(10).unwrap() as u8) - 8);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
return Result::Err(());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
// <----- TESTS ----->
|
// <----- TESTS ----->
|
||||||
|
|
||||||
|
|||||||
4371
ui/Cargo.lock
generated
Normal file
4371
ui/Cargo.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
@@ -4,3 +4,7 @@ version = "0.1.0"
|
|||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
eframe = "0.33.0"
|
||||||
|
egui = "0.33.0"
|
||||||
|
tokio-tungstenite = "0.28.0"
|
||||||
|
winit = "0.30.12"
|
||||||
|
|||||||
BIN
ui/fonts/DejaVuSans.ttf
Normal file
BIN
ui/fonts/DejaVuSans.ttf
Normal file
Binary file not shown.
431
ui/src/main.rs
431
ui/src/main.rs
@@ -1,3 +1,430 @@
|
|||||||
fn main() {
|
use eframe::egui;
|
||||||
println!("Hello, world!");
|
|
||||||
|
fn main() -> eframe::Result<()> {
|
||||||
|
let options = eframe::NativeOptions{
|
||||||
|
viewport: egui::ViewportBuilder::default()
|
||||||
|
.with_fullscreen(false)
|
||||||
|
.with_min_inner_size(egui::vec2(800.0, 600.0)) // Minimum width, height
|
||||||
|
.with_inner_size(egui::vec2(7680.0, 4320.0)), // Initial size
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
eframe::run_native(
|
||||||
|
"Knightly",
|
||||||
|
options,
|
||||||
|
Box::new(|cc| {
|
||||||
|
let mut fonts = egui::FontDefinitions::default();
|
||||||
|
fonts.font_data.insert(
|
||||||
|
"symbols".to_owned(),
|
||||||
|
egui::FontData::from_static(include_bytes!("../fonts/DejaVuSans.ttf")).into(),
|
||||||
|
);
|
||||||
|
fonts
|
||||||
|
.families
|
||||||
|
.entry(egui::FontFamily::Proportional)
|
||||||
|
.or_default()
|
||||||
|
.insert(0, "symbols".to_owned());
|
||||||
|
cc.egui_ctx.set_fonts(fonts);
|
||||||
|
Ok(Box::new(ChessApp::default()))
|
||||||
|
}),
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, PartialEq, Debug)]
|
||||||
|
enum Piece {
|
||||||
|
King(char),
|
||||||
|
Queen(char),
|
||||||
|
Rook(char),
|
||||||
|
Bishop(char),
|
||||||
|
Knight(char),
|
||||||
|
Pawn(char),
|
||||||
|
Empty,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Piece {
|
||||||
|
fn symbol(&self) -> &'static str {
|
||||||
|
match self {
|
||||||
|
Piece::King('w') => "♔",
|
||||||
|
Piece::Queen('w') => "♕",
|
||||||
|
Piece::Rook('w') => "♖",
|
||||||
|
Piece::Bishop('w') => "♗",
|
||||||
|
Piece::Knight('w') => "♘",
|
||||||
|
Piece::Pawn('w') => "♙",
|
||||||
|
Piece::King('b') => "♚",
|
||||||
|
Piece::Queen('b') => "♛",
|
||||||
|
Piece::Rook('b') => "♜",
|
||||||
|
Piece::Bishop('b') => "♝",
|
||||||
|
Piece::Knight('b') => "♞",
|
||||||
|
Piece::Pawn('b') => "♟︎",
|
||||||
|
Piece::Empty => "",
|
||||||
|
_ => "",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(PartialEq, Debug)]
|
||||||
|
enum Turn {
|
||||||
|
White,
|
||||||
|
Black,
|
||||||
|
}
|
||||||
|
|
||||||
|
enum AppState {
|
||||||
|
MainMenu,
|
||||||
|
InGame,
|
||||||
|
Settings,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct ChessApp {
|
||||||
|
fullscreen: bool,
|
||||||
|
resolutions: Vec<(u32, u32)>,
|
||||||
|
selected_resolution: usize,
|
||||||
|
state: AppState,
|
||||||
|
board: [[Piece; 8]; 8],
|
||||||
|
selected: Option<(usize, usize)>,
|
||||||
|
turn: Turn,
|
||||||
|
pending_settings: PendingSettings,
|
||||||
|
server_port: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
struct PendingSettings {
|
||||||
|
fullscreen: bool,
|
||||||
|
selected_resolution: usize,
|
||||||
|
server_port: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ChessApp {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self {
|
||||||
|
fullscreen: true,
|
||||||
|
resolutions: vec![
|
||||||
|
(1280, 720),
|
||||||
|
(1600, 900),
|
||||||
|
(1920, 1080),
|
||||||
|
(2560, 1440),
|
||||||
|
(3840, 2160),
|
||||||
|
],
|
||||||
|
selected_resolution: 2, // Default to 1920x1080
|
||||||
|
state: AppState::MainMenu,
|
||||||
|
board: Self::starting_board(),
|
||||||
|
selected: None,
|
||||||
|
turn: Turn::White,
|
||||||
|
pending_settings: PendingSettings::default(),
|
||||||
|
server_port: "8080".to_string(), // Default port
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChessApp {
|
||||||
|
fn starting_board() -> [[Piece; 8]; 8] {
|
||||||
|
use Piece::*;
|
||||||
|
[
|
||||||
|
[
|
||||||
|
Rook('b'),
|
||||||
|
Knight('b'),
|
||||||
|
Bishop('b'),
|
||||||
|
Queen('b'),
|
||||||
|
King('b'),
|
||||||
|
Bishop('b'),
|
||||||
|
Knight('b'),
|
||||||
|
Rook('b'),
|
||||||
|
],
|
||||||
|
[Pawn('b'); 8],
|
||||||
|
[Empty; 8],
|
||||||
|
[Empty; 8],
|
||||||
|
[Empty; 8],
|
||||||
|
[Empty; 8],
|
||||||
|
[Pawn('w'); 8],
|
||||||
|
[
|
||||||
|
Rook('w'),
|
||||||
|
Knight('w'),
|
||||||
|
Bishop('w'),
|
||||||
|
Queen('w'),
|
||||||
|
King('w'),
|
||||||
|
Bishop('w'),
|
||||||
|
Knight('w'),
|
||||||
|
Rook('w'),
|
||||||
|
],
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_click(&mut self, row: usize, col: usize) {
|
||||||
|
if let Some((r, c)) = self.selected {
|
||||||
|
let piece = self.board[r][c];
|
||||||
|
self.board[r][c] = Piece::Empty;
|
||||||
|
self.board[row][col] = piece;
|
||||||
|
self.selected = None;
|
||||||
|
self.turn = if self.turn == Turn::White {
|
||||||
|
Turn::Black
|
||||||
|
} else {
|
||||||
|
Turn::White
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
if self.board[row][col] != Piece::Empty {
|
||||||
|
self.selected = Some((row, col));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn apply_settings(&mut self, ctx: &egui::Context) {
|
||||||
|
self.fullscreen = self.pending_settings.fullscreen;
|
||||||
|
self.selected_resolution = self.pending_settings.selected_resolution;
|
||||||
|
self.server_port = self.pending_settings.server_port.clone();
|
||||||
|
|
||||||
|
if let Some(resolution) = self.resolutions.get(self.selected_resolution) {
|
||||||
|
ctx.send_viewport_cmd(egui::ViewportCommand::InnerSize(
|
||||||
|
egui::Vec2::new(resolution.0 as f32, resolution.1 as f32)
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.send_viewport_cmd(egui::ViewportCommand::Fullscreen(self.fullscreen));
|
||||||
|
}
|
||||||
|
|
||||||
|
fn enter_settings(&mut self) {
|
||||||
|
self.pending_settings.fullscreen = self.fullscreen;
|
||||||
|
self.pending_settings.selected_resolution = self.selected_resolution;
|
||||||
|
self.pending_settings.server_port = self.server_port.clone();
|
||||||
|
self.state = AppState::Settings;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl eframe::App for ChessApp {
|
||||||
|
fn update(&mut self, ctx: &egui::Context, frame: &mut eframe::Frame) {
|
||||||
|
match self.state {
|
||||||
|
AppState::MainMenu => {
|
||||||
|
egui::CentralPanel::default().show(ctx, |ui| {
|
||||||
|
ui.vertical_centered(|ui| {
|
||||||
|
ui.heading("♞ Knightly ♞");
|
||||||
|
ui.add_space(30.0);
|
||||||
|
|
||||||
|
if ui.add_sized([300.0, 60.0], egui::Button::new("Play")).clicked() {
|
||||||
|
self.state = AppState::InGame;
|
||||||
|
}
|
||||||
|
ui.add_space(8.0);
|
||||||
|
|
||||||
|
if ui.add_sized([300.0, 60.0], egui::Button::new("Settings")).clicked() {
|
||||||
|
self.enter_settings();
|
||||||
|
}
|
||||||
|
ui.add_space(8.0);
|
||||||
|
|
||||||
|
if ui
|
||||||
|
.add_sized([300.0, 60.0], egui::Button::new("Quit"))
|
||||||
|
.clicked()
|
||||||
|
{
|
||||||
|
std::process::exit(0);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
AppState::Settings => {
|
||||||
|
egui::CentralPanel::default().show(ctx, |ui| {
|
||||||
|
ui.vertical_centered(|ui| {
|
||||||
|
ui.heading("Settings");
|
||||||
|
ui.add_space(30.0);
|
||||||
|
|
||||||
|
// Fullscreen toggle
|
||||||
|
ui.horizontal(|ui| {
|
||||||
|
ui.label("Fullscreen:");
|
||||||
|
if ui.checkbox(&mut self.pending_settings.fullscreen, "").changed() {
|
||||||
|
// If enabling fullscreen, we might want to disable resolution selection
|
||||||
|
}
|
||||||
|
});
|
||||||
|
ui.add_space(10.0);
|
||||||
|
|
||||||
|
// Resolution dropdown
|
||||||
|
ui.horizontal(|ui| {
|
||||||
|
ui.label("Resolution:");
|
||||||
|
egui::ComboBox::new("resolution_combo", "")
|
||||||
|
.selected_text(format!(
|
||||||
|
"{}x{}",
|
||||||
|
self.resolutions[self.pending_settings.selected_resolution].0,
|
||||||
|
self.resolutions[self.pending_settings.selected_resolution].1
|
||||||
|
))
|
||||||
|
.show_ui(ui, |ui| {
|
||||||
|
for (i, &(width, height)) in self.resolutions.iter().enumerate() {
|
||||||
|
ui.selectable_value(
|
||||||
|
&mut self.pending_settings.selected_resolution,
|
||||||
|
i,
|
||||||
|
format!("{}x{}", width, height),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
ui.add_space(10.0);
|
||||||
|
|
||||||
|
// Server port input field
|
||||||
|
ui.horizontal(|ui| {
|
||||||
|
ui.label("Local Server Port:");
|
||||||
|
ui.add(egui::TextEdit::singleline(&mut self.pending_settings.server_port)
|
||||||
|
.desired_width(100.0)
|
||||||
|
.hint_text("8080"));
|
||||||
|
});
|
||||||
|
ui.add_space(30.0);
|
||||||
|
|
||||||
|
// Apply and Cancel buttons
|
||||||
|
ui.horizontal(|ui| {
|
||||||
|
if ui.add_sized([140.0, 40.0], egui::Button::new("Apply")).clicked() {
|
||||||
|
self.apply_settings(ctx);
|
||||||
|
self.state = AppState::MainMenu;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ui.add_sized([140.0, 40.0], egui::Button::new("Cancel")).clicked() {
|
||||||
|
self.state = AppState::MainMenu;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
AppState::InGame => {
|
||||||
|
egui::TopBottomPanel::top("menu_bar").show(ctx, |ui| {
|
||||||
|
ui.horizontal(|ui| {
|
||||||
|
if ui.button("Main Menu").clicked() {
|
||||||
|
self.state = AppState::MainMenu;
|
||||||
|
}
|
||||||
|
if ui.button("Settings").clicked() {
|
||||||
|
self.enter_settings();
|
||||||
|
}
|
||||||
|
if ui.button("New Game").clicked() {
|
||||||
|
*self = ChessApp::default();
|
||||||
|
self.state = AppState::InGame;
|
||||||
|
}
|
||||||
|
ui.separator();
|
||||||
|
ui.label(format!("Turn: {:?}", self.turn));
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
egui::CentralPanel::default().show(ctx, |ui| {
|
||||||
|
ui.vertical_centered(|ui| {
|
||||||
|
let full_avail = ui.available_rect_before_wrap();
|
||||||
|
let board_tile = (full_avail.width().min(full_avail.height())) / 8.0;
|
||||||
|
let board_size = board_tile * 8.0;
|
||||||
|
|
||||||
|
// Create a child UI at the board position
|
||||||
|
let (response, painter) = ui.allocate_painter(
|
||||||
|
egui::Vec2::new(board_size, board_size),
|
||||||
|
egui::Sense::click()
|
||||||
|
);
|
||||||
|
|
||||||
|
let board_rect = egui::Rect::from_center_size(
|
||||||
|
full_avail.center(),
|
||||||
|
egui::vec2(board_size, board_size)
|
||||||
|
);
|
||||||
|
|
||||||
|
// Draw the chess board
|
||||||
|
let tile_size = board_size / 8.0;
|
||||||
|
for row in 0..8 {
|
||||||
|
for col in 0..8 {
|
||||||
|
let color = if (row + col) % 2 == 0 {
|
||||||
|
egui::Color32::from_rgb(100, 97, 97)
|
||||||
|
} else {
|
||||||
|
egui::Color32::from_rgb(217, 217, 217)
|
||||||
|
};
|
||||||
|
|
||||||
|
let rect = egui::Rect::from_min_size(
|
||||||
|
egui::Pos2::new(
|
||||||
|
board_rect.min.x + col as f32 * tile_size,
|
||||||
|
board_rect.min.y + row as f32 * tile_size
|
||||||
|
),
|
||||||
|
egui::Vec2::new(tile_size, tile_size)
|
||||||
|
);
|
||||||
|
|
||||||
|
painter.rect_filled(rect, 0.0, color);
|
||||||
|
|
||||||
|
// Draw piece
|
||||||
|
let piece = self.board[row][col];
|
||||||
|
if piece != Piece::Empty {
|
||||||
|
let symbol = piece.symbol();
|
||||||
|
let font_id = egui::FontId::proportional(tile_size * 0.75);
|
||||||
|
painter.text(
|
||||||
|
rect.center(),
|
||||||
|
egui::Align2::CENTER_CENTER,
|
||||||
|
symbol,
|
||||||
|
font_id,
|
||||||
|
if matches!(piece, Piece::King('w') | Piece::Queen('w') | Piece::Rook('w') | Piece::Bishop('w') | Piece::Knight('w') | Piece::Pawn('w')) {
|
||||||
|
egui::Color32::WHITE
|
||||||
|
} else {
|
||||||
|
egui::Color32::BLACK
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Draw selection highlight
|
||||||
|
if self.selected == Some((row, col)) {
|
||||||
|
painter.rect_stroke(
|
||||||
|
rect,
|
||||||
|
0.0,
|
||||||
|
egui::Stroke::new(3.0, egui::Color32::RED),
|
||||||
|
egui::StrokeKind::Inside
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle clicks
|
||||||
|
if ui.ctx().input(|i| i.pointer.primary_clicked()) {
|
||||||
|
let click_pos = ui.ctx().input(|i| i.pointer.interact_pos()).unwrap();
|
||||||
|
if rect.contains(click_pos) {
|
||||||
|
self.handle_click(row, col);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
#[test]
|
||||||
|
fn test_initial_board_setup() {
|
||||||
|
let app = ChessApp::default();
|
||||||
|
assert!(matches!(app.board[0][0], Piece::Rook('b')));
|
||||||
|
assert!(matches!(app.board[7][0], Piece::Rook('w')));
|
||||||
|
|
||||||
|
assert!(matches!(app.board[1][0], Piece::Pawn('b')));
|
||||||
|
assert!(matches!(app.board[6][0], Piece::Pawn('w')));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_piece_symbols() {
|
||||||
|
assert_eq!(Piece::King('w').symbol(), "♔");
|
||||||
|
assert_eq!(Piece::King('b').symbol(), "♚");
|
||||||
|
assert_eq!(Piece::Empty.symbol(), "");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_piece_selection() {
|
||||||
|
let mut app = ChessApp::default();
|
||||||
|
app.handle_click(6, 0);
|
||||||
|
assert_eq!(app.selected, Some((6, 0)));
|
||||||
|
app.handle_click(6, 0);
|
||||||
|
assert_eq!(app.selected, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_piece_movement() {
|
||||||
|
let mut app = ChessApp::default();
|
||||||
|
// Select and move a piece
|
||||||
|
app.handle_click(6, 0); // Select white pawn
|
||||||
|
app.handle_click(5, 0); // Move to empty square
|
||||||
|
assert_eq!(app.board[6][0], Piece::Empty);
|
||||||
|
assert!(matches!(app.board[5][0], Piece::Pawn('w')));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_turn_switching() {
|
||||||
|
let mut app = ChessApp::default();
|
||||||
|
assert_eq!(app.turn, Turn::White);
|
||||||
|
app.handle_click(6, 0); // White selects
|
||||||
|
app.handle_click(5, 0); // White moves
|
||||||
|
assert_eq!(app.turn, Turn::Black); // Should now be Black's turn
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_server_port_default() {
|
||||||
|
let app = ChessApp::default();
|
||||||
|
assert_eq!(app.server_port, "8080");
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user