diff --git a/engine/engine.rs b/engine/engine.rs index 110a19a..f128df0 100644 --- a/engine/engine.rs +++ b/engine/engine.rs @@ -27,8 +27,8 @@ impl Engine { pub fn run(&mut self) -> Result<(), Box> { loop { // setup new game for players - self.players[self.x].start_new_game(Cell::CellX)?; - self.players[(self.x + 1) % 2].start_new_game(Cell::CellO)?; + self.players[self.x].start_new_game(Cell::X)?; + self.players[(self.x + 1) % 2].start_new_game(Cell::O)?; // reset board self.board.reset(); @@ -46,6 +46,7 @@ impl Engine { loop { println!("{}", self.board); + // request move from player, fail if there is error let m = self.players[self.turn].request_move(&self.board)?; if !self.board.is_valid_move(&m) { @@ -53,14 +54,21 @@ impl Engine { continue; } - self.board.put(m)?; + // apply move + self.board.apply(m)?; + // check if there is a winner if let Some(_winner) = self.board.has_winner() { println!("{}", self.board); println!("The winner is: {}", self.players[self.turn].name()); break; } + if !self.board.valid_moves_available() { + println!("{}", self.board); + break; + } + self.turn = (self.turn + 1) % 2 } diff --git a/game/entity.rs b/game/entity.rs index 201900f..6eb9520 100644 --- a/game/entity.rs +++ b/game/entity.rs @@ -1,22 +1,17 @@ #[derive(Clone, Copy, PartialEq)] -//pub struct Cell(u8); -//pub const Cell::CellEmpty: Cell = Cell(0); -//pub const Cell::CellX: Cell = Cell(1); -//pub const Cell::CellO: Cell = Cell(2); - pub enum Cell { - CellEmpty, - CellX, - CellO, + Empty, + X, + O, } impl std::fmt::Display for Cell { fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { let s: &str; match *self { - Cell::CellX => s = "X", - Cell::CellO => s = "O", - Cell::CellEmpty => s = " ", + Cell::X => s = "X", + Cell::O => s = "O", + Cell::Empty => s = " ", } write!(f, "{}", s) @@ -36,18 +31,18 @@ pub struct Board { impl Board { pub fn new() -> Board { - return Board { - cells: vec![Cell::CellEmpty; 9], - next: Cell::CellX, - }; + Board { + cells: vec![Cell::Empty; 9], + next: Cell::X, + } } pub fn reset(&mut self) { - self.cells = vec![Cell::CellEmpty; 9]; - self.next = Cell::CellX; + self.cells = vec![Cell::Empty; 9]; + self.next = Cell::X; } - pub fn put(&mut self, m: Move) -> Result<(), &str> { + pub fn apply(&mut self, m: Move) -> Result<(), &str> { let i = m.x * 1 + m.y * 3; if !self.is_valid_move(&m) { @@ -57,8 +52,8 @@ impl Board { self.cells[i] = m.piece; match self.next { - Cell::CellX => self.next = Cell::CellO, - Cell::CellO => self.next = Cell::CellX, + Cell::X => self.next = Cell::O, + Cell::O => self.next = Cell::X, _ => {} } @@ -68,41 +63,45 @@ impl Board { pub fn is_valid_move(&self, m: &Move) -> bool { let i = m.x * 1 + m.y * 3; - m.piece == self.next && self.cells[i] == Cell::CellEmpty + m.piece == self.next && self.cells[i] == Cell::Empty + } + + pub fn valid_moves_available(&self) -> bool { + // check for draw condition + let mut empty: usize = 0; + for i in 0..self.cells.len() { + if self.cells[i] == Cell::Empty { + empty += 1; + } + } + + empty > 0 } pub fn has_winner(&self) -> Option { // rows and cols for i in 0..3 { - if (self.cells[i] == self.cells[i + 3] && self.cells[i] == self.cells[i + 6]) - && (self.cells[i] != Cell::CellEmpty - && self.cells[i + 3] != Cell::CellEmpty - && self.cells[i + 6] != Cell::CellEmpty) + if (self.cells[i] != Cell::Empty) + && (self.cells[i] == self.cells[i + 3] && self.cells[i] == self.cells[i + 6]) { return Some(self.cells[i]); - } else if (self.cells[i * 3] == self.cells[i * 3 + 1] - && self.cells[i * 3] == self.cells[i * 3 + 2]) - && (self.cells[i * 3] != Cell::CellEmpty - && self.cells[i * 3 + 1] != Cell::CellEmpty - && self.cells[i * 3 + 2] != Cell::CellEmpty) + } else if (self.cells[i * 3] != Cell::Empty) + && (self.cells[i * 3] == self.cells[i * 3 + 1] + && self.cells[i * 3] == self.cells[i * 3 + 2]) { return Some(self.cells[i * 3]); } } // diagonals - if (self.cells[0] == self.cells[4] && self.cells[0] == self.cells[8]) - && (self.cells[0] != Cell::CellEmpty - && self.cells[4] != Cell::CellEmpty - && self.cells[8] != Cell::CellEmpty) + if (self.cells[0] != Cell::Empty) + && (self.cells[0] == self.cells[4] && self.cells[0] == self.cells[8]) { return Some(self.cells[0]); } - if (self.cells[2] == self.cells[4] && self.cells[2] == self.cells[6]) - && (self.cells[2] != Cell::CellEmpty - && self.cells[4] != Cell::CellEmpty - && self.cells[6] != Cell::CellEmpty) + if (self.cells[2] != Cell::Empty) + && (self.cells[2] == self.cells[4] && self.cells[2] == self.cells[6]) { return Some(self.cells[2]); } @@ -140,7 +139,10 @@ impl std::fmt::Display for Board { match self.has_winner() { Some(w) => s.push_str(&format!("The winner is {w}")), - None => s.push_str(&format!("Next: {}", self.next)), + None => match self.valid_moves_available() { + true => s.push_str(&format!("Next: {}", self.next)), + false => s.push_str("DRAW!"), + }, } write!(f, "{s}") diff --git a/player/human.rs b/player/human.rs index c498426..70b6e5d 100644 --- a/player/human.rs +++ b/player/human.rs @@ -11,7 +11,7 @@ impl PlayerConsole { pub fn new(name: &str) -> impl Player { PlayerConsole { name: name.to_owned(), - piece: Cell::CellEmpty, + piece: Cell::Empty, } }