from scipy.constants import c as clight, e as qe import matplotlib.pyplot as plt import numpy as np import PyECLOUD.PyEC4PyHT as PyEC4PyHT from PyHEADTAIL.particles.slicing import UniformBinSlicer import PyECLOUD.mystyle as ms from LHC_custom import LHC machine_configuration = 'HLLHC-injection' machine = LHC(n_segments=1, machine_configuration=machine_configuration) bunch = machine.generate_6D_Gaussian_bunch(n_macroparticles=300000, intensity=1.15e11, epsn_x=2.5e-6, epsn_y=2.5e-6, sigma_z=0.11) bunch.x[bunch.z < 5e-2] += 1e-3 ecloud_ele = PyEC4PyHT.Ecloud( slice_by_slice_mode=True, L_ecloud=1., slicer=None, Dt_ref=25e-12, pyecl_input_folder='pyecloud_config', ) n_slices = 150 z_cut = 2.5e-9 / 2 * clight