volIntensityScale.cpp

#include <iostream>
#include <DGtal/base/Common.h>
#include <DGtal/io/readers/VolReader.h>
#include <DGtal/io/writers/VolWriter.h>
#include "DGtal/io/writers/GenericWriter.h"
#include <DGtal/helpers/StdDefs.h>
#include <DGtal/images/ImageContainerBySTLVector.h>
#include <DGtal/images/ConstImageAdapter.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;
}


int main(int argc, char**argv)
{

  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::string inputFileName;
  std::string outputFileName {"result.vol"};
  int inMin {0};
  int inMax {255};
  int outMin {0};
  int outMax {255};
  
  
  app.description("Apply a linear rescaling of the image intensity from an input intensity interval [InMin, InMax] into an output interval [OutMin, OutMax].\n Basic usage:\n volIntensityScale --input <volFileName> --output <volOutputFileName>  (both files can be independently in vol, pgm3D, p3d format)\n Example: \n volIntensityScale ${DGtal}/examples/samples/lobster.vol  --inMin 0 --inMax 100 lobster0-100.vol");
  
  app.add_option("-i,--input,1", inputFileName, "Input vol file." )
  ->required()
  ->check(CLI::ExistingFile);
  
  app.add_option("-o,--output,2",outputFileName, "volumetric output file (.vol, .pgm, .pgm3d, .longvol) ", true);
  app.add_option("-m,--inMin", inMin,  "the min value of the input image.", true);
  app.add_option("-M,--inMax", inMax,  "the max value of the input image.", true);
  app.add_option("--outMin", outMin,  "the min value of the output image.", true);
  app.add_option("--outMax", outMax,  "the max value of the output image.", true);
  
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------

  
  trace.beginBlock("Loading file");

  typedef ImageContainerBySTLVector<Z3i::Domain, unsigned char>  MyImageC;
  typedef DGtal::functors::Rescaling<int ,unsigned char > RescalFCT;
  MyImageC image = VolReader< MyImageC,  RescalFCT  >::importVol( inputFileName,
                                                                  RescalFCT(inMin,
                                                                            inMax,
                                                                            outMin, outMax));  
  trace.endBlock();

  trace.beginBlock("Exporting...");
  bool res =  GenericWriter<MyImageC>::exportFile(outputFileName, image);
  trace.endBlock();
  if (res) return 0; else return 1;   
}