Added Q_D coefficient update to IMEX sweepers

Author: brownbaerchen

pySDC/implementations/sweeper_classes/imex_1st_order.py

@@ -61,16 +61,14 @@ def update_nodes(self):
         Returns:
             None
         """
-
-        # get current level and problem description
         L = self.level
         P = L.prob
+        M = self.coll.num_nodes
 
         # only if the level has been touched before
         assert L.status.unlocked
 
-        # get number of collocation nodes for easier access
-        M = self.coll.num_nodes
+        self.updateVariableCoeffs(L.status.sweep)
 
         # gather all terms which are known already (e.g. from the previous iteration)
         # this corresponds to u0 + QF(u^k) - QIFI(u^k) - QEFE(u^k) + tau

pySDC/implementations/sweeper_classes/imex_1st_order_MPI.py

@@ -43,14 +43,13 @@ def update_nodes(self):
         Returns:
             None
         """
-
         L = self.level
         P = L.prob
 
         # only if the level has been touched before
         assert L.status.unlocked
 
-        # get number of collocation nodes for easier access
+        self.updateVariableCoeffs(L.status.sweep)
 
         # gather all terms which are known already (e.g. from the previous iteration)
         # this corresponds to u0 + QF(u^k) - QdF(u^k) + tau

pySDC/implementations/sweeper_classes/imex_1st_order_mass.py

@@ -15,16 +15,14 @@ def update_nodes(self):
         Returns:
             None
         """
-
-        # get current level and problem description
         L = self.level
         P = L.prob
+        M = self.coll.num_nodes
 
         # only if the level has been touched before
         assert L.status.unlocked
 
-        # get number of collocation nodes for easier access
-        M = self.coll.num_nodes
+        self.updateVariableCoeffs(L.status.sweep)
 
         # gather all terms which are known already (e.g. from the previous iteration)
         # this corresponds to u0 + QF(u^k) - QIFI(u^k) - QEFE(u^k) + tau