Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

Add timestamps in stdout

Browse Source
This commit is contained in:
Thibault Barnouin
2022-01-14 19:44:05 +01:00
parent f9427d3125
commit 13dac2691d
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -1,6 +1,7 @@
#!/usr/bin/python #!/usr/bin/python
# -*- coding:utf-8 -*- # -*- coding:utf-8 -*-
from sys import exit as sysexit from sys import exit as sysexit
from time import time
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from lib.objects import Body, System from lib.objects import Body, System
@@ -41,12 +42,17 @@ def main():
bodylist.append(Body(m[j], q[j], v[j])) bodylist.append(Body(m[j], q[j], v[j]))
bin_syst = System(bodylist[0:2]) bin_syst = System(bodylist[0:2])
dyn_syst = System(bodylist, main=True) dyn_syst = System(bodylist, main=True)
print("Integration start...")
t1 = time()
if integrator.lower() in ['leapfrog', 'frogleap', 'frog']: if integrator.lower() in ['leapfrog', 'frogleap', 'frog']:
E, L, sma, ecc, phi = leapfrog(dyn_syst, bin_syst, duration, step, recover_param=True, display=display, savename=savename, gif=gif) E, L, sma, ecc, phi = leapfrog(dyn_syst, bin_syst, duration, step, recover_param=True, display=display, savename=savename, gif=gif)
elif integrator.lower() in ['hermite','herm']: elif integrator.lower() in ['hermite','herm']:
E, L, sma, ecc, phi = hermite(dyn_syst, bin_syst, duration, step, recover_param=True, display=display, savename=savename, gif=gif) E, L, sma, ecc, phi = hermite(dyn_syst, bin_syst, duration, step, recover_param=True, display=display, savename=savename, gif=gif)
t2=time()
print("...Integration end.\n Elapsed time {0:.3f} sec".format(t2-t1))
parameters = [duration, [step], dyn_syst, integrator, [a, e, psi]] parameters = [duration, [step], dyn_syst, integrator, [a, e, psi]]
display_parameters([E], [L], [sma], [ecc], [phi], parameters=parameters, savename=savename, display_param=display_param) display_parameters([E], [L], [sma], [ecc], [phi], parameters=parameters, savename=savename, display_param=display_param)
return 0 return 0