def test_ne():
objs = [
batoid.Mirror(batoid.Plane()),
batoid.Detector(batoid.Plane()),
batoid.Baffle(batoid.Plane()),
batoid.RefractiveInterface(batoid.Plane()),
batoid.Mirror(batoid.Paraboloid(0.1)),
batoid.Detector(batoid.Paraboloid(0.1)),
batoid.Baffle(batoid.Paraboloid(0.1)),
batoid.RefractiveInterface(batoid.Paraboloid(0.1)),
batoid.Mirror(batoid.Plane(), obscuration=batoid.ObscCircle(0.1)),
batoid.Mirror(batoid.Plane(), inMedium=batoid.ConstMedium(1.1)),
batoid.Mirror(batoid.Plane(), outMedium=batoid.ConstMedium(1.1)),
batoid.Mirror(batoid.Plane(), coordSys=batoid.CoordSys([0, 0, 1])),
batoid.CompoundOptic(
[batoid.Mirror(batoid.Plane()),
batoid.Mirror(batoid.Plane())]),
batoid.CompoundOptic(
[batoid.Mirror(batoid.Plane()),
batoid.Baffle(batoid.Plane())]),
batoid.CompoundOptic([
batoid.RefractiveInterface(batoid.Plane()),
batoid.RefractiveInterface(batoid.Plane())
]),
batoid.Lens([
batoid.RefractiveInterface(batoid.Plane()),
batoid.RefractiveInterface(batoid.Plane())
]),
]
all_obj_diff(objs)
def test_huygens_paraboloid(plot=False):
if __name__ == '__main__':
obscurations = [0.0, 0.25, 0.5, 0.75]
else:
obscurations = [0.25]
print("Testing HuygensPSF")
# Just do a single parabolic mirror test
focalLength = 1.5
diam = 0.3
R = 2*focalLength
for obscuration in obscurations:
telescope = batoid.CompoundOptic(
items = [
batoid.Mirror(
batoid.Paraboloid(R),
name="Mirror",
obscuration=batoid.ObscNegation(
batoid.ObscAnnulus(0.5*obscuration*diam, 0.5*diam)
)
),
batoid.Detector(
batoid.Plane(),
name="detector",
coordSys=batoid.CoordSys(origin=[0, 0, focalLength])
)
],
pupilSize=diam,
backDist=10.0,
inMedium=batoid.ConstMedium(1.0)
)
airy_size = 1.22*500e-9/diam * 206265
print()
print("Airy radius: {:4.2f} arcsec".format(airy_size))
# Start with the HuygensPSF
npix = 96
size = 3.0 # arcsec
dsize = size/npix
dsize_X = dsize*focalLength/206265 # meters
psf = batoid.huygensPSF(
telescope, 0.0, 0.0, 500e-9,
nx=npix, dx=dsize_X, dy=dsize_X
)
psf.array /= np.max(psf.array)
scale = np.sqrt(np.abs(np.linalg.det(psf.primitiveVectors))) # meters
scale *= 206265/focalLength # arcsec
obj = galsim.Airy(lam=500, diam=diam, obscuration=obscuration)
# Need to shift by half a pixel.
obj = obj.shift(scale/2, scale/2)
im = obj.drawImage(nx=npix, ny=npix, scale=scale, method='no_pixel')
arr = im.array/np.max(im.array)
gs_mom = galsim.hsm.FindAdaptiveMom(im)
psfim = galsim.Image(psf.array)
jt_mom = galsim.hsm.FindAdaptiveMom(psfim)
print("GalSim shape: ", gs_mom.observed_shape)
print("batoid shape: ", jt_mom.observed_shape)
print("GalSim centroid: ", gs_mom.moments_centroid)
print("batoid centroid: ", jt_mom.moments_centroid)
print("GalSim size: ", gs_mom.moments_sigma)
print("batoid size: ", jt_mom.moments_sigma)
print("GalSim rho4: ", gs_mom.moments_rho4)
print("batoid rho4: ", jt_mom.moments_rho4)
np.testing.assert_allclose(
gs_mom.observed_shape.g1,
jt_mom.observed_shape.g1,
rtol=0.0, atol=3e-3
)
np.testing.assert_allclose(
gs_mom.observed_shape.g2,
jt_mom.observed_shape.g2,
rtol=0.0, atol=3e-3
)
np.testing.assert_allclose(
gs_mom.moments_centroid.x,
jt_mom.moments_centroid.x,
rtol=0.0, atol=1e-9
)
np.testing.assert_allclose(
gs_mom.moments_centroid.y,
jt_mom.moments_centroid.y,
rtol=0.0, atol=1e-9
)
np.testing.assert_allclose(
gs_mom.moments_sigma,
jt_mom.moments_sigma,
rtol=1e-2 # why not better?!
)
np.testing.assert_allclose(
gs_mom.moments_rho4,
jt_mom.moments_rho4,
rtol=2e-2
)
if plot:
size = scale*npix
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(15, 4))
ax1 = fig.add_subplot(131)
im1 = ax1.imshow(
np.log10(arr),
extent=np.r_[-1,1,-1,1]*size/2,
vmin=-7, vmax=0
)
plt.colorbar(im1, ax=ax1, label=r'$\log_{10}$ flux')
ax1.set_title('GalSim')
ax1.set_xlabel("arcsec")
ax1.set_ylabel("arcsec")
sizeX = dsize_X * npix * 1e6 # microns
ax2 = fig.add_subplot(132)
im2 = ax2.imshow(
np.log10(psf.array),
extent=np.r_[-1,1,-1,1]*sizeX/2,
vmin=-7, vmax=0
)
plt.colorbar(im2, ax=ax2, label=r'$\log_{10}$ flux')
ax2.set_title('batoid')
ax2.set_xlabel(r"$\mu m$")
ax2.set_ylabel(r"$\mu m$")
ax3 = fig.add_subplot(133)
im3 = ax3.imshow(
(psf.array-arr)/np.max(arr),
vmin=-0.01, vmax=0.01,
cmap='seismic'
)
plt.colorbar(im3, ax=ax3, label="(batoid-GalSim)/max(GalSim)")
ax3.set_title('resid')
ax3.set_xlabel(r"$\mu m$")
ax3.set_ylabel(r"$\mu m$")
fig.tight_layout()
plt.show()
def test_traceSplit_simple():
telescope = batoid.CompoundOptic(
name = "simple",
stopSurface=batoid.Interface(batoid.Plane()),
items = [
batoid.Lens(
name = "L1",
items = [
batoid.RefractiveInterface(
batoid.Plane(),
name = "L1_entrance",
coordSys=batoid.CoordSys(origin=[0,0,0.3]),
inMedium=batoid.ConstMedium(1.0),
outMedium=batoid.ConstMedium(1.1)
),
batoid.RefractiveInterface(
batoid.Plane(),
name = "L1_exit",
coordSys=batoid.CoordSys(origin=[0,0,0.2]),
inMedium=batoid.ConstMedium(1.1),
outMedium=batoid.ConstMedium(1.0)
)
]
),
batoid.Mirror(
batoid.Plane(),
name="Mirror"
),
batoid.Detector(
batoid.Plane(),
name="detector",
coordSys=batoid.CoordSys(origin=[0, 0, 0.1])
)
],
pupilSize=1.0,
backDist=1.0,
inMedium=batoid.ConstMedium(1.0)
)
telescope['L1_entrance'].forwardCoating = batoid.SimpleCoating(0.02, 0.98)
telescope['L1_entrance'].reverseCoating = batoid.SimpleCoating(0.02, 0.98)
telescope['L1_exit'].forwardCoating = batoid.SimpleCoating(0.02, 0.98)
telescope['L1_exit'].reverseCoating = batoid.SimpleCoating(0.02, 0.98)
telescope['detector'].forwardCoating = batoid.SimpleCoating(0.02, 0.98)
rays = batoid.RayVector.asPolar(
telescope,
wavelength=500e-9,
theta_x=np.deg2rad(1.0),
theta_y=0.0,
nrad=10, naz=60
)
rForward, rReverse = telescope.traceSplit(rays.copy(), minFlux=1e-4)
for r in rForward:
r2 = telescope.trace(rays.copy(), path=r.path)
w = ~r2.vignetted
np.testing.assert_allclose(r.r, r2.r[w])
np.testing.assert_allclose(r.v, r2.v[w])
np.testing.assert_allclose(r.t, r2.t[w])
tf = telescope.traceFull(rays.copy(), path=r.path)
keys = []
for item in r.path:
j = 0
key = f"{item}_{j}"
while key in keys:
j += 1
key = f"{item}_{j}"
keys.append(key)
assert keys == [k for k in tf.keys()]
r3 = tf[keys[-1]]['out']
w = ~r3.vignetted
np.testing.assert_allclose(r.r, r3.r[w])
np.testing.assert_allclose(r.v, r3.v[w])
np.testing.assert_allclose(r.t, r3.t[w])
for r in rReverse:
r2 = telescope.trace(rays.copy(), path=r.path)
w = ~r2.vignetted
np.testing.assert_allclose(r.r, r2.r[w])
np.testing.assert_allclose(r.v, r2.v[w])
np.testing.assert_allclose(r.t, r2.t[w])
tf = telescope.traceFull(rays.copy(), path=r.path)
keys = []
for item in r.path:
j = 0
key = f"{item}_{j}"
while key in keys:
j += 1
key = f"{item}_{j}"
keys.append(key)
assert keys == [k for k in tf.keys()]
r3 = tf[keys[-1]]['out']
w = ~r3.vignetted
np.testing.assert_allclose(r.r, r3.r[w])
np.testing.assert_allclose(r.v, r3.v[w])
np.testing.assert_allclose(r.t, r3.t[w])