aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
blob: 2c633cca344ece6c1c0cf80a989c2dad3ea9aa1a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
extern crate sdl2;
extern crate rand;

use sdl2::event::Event;
use sdl2::keyboard::Keycode;
use sdl2::pixels::Color;

use rand::Rng;

pub mod vector2;
pub mod boid;
use boid::Boid;

const HEIGHT:            i32   = 480;
const WIDTH:             i32   = 640;
const BOIDS_NUM:         usize = 100;
// const NEIGHBOURS_RADIUS: i32   = ((HEIGHT + WIDTH) / 2) / 10;

fn main() {
    let sdl = sdl2::init().unwrap();
    let video_subsys = sdl.video().unwrap();

    let window = video_subsys.window("boids", WIDTH as u32, HEIGHT as u32).build().unwrap();
    let mut canvas = window.into_canvas().build().unwrap();

    let mut boids: Vec<Boid> = Vec::with_capacity(BOIDS_NUM);

    let mut rng = rand::thread_rng();
    for _ in 0..BOIDS_NUM {
        boids.push(Boid::new(
            rng.gen_range(0, WIDTH as i32),
            rng.gen_range(0, HEIGHT as i32),
            rng.gen_range(-2.0, 2.0),
            rng.gen_range(-2.0, 2.0),
        ));
    }

    let mut event_pump = sdl.event_pump().unwrap();
    'main: loop {
        for event in event_pump.poll_iter() {
            match event {
                Event::Quit { .. } |
                Event::KeyDown { keycode: Some(Keycode::Q), .. } |
                Event::KeyDown { keycode: Some(Keycode::Escape), .. }
                    => break 'main,
                _ => {},
            }
        }
        canvas.set_draw_color(Color::RGB(0, 0, 0));
        canvas.clear();
        canvas.set_draw_color(Color::RGB(200, 200, 200));
        for b in boids.iter() {
            b.draw(&mut canvas);
        }

        canvas.present();

        let prev_boids = boids.clone();

        for b in boids.iter_mut() {
            b.step(&prev_boids, WIDTH, HEIGHT);
        }

        std::thread::sleep(std::time::Duration::new(0, 50_000_000));
    }
}