curvature-comparator-ii-cnc-3d.cpp File Reference

#include <iostream>
#include <fstream>
#include <algorithm>
#include "DGtal/base/Common.h"
#include "DGtal/shapes/SurfaceMesh.h"
#include "DGtal/geometry/meshes/CorrectedNormalCurrentComputer.h"
#include "DGtal/helpers/Shortcuts.h"
#include "DGtal/helpers/ShortcutsGeometry.h"
#include "DGtal/io/writers/SurfaceMeshWriter.h"
#include "DGtal/io/colormaps/GradientColorMap.h"
#include "DGtal/io/colormaps/QuantifiedColorMap.h"

Go to the source code of this file.

Functions
DGtal::GradientColorMap< double >	makeColorMap (double min_value, double max_value)
	[curvature-comparator-Includes] More...
void	usage (int argc, char *argv[])
int	main (int argc, char *argv[])

Detailed Description

This program is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program. If not, see http://www.gnu.org/licenses/.

Author

Jacques-Olivier Lachaud (jacqu.nosp@m.es-o.nosp@m.livie.nosp@m.r.la.nosp@m.chaud.nosp@m.@uni.nosp@m.v-sav.nosp@m.oie..nosp@m.fr ) Laboratory of Mathematics (CNRS, UMR 5127), University of Savoie, France

Date

2022/02/27

An example file named curvature-comparator-ii-cnc-3d.

This file is part of the DGtal library.

Definition in file curvature-comparator-ii-cnc-3d.cpp.

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

[curvature-comparator-Typedefs]

[curvature-comparator-Typedefs]

[curvature-comparator-DigitalSurface]

[curvature-comparator-DigitalSurface]

[curvature-comparator-ii]

[curvature-comparator-ii]

[curvature-comparator-SurfaceMesh]

[curvature-comparator-SurfaceMesh]

[curvature-comparator-TrueCurvatures]

[curvature-comparator-TrueCurvatures]

[curvature-comparator-CNC]

[curvature-comparator-CNC]

[curvature-comparator-estimations]

[curvature-comparator-estimations]

[curvature-comparator-check]

[curvature-comparator-check]

[curvature-comparator-results]

[curvature-comparator-results]

Definition at line 126 of file curvature-comparator-ii-cnc-3d.cpp.

