PolyscopeViewer.h

#pragma once

 
#if !defined(DGTAL_WITH_POLYSCOPE_VIEWER)
#error DGTAL_WITH_POLYSCOPE_VIEWER flag not set during the cmake step
#endif
 
#include "DGtal/io/Display3D.h"
 
#include "polyscope/polyscope.h"
 
#include "polyscope/sparse_volume_grid.h"
#include "polyscope/curve_network.h"
#include "polyscope/surface_mesh.h"
#include "polyscope/volume_mesh.h"
#include "polyscope/volume_grid.h"
#include "polyscope/point_cloud.h"
#include "polyscope/pick.h"
 
namespace DGtal {
  namespace drawutils {
    // @brief Convert a DGtal color to a glm color
    glm::vec3 toglm(const DGtal::Color& col);
 
    // @brief Convert a vector of DGtal color to a vector of glm color
    std::vector<glm::vec3> toglm(const std::vector<DGtal::Color>& col);
 
    // @brief Convert an Eigen transform to glm matrix
    glm::mat4 toglm(const Eigen::Affine3d& transform);
  }
 
  template < typename Space = Z3i::Space, typename KSpace = Z3i::KSpace>
  class PolyscopeViewer : public Display3D<Space, KSpace> {
    public:
      PolyscopeViewer() : Display3D<Space, KSpace>() {
        polyscope::init();
      }
 
      PolyscopeViewer(const KSpace& k) : Display3D<Space, KSpace>(k) {
        polyscope::init();
      }
 
      // @brief Clear the view (ie. remove all polyscope structures)
      void clearView() override {
        polyscope::removeAllStructures();
      }
 
      // @brief Starts event loop (first render data)
      void show() override {
        renderNewData();
        renderClippingPlanes();
 
        polyscope::show();
      }
 
      // @brief Render new data
      void renderNewData() override;
 
      // @brief setCallback
      void setCallback(typename Display3D<Space, KSpace>::Callback* callback) override;
 
      // @brief Renders the clipping planes
      void renderClippingPlanes();
 
      ~PolyscopeViewer() {
        this->clear();
      }
 
    public:
      static constexpr double BallToCubeRatio = 0.025;
      static constexpr double VectorScale = 1. / 30.;
 
    private:
      void setGeneralProperties(polyscope::Structure* s, const DisplayData<typename Space::RealPoint>& d);
 
      void registerPointCloud (const std::string& n, const DisplayData<typename Space::RealPoint>& d);
      void registerLineNetwork(const std::string& n, const DisplayData<typename Space::RealPoint>& d);
      void registerSurfaceMesh(const std::string& n, const DisplayData<typename Space::RealPoint>& d);
      void registerVolumeMesh (const std::string& n, const DisplayData<typename Space::RealPoint>& d);
      void registerSparseVolumeGrid(const std::string& n, const DisplayData<typename Space::RealPoint>& d);
 
      // @brief Global callback handling UI, click and callback calls
      void polyscopeCallback();
  };
}
 
#include "PolyscopeViewer.ih"