update

Source/test_evolution.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+"""
+Test: Evolution
+
+Evolve a particle with a given energy in a potential and show the result path
+followed by the particle in a given time.
+
+@ Author: Moussouni, Yaël (MSc student) & Bhat, Junaid Ramzan (MSc student)
+@ Institution:  Université de Strasbourg, CNRS, Observatoire astronomique
+                de Strasbourg, UMR 7550, F-67000 Strasbourg, France
+@ Date: 2024-11-29
+"""
+
+import numpy as np
+import matplotlib.pyplot as plt
+import potentials as pot
+import energies as ene
+import integrator as itg
+import initial_conditions as init
+import poincare_sections as pcs
+
+if "YII_1" in plt.style.available: plt.style.use("YII_1")
+# Loading matplotlib style...
+
+# Constants
+N_grid = 5000
+h = 0.01
+N_inter = 5000
+eps = 1e-2
+
+x_0 = 0.0
+y_0 = 0.5
+E = 0.125
+
+# Main
+if __name__ == "__main__":
+    W_part = init.one_part(x_0, y_0,0,0)
+    POT = pot.hh_potential(W_part)[0]
+    u_0 = np.sqrt(2 * (E - POT))
+    v_0 = 0.0
+
+    W_part[1,0] = u_0
+    W_part[1,1] = v_0
+
+    pos_t, positions = itg.rk4(0, W_part, h, N_inter, pot.hh_evolution)
+
+    pos_x = positions[:,0,0]
+    pos_y = positions[:,0,1]
+    vel_x = positions[:,1,0]
+    vel_y = positions[:,1,1]
+
+    W_grid = init.mesh_grid(N_grid)
+    X_grid = W_grid[0]
+    Y_grid = W_grid[1]
+    potential = pot.hh_potential(W_grid, position_only=True)
+
+    fig, ax = plt.subplots(1)
+
+    pcm = ax.pcolormesh(X_grid, Y_grid, potential)
+    line = ax.plot(pos_x, pos_y, color="C3")
+    fig.colorbar(pcm, label="potential")
+
+    ax.set_xlabel("$x$")
+    ax.set_ylabel("$y$")
+    ax.set_aspect("equal")
+
+    plt.show(block=True)
+
+