adding "z-sublayers" input param in plane id generator, refs idaholab…

…#19217
yjung-anl · Nov 1, 2021 · beea3a5 · beea3a5
1 parent df40560
commit beea3a5
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/modules/reactor/src/meshgenerators/PlaneIDGenerator.C b/modules/reactor/src/meshgenerators/PlaneIDGenerator.C
@@ -23,6 +23,8 @@ PlaneIDGenerator::validParams()
 
   params.addRequiredParam<MeshGeneratorName>("input", "The mesh we want to modify");
   params.addRequiredParam<std::vector<Real>>("z_layers", "Z coordinates of axial layers");
+  params.addParam<std::vector<unsigned int>>("z_sublayers",
+                                             "The number of layers for each z-layers");
   params.addRequiredParam<std::string>("id_name", "Name of Integer ID set");
   params.addClassDescription(
       "This PlaneIDGenerator source code is to assigns plane reporting ID for existing 3D meshes");
@@ -33,9 +35,29 @@ PlaneIDGenerator::validParams()
 PlaneIDGenerator::PlaneIDGenerator(const InputParameters & parameters)
   : MeshGenerator(parameters),
     _input(getMesh("input")),
-    _layers(getParam<std::vector<Real>>("z_layers")),
     _element_id_name(getParam<std::string>("id_name"))
 {
+  if (!isParamValid("z_sublayers"))
+    _layers = getParam<std::vector<Real>>("z_layers");
+  else
+  {
+    _layers.clear();
+    std::vector<Real> base_z_layers = getParam<std::vector<Real>>("z_layers");
+    std::vector<unsigned int> z_sublayers = getParam<std::vector<unsigned int>>("z_sublayers");
+    if (base_z_layers.size() != z_sublayers.size() + 1)
+      mooseError("Sizes of 'z_layers' and 'z_sublayers' disagree");
+    _layers.push_back(base_z_layers[0]);
+    for (unsigned int i = 0; i < z_sublayers.size(); ++i)
+    {
+      Real layer_size = (base_z_layers[i + 1] - base_z_layers[i]) / (Real)z_sublayers[i];
+      for (unsigned int j = 0; j < z_sublayers[i]; ++j)
+        _layers.push_back(_layers.back() + layer_size);
+    }
+    for (auto z : _layers)
+    {
+      _console << z << std::endl;
+    }
+  }
 }
 
 std::unique_ptr<MeshBase>

diff --git a/...les/reactor/test/tests/meshgenerators/reporting_id/plane_id/gold/plane_reporting_id_out.e b/...les/reactor/test/tests/meshgenerators/reporting_id/plane_id/gold/plane_reporting_id_out.e
diff --git a/modules/reactor/test/tests/meshgenerators/reporting_id/plane_id/plane_reporting_id.i b/modules/reactor/test/tests/meshgenerators/reporting_id/plane_id/plane_reporting_id.i
@@ -14,12 +14,13 @@
     input = 'pin2d'
     heights = '5.0 5.0 5.0'
     direction = '0 0 1'
-    num_layers = '1 1 1'
+    num_layers = '2 2 2'
   []
   [pin3d_id]
     type = PlaneIDGenerator
     input = 'pin3d'
     z_layers = '0.0 5.0 10.0 15.0'
+    z_sublayers = ' 1 2 1'
     id_name = 'plane_id'
   []
 []