+#include <chrono>
+#include <iomanip>
#include <itkCommand.h>
#include <itkImage.h>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>
+#include <itkBinaryThresholdImageFilter.h>
#include <fpa/Image/MoriRegionGrow.h>
// -------------------------------------------------------------------------
static const unsigned int VDim = 3;
-typedef short TPixel;
-typedef itk::Image< TPixel, VDim > TImage;
-typedef itk::ImageFileReader< TImage > TReader;
-typedef itk::ImageFileWriter< TImage > TWriter;
-typedef fpa::Image::MoriRegionGrow< TImage, TImage > TFilter;
-typedef itk::ImageFileWriter< TFilter::TAuxiliaryImage > TAuxWriter;
+typedef short TPixel;
+typedef itk::Image< TPixel, VDim > TImage;
+typedef itk::ImageFileReader< TImage > TReader;
+typedef itk::ImageFileWriter< TImage > TWriter;
+typedef fpa::Image::MoriRegionGrow< TImage, TImage > TFilter;
+typedef itk::BinaryThresholdImageFilter< TImage, TImage > TThreshold;
+
+// -------------------------------------------------------------------------
+class ShowProgressObject
+{
+public:
+ ShowProgressObject( itk::ProcessObject* o )
+ {
+ this->m_Process = o;
+ }
+ void ShowProgress( )
+ {
+ std::cerr
+ << "\rProgress " << std::fixed << std::setprecision( 2 )
+ << ( this->m_Process->GetProgress( ) * 100 )
+ << " %" << std::flush;
+ }
+ itk::ProcessObject::Pointer m_Process;
+};
// -------------------------------------------------------------------------
int main( int argc, char* argv[] )
{
// Get arguments
- if( argc < 7 + VDim )
+ if( argc < 6 + VDim )
{
std::cerr
<< "Usage: " << argv[ 0 ]
- << " input_image output_image auxiliary_image lower upper delta";
+ << " input_image output_image lower upper delta";
for( unsigned int i = 0; i < VDim; ++i )
std::cerr << " s_" << i;
std::cerr << std::endl;
} // fi
std::string input_image_filename = argv[ 1 ];
std::string output_image_filename = argv[ 2 ];
- std::string auxiliary_image_filename = argv[ 3 ];
- TPixel lower = std::atof( argv[ 4 ] );
- TPixel upper = std::atof( argv[ 5 ] );
- TPixel delta = std::atof( argv[ 6 ] );
+ TPixel lower = std::atof( argv[ 3 ] );
+ TPixel upper = std::atof( argv[ 4 ] );
+ TPixel delta = std::atof( argv[ 5 ] );
TReader::Pointer reader = TReader::New( );
reader->SetFileName( input_image_filename );
+ try
+ {
+ reader->Update( );
+ }
+ catch( std::exception& err )
+ {
+ std::cerr << "ERROR: " << err.what( ) << std::endl;
+ return( 1 );
+
+ } // yrt
TFilter::Pointer filter = TFilter::New( );
filter->SetInput( reader->GetOutput( ) );
filter->SetThresholdRange( lower, upper, delta );
- TImage::IndexType seed;
+ TImage::PointType pnt;
for( int i = 0; i < VDim; ++i )
- seed[ i ] = std::atoi( argv[ i + 7 ] );
- filter->SetSeed( seed );
- filter->SetInsideValue( 255 );
- filter->SetOutsideValue( 0 );
+ pnt[ i ] = std::atof( argv[ i + 6 ] );
+ TImage::IndexType seed;
+ if( !( reader->GetOutput( )->TransformPhysicalPointToIndex( pnt, seed ) ) )
+ {
+ std::cerr << "ERROR: seed outside image." << std::endl;
+ return( 1 );
+
+ } // fi
+ filter->AddSeed( seed );
+
+ // to test ProgressReporter
+ /* TODO
+ ShowProgressObject progressWatch( filter );
+ typedef itk::SimpleMemberCommand< ShowProgressObject > CommandType;
+ CommandType::Pointer command = CommandType::New();
+ command->SetCallbackFunction( &progressWatch,
+ &ShowProgressObject::ShowProgress );
+ filter->AddObserver( itk::ProgressEvent( ), command );
+ */
+ std::chrono::time_point< std::chrono::system_clock > start, end;
+ start = std::chrono::system_clock::now( );
+ filter->Update( );
+ end = std::chrono::system_clock::now( );
+ std::chrono::duration< double > elapsed_seconds = end - start;
+
+ TThreshold::Pointer threshold = TThreshold::New( );
+ threshold->SetInput( filter->GetOutput( ) );
+ threshold->SetInsideValue( 255 );
+ threshold->SetOutsideValue( 0 );
+ threshold->SetLowerThreshold( 0 );
+ threshold->SetUpperThreshold( filter->GetOptimumThreshold( ) );
+
TWriter::Pointer writer = TWriter::New( );
- writer->SetInput( filter->GetOutput( ) );
+ writer->SetInput( threshold->GetOutput( ) );
writer->SetFileName( output_image_filename );
try
{
} // yrt
- TAuxWriter::Pointer aux_writer = TAuxWriter::New( );
- aux_writer->SetInput( filter->GetAuxiliaryImage( ) );
- aux_writer->SetFileName( auxiliary_image_filename );
- try
+ // Show data
+ TFilter::TCurve curve = filter->GetCurve( );
+ TFilter::TCurve::const_iterator data = curve.begin( );
+ for( ; data != curve.end( ); ++data )
{
- aux_writer->Update( );
+ std::cout << data->XValue << " " << data->YValue << " " << data->Diff1 << std::endl;
}
- catch( std::exception& err )
- {
- std::cerr << "ERROR: " << err.what( ) << std::endl;
- return( 1 );
-
- } // yrt
+ std::cout
+ << std::endl
+ << "# Opt: "
+ << curve[ filter->GetOptimumThreshold( ) ].XValue
+ << "("
+ << filter->GetOptimumThreshold( )
+ << ")"
+ << std::endl;
+ std::cout << "Time: " << elapsed_seconds.count( ) << "s" << std::endl;
return( 0 );
}