ChainControlDataSystem.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include "ChainControlData.h"
#include "MooseError.h"
#include "MooseUtils.h"

class ChainControl;

class ChainControlDataSystem
{
public:
  ChainControlDataSystem(MooseApp & app);

  bool hasChainControlData(const std::string & data_name) const;

  template <typename T>
  bool hasChainControlDataOfType(const std::string & data_name) const;

  template <typename T>
  ChainControlData<T> & getChainControlData(const std::string & data_name);

  template <typename T>
  ChainControlData<T> & declareChainControlData(const std::string & data_name,
                                                ChainControl & chain_control);

  void copyValuesBack();

  const std::map<std::string, std::unique_ptr<ChainControlDataBase>> &
  getChainControlDataMap() const;

  std::string outputChainControlMap() const;

private:
  MooseApp & _app;

  std::map<std::string, std::unique_ptr<ChainControlDataBase>> _chain_control_data_map;
};

template <typename T>
bool
ChainControlDataSystem::hasChainControlDataOfType(const std::string & data_name) const
{
  if (hasChainControlData(data_name))
    return dynamic_cast<ChainControlData<T> *>(_chain_control_data_map.at(data_name).get()) !=
           nullptr;
  else
    return false;
}

template <typename T>
ChainControlData<T> &
ChainControlDataSystem::getChainControlData(const std::string & data_name)
{
  if (hasChainControlData(data_name))
  {
    if (!hasChainControlDataOfType<T>(data_name))
      mooseError("The chain control data '",
                 data_name,
                 "' was requested with the type '",
                 MooseUtils::prettyCppType<T>(),
                 "' but has the type '",
                 _chain_control_data_map[data_name]->type(),
                 "'.");
  }
  else
    _chain_control_data_map[data_name] = std::make_unique<ChainControlData<T>>(_app, data_name);

  return static_cast<ChainControlData<T> &>(*_chain_control_data_map[data_name]);
}

template <typename T>
ChainControlData<T> &
ChainControlDataSystem::declareChainControlData(const std::string & data_name,
                                                ChainControl & chain_control)
{
  auto & data = getChainControlData<T>(data_name);
  if (!data.getDeclared())
  {
    data.setDeclared();
    data.setChainControl(chain_control);
  }
  else
    mooseError("The chain control data '", data_name, "' has already been declared.");

  return data;
}