Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

modify leapfrog to take velocities instead of momenta, add new initial conditions

Browse Source
This commit is contained in:
Thibault Barnouin
2021-11-05 17:12:16 +01:00
parent d5f948f4a1
commit fff1832e61
5 changed files with 51 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
lib/plots.py
View File

@@ -14,6 +14,11 @@ class DynamicUpdate():
plt.ion()
def set_lims(self, factor=1.5):
self.ax.set_xlim(factor*self.min_x, factor*self.max_x)
self.ax.set_ylim(factor*self.min_x, factor*self.max_x)
self.ax.set_zlim(factor*self.min_x, factor*self.max_x)
def on_launch(self):
#Set up plot
self.fig = plt.figure()
@@ -21,27 +26,27 @@ class DynamicUpdate():
self.lines, = self.ax.plot([],[],[],'o')
#Autoscale on unknown axis and known lims on the other
self.ax.set_autoscaley_on(True)
self.ax.set_xlim(self.min_x, self.max_x)
self.ax.set_ylim(self.min_x, self.max_x)
self.ax.set_zlim(self.min_x, self.max_x)
self.set_lims()
#Other stuff
self.ax.grid()
#self.ax.set_aspect('equal')
def on_running(self, xdata, ydata, zdata, step=None, label=None):
values = np.sqrt(np.sum((np.array((xdata,ydata,zdata))**2).T,axis=1))
self.min_x, self.max_x = -np.abs(values).max(), np.abs(values).max()
self.set_lims()
#Update data (with the new _and_ the old points)
self.lines.set_data_3d(xdata, ydata, zdata)
if not label is None:
self.ax.set_title(label)
#Need both of these in order to rescale
self.ax.relim()
self.ax.autoscale_view()
#We need to draw *and* flush
self.fig.canvas.draw()
self.fig.canvas.flush_events()
if not step is None and step%100==0:
self.lines.set_data_3d(xdata, ydata, zdata)
if not label is None:
self.ax.set_title(label)
#Need both of these in order to rescale
self.ax.relim()
self.ax.autoscale_view()
#We need to draw *and* flush
self.fig.canvas.draw()
self.fig.canvas.flush_events()
if not step is None and step%100==0:
self.fig.savefig("tmp/{0:05d}.png".format(step),bbox_inches="tight")
self.fig.savefig("tmp/{0:05d}.png".format(step),bbox_inches="tight")
#Example
def __call__(self):