Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

Rename class.py to objects.py to avoid library confict

Browse Source
This commit is contained in:
Thibault Barnouin
2021-10-22 13:00:19 +02:00
parent f10f23bdbc
commit b096021fc4
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/class.py → lib/objects.py
View File

@@ -7,7 +7,7 @@ import numpy as np
class Body: class Body:
def __init__(self, mass, position, velocity): def __init__(self, mass, position, velocity):
self.m = mass self.m = mass
self.q = position self.q = position
@@ -16,11 +16,10 @@ class Body:
def __repr__(self): # Called upon "print(body)" def __repr__(self): # Called upon "print(body)"
return "Body of mass: {0:.2f}kg, position: {1}, velocity: {2}".format(self.m, self.p, self.v) return "Body of mass: {0:.2f}kg, position: {1}, velocity: {2}".format(self.m, self.p, self.v)
def __str__(self): # Called upon "str(body)" def __str__(self): # Called upon "str(body)"
return "Body of mass: {0:.2f}kg, position: {1}, velocity: {2}".format(self.m, self.p, self.v) return "Body of mass: {0:.2f}kg, position: {1}, velocity: {2}".format(self.m, self.p, self.v)
class System: class System:
def __init__(self, bodylist): def __init__(self, bodylist):
@@ -54,12 +53,12 @@ class System:
return L return L
if __name__ == "__main__": if __name__ == "__main__":
# initialisation mass # initialisation mass
m1 = 10 m1 = 10
m2 = 1 m2 = 1
m3 = 1 m3 = 1
# initialisation position # initialisation position
q1 = np.array([0, 0, 0]) q1 = np.array([0, 0, 0])
q2 = np.array([1, 0, 0]) q2 = np.array([1, 0, 0])