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 "ADWeightedAverageMaterial.h"
      11             : 
      12             : registerMooseObject("ThermalHydraulicsApp", ADWeightedAverageMaterial);
      13             : 
      14             : InputParameters
      15         206 : ADWeightedAverageMaterial::validParams()
      16             : {
      17         206 :   InputParameters params = Material::validParams();
      18             : 
      19         206 :   params.addClassDescription("Weighted average of material properties using variables as weights");
      20             : 
      21         412 :   params.addRequiredParam<MaterialPropertyName>(
      22             :       "prop_name", "The name of the material property where the average is stored");
      23         412 :   params.addRequiredParam<std::vector<MaterialPropertyName>>("values",
      24             :                                                              "Vector of values to average");
      25         412 :   params.addRequiredCoupledVar("weights", "Vector of weights for each value");
      26             : 
      27         206 :   return params;
      28           0 : }
      29             : 
      30         159 : ADWeightedAverageMaterial::ADWeightedAverageMaterial(const InputParameters & parameters)
      31             :   : Material(parameters),
      32         159 :     _prop(declareADProperty<Real>(getParam<MaterialPropertyName>("prop_name"))),
      33         477 :     _n_values(getParam<std::vector<MaterialPropertyName>>("values").size())
      34             : {
      35             :   // make sure that number of weights equals the number of values
      36         159 :   if (coupledComponents("weights") != _n_values)
      37           0 :     mooseError(name(), ": The number of weights must equal the number of values");
      38             : 
      39             :   // get all of the variable values
      40             :   const std::vector<MaterialPropertyName> & prop_names =
      41         159 :       getParam<std::vector<MaterialPropertyName>>("values");
      42         477 :   for (unsigned int i = 0; i < _n_values; i++)
      43             :   {
      44         318 :     _values.push_back(&getADMaterialPropertyByName<Real>(prop_names[i]));
      45         636 :     _weights.push_back(&adCoupledValue("weights", i));
      46             :   }
      47         159 : }
      48             : 
      49             : void
      50       51979 : ADWeightedAverageMaterial::computeQpProperties()
      51             : {
      52       51979 :   ADReal weight_total = 0;
      53      155937 :   for (unsigned int i = 0; i < _n_values; i++)
      54      103958 :     weight_total += (*(_weights[i]))[_qp];
      55             : 
      56       51979 :   ADReal weighted_sum = 0;
      57      155937 :   for (unsigned int i = 0; i < _n_values; i++)
      58      207916 :     weighted_sum += (*(_weights[i]))[_qp] * (*(_values[i]))[_qp];
      59             : 
      60       51979 :   _prop[_qp] = weighted_sum / weight_total;
      61       51979 : }