summaryrefslogtreecommitdiff
path: root/src/modules/tractorbeam.rs
blob: 3bbb32a4920b4a6fc7d6117b82430258615f8a73 (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
101
102
103
104
105
106
107
108
109
110
111
112
113
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>,
}

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

    pub fn process(&mut self) {}

    pub fn get_client_data(&self, target: Option<&Mass>) -> String {
        let client_data = TractorbeamClientData {
            has_target: target.is_some(),
            status: self.status.clone(),
        };

        serde_json::to_string(&client_data).unwrap() + "\n"
    }

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

    pub fn get_acceleration(&self, ship_position: Vector, target_position: Vector) -> Vector {
        let acceleration = ship_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 > ship_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(),
        }
    }

    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,
        }
    }

    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,
        }
    }

    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,
        }
    }
}

#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct TractorbeamClientData {
    pub has_target: bool,
    pub status: TractorbeamStatus,
}

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

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