1 #include "ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h"
2 #include "MinimumSpanningTree.h"
4 #include <cpPlugins/Image.h>
5 #include <itkSimpleDataObjectDecorator.hxx>
6 #include <fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h>
7 #include <fpa/Base/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.hxx>
8 #include <fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.hxx>
10 // -------------------------------------------------------------------------
11 fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
12 ExtractEndPointsAndBifurcationsFromMinimumSpanningTree( )
15 this->_AddInput( "CostsImage" );
16 this->_AddInput( "DistanceMap" );
17 this->_AddInput( "MST" );
18 this->_AddOutput< cpPlugins::DataObject >( "EndPoints" );
19 this->_AddOutput< cpPlugins::DataObject >( "Bifurcations" );
20 this->_AddOutput< cpPlugins::DataObject >( "Collisions" );
22 this->m_Parameters.ConfigureAsBool( "SquaredDistanceMap" );
23 this->m_Parameters.SetBool( "SquaredDistanceMap", false );
26 // -------------------------------------------------------------------------
27 fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
28 ~ExtractEndPointsAndBifurcationsFromMinimumSpanningTree( )
32 // -------------------------------------------------------------------------
33 std::string fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
36 auto image = this->GetInputData( "CostsImage" )->GetITK< itk::DataObject >( );
37 std::string r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, float, 2 );
38 if( r != "" ) r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, double, 2 );
39 if( r != "" ) r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, float, 3 );
40 if( r != "" ) r = cpPlugin_Image_Demangle_Pixel_Dim( _GD0, image, double, 3 );
44 // -------------------------------------------------------------------------
45 template< class _TImage >
46 std::string fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
47 _GD0( _TImage* image )
49 typedef fpa::Image::MinimumSpanningTree< _TImage::ImageDimension > _TMST;
50 typedef fpa::Image::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree< _TImage, _TMST > _TFilter;
52 // Check input objects' integrity
55 "fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree: No valid input costs image."
57 auto dmap = this->GetInputData( "DistanceMap" )->GetITK< _TImage >( );
60 "fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree: Distance map does not have the same type as the costs image."
62 auto mst = this->GetInputData( "MST" )->GetITK< _TMST >( );
65 "fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree: No valid input tree."
68 // Create filter and connect inputs
69 _TFilter* filter = this->_CreateITK< _TFilter >( );
70 filter->SetCostsImage( image );
71 filter->SetDistanceMap( dmap );
72 filter->SetMinimumSpanningTree( mst );
73 filter->SetSquaredDistanceMap(
74 this->m_Parameters.GetBool( "SquaredDistanceMap" )
78 this->GetOutputData( "EndPoints" )->SetITK( filter->GetEndPoints( ) );
79 this->GetOutputData( "Bifurcations" )->SetITK( filter->GetBifurcations( ) );
80 this->GetOutputData( "Collisions" )->SetITK( filter->GetCollisions( ) );