meshViewer.cpp

 
#include <iostream>
#include <sstream>
#include "DGtal/base/Common.h"
 
#include "DGtal/io/viewers/PolyscopeViewer.h"
#include "DGtal/io/readers/MeshReader.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/readers/PointListReader.h"
 
#include "CLI11.hpp"
 
using namespace std;
using namespace DGtal;
 
// 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)
{
  float sx{1.0};
  float sy{1.0};
  float sz{1.0};
 
  unsigned int meshColorR{240};
  unsigned int meshColorG{240};
  unsigned int meshColorB{240};
  unsigned int meshColorA{255};
 
  unsigned int meshColorRLine{0};
  unsigned int meshColorGLine{0};
  unsigned int meshColorBLine{0};
  unsigned int meshColorALine{255};
 
  unsigned int sdpColorR{240};
  unsigned int sdpColorG{240};
  unsigned int sdpColorB{240};
  unsigned int sdpColorA{255};
 
  float lineWidth{1.5};
 
  std::vector<unsigned int> customColorMesh;
  std::vector<unsigned int> customColorSDP;
  std::vector<unsigned int> customLineColor;
  std::vector<unsigned int> customBGColor;
  std::vector<unsigned int> customAlphaMesh;
  std::vector<unsigned int> vectFieldIndices = {0, 1, 2, 3, 4, 5};
  std::string displayVectorField;
 
  std::string snapshotFile;
  std::string filenameSDP;
  double ballRadius{0.5};
  bool invertNormal{false};
  bool drawVertex{false};
 
  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::vector<std::string> inputFileNames;
  std::string outputFileName{"result.raw"};
  app.description("Display OFF mesh file by using PolyscopeViewers");
  app.add_option("-i,--input,1", inputFileNames, "inputFileNames.off files (.off), or OFS file (.ofs)")
      ->check(CLI::ExistingFile)
      ->required();
  app.add_option("-x,--scaleX", sx, "set the scale value in the X direction (default 1.0)");
  app.add_option("-y,--scaleY", sy, "set the scale value in the y direction (default 1.0)");
  app.add_option("-z,--scaleZ", sz, "set the scale value in the z direction (default 1.0)");
  app.add_option("--minLineWidth", lineWidth, "set the min line width of the mesh faces (default 1.5)");
  app.add_option("--customColorMesh", customColorMesh, "set the R, G, B, A components of the colors of the mesh faces and eventually the color R, G, B, A of the mesh edge lines (set by default to black).");
  app.add_option("--customAlphaMesh", customAlphaMesh, "set the alpha(A) components of the colors of the mesh faces (can be applied for each mesh).");
  app.add_option("--customColorSDP", customColorSDP, "set the R, G, B, A components of the colors of the sdp view")
      ->expected(4);
  app.add_option("--displayVectorField,-f", displayVectorField, "display a vector field from a simple sdp file (two points per line)");
  app.add_option("--vectorFieldIndex", vectFieldIndices, "specify special indices for the two point coordinates (instead usinf the default indices: 0 1, 2, 3, 4, 5)")
      ->expected(6);
  app.add_option("--customLineColor", customLineColor, "set the R, G, B components of the colors of the lines displayed from the --displayVectorField option (red by default).")
      ->expected(4);
  app.add_option("--SDPradius", ballRadius, "change the ball radius to display a set of discrete points (used with displaySDP option)");
  app.add_option("--displaySDP,-s", filenameSDP, "add the display of a set of discrete points as ball of radius 0.5.");
  app.add_flag("--invertNormal,-n", invertNormal, "invert face normal vectors.");
  app.add_flag("--drawVertex,-v", drawVertex, "draw the vertex of the mesh");
 
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------
 
  DGtal::Color vFieldLineColor = DGtal::Color::Red;
  if (customLineColor.size() == 4)
  {
    vFieldLineColor.setRGBi(customLineColor[0], customLineColor[1], customLineColor[2], 255);
  }
 
  if (customColorMesh.size() != 0)
  {
    if (customColorMesh.size() < 4)
    {
      trace.error() << "colors specification should contain R,G,B and Alpha values" << std::endl;
    }
  }
 
  if (customColorSDP.size() == 4)
  {
    sdpColorR = customColorSDP[0];
    sdpColorG = customColorSDP[1];
    sdpColorB = customColorSDP[2];
    sdpColorA = customColorSDP[3];
  }
 
 
  stringstream s;
  s << "meshViewer - DGtalTools: ";
  string name = inputFileNames[0].substr(inputFileNames[0].find_last_of("/")+1,inputFileNames[0].size()) ;
  s << " " <<  name << " (W key to display settings)";
  polyscope::options::programName = s.str();
  polyscope::options::buildGui=false;
  polyscope::options::groundPlaneMode = polyscope::GroundPlaneMode::None;
  polyscope::view::setNavigateStyle(polyscope::NavigateStyle::Free); 
 
  // Masquer le sol aussi
  PolyscopeViewer viewer;
 
  viewer.allowReuseList = true;
  trace.info() << "Importing mesh... ";
 
  std::vector<Mesh<DGtal::Z3i::RealPoint>> vectMesh;
  for (unsigned int i = 0; i < inputFileNames.size(); i++)
  {
    Mesh<DGtal::Z3i::RealPoint> aMesh(customColorMesh.size() != 4 && customColorMesh.size() != 8);
    aMesh << inputFileNames[i];
    // for obj mesh by default the mesh color face are not necessary uniform.
    if (aMesh.isStoringFaceColors() && customColorMesh.size() >= 4)
    {
      if (i * 8 < customColorMesh.size())
      {
        meshColorR = customColorMesh[i * 8];
      }
      if (i * 8 + 1 < customColorMesh.size())
      {
        meshColorG = customColorMesh[i * 8 + 1];
      }
      if (i * 8 + 2 < customColorMesh.size())
      {
        meshColorB = customColorMesh[i * 8 + 2];
      }
      if (i * 8 + 3 < customColorMesh.size())
      {
        meshColorA = customColorMesh[i * 8 + 3];
      }
      for (unsigned int j = 0; j < aMesh.nbFaces(); j++)
      {
        aMesh.setFaceColor(j, Color(meshColorR, meshColorG, meshColorB, meshColorA));
      }
    }
    else if (customAlphaMesh.size() > 0)
    {
      for (unsigned int j = 0; j < aMesh.nbFaces(); j++)
      {
        auto c = aMesh.getFaceColor(j);
        aMesh.setFaceColor(j, Color(c.red(), c.green(), c.blue(), customAlphaMesh.at(i < customAlphaMesh.size() ? i : 0)));
      }
    }
 
    vectMesh.push_back(aMesh);
  }
  
  bool import = vectMesh.size() == inputFileNames.size();
  if (!import)
  {
    trace.info() << "File import failed. " << std::endl;
    return 0;
  }
 
  trace.info() << "[done]. " << std::endl;
  if (filenameSDP != "")
  {
    if (customAlphaMesh.size() > 0)
    {
      trace.info() << "New meshViewer" << std::endl;
      auto vOrigins = PointListReader<PointVector<1, DGtal::int32_t>>::getPolygonsFromFile(filenameSDP);
      viewer << Color(sdpColorR, sdpColorG, sdpColorB, sdpColorA);
      for (auto l : vOrigins)
      {
        DGtal::Z3i::Point pt(l[0][0], l[1][0], l[2][0]);
        DGtal::Color cl(l[3][0], l[4][0], l[5][0], sdpColorA);
        viewer.drawColor(cl);
        viewer.drawBall(pt);
      }
    }
    else
    {
      vector<Z3i::RealPoint> vectPoints;
      vectPoints = PointListReader<Z3i::RealPoint>::getPointsFromFile(filenameSDP);
      viewer << Color(sdpColorR, sdpColorG, sdpColorB, sdpColorA);
      for (unsigned int i = 0; i < vectPoints.size(); i++)
      {
        viewer.drawBall(vectPoints.at(i));
      }
    }
  }
  if (invertNormal)
  {
    for (unsigned int i = 0; i < vectMesh.size(); i++)
    {
      vectMesh[i].invertVertexFaceOrder();
    }
  }
 
  for (unsigned int i = 0; i < vectMesh.size(); i++)
  {
    if (i * 8 < customColorMesh.size())
    {
      meshColorR = customColorMesh[i * 8];
    }
    if (i * 8 + 1 < customColorMesh.size())
    {
      meshColorG = customColorMesh[i * 8 + 1];
    }
    if (i * 8 + 2 < customColorMesh.size())
    {
      meshColorB = customColorMesh[i * 8 + 2];
    }
    if (i * 8 + 3 < customColorMesh.size())
    {
      meshColorA = customColorMesh[i * 8 + 3];
    }
    if (i * 8 + 4 < customColorMesh.size())
    {
      meshColorALine = customColorMesh[i * 8 + 4];
    }
    if (i * 8 + 5 < customColorMesh.size())
    {
      meshColorBLine = customColorMesh[i * 8 + 5];
    }
    if (i * 8 + 6 < customColorMesh.size())
    {
      meshColorRLine = customColorMesh[i * 8 + 6];
    }
    if (i * 8 + 7 < customColorMesh.size())
    {
      meshColorALine = customColorMesh[i * 8 + 7];
    }
 
    viewer << Color(meshColorR, meshColorG, meshColorB, meshColorA);
    viewer << vectMesh[i];
  }
 
  if (drawVertex)
  {
    for (unsigned int i = 0; i < vectMesh.size(); i++)
    {
      for (Mesh<DGtal::Z3i::RealPoint>::VertexStorage::const_iterator it = vectMesh[i].vertexBegin();
           it != vectMesh[i].vertexEnd(); ++it)
      {
        DGtal::Z3i::Point pt;
        pt[0] = (*it)[0];
        pt[1] = (*it)[1];
        pt[2] = (*it)[2];
        viewer << pt;
      }
    }
  }
 
  if (displayVectorField != "")
  {
    std::vector<unsigned int> vectFieldIndices1 = {vectFieldIndices[0], vectFieldIndices[1], vectFieldIndices[2]};
    std::vector<unsigned int> vectFieldIndices2 = {vectFieldIndices[3], vectFieldIndices[4], vectFieldIndices[5]};
    std::vector<DGtal::Z3i::RealPoint> vectPt1 = PointListReader<DGtal::Z3i::RealPoint>::getPointsFromFile(displayVectorField, vectFieldIndices1);
    std::vector<DGtal::Z3i::RealPoint> vectPt2 = PointListReader<DGtal::Z3i::RealPoint>::getPointsFromFile(displayVectorField, vectFieldIndices2);
    for (unsigned int i = 0; i < vectPt1.size(); i++)
    {
      viewer.drawColor(vFieldLineColor);
      viewer.drawLine(vectPt1[i], vectPt2[i]);
    }
  }
  unsigned int nbVertex = 0;
  unsigned int nbFaces = 0;
  for (auto const &m : vectMesh)
  {
    nbVertex += m.nbVertex();
    nbFaces += m.nbFaces();
  }
  stringstream ss;
  ss << "# faces: " << std::fixed << nbFaces << "    #vertex: " << nbVertex;
  trace.info() << "[display ready]" << std::endl;
  polyscope::state::userCallback = myCallback;
 
  viewer.show();
  return 0;
}