2 #include <itkCommand.h>
4 #include <itkImageFileReader.h>
5 #include <itkImageFileWriter.h>
6 #include <itkBinaryThresholdImageFilter.h>
8 #include <fpa/Image/MoriRegionGrow.h>
10 // -------------------------------------------------------------------------
11 static const unsigned int VDim = 3;
13 typedef itk::Image< TPixel, VDim > TImage;
14 typedef itk::ImageFileReader< TImage > TReader;
15 typedef itk::ImageFileWriter< TImage > TWriter;
16 typedef fpa::Image::MoriRegionGrow< TImage, TImage > TFilter;
17 typedef itk::BinaryThresholdImageFilter< TImage, TImage > TThreshold;
19 // -------------------------------------------------------------------------
20 class ShowProgressObject
23 ShowProgressObject( itk::ProcessObject* o )
30 << "\rProgress " << std::fixed << std::setprecision( 2 )
31 << ( this->m_Process->GetProgress( ) * 100 )
32 << " %" << std::flush;
34 itk::ProcessObject::Pointer m_Process;
37 // -------------------------------------------------------------------------
38 int main( int argc, char* argv[] )
44 << "Usage: " << argv[ 0 ]
45 << " input_image output_image auxiliary_image lower upper delta";
46 for( unsigned int i = 0; i < VDim; ++i )
47 std::cerr << " s_" << i;
48 std::cerr << std::endl;
52 std::string input_image_filename = argv[ 1 ];
53 std::string output_image_filename = argv[ 2 ];
54 std::string auxiliary_image_filename = argv[ 3 ];
55 TPixel lower = std::atof( argv[ 4 ] );
56 TPixel upper = std::atof( argv[ 5 ] );
57 TPixel delta = std::atof( argv[ 6 ] );
59 TReader::Pointer reader = TReader::New( );
60 reader->SetFileName( input_image_filename );
62 TFilter::Pointer filter = TFilter::New( );
63 filter->SetInput( reader->GetOutput( ) );
64 filter->SetThresholdRange( lower, upper, delta );
65 TImage::IndexType seed;
66 for( int i = 0; i < VDim; ++i )
67 seed[ i ] = std::atoi( argv[ i + 7 ] );
68 filter->AddSeed( seed );
70 // to test ProgressReporter
71 ShowProgressObject progressWatch( filter );
72 typedef itk::SimpleMemberCommand< ShowProgressObject > CommandType;
73 CommandType::Pointer command = CommandType::New();
74 command->SetCallbackFunction( &progressWatch,
75 &ShowProgressObject::ShowProgress );
76 filter->AddObserver( itk::ProgressEvent( ), command );
79 TThreshold::Pointer threshold = TThreshold::New( );
80 threshold->SetInput( filter->GetOutput( ) );
81 threshold->SetInsideValue( 255 );
82 threshold->SetOutsideValue( 0 );
83 threshold->SetLowerThreshold( 0 );
84 threshold->SetUpperThreshold( filter->GetOptimumThreshold( ) );
86 TWriter::Pointer writer = TWriter::New( );
87 writer->SetInput( threshold->GetOutput( ) );
88 writer->SetFileName( output_image_filename );
90 TWriter::Pointer aux_writer = TWriter::New( );
91 aux_writer->SetInput( filter->GetOutput( ) );
92 aux_writer->SetFileName( auxiliary_image_filename );
96 aux_writer->Update( );
98 catch( std::exception& err )
100 std::cerr << "ERROR: " << err.what( ) << std::endl;
106 TFilter::TCurve curve = filter->GetCurve( );
107 for( TFilter::TCurveData data: curve )
109 std::cout << data.XValue << " " << data.YValue << " " << data.Diff1 << std::endl;
114 << curve[ filter->GetOptimumThreshold( ) ].XValue
116 << filter->GetOptimumThreshold( )