{
  if ( argc <= 1 )
    {
      usage( argc, argv );
      return 0;
    }
  using namespace DGtal;
  using namespace DGtal::Z3i;
  typedef SurfaceMesh< RealPoint, RealVector >                    SM;
  typedef CorrectedNormalCurrentComputer< RealPoint, RealVector > CNC;
  typedef Shortcuts< KSpace >          SH;
  typedef SH::DigitalSurface           DigSurf;
  typedef ShortcutsGeometry< KSpace >  SHG;
  std::string  poly = argv[ 1 ]; // polynomial
  const double    B = argc > 2 ? atof( argv[ 2 ] ) : 1.0; // max ||_oo bbox
  const double    h = argc > 3 ? atof( argv[ 3 ] ) : 1.0; // gridstep  
  std::string  mode = argc > 4 ? argv[ 4 ] : "Const"; // either Const or Interp
  bool interpolated = mode == "Interp";
  if ( interpolated )
    trace.info() << "Using vertex-*Interpolated* Corrected Normal Current" << std::endl;
  else
    trace.info() << "Using face-*Constant* Corrected Normal Current" << std::endl;
  // Read polynomial and build digital surface
  auto params = SH::defaultParameters() | SHG::defaultParameters();
  // Choose depth-first traversal to speed-up II computations.
  params( "surfaceTraversal", "DepthFirst" );
  params( "t-ring", 3 );
  params( "polynomial", poly )( "gridstep", h ); 
  params( "minAABB", -B )( "maxAABB", B );
  params( "offset", 3.0 );
  auto shape       = SH::makeImplicitShape3D( params );
  auto K           = SH::getKSpace( params );
  auto dshape      = SH::makeDigitizedImplicitShape3D( shape, params );
  auto bimage      = SH::makeBinaryImage( dshape, params );
  if ( bimage == nullptr ) 
    {
      trace.error() <<  "Unable to read polynomial <"
                    << poly.c_str() << ">" << std::endl;
      return 1;
    }
  auto sembedder   = SH::getSCellEmbedder( K );
  auto embedder    = SH::getCellEmbedder( K );
  auto surface     = SH::makeDigitalSurface( bimage, K, params );
  auto surfels     = SH::getSurfelRange( surface, params );
  trace.info() << "- surface has " << surfels.size()<< " surfels." << std::endl;
 
  params( "r-radius", 3.0 );
  params( "alpha",    0.33 );
  double ii_r = 3.0 * pow( h, 0.33 );
  Clock c;
  trace.beginBlock( "Computing II curvatures" );
  std::vector< double > HII  = SHG::getIIMeanCurvatures    ( bimage, surfels, params );
  std::vector< double > GII  = SHG::getIIGaussianCurvatures( bimage, surfels, params );
  auto ii_t = trace.endBlock();
  
  SM smesh;
  std::vector< SM::Vertices > faces;
  SH::Cell2Index c2i;
  auto pointels = SH::getPointelRange( c2i, surface );
  auto vertices = SH::RealPoints( pointels.size() );
  std::transform( pointels.cbegin(), pointels.cend(), vertices.begin(),
                  [&] (const SH::Cell& c) { return h * embedder( c ); } ); 
  for ( auto&& surfel : surfels )
    {
      const auto primal_surfel_vtcs = SH::getPointelRange( K, surfel );
      SM::Vertices face;              
      for ( auto&& primal_vtx : primal_surfel_vtcs )
        face.push_back( c2i[ primal_vtx ] );
      faces.push_back( face );
    }
  smesh.init( vertices.cbegin(), vertices.cend(),
              faces.cbegin(),    faces.cend() );
  trace.info() << smesh << std::endl;
 
  trace.beginBlock( "Computing True curvatures" );
  auto exp_H = SHG::getMeanCurvatures( shape, K, surfels, params );
  auto exp_G = SHG::getGaussianCurvatures( shape, K, surfels, params );
  trace.endBlock();
 
  trace.beginBlock( "Computing CNC curvatures" );
  // Builds a CorrectedNormalCurrentComputer object onto the SurfaceMesh object
  CNC cnc( smesh );
  // Estimates normal vectors using Integral Invariant Normal estimator 
  trace.beginBlock( "Computing II normal vectors" );
  auto face_normals = SHG::getIINormalVectors( bimage, surfels, params );
  double cnc_tn = trace.endBlock();
  // Set corrected face normals => Corrected Normal Current with
  // constant per face corrected vector field.
  smesh.setFaceNormals( face_normals.cbegin(), face_normals.cend() ); // CCNC
  // Set corrected vertex normals => Corrected Normal Current with
  // smooth linearly interpolated per face corrected vector field.
  if ( interpolated ) smesh.computeVertexNormalsFromFaceNormals();    // ICNC
  // computes area, mean and Gaussian curvature measures
  auto mu0 = cnc.computeMu0();
  auto mu1 = cnc.computeMu1();
  auto mu2 = cnc.computeMu2();
 
  // estimates mean (H) and Gaussian (G) curvatures by measure normalization.
  double cnc_mr = 1.0 * sqrt( h );
  trace.info() << "CNC measuring radius = " << cnc_mr << std::endl;
  std::vector< double > H( smesh.nbFaces() );
  std::vector< double > G( smesh.nbFaces() );
  for ( auto f = 0; f < smesh.nbFaces(); ++f )
    {
      const auto b    = smesh.faceCentroid( f );
      const auto area = mu0.measure( b, cnc_mr, f );
      H[ f ] = cnc.meanCurvature    ( area, mu1.measure( b, cnc_mr, f ) );
      G[ f ] = cnc.GaussianCurvature( area, mu2.measure( b, cnc_mr, f ) );
    }
  double cnc_t = trace.endBlock();
  
  auto HII_min_max = std::minmax_element( HII.cbegin(), HII.cend() );
  auto GII_min_max = std::minmax_element( GII.cbegin(), GII.cend() );
  auto H_min_max = std::minmax_element( H.cbegin(), H.cend() );
  auto G_min_max = std::minmax_element( G.cbegin(), G.cend() );
  auto exp_H_min_max = std::minmax_element( exp_H.cbegin(), exp_H.cend() );
  auto exp_G_min_max = std::minmax_element( exp_G.cbegin(), exp_G.cend() );
  trace.info() << "Expected mean curvatures:"
            << " min=" << *exp_H_min_max.first << " max=" << *exp_H_min_max.second
            << std::endl;
  trace.info() << "Computed II mean curvatures:"
            << " min=" << *HII_min_max.first << " max=" << *HII_min_max.second
            << std::endl;
  trace.info() << "Computed CNC mean curvatures:"
            << " min=" << *H_min_max.first << " max=" << *H_min_max.second
            << std::endl;
  trace.info() << "Expected Gaussian curvatures:"
            << " min=" << *exp_G_min_max.first << " max=" << *exp_G_min_max.second
            << std::endl;
  trace.info() << "Computed II Gaussian curvatures:"
            << " min=" << *GII_min_max.first << " max=" << *GII_min_max.second
            << std::endl;
  trace.info() << "Computed CNC Gaussian curvatures:"
            << " min=" << *G_min_max.first << " max=" << *G_min_max.second
            << std::endl;
  const auto      error_HII = SHG::getScalarsAbsoluteDifference( HII, exp_H );
  const auto stat_error_HII = SHG::getStatistic( error_HII );
  const auto   error_HII_l2 = SHG::getScalarsNormL2( HII, exp_H );
  trace.info() << "|H-H_II|_oo = " << stat_error_HII.max() << std::endl;
  trace.info() << "|H-H_II|_2  = " << error_HII_l2 << std::endl;
  const auto      error_H = SHG::getScalarsAbsoluteDifference( H, exp_H );
  const auto stat_error_H = SHG::getStatistic( error_H );
  const auto   error_H_l2 = SHG::getScalarsNormL2( H, exp_H );
  trace.info() << "|H-H_CNC|_oo = " << stat_error_H.max() << std::endl;
  trace.info() << "|H-H_CNC|_2  = " << error_H_l2 << std::endl;
  const auto      error_GII = SHG::getScalarsAbsoluteDifference( GII, exp_G );
  const auto stat_error_GII = SHG::getStatistic( error_GII );
  const auto   error_GII_l2 = SHG::getScalarsNormL2( GII, exp_G );
  trace.info() << "|G-G_II|_oo = " << stat_error_GII.max() << std::endl;
  trace.info() << "|G-G_II|_2  = " << error_GII_l2 << std::endl;
  const auto      error_G = SHG::getScalarsAbsoluteDifference( G, exp_G );
  const auto stat_error_G = SHG::getStatistic( error_G );
  const auto   error_G_l2 = SHG::getScalarsNormL2( G, exp_G );
  trace.info() << "|G-G_CNC|_oo = " << stat_error_G.max() << std::endl;
  trace.info() << "|G-G_CNC|_2  = " << error_G_l2 << std::endl;
 
  std::cout << "# " << argv[ 0 ] << std::endl
            << "# polynomial: " << poly << std::endl
            << "# CNC mode:   " << mode << std::endl;
  //              1 2          3    4    5      6     7      8
  std::cout << "# h nb_surfels ii_t ii_r ii_Hoo ii_H2 ii_Goo ii_G2 ";
  //            9      10    11     12      13     14      15
  std::cout << "cnc_tn cnc_t cnc_mr cnc_Hoo cnc_H2 cnc_Goo cnc_G2" << std::endl;
  std::cout << h << " " << surfels.size() << " " << ii_t << " " << ii_r
            << " " << stat_error_HII.max() << " " << error_HII_l2
            << " " << stat_error_GII.max() << " " << error_GII_l2
            << " " << cnc_tn << " " << cnc_t << " " << cnc_mr
            << " " << stat_error_H.max() << " " << error_H_l2
            << " " << stat_error_G.max() << " " << error_G_l2 << std::endl;
  return 0;
}

