imex_sweeper.hpp

Go to the documentation of this file.

#ifndef _PFASST_ENCAP_IMEX_SWEEPER_HPP_
#define _PFASST_ENCAP_IMEX_SWEEPER_HPP_

#include <memory>
#include <vector>

#include "pfasst/encap/encapsulation.hpp"
#include "pfasst/encap/encap_sweeper.hpp"

using namespace std;

namespace pfasst
{
  namespace encap
  {
    using pfasst::encap::Encapsulation;

    template<typename time = time_precision>
    class IMEXSweeper
      : public pfasst::encap::EncapSweeper<time>
    {
      protected:


        vector<shared_ptr<Encapsulation<time>>> s_integrals;

        vector<shared_ptr<Encapsulation<time>>> fs_expl;
        shared_ptr<Encapsulation<time>> fs_expl_start;

        vector<shared_ptr<Encapsulation<time>>> fs_impl;

        virtual void integrate_end_state(time dt);

      public:
        IMEXSweeper() = default;
        virtual ~IMEXSweeper() = default;


        virtual void setup(bool coarse) override;

        virtual void predict(bool initial) override;

        virtual void sweep() override;

        virtual void advance() override;

        virtual void reevaluate(bool initial_only) override;

        virtual void integrate(time dt, vector<shared_ptr<Encapsulation<time>>> dst) const override;

        virtual void residual(time dt, vector<shared_ptr<Encapsulation<time>>> dst) const override;


        virtual void f_expl_eval(shared_ptr<Encapsulation<time>> f_expl_encap,
                                 shared_ptr<Encapsulation<time>> u_encap,
                                 time t);

        virtual void f_impl_eval(shared_ptr<Encapsulation<time>> f_impl_encap,
                                 shared_ptr<Encapsulation<time>> u_encap,
                                 time t);

        virtual void impl_solve(shared_ptr<Encapsulation<time>> f_encap,
                                shared_ptr<Encapsulation<time>> u_encap,
                                time t, time dt,
                                shared_ptr<Encapsulation<time>> rhs_encap);

      private:
        virtual void predict_with_left(bool initial);
        virtual void predict_without_left(bool initial);
        virtual void sweep_with_left();
        virtual void sweep_without_left();
    };
  }  // ::pfasst::encap
}  // ::pfasst

#include "pfasst/encap/imex_sweeper_impl.hpp"

#endif