be546c12b97f56b7de23ddf81daf708488898ee6
[kaka/rust-sdl-test.git] / src / core / app.rs
1 use boll::*;
2 use common::{Point2D, Rect};
3 use core::controller::ControllerManager;
4 use point; // defined in common, but loaded from main...
5 use rand::Rng;
6 use sdl2::event::{Event, WindowEvent};
7 use sdl2::gfx::primitives::DrawRenderer;
8 use sdl2::keyboard::Keycode;
9 use sdl2::pixels::Color;
10 use sdl2::rect::Rect as SDLRect;
11 use sdl2::render::{BlendMode, Canvas};
12 use sdl2::video::{FullscreenType, SwapInterval, Window};
13 use sdl2::{EventPump, VideoSubsystem};
14 use sprites::SpriteManager;
15 use std::f32::consts::PI;
16 use time::{Duration, Instant, prelude::*};
17
18 const FPS: u32 = 60;
19 const NS_PER_FRAME: u32 = 1_000_000_000 / FPS;
20
21 #[derive(Default)]
22 pub struct AppBuilder {
23     resolution: Rect<u16>,
24     state: Option<Box<dyn AppState>>,
25     title: Option<String>,
26 }
27
28 impl AppBuilder {
29     pub fn with_resolution(mut self, width: u16, height: u16) -> Self {
30         self.resolution = Rect { width, height };
31         self
32     }
33
34     pub fn with_state(mut self, state: Box<dyn AppState>) -> Self {
35         self.state = Some(state);
36         self
37     }
38
39     pub fn with_title(mut self, title: &str) -> Self {
40         self.title = Some(title.to_string());
41         self
42     }
43
44     pub fn build(self) -> Result<App, String> {
45         let context = sdl2::init().unwrap();
46         sdl2::image::init(sdl2::image::InitFlag::PNG)?;
47         let video = context.video()?;
48         //self.print_video_display_modes(&video);
49
50         let window = video
51             .window(
52                 &self.title.unwrap(),
53                 self.resolution.width.into(),
54                 self.resolution.height.into(),
55             )
56             .position_centered()
57             // .fullscreen()
58             // .fullscreen_desktop()
59             .opengl()
60             .build()
61             .unwrap();
62         context.mouse().show_cursor(false);
63
64         let mut canvas = window.into_canvas().build().unwrap();
65         canvas.set_blend_mode(BlendMode::Add);
66         canvas.set_draw_color(Color::RGB(0, 0, 0));
67         canvas.clear();
68         canvas.present();
69
70         video.gl_set_swap_interval(SwapInterval::VSync)?;
71
72         let event_pump = context.event_pump()?;
73         let sprites = SpriteManager::new(canvas.texture_creator());
74         let screen = canvas.output_size().unwrap();
75
76         Ok(App {
77             canvas,
78             event_pump,
79             sprites,
80             state: self.state.unwrap_or_else(|| Box::new(ActiveState::new(screen))),
81             ctrl_man: ControllerManager::new(context.joystick()?, context.haptic()?),
82         })
83     }
84
85     #[allow(dead_code)]
86     fn print_video_display_modes(&self, video: &VideoSubsystem) {
87         println!("video subsystem: {:?}", video);
88         println!("current_video_driver: {:?}", video.current_video_driver());
89         for display in 0..video.num_video_displays().unwrap() {
90             println!(
91                 "=== display {} - {} ===",
92                 display,
93                 video.display_name(display).unwrap()
94             );
95             println!(
96                 " display_bounds: {:?}",
97                 video.display_bounds(display).unwrap()
98             );
99             println!(
100                 " num_display_modes: {:?}",
101                 video.num_display_modes(display).unwrap()
102             );
103             println!(
104                 " desktop_display_mode: {:?}",
105                 video.desktop_display_mode(display).unwrap()
106             );
107             let current = video.current_display_mode(display).unwrap();
108             println!(
109                 " current_display_mode: {:?}",
110                 current
111             );
112             for idx in 0..video.num_display_modes(display).unwrap() {
113                 let mode = video.display_mode(display, idx).unwrap();
114                 println!(
115                     " {}{:2}: {:?}",
116                     if mode == current { "*" } else { " " },
117                     idx,
118                     mode
119                 );
120             }
121         }
122         println!("swap interval: {:?}", video.gl_get_swap_interval());
123     }
124 }
125
126 pub struct App {
127     canvas: Canvas<Window>,
128     event_pump: EventPump,
129     sprites: SpriteManager,
130     state: Box<dyn AppState>,
131     pub ctrl_man: ControllerManager,
132 }
133
134 impl App {
135     #[allow(clippy::new_ret_no_self)]
136     pub fn new() -> AppBuilder {
137         Default::default()
138     }
139
140     pub fn load_sprites(&mut self, sprites: &[(&str, &str)]) {
141         for (name, file) in sprites {
142             self.sprites.load(name, file);
143         }
144     }
145
146     pub fn start(&mut self) {
147         let mut last_time = Instant::now();
148
149         self.state.enter(&mut self.ctrl_man);
150
151         'running: loop {
152             if let Err(_) = self.handle_events() {
153                 break 'running;
154             }
155
156             let duration = Instant::now() - last_time;
157             last_time = Instant::now();
158
159             self.ctrl_man.update(duration);
160             self.state.update(duration);
161
162             self.render();
163         }
164
165         self.state.leave();
166     }
167
168     fn handle_events(&mut self) -> Result<(), ()> {
169         for event in self.event_pump.poll_iter() {
170             self.ctrl_man.handle_event(&event);
171             match event {
172                 Event::Quit { .. }
173                 | Event::KeyDown {
174                     keycode: Some(Keycode::Escape),
175                     ..
176                 } => {
177                     return Err(())
178                 }
179                 Event::KeyDown {
180                     keycode: Some(Keycode::F11),
181                     ..
182                 } => {
183                     match self.canvas.window().fullscreen_state() {
184                         FullscreenType::Off => self
185                             .canvas
186                             .window_mut()
187                             .set_fullscreen(FullscreenType::Desktop),
188                         _ => self.canvas.window_mut().set_fullscreen(FullscreenType::Off),
189                     }
190                     .unwrap();
191                 }
192                 Event::Window {
193                     win_event: WindowEvent::Resized(x, y),
194                     ..
195                 } => {
196                     println!("window resized({}, {})", x, y)
197                 }
198                 Event::Window {
199                     win_event: WindowEvent::Maximized,
200                     ..
201                 } => {
202                     println!("window maximized")
203                 }
204                 Event::Window {
205                     win_event: WindowEvent::Restored,
206                     ..
207                 } => {
208                     println!("window restored")
209                 }
210                 Event::Window {
211                     win_event: WindowEvent::Enter,
212                     ..
213                 } => {
214                     println!("window enter")
215                 }
216                 Event::Window {
217                     win_event: WindowEvent::Leave,
218                     ..
219                 } => {
220                     println!("window leave")
221                 }
222                 Event::Window {
223                     win_event: WindowEvent::FocusGained,
224                     ..
225                 } => {
226                     println!("window focus gained")
227                 }
228                 Event::Window {
229                     win_event: WindowEvent::FocusLost,
230                     ..
231                 } => {
232                     println!("window focus lost")
233                 }
234                 _ => self.state.handle_event(event),
235             }
236         }
237         Ok(())
238     }
239
240     fn render(&mut self) {
241         self.canvas.set_draw_color(Color::RGB(0, 0, 0));
242         self.canvas.clear();
243         self.state.render(&mut self.canvas, &mut self.sprites);
244         self.canvas.present();
245     }
246 }
247
248 pub trait AppState {
249     fn enter(&mut self, ctrl_man: &mut ControllerManager);
250     fn leave(&mut self);
251     fn update(&mut self, dt: Duration);
252     fn render(&mut self, canvas: &mut Canvas<Window>, sprites: &mut SpriteManager);
253     fn handle_event(&mut self, event: Event);
254 }
255
256 type Bollar = Vec<Box<dyn Boll>>;
257
258 #[derive(Default)]
259 pub struct ActiveState {
260     screen: Rect<u32>,
261     bolls: Bollar,
262     boll_size: u32,
263     mario_angle: f64,
264 }
265
266 impl ActiveState {
267     pub fn new(screen: (u32, u32)) -> ActiveState {
268         ActiveState {
269             bolls: Bollar::new(),
270             boll_size: 1,
271             screen: Rect::from(screen),
272             ..Default::default()
273         }
274     }
275
276     fn change_boll_count(&mut self, delta: i32) {
277         #[allow(clippy::comparison_chain)]
278         if delta > 0 {
279             for _i in 0..delta {
280                 self.add_boll();
281             }
282         } else if delta < 0 {
283             for _i in 0..(-delta) {
284                 self.bolls.pop();
285             }
286         }
287     }
288
289     fn add_boll(&mut self) {
290         let mut rng = rand::thread_rng();
291         self.bolls.push(Box::new(SquareBoll {
292             pos: point!(
293                 rng.gen_range(0, self.screen.width) as f64,
294                 rng.gen_range(0, self.screen.height) as f64
295             ),
296             vel: point!(rng.gen_range(-2.0, 2.0), rng.gen_range(-2.0, 2.0)),
297         }));
298     }
299 }
300
301 impl AppState for ActiveState {
302     fn enter(&mut self, _ctrl_man: &mut ControllerManager) {}
303
304     fn update(&mut self, dt: Duration) {
305         for b in &mut self.bolls {
306             b.update();
307         }
308
309         match dt {
310             ns if ns < (NS_PER_FRAME - 90_0000).nanoseconds() => self.change_boll_count(100),
311             ns if ns > (NS_PER_FRAME + 90_0000).nanoseconds() => self.change_boll_count(-100),
312             _ => {}
313         }
314     }
315
316     fn render(&mut self, canvas: &mut Canvas<Window>, sprites: &mut SpriteManager) {
317         /* draw square of blocks */ {
318             let blocks = 20;
319             let size = 32;
320             let offset = point!(
321                 (self.screen.width as i32 - (blocks + 1) * size) / 2,
322                 (self.screen.height as i32 - (blocks + 1) * size) / 2
323             );
324             let block = sprites.get("block");
325             for i in 0..blocks {
326                 canvas
327                     .copy(
328                         block,
329                         None,
330                         SDLRect::new((i) * size + offset.x, offset.y, size as u32, size as u32),
331                     )
332                     .unwrap();
333                 canvas
334                     .copy(
335                         block,
336                         None,
337                         SDLRect::new(
338                             (blocks - i) * size + offset.x,
339                             (blocks) * size + offset.y,
340                             size as u32,
341                             size as u32,
342                         ),
343                     )
344                     .unwrap();
345                 canvas
346                     .copy(
347                         block,
348                         None,
349                         SDLRect::new(
350                             offset.x,
351                             (blocks - i) * size + offset.y,
352                             size as u32,
353                             size as u32,
354                         ),
355                     )
356                     .unwrap();
357                 canvas
358                     .copy(
359                         block,
360                         None,
361                         SDLRect::new(
362                             (blocks) * size + offset.x,
363                             (i) * size + offset.y,
364                             size as u32,
365                             size as u32,
366                         ),
367                     )
368                     .unwrap();
369             }
370         }
371
372         /* draw mario */ {
373             let size = 64;
374             let offset = point!(
375                 (self.screen.width as i32 - size) / 2,
376                 (self.screen.height as i32 - size) / 2
377             );
378             let radius = 110.0 + size as f32 * 0.5;
379             let angle = (self.mario_angle as f32 - 90.0) * PI / 180.0;
380             let offset2 = point!((angle.cos() * radius) as i32, (angle.sin() * radius) as i32);
381             canvas
382                 .copy_ex(
383                     sprites.get("mario"),
384                     None,
385                     SDLRect::new(
386                         offset.x + offset2.x,
387                         offset.y + offset2.y,
388                         size as u32,
389                         size as u32,
390                     ),
391                     self.mario_angle,
392                     sdl2::rect::Point::new(size / 2, size / 2),
393                     false,
394                     false,
395                 )
396                 .unwrap();
397             self.mario_angle = (self.mario_angle + 1.0) % 360.0;
398         }
399
400         /* draw circles and ellipses*/ {
401             let p = point!((self.screen.width / 2) as i16, (self.screen.height / 2) as i16);
402             canvas
403                 .circle(p.x, p.y, 100, Color::RGB(255, 255, 255))
404                 .unwrap();
405             canvas
406                 .aa_circle(p.x, p.y, 110, Color::RGB(255, 255, 255))
407                 .unwrap();
408             canvas
409                 .ellipse(p.x, p.y, 50, 100, Color::RGB(255, 255, 255))
410                 .unwrap();
411             canvas
412                 .aa_ellipse(p.x, p.y, 110, 55, Color::RGB(255, 255, 255))
413                 .unwrap();
414         }
415
416         for b in &self.bolls {
417             b.draw(canvas, self.boll_size);
418         }
419     }
420
421     fn leave(&mut self) {
422         println!("number of bolls: {}", self.bolls.len());
423     }
424
425     fn handle_event(&mut self, event: Event) {
426         match event {
427             Event::KeyDown {
428                 keycode: Some(Keycode::KpPlus),
429                 ..
430             } => self.boll_size = std::cmp::min(self.boll_size + 1, 32),
431             Event::KeyDown {
432                 keycode: Some(Keycode::KpMinus),
433                 ..
434             } => self.boll_size = std::cmp::max(self.boll_size - 1, 1),
435             Event::MouseMotion { x, y, .. } => self.bolls.push(Box::new(CircleBoll::new(
436                 point!(x as f64, y as f64),
437                 point!(0.0, 0.0),
438             ))),
439             _ => {}
440         }
441     }
442 }