caught exception in estimate_iterations_kin_wave when all map values are zero.

Author: hcwinsemius

wflow/wflow_funcs.py

@@ -99,10 +99,12 @@ def set_dd(ldd, _ldd_us=_ldd_us, pit_value=_pits):
 ###############################################################################
 
 def estimate_iterations_kin_wave(Q, Beta, alpha, timestepsecs, dx, mv):
-    
-    celerity = pcr.ifthen(Q > 0.0, 1.0 / (alpha * Beta * Q**(Beta-1)))
-    courant = (timestepsecs / dx) * celerity
-    np_courant = pcr.pcr2numpy(courant, mv)
+    if (pcr.pcr2numpy(Q, mv)).max() > 0:
+        celerity = pcr.ifthen(Q > 0.0, 1.0 / (alpha * Beta * Q**(Beta-1)))
+        courant = (timestepsecs / dx) * celerity
+        np_courant = pcr.pcr2numpy(courant, mv)
+    else:
+        np_courant = np.zeros(pcr.pcr2numpy(Q, mv).shape) + mv
     np_courant[np_courant==mv] = np.nan
     try:
         it_kin = int(np.ceil(1.25*(np.nanpercentile(np_courant,95))))