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
use geom::{Angle, Speed};
use widgetry::{EventCtx, Key};

// TODO The timestep accumulation seems fine. What's wrong? Clamping errors repeated?
const HACK: f64 = 5.0;

pub struct InstantController {
    /// Which of the 8 directions are we facing, based on the last set of keys pressed down?
    pub facing: Angle,
}

impl InstantController {
    pub fn new() -> InstantController {
        InstantController {
            facing: Angle::ZERO,
        }
    }

    pub fn displacement(&mut self, ctx: &mut EventCtx, speed: Speed) -> Option<(f64, f64)> {
        let dt = ctx.input.nonblocking_is_update_event()?;
        // Work around a few bugs here.
        //
        // 1) The Santa sprites are all facing 180 degrees, not 0, so invert X.
        // 2) Invert y so that negative is up.
        //
        // It's confusing, but self.facing winds up working for rotating the sprite, and the output
        // displacement works.
        self.facing = angle_from_arrow_keys(ctx)?.opposite();
        let magnitude = (dt * HACK * speed).inner_meters();
        let (sin, cos) = self.facing.normalized_radians().sin_cos();
        Some((-magnitude * cos, -magnitude * sin))
    }
}

pub fn angle_from_arrow_keys(ctx: &EventCtx) -> Option<Angle> {
    let mut x: f64 = 0.0;
    let mut y: f64 = 0.0;
    if ctx.is_key_down(Key::LeftArrow) || ctx.is_key_down(Key::A) {
        x -= 1.0;
    }
    if ctx.is_key_down(Key::RightArrow) || ctx.is_key_down(Key::D) {
        x += 1.0;
    }
    if ctx.is_key_down(Key::UpArrow) || ctx.is_key_down(Key::W) {
        y -= 1.0;
    }
    if ctx.is_key_down(Key::DownArrow) || ctx.is_key_down(Key::S) {
        y += 1.0;
    }

    if x == 0.0 && y == 0.0 {
        return None;
    }
    Some(Angle::new_rads(y.atan2(x)))
}