3dSDPViewer.cpp

 
#include <iostream>
 
#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/viewers/PolyscopeViewer.h"
#include "DGtal/io/readers/TableReader.h"
#include "DGtal/io/readers/PointListReader.h"
#include "DGtal/io/readers/MeshReader.h"
#include "DGtal/topology/helpers/Surfaces.h"
#include "DGtal/topology/SurfelAdjacency.h"
#include "DGtal/shapes/Mesh.h"
#include "DGtal/io/Color.h"
#include "DGtal/io/colormaps/HueShadeColorMap.h"
#include "DGtal/io/readers/GenericReader.h"
 
#include "CLI11.hpp"
 
using namespace std;
using namespace DGtal;
using namespace Z3i;
 
typedef PolyscopeViewer<Z3i::Space, Z3i::KSpace> Viewer;
 
// Variable globale pour activer/désactiver l'UI
bool show_ui = false;
 
void myCallback() {
    ImGuiIO& io = ImGui::GetIO();
 
    
    // Si la touche W est enfoncée ce frame
    if (ImGui::IsKeyPressed(ImGuiKey_W)) {
        show_ui = !show_ui;
        polyscope::options::buildGui = show_ui;
    }
 
  
}
 
 
 
int main(int argc, char **argv)
{
  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::string inputFileName;
  std::vector<unsigned int> vectIndexSDP{0, 1, 2};
  Color lineColor(100, 100, 250);
  Color pointColor(250, 250, 250);
  std::vector<int> vectColorPt;
  std::vector<int> vectColorLine;
  std::string meshName;
  std::string typePrimitive{"voxel"};
  std::string vectorsFileName;
  std::vector<unsigned int> vectColMesh;
  std::vector<unsigned int> vectIndexColorImport{3, 4, 5};
  bool importColorLabels{false};
  bool noPointDisplay{false};
  bool drawLines{false};
  bool importColors{false};
  
  float sx{1.0};
  float sy{1.0};
  float sz{1.0};
  double lineSize{0.2};
  double filterValue{100.0};
  double constantNorm{0.0};
  double percentageFilterVect{100.0};
  unsigned int customAlphaMesh;
 
  unsigned int colorLabelIndex = 3;
 
  app.description("Display sequence of 3d discrete points by using PolyscopeViewer.");
  app.add_option("-i,--input,1", inputFileName, "input file: sdp (sequence of discrete points).")
      ->required()
      ->check(CLI::ExistingFile);
  app.add_option("--SDPindex", vectIndexSDP, "specify the sdp index (by default 0,1,2).")
      ->expected(3);
  app.add_option("--pointColor,-c", vectColorPt, "set the color of  points: r g b a.")
      ->expected(4);
  app.add_option("--lineColor,-l", vectColorLine, "set the color of line: r g b a.")
      ->expected(4);
  app.add_option("--addMesh,-m", meshName, "append a mesh (off/obj) to the point set visualization.");
  app.add_option("--customAlphaMesh", customAlphaMesh, "set the alpha components of the colors of the mesh faces (can be applied for each mesh).");
  auto optMesh = app.add_option("--customColorMesh", vectColMesh, "set the R, G, B, A components of the colors of the mesh faces (mesh added with option --addMesh).")
                     ->expected(4);
  auto importColOpt = app.add_flag("--importColors", importColors, "import point colors from the input file (R G B colors should be by default at index 3, 4, 5).");
  app.add_option("--setColorsIndex", vectIndexColorImport, "customize the index of the imported colors in the source file (used by --importColor). By default the initial indexes are 3, 4, 5.")
      ->expected(3)
      ->needs(importColOpt);
 
  app.add_flag("--importColorLabels", importColorLabels, "import color labels from the input file (label index  should be by default at index 3).");
  app.add_option("--setColorLabelIndex", colorLabelIndex, "customize the index of the imported color labels in the source file (used by -importColorLabels).");
  app.add_option("--filter,-f", filterValue, "filter input file in order to display only the [arg] percent of the input 3D points (uniformly selected).");
  app.add_flag("--noPointDisplay", noPointDisplay, "usefull for instance to only display the lines between points.");
  app.add_flag("--drawLines", drawLines, "draw the line between discrete points.");
  app.add_option("--scaleX,-x", sx, "set the scale value in the X direction");
  app.add_option("--scaleY,-y", sy, "set the scale value in the Y direction");
  app.add_option("--scaleZ,-z", sy, "set the scale value in the Z direction");
  app.add_option("--lineSize", lineSize, "defines the line size (used when the --drawLines or --drawVectors option is selected).");
  app.add_option("--primitive,-p", typePrimitive, "set the primitive to display the set of points.")
      ->check(CLI::IsMember({"voxel", "sphere"}));
  app.add_option("--drawVectors,-v", vectorsFileName, "SDP vector file: draw a set of vectors from the given file (each vector are determined by two consecutive point given, each point represented by its coordinates on a single line.");
  app.add_option("--unitVector,-u", constantNorm, "specifies that the SDP vector file format (of --drawVectors option) should be interpreted as unit vectors (each vector position is be defined from the input point (with input order) with a constant norm defined by [arg]).");
  app.add_option("--filterVectors", percentageFilterVect, "filters vector input file in order to display only the [arg] percent of the input vectors (uniformly selected, to be used with option --drawVectors else no effect). ");
 
 
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------
 
  if (vectColorLine.size() == 4)
  {
    lineColor.setRGBi(vectColorLine[0], vectColorLine[1], vectColorLine[2], vectColorLine[3]);
  }
  if (vectColorPt.size() == 4)
  {
    pointColor.setRGBi(vectColorPt[0], vectColorPt[1], vectColorPt[2], vectColorPt[3]);
  }
    stringstream s;
    s << "3dSDPViewer - DGtalTools: ";
    string bname = inputFileName.substr(inputFileName.find_last_of("/")+1,inputFileName.size()) ;
    s << " " <<  bname ;
    polyscope::options::programName = s.str();
 
  bool useUnitVector = constantNorm != 0.0;
  polyscope::view::setNavigateStyle(polyscope::NavigateStyle::Free);
  polyscope::options::groundPlaneMode = polyscope::GroundPlaneMode::None;
  polyscope::options::buildGui=false;
  
  
  typedef PolyscopeViewer<Z3i::Space, Z3i::KSpace> Viewer;
  Z3i::KSpace K;
  Viewer viewer(K);
  viewer << pointColor;
  viewer.allowReuseList = true;
  
  if (typePrimitive == "sphere") {
    viewer.drawAsBalls();
  }
 
  // Get vector of colors if imported.
  std::vector<Color> vectColors;
  if (importColors)
  {
    std::vector<unsigned int> r = TableReader<unsigned int>::getColumnElementsFromFile(inputFileName, vectIndexColorImport[0]);
    std::vector<unsigned int> g = TableReader<unsigned int>::getColumnElementsFromFile(inputFileName, vectIndexColorImport[1]);
    std::vector<unsigned int> b = TableReader<unsigned int>::getColumnElementsFromFile(inputFileName, vectIndexColorImport[2]);
    for (unsigned int i = 0; i < r.size(); i++)
    {
      vectColors.push_back(Color(r[i], g[i], b[i]));
    }
  }
 
  // Get vector of colors if imported.
  std::vector<int> vectColorLabels;
  unsigned int maxLabel = 1;
  if (importColorLabels)
  {
    vectColorLabels = TableReader<int>::getColumnElementsFromFile(inputFileName, colorLabelIndex);
    maxLabel = *(std::max_element(vectColorLabels.begin(), vectColorLabels.end()));
  }
  HueShadeColorMap<unsigned int> aColorMap(0, maxLabel);
 
  vector<Z3i::RealPoint> vectVoxels;
  vectVoxels = PointListReader<Z3i::RealPoint>::getPointsFromFile(inputFileName, vectIndexSDP);
 
  int name = 0;
  if (!noPointDisplay)
  {
    int step = max(1, (int)(100 / filterValue));
    for (unsigned int i = 0; i < vectVoxels.size(); i = i + step)
    {
      if (importColors)
      {
        pointColor = vectColors[i];
 
      }
      else if (importColorLabels)
      {
        unsigned int index = vectColorLabels[i];
        pointColor = aColorMap(index);
      }      
      viewer << WithQuantity(Z3i::Point((int)vectVoxels.at(i)[0],
                                        (int)vectVoxels.at(i)[1],
                                        (int)vectVoxels.at(i)[2]), "label", pointColor);
      
      
    }
 
    viewer << lineColor;
    if (drawLines)
    {
      for (unsigned int i = 1; i < vectVoxels.size(); i++)
      {
        viewer.drawLine(vectVoxels.at(i - 1), vectVoxels.at(i));
      }
    }
 
    if (vectorsFileName != "")
    {
      std::vector<Z3i::RealPoint> vectorsPt = PointListReader<Z3i::RealPoint>::getPointsFromFile(vectorsFileName);
      if (vectorsPt.size() % 2 == 1)
      {
        trace.info() << "Warning the two set of points doesn't contains the same number of points, some vectors will be skipped." << std::endl;
      }
 
      double percentage = percentageFilterVect;
      int step = max(1, (int)(100 / percentage));
      
      // Splits vectors into their own line groups
      viewer.endCurrentGroup();  
      if (useUnitVector)
      {
        for (unsigned int i = 0; i < std::min(vectVoxels.size(), vectorsPt.size()); i = i + 2 * step)
        {
          viewer.drawLine(vectVoxels.at(i), vectVoxels.at(i) + vectorsPt.at(i) * constantNorm);
        }
      }
      else
      {
        for (unsigned int i = 0; i < vectorsPt.size() - 1; i = i + 2 * step)
        {
          viewer.drawLine(vectorsPt.at(i), vectorsPt.at(i + 1));
        }
      }
    }
 
    if (meshName != "")
    {
      bool customColorMesh = vectColMesh.size() == 4;
      if (customColorMesh)
      {
        viewer.drawColor(DGtal::Color(vectColMesh[0], vectColMesh[1], vectColMesh[2], vectColMesh[3]));
      }
 
      Mesh<Z3i::RealPoint> mesh(!customColorMesh);
      mesh << meshName;
      if (customAlphaMesh)
      {
        for (unsigned int j = 0; j < mesh.nbFaces(); j++)
        {
          auto c = mesh.getFaceColor(j);
          mesh.setFaceColor(j, Color(c.red(), c.green(), c.blue(), customAlphaMesh));
        }
        viewer << mesh;
      }
    }
    polyscope::state::userCallback = myCallback;
    viewer.show();
    return 0;
  }
}