Avoid some dividing by zero in bad corner cases.

schlafly · May 9, 2022 · d0bb2eb · d0bb2eb
1 parent 750e2f7
commit d0bb2eb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/crowdsource/crowdsource_base.py b/crowdsource/crowdsource_base.py
@@ -503,7 +503,8 @@ def compute_centroids(x, y, psflist, flux, im, resid, weight,
     # stamps: 0: neighbor-subtracted images,
     # 1: images,
     # 2: psfs with shifts
-    # 3: psfs without shifts
+    # 3: weights
+    # 4: psfs without shifts
     res = (xcen, ycen, (modelst+residst, imst, modelst, weightst, psfst))
     return res
 
@@ -943,9 +944,9 @@ def compute_stats(xs, ys, impsfstack, psfstack, weightstack, imstack, flux):
     fracfluxd = fracfluxd + (fracfluxd == 0)*1e-20
     fracflux = (fracfluxn / fracfluxd).astype('f4')
     fluxlbs, dfluxlbs = compute_lbs_flux(impsfstack, psfstack, weightstack,
-                                         flux/norm)
+                                         flux/(norm+(norm == 0)))
     fluxiso, xiso, yiso = compute_iso_fit(impsfstack, psfstack, weightstack,
-                                          flux/norm, psfderiv)
+                                          flux/(norm+(norm == 0)), psfderiv)
     fluxlbs = fluxlbs.astype('f4')
     dfluxlbs = dfluxlbs.astype('f4')
     fwhm = psfmod.neff_fwhm(psfstack).astype('f4')

diff --git a/crowdsource/psf.py b/crowdsource/psf.py
@@ -51,6 +51,7 @@ def central_stamp(stamp, censize=19):
 def neff_fwhm(stamp):
     """FWHM-like quantity derived from N_eff = numpy.sum(PSF**2.)**-1"""
     norm = numpy.sum(stamp, axis=(-1, -2), keepdims=True)
+    norm += (norm == 0)
     return 1.18 * (numpy.pi*numpy.sum((stamp/norm)**2., axis=(-1, -2)))**(-0.5)