68 lines
2.1 KiB
Rust
68 lines
2.1 KiB
Rust
use super::variables::{Position, Size};
|
|
|
|
use crossterm::{style, Color, Colorize, Crossterm, Screen, TerminalCursor};
|
|
|
|
use rand::distributions::{IndependentSample, Range};
|
|
|
|
use std::collections::HashMap;
|
|
|
|
use rand;
|
|
|
|
pub struct Map {
|
|
pub size: Size,
|
|
pub foot_pos: Position,
|
|
}
|
|
|
|
impl Map {
|
|
pub fn new(size: Size) -> Self {
|
|
Map {
|
|
size: size,
|
|
foot_pos: Position::new(0, 0),
|
|
}
|
|
}
|
|
|
|
// render the map on the screen.
|
|
pub fn render_map(&mut self, screen: &Screen, free_positions: &mut HashMap<String, Position>) {
|
|
let crossterm = Crossterm::from_screen(screen);
|
|
let cursor = crossterm.cursor();
|
|
let terminal = crossterm.terminal();
|
|
|
|
for y in 0..self.size.height {
|
|
for x in 0..self.size.height {
|
|
if (y == 0 || y == self.size.height - 1) || (x == 0 || x == self.size.width - 1) {
|
|
cursor.goto(x as u16, y as u16);
|
|
"█".magenta().paint(&screen.stdout);
|
|
} else {
|
|
free_positions.insert(format!("{},{}", x, y), Position::new(x, y));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn is_out_of_bounds(&self, new_pos: Position) -> bool {
|
|
if (new_pos.x == 0 || new_pos.x == self.size.width)
|
|
|| (new_pos.y == 0 || new_pos.y == self.size.height)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
pub fn is_food_eaten(&self, snake_head: Position) -> bool {
|
|
snake_head.x == self.foot_pos.x && snake_head.y == self.foot_pos.y
|
|
}
|
|
|
|
pub fn spawn_food(&mut self, free_positions: &HashMap<String, Position>, screen: &Screen) {
|
|
let index = Range::new(0, free_positions.len()).ind_sample(&mut rand::thread_rng());
|
|
self.foot_pos = free_positions.values().skip(index).next().unwrap().clone();
|
|
self.draw_food(screen);
|
|
}
|
|
|
|
fn draw_food(&self, screen: &Screen) {
|
|
let cursor = TerminalCursor::from_output(&screen.stdout);
|
|
cursor.goto(self.foot_pos.x as u16, self.foot_pos.y as u16);
|
|
"$".green().paint(&screen.stdout);
|
|
}
|
|
}
|