Line data    Source code

       1             : //* This file is part of the MOOSE framework
       2             : //* https://mooseframework.inl.gov
       3             : //*
       4             : //* All rights reserved, see COPYRIGHT for full restrictions
       5             : //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
       6             : //*
       7             : //* Licensed under LGPL 2.1, please see LICENSE for details
       8             : //* https://www.gnu.org/licenses/lgpl-2.1.html
       9             : 
      10             : #include "StochasticToolsApp.h"
      11             : #include "Moose.h"
      12             : #include "AppFactory.h"
      13             : #include "MooseSyntax.h"
      14             : 
      15             : InputParameters
      16       42865 : StochasticToolsApp::validParams()
      17             : {
      18       42865 :   InputParameters params = MooseApp::validParams();
      19             : 
      20       42865 :   params.set<bool>("use_legacy_material_output") = false;
      21       42865 :   params.set<bool>("use_legacy_initial_residual_evaluation_behavior") = false;
      22             : 
      23       42865 :   return params;
      24           0 : }
      25             : 
      26             : registerKnownLabel("StochasticToolsApp");
      27             : 
      28           0 : StochasticToolsApp::StochasticToolsApp(const InputParameters & parameters) : MooseApp(parameters)
      29             : {
      30           0 :   StochasticToolsApp::registerAll(_factory, _action_factory, _syntax);
      31           0 : }
      32             : 
      33           0 : StochasticToolsApp::~StochasticToolsApp() {}
      34             : 
      35             : void
      36       42861 : StochasticToolsApp::registerAll(Factory & f, ActionFactory & af, Syntax & syntax)
      37             : {
      38       42861 :   Registry::registerObjectsTo(f, {"StochasticToolsApp"});
      39       42861 :   Registry::registerActionsTo(af, {"StochasticToolsApp"});
      40             : 
      41             :   // Adds [Trainers] block
      42       85722 :   registerSyntaxTask("AddSurrogateAction", "Trainers/*", "add_trainer");
      43       85722 :   registerMooseObjectTask("add_trainer", SurrogateTrainer, false);
      44       85722 :   addTaskDependency("add_trainer", "add_sampler");
      45             : 
      46             :   // Adds [Surrogates] block
      47       85722 :   registerSyntaxTask("AddSurrogateAction", "Surrogates/*", "add_surrogate");
      48       85722 :   registerMooseObjectTask("add_surrogate", SurrogateModel, false);
      49       85722 :   addTaskDependency("add_surrogate", "add_trainer");
      50             : 
      51             :   // Adds action for loading Surrogate data
      52       42861 :   registerTask("load_surrogate_data", true);
      53       85722 :   addTaskDependency("load_surrogate_data", "add_surrogate");
      54             : 
      55             :   // Adds action for loading mapping data
      56       42861 :   registerTask("load_mapping_data", true);
      57       85722 :   addTaskDependency("load_mapping_data", "add_variable_mapping");
      58             : 
      59             :   // General StochasticTools action
      60       42861 :   registerTask("auto_create_mesh", false);
      61       42861 :   registerTask("auto_create_problem", false);
      62       42861 :   registerTask("auto_create_executioner", false);
      63       85722 :   registerSyntaxTask("StochasticToolsAction", "StochasticTools", "auto_create_mesh");
      64       85722 :   registerSyntaxTask("StochasticToolsAction", "StochasticTools", "auto_create_problem");
      65       85722 :   registerSyntaxTask("StochasticToolsAction", "StochasticTools", "auto_create_executioner");
      66             : 
      67             :   // StochasticResults
      68       42861 :   registerTask("declare_stochastic_results_vectors", true);
      69       85722 :   addTaskDependency("declare_stochastic_results_vectors", "add_vector_postprocessor");
      70       85722 :   addTaskDependency("add_reporter", "declare_stochastic_results_vectors");
      71             : 
      72             :   // Covariance functions (Gaussian Process)
      73       85722 :   registerSyntaxTask("AddCovarianceAction", "Covariance/*", "add_covariance");
      74       85722 :   registerMooseObjectTask("add_covariance", CovarianceFunctionBase, false);
      75       85722 :   addTaskDependency("add_covariance", "add_user_object");
      76       85722 :   addTaskDependency("add_distribution", "add_covariance");
      77             :   // Mapping objects
      78       85722 :   registerSyntaxTask("AddVariableMappingAction", "VariableMappings/*", "add_variable_mapping");
      79       85722 :   registerMooseObjectTask("add_variable_mapping", VariableMappingBase, false);
      80       85722 :   addTaskDependency("add_variable_mapping", "add_reporter");
      81             :   // Adds action for loading Covariance data in model
      82       42861 :   registerTask("load_covariance_data", true);
      83       85722 :   addTaskDependency("load_covariance_data", "load_surrogate_data");
      84       85722 :   addTaskDependency("setup_function_complete", "load_covariance_data");
      85       85722 :   addTaskDependency("setup_mesh", "auto_create_mesh");
      86       85722 :   addTaskDependency("create_problem", "auto_create_problem");
      87       85722 :   addTaskDependency("setup_executioner", "auto_create_executioner");
      88             :   // Likelihood functions (Bayesian inference)
      89       85722 :   registerSyntaxTask("AddLikelihoodAction", "Likelihood/*", "add_likelihood");
      90       85722 :   registerMooseObjectTask("add_likelihood", LikelihoodFunctionBase, false);
      91       85722 :   addTaskDependency("add_likelihood", "add_user_object");
      92       85722 :   addTaskDependency("add_distribution", "add_likelihood");
      93             : 
      94       85722 :   registerSyntaxTask("AdaptiveSamplerAction", "Samplers", "add_user_object");
      95       85722 :   registerSyntaxTask("AdaptiveSamplerAction", "Samplers", "add_postprocessor");
      96             : 
      97             :   // Adds [ParameterStudy] block
      98       85722 :   registerSyntax("ParameterStudyAction", "ParameterStudy");
      99       42861 : }
     100             : 
     101             : void
     102        7383 : StochasticToolsApp::registerApps()
     103             : {
     104        7383 :   registerApp(StochasticToolsApp);
     105        7383 : }
     106             : 
     107             : void
     108          40 : StochasticToolsApp::requiresTorch(const MooseObject &
     109             : #ifndef MOOSE_LIBTORCH_ENABLED
     110             :                                       obj
     111             : #endif
     112             : )
     113             : {
     114             : #ifndef MOOSE_LIBTORCH_ENABLED
     115           0 :   obj.mooseError("PyTorch C++ API (libtorch) must be installed to use this object, see "
     116             :                  "https://mooseframework.inl.gov/modules/stochastic_tools/install_pytorch.html for "
     117             :                  "instruction.");
     118             : #endif
     119          40 : }
     120             : 
     121             : extern "C" void
     122           0 : StochasticToolsApp__registerAll(Factory & f, ActionFactory & af, Syntax & s)
     123             : {
     124           0 :   StochasticToolsApp::registerAll(f, af, s);
     125           0 : }
     126             : extern "C" void
     127           0 : StochasticToolsApp__registerApps()
     128             : {
     129           0 :   StochasticToolsApp::registerApps();
     130           0 : }