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
use na::RealField;
use crate::force_generator::ForceGenerator;
use crate::math::{Force, ForceType, Vector, Velocity};
use crate::object::{BodyHandle, BodyPartHandle, BodySet};
use crate::solver::IntegrationParameters;
pub struct ConstantAcceleration<N: RealField, Handle: BodyHandle> {
parts: Vec<BodyPartHandle<Handle>>,
acceleration: Velocity<N>,
}
impl<N: RealField, Handle: BodyHandle> ConstantAcceleration<N, Handle> {
#[cfg(feature = "dim3")]
pub fn new(linear_acc: Vector<N>, angular_acc: Vector<N>) -> Self {
ConstantAcceleration {
parts: Vec::new(),
acceleration: Velocity::new(linear_acc, angular_acc),
}
}
#[cfg(feature = "dim2")]
pub fn new(linear_acc: Vector<N>, angular_acc: N) -> Self {
ConstantAcceleration {
parts: Vec::new(),
acceleration: Velocity::new(linear_acc, angular_acc),
}
}
pub fn add_body_part(&mut self, body: BodyPartHandle<Handle>) {
self.parts.push(body)
}
}
impl<N: RealField, Handle: BodyHandle> ForceGenerator<N, Handle>
for ConstantAcceleration<N, Handle>
{
fn apply(
&mut self,
_: &IntegrationParameters<N>,
bodies: &mut dyn BodySet<N, Handle = Handle>,
) {
let acceleration = self.acceleration;
self.parts.retain(|h| {
if let Some(body) = bodies.get_mut(h.0) {
body.apply_force(
h.1,
&Force::new(acceleration.linear, acceleration.angular),
ForceType::AccelerationChange,
false,
);
true
} else {
false
}
});
}
}