VertexCFD_TempusObserver_WriteMatrix.hpp

#ifndef VERTEXCFD_TEMPUSOBSERVER_WRITEMATRIX_HPP
#define VERTEXCFD_TEMPUSOBSERVER_WRITEMATRIX_HPP
 
#include <Tempus_Integrator.hpp>
#include <Tempus_IntegratorObserver.hpp>
 
#include <Panzer_GlobalIndexer.hpp>
#include <Teuchos_ParameterList.hpp>
 
namespace VertexCFD
{
namespace TempusObserver
{
//---------------------------------------------------------------------------//
template<class Scalar>
class WriteMatrix : virtual public Tempus::IntegratorObserver<Scalar>
{
  public:
    WriteMatrix(const Teuchos::ParameterList& output_params);
 
    void observeStartIntegrator(
        const Tempus::Integrator<Scalar>& integrator) override;
 
    void
    observeStartTimeStep(const Tempus::Integrator<Scalar>& integrator) override;
 
    void
    observeNextTimeStep(const Tempus::Integrator<Scalar>& integrator) override;
 
    void
    observeBeforeTakeStep(const Tempus::Integrator<Scalar>& integrator) override;
 
    void
    observeAfterTakeStep(const Tempus::Integrator<Scalar>& integrator) override;
 
    void observeAfterCheckTimeStep(
        const Tempus::Integrator<Scalar>& integrator) override;
 
    void
    observeEndTimeStep(const Tempus::Integrator<Scalar>& integrator) override;
 
    void
    observeEndIntegrator(const Tempus::Integrator<Scalar>& integrator) override;
 
  private:
    Teuchos::Array<int> _write_steps;
    std::string _jacobian_prefix;
    std::string _residual_prefix;
    bool _write_residual;
 
    // Extract Thyra linear operator from linear solver
    Teuchos::RCP<const Thyra::LinearOpBase<Scalar>> extractLinearOp(
        const Teuchos::RCP<const Thyra::LinearOpWithSolveBase<Scalar>>&
            linear_solver) const;
 
    // Extract forward op src (the Jacobian) from a linear op with solve
    template<class SolverType>
    Teuchos::RCP<const Thyra::LinearOpBase<Scalar>> tryExtractForwardOp(
        const Teuchos::RCP<const Thyra::LinearOpWithSolveBase<Scalar>>&
            linear_solver) const;
 
    // Write Jacobian matrix to file, toggling on Epetra vs. Tpetra
    void
    write_jacobian(const Teuchos::RCP<const Thyra::LinearOpBase<Scalar>>& matrix,
                   const std::string& filename,
                   const std::string& description) const;
 
    // Write residual to file, toggling on Epetra vs. Tpetra
    void
    write_residual(const Teuchos::RCP<const Thyra::VectorBase<Scalar>>& vec,
                   const std::string& filename,
                   const std::string& description) const;
};
 
//---------------------------------------------------------------------------//
 
} // end namespace TempusObserver
} // end namespace VertexCFD
 
#include "VertexCFD_TempusObserver_WriteMatrix_impl.hpp"
 
#endif // end VERTEXCFD_TEMPUSOBSERVER_WRITEMATRIX_HPP