diff --git a/python/dune/perftool/pdelab/driver.py b/python/dune/perftool/pdelab/driver.py index 872f6bd809ade33a923ab59907b3c58474ecd2d5..c904f8ae65c1e8c75699ca95c0255606f95eb860 100644 --- a/python/dune/perftool/pdelab/driver.py +++ b/python/dune/perftool/pdelab/driver.py @@ -755,11 +755,20 @@ def print_matrix(): "Dune::printmatrix(std::cout, native(m),\"global stiffness matrix\",\"row\",9,1);"] +@preamble +def define_gfs_name(element): + gfs = name_gfs(element) + #TODO make something sensible here + return "{}.name(\"bla\");".format(gfs) + + @preamble def vtkoutput(): + element = _form.coefficients()[0].element() + define_gfs_name(element) include_file("dune/pdelab/gridfunctionspace/vtk.hh", filetag="driver") vtkwriter = name_vtkwriter() - gfs = name_gfs(_form.coefficients()[0].element()) + gfs = name_gfs(element) vec = name_vector() vtkfile = name_vtkfile() dune_solve()