gmsh_reader.h

/***************************************************************************
 * LehrFEM++ - A simple C++ finite element library for teaching
 * Developed from 2018 at the Seminar of Applied Mathematics of ETH Zurich,
 * lead developers Dr. R. Casagrande and Prof. R. Hiptmair
 ***************************************************************************/
 
#ifndef INCG7fedf7cf1a0246a98b2bf431cfa34da2
#define INCG7fedf7cf1a0246a98b2bf431cfa34da2
#include <lf/mesh/mesh.h>
#include <lf/mesh/utils/utils.h>
 
#include <map>
#include <string>
#include <variant>
#include <vector>
 
#include "gmsh_file_v2.h"
#include "gmsh_file_v4.h"
 
namespace lf::io {
 
class GmshReader {
 public:
  using size_type = mesh::Mesh::size_type;
  using dim_t = base::RefEl::dim_t;
  using GmshFileVariant = std::variant<GMshFileV2, GMshFileV4>;
 
  [[nodiscard]] std::shared_ptr<mesh::Mesh> mesh() { return mesh_; }
 
  [[nodiscard]] std::shared_ptr<const mesh::Mesh> mesh() const { return mesh_; }
 
  [[nodiscard]] size_type PhysicalEntityName2Nr(const std::string& name,
                                                dim_t codim = -1) const;
 
  [[nodiscard]] std::string PhysicalEntityNr2Name(size_type number,
                                                  dim_t codim = -1) const;
 
  [[nodiscard]] std::vector<std::pair<size_type, std::string>> PhysicalEntities(
      dim_t codim) const;
 
  [[nodiscard]] std::vector<size_type> PhysicalEntityNr(
      const mesh::Entity& e) const;
 
  [[nodiscard]] bool IsPhysicalEntity(const mesh::Entity& e,
                                      size_type physical_entity_nr) const;
 
  GmshReader(std::unique_ptr<mesh::MeshFactory> factory,
             const GmshFileVariant& msh_file);
 
  GmshReader(std::unique_ptr<mesh::MeshFactory> factory,
             const std::string& filename);
 
 private:
  std::shared_ptr<mesh::Mesh> mesh_;
 
  std::unique_ptr<mesh::MeshFactory> mesh_factory_;
 
  std::shared_ptr<mesh::utils::AllCodimMeshDataSet<std::vector<size_type>>>
      physical_nrs_;
 
  std::multimap<std::string, std::pair<size_type, dim_t>> name_2_nr_;
 
  std::multimap<size_type, std::pair<std::string, dim_t>> nr_2_name_;
 
  void InitGmshFile(const GMshFileV2& msh_file);
  void InitGmshFile(const GMshFileV4& msh_file);
};
 
std::variant<GMshFileV2, GMshFileV4> ReadGmshFile(const std::string& filename);
 
}  // namespace lf::io
 
#endif  // INCG7fedf7cf1a0246a98b2bf431cfa34da2