MichaelBradley/nbody
1
0
Fork 0
Code Issues Pull requests Activity Actions

Reduce dimensional alterations

Browse source
This commit is contained in:
Michael Bradley 2023-10-07 18:30:57 -04:00
parent 9e8d8fe620
commit 3ebf5b9baf
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
physics.py
View file

@ -11,7 +11,7 @@ def rotations(a: np.ndarray):
def n_body(pos: np.ndarray, vel: np.ndarray, mass: np.ndarray):
for (o_pos, o_mass) in zip(rotations(pos), rotations(mass)):
for (o_pos, o_mass) in zip(rotations(pos), rotations(mass[:, np.newaxis])):
dist = o_pos - pos
vel += G * (dist / np.linalg.norm(dist, axis=1)[:, np.newaxis]) * o_mass[:, np.newaxis] / np.sum(dist ** 2, axis=1)[:, np.newaxis]
vel += G * (dist / np.linalg.norm(dist, axis=1)[:, np.newaxis]) * o_mass / np.sum(dist ** 2, axis=1)[:, np.newaxis]
pos += vel