#ifndef __CTBronchi__MoriLabelling__h__
#define __CTBronchi__MoriLabelling__h__
-#include <itkImageToImageFilter.h>
-#include <itkHessianRecursiveGaussianImageFilter.h>
-#include <itkHessian3DToVesselnessMeasureImageFilter.h>
-
-#include <itkImageFileWriter.h>
-
+#include <fpa/Filters/Image/DefaultTraits.h>
+#include <fpa/Filters/MarksInterface.h>
+#include <fpa/Filters/Image/LabelsSeedFilter.h>
+#include <fpa/Filters/Image/LabelsSeedInterface.h>
+#include <fpa/Filters/Image/RegionGrow.h>
+#include <fpa/Functors/RegionGrow/BinaryThreshold.h>
+
+/* TODO
+ #include <fpa/Base/RegionGrow.h>
+ #include <fpa/Base/MarksInterface.h>
+ #include <fpa/Base/Functors/RegionGrow/BinaryThreshold.h>
+ #include <fpa/Image/Algorithm.h>
+ #include <fpa/Image/DefaultTraits.h>
+ #include <fpa/Image/LabelledSeedsInterface.h>
+*/
namespace CTBronchi
{
/**
*/
template< class _TInputImage, class _TLabelImage >
- class MoriLabelling
- : public itk::ImageToImageFilter< _TLabelImage, _TLabelImage >
+ class MoriLabellingTraits
+ : public fpa::Filters::Image::DefaultTraits< _TInputImage, _TLabelImage, typename _TLabelImage::PixelType >
{
public:
- typedef MoriLabelling Self;
- typedef itk::ImageToImageFilter< _TLabelImage, _TLabelImage > Superclass;
- typedef itk::SmartPointer< Self > Pointer;
- typedef itk::SmartPointer< const Self > ConstPointer;
-
typedef _TInputImage TInputImage;
typedef _TLabelImage TLabelImage;
- typedef typename TInputImage::PixelType TPixel;
- typedef typename TLabelImage::PixelType TLabel;
- typedef typename TLabelImage::RegionType TRegion;
+ typedef fpa::Filters::Image::DefaultTraits< TInputImage, TLabelImage, typename TLabelImage::PixelType > Superclass;
- public:
- itkNewMacro( Self );
- itkTypeMacro( MoriLabelling, itk::ImageToImageFilter );
-
- itkGetConstMacro( UpperThreshold, TPixel );
- itkSetMacro( UpperThreshold, TPixel );
+ typedef typename Superclass::TInternalTraits TInternalTraits;
+ typedef typename Superclass::TFilter TFilter;
+ typedef fpa::Filters::MarksInterface< TInternalTraits > TMarksInterface;
+ typedef fpa::Filters::Image::LabelsSeedInterface< TInternalTraits > TSeedsInterface;
+ };
- itkGetConstMacro( InsideValue, TLabel );
- itkSetMacro( InsideValue, TLabel );
+ /**
+ */
+ template< class _TInputImage, class _TLabelImage >
+ class MoriLabelling
+ : public fpa::Filters::Image::LabelsSeedFilter< fpa::Filters::Image::Algorithm< fpa::Filters::RegionGrow< CTBronchi::MoriLabellingTraits< _TInputImage, _TLabelImage > > >, _TLabelImage >
+ {
+ public:
+ typedef _TInputImage TInputImage;
+ typedef _TLabelImage TLabelImage;
+ typedef CTBronchi::MoriLabellingTraits< TInputImage, TLabelImage > TTraits;
+ fpaTraitsMacro( typename, TTraits );
- itkGetConstMacro( InsideLabel, TLabel );
- itkSetMacro( InsideLabel, TLabel );
+ typedef fpa::Filters::Image::Algorithm< fpa::Filters::RegionGrow< TTraits > > TAlgorithm;
+ typedef fpa::Filters::Image::LabelsSeedFilter< TAlgorithm, _TLabelImage > Superclass;
+ typedef MoriLabelling Self;
+ typedef itk::SmartPointer< Self > Pointer;
+ typedef itk::SmartPointer< const Self > ConstPointer;
- itkGetConstMacro( OutsideLabel, TLabel );
- itkSetMacro( OutsideLabel, TLabel );
+ typedef fpa::Functors::RegionGrow::BinaryThreshold< TInputValue > TLabelFunctor;
public:
- const TLabelImage* GetInputLabelImage( ) const;
- void SetInputLabelImage( TLabelImage* image );
+ itkNewMacro( Self );
+ itkTypeMacro( CTBronchi::MoriLabelling, fpa::Filters::Image::RegionGrow );
- const TInputImage* GetInputRawImage( ) const;
- void SetInputRawImage( TInputImage* image );
+ public:
+ TInputValue GetUpperThreshold( ) const;
+ void SetUpperThreshold( TInputValue t );
protected:
MoriLabelling( );
virtual ~MoriLabelling( );
- virtual void BeforeThreadedGenerateData( ) override
- {
- this->Superclass::BeforeThreadedGenerateData( );
-
- const TInputImage* raw = this->GetInputRawImage( );
- typename TInputImage::SpacingType spac = raw->GetSpacing( );
- double sigma = spac[ 0 ];
- for( unsigned int d = 1; d < TInputImage::ImageDimension; ++d )
- sigma = ( spac[ d ] < sigma )? spac[ d ]: sigma;
- sigma *= 1.5;
-
- typedef itk::HessianRecursiveGaussianImageFilter< TInputImage > _THessian;
- typename _THessian::Pointer hessian = _THessian::New( );
- hessian->SetInput( raw );
- hessian->SetSigma( sigma );
-
- typedef itk::Hessian3DToVesselnessMeasureImageFilter< double > _TVesselness;
- typename _TVesselness::Pointer vesselness = _TVesselness::New( );
- vesselness->SetInput( hessian->GetOutput( ) );
- vesselness->SetAlpha1( 0.5 );
- vesselness->SetAlpha2( 2.0 );
-
- typename itk::ImageFileWriter< typename _TVesselness::OutputImageType >::Pointer w =
- itk::ImageFileWriter< typename _TVesselness::OutputImageType >::New( );
- w->SetInput( vesselness->GetOutput( ) );
- w->SetFileName( "vessel.mhd" );
- w->Update( );
- }
-
- virtual void ThreadedGenerateData( const TRegion& region, itk::ThreadIdType threadId ) override;
+ virtual void _UpdateOutputValue( TNode& n ) override;
private:
// Purposely not implemented.
Self& operator=( const Self& other );
protected:
- TPixel m_UpperThreshold;
- TLabel m_InsideValue;
- TLabel m_InsideLabel;
- TLabel m_OutsideLabel;
+ typename TLabelFunctor::Pointer m_LabelFunctor;
};
} // ecapseman