itkExtendedBrainMaskWithSmoothDropOffCompositeFilter.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/

#ifndef itkExtendedBrainMaskWithSmoothDropOffCompositeFilter_h
#define itkExtendedBrainMaskWithSmoothDropOffCompositeFilter_h


#include <itkImageToImageFilter.h>
#include "itkInjectSourceImageGreaterThanZeroIntoTargetImageFilter.h"
#include <itkBinaryThresholdImageFilter.h>
#include <itkBinaryCrossStructuringElement.h>
#include <itkBinaryDilateImageFilter.h>
#include <itkDiscreteGaussianImageFilter.h>

namespace itk 
{

template< class TImageType>
class ITK_EXPORT ExtendedBrainMaskWithSmoothDropOffCompositeFilter : public ImageToImageFilter<TImageType, TImageType>
{
  
public:

  // Standard ITK style typedefs.
  typedef ExtendedBrainMaskWithSmoothDropOffCompositeFilter Self;
  typedef ImageToImageFilter<TImageType, TImageType>        Superclass;
  typedef SmartPointer<Self>                                Pointer;
  typedef SmartPointer<const Self>                          ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(ExtendedBrainMaskWithSmoothDropOffCompositeFilter, ImageToImageFilter);

  // typedefs for internal filters.
  typedef TImageType                                                                         ImageType;
  typedef typename ImageType::Pointer                                                        ImagePointer;
  typedef BinaryThresholdImageFilter<TImageType, TImageType>                                 BinaryThresholdFilterType;
  typedef typename BinaryThresholdFilterType::Pointer                                        BinaryThresholdFilterPointer;
  typedef BinaryCrossStructuringElement<typename TImageType::PixelType, 
                                        TImageType::ImageDimension>                          StructuringElementType;
  typedef BinaryDilateImageFilter<TImageType, TImageType, StructuringElementType>            BinaryDilateFilterType;
  typedef typename  BinaryDilateFilterType::Pointer                                          BinaryDilateFilterPointer;
  typedef DiscreteGaussianImageFilter<TImageType, TImageType>                                GaussianFilterType;
  typedef typename GaussianFilterType::Pointer                                               GaussianFilterPointer;
  typedef InjectSourceImageGreaterThanZeroIntoTargetImageFilter<TImageType, 
                                                                TImageType, 
                                                                TImageType>                  InjectSourceImageGreaterThanZeroIntoTargetImageFilterType;
  typedef typename InjectSourceImageGreaterThanZeroIntoTargetImageFilterType::Pointer        InjectSourceImageGreaterThanZeroIntoTargetImageFilterPointer;
  
  void PrintSelf( std::ostream& os, itk::Indent indent ) const;

  itkGetMacro( InitialThreshold, float);
  itkSetMacro( InitialThreshold, float);
  itkGetMacro( FirstNumberOfDilations, unsigned int);
  itkSetMacro( FirstNumberOfDilations, unsigned int);
  itkGetMacro( SecondNumberOfDilations, unsigned int);
  itkSetMacro( SecondNumberOfDilations, unsigned int);
  itkGetMacro( GaussianFWHM, float);
  itkSetMacro( GaussianFWHM, float);

protected:

  ExtendedBrainMaskWithSmoothDropOffCompositeFilter();
  virtual ~ExtendedBrainMaskWithSmoothDropOffCompositeFilter();
  void GenerateData();

private:

  ExtendedBrainMaskWithSmoothDropOffCompositeFilter(Self&);   // intentionally not implemented
  void operator=(const Self&);                                // intentionally not implemented

  float        m_InitialThreshold;
  unsigned int m_FirstNumberOfDilations;
  unsigned int m_SecondNumberOfDilations;
  float        m_GaussianFWHM; 
  
  BinaryThresholdFilterPointer m_ThresholdFilter;
  BinaryDilateFilterPointer m_FirstDilateFilter;
  BinaryDilateFilterPointer m_SecondDilateFilter;
  GaussianFilterPointer m_GaussianFilter;
  InjectSourceImageGreaterThanZeroIntoTargetImageFilterPointer m_InjectorFilter;
}; // end class

} // end namespace

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkExtendedBrainMaskWithSmoothDropOffCompositeFilter.txx"
#endif

#endif