AddGeochemicalModelInterrogatorAction.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 "AddGeochemicalModelInterrogatorAction.h"
#include "GeochemicalModelInterrogator.h"
#include "Executioner.h"
#include "FEProblem.h"

registerMooseAction("GeochemistryApp", AddGeochemicalModelInterrogatorAction, "setup_mesh");
registerMooseAction("GeochemistryApp", AddGeochemicalModelInterrogatorAction, "init_mesh");
registerMooseAction("GeochemistryApp", AddGeochemicalModelInterrogatorAction, "create_problem");
registerMooseAction("GeochemistryApp", AddGeochemicalModelInterrogatorAction, "setup_executioner");
registerMooseAction("GeochemistryApp", AddGeochemicalModelInterrogatorAction, "add_output");

InputParameters
AddGeochemicalModelInterrogatorAction::validParams()
{
  InputParameters params = Action::validParams();
  params += GeochemicalModelInterrogator::sharedParams();
  params.addClassDescription("Action that sets up the geochemical model interrogator");

  return params;
}

AddGeochemicalModelInterrogatorAction::AddGeochemicalModelInterrogatorAction(
    const InputParameters & params)
  : Action(params)
{
}

void
AddGeochemicalModelInterrogatorAction::act()
{
  // Set up an arbitrary mesh
  if (_current_task == "setup_mesh")
  {
    const std::string class_name = "GeneratedMesh";
    InputParameters params = _factory.getValidParams(class_name);
    params.set<MooseEnum>("dim") = "1";
    _mesh = _factory.create<MooseMesh>(class_name, "mesh", params);
  }
  // Initialize the arbitrary mesh
  else if (_current_task == "init_mesh")
  {
    _mesh->init();
  }
  // Create a "solve=false" FEProblem
  else if (_current_task == "create_problem")
  {
    const std::string class_name = "FEProblem";
    InputParameters params = _factory.getValidParams(class_name);
    params.set<MooseMesh *>("mesh") = _mesh.get();
    params.set<bool>("use_nonlinear") = true;
    params.set<bool>("solve") = false;
    _problem = _factory.create<FEProblemBase>(class_name, "Problem", params);
    _problem->setKernelCoverageCheck(FEProblemBase::CoverageCheckMode::FALSE);
  }
  // Set up an arbitrary steady executioner
  else if (_current_task == "setup_executioner")
  {
    const std::string class_name = "Steady";
    InputParameters params = _factory.getValidParams(class_name);
    params.set<FEProblemBase *>("_fe_problem_base") = _problem.get();
    params.set<FEProblem *>("_fe_problem") = (std::dynamic_pointer_cast<FEProblem>(_problem)).get();
    std::shared_ptr<Executioner> executioner =
        _factory.create<Executioner>(class_name, "Executioner", params);
    _app.setExecutioner(std::move(executioner));
  }
  // Create a console that executes only on FINAL and does not print system info
  else if (_current_task == "add_output")
  {
    const std::string class_name = "GeochemicalModelInterrogator";
    auto params = _factory.getValidParams(class_name);
    params.set<UserObjectName>("model_definition") = getParam<UserObjectName>("model_definition");
    // Only pass parameters that were supplied to this action
    if (isParamValid("swap_out_of_basis"))
      params.set<std::vector<std::string>>("swap_out_of_basis") =
          getParam<std::vector<std::string>>("swap_out_of_basis");
    if (isParamValid("swap_into_basis"))
      params.set<std::vector<std::string>>("swap_into_basis") =
          getParam<std::vector<std::string>>("swap_into_basis");
    if (isParamValid("activity_species"))
      params.set<std::vector<std::string>>("activity_species") =
          getParam<std::vector<std::string>>("activity_species");
    if (isParamValid("activity_values"))
      params.set<std::vector<Real>>("activity_values") =
          getParam<std::vector<Real>>("activity_values");
    params.set<unsigned int>("precision") = getParam<unsigned int>("precision");
    params.set<std::string>("equilibrium_species") = getParam<std::string>("equilibrium_species");
    params.set<MooseEnum>("interrogation") = getParam<MooseEnum>("interrogation");
    params.set<Real>("temperature") = getParam<Real>("temperature");
    params.set<Real>("stoichiometry_tolerance") = getParam<Real>("stoichiometry_tolerance");
    _problem->addOutput(class_name, "geochemical_model_interrogator", params);
  }
}