ImageSubdomainGenerator.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 "ImageSubdomainGenerator.h"
#include "CastUniquePointer.h"
 
#include "libmesh/elem.h"
 
// provides round, not std::round (see http://www.cplusplus.com/reference/cmath/round/)
#include <cmath>
 
registerMooseObject("MooseApp", ImageSubdomainGenerator);
 
InputParameters
ImageSubdomainGenerator::validParams()
{
  InputParameters params = MeshGenerator::validParams();
  params.addClassDescription("Samples an image at the coordinates of each element centroid, using "
                             "the resulting pixel color value as each element's subdomain ID");
  params += MeshBaseImageSampler::validParams();
 
  params.addRequiredParam<MeshGeneratorName>("input", "The mesh we want to modify");
 
  return params;
}
 
ImageSubdomainGenerator::ImageSubdomainGenerator(const InputParameters & parameters)
  : MeshGenerator(parameters), MeshBaseImageSampler(parameters), _input(getMesh("input"))
{
}
 
std::unique_ptr<MeshBase>
ImageSubdomainGenerator::generate()
{
  std::unique_ptr<MeshBase> mesh = std::move(_input);
 
  // Initialize the ImageSampler
  setupImageSampler(*mesh);
 
  // Loop over the elements and sample the image at the element centroid and use the value for the
  // subdomain id
  for (auto & elem : mesh->active_element_ptr_range())
  {
    subdomain_id_type id = static_cast<subdomain_id_type>(round(sample(elem->vertex_average())));
    elem->subdomain_id() = id;
  }
 
  return dynamic_pointer_cast<MeshBase>(mesh);
}