itkDBCImageFilter.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 itkDBCImageFilter_h
#define itkDBCImageFilter_h

#include <itkImageToImageFilter.h>
#include <vector>

namespace itk 
{
template <class TImageType, class TMaskType>
class ITK_EXPORT DBCImageFilter : 
  public ImageToImageFilter<TImageType, TImageType>
{
public:
  typedef DBCImageFilter Self;
  typedef ImageToImageFilter<TImageType,TImageType> Superclass;
  typedef SmartPointer<Self> Pointer;
  typedef SmartPointer<const Self> ConstPointer;
  itkNewMacro(Self);  
  itkTypeMacro(DBCImageFilter, ImageToImageFilter);
  void AddImage(typename TImageType::ConstPointer inputImage, typename TMaskType::ConstPointer inputMask)
  {
    this->m_InputImages.push_back(inputImage); 
    this->m_InputImageMasks.push_back(inputMask); 
  }
  void ClearImage()
  {
    this->m_InputImages.clear(); 
    this->m_InputImageMasks.clear(); 
  }
  void CalculateBiasFields(); 
  void ApplyBiasFields(); 
  TImageType* GetOutputImage(unsigned int i)
  {
    return this->m_OutputImages[i]; 
  }

protected:
  DBCImageFilter();
  virtual ~DBCImageFilter() {}
  void GenerateData();
  void InputSanityCheck()
  {
    if (this->m_InputImages.size() != this->m_InputImageMasks.size())
    {
      itkExceptionMacro("Errr... you must supply the same number of images and masks."); 
    }
    if (this->m_InputImages.size() < 2)
    {
      itkExceptionMacro("Errr... you must supply at least two images. Hello? We are doing DBC."); 
    }
  }
  std::vector<typename TImageType::ConstPointer> m_InputImages; 
  std::vector<typename TMaskType::ConstPointer> m_InputImageMasks; 
  std::vector<typename TImageType::Pointer> m_BiasFields; 
  std::vector<typename TImageType::Pointer> m_OutputImages; 
  int m_InputRegionExpansion; 
  int m_InputRadius; 
  int m_InputMode; 
  
private:
  DBCImageFilter(const Self&); 
  void operator=(const Self&); 
  
};

}

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




#endif /*ITKDBCImageFilter_H_*/