...

[FrontAlgorithms.git] / plugins / Plugins / ImageDijkstra.cxx

diff --git a/plugins/Plugins/ImageDijkstra.cxx b/plugins/Plugins/ImageDijkstra.cxx
index c5d34a4290453646061f6e28687baeb92eeaf8ee..7017c77af94f38605b68ae00327f7cf1e79b849f 100644 (file)
--- a/plugins/Plugins/ImageDijkstra.cxx
+++ b/plugins/Plugins/ImageDijkstra.cxx
@@ -71,9 +71,12 @@ _GD1( _TInputImage* image )
   // Configure filter
   auto filter = this->_CreateITK< _TFilter >( );
   filter->SetInput( image );
-  filter->SetNeighborhoodFunction( neig );
-  filter->SetCostFunction( cost );
-  filter->SetCostConversionFunction( conv );
+  if( neig != NULL )
+    filter->SetNeighborhoodFunction( neig );
+  if( cost != NULL )
+    filter->SetCostFunction( cost );
+  if( conv != NULL )
+    filter->SetCostConversionFunction( conv );
   filter->SetStopAtOneFront( this->m_Parameters.GetBool( "StopAtOneFront" ) );
 
   // Assign seeds
@@ -84,7 +87,7 @@ _GD1( _TInputImage* image )
     typename _TInputImage::IndexType idx;
     unsigned int dim =
       ( _TInputImage::ImageDimension < 3 )? _TInputImage::ImageDimension: 3;
-    for( unsigned int i = 0; i < seeds->GetNumberOfPoints( ); ++i )
+    for( int i = 0; i < seeds->GetNumberOfPoints( ); ++i )
     {
       double buf[ 3 ];
       seeds->GetPoint( i, buf );