general debug and display angular momentum and mechanical energy curves

2021-10-30 13:24:02 +02:00
parent a82d316cc9
commit 9a16400a89
3 changed files with 20 additions and 10 deletions
--- a/lib/objects.py
+++ b/lib/objects.py
@@ -55,9 +55,8 @@ class System:
        return coord

    def COMShift(self): #Shift coordinates of bodies in system to COM frame and set COM at rest
-        comcoord = self.COM
        for body in self.bodylist:
-            body.q = body.q-self.COM()
+            body.q = body.q - self.COM()
            body.p = body.p - self.COMV()
        return 0