(ref, bodies=SYSTEM, px=0.0, py=0.0, pz=0.0)
| 84 | |
| 85 | |
| 86 | def offset_momentum(ref, bodies=SYSTEM, px=0.0, py=0.0, pz=0.0): |
| 87 | for (r, [vx, vy, vz], m) in bodies: |
| 88 | px -= vx * m |
| 89 | py -= vy * m |
| 90 | pz -= vz * m |
| 91 | (r, v, m) = ref |
| 92 | v[0] = px / m |
| 93 | v[1] = py / m |
| 94 | v[2] = pz / m |
| 95 | |
| 96 | |
| 97 | def main(n, ref='sun'): |