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
|
# -*- coding: utf-8 -*-
__author__ = "Konstantin Klementiev"
__date__ = "20 Jun 2022"
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.path import Path as mplPath
import matplotlib.patches as patches
import sys
import os, sys; sys.path.append(os.path.join('..', '..')) # analysis:ignore
import xrt.backends.raycing.apertures as rapts
ap = rapts.GridAperture(dx=0.1, dz=0.1, px=0.15, pz=0.15, nx=7, nz=7)
# ap = rapts.SiemensStar(nSpokes=24, r=0.1)
# ap = rapts.SiemensStar(nSpokes=9, rx=0.1, rz=0.08, vortex=2, vortexNradial=9)
def main():
vertices = ap.opening
ntest = 75
xtest = np.linspace(*ap.limOptX, ntest)
ztest = np.linspace(*ap.limOptY, ntest)
xtestm, ztestm = np.meshgrid(xtest, ztest)
xtestm = xtestm.ravel()
ztestm = ztestm.ravel()
footprint = mplPath(vertices)
badIndices = np.invert(footprint.contains_points(np.array(
list(zip(xtestm, ztestm)))))
color = ['r' if ind else 'g' for ind in badIndices]
fig, ax = plt.subplots()
ax.set_aspect(1)
ax.set_xlabel('x (mm)')
ax.set_ylabel('z (mm)')
patch = patches.PathPatch(footprint, facecolor='orange', lw=2)
ax.add_patch(patch)
ax.scatter(xtestm, ztestm, s=2, c=color)
ax.set_xlim(*ap.limOptX)
ax.set_ylim(*ap.limOptY)
plt.show()
if __name__ == '__main__':
main()
|
