#include <ctime>
#include <iostream>
#include <string>

#include <itkConstNeighborhoodIterator.h>
#include <itkDanielssonDistanceMapImageFilter.h>
#include <itkImage.h>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>
#include <itkImageToVTKImageFilter.h>

#include <fpa/Image/RegionGrowWithMultipleThresholds.h>
#include <fpa/VTK/ImageMPR.h>
#include <fpa/VTK/Image3DObserver.h>

// -------------------------------------------------------------------------
const unsigned int Dim = 3;
typedef short                                TPixel;
typedef double                               TScalar;
typedef itk::Image< TPixel, Dim >            TImage;
typedef itk::Image< TScalar, Dim >           TDistanceMap;
typedef itk::ImageToVTKImageFilter< TImage > TVTKImage;

typedef itk::ImageFileReader< TImage >                       TImageReader;
typedef itk::ImageFileWriter< TImage >                       TImageWriter;
typedef itk::ImageFileWriter< TDistanceMap >           TDistanceMapWriter;
typedef fpa::Image::RegionGrowWithMultipleThresholds< TImage > TSegmentor;
typedef
itk::DanielssonDistanceMapImageFilter< TImage, TDistanceMap >
TDistanceFilter;

typedef fpa::VTK::ImageMPR TMPR;
typedef fpa::VTK::Image3DObserver< TSegmentor, vtkRenderWindow > TSegmentorObs;

// -------------------------------------------------------------------------
int main( int argc, char* argv[] )
{
  if( argc < 7 )
  {
    std::cerr
      << "Usage: " << argv[ 0 ]
      << " input_image thr_0 thr_1 step"
      << " output_segmentation output_distancemap"
      << " visual_debug"
      << std::endl;
    return( 1 );

  } // fi
  std::string input_image_fn = argv[ 1 ];
  TPixel thr_0 = TPixel( std::atof( argv[ 2 ] ) );
  TPixel thr_1 = TPixel( std::atof( argv[ 3 ] ) );
  unsigned int step = std::atoi( argv[ 4 ] );
  std::string output_segmentation_fn = argv[ 5 ];
  std::string output_distancemap_fn = argv[ 6 ];
  bool visual_debug = false;
  if( argc > 7 )
    visual_debug = ( std::atoi( argv[ 7 ] ) == 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
  TImage::ConstPointer input_image = input_image_reader->GetOutput( );

  // Show image
  TVTKImage::Pointer vtk_image = TVTKImage::New( );
  vtk_image->SetInput( input_image );
  vtk_image->Update( );

  TMPR view;
  view.SetBackground( 0.3, 0.2, 0.8 );
  view.SetSize( 800, 800 );
  view.SetImage( vtk_image->GetOutput( ) );

  // Wait for a seed to be given
  while( view.GetNumberOfSeeds( ) == 0 )
    view.Start( );

  // Get seed from user
  double seed[ 3 ];
  view.GetSeed( 0, seed );
  TImage::PointType seed_pnt;
  seed_pnt[ 0 ] = seed[ 0 ];
  seed_pnt[ 1 ] = seed[ 1 ];
  seed_pnt[ 2 ] = seed[ 2 ];
  TImage::IndexType seed_idx;
  input_image->TransformPhysicalPointToIndex( seed_pnt, seed_idx );

  // Segment input image
  TSegmentor::Pointer segmentor = TSegmentor::New( );
  segmentor->AddThresholds( thr_0, thr_1, step );
  segmentor->AddSeed( seed_idx, 0 );
  segmentor->SetInput( input_image );
  segmentor->SetNeighborhoodOrder( 1 );
  segmentor->SetDifferenceThreshold( double( 3 ) );
  segmentor->SetInsideValue( TPixel( 0 ) );  // WARNING: This is to optimize
  segmentor->SetOutsideValue( TPixel( 1 ) ); // distance map calculation
  if( visual_debug )
  {
    // Configure observer
    TSegmentorObs::Pointer obs = TSegmentorObs::New( );
    obs->SetImage( input_image, view.GetWindow( ) );
    segmentor->AddObserver( itk::AnyEvent( ), obs );
    segmentor->ThrowEventsOn( );
  }
  else
    segmentor->ThrowEventsOff( );
  std::clock_t start = std::clock( );
  segmentor->Update( );
  std::clock_t end = std::clock( );
  double seconds = double( end - start ) / double( CLOCKS_PER_SEC );
  std::cout << "Segmentation time = " << seconds << std::endl;

  // Extract distance map
  TDistanceFilter::Pointer distanceMap = TDistanceFilter::New( );
  distanceMap->SetInput( segmentor->GetOutput( ) );
  distanceMap->InputIsBinaryOn( );
  distanceMap->SquaredDistanceOn( );
  start = std::clock( );
  distanceMap->Update( );
  end = std::clock( );
  seconds = double( end - start ) / double( CLOCKS_PER_SEC );
  std::cout << "Distance map time = " << seconds << std::endl;

  std::cout << "Final seed: " << seed_idx << std::endl;

  TDistanceMapWriter::Pointer distancemap_writer =
    TDistanceMapWriter::New( );
  distancemap_writer->SetInput( distanceMap->GetOutput( ) );
  distancemap_writer->SetFileName( output_distancemap_fn );
  distancemap_writer->Update( );

  TImageWriter::Pointer segmentation_writer =
    TImageWriter::New( );
  segmentation_writer->SetInput( segmentor->GetOutput( ) );
  segmentation_writer->SetFileName( output_segmentation_fn );
  segmentation_writer->Update( );

  // Show result
  /*
    TVTKImage::Pointer output_image_vtk = TVTKImage::New( );
    output_image_vtk->SetInput( segmentor->GetOutput( ) );
    output_image_vtk->Update( );

    vtkSmartPointer< vtkImageMarchingCubes > mc =
    vtkSmartPointer< vtkImageMarchingCubes >::New( );
    mc->SetInputData( output_image_vtk->GetOutput( ) );
    mc->SetValue(
    0,
    double( segmentor->GetInsideValue( ) ) * double( 0.95 )
    );
    mc->Update( );

    // Let some interaction and close program
    view.AddPolyData( mc->GetOutput( ), 0.1, 0.6, 0.8, 0.5 );
    view.Start( );

    // Write resulting image
    TImageWriter::Pointer output_image_writer = TImageWriter::New( );
    output_image_writer->SetInput( segmentor->GetOutput( ) );
    output_image_writer->SetFileName( output_image_fn );
    try
    {
    output_image_writer->Update( );
    }
    catch( itk::ExceptionObject& err )
    {
    std::cerr << "Error caught: " << err << std::endl;
    return( 1 );

    } // yrt
  */
  return( 0 );
}

// eof - $RCSfile$