Add constrained physics sim

Performs slightly worse, sends less bodies flying off.
This commit is contained in:
Michael Bradley 2023-10-07 21:42:30 -04:00
parent d3a793c3a3
commit 86d5ca98bf
2 changed files with 23 additions and 2 deletions

View file

@ -3,7 +3,7 @@ import matplotlib.cm as cm
import matplotlib.pyplot as plt
import numpy as np
from physics import n_body_matrix
import physics
def parse_csv(filename: str):
@ -54,7 +54,7 @@ class Animator:
return self.scat,
def update(self, *_args, **_kwargs):
n_body_matrix(self.pos, self.vel, self.mass)
physics.n_body_matrix_constrained(self.pos, self.vel, self.mass)
self.scat.set_offsets(self.pos)
return self.scat,