--- /dev/null
+#include <cstdlib>
+#include <iostream>
+#include <fstream>
+#include <string>
+#include <vector>
+
+#include <itkImage.h>
+#include <itkImageToVTKImageFilter.h>
+
+#include <itkImageFileReader.h>
+#include <itkImageFileWriter.h>
+
+#include <fpa/VTK/ImageMPR.h>
+#include <fpa/Image/IncrementalRegionGrow.h>
+#include <fpa/Image/Functors/RegionGrowThresholdFunction.h>
+
+/*
+ #include <vtkCamera.h>
+ #include <vtkImageActor.h>
+ #include <vtkInteractorStyleImage.h>
+ #include <vtkPointHandleRepresentation3D.h>
+ #include <vtkProperty.h>
+ #include <vtkRenderer.h>
+ #include <vtkRenderWindow.h>
+ #include <vtkRenderWindowInteractor.h>
+ #include <vtkSeedRepresentation.h>
+ #include <vtkSeedWidget.h>
+ #include <vtkSmartPointer.h>
+
+ #include <fpa/Image/Functors/ImageAbsoluteDifferenceCostFunction.h>
+ #include <fpa/VTK/Image3DObserver.h>
+*/
+
+// -------------------------------------------------------------------------
+const unsigned int Dim = 3;
+typedef short TInputPixel;
+typedef short TOutputPixel;
+
+typedef itk::Image< TInputPixel, Dim > TInputImage;
+typedef itk::Image< TOutputPixel, Dim > TOutputImage;
+typedef itk::ImageToVTKImageFilter< TInputImage > TVTKInputImage;
+
+// -------------------------------------------------------------------------
+int main( int argc, char* argv[] )
+{
+ if( argc < 8 )
+ {
+ std::cerr
+ << "Usage: " << argv[ 0 ] << std::endl
+ << "\tinput_image" << std::endl
+ << "\toutput_image" << std::endl
+ << "\tneighborhood_order" << std::endl
+ << "\tstop_at_one_front" << std::endl
+ << "\tinit_threshold" << std::endl
+ << "\tend_threshold" << std::endl
+ << "\tsamples" << std::endl
+ << "\t[input_seeds]" << std::endl;
+ return( 1 );
+
+ } // fi
+ std::string input_image_fn = argv[ 1 ];
+ std::string output_image_fn = argv[ 2 ];
+ unsigned int neighborhood_order = std::atoi( argv[ 3 ] );
+ bool stop_at_one_front = ( std::atoi( argv[ 4 ] ) != 0 );
+ TInputPixel init_threshold = TInputPixel( std::atof( argv[ 5 ] ) );
+ TInputPixel end_threshold = TInputPixel( std::atof( argv[ 6 ] ) );
+ unsigned int samples = TInputPixel( std::atof( argv[ 7 ] ) );
+ std::string input_seeds_fn = ( argc >= 9 )? argv[ 8 ]: "";
+
+ // Read image
+ itk::ImageFileReader< TInputImage >::Pointer input_image_reader =
+ itk::ImageFileReader< TInputImage >::New( );
+ input_image_reader->SetFileName( input_image_fn );
+ try
+ {
+ input_image_reader->Update( );
+ }
+ catch( itk::ExceptionObject& err )
+ {
+ std::cerr
+ << "Error while reading image from " << input_image_fn << ": "
+ << err << std::endl;
+ return( 1 );
+
+ } // yrt
+ TInputImage::ConstPointer input_image = input_image_reader->GetOutput( );
+ TVTKInputImage::Pointer vtk_input_image = TVTKInputImage::New( );
+ vtk_input_image->SetInput( input_image );
+ vtk_input_image->Update( );
+
+ // Show input image and let some interaction
+ fpa::VTK::ImageMPR view;
+ view.SetBackground( 0.3, 0.2, 0.8 );
+ view.SetSize( 600, 600 );
+ view.SetImage( vtk_input_image->GetOutput( ) );
+ view.Render( );
+
+ // Read seeds
+ std::vector< TInputImage::IndexType > input_seeds;
+ if( input_seeds_fn != "" )
+ {
+ std::ifstream input_seeds_str( input_seeds_fn.c_str( ) );
+ if( input_seeds_str )
+ {
+ unsigned int nSeeds;
+ input_seeds_str >> nSeeds;
+ for( unsigned int i = 0; i < nSeeds; ++i )
+ {
+ TInputImage::PointType pnt;
+ input_seeds_str >> pnt[ 0 ] >> pnt[ 1 ] >> pnt[ 2 ];
+
+ TInputImage::IndexType idx;
+ input_image_reader->GetOutput( )->
+ TransformPhysicalPointToIndex( pnt, idx );
+ input_seeds.push_back( idx );
+
+ view.AddSeed( pnt[ 0 ], pnt[ 1 ], pnt[ 2 ] );
+
+ } // rof
+ input_seeds_str.close( );
+ }
+ else
+ std::cerr << "Error reading \"" << input_seeds_fn << "\"" << std::endl;
+
+ } // fi
+
+ // Allow some interaction
+ view.Render( );
+ if( input_seeds.size( ) == 0 )
+ {
+ while( view.GetNumberOfSeeds( ) == 0 )
+ view.Start( );
+
+ for( unsigned int i = 0; i < view.GetNumberOfSeeds( ); ++i )
+ {
+ double p[ 3 ];
+ view.GetSeed( i, p );
+
+ TInputImage::PointType pnt;
+ pnt[ 0 ] = p[ 0 ];
+ pnt[ 1 ] = p[ 1 ];
+ pnt[ 2 ] = p[ 2 ];
+
+ TInputImage::IndexType idx;
+ input_image_reader->GetOutput( )->
+ TransformPhysicalPointToIndex( pnt, idx );
+ input_seeds.push_back( idx );
+
+ } // rof
+ }
+ else
+ view.Start( );
+
+ // Prepare region grow filter and cost function
+ typedef fpa::Image::
+ IncrementalRegionGrow< TInputImage, TOutputImage > TFilter;
+ typedef fpa::Image::Functors::
+ RegionGrowThresholdFunction< TInputImage > TFunction;
+
+ TFilter::Pointer filter = TFilter::New( );
+ filter->SetInput( input_image_reader->GetOutput( ) );
+
+ /*
+ typedef fpa::Image::Dijkstra< TInputImage, TOutputImage > TFilter;
+ typedef fpa::Image::Functors::
+ ImageAbsoluteDifferenceCostFunction< TFilter::TInputImage, TFilter::TResult >
+ TCost;
+
+ TCost::Pointer cost = TCost::New( );
+
+ TFilter::Pointer filter = TFilter::New( );
+ filter->SetInput( input_image );
+ filter->SetCostFunction( cost );
+ filter->SetConversionFunction( NULL );
+ filter->SetNeighborhoodOrder( neighborhood_order );
+ filter->SetStopAtOneFront( stop_at_one_front );
+
+ // Get user-given seeds
+ std::ifstream input_seeds_str( input_seeds_fn.c_str( ) );
+ if( !input_seeds_str )
+ {
+ std::cerr << "Error opening \"" << input_seeds_fn << "\"" << std::endl;
+ return( 1 );
+
+ } // fi
+
+ unsigned int nSeeds;
+ input_seeds_str >> nSeeds;
+ std::vector< TInputImage::IndexType > input_seeds;
+ for( unsigned int s = 0; s < nSeeds; s++ )
+ {
+ TInputImage::IndexType idx;
+ input_seeds_str >> idx[ 0 ] >> idx[ 1 ] >> idx[ 2 ];
+ input_seeds.push_back( idx );
+
+ } // rof
+ input_seeds_str.close( );
+
+ filter->AddSeed( input_seeds[ init_seed ], 0 );
+ filter->AddSeed( input_seeds[ end_seed ], 0 );
+
+ // Prepare graphical debugger
+ typedef fpa::VTK::Image3DObserver< TFilter, vtkRenderWindow > TDebugger;
+ TDebugger::Pointer debugger = TDebugger::New( );
+ debugger->SetRenderWindow( view.GetWindow( ) );
+ debugger->SetRenderPercentage( 0.0001 );
+ filter->AddObserver( itk::AnyEvent( ), debugger );
+ filter->ThrowEventsOn( );
+
+ // Go!
+ filter->Update( );
+
+ // Save final total cost map
+ itk::ImageFileWriter< TOutputImage >::Pointer output_image_writer =
+ itk::ImageFileWriter< TOutputImage >::New( );
+ output_image_writer->SetFileName( output_image_fn );
+ output_image_writer->SetInput( filter->GetOutput( ) );
+ try
+ {
+ output_image_writer->Update( );
+ }
+ catch( itk::ExceptionObject& err )
+ {
+ std::cerr
+ << "Error while writing image to " << output_image_fn << ": "
+ << err << std::endl;
+ return( 1 );
+
+ } // yrt
+ */
+
+ return( 0 );
+}
+
+// eof - $RCSfile$
git://git.creatis.insa-lyon.fr / FrontAlgorithms.git / blobdiff