SurrogateModelInterface.C

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

#include "SurrogateModelInterface.h"
#include "SurrogateTrainer.h"
#include "SurrogateModel.h"
#include "SubProblem.h"
#include "MooseTypes.h"

InputParameters
SurrogateModelInterface::validParams()
{
  return emptyInputParameters();
}

SurrogateModelInterface::SurrogateModelInterface(const MooseObject * moose_object)
  : _smi_params(moose_object->parameters()),
    _smi_feproblem(*_smi_params.get<FEProblemBase *>("_fe_problem_base")),
    _smi_tid(_smi_params.have_parameter<THREAD_ID>("_tid") ? _smi_params.get<THREAD_ID>("_tid") : 0)
{
}

template <>
SurrogateModel &
SurrogateModelInterface::getSurrogateModelByName(const UserObjectName & name) const
{
  std::vector<SurrogateModel *> models;
  _smi_feproblem.theWarehouse()
      .query()
      .condition<AttribName>(name)
      .condition<AttribSystem>("SurrogateModel")
      .queryInto(models);
  if (models.empty())
    mooseError("Unable to find a SurrogateModel object with the name '" + name + "'");
  return *(models[0]);
}

template <>
SurrogateModel &
SurrogateModelInterface::getSurrogateModel(const std::string & name) const
{
  return getSurrogateModelByName<SurrogateModel>(_smi_params.get<UserObjectName>(name));
}

template <>
SurrogateTrainerBase &
SurrogateModelInterface::getSurrogateTrainerByName(const UserObjectName & name) const
{
  return _smi_feproblem.getUserObject<SurrogateTrainerBase>(name);
}

template <>
SurrogateTrainerBase &
SurrogateModelInterface::getSurrogateTrainer(const std::string & name) const
{
  return getSurrogateTrainerByName<SurrogateTrainerBase>(_smi_params.get<UserObjectName>(name));
}