volImageMetrics.cpp

#include <iostream>
#include <limits>

#include <DGtal/base/Common.h>
#include <DGtal/io/readers/GenericReader.h>
#include <DGtal/io/writers/GenericWriter.h>
#include <DGtal/helpers/StdDefs.h>
#include <DGtal/images/Image.h>
#include <DGtal/images/ImageContainerBySTLVector.h>
#include <DGtal/images/imagesSetsUtils/SetFromImage.h>
#include <DGtal/math/Statistic.h>

#include "CLI11.hpp"

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

typedef ImageContainerBySTLVector < Z3i::Domain,  int > Image3D;
typedef ImageContainerBySTLVector < Z2i::Domain,  int > Image2D;

double
getRMSE(const Image3D & imageA, const Image3D &imageB){
  double sumDiff=0;
  for(Image3D::Domain::ConstIterator it = imageA.domain().begin(); it!=imageA.domain().end(); it++){
    sumDiff+=(imageA(*it)-imageB(*it))*(imageA(*it)-imageB(*it));
  }
  return sqrt(sumDiff/imageA.domain().size());
}

double
getPSNR(const Image3D & imageA, const Image3D &imageB, double rmsd){
  unsigned long long int d =  std::numeric_limits<Image3D::Value>::max();
  return 10.0*log10(d*d/rmsd);
}


int main(int argc, char**argv)
{
  
  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::string inputFileNameVolA;
  std::string inputFileNameVolB;

  app.description("Apply basic image measures (RMSE, PSNR) between two volumetric images A and B. \n Basic usage:\n \t volImageMetrics --volA <volAFilename> --volB <volBFilename> \n Typical use :\n  volImageMetrics -a imageA.vol  -b imageB.vol \n");
  app.add_option("-a,--volA,1", inputFileNameVolA, "Input filename of volume A (vol format, and other pgm3d can also be used)." )
  ->required()
  ->check(CLI::ExistingFile);
  app.add_option("-b,--volB,2", inputFileNameVolB, "Input filename of volume B (vol format, and other pgm3d can also be used)." )
  ->required()
  ->check(CLI::ExistingFile);
  
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------

  Image3D imageA = GenericReader<Image3D>::import(inputFileNameVolA);
  Image3D imageB = GenericReader<Image3D>::import(inputFileNameVolB);
   
  std::cout << "# Image based measures (generated with volImageMetrics) given with the image A: "<< inputFileNameVolA<< " and the image B: "<< inputFileNameVolB << endl;
  std::cout << "#  RMSE PSNR "<< endl;    
  
  double rmse= getRMSE(imageA, imageB);
  double psnr= getPSNR(imageA, imageB, rmse);
    
  std::cout << " " << rmse << " " << psnr << endl;
  return 1;
}