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
90
|
"""
Example how to export a GenX sample model to the simple model language.
Requires genx3 or genx3server (no WX) to be installed:
pip install genx3server
If ran without installing orsopy, remember to put folder in PYTHONPATH.
Be aware of other orsopy version that might be installed by pip.
To be sure run: pip uninstall orsopy
"""
import os
import sys
from genx import api
# improve load time by ignoring numba
from genx.models import lib as gmlib
from orsopy.fileio import model_language as ml
gmlib.USE_NUMBA = False
def get_materialn(li):
sld = complex(li.b * li.dens) * 1e-6
return ml.Material(sld=ml.ComplexValue(sld.real, sld.imag))
def get_materialx(li):
sld = complex(li.f * li.dens) * 1e-6
return ml.Material(sld=ml.ComplexValue(sld.real, sld.imag))
get_material = get_materialn
def get_layer(li):
return ml.Layer(thickness=li.d, material=get_material(li), roughness=li.sigma)
def main():
fname = sys.argv[1]
model, optimizer = api.load(fname)
model.compile_script()
sm = model.script_module
global get_material
if sm.inst.probe == "x-ray":
get_material = get_materialx
names = list(sm.__dict__.keys())
objects = list(sm.__dict__.values())
defaults = ml.ModelParameters(length_unit="angstrom", sld_unit="1/angstrom^2")
layers = {}
materials = {}
materials["ambient"] = get_material(sm.Amb)
materials["substrate"] = get_material(sm.Sub)
sub_stacks = {}
stack_order = ["ambient"]
for si in reversed(sm.sample.Stacks):
if len(si.Layers) == 0:
continue
ni = names[objects.index(si)]
stack_order.append(ni)
layer_order = []
for lj in reversed(si.Layers):
nj = names[objects.index(lj)]
layers[nj] = get_layer(lj)
layer_order.append(nj)
sub_stacks[ni] = ml.SubStack(repetitions=si.Repetitions, stack=" | ".join(layer_order))
stack_order.append("substrate")
stack_str = " | ".join(stack_order)
sample = ml.SampleModel(
stack=stack_str,
origin=f'GenX model "{os.path.basename(fname)}"',
sub_stacks=sub_stacks,
layers=layers,
materials=materials,
globals=defaults,
)
print(sample.to_yaml())
if len(sys.argv) > 2:
open(sys.argv[2], "w").write(sample.to_yaml())
if __name__ == "__main__":
main()
|
