testRigidTransformation3D.cpp

Go to the documentation of this file.

#include <iostream>
#include <boost/graph/graph_concepts.hpp>
#include "DGtal/base/Common.h"
#include "ConfigTest.h"
#include "DGtal/helpers/StdDefs.h"
#include <DGtal/images/ImageSelector.h>
#include <DGtal/images/ImageContainerBySTLVector.h>
#include "DGtal/images/ConstImageAdapter.h"
#include "DGtal/images/RigidTransformation3D.h"
#include "DGtal/io/readers/PGMReader.h"
#include "DGtal/io/readers/VolReader.h"
#include "DGtal/io/writers/GenericWriter.h"

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

// Functions for testing class RigidTransformation3D.

class testRigidTransformation3D
{
  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 > DomainTrans;
  typedef DomainTrans::Bounds Bounds;
private:
  Image binary;
  ForwardTrans forwardTrans;
  BackwardTrans backwardTrans;
  Identity idD;
  DomainTrans domainForwardTrans;
public:
  // Setup part
  testRigidTransformation3D() :
  binary ( PGMReader<Image>::importPGM3D ( testPath + "samples/cat10.pgm3d" ) ),
  forwardTrans ( RealPoint ( 5, 5, 5 ), RealVector ( 1, 0, 1 ), M_PI_4, RealVector( 3, -3, 3 ) ),
  backwardTrans( RealPoint ( 5, 5, 5 ), RealVector ( 1, 0, 1 ), M_PI_4, RealVector( 3, -3, 3 ) ),
  domainForwardTrans(forwardTrans)
  {}
 
  bool forwardTransformation ()
  {
    Bounds bounds = domainForwardTrans ( binary.domain() );
    Domain d ( bounds.first, bounds.second );
    Image transformed ( d );
    for ( Domain::ConstIterator it = binary.domain().begin(); it != binary.domain().end(); ++it )
    {
      transformed.setValue ( forwardTrans ( *it ), binary ( *it ) );
    }
    transformed >> "binary_after_forward.pgm3d";
    return true;
  }
  bool backwardTransformation ()
  {
    Bounds bounds = domainForwardTrans ( binary.domain() );
    Domain d ( bounds.first, bounds.second );
    MyImageBackwardAdapter adapter ( binary, d, backwardTrans, idD );
    adapter >> "binary_after_backward.pgm3d";
    return true;
  }
};

// Standard services - public :
 
int main( int, char** )
{
  bool res = true;
  testRigidTransformation3D rigidTest;
  trace.beginBlock ( "Testing RigidTransformation3D" );
    res &= rigidTest.forwardTransformation();
    res &= rigidTest.backwardTransformation();
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();
  return res ? 0 : 1;
}
//                                                                           //