__author__ = 'swebb'
from radtrack.genesis.RbGenLattice import GenLattice
from matplotlib import pyplot as plt
import matplotlib as mpl
mpl.rc('text', usetex=True)
font = {'family' : 'normal',
'weight' : 'bold',
'size' : 18}
mpl.rc('font', **font)
import numpy as np
# A doublet cell
my_lattice = GenLattice('doublet_lattice.lat')
f = 1.1 #meters
lundulator = 2.27e-2
lquad = 2*lundulator
print 'lquad =', lquad
kq = 1/(f*lquad)
print 'kq =', kq
ldrift1 = 10*lundulator
print 'l =', ldrift1
ldrift2 = 90*lundulator
print 'L =', ldrift2
n_cells = 1
from radtrack.genesis.RbGenLattice import GenLattice
from matplotlib import pyplot as plt
import matplotlib as mpl
mpl.rc('text', usetex=True)
font = {'family' : 'normal',
'weight' : 'bold',
'size' : 18}
mpl.rc('font', **font)
import numpy as np
from scipy import constants as consts
from radtrack.genesis.rbExecuteGenesis import rbExecuteGenesis
# A doublet cell
my_lattice = GenLattice('ttf.lat')
gamma0 = 1956.947
emittance = 2.000000e-6 #2 mm-mrad
l_undulator = 2.73e-2
dBdx_quad = 37. #T/m
k_quadrupole = dBdx_quad*consts.e/(consts.m_e*gamma0*consts.c)
aw = 0.896
l_quad = 3*l_undulator
l_drift1 = 16*l_undulator
l_drift2 = 2*l_undulator
l_und_segment = 163*l_undulator
und_spacing = 26*l_undulator
n_cells = 6
pos = 0.
