Closed bwheelz36 closed 1 year ago
from pathlib import Path import sys sys.path.append('../') # not necessary when the library is installed from ParticlePhaseSpace import DataLoaders from ParticlePhaseSpace import PhaseSpace test_data_loc = Path(r'../tests/test_data/coll_PhaseSpace_xAng_0.00_yAng_0.00_angular_error_0.0.phsp').absolute() ps_data = DataLoaders.Load_TopasData(test_data_loc) PS = PhaseSpace(ps_data) PS.plot.particle_positions_hist_2D(xlim=[-5, 5], ylim=[-5,5], grid=True, log_scale=True) PS.plot.transverse_trace_space_hist_2D(xlim=[-2,2], ylim=[-1, 1], plot_twiss_ellipse=True, grid=False) PS.plot.energy_hist_1D(grid=True)