--- /dev/null
+// =========================================================================
+// @author Leonardo Florez Valencia
+// @email florez-l@javeriana.edu.co
+// =========================================================================
+
+#include <itkImage.h>
+#include <itkImageFileReader.h>
+#include <itkImageFileWriter.h>
+#include <fpa/Filters/Image/Dijkstra.h>
+#include <fpa/Functors/Dijkstra/Invert.h>
+
+// -------------------------------------------------------------------------
+const unsigned int Dim = 2;
+typedef unsigned char TInputPixel;
+typedef float TOutputPixel;
+typedef itk::Image< TInputPixel, Dim > TInputImage;
+typedef itk::Image< TOutputPixel, Dim > TOutputImage;
+
+// -------------------------------------------------------------------------
+int main( int argc, char* argv[] )
+{
+ // Get arguments
+ if( argc < 7 )
+ {
+ std::cerr
+ << "Usage: " << argv[ 0 ]
+ << " input_image output_image output_marks output_mst alpha beta [seeds]"
+ << std::endl;
+ return( 1 );
+
+ } // fi
+ std::string input_image_filename = argv[ 1 ];
+ std::string output_image_filename = argv[ 2 ];
+ std::string output_marks_filename = argv[ 3 ];
+ std::string output_mst_filename = argv[ 4 ];
+ double alpha = std::atof( argv[ 5 ] );
+ double beta = std::atof( argv[ 6 ] );
+
+ // Read image
+ typedef itk::ImageFileReader< TInputImage > TInputImageReader;
+ TInputImageReader::Pointer input_image_reader = TInputImageReader::New( );
+ input_image_reader->SetFileName( input_image_filename );
+
+ // Prepare weight functor
+ typedef fpa::Functors::Dijkstra::Invert< TOutputPixel > TWeight;
+ TWeight::Pointer weight = TWeight::New( );
+ weight->SetAlpha( alpha );
+ weight->SetBeta( beta );
+
+ // Prepare filter
+ typedef fpa::Filters::Image::Dijkstra< TInputImage, TOutputImage > TFilter;
+ TFilter::Pointer filter = TFilter::New( );
+ filter->SetInput( input_image_reader->GetOutput( ) );
+ filter->SetWeightFunction( weight );
+
+ // Get all seeds
+ for( int i = 7; i < argc; i += Dim )
+ {
+ TInputImage::IndexType seed;
+ for( int j = 0; j < Dim; ++j )
+ if( i + j < argc )
+ seed[ j ] = std::atoi( argv[ i + j ] );
+ filter->AddSeed( seed );
+
+ } // rof
+
+ // Execute filter
+ filter->Update( );
+
+ // Save results
+ typedef itk::ImageFileWriter< TFilter::TOutputImage > TOutputWriter;
+ TOutputWriter::Pointer output_writer = TOutputWriter::New( );
+ output_writer->SetInput( filter->GetOutput( ) );
+ output_writer->SetFileName( output_image_filename );
+
+ typedef itk::ImageFileWriter< TFilter::TMarksImage > TMarksWriter;
+ TMarksWriter::Pointer marks_writer = TMarksWriter::New( );
+ marks_writer->SetInput( filter->GetMarks( ) );
+ marks_writer->SetFileName( output_marks_filename );
+
+ typedef itk::ImageFileWriter< TFilter::TMST > TMSTWriter;
+ TMSTWriter::Pointer mst_writer = TMSTWriter::New( );
+ mst_writer->SetInput( filter->GetMinimumSpanningTree( ) );
+ mst_writer->SetFileName( output_mst_filename );
+
+ try
+ {
+ output_writer->Update( );
+ marks_writer->Update( );
+ mst_writer->Update( );
+ }
+ catch( std::exception& err )
+ {
+ std::cerr << "Error caught: " << err.what( ) << std::endl;
+ return( 1 );
+
+ } // yrt
+
+ return( 0 );
+}
+
+// eof - $RCSfile$