107 lines
3.2 KiB
Rust
107 lines
3.2 KiB
Rust
use crate::{
|
|
asteroids::{Asteroid, AsteroidSize},
|
|
nn::NN,
|
|
player::Player,
|
|
};
|
|
use macroquad::{prelude::*, rand::gen_range};
|
|
|
|
#[derive(Default)]
|
|
pub struct World {
|
|
player: Player,
|
|
asteroids: Vec<Asteroid>,
|
|
pub score: u32,
|
|
pub over: bool,
|
|
max_asteroids: usize,
|
|
}
|
|
|
|
impl World {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
player: Player::new(),
|
|
max_asteroids: 28,
|
|
..Default::default()
|
|
}
|
|
}
|
|
|
|
pub fn simulate(brain: NN) -> Self {
|
|
Self {
|
|
player: Player::simulate(brain, 28),
|
|
max_asteroids: 28,
|
|
..Default::default()
|
|
}
|
|
}
|
|
|
|
pub fn see_brain(&self) -> &NN {
|
|
self.player.brain.as_ref().unwrap()
|
|
}
|
|
|
|
pub fn update(&mut self) {
|
|
self.player.update();
|
|
let mut to_add: Vec<Asteroid> = Vec::new();
|
|
for asteroid in &mut self.asteroids {
|
|
asteroid.update();
|
|
if self.player.check_player_collision(asteroid) {
|
|
self.over = true;
|
|
}
|
|
if self.player.check_bullet_collisions(asteroid) {
|
|
self.score += 1;
|
|
match asteroid.size {
|
|
AsteroidSize::Large => {
|
|
let rand = vec2(gen_range(-0.8, 0.8), gen_range(-0.8, 0.8));
|
|
to_add.push(Asteroid::new_from(
|
|
asteroid.pos,
|
|
asteroid.vel + rand,
|
|
AsteroidSize::Medium,
|
|
));
|
|
to_add.push(Asteroid::new_from(
|
|
asteroid.pos,
|
|
asteroid.vel - rand,
|
|
AsteroidSize::Medium,
|
|
));
|
|
}
|
|
AsteroidSize::Medium => {
|
|
let rand = vec2(gen_range(-0.6, 0.6), gen_range(-0.6, 0.6));
|
|
to_add.push(Asteroid::new_from(
|
|
asteroid.pos,
|
|
asteroid.vel + rand,
|
|
AsteroidSize::Small,
|
|
));
|
|
to_add.push(Asteroid::new_from(
|
|
asteroid.pos,
|
|
asteroid.vel - rand,
|
|
AsteroidSize::Small,
|
|
));
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
}
|
|
self.asteroids.append(&mut to_add);
|
|
self.asteroids.retain(|asteroid| asteroid.alive);
|
|
if self.asteroids.iter().fold(0, |acc, x| {
|
|
acc + match x.size {
|
|
AsteroidSize::Large => 4,
|
|
AsteroidSize::Medium => 2,
|
|
AsteroidSize::Small => 1,
|
|
}
|
|
}) < self.max_asteroids
|
|
{
|
|
self.asteroids.push(Asteroid::new(AsteroidSize::Large));
|
|
}
|
|
}
|
|
|
|
pub fn draw(&self) {
|
|
self.player.draw();
|
|
for asteroid in &self.asteroids {
|
|
asteroid.draw();
|
|
}
|
|
// draw_text(
|
|
// &format!("Score {}", self.score),
|
|
// 20. - screen_width() * 0.5,
|
|
// 30. - screen_height() * 0.5,
|
|
// 32.,
|
|
// WHITE,
|
|
// );
|
|
}
|
|
}
|