diff --git a/boosterubung.py b/boosterubung.py index 2920b884044e1313d79c81d12f887587836e00b4..720fe81ce5f47d737ae26245b39fc0b1b613f802 100755 --- a/boosterubung.py +++ b/boosterubung.py @@ -6,7 +6,12 @@ import matplotlib.pylab as mp ring=at.load_lattice("../../machine/lattice/elettra/booster.mat",energy=2.4e9) refpts = range(len(ring) + 1) -elemdata0, ringdata,elemdata=at.linopt6(ring,refpts) + +#elemdata0, ringdata,elemdata=at.linopt6(ring,refpts) + +#--- lionop needs radiation off +ring.radiation_off() +elemdata0, tune, chrom, elemdata=at.linopt(ring,0.0001,list(refpts)) s_pos = elemdata['s_pos'] beta_x =elemdata['beta'][:,0] @@ -26,8 +31,8 @@ closed_orbit_xp=elemdata['closed_orbit'][:,1] closed_orbit_y=elemdata['closed_orbit'][:,2] closed_orbit_yp=elemdata['closed_orbit'][:,3] -[tuneH,tuneV,tuneL]=ring.get_tune() -[chromH,chromV,chromL]=ring.get_chrom() +#[tuneH,tuneV,tuneL]=ring.get_tune() +#[chromH,chromV,chromL]=ring.get_chrom() e_eV=ring.energy diff --git a/elettratest.py b/elettratest.py new file mode 100755 index 0000000000000000000000000000000000000000..3a23f635985a903ed93a883d4ff1d7ab145f82b4 --- /dev/null +++ b/elettratest.py @@ -0,0 +1,56 @@ +#!/usr/bin/python3 + +import at +import matplotlib.pylab as mp + +ring=at.load_lattice("../../machine/lattice/elettra/elettra.m",energy=2.0e9) + +refpts = range(len(ring) + 1) + +elemdata0, ringdata,elemdata=at.linopt6(ring,refpts) + +#--- lionop needs radiation off +#ring.radiation_off() +#elemdata0, tune, chrom, elemdata=at.linopt(ring,0.0001,list(refpts)) + +s_pos = elemdata['s_pos'] +beta_x =elemdata['beta'][:,0] +beta_y =elemdata['beta'][:,1] + +alpha_x = elemdata['alpha'][:,0] +alpha_y = elemdata['alpha'][:,1] + +dispersion_x=elemdata['dispersion'][:,0] +dispersion_xp=elemdata['dispersion'][:,1] +dispersion_y=elemdata['dispersion'][:,2] +dispersion_yp=elemdata['dispersion'][:,3] + + +closed_orbit_x=elemdata['closed_orbit'][:,0] +closed_orbit_xp=elemdata['closed_orbit'][:,1] +closed_orbit_y=elemdata['closed_orbit'][:,2] +closed_orbit_yp=elemdata['closed_orbit'][:,3] + +#[tuneH,tuneV,tuneL]=ring.get_tune() +#[chromH,chromV,chromL]=ring.get_chrom() +e_eV=ring.energy + + +mp.figure() +mp.subplot(211) +mp.plot(s_pos,beta_x) +mp.plot(s_pos,beta_y) +mp.plot(s_pos,dispersion_x) +mp.xlabel('s [m]') +mp.ylabel('m') +mp.title('beta_x beta_y disp_x' ) + + +mp.subplot(212) +mp.plot(s_pos,closed_orbit_x) +mp.xlabel('s [m]') +mp.ylabel('m') +mp.title('orb_x' ) +mp.show() +print('tune H:',tuneH,'tune V:',tuneV,'tune L:',tuneL) +print('E:', e_eV/1e9,' GeV')