From 4c7fee77d33cdb90da59a4659402311e61583326 Mon Sep 17 00:00:00 2001
From: Elias Pipping <elias.pipping@fu-berlin.de>
Date: Sun, 11 Mar 2012 21:54:36 +0100
Subject: [PATCH] Cleanup

---
 src/one-body-sample.cc | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/src/one-body-sample.cc b/src/one-body-sample.cc
index 09eea49e..4d08d013 100644
--- a/src/one-body-sample.cc
+++ b/src/one-body-sample.cc
@@ -275,17 +275,11 @@ int main(int argc, char *argv[]) {
         std::cout.flush();
       }
 
-      // b = neumann
-      assemble_neumann<GridType, GridView, SmallVector, P1Basis>(
-          leafView, p1Basis, neumannNodes, b4, functions.get("sampleFunction"),
-          h * run);
-      b5 = b4;
-
-      // b -= linear update
-      stiffnessMatrix.mmv(u4, b4);
-      stiffnessMatrix.mmv(u5, b5);
-
       if (parset.get<bool>("solver.tnnmg.use")) {
+        assemble_neumann<GridType, GridView, SmallVector, P1Basis>(
+            leafView, p1Basis, neumannNodes, b4,
+            functions.get("sampleFunction"), h * run);
+        stiffnessMatrix.mmv(u4, b4);
         for (int state_fpi = 0;
              state_fpi < parset.get<int>("solver.tnnmg.fixed_point_iterations");
              ++state_fpi) {
@@ -339,6 +333,10 @@ int main(int argc, char *argv[]) {
       s4_old = *s4_new;
 
       if (parset.get<bool>("benchmarks.fpi.enable")) {
+        assemble_neumann<GridType, GridView, SmallVector, P1Basis>(
+            leafView, p1Basis, neumannNodes, b5,
+            functions.get("sampleFunction"), h * run);
+        stiffnessMatrix.mmv(u5, b5);
         for (int state_fpi = 0;
              state_fpi < parset.get<int>("benchmarks.fpi.iterations");
              ++state_fpi) {
-- 
GitLab