volAddBorder.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 ( 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"};
  bool addInside {false};

  app.description("Add a border of one voxel with value 0 around a vol file.\n \tvolAddBorder --input <volFileName> --o <volOutputFileName>\n Example: \n \t volAddBorder -i $DGtal/examples/samples/Al.100.vol -o Al.100border.vol");
  
  app.add_option("-i,--input,1", inputFileName, "Input vol file." )
  ->required()
  ->check(CLI::ExistingFile);
  app.add_flag("--inside", addInside , "Sets zero value to domain boundary voxels without changing the domain extent.");
  app.add_option("--output,-o,2", outputFileName, "Output filename.");
  
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------

  
  typedef ImageContainerBySTLVector<Z3i::Domain, unsigned char>  MyImageC;
  
  MyImageC  imageC = VolReader< MyImageC >::importVol ( inputFileName );
  Z3i::Domain rDom  ( imageC.domain().lowerBound() -  Vector().diagonal(addInside ? 0: 1),
                      imageC.domain().upperBound() + Vector().diagonal(addInside ? 0: 1));
  MyImageC  outputImage(rDom);
  Z3i::Domain iDom  ( imageC.domain().lowerBound() +  Vector().diagonal( 1),
                      imageC.domain().upperBound() - Vector().diagonal( 1));

  //Fast Copy
  for(MyImageC::Domain::ConstIterator it = imageC.domain().begin(),
          itend = imageC.domain().end(); it != itend; ++it){
     if(!addInside){
          outputImage.setValue( *it , imageC(*it));
     }
     else
     {
       if (!iDom.isInside(*it)){
         imageC.setValue( *it , 0); 
       }
     }      
     
  }  
  bool res = true;
  if (!addInside) {
    res=  VolWriter< MyImageC>::exportVol(outputFileName, outputImage);
  }
  else{
    res=  VolWriter< MyImageC>::exportVol(outputFileName, imageC);
  }
  if (res) return 0; else return 1;
}