EBGeometry_VTK.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Robert Marskar <robert.marskar@sintef.no>
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef EBGEOMETRY_VTK_HPP
#define EBGEOMETRY_VTK_HPP
 
// Std includes
#include <cstddef>
#include <map>
#include <memory>
#include <string>
#include <type_traits>
#include <vector>
 
// Our includes
#include "EBGeometry_DCEL.hpp"
#include "EBGeometry_Vec.hpp"
 
namespace EBGeometry {
 
template <typename T>
class VTK
{
  static_assert(std::is_floating_point_v<T>, "VTK<T>: T must be a floating-point type");
 
public:
  VTK() noexcept = default;
 
  VTK(const std::string& a_id) noexcept;
 
  VTK(const VTK& a_other) = default;
 
  VTK(VTK&& a_other) noexcept = default;
 
  VTK&
  operator=(const VTK& a_other) = default;
 
  VTK&
  operator=(VTK&& a_other) noexcept = default;
 
  ~VTK() noexcept = default;
 
  [[nodiscard]] std::string&
  getID() noexcept;
 
  [[nodiscard]] const std::string&
  getID() const noexcept;
 
  [[nodiscard]] std::vector<Vec3T<T>>&
  getVertexCoordinates() noexcept;
 
  [[nodiscard]] const std::vector<Vec3T<T>>&
  getVertexCoordinates() const noexcept;
 
  [[nodiscard]] std::vector<std::vector<size_t>>&
  getFacets() noexcept;
 
  [[nodiscard]] const std::vector<std::vector<size_t>>&
  getFacets() const noexcept;
 
  [[nodiscard]] std::vector<T>&
  getPointDataScalars(const std::string a_name);
 
  [[nodiscard]] const std::vector<T>&
  getPointDataScalars(const std::string a_name) const;
 
  [[nodiscard]] std::vector<T>&
  getCellDataScalars(const std::string a_name);
 
  [[nodiscard]] const std::vector<T>&
  getCellDataScalars(const std::string a_name) const;
 
  void
  setPointDataScalars(const std::string a_name, std::vector<T> a_data);
 
  void
  setCellDataScalars(const std::string a_name, std::vector<T> a_data);
 
  template <typename Meta>
  [[nodiscard]] std::shared_ptr<EBGeometry::DCEL::MeshT<T, Meta>>
  convertToDCEL() const noexcept;
 
protected:
  std::string m_id;
 
  std::vector<Vec3T<T>> m_vertexCoordinates;
 
  std::vector<std::vector<size_t>> m_facets;
 
  std::map<std::string, std::vector<T>> m_pointDataScalars;
 
  std::map<std::string, std::vector<T>> m_cellDataScalars;
};
 
} // namespace EBGeometry
 
#include "EBGeometry_VTKImplem.hpp"
 
#endif