fix error message when using HLLC with sphericalpolar geometry

python-hydro · Jul 31, 2024 · 1a6b0d4 · 1a6b0d4
1 parent 6bb2b66
commit 1a6b0d4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/pyro/compressible/interface.py b/pyro/compressible/interface.py
@@ -888,10 +888,6 @@ def riemann_hllc(idir, ng, coord_type,
         Conserved flux
     """
 
-
-    if coord_type == 1:
-        msg.fail("ERROR: HLLC Riemann Solver is not supported for SphericalPolar Geometry")
-
     qx, qy, nvar = U_l.shape
 
     F = np.zeros((qx, qy, nvar))

diff --git a/pyro/compressible/simulation.py b/pyro/compressible/simulation.py
@@ -111,6 +111,12 @@ def initialize(self, extra_vars=None, ng=4):
         my_grid = grid_setup(self.rp, ng=ng)
         my_data = self.data_class(my_grid)
 
+        # Make sure we use CGF for riemann solver when we do SphericalPolar
+        riemann = self.rp.get_param("compressible.riemann")
+        if my_grid.coord_type == 1 and riemann == "HLLC":
+            msg.fail("ERROR: Only CGF Riemann Solver is supported " +
+                     "with SphericalPolar Geometry")
+
         # define solver specific boundary condition routines
         bnd.define_bc("hse", BC.user, is_solid=False)
         bnd.define_bc("ramp", BC.user, is_solid=False)  # for double mach reflection problem