Added a step size hook

Author: Thomas Baumann

pySDC/implementations/convergence_controller_classes/adaptivity.py

@@ -7,6 +7,7 @@
     BasicRestartingNonMPI,
 )
 from pySDC.implementations.controller_classes.controller_nonMPI import controller_nonMPI
+from pySDC.implementations.hooks.log_step_size import LogStepSize
 
 
 class AdaptivityBase(ConvergenceController):
@@ -35,6 +36,7 @@ def setup(self, controller, params, description, **kwargs):
             "control_order": -50,
             "beta": 0.9,
         }
+        controller.add_hook(LogStepSize)
         return {**defaults, **params}
 
     def dependencies(self, controller, description, **kwargs):

pySDC/implementations/convergence_controller_classes/estimate_embedded_error.py

@@ -44,7 +44,7 @@ def get_implementation(cls, flavor):
         elif flavor == 'nonMPI':
             return EstimateEmbeddedErrorNonMPI
         else:
-            raise NotImplementedError(f'Flavor {flavor} of EmstimateEmbeddedError is not implemented!')
+            raise NotImplementedError(f'Flavor {flavor} of EstimateEmbeddedError is not implemented!')
 
     def setup(self, controller, params, description, **kwargs):
         """

pySDC/implementations/hooks/log_step_size.py

@@ -0,0 +1,31 @@
+from pySDC.core.Hooks import hooks
+
+
+class LogStepSize(hooks):
+    """
+    Store the step size at the end of each step as "dt".
+    """
+
+    def post_step(self, step, level_number):
+        """
+        Record step size
+
+        Args:
+            step (pySDC.Step.step): the current step
+            level_number (int): the current level number
+
+        Returns:
+            None
+        """
+        # some abbreviations
+        L = step.levels[level_number]
+
+        self.add_to_stats(
+            process=step.status.slot,
+            time=L.time,
+            level=L.level_index,
+            iter=step.status.iter,
+            sweep=L.status.sweep,
+            type='dt',
+            value=L.dt,
+        )

pySDC/projects/Resilience/hook.py

@@ -2,9 +2,10 @@
 from pySDC.implementations.hooks.log_solution import LogSolution
 from pySDC.implementations.hooks.log_embedded_error_estimate import LogEmbeddedErrorEstimate
 from pySDC.implementations.hooks.log_extrapolated_error_estimate import LogExtrapolationErrorEstimate
+from pySDC.implementations.hooks.log_step_size import LogStepSize
 
 
-hook_collection = [LogSolution, LogEmbeddedErrorEstimate, LogExtrapolationErrorEstimate]
+hook_collection = [LogSolution, LogEmbeddedErrorEstimate, LogExtrapolationErrorEstimate, LogStepSize]
 
 
 class log_data(hooks):
@@ -28,15 +29,6 @@ def post_step(self, step, level_number):
 
         L.sweep.compute_end_point()
 
-        self.add_to_stats(
-            process=step.status.slot,
-            time=L.time,
-            level=L.level_index,
-            iter=0,
-            sweep=L.status.sweep,
-            type='dt',
-            value=L.dt,
-        )
         self.add_to_stats(
             process=step.status.slot,
             time=L.time,