64 lines
2.2 KiB
Rust
64 lines
2.2 KiB
Rust
use crate::constants::*;
|
|
use macroquad::color::*;
|
|
use macroquad::math::Vec2;
|
|
use macroquad::prelude::Image;
|
|
use macroquad::rand::RandomRange;
|
|
use std::f32::consts::PI;
|
|
|
|
#[derive(Debug)]
|
|
pub(crate) struct Ant {
|
|
pos: Vec2,
|
|
dir: f32,
|
|
}
|
|
|
|
impl Ant {
|
|
pub(crate) fn init(num: i32) -> Vec<Self> {
|
|
let dir_steps = PI * 2. / num as f32;
|
|
(0..num)
|
|
.into_iter()
|
|
.map(|n| Ant {
|
|
pos: Vec2::new((MAP_WIDTH / 2) as f32, (MAP_HEIGHT / 2) as f32),
|
|
dir: n as f32 * dir_steps,
|
|
})
|
|
.collect()
|
|
}
|
|
pub(crate) fn walk(&mut self, frame_time: f32, trail_map: &mut Image) {
|
|
self.sense(trail_map);
|
|
let new_dir = PI / 8. * RandomRange::gen_range(-2, 2) as f32;
|
|
self.dir += new_dir;
|
|
if self.dir > 2. * PI {
|
|
self.dir -= 2. * PI;
|
|
}
|
|
|
|
let acceleration = Vec2::new(self.dir.sin(), -self.dir.cos());
|
|
self.pos += acceleration * ANT_SPEED * frame_time;
|
|
// if we are out of the map, turn around 180 degrees (or pi in radian)
|
|
if self.pos.x < 0.0
|
|
|| self.pos.x >= (MAP_WIDTH - 1) as f32
|
|
|| self.pos.y < 0.0
|
|
|| self.pos.y >= (MAP_HEIGHT - 1) as f32
|
|
{
|
|
self.dir += PI;
|
|
}
|
|
self.pos.x = self.pos.x.clamp(0., (MAP_WIDTH - 1) as f32);
|
|
self.pos.y = self.pos.y.clamp(0., (MAP_HEIGHT - 1) as f32);
|
|
}
|
|
fn sense(&self, trail_map: &mut Image) {
|
|
trail_map.set_pixel(self.dir.sin() as u32, -self.dir.cos() as u32, YELLOW);
|
|
trail_map.set_pixel(self.dir.sin() as u32, -self.dir.cos() as u32, YELLOW);
|
|
trail_map.set_pixel(self.dir.sin() as u32, -self.dir.cos() as u32, YELLOW);
|
|
}
|
|
// TODO: change color over time or based on other factors
|
|
pub(crate) fn draw(&self, trail_map: &mut Image) {
|
|
let radius: i32 = 3;
|
|
for x in -radius..radius {
|
|
for y in -radius..radius {
|
|
let x = (self.pos.x as i32 + x) as u32;
|
|
let y = (self.pos.y as i32 + y) as u32;
|
|
if x < (MAP_WIDTH - 1) && y < (MAP_HEIGHT - 1) {
|
|
trail_map.set_pixel(x, y, RED);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|