Tempus_StepperRKAppActionComposite.hpp

Go to the documentation of this file.

// @HEADER
// ****************************************************************************
//                Tempus: Copyright (2017) Sandia Corporation
//
// Distributed under BSD 3-clause license (See accompanying file Copyright.txt)
// ****************************************************************************
// @HEADER

#ifndef Tempus_StepperRKAppActionComposite_hpp
#define Tempus_StepperRKAppActionComposite_hpp

#include "Tempus_config.hpp"
#include "Tempus_StepperRKAppAction.hpp"
#include <vector>

namespace Tempus {

template<class Scalar>
class StepperRKAppActionComposite
  : virtual public Tempus::StepperRKAppAction<Scalar>
{
public:

  StepperRKAppActionComposite() {}

  virtual ~StepperRKAppActionComposite() {}

  virtual void execute(
    Teuchos::RCP<SolutionHistory<Scalar> > sh,
    Teuchos::RCP<StepperRKBase<Scalar> > stepper,
    const typename StepperRKAppAction<Scalar>::ACTION_LOCATION actLoc)
  {
    for(auto& a : appActions_)
      a->execute(sh, stepper, actLoc);
  }

  // Add AppAction to the AppAction vector.
  void addRKAppAction(Teuchos::RCP<StepperRKAppAction<Scalar> > appAction)
  {
    appActions_.push_back(appAction);
  }

  // Clear the AppAction vector.
  void clearRKAppActions()
  { appActions_.clear(); }

  // Return the size of the AppAction vector.
  std::size_t getSize() const { return appActions_.size(); }

private:

  std::vector<Teuchos::RCP<StepperRKAppAction<Scalar > > > appActions_;

};

} // namespace Tempus
#endif // Tempus_StepperRKAppActionComposite_hpp