References DGtal::Trace::beginBlock(), DGtal::Trace::endBlock(), DGtal::Trace::error(), DGtal::Trace::info(), and K.

makeColorMap()

DGtal::GradientColorMap< double > makeColorMap	(	double	min_value,
		double	max_value
	)

[curvature-comparator-Includes]

[curvature-comparator-Includes]

Examples

geometry/meshes/curvature-comparator-ii-cnc-3d.cpp, geometry/meshes/curvature-measures-icnc-3d.cpp, geometry/meshes/curvature-measures-icnc-XY-3d.cpp, geometry/meshes/curvature-measures-nc-3d.cpp, geometry/meshes/curvature-measures-nc-XY-3d.cpp, geometry/meshes/digpoly-curvature-measures-cnc-3d.cpp, geometry/meshes/digpoly-curvature-measures-cnc-XY-3d.cpp, geometry/meshes/obj-curvature-measures-icnc-3d.cpp, geometry/meshes/obj-curvature-measures-icnc-XY-3d.cpp, geometry/meshes/vol-curvature-measures-icnc-3d.cpp, and geometry/meshes/vol-curvature-measures-icnc-XY-3d.cpp.

Definition at line 89 of file curvature-comparator-ii-cnc-3d.cpp.

{
  DGtal::GradientColorMap< double > gradcmap( min_value, max_value );
  gradcmap.addColor( DGtal::Color( 0, 0, 255 ) );
  gradcmap.addColor( DGtal::Color( 0, 255, 255 ) );
  gradcmap.addColor( DGtal::Color( 255, 255, 255 ) );
  gradcmap.addColor( DGtal::Color( 255, 255, 0 ) );
  gradcmap.addColor( DGtal::Color( 255, 0, 0 ) );
  return gradcmap;
}

