summaryrefslogtreecommitdiff
path: root/src/modules/tractorbeam.rs
blob: c77ea1fad2d7b918d79544e9fe3ef3a72f8549b3 (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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
use crate::constants;
use crate::mass::Mass;
use crate::math::Vector;

#[derive(Serialize, Deserialize, Debug, Clone, Default)]
pub struct Tractorbeam {
    pub status: TractorbeamStatus,
    strength: f64,
    desired_distance: Option<f64>,
}

#[derive(Serialize, Deserialize, Debug, Clone)]
pub enum TractorbeamStatus {
    None,
    Push,
    Pull,
    Bring,
}

impl Default for TractorbeamStatus {
    fn default() -> Self {
        TractorbeamStatus::None
    }
}

impl Tractorbeam {
    pub fn new() -> Tractorbeam {
        Tractorbeam {
            status: TractorbeamStatus::None,
            strength: constants::SHIP_TRACTORBEAM_STRENGTH,
            desired_distance: None,
        }
    }

    pub fn give_recv(&mut self, recv: String) {
        match recv.as_str() {
            "o" => self.toggle_pull(),
            "p" => self.toggle_push(),
            "b" => self.toggle_bring(5.0),
            _ => (),
        }
    }

    pub fn toggle_pull(&mut self) {
        self.status = match self.status {
            TractorbeamStatus::None => TractorbeamStatus::Pull,
            TractorbeamStatus::Push => TractorbeamStatus::Pull,
            TractorbeamStatus::Bring => TractorbeamStatus::Pull,
            TractorbeamStatus::Pull => TractorbeamStatus::None,
        }
    }

    pub fn toggle_push(&mut self) {
        self.status = match self.status {
            TractorbeamStatus::None => TractorbeamStatus::Push,
            TractorbeamStatus::Pull => TractorbeamStatus::Push,
            TractorbeamStatus::Bring => TractorbeamStatus::Push,
            TractorbeamStatus::Push => TractorbeamStatus::None,
        }
    }

    pub fn toggle_bring(&mut self, desired_distance: f64) {
        self.desired_distance = Some(desired_distance);
        self.status = match self.status {
            TractorbeamStatus::None => TractorbeamStatus::Bring,
            TractorbeamStatus::Pull => TractorbeamStatus::Bring,
            TractorbeamStatus::Push => TractorbeamStatus::Bring,
            TractorbeamStatus::Bring => TractorbeamStatus::None,
        }
    }

    pub fn off(&mut self) {
        self.status = TractorbeamStatus::None;
    }

    pub fn get_acceleration(&self, position: Vector, target: Mass) -> Vector {
        let acceleration = position.clone() - target.position.clone();
        match self.status {
            TractorbeamStatus::Push => acceleration.unitize() * -0.05,
            TractorbeamStatus::Pull => acceleration.unitize() * 0.05,
            TractorbeamStatus::Bring => match self.desired_distance {
                Some(desired_distance) => {
                    if desired_distance > position.distance_from(target.position) {
                        acceleration.unitize() * -0.05
                    //some sort of velocity limiter
                    //if target.speed_torwards(ship) < 10.0 {
                    //    acceleration.unitize() * -0.05
                    //} else {
                    //    Vector::default()
                    //}
                    } else {
                        acceleration.unitize() * 0.05
                    }
                }
                None => Vector::default(),
            },
            TractorbeamStatus::None => Vector::default(),
        }
    }
}