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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
using CrystFEL
using GLMakie
function readmillepede(filename)
motions = let fh = open(filename, "r")
readline(fh) # discard header line
motions = map(eachline(fh)) do line
sgroupser,sdelta,_ = split(line)
(parse(Int,sgroupser), parse(Float64,sdelta))
end
motions
end
end
function translate!(panel::DetGeomPanel, vec)
panel.corner += vec/panel.pixel_pitch
end
function translate!(group::DetGeomGroup, vec)
for child in group.children
translate!(child, vec)
end
end
function rotate2d!(p, c, ang)
nx = c[1] + (p[1]-c[1])*cos(ang) - (p[2]-c[2])*sin(ang);
ny = c[2] + (p[1]-c[1])*sin(ang) + (p[2]-c[2])*cos(ang);
p[1] = nx
p[2] = ny
end
function rotatex!(panel::DetGeomPanel, center, angle)
rotate2d!(view(panel.corner, 2:3), center[2:3]/panel.pixel_pitch, angle);
rotate2d!(view(panel.fsvec, 2:3), [0,0], angle);
rotate2d!(view(panel.ssvec, 2:3), [0,0], angle);
end
function rotatey!(panel::DetGeomPanel, center, angle)
rotate2d!(view(panel.corner, 3:-2:1), center[3:-2:1]/panel.pixel_pitch, angle);
rotate2d!(view(panel.fsvec, 3:-2:1), [0,0], angle);
rotate2d!(view(panel.ssvec, 3:-2:1), [0,0], angle);
end
function rotatez!(panel::DetGeomPanel, center, angle)
rotate2d!(view(panel.corner, 1:2), center[1:2]/panel.pixel_pitch, angle);
rotate2d!(view(panel.fsvec, 1:2), [0,0], angle);
rotate2d!(view(panel.ssvec, 1:2), [0,0], angle);
end
function rotatex!(group::DetGeomGroup, center, angle)
for child in group.children
rotatex!(child, center, angle)
end
end
function rotatey!(group::DetGeomGroup, center, angle)
for child in group.children
rotatey!(child, center, angle)
end
end
function rotatez!(group::DetGeomGroup, center, angle)
for child in group.children
rotatez!(child, center, angle)
end
end
function plotgeom!(dtempl, motions)
f = Figure(size=(1024,1024))
ax = Axis3(f[1,1], aspect=:equal, limits=(nothing,nothing,nothing,nothing,0,0.3),
xlabel="x (meters)", ylabel="y (meters)", zlabel="z (meters)")
sl = Slider(f[2,1], range=0:0.01:1)
btoggle = Toggle(f, halign=:left, tellwidth=false)
f[3,1] = grid!([1,1]=>Label(f, "Show beams", halign=:right, tellwidth=false), [1,2]=>btoggle)
cfimage = CrystFEL.Image(dtempl)
triangles = lift(sl.value) do trscale
dgmoved = cfimage.detgeom # Makes a fresh copy of underlying C structure
# Apply all `motions` in the same way as align_detector
for m in motions
param = m[1] % 100
group = findgroup(dgmoved, m[1]-param)
if param == 1
translate!(group, trscale*[-m[2],0,0])
elseif param == 2
translate!(group, trscale*[0,-m[2],0])
elseif param == 3
translate!(group, trscale*[0,0,-m[2]])
elseif param == 4
rotatex!(group, group.center, trscale*-m[2])
elseif param == 5
rotatey!(group, group.center, trscale*-m[2])
elseif param == 6
rotatez!(group, group.center, trscale*-m[2])
end
end
# Generate triangle mesh from panel descriptions
points = Point3f[]
for panel in dgmoved.panels
corner = panel.pixel_pitch*panel.corner
fs = panel.pixel_pitch*panel.w*panel.fsvec
ss = panel.pixel_pitch*panel.h*panel.ssvec
push!(points, corner)
push!(points, corner+fs)
push!(points, corner+fs+ss)
push!(points, corner+fs+ss)
push!(points, corner+ss)
push!(points, corner)
end
return points
end
beams = map(cfimage.detgeom.panels) do p
cen = (p.corner+p.fsvec*p.w/2+p.ssvec*p.h/2)*1.5*p.pixel_pitch
((0,0,0), Tuple(cen))
end
mesh!(triangles, color="#7f7f7f")
beamv = lines!(collect(Iterators.flatten(beams[1:4:end])), color="#008080")
connect!(beamv.visible, btoggle.active)
return f
end
|
