LoadModelDataAction.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 "Action.h"
#include "Attributes.h"
#include "MooseApp.h"
#include "RestartableDataReader.h"
#include "RestartableModelInterface.h"
 
template <typename T>
class LoadModelDataAction : public Action
{
public:
  static InputParameters validParams();
 
  LoadModelDataAction(const InputParameters & params) : Action(params) {}
 
  virtual void act() override;
 
private:
  void load(const T & object);
};
 
template <typename T>
InputParameters
LoadModelDataAction<T>::validParams()
{
  return Action::validParams();
}
 
template <typename T>
void
LoadModelDataAction<T>::act()
{
  static_assert(std::is_base_of<RestartableModelInterface, T>::value,
                "You must derive from RestartableModelInterface to use this action");
 
  // We fetch the mapping objects and then load the necessary data
  std::vector<T *> objects;
  static const auto attribute_name = T::validParams().getSystemAttributeName();
 
  _app.theWarehouse().query().template condition<AttribSystem>(attribute_name).queryInto(objects);
  for (auto object_ptr : objects)
    if (object_ptr->hasModelData())
      load(*object_ptr);
}
 
template <typename T>
void
LoadModelDataAction<T>::load(const T & object)
{
  if (object.hasExplicitModelData())
  {
    // Create the object that will load in data
    RestartableDataReader reader(
        _app, _app.getRestartableDataMap(object.modelMetaDataName()), _app.forceRestart());
    reader.setErrorOnLoadWithDifferentNumberOfProcessors(false);
 
    // Read the supplied file
    const std::string filename = object.getModelDataFileName();
    try
    {
      reader.setInput(filename);
      reader.restore();
    }
    catch (...)
    {
      paramError("filename", "The supplied file '", filename, "' failed to load.");
    }
  }
  else if (_app.isRecovering())
    // Restore model data from the recover checkpoint into the already-declared map
    _app.possiblyLoadRestartableMetaData(object.modelMetaDataName(),
                                         _app.getRestartRecoverFileBase() +
                                             MooseApp::checkpointSuffix());
}