References DGtal::GradientColorMap< PValue, PDefaultPreset, PDefaultFirstColor, PDefaultLastColor >::addColor().

usage()

void usage	(	int	argc,
		char *	argv[]
	)

Definition at line 100 of file curvature-comparator-ii-cnc-3d.cpp.

{
  using namespace DGtal;
  using namespace DGtal::Z3i;
  typedef Shortcuts< KSpace >          SH;
  std::cout << "Usage: " << std::endl
            << "\t" << argv[ 0 ] << " <P> <B> <h> <mode>" << std::endl
            << std::endl
            << "Compare Integral Invariant (II) curvature estimations "  << std::endl
            << "with Corrected Normal Current (CNC) estimations, either "<< std::endl
            << "interpolated (Interp) or constant per face (Const)."     << std::endl
            << "- builds the surface mesh from polynomial <P>"           << std::endl
            << "- <B> defines the digitization space size [-B,B]^3"      << std::endl
            << "- <h> is the gridstep digitization"                      << std::endl
            << "- <mode> is either Const for constant corrected normal"  << std::endl
            << "  vector field or Interp for interpolated corrected"     << std::endl
            << "  normal vector field."                                  << std::endl
            << "It outputs timings and accuracy statistics for both "    << std::endl
            << "methods."                                                << std::endl;
  std::cout << "You may either write your own polynomial as 3*x^2-z^2*x*y+1" << std::endl
            <<"or use a predefined polynomial in the following list:" << std::endl;
  auto L = SH::getPolynomialList();
  for ( const auto& p : L )
    std::cout << p.first << " : " << p.second << std::endl;
}