testGeodesicsInHeat.cpp

Go to the documentation of this file.

 
#include <iostream>
#include "DGtal/base/Common.h"
#include "ConfigTest.h"
#include "DGtalCatch.h"
#include "DGtal/helpers/StdDefs.h"
 
#include "DGtal/dec/PolygonalCalculus.h"
#include "DGtal/dec/GeodesicsInHeat.h"
#include "DGtal/shapes/SurfaceMesh.h"
#include "DGtal/shapes/MeshHelpers.h"
 
using namespace std;
using namespace DGtal;
using namespace Z3i;
 
TEST_CASE( "Testing GeodesicsInHeat" )
{
  typedef SurfaceMesh< RealPoint,RealPoint > Mesh;
  std::vector<RealPoint> positions = { RealPoint( 0, 0, 0 ) ,
    RealPoint( 1, 0, 0 ) ,
    RealPoint( 0, 1, 0 ) ,
    RealPoint( 1, 1, 0 ) ,
    RealPoint( 0, 0, 1 ) ,
    RealPoint( 1, 0, 1 ) ,
    RealPoint( 0, 1, 1 ) ,
    RealPoint( 1, 1, 1 ) ,
    RealPoint( 1, 0, 2 ) ,
    RealPoint( 0, 0, 2 ) };
  std::vector<Mesh::Vertices> faces = { { 1, 0, 2, 3 },
    { 0, 1, 5, 4 } ,
    { 1, 3, 7, 5 } ,
    { 3, 2, 6, 7 } ,
    { 2, 0, 4, 6 } ,
    { 4, 5, 8, 9 } };
  
  Mesh box(positions.cbegin(), positions.cend(),
           faces.cbegin(), faces.cend());
  
  PolygonalCalculus<RealPoint,RealVector> boxCalculus(box);
  
  
  SECTION("Construction and basic operators")
  {
    GeodesicsInHeat<PolygonalCalculus<RealPoint,RealVector>> heat(boxCalculus);
    REQUIRE( heat.isValid() == false );
    
    trace.beginBlock("init solvers");
    heat.init(0.1);
    trace.endBlock();
    REQUIRE( heat.isValid()  );
    
    heat.addSource(0);
    GeodesicsInHeat<PolygonalCalculus<RealPoint,RealVector>>::Vector d = heat.compute();
    REQUIRE( (size_t)d.size() == (size_t)positions.size() );
    REQUIRE( d[5] == Approx(1.444608) );
    heat.clearSource();
    auto sources = heat.source();
    REQUIRE(sources.sum() == 0);
  }
}