#include "ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h"

#include <fpa_Instances/Backtracking.h>
#include <cpPlugins/Image.h>
#include <cpPlugins/ImageIndexesContainer.h>
#include <plugins/fpa/ImageSkeleton.h>

// -------------------------------------------------------------------------
fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
ExtractEndPointsAndBifurcationsFromMinimumSpanningTree( )
  : Superclass( )
{
  this->_AddInput( "CostsImage" );
  this->_AddInput( "DistanceMap" );
  this->_AddInput( "MST" );
  this->_AddOutput< cpPlugins::ImageIndexesContainer >( "EndPoints" );
  this->_AddOutput< cpPlugins::ImageIndexesContainer >( "Bifurcations" );
  this->_AddOutput< cpPlugins::ImageIndexesContainer >( "Collisions" );
  this->_AddOutput< fpaPlugins::ImageSkeleton >( "Skeleton" );

  this->m_Parameters.ConfigureAsBool( "SquaredDistanceMap" );
  this->m_Parameters.SetBool( "SquaredDistanceMap", false );
}

// -------------------------------------------------------------------------
fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
~ExtractEndPointsAndBifurcationsFromMinimumSpanningTree( )
{
}

// -------------------------------------------------------------------------
void fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
_GenerateData( )
{
  auto image = this->GetInputData< itk::DataObject >( "CostsImage" );
  cpPlugins_Image_Demangle_Pixel_AllFloats     ( _GD0, image, 2 );
  else cpPlugins_Image_Demangle_Pixel_AllFloats( _GD0, image, 3 );
  else this->_Error( "Invalid input costs." );
}

// -------------------------------------------------------------------------
template< class _TImage >
void fpaPlugins::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree::
_GD0( _TImage* image )
{
  typedef fpa::Image::MinimumSpanningTree< _TImage::ImageDimension > _TMST;
  typedef fpa::Image::ExtractEndPointsAndBifurcationsFromMinimumSpanningTree< _TImage, _TMST > _TFilter;

  // Check input objects' integrity
  auto dmap = this->GetInputData< _TImage >( "DistanceMap" );
  if( dmap == NULL )
    this->_Error( "Distance map does not have the same type as the costs image." );
  auto mst = this->GetInputData< _TMST >( "MST" );
  if( mst == NULL )
    this->_Error( "No valid input tree." );

  // Create filter and connect inputs
  _TFilter* filter = this->_CreateITK< _TFilter >( );
  filter->SetCostsImage( image );
  filter->SetDistanceMap( dmap );
  filter->SetMinimumSpanningTree( mst );
  filter->SetSquaredDistanceMap(
    this->m_Parameters.GetBool( "SquaredDistanceMap" )
    );
  filter->Update( );

  this->GetOutput( "EndPoints" )->SetITK( filter->GetEndPoints( ) );
  this->GetOutput( "Bifurcations" )->SetITK( filter->GetBifurcations( ) );
  this->GetOutput( "Collisions" )->SetITK( filter->GetCollisions( ) );
  this->GetOutput( "Skeleton" )->SetITK( filter->GetSkeleton( ) );
}

// eof - $RCSfile$