From 4332f9d063490a307de8045bceb24f35ad02148d Mon Sep 17 00:00:00 2001
From: Thomas Gibson <gibsonthomas1120@hotmail.com>
Date: Thu, 3 Jun 2021 17:32:13 -0500
Subject: [PATCH] Add elementwise_integral reduction
---
grudge/op.py | 21 ++++++++++++++++++++-
1 file changed, 20 insertions(+), 1 deletion(-)
diff --git a/grudge/op.py b/grudge/op.py
index 152e466d..d0c43741 100644
--- a/grudge/op.py
+++ b/grudge/op.py
@@ -55,6 +55,7 @@ Elementwise reductions
----------------------
.. autofunction:: elementwise_sum
+.. autofunction:: elementwise_integral
"""
__copyright__ = """
@@ -1185,7 +1186,6 @@ def elementwise_sum(dcoll: DiscretizationCollection, *args):
)
actx = vec.array_context
- vec = project(dcoll, "vol", dd, vec)
return DOFArray(
actx,
@@ -1198,6 +1198,25 @@ def elementwise_sum(dcoll: DiscretizationCollection, *args):
)
)
+
+def elementwise_integral(dcoll: DiscretizationCollection, dd, vec):
+ """Numerically integrates a function represented by a
+ :class:`~meshmode.dof_array.DOFArray` of degrees of freedom in
+ each element of a discretization, given by *dd*.
+
+ :arg dd: a :class:`~grudge.dof_desc.DOFDesc`, or a value convertible to one.
+ :arg vec: a :class:`~meshmode.dof_array.DOFArray`
+ :returns: a :class:`~meshmode.dof_array.DOFArray` containing the
+ elementwise integral if *vec*.
+ """
+
+ dd = dof_desc.as_dofdesc(dd)
+
+ ones = dcoll.discr_from_dd(dd).zeros(vec.array_context) + 1.0
+ return elementwise_sum(
+ dcoll, dd, vec * _apply_mass_operator(dcoll, dd, dd, ones)
+ )
+
# }}}
--
GitLab