From 77378b4d06c0cd25d24c8512e04f18bbedce64a1 Mon Sep 17 00:00:00 2001
From: Marcel Koch <marcel.koch@uni-muenster.de>
Date: Tue, 16 Oct 2018 11:53:29 +0200
Subject: [PATCH] use different dof numbers for DG and CG cases

---
 python/dune/perftool/pdelab/driver/timings.py | 22 ++++++++++++++-----
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/python/dune/perftool/pdelab/driver/timings.py b/python/dune/perftool/pdelab/driver/timings.py
index 4ba2d831..cdeb6b1d 100644
--- a/python/dune/perftool/pdelab/driver/timings.py
+++ b/python/dune/perftool/pdelab/driver/timings.py
@@ -44,12 +44,22 @@ def dump_dof_numbers(stream):
     include_file("dune/pdelab/common/partitionviewentityset.hh", filetag="driver")
     gvt = type_leafview()
     gv = name_leafview()
-    return ["Dune::PDELab::NonOverlappingEntitySet<{}> es({});".format(gvt, gv),
-            "{} << \"{} \" << {} << \" dofs dofs \" << {}.maxLocalSize() * es.size(0) << std::endl;".format(stream,
-                                                                                                            level,
-                                                                                                            ident,
-                                                                                                            name_trial_gfs())
-            ]
+
+    from dune.perftool.pdelab.driver import get_trial_element, isDG
+    if isDG(get_trial_element()):
+        return ["Dune::PDELab::NonOverlappingEntitySet<{}> es({});".format(gvt, gv),
+                "{} << \"{} \" << {} << \" dofs dofs \" << {}.maxLocalSize() * es.size(0) << std::endl;".format(stream,
+                                                                                                                level,
+                                                                                                                ident,
+                                                                                                                name_trial_gfs())
+                ]
+    else:
+        return ["{} << \"{} \" << {} << \" dofs dofs \" << {}.() << std::endl;".format(stream,
+                                                                                       level,
+                                                                                       ident,
+                                                                                       name_trial_gfs())
+                ]
+
 
 
 @preamble(section="timings")
-- 
GitLab