1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
# -*- coding: utf-8 -*-
__author__ = "Konstantin Klementiev", "Roman Chernikov"
__date__ = "08 Mar 2016"
import os, sys; sys.path.append(os.path.join('..', '..', '..')) # analysis:ignore
import numpy as np
import BalderDMM
import xrt.plotter as xrtp
import xrt.runner as xrtr
#import xrt.backends.raycing.materials as rm
stripe = 'Si'
E0 = 8000
dE = 1200
def define_plots(beamLine):
plots = []
plot = xrtp.XYCPlot(
'beamFSMDCM', (1,),
xaxis=xrtp.XYCAxis(r'$x$', 'mm'), yaxis=xrtp.XYCAxis(r'$z$', 'mm'),
caxis=xrtp.XYCAxis('energy', 'keV', fwhmFormatStr='%.2f'), title='DCM')
plot.xaxis.limits = [-7., 7.]
plot.yaxis.limits = [20.3-7., 20.3+7.]
plot.fluxFormatStr = '%.1p'
plot.textPanel = plot.fig.text(0.88, 0.8, '',
transform=plot.fig.transFigure, size=14,
color='r', ha='center')
plot.baseName = 'afterDMM'
plots.append(plot)
plot = xrtp.XYCPlot(
'beamDCMlocal1', (1,),
xaxis=xrtp.XYCAxis(r'$x$', 'mm'), yaxis=xrtp.XYCAxis(r'$y$', 'mm'),
caxis=xrtp.XYCAxis('energy', 'keV', fwhmFormatStr='%.2f'),
title='Xtal1 local')
plot.xaxis.limits = [-86., 86.]
plot.yaxis.limits = [-86., 86.]
plot.fluxFormatStr = '%.1p'
plot.textPanel = plot.fig.text(0.88, 0.8, '',
transform=plot.fig.transFigure, size=14,
color='r', ha='center')
plot.baseName = '1stML'
plots.append(plot)
for plot in plots:
plot.caxis.limits = [(E0 - dE)*1e-3, (E0 + dE)*1e-3]
plot.caxis.offset = E0*1e-3
return plots
def plot_generator(plots, beamLine):
energies = np.linspace(E0 - 500, E0 + 500, 7)
# energies = E0,
for energy in energies:
BalderDMM.align_beamline(beamLine, energy=energy)
thetaDeg = np.degrees(
beamLine.dmm.bragg - 2*beamLine.vcm.pitch)
for plot in plots:
baseName = '{0}_{1:05.0f}'.format(plot.baseName, thetaDeg*1e4)
plot.saveName = baseName + '.png'
# plot.persistentName = baseName + '.pickle'
if hasattr(plot, 'textPanel'):
plot.textPanel.set_text(
'$\\theta$ = {0:.3f}$^o$'.format(thetaDeg))
if BalderDMM.showIn3D:
beamLine.glowFrameName = 'DMM_{0:05.0f}.jpg'.format(thetaDeg*1e4)
yield
def main():
myBalder = BalderDMM.build_beamline(
stripe=stripe, eMinRays=E0-dE, eMaxRays=E0+dE)
if BalderDMM.showIn3D:
myBalder.glow(scale=[500, 10, 500], centerAt='VCM', startFrom=1,
generator=plot_generator, generatorArgs=[[], myBalder])
return
plots = define_plots(myBalder)
xrtr.run_ray_tracing(plots, repeats=120, generator=plot_generator,
beamLine=myBalder,
globalNorm=True,
processes=1)
#this is necessary to use multiprocessing in Windows, otherwise the new Python
#contexts cannot be initialized:
if __name__ == '__main__':
main()
|
