shapeDTViewer.cpp

#include <iostream>
#include <fstream>
#include <algorithm>

#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
#include "ConfigExamples.h"

#include "DGtal/shapes/Shapes.h" 
#include "DGtal/shapes/ShapeFactory.h" 

#include <QtGui/qapplication.h>
#include "DGtal/io/viewers/Viewer3D.h" 
#include "DGtal/io/DrawWithDisplay3DModifier.h"

#include "DGtal/geometry/volumes/distance/DistanceTransformation.h"
#include "DGtal/kernel/sets/SetPredicate.h"
#include "DGtal/io/colormaps/GradientColorMap.h"

int main(int argc, char **argv)
{
  QApplication application(argc,argv);
  
  DGtal::Viewer3D viewer;
  
  DGtal::Z3i::Point center(0,0,0);
  DGtal::ImplicitRoundedHyperCube<Z3i::Space> myCube( center, 20, 2.8);
  DGtal::Z3i::Domain domain(myCube.getLowerBound(), 
                            myCube.getUpperBound());
  
  DGtal::Z3i::DigitalSet mySet(domain);
  
  DGtal::Shapes<DGtal::Z3i::Domain>::euclideanShaper( mySet, myCube);


  viewer.show();
  // viewer << mySet << DGtal::Display3D::updateDisplay; 
  

  typedef DGtal::SetPredicate<DGtal::Z3i::DigitalSet> Predicate;
  Predicate aPredicate(mySet);
  
  typedef DGtal::DistanceTransformation<Z3i::Space, Predicate, 2> DTL2;
  typedef DTL2::OutputImage OutputImage;
  DTL2 dt(domain,aPredicate);
  
  OutputImage result = dt.compute();

  OutputImage::Value maxDT = (*std::max_element(result.begin(), 
                                                result.end()));
  
  
  GradientColorMap<OutputImage::Value> gradient( 0, maxDT);
  gradient.addColor(DGtal::Color::Blue);
  gradient.addColor(DGtal::Color::Green);
  gradient.addColor(DGtal::Color::Yellow);
  gradient.addColor(DGtal::Color::Red);
  

  for(Z3i::Domain::ConstIterator it = domain.begin(),
        itend = domain.end(); it != itend;
      ++it)
    if (result(*it) != 0)
      {
        OutputImage::Value  val= result( *it );     
        DGtal::Color c= gradient(val);
        
        viewer <<  DGtal::CustomColors3D(c,c) << *it    ; 
          
      }  

  
  viewer << DGtal::ClippingPlane(1,0,0,0);
  //@todo updateDisplay is in Display3D or Viewer3D (cf doc)?
  viewer << DGtal::Display3D::updateDisplay; 
  
  return application.exec();
  

}