testRigidTransformation2D.cpp

Go to the documentation of this file.

 
#include <iostream>
#include <cmath>
#include <DGtal/images/ImageSelector.h>
#include <DGtal/images/ImageContainerBySTLVector.h>
#include "DGtal/images/ConstImageAdapter.h"
#include "DGtal/base/Common.h"
#include "ConfigTest.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/images/RigidTransformation2D.h"
#include "DGtal/io/readers/PGMReader.h"
#include "DGtal/io/writers/GenericWriter.h"
 
 
using namespace std;
using namespace DGtal;
using namespace functors;
using namespace Z2i;
 
// Functions for testing class testRigidTransformation2D.
 
class testRigidTransformation2D
{
  typedef ImageSelector<Domain, unsigned char >::Type Image;
  typedef ForwardRigidTransformation2D < Space > ForwardTrans;
  typedef BackwardRigidTransformation2D < Space > BackwardTrans;
  typedef ConstImageAdapter<Image, Domain, BackwardTrans, Image::Value, Identity > MyImageBackwardAdapter;
  typedef DomainRigidTransformation2D < Domain, ForwardTrans > DomainTrans;
  typedef DomainTrans::Bounds Bounds;
private:
  Image binary;
  Image gray;
  ForwardTrans forwardTrans;
  BackwardTrans backwardTrans;
  Identity idD;
  DomainTrans domainForwardTrans;
public:
  // Setup part
  testRigidTransformation2D() : 
  binary ( Domain ( Point ( 0,0 ), Point ( 10, 10 ) ) ),
  gray ( PGMReader<Image>::importPGM ( testPath + "samples/church-small.pgm" ) ),
  forwardTrans ( Point ( 5, 5 ), M_PI_4, RealVector( 3, -3 ) ),
  backwardTrans ( Point ( 5, 5 ), M_PI_4, RealVector( 3, -3 ) ),
  domainForwardTrans ( forwardTrans )
    {
      binary.setValue ( Point ( 3,3 ), 255 );
      binary.setValue ( Point ( 3,4 ), 255 );
      binary.setValue ( Point ( 4,3 ), 255 );
      binary.setValue ( Point ( 4,4 ), 255 );
      
      binary >> "binary.pgm";
      gray >> "gray.pgm"; 
    }
    
    bool forwardTransformationBinary ()
    {
      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.pgm";
      return true;
    }
    
    bool backwardTransformationBinary ()
    {
      Bounds bounds = domainForwardTrans ( binary.domain() );
      Domain d ( bounds.first, bounds.second );
      MyImageBackwardAdapter adapter ( binary, d, backwardTrans, idD );
      adapter >> "binary_after_backward.pgm";
      return true;
    }
    
    bool backwardTransformationGray ()
    {
      Bounds bounds = domainForwardTrans ( gray.domain() );
      Domain d ( bounds.first, bounds.second );
      MyImageBackwardAdapter adapter ( gray, d, backwardTrans, idD );
      adapter >> "gray_after_backward.pgm";
      return true;
    }
    
    bool forwardTransformationGray ()
    {
      Bounds bounds = domainForwardTrans ( gray.domain() );
      Domain d ( bounds.first, bounds.second );
      Image transformed ( d );
      for ( Domain::ConstIterator it = gray.domain().begin(); it != gray.domain().end(); ++it )
      {
        transformed.setValue ( forwardTrans ( *it ), gray ( *it ) );
      }
      transformed >> "gray_after_forward.pgm";
      return true;
    }
};
 
// Standard services - public :
 
int main( int, char** )
{
  bool res = true;
  testRigidTransformation2D rigidTest;
  trace.beginBlock ( "Testing RigidTransformation2D" );
    res &= rigidTest.forwardTransformationBinary();
    res &= rigidTest.backwardTransformationBinary();
    res &= rigidTest.backwardTransformationGray();
    res &= rigidTest.forwardTransformationGray();
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();
  return res ? 0 : 1;
}
//                                                                           //