itkCorrectGMUsingNeighbourhoodFilter.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 itkCorrectGMUsingNeighbourhoodFilter_h
#define itkCorrectGMUsingNeighbourhoodFilter_h

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

namespace itk
{
template <class TImageType>
class ITK_EXPORT CorrectGMUsingNeighbourhoodFilter : 
    public BaseCTESegmentationFilter< TImageType >
{
public:
  typedef CorrectGMUsingNeighbourhoodFilter          Self;
  typedef BaseCTESegmentationFilter<TImageType>      Superclass;
  typedef SmartPointer<Self>                         Pointer;
  typedef SmartPointer<const Self>                   ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(CorrectGMUsingNeighbourhoodFilter, BaseCTESegmentationFilter);
  
  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 OutputImageType::RegionType         OutputImageRegionType;
  typedef typename OutputImageType::IndexType          OutputImageIndexType;
  typedef typename OutputImageType::SizeType           OutputImageSizeType;
  typedef typename Superclass::OutputImagePointer      OutputImagePointer;
  typedef typename Superclass::OutputImageConstPointer OutputImageConstPointer;

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

  const std::vector<InputIndexType>& GetListOfGreyMatterPixelsBeforeCorrection() const;
  
  const std::vector<InputIndexType>& GetListOfGreyMatterPixelsAfterCorrection() const;
  
  itkGetMacro(NumberReclassified, unsigned long int);
  
  itkSetMacro(UseFullNeighbourHood, bool);
  itkGetMacro(UseFullNeighbourHood, bool);
  
protected:
  
  CorrectGMUsingNeighbourhoodFilter();
  virtual ~CorrectGMUsingNeighbourhoodFilter()  {};

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

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

  CorrectGMUsingNeighbourhoodFilter(const Self&); // purposely not implemented
  void operator=(const Self&); // purposely not implemented

  std::vector<InputIndexType> m_ListOfGreyMatterVoxelsBeforeCorrection;
  
  std::vector<InputIndexType> m_ListOfGreyMatterVoxelsAfterCorrection;
  
  bool m_UseFullNeighbourHood;
  
  unsigned long int m_NumberReclassified;
};
  
} // end namespace itk

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

#endif