esbo_ds/ESBO-ETC
esbo_ds/ESBO-ETC
11
0
Fork 1
Code Issues 1 Pull Requests Releases 1 Wiki Activity

Rasterize circle on given grid

Browse Source
This commit is contained in:
Lukas Klass 2020-05-12 09:15:40 +02:00
parent a78c882227
commit 766a0872cc
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
esbo_etc/classes/psf/Zemax.py
View File

@ -142,7 +142,8 @@ class Zemax(IPSF):
total = np.sum(psf) total = np.sum(psf)
# Iterate the optimal radius for the contained energy # Iterate the optimal radius for the contained energy
r = bisect(lambda r_c: contained_energy.value - np.sum( r = bisect(lambda r_c: contained_energy.value - np.sum(
psf * rasterizeCircle(psf.shape[0], r_c, center_point[0], center_point[1])) / total, 0, r_max, xtol=1e-1) psf * rasterizeCircle(np.zeros((psf.shape[0], psf.shape[0])), r_c, center_point[0],
center_point[1])) / total, 0, r_max, xtol=1e-1)
# Calculate the reduced observation angle in lambda / d_ap # Calculate the reduced observation angle in lambda / d_ap
# noinspection PyTypeChecker # noinspection PyTypeChecker
reduced_observation_angle = r / psf_osf * self.__grid_delta[0] / ( reduced_observation_angle = r / psf_osf * self.__grid_delta[0] / (

11
esbo_etc/lib/helpers.py
View File

@ -42,14 +42,14 @@ def isLambda(obj: object):
return isinstance(obj, type(lambda: None)) and obj.__name__ == (lambda: None).__name__ return isinstance(obj, type(lambda: None)) and obj.__name__ == (lambda: None).__name__
def rasterizeCircle(n: int, radius: float, xc: float, yc: float): def rasterizeCircle(grid: np.ndarray, radius: float, xc: float, yc: float):
""" """
Map a circle on a rectangular grid. Map a circle on a rectangular grid.
Parameters Parameters
---------- ----------
n : int grid : ndarray
Size of the rectangular grid to map the circle on. The grid to map the circle onto.
radius : float radius : float
Radius of the circle to be mapped. Radius of the circle to be mapped.
xc : float xc : float
@ -62,7 +62,6 @@ def rasterizeCircle(n: int, radius: float, xc: float, yc: float):
grid: ndarray grid: ndarray
The grid with the circle mapped onto. Each point contained within the circle is marked as 1. The grid with the circle mapped onto. Each point contained within the circle is marked as 1.
""" """
grid = np.zeros((n, n)) # Initialize an empty grid
xc_pix = int(round(xc)) # X center in pixel coordinates xc_pix = int(round(xc)) # X center in pixel coordinates
x_shift = xc_pix - xc # X shift of the circle center x_shift = xc_pix - xc # X shift of the circle center
yc_pix = int(round(yc)) # Y center in pixel coordinates yc_pix = int(round(yc)) # Y center in pixel coordinates
@ -73,9 +72,9 @@ def rasterizeCircle(n: int, radius: float, xc: float, yc: float):
grid[yc_pix, xc_pix] = 1 # Set the center pixel by default grid[yc_pix, xc_pix] = 1 # Set the center pixel by default
# Create meshgrid for the x and y range of the circle # Create meshgrid for the x and y range of the circle
dx, dy = np.meshgrid(range(- radius_pix if xc_pix - radius_pix >= 0 else - xc_pix, dx, dy = np.meshgrid(range(- radius_pix if xc_pix - radius_pix >= 0 else - xc_pix,
radius_pix + 1 if n > (xc_pix + radius_pix + 1) else n - xc_pix), radius_pix + 1 if grid.shape[1] > (xc_pix + radius_pix + 1) else grid.shape[1] - xc_pix),
range(- radius_pix if yc_pix - radius_pix >= 0 else - yc_pix, range(- radius_pix if yc_pix - radius_pix >= 0 else - yc_pix,
radius_pix + 1 if n > (yc_pix + radius_pix + 1) else n - yc_pix)) radius_pix + 1 if grid.shape[0] > (yc_pix + radius_pix + 1) else grid.shape[0] - yc_pix))
dx2 = (dx + x_shift) ** 2 # Square of the x-component of the current pixels radius dx2 = (dx + x_shift) ** 2 # Square of the x-component of the current pixels radius
dx_side2 = (dx + x_shift + ((dx < 0) - 0.5)) ** 2 # Square of the x-component of the neighbouring pixels radius dx_side2 = (dx + x_shift + ((dx < 0) - 0.5)) ** 2 # Square of the x-component of the neighbouring pixels radius
dy2 = (dy + y_shift) ** 2 # Square of the y-component of the current pixels radius dy2 = (dy + y_shift) ** 2 # Square of the y-component of the current pixels radius