Skip to content
Closed
Show file tree
Hide file tree
Changes from 9 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions examples/advection/surface.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,9 +158,9 @@ def main(ctx_factory, dim=2, order=4, use_quad=False, visualize=False):
discr_tag_to_group_factory[qtag] = \
QuadratureSimplexGroupFactory(order=4*order)

from grudge import DiscretizationCollection
from grudge import make_discretization_collection

dcoll = DiscretizationCollection(
dcoll = make_discretization_collection(
actx, mesh,
discr_tag_to_group_factory=discr_tag_to_group_factory
)
Expand Down
4 changes: 2 additions & 2 deletions examples/advection/var-velocity.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,9 +146,9 @@ def main(ctx_factory, dim=2, order=4, use_quad=False, visualize=False):
else:
discr_tag_to_group_factory = {}

from grudge import DiscretizationCollection
from grudge import make_discretization_collection

dcoll = DiscretizationCollection(
dcoll = make_discretization_collection(
actx, mesh, order=order,
discr_tag_to_group_factory=discr_tag_to_group_factory
)
Expand Down
4 changes: 2 additions & 2 deletions examples/advection/weak.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,9 +134,9 @@ def main(ctx_factory, dim=2, order=4, visualize=False):
[np.linspace(-d/2, d/2, npoints) for _ in range(dim)],
order=order)

from grudge import DiscretizationCollection
from grudge import make_discretization_collection

dcoll = DiscretizationCollection(actx, mesh, order=order)
dcoll = make_discretization_collection(actx, mesh, order=order)

# }}}

Expand Down
4 changes: 2 additions & 2 deletions examples/geometry.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
from arraycontext import thaw
from grudge.array_context import PyOpenCLArrayContext

from grudge import DiscretizationCollection, shortcuts
from grudge import make_discretization_collection, shortcuts


def main(write_output=True):
Expand All @@ -49,7 +49,7 @@ def main(write_output=True):
from meshmode.mesh.generation import generate_warped_rect_mesh
mesh = generate_warped_rect_mesh(dim=2, order=4, nelements_side=6)

dcoll = DiscretizationCollection(actx, mesh, order=4)
dcoll = make_discretization_collection(actx, mesh, order=4)

nodes = thaw(dcoll.nodes(), actx)
bdry_nodes = thaw(dcoll.nodes(dd=BTAG_ALL), actx)
Expand Down
4 changes: 2 additions & 2 deletions examples/maxwell/cavities.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
from grudge.array_context import PyOpenCLArrayContext

from grudge.shortcuts import set_up_rk4
from grudge import DiscretizationCollection
from grudge import make_discretization_collection

from grudge.models.em import get_rectangular_cavity_mode

Expand All @@ -59,7 +59,7 @@ def main(ctx_factory, dim=3, order=4, visualize=False):
b=(1.0,)*dim,
nelements_per_axis=(4,)*dim)

dcoll = DiscretizationCollection(actx, mesh, order=order)
dcoll = make_discretization_collection(actx, mesh, order=order)

if 0:
epsilon0 = 8.8541878176e-12 # C**2 / (N m**2)
Expand Down
4 changes: 2 additions & 2 deletions examples/old_symbolics/dagrt-fusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@
from pytools import memoize
from pytools.obj_array import flat_obj_array

from grudge import sym, bind, DiscretizationCollection
from grudge import sym, bind, make_discretization_collection
from leap.rk import LSRK4MethodBuilder

from pyopencl.tools import ( # noqa
Expand Down Expand Up @@ -573,7 +573,7 @@ def get_wave_op_with_discr(actx, dims=2, order=4):

logger.debug("%d elements", mesh.nelements)

discr = DiscretizationCollection(actx, mesh, order=order)
discr = make_discretization_collection(actx, mesh, order=order)

from symbolic_wave_op import WeakWaveOperator
from meshmode.mesh import BTAG_ALL, BTAG_NONE
Expand Down
4 changes: 2 additions & 2 deletions examples/wave/var-propagation-speed.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
from grudge.array_context import PyOpenCLArrayContext

from grudge.shortcuts import set_up_rk4
from grudge import DiscretizationCollection
from grudge import make_discretization_collection

from pytools.obj_array import flat_obj_array

Expand All @@ -59,7 +59,7 @@ def main(ctx_factory, dim=2, order=4, visualize=False):
b=(0.5,)*dim,
nelements_per_axis=(20,)*dim)

dcoll = DiscretizationCollection(actx, mesh, order=order)
dcoll = make_discretization_collection(actx, mesh, order=order)

def source_f(actx, dcoll, t=0):
source_center = np.array([0.1, 0.22, 0.33])[:dcoll.dim]
Expand Down
8 changes: 5 additions & 3 deletions examples/wave/wave-min-mpi.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
from grudge.array_context import PyOpenCLArrayContext

from grudge.shortcuts import set_up_rk4
from grudge import DiscretizationCollection
from grudge import make_discretization_collection

from mpi4py import MPI

Expand Down Expand Up @@ -79,8 +79,10 @@ def main(ctx_factory, dim=2, order=4, visualize=False):
else:
local_mesh = mesh_dist.receive_mesh_part()

dcoll = DiscretizationCollection(actx, local_mesh, order=order,
mpi_communicator=comm)
dcoll = make_discretization_collection(
actx, local_mesh, order=order,
mpi_communicator=comm
)

def source_f(actx, dcoll, t=0):
source_center = np.array([0.1, 0.22, 0.33])[:dcoll.dim]
Expand Down
8 changes: 5 additions & 3 deletions examples/wave/wave-op-mpi.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@

from meshmode.mesh import BTAG_ALL, BTAG_NONE # noqa

from grudge.discretization import DiscretizationCollection
from grudge.discretization import make_discretization_collection
from grudge.shortcuts import make_visualizer

import grudge.op as op
Expand Down Expand Up @@ -176,8 +176,10 @@ def main(ctx_factory, dim=2, order=3, visualize=False):
else:
local_mesh = mesh_dist.receive_mesh_part()

dcoll = DiscretizationCollection(actx, local_mesh, order=order,
mpi_communicator=comm)
dcoll = make_discretization_collection(
actx, local_mesh, order=order,
mpi_communicator=comm
)

fields = flat_obj_array(
bump(actx, dcoll),
Expand Down
4 changes: 2 additions & 2 deletions examples/wave/wave-op-var-velocity.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@

from meshmode.mesh import BTAG_ALL, BTAG_NONE # noqa

from grudge.discretization import DiscretizationCollection
from grudge.discretization import make_discretization_collection
from grudge.dof_desc import DISCR_TAG_BASE, DISCR_TAG_QUAD, DOFDesc
from grudge.shortcuts import make_visualizer

Expand Down Expand Up @@ -179,7 +179,7 @@ def main(ctx_factory, dim=2, order=3, visualize=False):
from meshmode.discretization.poly_element import \
QuadratureSimplexGroupFactory, \
default_simplex_group_factory
dcoll = DiscretizationCollection(
dcoll = make_discretization_collection(
actx, mesh,
discr_tag_to_group_factory={
DISCR_TAG_BASE: default_simplex_group_factory(base_dim=dim, order=order),
Expand Down
14 changes: 11 additions & 3 deletions grudge/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,7 @@
__copyright__ = "Copyright (C) 2015 Andreas Kloeckner"
__copyright__ = """
Copyright (C) 2015 Andreas Kloeckner
Copyright (C) 2021 University of Illinois Board of Trustees
"""

__license__ = """
Permission is hereby granted, free of charge, to any person obtaining a copy
Expand All @@ -23,8 +26,13 @@
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"
]
Loading