implement a clean_state routine for compressible (#329)

this will enforce small_dens

Author: zingale

pyro/compressible/_defaults

@@ -21,6 +21,8 @@ grav = 0.0                ; gravitational acceleration (in y-direction)
 
 riemann = HLLC            ; HLLC or CGF
 
+small_dens = -1.e200      ; minimum allowed density
+small_eint = -1.e200      ; minimum allowed specific internal energy
 
 [sponge]
 do_sponge = 0             ; do we include a sponge source term

pyro/compressible/simulation.py

@@ -293,6 +293,8 @@ def evolve(self):
         timestep dt.
         """
 
+        self.clean_state(self.cc_data.data)
+
         tm_evolve = self.tc.timer("evolve")
         tm_evolve.begin()
 
@@ -437,6 +439,12 @@ def evolve(self):
 
         tm_evolve.end()
 
+    def clean_state(self, U):
+        """enforce minimum density and eint on the conserved state U"""
+
+        U.v(n=self.ivars.idens)[:, :] = np.maximum(U.v(n=self.ivars.idens),
+                                                   self.rp.get_param("compressible.small_dens"))
+
     def dovis(self):
         """
         Do runtime visualization.

pyro/compressible_fv4/_defaults

@@ -23,6 +23,7 @@ grav = 0.0                ; gravitational acceleration (in y-direction)
 
 riemann = CGF
 
+small_dens = -1.e200      ; minimum allowed density
 
 [sponge]
 do_sponge = 0             ; do we include a sponge source term

pyro/compressible_fv4/simulation.py

@@ -19,6 +19,8 @@ def substep(self, myd):
         compute the advective source term for the given state
         """
 
+        self.clean_state(myd.data)
+
         myg = myd.grid
 
         # compute the source terms -- we need to do this first

pyro/compressible_rk/_defaults

@@ -25,6 +25,7 @@ riemann = HLLC            ; HLLC or CGF
 
 well_balanced = 0         ; use a well-balanced scheme to keep the model in hydrostatic equilibrium
 
+small_dens = -1.e200      ; minimum allowed density
 
 [sponge]
 do_sponge = 0             ; do we include a sponge source term

pyro/compressible_rk/simulation.py

@@ -15,6 +15,8 @@ def substep(self, myd):
         conservative state defined as part of myd
         """
 
+        self.clean_state(myd.data)
+
         myg = myd.grid
 
         # source terms -- note: this dt is the entire dt, not the

pyro/compressible_sdc/_defaults

@@ -19,6 +19,7 @@ grav = 0.0                ; gravitational acceleration (in y-direction)
 
 riemann = CGF
 
+small_dens = -1.e200      ; minimum allowed density
 
 [sponge]
 do_sponge = 0             ; do we include a sponge source term