TransientProblemInterface has TimeStepAdapt object and getters fo…

…r it
andrsd · Jan 15, 2025 · de292c0 · de292c0
1 parent 2459acc
commit de292c0
Show file tree

Hide file tree

Showing 5 changed files with 43 additions and 1 deletion.
diff --git a/include/godzilla/TSAbstract.h b/include/godzilla/TSAbstract.h
@@ -75,7 +75,11 @@ class TSAbstract {
 protected:
     TS get_ts();
 
+    /// Get time step adaptor
+    ///
+    /// @return Time step adaptor
     TimeStepAdapt & get_adapt();
+    const TimeStepAdapt & get_adapt() const;
 
     TSConvergedReason get_reason() const;
     void set_reason(TSConvergedReason reason);

diff --git a/include/godzilla/TimeStepAdapt.h b/include/godzilla/TimeStepAdapt.h
@@ -21,7 +21,6 @@ class TimeStepAdapt {
     };
 
     TimeStepAdapt();
-    TimeStepAdapt(TS ts, TSAdapt tsadapt);
 
     /// Add a candidate scheme for the adaptive controller to select from
     ///
@@ -152,6 +151,8 @@ class TimeStepAdapt {
     void set_type(Type type);
 
 private:
+    TimeStepAdapt(TS ts, TSAdapt tsadapt);
+
     TS ts;
     TSAdapt tsadapt;
 

diff --git a/include/godzilla/TransientProblemInterface.h b/include/godzilla/TransientProblemInterface.h
@@ -17,6 +17,7 @@ namespace godzilla {
 class Problem;
 class Parameters;
 class TimeSteppingAdaptor;
+class TimeStepAdapt;
 
 /// Interface for transient simulations
 ///
@@ -155,6 +156,16 @@ class TransientProblemInterface {
         return *static_cast<TSAbstract *>(this->ts->data);
     }
 
+    /// Get time step adaptivity
+    ///
+    /// @return Time step adaptivity object
+    const TimeStepAdapt & get_time_step_adapt() const;
+
+    /// Get time step adaptivity
+    ///
+    /// @return Time step adaptivity object
+    TimeStepAdapt & get_time_step_adapt();
+
 protected:
     /// Get underlying non-linear solver
     SNESolver get_snes() const;
@@ -253,6 +264,8 @@ class TransientProblemInterface {
 
     /// PETSc TS object
     TS ts;
+    /// Time step adapt object
+    TimeStepAdapt time_step_adapt;
     /// Method for monitoring the solve
     Delegate<void(Int it, Real rnorm, const Vector & x)> monitor_method;
     /// Method for computing right-hand side

diff --git a/src/TSAbstract.cpp b/src/TSAbstract.cpp
@@ -172,6 +172,13 @@ TSAbstract::get_adapt()
     return this->adapt;
 }
 
+const TimeStepAdapt &
+TSAbstract::get_adapt() const
+{
+    CALL_STACK_MSG();
+    return this->adapt;
+}
+
 TSConvergedReason
 TSAbstract::get_reason() const
 {

diff --git a/src/TransientProblemInterface.cpp b/src/TransientProblemInterface.cpp
@@ -4,6 +4,7 @@
 #include "godzilla/Godzilla.h"
 #include "godzilla/CallStack.h"
 #include "godzilla/Problem.h"
+#include "godzilla/TimeStepAdapt.h"
 #include "godzilla/TimeSteppingAdaptor.h"
 #include "godzilla/TransientProblemInterface.h"
 #include "godzilla/LoggingInterface.h"
@@ -129,6 +130,8 @@ TransientProblemInterface::TransientProblemInterface(Problem * problem, const Pa
 {
     CALL_STACK_MSG();
     PETSC_CHECK(TSCreate(this->problem->get_comm(), &this->ts));
+    this->time_step_adapt = TimeStepAdapt::from_ts(this->ts);
+
     if (this->tpi_params.is_param_valid("end_time") && this->tpi_params.is_param_valid("num_steps"))
         this->problem->log_error(
             "Cannot provide 'end_time' and 'num_steps' together. Specify one or the other.");
@@ -228,6 +231,20 @@ TransientProblemInterface::get_step_number() const
     return this->step_num;
 }
 
+const TimeStepAdapt &
+TransientProblemInterface::get_time_step_adapt() const
+{
+    CALL_STACK_MSG();
+    return this->time_step_adapt;
+}
+
+TimeStepAdapt &
+TransientProblemInterface::get_time_step_adapt()
+{
+    CALL_STACK_MSG();
+    return this->time_step_adapt;
+}
+
 void
 TransientProblemInterface::init()
 {