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