tutorials
/
learn-rust-the-dangerous-way
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Make output_Energy a safe function

This commit is contained in:
Timothy Warren 2020-01-13 11:29:54 -05:00
parent 7b655129e8
commit 7c511f9b48
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main.rs
View File

@ -101,7 +101,7 @@ fn offset_Momentum(bodies: &mut [body; BODIES_COUNT]) {
}
// Output the total energy of the system.
unsafe fn output_Energy(bodies: &mut [body; BODIES_COUNT]) {
fn output_Energy(bodies: &mut [body; BODIES_COUNT]) {
let mut energy = 0.;
for i in 0..BODIES_COUNT {
// Add the kinetic energy for each body.
@ -114,14 +114,11 @@ unsafe fn output_Energy(bodies: &mut [body; BODIES_COUNT]) {
// Add the potential energy between this body and
// every other body
for j in i + 1..BODIES_COUNT {
let mut position_Delta = [mem::MaybeUninit::<f64>::uninit(); 3];
let mut position_Delta = [0.; 3];
for m in 0..3 {
position_Delta[m]
.as_mut_ptr()
.write(bodies[i].position[m] - bodies[j].position[m]);
position_Delta[m] = bodies[i].position[m] - bodies[j].position[m];
}
let position_Delta: [f64; 3] = mem::transmute(position_Delta);
energy -= bodies[i].mass * bodies[j].mass
/ f64::sqrt(