generateVoxelComplexTables.cpp

Go to the documentation of this file.

#include <iostream>
#include <vector>
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/topology/VoxelComplexFunctions.h"
#include "DGtal/topology/tables/NeighborhoodTablesGenerators.h"

 
using namespace std;
using namespace DGtal;

 
 
 
int main( int argc, char** argv )
{
  typedef std::bitset<67108864> ConfigMap; // 2^26
 
  using namespace Z3i;
  using VoxelComplex = VoxelComplex<KSpace>;
 
  std::function< bool(
                 const VoxelComplex & ,
                 const typename VoxelComplex::Cell & )
               > skelFunction;
  string error_message(
      "Provide one of the following arguments for select function:\n"
      "- skelIsthmus \n"
      "- oneIsthmus \n"
      "- twoIsthmus \n");
  if (argc != 2 ){
    cout << error_message << std::endl;
    return 1;
  }
  std::string input_str = std::string(argv[1]);
  if (input_str == "skelIsthmus")
    skelFunction = functions::skelIsthmus<VoxelComplex>;
  else if (input_str == "oneIsthmus")
    skelFunction = functions::oneIsthmus<VoxelComplex>;
  else if (input_str == "twoIsthmus")
    skelFunction = functions::twoIsthmus<VoxelComplex>;
  else{
    cout << error_message << endl;
    return 1;
  }
 
  trace.beginBlock ( "Generate " + input_str + " table for 26_6 topology" );
  // Too big for stack. Use heap instead.
  auto table26_6 = make_shared<ConfigMap>();
  functions::generateVoxelComplexTable< VoxelComplex >(
      *table26_6,
      skelFunction );
 
  string filename = input_str + "_table26_6.txt";
  trace.info() << "Save to file... " + filename << std::endl;
  ofstream file26_6( filename );
  file26_6 << *table26_6;
  file26_6.close();
 
  trace.endBlock();
 
  return 0;
}
//                                                                           //