3dHeightMapViewer.cpp

Go to the documentation of this file.

#include <iostream>
#include <climits>

#include "DGtal/base/Common.h"
#include "DGtal/base/BasicFunctors.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/readers/GenericReader.h"
#include "DGtal/io/viewers/Viewer3D.h"
#include <DGtal/io/colormaps/GradientColorMap.h>
#include <DGtal/io/colormaps/GrayscaleColorMap.h>
#include <DGtal/images/ImageContainerBySTLVector.h>

#include "DGtal/io/Color.h"

#include <DGtal/base/ConstAlias.h>
#include <DGtal/kernel/BasicPointFunctors.h>
#include <DGtal/topology/helpers/Surfaces.h>
#include <DGtal/io/colormaps/GradientColorMap.h>

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

#include "CLI11.hpp"



template < typename Space = DGtal::Z3i::Space, typename KSpace = DGtal::Z3i::KSpace>
struct Viewer3DImageSpec: public DGtal::Viewer3D <Space, KSpace>
{
  Viewer3DImageSpec(Z2i::RealPoint pup, Z2i::RealPoint plow): myZScale(1.0), myZposClipping(10.0),
                                                              myPUpper(pup), myPLower(plow){
    DGtal::Viewer3D<>::update();
  };

virtual  void
init(){
   DGtal::Viewer3D<>::init();
   QGLViewer::setKeyDescription ( Qt::Key_Up, "Move Up the cutting plane in the Z axis direction." );
   QGLViewer::setKeyDescription ( Qt::Key_Down, "Move Down the cutting plane in the Z axis direction." );
   QGLViewer::setKeyDescription ( Qt::Key_Shift, "Change the cutting plane move with step 1 (5 by default)" );
   QGLViewer::setKeyDescription ( Qt::Key_Plus, "Increase the openGL scale by 0.25 " );
   QGLViewer::setKeyDescription ( Qt::Key_Minus, "Decrease the openGL scale by 0.25 " );
};


protected:
  double myZScale;
  double myZposClipping;
  Z2i::RealPoint myPUpper, myPLower;

  virtual void draw (  ){
    DGtal::Viewer3D<>::draw();
    glPushMatrix();
    glMultMatrixd ( DGtal::Viewer3D<>::manipulatedFrame()->matrix() );
    glPushMatrix();
    glScalef(1.0, 1.0, myZScale);
    glEnable( GL_LIGHTING );
    glBegin( GL_QUADS );
    glColor4ub( 50, 50, 240,  150 );
    glNormal3f( 0, 0 , 1.0 );
    glVertex3f( myPLower[0], myPLower[1] , myZposClipping );
    glVertex3f( myPUpper[0], myPLower[1] , myZposClipping );
    glVertex3f( myPUpper[0], myPUpper[1], myZposClipping );
    glVertex3f( myPLower[0], myPUpper[1],  myZposClipping );
    glEnd();
    glPopMatrix();
    glPopMatrix();
  }
  virtual void keyPressEvent ( QKeyEvent *e ){
    bool handled = false;
    if( e->key() == Qt::Key_Up){
      if((e->modifiers() & Qt::MetaModifier)){
        myZposClipping+=1;
      }else{
        myZposClipping+=5;
      }

      DGtal::Viewer3D<>::update();
      handled=true;
    }
    if( e->key() == Qt::Key_Down){
      if((e->modifiers() & Qt::MetaModifier)){
        myZposClipping-=1;
      }else{
        myZposClipping-=5;
      }

      DGtal::Viewer3D<>::update();
      handled=true;
    }

    if( e->key() == Qt::Key_Plus){
      myZScale+=0.25;
      DGtal::Viewer3D<Space, KSpace>::setGLScale(1.0, 1.0, myZScale);
      DGtal::Viewer3D<Space, KSpace>::update();
      handled=true;
    }
    if( e->key() == Qt::Key_Minus){
      myZScale-=0.25;
      DGtal::Viewer3D<Space, KSpace>::setGLScale(1.0, 1.0, myZScale);
      DGtal::Viewer3D<>::update();
      handled=true;
    }
    if ( !handled )
      DGtal::Viewer3D<>::keyPressEvent ( e );
  }
};

