Tempus_TimeStepControlStrategyComposite.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_TimeStepControlStrategyComposite_hpp
#define Tempus_TimeStepControlStrategyComposite_hpp

#include "Tempus_config.hpp"
#include "Tempus_TimeStepControlStrategy.hpp"
#include "Tempus_TimeStepControlStrategyConstant.hpp"
#include "Tempus_TimeStepControlStrategyBasicVS.hpp"
#include "Tempus_TimeStepControlStrategyIntegralController.hpp"
#include "Tempus_SolutionHistory.hpp"

namespace Tempus {

template <class Scalar>
class TimeStepControlStrategyComposite
  : virtual public TimeStepControlStrategy<Scalar> {
 public:
  TimeStepControlStrategyComposite()
  {
    this->setStrategyType("Composite");
    this->setStepType("Variable");
    this->isInitialized_ = false;
  }

  virtual ~TimeStepControlStrategyComposite() {}

  virtual void setNextTimeStep(const TimeStepControl<Scalar>& tsc,
                               Teuchos::RCP<SolutionHistory<Scalar>> sh,
                               Status& integratorStatus) override
  {
    for (auto& s : strategies_) s->setNextTimeStep(tsc, sh, integratorStatus);
  }


  std::string description() const override
  {
    return "Tempus::TimeStepControlComposite";
  }

  void describe(Teuchos::FancyOStream& out,
                const Teuchos::EVerbosityLevel verbLevel) const override
  {
    auto l_out = Teuchos::fancyOStream(out.getOStream());
    Teuchos::OSTab ostab(*l_out, 2, this->description());
    l_out->setOutputToRootOnly(0);

    *l_out << "\n--- " << this->description() << " ---" << std::endl;

    if (Teuchos::as<int>(verbLevel) >= Teuchos::as<int>(Teuchos::VERB_MEDIUM)) {
      *l_out << "  Strategy Type = " << this->getStrategyType() << std::endl
             << "  Step Type     = " << this->getStepType() << std::endl;

      std::stringstream sList;
      for (std::size_t i = 0; i < strategies_.size(); ++i) {
        sList << strategies_[i]->getStrategyType();
        if (i < strategies_.size() - 1) sList << ", ";
      }
      *l_out << "  Strategy List = " << sList.str() << std::endl;

      for (auto& s : strategies_) s->describe(*l_out, verbLevel);

      *l_out << std::string(this->description().length() + 8, '-') << std::endl;
    }
  }

  void addStrategy(
      const Teuchos::RCP<TimeStepControlStrategy<Scalar>>& strategy)
  {
    if (Teuchos::nonnull(strategy)) {
      if (this->size() == 0) this->setStepType(strategy->getStepType());

      TEUCHOS_TEST_FOR_EXCEPTION(
          this->getStepType() != strategy->getStepType(), std::logic_error,
          "Error - Cannot mix 'Constant' and 'Variable' step types.\n"
          "strategies in composite!  Need at least one.\n");

      strategies_.push_back(strategy);
    }
  }

  virtual int size() const { return strategies_.size(); }

  virtual std::vector<Teuchos::RCP<TimeStepControlStrategy<Scalar>>>
  getStrategies() const
  {
    return strategies_;
  }

  void clearStrategies() { strategies_.clear(); }

  virtual void initialize() const override
  {
    TEUCHOS_TEST_FOR_EXCEPTION(
        strategies_.size() == 0, std::logic_error,
        "Error - No strategies in composite!  Need at least one.\n");

    for (auto& s : strategies_) s->initialize();

    auto strategy0 = strategies_[0];
    for (auto& s : strategies_) {
      TEUCHOS_TEST_FOR_EXCEPTION(s->isInitialized() == false, std::logic_error,
                                 "Error - Composite strategy, "
                                     << s->getName()
                                     << " is not initialized!\n");

      if (strategy0->getStepType() != s->getStepType()) {
        std::ostringstream msg;
        msg << "Error - All the Strategy Step Types must match.\n";
        for (std::size_t i = 0; i < strategies_.size(); ++i) {
          msg << "  Strategy[" << i << "] = " << strategies_[i]->getStepType()
              << "  (" << strategies_[i]->getName() << ")\n";
        }
        TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error, msg.str());
      }
    }

    this->isInitialized_ = true;  // Only place where this is set to true!
  }

  virtual Teuchos::RCP<const Teuchos::ParameterList> getValidParameters()
      const override
  {
    Teuchos::RCP<Teuchos::ParameterList> pl =
        Teuchos::parameterList("Time Step Control Strategy");

    pl->set<std::string>("Strategy Type", this->getStrategyType(), "Composite");

    std::stringstream sList;
    for (std::size_t i = 0; i < strategies_.size(); ++i) {
      sList << strategies_[i]->getStrategyType();
      if (i < strategies_.size() - 1) sList << ", ";
    }
    pl->set<std::string>("Strategy List", sList.str());

    for (auto& s : strategies_) pl->set(s->getName(), *s->getValidParameters());

    return pl;
  }

