plugins/fpa/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.cxx

   1 #include "ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h"
   2 #include "MinimumSpanningTree.h"
   3
   4 #include <cpPlugins/Image.h>
   5 #include <fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h>
   6 #include <fpa/Base/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.hxx>
   7 #include <fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.hxx>
   8
   9 // -------------------------------------------------------------------------
  10 fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
  11 ExtractEndPointsAndBifurcationsFromMinimumSpanningTree( )
  12   : Superclass( )
  13 {
  14   this->_AddInput( "CostsImage" );
  15   this->_AddInput( "DistanceMap" );
  16   this->_AddInput( "MST" );
  17   this->_AddOutput< cpPlugins::DataObject >( "EndPoints" );
  18   this->_AddOutput< cpPlugins::DataObject >( "Bifurcations" );
  19   this->_AddOutput< cpPlugins::DataObject >( "Collisions" );
  20
  21   this->m_Parameters.ConfigureAsBool( "SquaredDistanceMap" );
  22   this->m_Parameters.SetBool( "SquaredDistanceMap", false );
  23 }
  24
  25 // -------------------------------------------------------------------------
  26 fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
  27 ~ExtractEndPointsAndBifurcationsFromMinimumSpanningTree( )
  28 {
  29 }
  30
  31 // -------------------------------------------------------------------------
  32 std::string fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
  33 _GenerateData( )
  34 {
  35   auto image = this->GetInputData( "CostsImage" )->GetITK< itk::DataObject >( );
  36   std::string   r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, float, 2 );
  37   if( r != "" ) r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, double, 2 );
  38   if( r != "" ) r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, float, 3 );
  39   if( r != "" ) r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, double, 3 );
  40   return( r );
  41 }
  42
  43 // -------------------------------------------------------------------------
  44 template< class _TImage >
  45 std::string fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
  46 _GD0( _TImage* image )
  47 {
  48   typedef fpa::Image::MinimumSpanningTree< _TImage::ImageDimension > _TMST;
  49   typedef fpa::Image::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree< _TImage, _TMST > _TFilter;
  50
  51   // Check input objects' integrity
  52   if( image == NULL )
  53     return(
  54       "fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree: No valid input costs image."
  55       );
  56   auto dmap = this->GetInputData( "DistanceMap" )->GetITK< _TImage >( );
  57   if( dmap == NULL )
  58     return(
  59       "fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree: Distance map does not have the same type as the costs image."
  60       );
  61   auto mst = this->GetInputData( "MST" )->GetITK< _TMST >( );
  62   if( mst == NULL )
  63     return(
  64       "fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree: No valid input tree."
  65       );
  66
  67   // Create filter and connect inputs
  68   _TFilter* filter = this->_CreateITK< _TFilter >( );
  69   filter->SetCostsImage( image );
  70   filter->SetDistanceMap( dmap );
  71   filter->SetMinimumSpanningTree( mst );
  72   filter->SetSquaredDistanceMap(
  73     this->m_Parameters.GetBool( "SquaredDistanceMap" )
  74     );
  75   filter->Update( );
  76
  77   this->GetOutputData( "EndPoints" )->SetITK( filter->GetEndPoints( ) );
  78   this->GetOutputData( "Bifurcations" )->SetITK( filter->GetBifurcations( ) );
  79   this->GetOutputData( "Collisions" )->SetITK( filter->GetCollisions( ) );
  80   return( "" );
  81 }
  82
  83 // eof - $RCSfile$