Page MenuHomec4science
Files F120807831

effective_log_rz.py
No OneTemporary
Actions

File Metadata

Created
Mon, Jul 7, 05:51

effective_log_rz.py
View Options

import numpy as np
import matplotlib.pyplot as plt
import argparse
import pickle as pkl
from poincarepy import PoincareCollection, Tomography, PoincareMapper
import poincarepy.potentials as potentials
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Poincare x vx Section")
# Integration Parameters (Collection parameters)
parser.add_argument("-tmax",type=float,default=200,help="Maximal integration time. If --no_count, this will be the obligatory final time")
parser.add_argument("-N_points",type=int,default= 40,help="Terminate integration after n crossings of the plane (=nb of points in the Poincaré map)")
parser.add_argument("-N_orbits",type=int,default=15,help="Number of orbits to sample if --fill is passed")
# Tomography Parameters
parser.add_argument("-Emin",type=float,default=-1)
parser.add_argument("-Emax",type=float,default=0.)
parser.add_argument("-N_E",type=int,default=20,help="Number of energy slices in tomographic mode")
parser.add_argument("--no_orbit_redraw",action='store_false')
# Script Parameters
parser.add_argument("--progress",action='store_true',help="Use tqdm to show progress bar")
parser.add_argument("-save",type=str,default=None)
parser.add_argument("-open",type=str,default=None)
args = parser.parse_args()
# Run without loading previous pkl file
if args.open is None:
# Define a potential
pot = potentials.EffectiveLogarithmic_cylindrical(rc=0.,q=0.9,Lz=0.2,v0=1)
xlim = [0.001,2] # Range of R's in which to search for true physical min/max R per energy
# Mapper with default parameters for integration time etc
mapper = PoincareMapper(pot,max_integ_time=args.tmax)
# Print some info
print("Generating {:n} Poincare maps in the potential:\n".format(args.N_E))
print(pot.info())
print("from E={:.2f} to E={:.2f}, with {:n} orbits per map and {:n} points (crossings) per orbit.".format(args.Emin,args.Emax,args.N_orbits,args.N_points))
print("Maximum integration time is t_max={:.2f}".format(args.tmax),end="")
if args.save is not None:
print(", output will be saved to " + args.save + ".")
else:
print(", output will not be saved.")
# Create Poincare sections over a range of energies
energies = np.linspace(args.Emin,args.Emax,args.N_E)
sections, orbits, zvcs = mapper.section_collection(energies,xlim,args.N_orbits,args.N_points)
# Create PoincareCollection object
col = PoincareCollection(energies,orbits,sections,zvcs,mapper)
if args.save is not None:
col.save(args.save)
# Run by loading previous pkl file
else:
col = PoincareCollection.load(args.open)
""" Show Results"""
polar_axlabels = ["$r$","$\dot{r}$","$r$","$z$"]
title = r'$\Phi(r,z) = \frac{1}{2} v_0^2 \ln{\left(r_c^2 + r^2 + \frac{z^2}{q^2}\right)} + \frac{L_z^2}{2r^2}$'
tom = Tomography(col,title=title,axlabels=polar_axlabels)

Event Timeline

c4science · Help