 private:
  std::vector<Teuchos::RCP<TimeStepControlStrategy<Scalar>>> strategies_;
};

// Nonmember constructor - ParameterList
// ------------------------------------------------------------------------
template <class Scalar>
Teuchos::RCP<TimeStepControlStrategyComposite<Scalar>>
createTimeStepControlStrategyComposite(
    Teuchos::RCP<Teuchos::ParameterList> const& pList,
    std::string name = "Composite")
{
  using Teuchos::ParameterList;
  using Teuchos::RCP;

  std::vector<std::string> tscsList;

  TEUCHOS_TEST_FOR_EXCEPTION(
      pList->get<std::string>("Strategy Type") != "Composite", std::logic_error,
      "Error - Strategy Type != 'Composite'.  (='" +
          pList->get<std::string>("Strategy Type") + "')\n");

  // string tokenizer
  tscsList.clear();
  std::string str = pList->get<std::string>("Strategy List");
  std::string delimiters(",");
  const char* WhiteSpace = " \t\v\r\n";
  // Skip delimiters at the beginning
  std::string::size_type lastPos = str.find_first_not_of(delimiters, 0);
  // Find the first delimiter
  std::string::size_type pos = str.find_first_of(delimiters, lastPos);
  while ((pos != std::string::npos) || (lastPos != std::string::npos)) {
    // Found a token, add it to the vector
    std::string token = str.substr(lastPos, pos - lastPos);

    std::size_t start = token.find_first_not_of(WhiteSpace);
    std::size_t end   = token.find_last_not_of(WhiteSpace);
    token =
        (start == end ? std::string() : token.substr(start, end - start + 1));

    tscsList.push_back(token);
    if (pos == std::string::npos) break;

    lastPos = str.find_first_not_of(delimiters, pos);  // Skip delimiters
    pos     = str.find_first_of(delimiters, lastPos);  // Find next delimiter
  }

  auto tscsc = Teuchos::rcp(new TimeStepControlStrategyComposite<Scalar>());

  // For each sublist name tokenized, add the TSCS
  for (auto tscsName : tscsList) {
    RCP<ParameterList> pl =
        Teuchos::rcp(new ParameterList(pList->sublist(tscsName, true)));

    auto strategyType = pl->get<std::string>("Strategy Type", "Unknown");
    if (strategyType == "Constant") {
      tscsc->addStrategy(
          createTimeStepControlStrategyConstant<Scalar>(pl, tscsName));
    }
    else if (strategyType == "Basic VS") {
      tscsc->addStrategy(
          createTimeStepControlStrategyBasicVS<Scalar>(pl, tscsName));
    }
    else if (strategyType == "Integral Controller") {
      tscsc->addStrategy(
          createTimeStepControlStrategyIntegralController<Scalar>(pl,
                                                                  tscsName));
    }
    else if (strategyType == "Composite") {
      tscsc->addStrategy(
          createTimeStepControlStrategyComposite<Scalar>(pl, tscsName));
    }
    else {
      RCP<Teuchos::FancyOStream> out =
          Teuchos::fancyOStream(Teuchos::rcpFromRef(std::cout));
      out->setOutputToRootOnly(0);
      Teuchos::OSTab ostab(out, 1, "createTimeStepControlStrategyComposite()");
      *out << "Warning -- Unknown strategy type!\n"
           << "'Strategy Type' = '" << strategyType << "'\n"
           << "Should call addStrategy() with this\n"
           << "(app-specific?) strategy, and initialize().\n"
           << std::endl;
    }
  }

  tscsc->setName(name);

  if (tscsc->size() == 0) {
    RCP<Teuchos::FancyOStream> out =
        Teuchos::fancyOStream(Teuchos::rcpFromRef(std::cout));
    out->setOutputToRootOnly(0);
    Teuchos::OSTab ostab(out, 1, "createTimeStepControlStrategyComposite()");
    *out << "Warning -- Did not find a Tempus strategy to create!\n"
         << "Should call addStrategy() with (app-specific?) strategy(ies),\n"
         << "and initialize().\n"
         << std::endl;
  }
  else {
    tscsc->initialize();
  }

  return tscsc;
}

template <class Scalar>
Teuchos::RCP<Teuchos::ParameterList> getTimeStepControlStrategyCompositePL()
{
  auto t    = rcp(new Tempus::TimeStepControlStrategyComposite<Scalar>());
  auto tscs = rcp(new Tempus::TimeStepControlStrategyConstant<Scalar>());
  t->addStrategy(tscs);
  return Teuchos::rcp_const_cast<Teuchos::ParameterList>(
      t->getValidParameters());
}

}  // namespace Tempus
#endif  // Tempus_TimeStepControlStrategy_hpp