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_prescreen():
"""Add an OPDScreen in front of LSST entrance pupil. The OPD that comes out
should be _negative_ the added phase delay by convention.
"""
lsst = batoid.Optic.fromYaml("LSST_r.yaml")
wavelength = 620e-9
z_ref = batoid.zernikeGQ(
lsst, 0, 0, wavelength, rings=10, reference='chief', jmax=37, eps=0.61
)
rng = np.random.default_rng(577)
for i in range(4, 38):
amplitude = rng.uniform(0.1, 0.2)
zern = batoid.Zernike(
np.array([0]*i+[amplitude])*wavelength,
R_outer=4.18, R_inner=0.61*4.18
)
tel = batoid.CompoundOptic(
(
batoid.optic.OPDScreen(
batoid.Plane(),
zern,
name='PS',
obscuration=batoid.ObscNegation(batoid.ObscCircle(5.0)),
coordSys=lsst.stopSurface.coordSys
),
*lsst.items
),
name='PS0',
backDist=lsst.backDist,
pupilSize=lsst.pupilSize,
inMedium=lsst.inMedium,
stopSurface=lsst.stopSurface,
sphereRadius=lsst.sphereRadius,
pupilObscuration=lsst.pupilObscuration
)
do_pickle(tel)
zGQ = batoid.zernikeGQ(
tel, 0, 0, wavelength, rings=10, reference='chief', jmax=37, eps=0.61
)
zTA = batoid.zernikeTA(
tel, 0, 0, wavelength, nrad=10, naz=60, reference='chief', jmax=37, eps=0.61
)
z_expect = np.zeros_like(zGQ)
z_expect[i] = -amplitude # Longer OPL => negative OPD
np.testing.assert_allclose(
(zGQ-z_ref)[4:], z_expect[4:],
rtol=0, atol=5e-4
)
# Distortion makes this comparison less precise
np.testing.assert_allclose(
zGQ[4:], zTA[4:],
rtol=0, atol=5e-3
)
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_zeroscreen():
"""Add a zero phase OPDScreen in front of LSST entrance pupil. Should have
_no_ effect.
"""
lsst = batoid.Optic.fromYaml("LSST_r.yaml")
screens = [
batoid.optic.OPDScreen(
batoid.Plane(),
batoid.Plane(),
name='PS',
coordSys=lsst.stopSurface.coordSys
),
batoid.optic.OPDScreen(
batoid.Paraboloid(100.0),
batoid.Plane(),
name='PS',
coordSys=lsst.stopSurface.coordSys
),
batoid.optic.OPDScreen(
batoid.Quadric(11.0, -0.5),
batoid.Plane(),
name='PS',
coordSys=lsst.stopSurface.coordSys
),
batoid.optic.OPDScreen(
batoid.Zernike([0, 0, 0, 0, 300e-9, 0, 0, 400e-9, -600e-9]),
batoid.Zernike([0]*22),
name='PS',
coordSys=lsst.stopSurface.coordSys
)
]
for screen in screens:
tel = batoid.CompoundOptic(
(screen, *lsst.items),
name='PS0',
backDist=lsst.backDist,
pupilSize=lsst.pupilSize,
inMedium=lsst.inMedium,
stopSurface=lsst.stopSurface,
sphereRadius=lsst.sphereRadius,
pupilObscuration=lsst.pupilObscuration
)
do_pickle(tel)
rng = np.random.default_rng(57)
thx = np.deg2rad(rng.uniform(-1, 1))
thy = np.deg2rad(rng.uniform(-1, 1))
rays = batoid.RayVector.asPolar(
optic=tel, wavelength=620e-9,
theta_x=thx, theta_y=thy,
nrad=5, naz=60
)
tf1 = tel.traceFull(rays)
tf2 = lsst.traceFull(rays)
np.testing.assert_allclose(
tf1['PS']['in'].v,
tf1['PS']['out'].v,
rtol=0, atol=1e-14
)
for key in tf2:
rays_allclose(
tf1[key]['out'],
tf2[key]['out'],
atol=1e-13
)
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])
