9efa97de386265de519310f5f1b263c1f67cb047
[kaka/rust-sdl-test.git] / src / main.rs
1 extern crate rand;
2 extern crate sdl2;
3 extern crate time;
4
5 use std::f32::consts::PI;
6
7 use rand::Rng;
8 use sdl2::event::Event;
9 use sdl2::EventPump;
10 use sdl2::gfx::primitives::DrawRenderer;
11 use sdl2::keyboard::Keycode;
12 use sdl2::pixels::Color;
13 use sdl2::rect::Rect;
14 use sdl2::render::BlendMode;
15 use sdl2::render::Canvas;
16 use sdl2::video::FullscreenType;
17 use sdl2::video::Window;
18 use time::PreciseTime;
19
20 use boll::{Boll, CircleBoll, SquareBoll};
21 use common::Point2D;
22 use sprites::SpriteManager;
23 use sdl2::video::WindowContext;
24 use sdl2::render::TextureCreator;
25
26 #[macro_use] mod common;
27 mod boll;
28 mod sprites;
29
30 const SCREEN_WIDTH: u32 = 1280;
31 const SCREEN_HEIGHT: u32 = (SCREEN_WIDTH as f64 * (1440.0 / 2560.0)) as u32;
32 const FPS: u32 = 60;
33 const NS_PER_FRAME: u32 = 1_000_000_000 / FPS;
34
35 type Bollar = Vec<Box<dyn Boll>>;
36
37 fn init() -> (Canvas<Window>, EventPump) {
38     let context = sdl2::init().unwrap();
39     let video = context.video().unwrap();
40     let _image_context = sdl2::image::init(sdl2::image::InitFlag::PNG).unwrap();
41     let window = video.window("SDL test", SCREEN_WIDTH, SCREEN_HEIGHT)
42         .position_centered()
43         .opengl()
44         .build()
45         .unwrap();
46     context.mouse().show_cursor(false);
47     let mut canvas = window.into_canvas().build().unwrap();
48     canvas.set_blend_mode(BlendMode::Add);
49     canvas.set_draw_color(Color::RGB(0, 0, 0));
50     canvas.clear();
51     canvas.present();
52     let event_pump = context.event_pump().unwrap();
53     (canvas, event_pump)
54 }
55
56 fn load_sprites(texture_creator: &TextureCreator<WindowContext>) -> SpriteManager {
57     let mut texman = SpriteManager::new(texture_creator);
58     texman.load("block", "res/block.bmp");
59     texman.load("mario", "res/mario-trans.png");
60     texman
61 }
62
63 fn main() {
64     println!("starting...");
65     let (mut canvas, mut event_pump) = init();
66
67     let mut frame_count: u64 = 0;
68     let mut fps_time = PreciseTime::now();
69
70     let mut bolls: Bollar = Bollar::new();
71     let mut boll_size = 1;
72
73     let texture_creator = canvas.texture_creator();
74     let sprites = load_sprites(&texture_creator);
75     let mut mario_angle = 0.0;
76
77     'running: loop {
78         let loop_start = PreciseTime::now();
79         canvas.set_draw_color(Color::RGB(0, 0, 0));
80         canvas.clear();
81         {
82             let blocks = 20;
83             let size = 32;
84             let offset = point!((SCREEN_WIDTH as i32 - (blocks + 1) * size) / 2, (SCREEN_HEIGHT as i32 - (blocks + 1) * size) / 2);
85             for i in 0..blocks {
86                 canvas.copy(sprites.get("block"), None, Rect::new((i) * size + offset.x, (0) * size + offset.y, size as u32, size as u32)).unwrap();
87                 canvas.copy(sprites.get("block"), None, Rect::new((blocks - i) * size + offset.x, (blocks) * size + offset.y, size as u32, size as u32)).unwrap();
88                 canvas.copy(sprites.get("block"), None, Rect::new((0) * size + offset.x, (blocks - i) * size + offset.y, size as u32, size as u32)).unwrap();
89                 canvas.copy(sprites.get("block"), None, Rect::new((blocks) * size + offset.x, (i) * size + offset.y, size as u32, size as u32)).unwrap();
90             }
91         }
92         {
93             let size = 64;
94             let offset = point!((SCREEN_WIDTH as i32 - size) / 2, (SCREEN_HEIGHT as i32 - size) / 2);
95             let radius = 110.0 + size as f32 * 0.5;
96             let angle = (mario_angle as f32 - 90.0) * PI / 180.0;
97             let offset2 = point!((angle.cos() * radius) as i32, (angle.sin() * radius) as i32);
98             canvas.copy_ex(sprites.get("mario"), None, Rect::new(offset.x + offset2.x, offset.y + offset2.y, size as u32, size as u32), mario_angle, sdl2::rect::Point::new(size / 2, size / 2), false, false).unwrap();
99             mario_angle += 1.0;
100             if mario_angle >= 360.0 { mario_angle -= 360.0 }
101         }
102         {
103             let p = point!((SCREEN_WIDTH / 2) as i16, (SCREEN_HEIGHT / 2) as i16);
104             canvas.circle(p.x, p.y, 100, Color::RGB(255, 255, 255)).unwrap();
105             canvas.aa_circle(p.x, p.y, 110, Color::RGB(255, 255, 255)).unwrap();
106             canvas.ellipse(p.x, p.y, 50, 100, Color::RGB(255, 255, 255)).unwrap();
107             canvas.aa_ellipse(p.x, p.y, 110, 55, Color::RGB(255, 255, 255)).unwrap();
108         }
109
110         for b in &mut bolls {
111             b.update();
112             b.draw(&mut canvas, boll_size);
113         }
114
115 //        window.gl_swap_window();
116         for event in event_pump.poll_iter() {
117             match event {
118                 Event::Quit { .. } | Event::KeyDown { keycode: Some(Keycode::Escape), .. } => {
119                     break 'running;
120                 }
121                 Event::KeyDown { keycode: Some(Keycode::F11), .. } => {
122                     match canvas.window().fullscreen_state() {
123                         FullscreenType::Off => canvas.window_mut().set_fullscreen(FullscreenType::Desktop),
124                         _                   => canvas.window_mut().set_fullscreen(FullscreenType::Off)
125                     }.unwrap();
126                 }
127                 Event::KeyDown { keycode: Some(Keycode::KpPlus), .. } => { boll_size = std::cmp::min(boll_size + 1, 32) }
128                 Event::KeyDown { keycode: Some(Keycode::KpMinus), .. } => { boll_size = std::cmp::max(boll_size - 1, 1) }
129                 Event::MouseMotion { x, y, .. } => {
130                     bolls.push(Box::new(CircleBoll::new(
131                         point!(x as f64, y as f64),
132                         point!(0.0, 0.0),
133                     )))
134                 }
135                 _ => {}
136             }
137         }
138         canvas.present();
139
140         match loop_start.to(PreciseTime::now()).num_nanoseconds() {
141             Some(ns) if ns < (NS_PER_FRAME - 50_0000) as i64 => { change_boll_count(&mut bolls, 100) }
142             Some(ns) if ns > (NS_PER_FRAME + 50_0000) as i64 => { change_boll_count(&mut bolls, -100) }
143             _ => {}
144         }
145
146         frame_count += 1;
147         if frame_count == FPS as u64 {
148             let duration = fps_time.to(PreciseTime::now()).num_nanoseconds().unwrap() as f64 / 1_000_000_000.0;
149             println!("fps: {}", frame_count as f64 / duration);
150             frame_count = 0;
151             fps_time = PreciseTime::now();
152         }
153     }
154
155     println!("number of bolls: {}", bolls.len());
156 }
157
158 fn change_boll_count(mut bolls: &mut Bollar, delta: i32) {
159     if delta > 0 {
160         for _i in 0..delta {
161             add_boll(&mut bolls);
162         }
163     } else if delta < 0 {
164         for _i in 0..delta {
165             bolls.pop();
166         }
167     }
168 }
169
170 fn add_boll(bolls: &mut Bollar) {
171     let mut rng = rand::thread_rng();
172     bolls.push(Box::new(SquareBoll {
173         pos: point!(rng.gen_range(0, SCREEN_WIDTH) as f64, rng.gen_range(0, SCREEN_HEIGHT) as f64),
174         vel: point!(rng.gen_range(-2.0, 2.0), rng.gen_range(-2.0, 2.0)),
175     }));
176 }