Support multiple volumes in a discretization collection

Author: inducer

doc/conf.py

@@ -40,6 +40,7 @@ def get_version():
     "https://documen.tician.de/arraycontext/": None,
     "https://documen.tician.de/meshmode/": None,
     "https://documen.tician.de/loopy/": None,
+    "https://mpi4py.readthedocs.io/en/stable": None,
     }
 
 # index-page demo uses pyopencl via plot_directive

examples/hello-grudge.py

@@ -15,7 +15,7 @@
 from meshmode.mesh.generation import generate_box_mesh
 from meshmode.array_context import PyOpenCLArrayContext
 from arraycontext import thaw
-from grudge.dof_desc import DTAG_BOUNDARY, FACE_RESTR_INTERIOR
+from grudge.dof_desc import BoundaryDomainTag, FACE_RESTR_INTERIOR
 
 
 ctx = cl.create_some_context()
@@ -52,8 +52,8 @@ def flux(dcoll, u_tpair):
 
 
 vol_discr = dcoll.discr_from_dd("vol")
-left_bndry = DTAG_BOUNDARY("left")
-right_bndry = DTAG_BOUNDARY("right")
+left_bndry = BoundaryDomainTag("left")
+right_bndry = BoundaryDomainTag("right")
 
 x_vol = thaw(dcoll.nodes(), actx)
 x_bndry = thaw(dcoll.discr_from_dd(left_bndry).nodes(), actx)

examples/old_symbolics/dagrt-fusion.py

@@ -1206,15 +1206,17 @@ def problem_stats(order=3):
     with open_output_file("grudge-problem-stats.txt") as outf:
         _, dg_discr_2d = get_wave_op_with_discr(
             actx, dims=2, order=order)
-        print("Number of 2D elements:", dg_discr_2d.mesh.nelements, file=outf)
+        mesh_2d = dg_discr_2d.discr_from_dd("vol").mesh
+        print("Number of 2D elements:", mesh_2d.nelements, file=outf)
         vol_discr_2d = dg_discr_2d.discr_from_dd("vol")
         dofs_2d = {group.nunit_dofs for group in vol_discr_2d.groups}
         from pytools import one
         print("Number of DOFs per 2D element:", one(dofs_2d), file=outf)
 
         _, dg_discr_3d = get_wave_op_with_discr(
             actx, dims=3, order=order)
-        print("Number of 3D elements:", dg_discr_3d.mesh.nelements, file=outf)
+        mesh_3d = dg_discr_3d.discr_from_dd("vol").mesh
+        print("Number of 3D elements:", mesh_3d.nelements, file=outf)
         vol_discr_3d = dg_discr_3d.discr_from_dd("vol")
         dofs_3d = {group.nunit_dofs for group in vol_discr_3d.groups}
         from pytools import one

grudge/__init__.py

@@ -23,8 +23,9 @@
 import grudge.symbolic as sym
 from grudge.execution import bind
 
-from grudge.discretization import DiscretizationCollection
+from grudge.discretization import (
+        DiscretizationCollection, make_discretization_collection)
 
 __all__ = [
-    "sym", "bind", "DiscretizationCollection"
+    "sym", "bind", "DiscretizationCollection", "make_discretization_collection"
 ]