Adding preIterate and postIterate methods.

examples/heat-equation/heatEquationSolver.h

@@ -76,12 +76,18 @@ class heatEquationSolver
    void bindAuxiliaryDofs( const MeshType& mesh,
                            DofVectorType& auxiliaryDofs );
 
+   bool preIterate( const RealType& time,
+                    const RealType& tau,
+                    const MeshType& mesh,
+                    DofVectorType& u );
+
    void getExplicitRHS( const RealType& time,
                         const RealType& tau,
                         const MeshType& mesh,
                         DofVectorType& _u,
                         DofVectorType& _fu );
 
+
    void assemblyLinearSystem( const RealType& time,
                               const RealType& tau,
                               const MeshType& mesh,
@@ -89,6 +95,12 @@ class heatEquationSolver
                               MatrixType& matrix,
                               DofVectorType& rightHandSide );
 
+   bool postIterate( const RealType& time,
+                     const RealType& tau,
+                     const MeshType& mesh,
+                     DofVectorType& u );
+
+
    tnlSolverMonitor< RealType, IndexType >* getSolverMonitor();
    
    protected:

examples/heat-equation/heatEquationSolver_impl.h

@@ -183,6 +183,21 @@ makeSnapshot( const RealType& time,
    return true;
 }
 
+template< typename Mesh,
+          typename DifferentialOperator,
+          typename BoundaryCondition,
+          typename RightHandSide >
+bool
+heatEquationSolver< Mesh, DifferentialOperator, BoundaryCondition, RightHandSide >::
+preIterate( const RealType& time,
+            const RealType& tau,
+            const MeshType& mesh,
+            DofVectorType& u )
+{
+   return true;
+}
+
+
 template< typename Mesh,
           typename DifferentialOperator,
           typename BoundaryCondition,
@@ -245,6 +260,21 @@ assemblyLinearSystem( const RealType& time,
    //abort();
 }
 
+template< typename Mesh,
+          typename DifferentialOperator,
+          typename BoundaryCondition,
+          typename RightHandSide >
+bool
+heatEquationSolver< Mesh, DifferentialOperator, BoundaryCondition, RightHandSide >::
+postIterate( const RealType& time,
+             const RealType& tau,
+             const MeshType& mesh,
+             DofVectorType& u )
+{
+   return true;
+}
+
+
 template< typename Mesh,
           typename DifferentialOperator,
           typename BoundaryCondition,

src/implementation/operators/tnlDirichletBoundaryConditions_impl.h

@@ -118,7 +118,7 @@ setBoundaryConditions( const RealType& time,
                        DofVectorType& fu )
 {
    fu[ index ] = 0;
-   u[ index ] = function.getValue( mesh.getCellCenter( coordinates ), time );;
+   u[ index ] = function.getValue( mesh.getCellCenter( coordinates ), time );
 }
 
 template< typename MeshReal,

src/implementation/solvers/pde/tnlExplicitTimeStepper_impl.h

@@ -112,7 +112,25 @@ void tnlExplicitTimeStepper< Problem, OdeSolver >::getExplicitRHS( const RealTyp
                                                                    DofVectorType& _u,
                                                                    DofVectorType& _fu )
 {
+   if( ! this->problem->preIterate( time,
+                                    tau,
+                                    *( this->mesh),
+                                    _u ) )
+   {
+      cerr << endl << "Preiteration failed." << endl;
+      return;
+      //return false; // TODO: throw exception
+   }
    this->problem->getExplicitRHS( time, tau, *( this->mesh ), _u, _fu );
+   if( ! this->problem->postIterate( time,
+                                     tau,
+                                     *( this->mesh ),
+                                     _u ) )
+   {
+      cerr << endl << "Postiteration failed." << endl;
+      return;
+      //return false; // TODO: throw exception
+   }
 }
 
 #endif /* TNLEXPLICITTIMESTEPPER_IMPL_H_ */

src/implementation/solvers/pde/tnlSemiImplicitTimeStepper_impl.h

@@ -139,6 +139,14 @@ solve( const RealType& time,
    {
       RealType currentTau = Min( this->timeStep, stopTime - t );
 
+      if( ! this->problem->preIterate( t,
+                                       currentTau,
+                                       mesh,
+                                       dofVector ) )
+      {
+         cerr << endl << "Preiteration failed." << endl;
+         return false;
+      }
       if( verbose )
          cout << "                                                                  Assembling the linear system ... \r" << flush;
       this->problem->assemblyLinearSystem( t,
@@ -154,6 +162,14 @@ solve( const RealType& time,
          cerr << "The linear system solver did not converge." << endl;
          return false;
       }
+      if( ! this->problem->postIterate( t,
+                                        currentTau,
+                                        mesh,
+                                        dofVector ) )
+      {
+         cerr << endl << "Postiteration failed." << endl;
+         return false;
+      }
       t += currentTau;
    }
    return true;