def build_beamline(): # for test=True
beamLine = raycing.BeamLine()
gsource = rs.GeometricSource(beamLine,
'GeometricSource', (0, 0, 0),
nrays=nrays,
distx='annulus',
distxprime=None,
distzprime=None,
distE='lines',
energies=[E_FZP],
polarization='h')
beamLine.fzp = roe.NormalFZP(beamLine,
'FZP', [0, p, 0],
pitch=np.pi / 2,
material=mGold,
f=f,
E=E_FZP,
thinnestZone=thinnestZone,
isCentralZoneBlack=True)
gsource.dx = (0, beamLine.fzp.rn[-1])
beamLine.fzp.area = np.pi * beamLine.fzp.rn[-1]**2 / 2
print('FZP radius=', beamLine.fzp.rn[-1])
r0max = beamLine.fzp.rn[-1] * 5
rNmax = thinnestZone * 5
beamLine.yglo = np.repeat(
[10000 if i == 0 else f / i for i in range(maxOrder + 1)], Nr)
beamLine.dr = np.array([
r0max / (Nr - 1) if i == 0 else rNmax / (Nr - 1)
for i in range(maxOrder + 1)
])
beamLine.zglo = (np.arange(Nr) * beamLine.dr[:, np.newaxis]).flatten()
beamLine.xglo = np.zeros_like(beamLine.yglo)
return beamLine
def build_beamline(azimuth=0):
beamLine = raycing.BeamLine(azimuth=azimuth, height=0)
beamLine.source = rs.Undulator(
beamLine, 'Softi53', nrays=nrays,
eE=3.0, eI=0.5, eEspread=0.0008*energySpreadFactor,
eEpsilonX=0.263*emittanceFactor, eEpsilonZ=0.008*emittanceFactor,
betaX=9., betaZ=2.,
period=48., n=81, targetE=(E0, targetHarmonic),
eMin=E0-dE*vFactor, eMax=E0+dE*vFactor,
xPrimeMax=acceptanceHor/2*1e3*hShrink,
zPrimeMax=acceptanceVer/2*1e3*vShrink,
xPrimeMaxAutoReduce=False, zPrimeMaxAutoReduce=False,
targetOpenCL='CPU',
uniformRayDensity=True,
filamentBeam=(what != 'rays'))
opening = [-acceptanceHor*pFE/2*hShrink, acceptanceHor*pFE/2*hShrink,
-acceptanceVer*pFE/2*vShrink, acceptanceVer*pFE/2*vShrink]
beamLine.slitFE = ra.RectangularAperture(
beamLine, 'FE slit', kind=['left', 'right', 'bottom', 'top'],
opening=opening)
beamLine.fsm0 = rsc.Screen(beamLine, 'FSM-M1')
xShrink = vShrink if what == 'wave' else 1
yShrink = hShrink if what == 'wave' else 1
beamLine.m1 = roe.ToroidMirror(
beamLine, 'M1', surface=('Au',), material=(material,),
limPhysX=(-5*xShrink, 5*xShrink), limPhysY=(-150*yShrink, 150*yShrink),
positionRoll=np.pi/2, R=1e22,
alarmLevel=0.1)
# beamLine.fsm1 = rsc.Screen(beamLine, 'FSM-M1')
xShrink = hShrink if what == 'wave' else 1
yShrink = vShrink if what == 'wave' else 1
beamLine.m2 = roe.OE(
beamLine, 'M2', surface=('Au',), material=(material,),
limPhysX=(-5*xShrink, 5*xShrink), limPhysY=(-225*yShrink, 225*yShrink),
alarmLevel=0.1)
gratingKW = dict(
positionRoll=np.pi, limPhysX=(-2*xShrink, 2*xShrink),
limPhysY=(-40*yShrink, 40*yShrink), alarmLevel=0.1)
if what == 'rays':
beamLine.pg = Grating(beamLine, 'PlaneGrating',
material=gratingMaterial, **gratingKW)
beamLine.pg.material.efficiency = [(1, 0.4)]
else:
beamLine.pg = roe.BlazedGrating(
beamLine, 'BlazedGrating', material=material, blaze=blaze,
rho=rho, **gratingKW)
beamLine.pg.order = 1
# beamLine.fsmPG = rsc.Screen(beamLine, 'FSM-PG')
beamLine.m3 = roe.ToroidMirror(
beamLine, 'M3', surface=('Au',), material=(material,),
positionRoll=-np.pi/2, limPhysX=(-2*vShrink, 2*vShrink),
limPhysY=(-80*hShrink*hExtraFactor, 80*hShrink*hExtraFactor),
#limPhysY=(-80*hShrink, 80*hShrink),
alarmLevel=0.1)
beamLine.fsm3 = rsc.Screen(beamLine, 'FSM-M3')
# beamLine.exitSlit = ra.RoundAperture(
# beamLine, 'ExitSlit', r=ESradius, alarmLevel=None)
beamLine.exitSlit = ra.RectangularAperture(
beamLine, 'ExitSlit',
opening=[-ESdX/2*hFactor*hShrink, ESdX/2*hFactor*hShrink,
-ESdZ/2*vFactor*vShrink, ESdZ/2*vFactor*vShrink])
beamLine.fzp = roe.NormalFZP(
beamLine, 'FZP', pitch=np.pi/2, material=mAuFZP, f=focus, E=E0,
N=Nzone)
beamLine.fzp.order = 1
beamLine.fzp.material.efficiency = [(1, 0.1)] # used with rays
print('outerzone = {0} mm'.format(beamLine.fzp.rn[-1]-beamLine.fzp.rn[-2]))
beamLine.fzp.area = np.pi * beamLine.fzp.rn[-1]**2 / 2
beamLine.fsmFZP = rsc.Screen(beamLine, 'FSM-FZP')
beamLine.fsmExp = rsc.Screen(beamLine, 'FSM-Exp')
return beamLine