itkRegionBasedLevelSetFunctionData.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkRegionBasedLevelSetFunctionData.h,v $
  Language:  C++
  Date:      $Date: 2009-06-08 04:36:32 $
  Version:   $Revision: 1.2 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkRegionBasedLevelSetFunctionData_h
#define __itkRegionBasedLevelSetFunctionData_h

#include "itkLightObject.h"

#include "itkVector.h"
#include "itkListSample.h"
#include "itkKdTree.h"
#include "itkKdTreeGenerator.h"

#include "itkImageRegionIterator.h"
#include "itkImageRegionIteratorWithIndex.h"

namespace itk
{

template < class TInputImage, class TFeatureImage >
class RegionBasedLevelSetFunctionData : public LightObject
{
public:

  typedef RegionBasedLevelSetFunctionData                   Self;
  typedef LightObject                                       Superclass;
  typedef SmartPointer<Self>                                Pointer;
  typedef SmartPointer<const Self>                          ConstPointer;

  itkStaticConstMacro( ImageDimension, unsigned int, TFeatureImage::ImageDimension );

  itkNewMacro( Self );

  itkTypeMacro(RegionBasedLevelSetFunctionData, LightObject);

  typedef TInputImage                                   InputImageType;
  typedef typename InputImageType::Pointer              InputImagePointer;
  typedef typename InputImageType::ConstPointer         InputImageConstPointer;
  typedef typename InputImageType::PixelType            InputPixelType;
  typedef typename InputImageType::RegionType           InputRegionType;
  typedef typename InputImageType::SizeType             InputSizeType;
  typedef typename InputSizeType::SizeValueType         InputSizeValueType;
  typedef typename InputImageType::SpacingType          InputSpacingType;
  typedef typename InputImageType::IndexType            InputIndexType;
  typedef typename InputIndexType::IndexValueType       InputIndexValueType;
  typedef typename InputImageType::PointType            InputPointType;

  typedef TFeatureImage                                 FeatureImageType;
  typedef typename FeatureImageType::Pointer            FeatureImagePointer;
  typedef typename FeatureImageType::ConstPointer       FeatureImageConstPointer;
  typedef typename FeatureImageType::PixelType          FeaturePixelType;
  typedef typename FeatureImageType::RegionType         FeatureRegionType;
  typedef typename FeatureImageType::SizeType           FeatureSizeType;
  typedef typename FeatureSizeType::SizeValueType       FeatureSizeValueType;
  typedef typename FeatureImageType::SpacingType        FeatureSpacingType;
  typedef typename FeatureImageType::IndexType          FeatureIndexType;
  typedef typename FeatureImageType::PointType          FeaturePointType;

  // Allocates m_HeavisideFunctionOfLevelSetImage to have same origin, 
  // spacing and size as image. Also sets the m_Start and m_End indices.
  void CreateHeavisideFunctionOfLevelSetImage( const InputImageType * image );

  // Checks if the given index lies in the domain of the current
  // level-set function. The domain is defined by the start and end indices.
  template< class TIndex >
  bool VerifyInsideRegion( const TIndex& featureIndex )
    {
    for( unsigned int j = 0; j < ImageDimension; j++ )
      {
      if(  (featureIndex[j] < static_cast< InputIndexValueType >(this->m_Start[j]) )
        || (featureIndex[j] > static_cast< InputIndexValueType >(this->m_End[j]))  )
        {
        return false;
        }
      }
    return true;
    }

  // Get the index into the domain of the current level-set function
  InputIndexType GetIndex( const FeatureIndexType& featureIndex );

  // Get the index in the domain of the feature image
  FeatureIndexType GetFeatureIndex( const InputIndexType& inputIndex );


  double              m_WeightedNumberOfPixelsInsideLevelSet;
  double              m_WeightedNumberOfPixelsOutsideLevelSet;

  InputImagePointer   m_HeavisideFunctionOfLevelSetImage;
  InputIndexType      m_Start;
  InputIndexType      m_End;

protected:
  RegionBasedLevelSetFunctionData();
  virtual ~RegionBasedLevelSetFunctionData() {}

private:
  RegionBasedLevelSetFunctionData(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
};

} //end namespace itk

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