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
|
#!/usr/bin/env python3
"""
Specular simulation with footprint correction for a square beam
"""
import bornagain as ba
from bornagain import angstrom, ba_plot as bp, deg, std_samples
sample = std_samples.alternating_layers()
def simulate(footprint, title):
n = <%= test_mode ? 50 : 500 %>
scan = ba.AlphaScan(n, 0.6*deg/n, 0.6*deg)
scan.setWavelength(1.54*angstrom)
scan.setFootprint(footprint)
simulation = ba.SpecularSimulation(scan, sample)
result = simulation.simulate()
result.setTitle(title)
return result
if __name__ == '__main__':
beam_sample_ratio = 0.01 # beam-to-sample size ratio
results = [
simulate(ba.FootprintSquare(beam_sample_ratio), "With footprint"),
simulate(None, "Without footprint"),
]
<%- if test_mode -%>
from bornagain import ba_check
ba_check.persistence_test(results)
<%- elsif figure_mode -%>
plotargs = bp.parse_commandline()
plotargs['legendloc'] = 'lower left'
bp.plot_multicurve(results, **plotargs)
bp.export(**plotargs)
<%- else -%>
bp.plot_multicurve(results, legendloc='lower left')
bp.plt.show()
<%- end -%>
|
