parallelIIShortcuts-viewer.cpp

Go to the documentation of this file.

 
#include <iostream>
#include "ConfigExamples.h"
 
// Helpers
#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
 
#include "DGtal/kernel/domains/DomainSplitter.h"
#include "DGtal/helpers/Shortcuts.h"
#include "DGtal/helpers/ShortcutsGeometry.h"
 
// Visualization
#include "DGtal/io/viewers/PolyscopeViewer.h"
#include "DGtal/io/colormaps/GradientColorMap.h"
 
 
using namespace DGtal;
 
// Using standard 3D digital space.
typedef Shortcuts<Z3i::KSpace> SH3;
typedef ShortcutsGeometry<Z3i::KSpace> SHG3;
 
int main()
{
 
  auto params = SH3::defaultParameters() | SHG3::defaultParameters();
  std::string filename = examplesPath + std::string("/samples/bunny-128.vol");
  auto binary_image    = SH3::makeBinaryImage(filename, params );
  auto K               = SH3::getKSpace( binary_image, params );
  auto surface         = SH3::makeDigitalSurface( binary_image, K, params );
  auto surfels   = SH3::getSurfelRange( surface, params );
 
  //Parallel
  const auto axis = 2;
  const auto nbThreads = 4;
  trace.beginBlock(std::to_string(nbThreads) + "threads on axis "+std::to_string(axis));
  auto curv = SHG3::getIIMeanCurvatures( binary_image, surfels,
                                        params( "ii-thread-number", nbThreads )
                                        ( "ii-split-axis", axis )
                                        ( "r-radius", 8 ));
  trace.endBlock();
 
  PolyscopeViewer viewer;
  std::string objectName = "Surfels";
  viewer.draw(surfels, objectName); // Draws the object independently
  viewer.addQuantity(objectName, "Mean curvature", curv);
 
  AxisDomainSplitter<Z3i::Domain> splitter(axis);
  AxisDomainSplitter<Z3i::Domain>::SplitDomainsInfo splits = splitter(binary_image->domain(), nbThreads);
  HueShadeColorMap<unsigned int> cmap(0,(unsigned int)splits.size());
  for(auto i=0; i< splits.size(); ++i)
  {
    viewer << cmap(i);
    viewer << splits[i].domain;
  }
  viewer.show();
 
  return 0;
}
//                                                                           //