MeshExtruderGenerator.h

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

#pragma once

#include "MeshGenerator.h"

#include "libmesh/mesh_generation.h"

class MeshExtruderGenerator : public MeshGenerator
{
public:
  static InputParameters validParams();

  MeshExtruderGenerator(const InputParameters & parameters);

  std::unique_ptr<MeshBase> generate() override;

protected:
  std::unique_ptr<MeshBase> & _input;

  const RealVectorValue _extrusion_vector;
  unsigned int _num_layers;
  std::vector<SubdomainID> _existing_subdomains;
  std::vector<unsigned int> _layers;
  std::vector<unsigned int> _new_ids;

  class QueryElemSubdomainID : public MeshTools::Generation::QueryElemSubdomainIDBase
  {
  public:
    QueryElemSubdomainID(const std::vector<SubdomainID> & existing_subdomains,
                         std::vector<unsigned int> layers,
                         const std::vector<unsigned int> & new_ids,
                         unsigned int num_layers);

    virtual subdomain_id_type get_subdomain_for_layer(const Elem * old_elem, unsigned int layer);

  private:
    std::map<unsigned int, std::map<SubdomainID, unsigned int>> _layer_data;

    unsigned int _num_layers;
  };

private:
  void changeID(MeshBase & mesh, const std::vector<BoundaryName> & names, BoundaryID old_id);
};