crankNicolsonScheme.hpp

#ifndef MARIAN_CRANKNICOLSONSCHEME_HPP
#define MARIAN_CRANKNICOLSONSCHEME_HPP

#include <FDM/schemes/fdScheme.hpp>
#include <FDM/tridiagonalSolver.hpp>

namespace marian {
  class CrankNicolsonScheme : public DCFDScheme<CrankNicolsonScheme> {
  public:
    CrankNicolsonScheme(){};
    CrankNicolsonScheme(SmartPointer<TridiagonalSolver> solver): solver_(solver) {};
    
    void setSolver(const SmartPointer<TridiagonalSolver>& solver) override {
      solver_ = solver;
    }
    std::vector<double> solve(std::vector<double> f,
                  const std::vector<SmartPointer<BoundaryCondition> >& bcs,
                  const std::vector<double>& time_grid,
                  const TridiagonalOperator& L) override;
    std::vector<double> solveAndSave(std::vector<double> f,
                     const std::vector<SmartPointer<BoundaryCondition> >& bcs,
                     const std::vector<double>& spatial_grid,
                     const std::vector<double>& time_grid,
                     const TridiagonalOperator& L,
                     const std::string file_name) override;
    std::string info() const override {
      return "CrankNicolson";
    }
  private:
    SmartPointer<TridiagonalSolver> solver_;   
  };
  
} // namespace marian


#endif /* MARIAN_CRANKNICOLSONSCHEME_HPP */