+++ /dev/null
-#include <iostream>
-#include <limits>
-#include <string>
-
-#include <itkImage.h>
-#include <itkImageFileReader.h>
-#include <itkRGBAPixel.h>
-#include <itkSignedDanielssonDistanceMapImageFilter.h>
-#include <itkImageToVTKImageFilter.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/Dijkstra.h>
-#include <fpa/Base/Functors/InvertCostFunction.h>
-#include <fpa/Base/TreeExtractor.h>
-#include <fpa/VTK/Image2DObserver.h>
-
-// -------------------------------------------------------------------------
-const unsigned int Dim = 2;
-typedef unsigned char TPixel;
-typedef itk::RGBAPixel< TPixel > TRGBAPixel;
-typedef double TScalar;
-typedef itk::Image< TPixel, Dim > TImage;
-typedef itk::Image< TScalar, Dim > TDistanceMap;
-typedef itk::Image< TRGBAPixel, Dim > TRGBAImage;
-typedef itk::ImageToVTKImageFilter< TImage > TVTKImage;
-typedef itk::ImageToVTKImageFilter< TRGBAImage > TVTKRGBAImage;
-
-typedef itk::ImageFileReader< TImage > TImageReader;
-typedef
-itk::SignedDanielssonDistanceMapImageFilter< TImage, TDistanceMap >
-TDistanceFilter;
-
-typedef fpa::Image::Dijkstra< TDistanceMap, TScalar > TFrontAlgorithm;
-typedef fpa::Base::TreeExtractor< TFrontAlgorithm > TExtractor;
-typedef fpa::Base::Functors::InvertCostFunction< TScalar > TScalarFunction;
-
-typedef
-fpa::VTK::Image2DObserver< TExtractor, vtkRenderWindow >
-TObserver;
-
-// -------------------------------------------------------------------------
-int main( int argc, char* argv[] )
-{
- if( argc < 2 )
- {
- std::cerr
- << "Usage: " << argv[ 0 ]
- << " input_image [stop_at_one_front]" << std::endl;
- return( 1 );
-
- } // fi
- std::string input_image_fn = argv[ 1 ];
- bool stop_at_one_front = false;
- if( 2 < argc )
- stop_at_one_front = ( std::atoi( argv[ 2 ] ) == 1 );
-
- // Read image
- TImageReader::Pointer input_image_reader = TImageReader::New( );
- input_image_reader->SetFileName( input_image_fn );
- try
- {
- input_image_reader->Update( );
- }
- catch( itk::ExceptionObject& err )
- {
- std::cerr << "Error caught: " << err << std::endl;
- return( 1 );
-
- } // yrt
-
- TDistanceFilter::Pointer dist_filter = TDistanceFilter::New( );
- dist_filter->SetInput( input_image_reader->GetOutput( ) );
- dist_filter->InsideIsPositiveOn( );
- dist_filter->SquaredDistanceOff( );
- dist_filter->Update( );
-
- TImage::ConstPointer input_image = input_image_reader->GetOutput( );
-
- TVTKImage::Pointer vtk_image = TVTKImage::New( );
- vtk_image->SetInput( input_image );
- vtk_image->Update( );
-
- // VTK visualization
- vtkSmartPointer< vtkImageActor > actor =
- vtkSmartPointer< vtkImageActor >::New( );
- actor->SetInputData( vtk_image->GetOutput( ) );
-
- vtkSmartPointer< vtkRenderer > renderer =
- vtkSmartPointer< vtkRenderer >::New( );
- renderer->SetBackground( 0.1, 0.2, 0.7 );
- renderer->AddActor( actor );
- vtkSmartPointer< vtkRenderWindow > window =
- vtkSmartPointer< vtkRenderWindow >::New( );
- window->SetSize( 800, 800 );
- window->AddRenderer( renderer );
-
- // VTK interaction
- vtkSmartPointer< vtkInteractorStyleImage > imageStyle =
- vtkSmartPointer< vtkInteractorStyleImage >::New( );
- vtkSmartPointer< vtkRenderWindowInteractor > interactor =
- vtkSmartPointer< vtkRenderWindowInteractor >::New( );
- interactor->SetInteractorStyle( imageStyle );
- window->SetInteractor( interactor );
- window->Render( );
-
- // Create the widget and its representation
- vtkSmartPointer< vtkPointHandleRepresentation3D > handle =
- vtkSmartPointer< vtkPointHandleRepresentation3D >::New( );
- handle->GetProperty( )->SetColor( 1, 0, 0 );
- vtkSmartPointer< vtkSeedRepresentation > rep =
- vtkSmartPointer< vtkSeedRepresentation >::New( );
- rep->SetHandleRepresentation( handle );
-
- vtkSmartPointer< vtkSeedWidget > widget =
- vtkSmartPointer< vtkSeedWidget >::New( );
- widget->SetInteractor( interactor );
- widget->SetRepresentation( rep );
-
- // Let some interaction
- interactor->Initialize( );
- window->Render( );
- widget->On( );
- interactor->Start( );
-
- // Configure observer
- TObserver::Pointer obs = TObserver::New( );
- obs->SetImage( dist_filter->GetOutput( ), window );
-
- // Configure membership function
- TScalarFunction::Pointer cost_function = TScalarFunction::New( );
-
- // Configure algorithm
- TExtractor::Pointer algorithm = TExtractor::New( );
- for( unsigned int s = 0; s < rep->GetNumberOfSeeds( ); s++ )
- {
- double pos[ 3 ];
- rep->GetSeedWorldPosition( s, pos );
-
- TImage::PointType pnt;
- pnt[ 0 ] = pos[ 0 ];
- pnt[ 1 ] = pos[ 1 ];
-
- TImage::IndexType idx;
- if( input_image->TransformPhysicalPointToIndex( pnt, idx ) )
- {
- algorithm->AddSeed( idx, 0 );
- if( s == 0 )
- algorithm->SetRoot( idx );
- else
- algorithm->AddLeaf( idx );
-
- } // fi
-
- } // rof
- algorithm->AddObserver( itk::AnyEvent( ), obs );
- algorithm->ThrowEventsOn( );
- algorithm->SetInput( dist_filter->GetOutput( ) );
- algorithm->SetNeighborhoodOrder( 1 );
- algorithm->SetStopAtOneFront( stop_at_one_front );
- algorithm->SetCostConversion( cost_function );
- algorithm->Update( );
-
- // Create image with tree
- TPixel transparent_color[] = { 0, 0, 0, 0 };
- TPixel solid_color[] = { 255, 255, 0, 255 };
- TRGBAImage::Pointer tree_image = TRGBAImage::New( );
- tree_image->CopyInformation( input_image );
- tree_image->SetRequestedRegionToLargestPossibleRegion( );
- tree_image->SetBufferedRegion( tree_image->GetRequestedRegion( ) );
- tree_image->Allocate( );
- tree_image->FillBuffer( TRGBAPixel( transparent_color ) );
-
- const TExtractor::TTree& tree = algorithm->GetTree( );
- TExtractor::TTree::const_iterator tIt = tree.begin( );
- for( ; tIt != tree.end( ); ++tIt )
- tree_image->SetPixel( tIt->first, TRGBAPixel( solid_color ) );
-
- TVTKRGBAImage::Pointer vtk_tree_image = TVTKRGBAImage::New( );
- vtk_tree_image->SetInput( tree_image );
- vtk_tree_image->Update( );
-
- vtkSmartPointer< vtkImageActor > tree_actor =
- vtkSmartPointer< vtkImageActor >::New( );
- tree_actor->SetInputData( vtk_tree_image->GetOutput( ) );
-
- renderer->AddActor( tree_actor );
- window->Render( );
-
- // One last interaction
- interactor->Start( );
-
- return( 0 );
-}
-
-// eof - $RCSfile$