7 #include <itkImageFileReader.h>
8 #include <itkImageToVTKImageFilter.h>
10 #include <vtkImageActor.h>
11 #include <vtkProperty.h>
12 #include <vtkRenderer.h>
13 #include <vtkRenderWindow.h>
14 #include <vtkRenderWindowInteractor.h>
15 #include <vtkSmartPointer.h>
16 #include <vtkSphereSource.h>
18 #include <fpa/Image/RegionGrowWithMultipleThresholds.h>
19 #include <fpa/VTK/ImageMPR.h>
20 #include <fpa/VTK/Image3DObserver.h>
22 // -------------------------------------------------------------------------
23 const unsigned int Dim = 3;
25 typedef double TScalar;
26 typedef itk::Image< TPixel, Dim > TImage;
27 typedef itk::ImageToVTKImageFilter< TImage > TVTKImage;
29 typedef itk::ImageFileReader< TImage > TImageReader;
32 fpa::Image::RegionGrowWithMultipleThresholds< TImage >
35 fpa::VTK::Image3DObserver< TFrontAlgorithm, vtkRenderWindow >
38 // -------------------------------------------------------------------------
39 int main( int argc, char* argv[] )
44 << "Usage: " << argv[ 0 ]
45 << " input_image thr_0 thr_1 n_samples" << std::endl;
49 std::string input_image_fn = argv[ 1 ];
50 TPixel thr_0 = TPixel( std::atof( argv[ 2 ] ) );
51 TPixel thr_1 = TPixel( std::atof( argv[ 3 ] ) );
52 unsigned int n_samples = std::atoi( argv[ 4 ] );
55 TImageReader::Pointer input_image_reader = TImageReader::New( );
56 input_image_reader->SetFileName( input_image_fn );
59 input_image_reader->Update( );
61 catch( itk::ExceptionObject& err )
63 std::cerr << "Error caught: " << err << std::endl;
67 TImage::ConstPointer input_image = input_image_reader->GetOutput( );
68 TImage::SpacingType spac = input_image->GetSpacing( );
69 double min_spac = spac[ 0 ];
70 for( unsigned int d = 1; d < Dim; d++ )
71 min_spac = ( spac[ d ] < min_spac )? spac[ d ]: min_spac;
73 TVTKImage::Pointer vtk_image = TVTKImage::New( );
74 vtk_image->SetInput( input_image );
77 fpa::VTK::ImageMPR view;
78 view.SetBackground( 0.3, 0.2, 0.8 );
79 view.SetSize( 800, 800 );
80 view.SetImage( vtk_image->GetOutput( ) );
82 // Wait for a seed to be given
83 while( view.GetNumberOfSeeds( ) == 0 )
88 view.GetSeed( 0, seed );
89 TImage::PointType seed_pnt;
90 seed_pnt[ 0 ] = seed[ 0 ];
91 seed_pnt[ 1 ] = seed[ 1 ];
92 seed_pnt[ 2 ] = seed[ 2 ];
93 TImage::IndexType seed_idx;
94 input_image->TransformPhysicalPointToIndex( seed_pnt, seed_idx );
97 TObserver::Pointer obs = TObserver::New( );
98 obs->SetImage( input_image, view.GetWindow( ) );
100 // Configure algorithm
101 TFrontAlgorithm::Pointer algorithm = TFrontAlgorithm::New( );
102 algorithm->AddThresholds( thr_0, thr_1, n_samples );
103 algorithm->AddSeed( seed_idx, 0 );
104 algorithm->AddObserver( itk::AnyEvent( ), obs );
105 algorithm->ThrowEventsOn( );
106 algorithm->SetInput( input_image );
107 algorithm->SetNeighborhoodOrder( 1 );
108 algorithm->SetDerivativeThreshold( double( 3 ) );
109 algorithm->Update( );
111 // Let some interaction and close program