Some tools for Hamiltonian systems
Project description
pyHamSys
pyHamSys is a Python package for scientific computing involving Hamiltonian systems
Installation:
pip install pyhamsys
Symplectic Integrators
pyHamSys includes a class SymplecticIntegrator containing the following symplectic splitting integrators:
Verlet
(order 2, all purpose), also referred to as Strang or Störmer-Verlet splitting- From Forest, Ruth, Physica D 43, 105 (1990):
FR
(order 4, all purpose)
- From Yoshida, Phys. Lett. A 150, 262 (1990):
Yo#
: # should be replaced by an even integer, e.g.,Yo6
for 6th order symplectic integrator (all purpose)Yos6
: (order 6, all purpose) optimized symplectic integrator (solution A from Table 1)
- From McLachlan, SIAM J. Sci. Comp. 16, 151 (1995):
M2
(order 2, all purpose)M4
(order 4, all purpose)
- From Omelyan, Mryglod, Folk, Comput. Phys. Commun. 146, 188 (2002):
EFRL
(order 4) optimized for H = A + BPEFRL
andVEFRL
(order 4) optimized for H = A(p) + B(q). ForPEFRL
, chi should be exp(hA)exp(hB). ForVEFRL
, chi should be exp(hB)exp(hA).
- From Blanes, Moan, J. Comput. Appl. Math. 142, 313 (2002):
BM4
(order 4, all purpose) refers to S6BM6
(order 6, all purpose) refers to S10RKN4b
(order 4) refers to SRKN6b optimized for H = A(p) + B(q). Here chi should be exp(hB)exp(hA).RKN6b
(order 6) refers to SRKN11b optimized for H = A(p) + B(q). Here chi should be exp(hB)exp(hA).RKN6a
(order 6) refers to SRKN14a optimized for H = A(p) + B(q). Here chi should be exp(hA)exp(hB).
- From Blanes, Casas, Farrés, Laskar, Makazaga, Murua, Appl. Numer. Math. 68, 58 (2013):
ABA104
(order (10,4)) optimized for H = A + ε B. Here chi should be exp(hA)exp(hB).ABA864
(order (8,6,4)) optimized for H = A + ε B. Here chi should be exp(hA)exp(hB).ABA1064
(order (10,6,4)) optimized for H = A + ε B. Here chi should be exp(hA)exp(hB).
All purpose integrators are for any splitting of the Hamiltonian H=∑k Xk in any order of the functions Xk. Otherwise, the order of the operators is specified for each integrator.
Usage: integrator = SymplecticIntegrator(name, step) where name is one of the names listed above and step is the time step of the integrator (float).
The function integrator._integrate
integrates the Hamiltonian flow by one step.
The function integrator.integrate
integrates the Hamiltonian flow from the initial conditions specified by the initial state vector y using integrator, one of the selected symplectic splitting integrators. It returns the value of y at times defines by the float, list or array times.
Parameters:
- chi : function of (h, y), y being the state vector. Function returning exp(h Xn)...exp(h X1) y. If the selected integrator is not all purpose, refer to the list above for the specific ordering of the operators. The operator Xk is the Liouville operator associated with the function Xk, i.e., Xk = {Xk, ·}.
- chi_star : function of (h, y). Function returning exp(h X1)...exp(h Xn) y.
- y : initial state vector (numpy array)
- times : times at which the values of the state vector are computed
- command : function of (t, y). Function to be run at each time step (e.g., plotting an observable associated with the state vector, or register specific events).
- autonomous : boolean. If autonomous is False, the state vector y should be of the form y = [t, x], where the first coordinate is time.
Returns:
- If times is a float of integer, the output is a tuple (t, y or x) where y is the value of the state vector and y = [t, x] if autonomous is False.
- If times is a list or array, returns the times and values of y or x at times.
- If times is a list or array with a single element, returns the times and values of y or x at all computed times.
References:
- Hairer, Lubich, Wanner, 2003, Geometric Numerical Integration: Structure-Preserving Algorithms for Ordinary Differential Equations (Springer)
- McLachlan, Tuning symplectic integrators is easy and worthwhile, Commun. Comput. Phys. 31, 987 (2022); arxiv:2104.10269
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.