From 3934cfd6dc69fcc022a23652344ac2683f506ff5 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 25 Aug 2020 18:04:03 -0500
Subject: [PATCH] Eager nodes: allow passing dd
---
grudge/eager.py | 13 +++++++++++--
1 file changed, 11 insertions(+), 2 deletions(-)
diff --git a/grudge/eager.py b/grudge/eager.py
index a79d9c7a..407c182e 100644
--- a/grudge/eager.py
+++ b/grudge/eager.py
@@ -77,8 +77,17 @@ class EagerDGDiscretization(DGDiscretizationWithBoundaries):
return self.connection_from_dds(src, tgt)(vec)
- def nodes(self):
- return self._volume_discr.nodes()
+ def nodes(self, dd=None):
+ r"""Return the nodes of a discretization.
+
+ :arg dd: a :class:`~grudge.sym.DOFDesc`, or a value convertible to one.
+ Defaults to the base volume discretization.
+ :returns: an object array of :class:`~meshmode.dof_array.DOFArray`\ s
+ """
+ if dd is None:
+ return self._volume_discr.nodes()
+ else:
+ return self.discr_from_dd(dd).nodes()
@memoize_method
def _bound_grad(self):
--
GitLab