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

Revert "Make output_Energy safe" 

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

15
src/nbody-2.rs
View File

@ -102,7 +102,7 @@ fn offset_Momentum(bodies: &mut [body; BODIES_COUNT]) {
}
// Output the total energy of the system.
fn output_Energy(bodies: &mut [body; BODIES_COUNT]) {
unsafe 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.
@ -115,18 +115,21 @@ 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 = [0.; 3];
let mut position_Delta = [mem::MaybeUninit::<f64>::uninit(); 3];
for m in 0..3 {
position_Delta[m] =
bodies[i].position[m] - bodies[j].position[m];
position_Delta[m]
.as_mut_ptr()
.write(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(
energy -= bodies[i].mass * bodies[j].mass
/ f64::sqrt(
position_Delta[0] * position_Delta[0]
+ position_Delta[1] * position_Delta[1]
+ position_Delta[2] * position_Delta[2]);
+ position_Delta[2] * position_Delta[2],
);
}
}