#include "itkBinaryWellComposed2DImageFilter.h"

#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"

#include "itkRandomImageSource.h"
#include "itkConstantPadImageFilter.h"

int main( int argc, char * argv[])
{
  const unsigned int      ImageDimension = 2;
  typedef int             PixelType;
  typedef itk::Image<PixelType,  ImageDimension> ImageType;

/*
 itk::RandomImageSource<ImageType>::Pointer random 
     = itk::RandomImageSource<ImageType>::New();
  float spacing[ImageDimension];
  unsigned long size[ImageDimension];
  float origin[ImageDimension];
  for ( unsigned int i = 0; i < ImageDimension; i++ )
    {
    spacing[i] = 1;
    size[i] = 17;
    origin[i] = 0;
    }
  random->SetMax( 2 );
  random->SetMin( 0 );
  random->SetOrigin( origin );
  random->SetSpacing( spacing );
  random->SetSize( size );  
  random->Update();

  typedef itk::ConstantPadImageFilter<ImageType, ImageType> PadderType;
  PadderType::Pointer padder = PadderType::New();
  padder->SetInput( random->GetOutput() );
  padder->SetConstant( 0 );
  unsigned long lowerfactors[ImageDimension];
  unsigned long upperfactors[ImageDimension];
  for ( unsigned int i = 0; i < ImageDimension; i++ )
    {
    lowerfactors[i] = 5;
    upperfactors[i] = 5;
    }
  padder->SetPadLowerBound( lowerfactors );
  padder->SetPadUpperBound( upperfactors );
  padder->UpdateLargestPossibleRegion();

  typedef itk::BinaryWellComposed2DImageFilter<ImageType> FilterType;
  FilterType::Pointer filter = FilterType::New();
  filter->SetInput(padder->GetOutput());
  filter->SetBackgroundValue( static_cast<PixelType>( 0 ) );
  filter->SetForegroundValue( static_cast<PixelType>( 1 ) );
  filter->Update();
*/

  typedef itk::ImageFileReader<ImageType> ReaderType;
  ReaderType::Pointer reader = ReaderType::New();
  reader->SetFileName( "random2d.hdr" );
  reader->Update();

  typedef itk::BinaryWellComposed2DImageFilter<ImageType> FilterType;
  FilterType::Pointer filter = FilterType::New();
  filter->SetInput( reader->GetOutput() );
  filter->DebugOn();
  filter->SetBackgroundValue( static_cast<PixelType>( 0 ) );
  filter->SetForegroundValue( static_cast<PixelType>( 1 ) );
  filter->SetFullInvariance( true );

  typedef itk::ImageFileWriter<ImageType> WriterType;
  WriterType::Pointer writer = WriterType::New();
  writer->SetInput( filter->GetOutput() );
  writer->SetFileName( "random2d_out.hdr" );
  writer->Update();
  
  return 0;
};