lib/fpa/Filters/Image/DefaultTraits.h

   1 // =========================================================================
   2 // @author Leonardo Florez Valencia
   3 // @email florez-l@javeriana.edu.co
   4 // =========================================================================
   5 #ifndef __fpa__Filters__Image__DefaultTraits__h__
   6 #define __fpa__Filters__Image__DefaultTraits__h__
   7
   8 #include <set>
   9 #include <vector>
  10 #include <itkConceptChecking.h>
  11 #include <itkImageToImageFilter.h>
  12 #include <fpa/Filters/MarksWithCollisionsInterface.h>
  13 #include <fpa/Filters/SeedsInterface.h>
  14
  15 namespace fpa
  16 {
  17   namespace Filters
  18   {
  19     namespace Image
  20     {
  21       /**
  22        */
  23       template< class _TInputImage, class _TOutputImage, class _TMark >
  24       class DefaultTraits
  25       {
  26       public:
  27         typedef _TInputImage  TInputImage;
  28         typedef _TOutputImage TOutputImage;
  29
  30         class TInternalTraits
  31         {
  32         public:
  33           typedef _TMark TMark;
  34           typedef typename TInputImage::IndexType        TVertex;
  35           typedef typename TInputImage::PixelType        TInputValue;
  36           typedef typename TInputImage::PointType        TPoint;
  37           typedef typename TOutputImage::PixelType       TOutputValue;
  38           typedef typename TVertex::LexicographicCompare TVertexCompare;
  39           itkStaticConstMacro(
  40             Dimension, unsigned int, _TInputImage::ImageDimension
  41             );
  42
  43           typedef std::vector< TVertex > TNeighborhood;
  44           struct TSeed
  45           {
  46             TVertex Vertex;
  47             TPoint  Point;
  48             bool    IsPoint;
  49             TMark   FrontId;
  50             TSeed( const TMark& fid = 0 )
  51               : FrontId( fid )
  52               { }
  53           };
  54           typedef std::vector< TSeed > TSeeds;
  55
  56           struct TNode
  57           {
  58             TVertex Vertex;
  59             TVertex Parent;
  60             TMark   FrontId;
  61
  62             // Hack to hide the fact that nodes need to be (re-)initialized
  63             mutable TOutputValue Value;
  64           };
  65           struct TNodeCompare
  66           {
  67             bool operator()( const TNode& a, const TNode& b ) const
  68               {
  69                 static TVertexCompare c;
  70                 return( c( a.Vertex, b.Vertex ) );
  71               }
  72           };
  73           typedef std::set< TNode, TNodeCompare > TNodes;
  74         };
  75
  76         typedef itk::ImageToImageFilter< TInputImage, TOutputImage > TFilter;
  77         typedef fpa::Filters::MarksWithCollisionsInterface< TInternalTraits >  TMarksInterface;
  78         typedef fpa::Filters::SeedsInterface< TInternalTraits > TSeedsInterface;
  79
  80       protected:
  81         itkConceptMacro(
  82           Check_Dimensions,
  83           ( itk::Concept::SameDimension< TInputImage::ImageDimension, TOutputImage::ImageDimension > )
  84           );
  85       };
  86
  87     } // ecapseman
  88
  89   } // ecapseman
  90
  91 } // ecapseman
  92
  93 #endif // __fpa__Filters__Image__DefaultTraits__h__
  94 // eof - $RCSfile$