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
|
use std::collections::HashMap;
use std::time::SystemTime;
use crate::constants;
use crate::mass::Mass;
use crate::math::Vector;
#[derive(Serialize, Deserialize, Debug, Clone, Default)]
pub struct Navigation {
pub range: f64,
pub status: Status,
pub target_name: Option<String>,
time: u64,
start: Option<SystemTime>,
}
impl Navigation {
pub fn new() -> Navigation {
Navigation {
target_name: None,
range: constants::SHIP_NAVIGATION_RANGE,
status: Status::None,
time: constants::SHIP_NAVIGATION_TIME,
start: None,
}
}
pub fn process(&mut self, ship_position: Vector, masses: &mut HashMap<String, Mass>) {
self.verify_target(ship_position, masses);
if let Some(timer) = self.start {
if timer.elapsed().unwrap().as_secs() > self.time {
self.status = Status::Targeted;
self.start = None;
}
}
}
pub fn give_received_data(&mut self, recv: String) {
if !recv.is_empty() {
self.start = Some(SystemTime::now());
self.status = Status::Targeting;
self.target_name = Some(recv);
}
}
pub fn get_client_data(&self, ship_position: Vector, masses: &HashMap<String, Mass>) -> String {
let client_data = ClientData {
ship_position: ship_position.clone(),
status: self.status.clone(),
target_name: self.target_name.clone(),
available_targets: masses
.iter()
.filter(|(_, mass)| ship_position.distance_from(mass.position.clone()) < self.range)
.map(|(name, mass)| (name.to_string(), mass.position.clone()))
.collect(),
};
serde_json::to_string(&client_data).unwrap() + "\n"
}
fn verify_target(&mut self, ship_position: Vector, masses: &HashMap<String, Mass>) {
if let Some(name) = self.target_name.clone() {
let target = masses.get(&name).unwrap();
if target.position.distance_from(ship_position) > self.range {
self.target_name = None;
self.status = Status::None;
}
}
}
}
#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct ClientData {
pub ship_position: Vector,
pub available_targets: Vec<(String, Vector)>,
pub status: Status,
pub target_name: Option<String>,
}
#[derive(Serialize, Deserialize, Debug, Clone, PartialEq)]
pub enum Status {
None,
Targeting,
Targeted,
}
impl Default for Status {
fn default() -> Self {
Status::None
}
}
|
