#include <itkCommand.h>
#include <itkImage.h>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.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;

// -------------------------------------------------------------------------
int main( int argc, char* argv[] )
{
  // Get arguments
  if( argc < 7 + VDim )
  {
    std::cerr
      << "Usage: " << argv[ 0 ]
      << " input_image output_image auxiliary_image lower upper delta";
    for( unsigned int i = 0; i < VDim; ++i )
      std::cerr << " s_" << i;
    std::cerr << std::endl;
    return( 1 );

  } // 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 ] );

  TReader::Pointer reader = TReader::New( );
  reader->SetFileName( input_image_filename );

  TFilter::Pointer filter = TFilter::New( );
  filter->SetInput( reader->GetOutput( ) );
  filter->SetThresholdRange( lower, upper, delta );
  TImage::IndexType seed;
  for( int i = 0; i < VDim; ++i )
    seed[ i ] = std::atoi( argv[ i + 7 ] );
  filter->SetSeed( seed );
  filter->SetInsideValue( 255 );
  filter->SetOutsideValue( 0 );

  TWriter::Pointer writer = TWriter::New( );
  writer->SetInput( filter->GetOutput( ) );
  writer->SetFileName( output_image_filename );
  try
  {
    writer->Update( );
  }
  catch( std::exception& err )
  {
    std::cerr << "ERROR: " << err.what( ) << std::endl;
    return( 1 );

  } // yrt

  TAuxWriter::Pointer aux_writer = TAuxWriter::New( );
  aux_writer->SetInput( filter->GetAuxiliaryImage( ) );
  aux_writer->SetFileName( auxiliary_image_filename );
  try
  {
    aux_writer->Update( );
  }
  catch( std::exception& err )
  {
    std::cerr << "ERROR: " << err.what( ) << std::endl;
    return( 1 );

  } // yrt
  return( 0 );
}

// eof - $RCSfile$