3dSDPViewer.cpp

#include <iostream>

#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/viewers/Viewer3D.h"
#include "DGtal/io/DrawWithDisplay3DModifier.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 "DGtal/io/DrawWithDisplay3DModifier.h"


#include <boost/program_options/options_description.hpp>
#include <boost/program_options/parsers.hpp>
#include <boost/program_options/variables_map.hpp>


using namespace std;
using namespace DGtal;
using namespace Z3i;

namespace po = boost::program_options;
typedef Viewer3D<Z3i::Space, Z3i::KSpace> Viewer;


// call back function to display voxel coordinates
int
displayCoordsCallBack( void* viewer, int name, void* data )
{
  vector<Z3i::RealPoint> *vectVoxels = (vector<Z3i::RealPoint> *) data;
  std::stringstream ss;
  ss << "Selected voxel: (" << (*vectVoxels)[name][0] <<  ", ";
  ss << (*vectVoxels)[name][1] <<  ", ";
  ss << (*vectVoxels)[name][2] <<  ") ";
  ((Viewer *) viewer)->displayMessage(QString(ss.str().c_str()), 100000);
  
  return 0;
}


int main( int argc, char** argv )
{
  
  
  // parse command line ----------------------------------------------
  po::options_description general_opt(" Allowed options are");
  general_opt.add_options()
  ("help,h", "display this message")
  ("input,i", po::value<std::string>(), "input file: sdp (sequence of discrete points)" )
  ("SDPindex", po::value<std::vector <unsigned int> >()->multitoken(), "specify the sdp index (by default 0,1,2).")
  ("pointColor,c", po::value<std::vector <int> >()->multitoken(), "set the color of  points: r g b a " )
  ("lineColor,l",po::value<std::vector <int> >()->multitoken(), "set the color of line: r g b a " )
  ("addMesh,m", po::value<std::string>(), "append a mesh (off/obj) to the point set visualization.")
  ("customColorMesh",po::value<std::vector<unsigned int> >()->multitoken(), "set the R, G, B, A components of the colors of the mesh faces (mesh added with option --addMesh). " )   
  ("importColors", "import point colors from the input file (R G B colors should be by default at index 3, 4, 5).")
  ("importColorLabels", "import color labels from the input file (label index  should be by default at index 3).")
  ("setColorsIndex", po::value<std::vector<unsigned int> >()->multitoken(), "customize the index of the imported colors in the source file (used by -importColor).")
  ("setColorLabelIndex", po::value<unsigned int >()->default_value(3), "customize the index of the imported color labels in the source file (used by -importColorLabels).")
  ("filter,f",po::value<double>()->default_value(100.0), "filter input file in order to display only the [arg] pourcent of the input 3D points (uniformly selected)." )
  ("noPointDisplay", "usefull for instance to only display the lines between points.")
  ("drawLines", "draw the line between discrete points." )
  ("scaleX,x",  po::value<float>()->default_value(1.0), "set the scale value in the X direction (default 1.0)" )
  ("scaleY,y",  po::value<float>()->default_value(1.0), "set the scale value in the Y direction (default 1.0)" )
  ("scaleZ,z",  po::value<float>()->default_value(1.0), "set the scale value in the Z direction (default 1.0)")
  ("sphereResolution",  po::value<unsigned int>()->default_value(30), "defines the sphere resolution (used when the primitive is set to the sphere). (default resolution: 30)")
  ("sphereRadius,s",  po::value<double>()->default_value(0.2), "defines the sphere radius (used when the primitive is set to the sphere). (default value 0.2)")
  ("sphereRadiusFromInput", "takes, as sphere radius, the 4th field of the sdp input file.")
  ("lineSize",  po::value<double>()->default_value(0.2), "defines the line size (used when the --drawLines or --drawVectors option is selected). (default value 0.2))")
  ("primitive,p", po::value<std::string>()->default_value("voxel"), "set the primitive to display the set of points (can be sphere, voxel (default), or glPoints (opengl points).")
  ("drawVectors,v", po::value<std::string>(), "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.")
  ("interactiveDisplayVoxCoords", "by using this option the pixel coordinates can be displayed after selection (shift+left click on voxel)." );
  
  
  bool parseOK=true;
  bool cannotStart= false;
  
  
  
  po::variables_map vm;
  try{
    po::store(po::parse_command_line(argc, argv, general_opt), vm);
  }catch(const std::exception& ex){
    parseOK=false;
    trace.error()<< "Error checking program options: "<< ex.what()<< endl;
  }
  po::notify(vm);
  std::string typePrimitive;
  double sphereRadius = 0.2;
  std::vector<double> vectSphereRadius;
  unsigned int sphereResolution = vm["sphereResolution"].as<unsigned int>();
  double lineSize =0.2;
  bool useMultiRad = vm.count("sphereRadiusFromInput");
  
  Color lineColor(100, 100, 250);
  Color pointColor(250, 250, 250);
  if(parseOK)
  {
    typePrimitive = vm["primitive"].as<std::string>();
    sphereRadius = vm["sphereRadius"].as<double>();
    lineSize = vm["lineSize"].as<double>();
  }
  
  if (parseOK && typePrimitive !="voxel"
      && typePrimitive !="glPoints" 
      && typePrimitive  != "sphere" )
  {
    trace.error() << " The primitive should be sphere or voxel (primitive: "
    << typePrimitive << " not implemented)" << std::endl;
    cannotStart = true;
  }
  
  if(parseOK && vm.count("lineColor"))
  {
    std::vector<int> vcol= vm["lineColor"].as<std::vector<int > >();
    if(vcol.size()<4)
    {
      trace.error() << " Not enough parameter: color specification should contains four elements: red, green, blue and alpha values "
      << "(Option --lineColor ignored). "  << std::endl;
    }
    lineColor.setRGBi(vcol[0], vcol[1], vcol[2], vcol[3]);
  }
  if(parseOK && vm.count("pointColor"))
  {
    std::vector<int> vcol= vm["pointColor"].as<std::vector<int > >();
    if(vcol.size()<4)
    {
      trace.error() << " Not enough parameter: color specification should contains four elements: red, green, blue and alpha values "
      << "(Option --pointColor ignored)."  << std::endl;
    }
    pointColor.setRGBi(vcol[0], vcol[1], vcol[2], vcol[3]);
  }
  
  if( !parseOK || cannotStart ||  vm.count("help")||argc<=1)
  {
    trace.info() << "Usage: " << argv[0] << " [input]\n"
    << "Display sequence of 3d discrete points by using QGLviewer."
    << general_opt << "\n";
    return 0;
  }
  
  string inputFilename = vm["input"].as<std::string>();
  
  
  QApplication application(argc,argv);
  
  float sx = vm["scaleX"].as<float>();
  float sy = vm["scaleY"].as<float>();
  float sz = vm["scaleZ"].as<float>();

  bool importColorLabels = vm.count("importColorLabels");
  bool importColors = vm.count("importColors");
  bool interactiveDisplayVoxCoords = vm.count("interactiveDisplayVoxCoords");
  typedef Viewer3D<Z3i::Space, Z3i::KSpace> Viewer;
  Z3i::KSpace K;
  Viewer viewer( K );
  viewer.setWindowTitle("3dSPD Viewer");
  viewer.show();
  viewer.setGLScale(sx, sy, sz);
  viewer.myGLLineMinWidth = lineSize;
  viewer << CustomColors3D(pointColor, pointColor);
  
  
  // Get vector of colors if imported.
  std::vector<Color> vectColors;
  if(vm.count("importColors"))
  {
    std::vector<unsigned int > vectIndex;
    if(vm.count("setColorsIndex"))
    {
      vectIndex = vm["setColorsIndex"].as<std::vector<unsigned int > >();
      if(vectIndex.size()!=3)
      {
        trace.error() << "you need to specify the three indexes of color." << std::endl;
        return 0;
      }
    }
    else
    {
      vectIndex.push_back(3);
      vectIndex.push_back(4);
      vectIndex.push_back(5);
    }
    
    
    std::vector<unsigned int> r = TableReader<unsigned int>::getColumnElementsFromFile(inputFilename,vectIndex[0]);
    std::vector<unsigned int> g = TableReader<unsigned int>::getColumnElementsFromFile(inputFilename,vectIndex[1]);
    std::vector<unsigned int> b = TableReader<unsigned int>::getColumnElementsFromFile(inputFilename,vectIndex[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(vm.count("importColorLabels"))
  {
    unsigned int index = vm["setColorLabelIndex"].as<unsigned int >();
    vectColorLabels = TableReader< int>::getColumnElementsFromFile(inputFilename,index);
    maxLabel = *(std::max_element(vectColorLabels.begin(), vectColorLabels.end()));
  }
  HueShadeColorMap<unsigned int> aColorMap(0, maxLabel);
  
  
  if(useMultiRad)
  {
    vectSphereRadius = TableReader<double>::getColumnElementsFromFile(inputFilename,3);
  }
  
  vector<Z3i::RealPoint> vectVoxels;
  if(vm.count("SDPindex"))
  {
    std::vector<unsigned int > vectIndex = vm["SDPindex"].as<std::vector<unsigned int > >();
    if(vectIndex.size()!=3)
    {
      trace.error() << "you need to specify the three indexes of vertex." << std::endl;
      return 0;
    }
    vectVoxels = PointListReader<Z3i::RealPoint>::getPointsFromFile(inputFilename, vectIndex);
    
  }else{
    vectVoxels = PointListReader<Z3i::RealPoint>::getPointsFromFile(inputFilename);
  }
  int name = 0;
  if(!vm.count("noPointDisplay")){
    if (typePrimitive == "glPoints")
      {
        viewer.setUseGLPointForBalls(true);
      }

    double percent = vm["filter"].as<double>();
    int step = max(1, (int) (100/percent));
    for(unsigned int i=0;i< vectVoxels.size(); i=i+step){
      if(importColors)
        {
          Color col = vectColors[i];
          viewer.setFillColor(col);
        }
      else if(importColorLabels)
        {
          unsigned int index = vectColorLabels[i];
          Color col = aColorMap(index);
          viewer.setFillColor(col);
        }
      
      if(typePrimitive=="voxel" ){
        if (interactiveDisplayVoxCoords)
        {
          viewer << SetName3D( name++ ) ;
        }
        viewer << Z3i::Point((int)vectVoxels.at(i)[0],
                             (int)vectVoxels.at(i)[1],
                             (int)vectVoxels.at(i)[2]);
      }
      else
        {
          viewer.addBall(vectVoxels.at(i), sphereRadius, sphereResolution);
        }
    }
    
    viewer << CustomColors3D(lineColor, lineColor);
    if(vm.count("drawLines"))
    {
      for(unsigned int i=1;i< vectVoxels.size(); i++)
      {
        viewer.addLine(vectVoxels.at(i-1), vectVoxels.at(i), lineSize);
      }
    }
    
    
    if(vm.count("drawVectors"))
    {
      std::string vectorsFileName = vm["drawVectors"].as<std::string>();
      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;
      }
      for(unsigned int i =0; i<vectorsPt.size()-1; i=i+2)
      {
        viewer.addLine(vectorsPt.at(i),vectorsPt.at(i+1), lineSize);
      }
      
    }
    if(vm.count("addMesh"))
    {
      bool customColorMesh =  vm.count("customColorMesh");
      if(customColorMesh){
        std::vector<unsigned int > vectCol = vm["customColorMesh"].as<std::vector<unsigned int> >();
        if(vectCol.size()!=4){
          trace.error() << "colors specification should contain R,G,B and Alpha values"<< std::endl;
        }
        viewer.setFillColor(DGtal::Color(vectCol[0], vectCol[1], vectCol[2], vectCol[3]));
      }
      std::string meshName = vm["addMesh"].as<std::string>();
      Mesh<Z3i::RealPoint> mesh(!customColorMesh);
      mesh << meshName ;
      viewer << mesh;
    }
    if (interactiveDisplayVoxCoords)
    {
      viewer << SetSelectCallback3D( displayCoordsCallBack,  &vectVoxels,  0, vectVoxels.size()-1 );
    }
    
    
    viewer << Viewer3D<>::updateDisplay;
    return application.exec();
  }
}