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
|
#!/usr/bin/env python3
"""
An example of taking into account beam angular and wavelength
divergence in reflectometry calculations with BornAgain.
"""
import bornagain as ba
from bornagain import angstrom, ba_plot as bp, deg, std_samples
# input parameters
wavelength = 1.54*angstrom
# convolution parameters
d_wl = 0.01*wavelength # spread width for wavelength
d_ang = 0.01*deg # spread width for incident angle
def get_sample():
return std_samples.alternating_layers()
def get_simulation(sample):
"""
A specular simulation with beam and detector defined.
"""
n = <%= test_mode ? 50 : 500 %>
scan = ba.AlphaScan(n, 2*deg/n, 2*deg)
scan.setWavelength(wavelength)
alpha_distr = ba.DistributionGaussian(0, d_ang, 25, 3.)
scan.setGrazingAngleDistribution(alpha_distr)
wavelength_distr = ba.DistributionGaussian(0, d_wl, 25, 3.)
scan.setWavelengthDistribution(wavelength_distr)
return ba.SpecularSimulation(scan, sample)
return simulation
if __name__ == '__main__':
sample = get_sample()
simulation = get_simulation(sample)
result = simulation.simulate()
<%- if test_mode -%>
from bornagain import ba_check
ba_check.persistence_test(result)
<%- elsif figure_mode -%>
plotargs = bp.parse_commandline()
bp.plot_datafield(result, **plotargs)
bp.export(**plotargs)
<%- else -%>
bp.plot_datafield(result)
bp.plt.show()
<%- end -%>
|
