itkSetGreyBoundaryToWhiteOrCSFFilter.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 itkSetGreyBoundaryToWhiteOrCSFFilter_h
#define itkSetGreyBoundaryToWhiteOrCSFFilter_h

#include <itkImage.h>
#include "itkBaseCTESegmentationFilter.h"

namespace itk
{
template <class TImageType, typename TScalarType, unsigned int NDimensions>
class ITK_EXPORT SetGreyBoundaryToWhiteOrCSFFilter : 
    public BaseCTESegmentationFilter< TImageType >
{
public:
  typedef SetGreyBoundaryToWhiteOrCSFFilter          Self;
  typedef BaseCTESegmentationFilter<TImageType>      Superclass;
  typedef SmartPointer<Self>                         Pointer;
  typedef SmartPointer<const Self>                   ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(SetGreyBoundaryToWhiteOrCSFFilter, BaseCTESegmentationFilter);

  itkStaticConstMacro(Dimension, unsigned int, NDimensions);

  typedef typename Superclass::InputImageType             InputImageType;
  typedef typename Superclass::InputPixelType             InputPixelType;
  typedef typename Superclass::InputIndexType             InputIndexType;
  typedef typename Superclass::InputImagePointer          InputImagePointer;
  typedef typename Superclass::InputImageConstPointer     InputImageConstPointer;
  typedef typename Superclass::OutputPixelType            OutputPixelType;
  typedef typename Superclass::OutputImageType            OutputImageType;
  typedef typename Superclass::OutputImagePointer         OutputImagePointer;
  typedef typename Superclass::OutputImageConstPointer    OutputImageConstPointer;
  typedef TScalarType                                     ThicknessPixelType;
  typedef Image<ThicknessPixelType, NDimensions>          ThicknessImageType;

  void SetLabelImage(const InputImageType *image) {this->SetNthInput(0, const_cast<InputImageType *>(image)); }

  void SetOneLayerImage(const InputImageType *image) {this->SetNthInput(1, const_cast<InputImageType *>(image)); }

  void SetThicknessImage(const ThicknessImageType *image) {this->SetNthInput(2, const_cast<ThicknessImageType *>(image)); }

  itkSetMacro(ExpectedVoxelSize, float);
  itkGetMacro(ExpectedVoxelSize, float);
  
  itkSetMacro(TaggedCSFLabel, InputPixelType);
  itkGetMacro(TaggedCSFLabel, InputPixelType);
  
  itkGetMacro(NumberOfGreyBefore, unsigned long int);
  
  itkGetMacro(NumberOfGreyAfter, unsigned long int);
  
protected:
  
  SetGreyBoundaryToWhiteOrCSFFilter();
  virtual ~SetGreyBoundaryToWhiteOrCSFFilter()  {};

  virtual void PrintSelf(std::ostream&, Indent) const;

  /* The main filter method. Note, single threaded. */
  virtual void GenerateData();
  
private:

  SetGreyBoundaryToWhiteOrCSFFilter(const Self&); // purposely not implemented
  void operator=(const Self&); // purposely not implemented
  
  float m_ExpectedVoxelSize;
  
  InputPixelType m_TaggedCSFLabel;
  
  unsigned long int m_NumberOfGreyBefore;
  
  unsigned long int m_NumberOfGreyAfter;
  
};
  
} // end namespace itk

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

#endif