save stokes flow solution

LLNL · Mar 18, 2024 · 272c85c · 272c85c
1 parent 0e7cf67
commit 272c85c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/include/multiblock_solver.hpp b/include/multiblock_solver.hpp
@@ -137,6 +137,8 @@ friend class ParameterizedProblem;
    BlockVector* GetSolution() { return U; }
    BlockVector* GetSolutionCopy() { return new BlockVector(*U); }
 
+   void SetSolutionSaveMode(const bool save_sol_);
+
    void GetVariableVector(const int &var_idx, BlockVector &global, BlockVector &var);
    void SetVariableVector(const int &var_idx, BlockVector &var, BlockVector &global);
 

diff --git a/src/advdiff_solver.cpp b/src/advdiff_solver.cpp
@@ -241,6 +241,7 @@ void AdvDiffSolver::GetFlowField(ParameterizedProblem *flow_problem)
 
    stokes_solver = new StokesSolver;
    stokes_solver->InitVariables();
+   stokes_solver->SetSolutionSaveMode(save_flow);
 
    if (load_flow && FileExists(flow_file))
       stokes_solver->LoadSolution(flow_file);

diff --git a/src/multiblock_solver.cpp b/src/multiblock_solver.cpp
@@ -125,7 +125,13 @@ void MultiBlockSolver::ParseInputs()
    train_mode = SetTrainMode();
 
    // save solution if single run.
-   save_sol = config.GetOption<bool>("save_solution/enabled", false);
+   SetSolutionSaveMode(config.GetOption<bool>("save_solution/enabled", false));
+}
+
+void MultiBlockSolver::SetSolutionSaveMode(const bool save_sol_)
+{
+   // save solution if single run.
+   save_sol = save_sol_;
    if (save_sol)
    {
       // Default file path if no input file name is provided.