volFillInterior.cpp

#include <iostream>
#include <DGtal/base/Common.h>
#include <DGtal/io/readers/VolReader.h>
#include <DGtal/io/writers/VolWriter.h>
#include <DGtal/helpers/StdDefs.h>
#include <DGtal/images/Image.h>
#include <DGtal/images/ImageContainerBySTLVector.h>

#include "CLI11.hpp"

using namespace std;
using namespace DGtal;
using namespace Z3i;

void missingParam ( const std::string &param )
{
  trace.error() <<" Parameter: "<<param<<" is required..";
  trace.info() <<std::endl;
  exit ( 1 );
}


int main(int argc, char**argv)
{
  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::string inputFileName;
  std::string outputFileName {"result.vol"};

  app.description("Fill the interior of a voxel set by filling the exterior using the 6-adjacency.\nThe exterior is the set of voxels with value zero and the interior voxels have value 128\n Basic usage:\n\tvolFillInterior <volFileName> <volOutputFileName> ");
  
  app.add_option("-i,--input,1", inputFileName, "Input vol file." )
  ->required()
  ->check(CLI::ExistingFile);
  app.add_option("-o,--output,2",outputFileName, "Output filename.", true);
  
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------

  
  trace.beginBlock("Loading");
  typedef ImageContainerBySTLVector<Z3i::Domain, unsigned char>  MyImageC;
  MyImageC  image = VolReader< MyImageC >::importVol ( inputFileName );
  trace.info() << image << std::endl;
  trace.endBlock();
  
  //Flag image
  ImageContainerBySTLVector<Z3i::Domain, bool>  imageFlag(image.domain());
  for(auto &p: imageFlag.domain())
  {
    if (image(p) != 0)
      imageFlag.setValue(p, true);
  }
  
  std::stack<Z3i::Point> pstack;
  pstack.push(*(image.domain().begin()));
  FATAL_ERROR_MSG(image(pstack.top())==0, "Starting point of the domain must be equal to zero.");
  
  //6-Pencil for the exterior propagation
  std::vector<Z3i::Point> pencil6= { {1,0,0}, {0,1,0}, {0,0,1},{-1,0,0}, {0,-1,0}, {0,0,-1}  };
  
  trace.beginBlock("Filling");
  while (!pstack.empty())
  {
    Z3i::Point p = pstack.top();
    pstack.pop();
    imageFlag.setValue(p, true);

    for(auto & delta: pencil6)
      if ((image.domain().isInside(p + delta)) &&
          (imageFlag( p + delta) == false))
        pstack.push( p + delta);
  }
  trace.endBlock();
  
  trace.beginBlock("Complement");
  for(auto &p :  image.domain())
    if ((image(p) == 0) && (!imageFlag(p)))
      image.setValue(p,128);
  trace.endBlock();
  
  trace.beginBlock("Saving");
  bool res =  VolWriter< MyImageC >::exportVol(outputFileName, image);
  trace.endBlock();
  
  if (res) return 0; else return 1;
}