// Defining a Helper to get the 3D point functor from an 2DImage
template<typename TImage2D, typename TPoint3D >
struct Image3DPredicatFrom2DImage{
  typedef  TPoint3D Point3D;
  Image3DPredicatFrom2DImage(DGtal::ConstAlias<TImage2D> anImage, double aScale):myImageRef(anImage),
                                                                                 myScale(aScale){
  }
  inline
  bool operator()(const Point3D &aPoint)  const {
    functors::Projector<SpaceND<2, typename TImage2D::Integer> > projXY;
    return  (*myImageRef)(projXY(aPoint))*myScale >= aPoint[2];
  }
  CountedConstPtrOrConstPtr<TImage2D> myImageRef;
  double myScale;
};








int main( int argc, char** argv )
{
  
  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::string inputFileName;
  double scale {1.0};
  bool colorMap {false};
  std::string colorTextureImage;
  app.description("Displays 2D image as heightmap by using QGLviewer.\n Exemple of use:  visualisation/3dHeightMapViewer  ${DGtal}/examples/samples/church.pgm -s 0.2");
  
  app.add_option("-i,--input,1", inputFileName, "2d input image representing the height map (given as grayscape image cast into 8 bits)." )
  ->required()
  ->check(CLI::ExistingFile);
  app.add_option("--scale,-s",scale,  "set the scale of the maximal level. (default 1.0)");
  app.add_flag("--colorMap,-c", colorMap, "define the heightmap color with a pre-defined colormap (GradientColorMap)");
  app.add_option("--colorTextureImage,-t", colorTextureImage,  "define the heightmap color from a given color image (32 bits image).");
  
  
  
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------

  
  typedef DGtal::ImageContainerBySTLVector<Z2i::Domain, unsigned char> Image2DG ;
  typedef DGtal::ImageContainerBySTLVector<Z2i::Domain, unsigned int> Image2DCol ;

  Image2DG image = GenericReader<Image2DG>::import( inputFileName );
  Image2DCol imageTexture(image.domain());
  int maxHeight = (int)(std::numeric_limits<Image2DG::Value>::max()*scale);
  trace.info()<< "Max height from scale:" << maxHeight << std::endl;

  if(colorTextureImage != ""){
    imageTexture =  GenericReader<Image2DCol>::import( colorTextureImage );
  }

  QApplication application(argc,argv);
  Z2i::RealPoint plow (image.domain().lowerBound()[0]-0.5,
                       image.domain().lowerBound()[1]-0.5);
  
  Z2i::RealPoint pup (image.domain().upperBound()[0]+0.5,
                      image.domain().upperBound()[1]+0.5);
  
  Viewer3DImageSpec<> viewer(plow, pup) ;
  viewer.setWindowTitle("Height Map Viewer");
  viewer.show();


  KSpace K;
  K.init(Z3i::Point(0,0,0),Z3i::Point(image.domain().upperBound()[0], image.domain().upperBound()[1], maxHeight+1), true);
  std::set<KSpace::SCell> boundVect;
  Image3DPredicatFrom2DImage<Image2DG, Z3i::Point> image3Dpredicate(image, scale);
  trace.info() << "Constructing boundary... ";
  Surfaces<KSpace>::sMakeBoundary (boundVect, K, image3Dpredicate, Z3i::Point(0,0,0),
                                   Z3i::Point(image.domain().upperBound()[0], image.domain().upperBound()[1], maxHeight+1));
  trace.info() << "[done]"<< std::endl;

  viewer << SetMode3D((*(boundVect.begin())).className(), "Basic" );
  GradientColorMap<Image2DG::Value,CMAP_JET>  gradientShade( 0, std::numeric_limits<Image2DG::Value>::max());
  GrayscaleColorMap<Image2DG::Value>  grayShade(0, std::numeric_limits<Image2DG::Value>::max());


  for(std::set<KSpace::SCell>::const_iterator it = boundVect.begin();
      it!= boundVect.end(); it++){
    Z3i::Point pt = K.sCoords(K.sDirectIncident( *it, 2 ));
    functors::Projector<SpaceND<2,int> > proj;
    Image2DG::Value val = image(proj(pt));
    if(colorMap){
      viewer.setFillColor(gradientShade(val));
    }else if (colorTextureImage != "") {
      viewer.setFillColor(Color(imageTexture(proj(pt))));
    }else{
      viewer.setFillColor(grayShade(val));
    }
  viewer << *it;
  }

  viewer << Viewer3D<>::updateDisplay;
  return application.exec();
}