From 20b12937e028b3e0069417b7766ffb42ae700221 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 1 Jul 2020 13:23:44 -0500
Subject: [PATCH] Introduce Discretization.ndofs

---
 meshmode/discretization/__init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/meshmode/discretization/__init__.py b/meshmode/discretization/__init__.py
index f1dd8915..d19e5b92 100644
--- a/meshmode/discretization/__init__.py
+++ b/meshmode/discretization/__init__.py
@@ -188,6 +188,8 @@ class Discretization(object):
 
     .. attribute:: ambient_dim
 
+    .. attribute:: ndofs
+
     .. attribute :: groups
 
     .. automethod:: empty
@@ -241,6 +243,10 @@ class Discretization(object):
     def ambient_dim(self):
         return self.mesh.ambient_dim
 
+    @property
+    def ndofs(self):
+        return sum(grp.ndofs for grp in self.groups)
+
     def _new_array(self, actx, creation_func, dtype=None):
         if dtype is None:
             dtype = self.real_dtype
-- 
GitLab