[Cleanup] Store Dirichlet values

src/timestepping/backward_euler.cc

@@ -23,7 +23,7 @@ void BackwardEuler<Vector, Matrix, Function, dim>::setup(
     Vector const &ell, double _tau, double relativeTime, Vector &rhs,
     Vector &iterate, Matrix &AM) {
   postProcessCalled = false;
-
+  dirichletFunction.evaluate(relativeTime, dirichletValue);
   tau = _tau;
 
   /* We start out with the formulation
@@ -72,12 +72,8 @@ void BackwardEuler<Vector, Matrix, Function, dim>::setup(
 
   for (size_t i = 0; i < dirichletNodes.size(); ++i)
     for (size_t j = 0; j < dim; ++j)
-      if (dirichletNodes[i][j]) {
-        if (j == 0)
-          dirichletFunction.evaluate(relativeTime, iterate[i][j]);
-        else
-          iterate[i][j] = 0;
-      }
+      if (dirichletNodes[i][j])
+        iterate[i][j] = (j == 0) ? dirichletValue : 0;
 }
 
 template <class Vector, class Matrix, class Function, size_t dim>

src/timestepping/backward_euler.hh

@@ -25,6 +25,7 @@ class BackwardEuler : public TimeSteppingScheme<Vector, Matrix, Function, dim> {
   Vector v;
   Dune::BitSetVector<dim> const &dirichletNodes;
   Function const &dirichletFunction;
+  double dirichletValue;
 
   Vector u_o;
   Vector v_o;

src/timestepping/newmark.cc

@@ -27,7 +27,7 @@ void Newmark<Vector, Matrix, Function, dim>::setup(Vector const &ell,
                                                    Vector &rhs, Vector &iterate,
                                                    Matrix &AM) {
   postProcessCalled = false;
-
+  dirichletFunction.evaluate(relativeTime, dirichletValue);
   tau = _tau;
 
   /* We start out with the formulation
@@ -79,12 +79,8 @@ void Newmark<Vector, Matrix, Function, dim>::setup(Vector const &ell,
 
   for (size_t i = 0; i < dirichletNodes.size(); ++i)
     for (size_t j = 0; j < dim; ++j)
-      if (dirichletNodes[i][j]) {
-        if (j == 0)
-          dirichletFunction.evaluate(relativeTime, iterate[i][j]);
-        else
-          iterate[i][j] = 0;
-      }
+      if (dirichletNodes[i][j])
+        iterate[i][j] = (j == 0) ? dirichletValue : 0;
 }
 
 template <class Vector, class Matrix, class Function, size_t dim>

src/timestepping/newmark.hh

@@ -27,6 +27,7 @@ class Newmark : public TimeSteppingScheme<Vector, Matrix, Function, dim> {
   Vector a;
   Dune::BitSetVector<dim> const &dirichletNodes;
   Function const &dirichletFunction;
+  double dirichletValue;
 
   Vector u_o;
   Vector v_o;