rigidTransform3D.cpp

 
#include <iostream>
#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/ImageSelector.h>
#include <DGtal/images/ConstImageAdapter.h>
#include <DGtal/images/RigidTransformation3D.h>
 
#include "CLI11.hpp"
 
using namespace std;
using namespace DGtal;
using namespace Z3i;
using namespace functors;
 
int main(int argc, char**argv)
{
 
  // parse command line CLI ----------------------------------------------
  CLI::App app;
  std::string filename;
  std::string outputFileName;
  std::string model;
  double angle {0};
  double ox {0}, oy {0}, oz {0};
  double ax {1}, ay {0}, az {0};
  double tx {0}, ty {0}, tz {0};
 
  app.description("Apply rigid transformation on a given volumic image.\n Typical use example:\n \t rigidTrans3D  <RawFileName>  <VolOutputFileName> --ox 1.0 --oy 1.0 --oz 1 -a 1.2 --ax 1 --ay 1 --az 0 --tx 1 --ty 0 --tz 0 --m <forward|backward>\n");
  app.add_option("-i,--input,1",filename,"Input file.")->required()->check(CLI::ExistingFile);
  app.add_option("-o,--output,2",outputFileName,"Output file.")->required();
  app.add_option("-m,--model",model,"Transformation model: backward, forward.")->required();
  app.add_option("-a,--angle",angle,"Rotation angle in radians (default 0)",true);
  app.add_option("--ox",ox,"X coordinate of origin (default 0)",true);
  app.add_option("--oy",oy,"Y coordinate of origin (default 0)",true);
  app.add_option("--oz",oz,"Z coordinate of origin (default 0)",true);
  app.add_option("--ax",ax,"X component of rotation axis (default 1)",true);
  app.add_option("--ay",ay,"Y component of rotation axis (default 0)",true);
  app.add_option("--az",az,"Z component of rotation axis (default 0)",true);
  app.add_option("--tx",tx,"X component of translation vector (default 0)",true);
  app.add_option("--ty",ty,"Y component of translation vector (default 0)",true);
  app.add_option("--tz",tz,"Y component of translation vector (default 0)",true);
 
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------
 
  typedef ImageSelector<Domain, unsigned char >::Type Image;
  typedef ForwardRigidTransformation3D < Space > ForwardTrans;
  typedef BackwardRigidTransformation3D < Space > BackwardTrans;
  typedef ConstImageAdapter<Image, Domain, BackwardTrans, Image::Value, Identity > MyImageBackwardAdapter;
  typedef DomainRigidTransformation3D < Domain, ForwardTrans > MyTransformedDomain;
  typedef MyTransformedDomain::Bounds Bounds;
 
  Image image = GenericReader <Image, 3>::import(filename);
  ForwardTrans forwardTrans( RealPoint ( ox, oy, oz ), RealPoint ( ax, ay, az ), angle, RealVector( tx, ty, tz ) );
  BackwardTrans backwardTrans( RealPoint ( ox, oy, oz ), RealPoint ( ax, ay, az ), angle, RealVector( tx, ty, tz ) );
  MyTransformedDomain domainForwardTrans ( forwardTrans );
  Bounds bounds = domainForwardTrans ( image.domain() );
  Domain transformedDomain ( bounds.first, bounds.second );
  Identity idD;
 
  if ( model == "forward" )
  {
      Image forwardTransformedImage ( transformedDomain );
      for ( Domain::ConstIterator it = image.domain().begin(); it != image.domain().end(); ++it )
      {
          forwardTransformedImage.setValue ( forwardTrans ( *it ), image ( *it ) );
      }
      GenericWriter<Image, 3, unsigned char, Identity>::exportFile(outputFileName, forwardTransformedImage," ", idD);
  }
  else
  {
      MyImageBackwardAdapter backwardImageAdapter ( image, transformedDomain , backwardTrans, idD );
      GenericWriter<MyImageBackwardAdapter, 3, unsigned char, Identity>::exportFile(outputFileName, backwardImageAdapter, " ", idD);
  }
  return 0;
}