raw2vol.cpp

#include <iostream>
#include <DGtal/base/Common.h>
#include <DGtal/io/readers/RawReader.h>
#include <DGtal/io/writers/VolWriter.h>
#include <DGtal/io/readers/VolReader.h>
#include <DGtal/io/writers/RawWriter.h>
#include <DGtal/helpers/StdDefs.h>
#include "DGtal/io/colormaps/GrayscaleColorMap.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"};
   unsigned int x, y, z;
   app.description("Converts a  8-bit raw file to  vol.\n Basic example:\n \t raw2vol <RawFileName>  <VolOutputFileName> -x 128 -y 128 -z 128");
   app.add_option("-i,--input,1", inputFileName, "Input raw file." )
    ->required()
    ->check(CLI::ExistingFile);
   app.add_option("-o,--output,2",outputFileName,"Output vol filename.", true);
   app.add_option("--x,-x", x, "x extent." )
   ->required();
   app.add_option("--y,-y", y, "y extent." )
     ->required();
   app.add_option("--z,-z", z, "z extent." )
     ->required(); 


  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 = RawReader< MyImageC >::importRaw8 ( inputFileName, Z3i::Vector(x,y,z));
  bool res =  VolWriter< MyImageC>::exportVol(outputFileName, imageC);

  if (res)
     return EXIT_SUCCESS;
  else 
    {
      trace.error()<< "Error while exporting the volume."<<std::endl;
      return EXIT_FAILURE;
    }
}