itkRegionBasedLevelSetFunctionSharedData.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkRegionBasedLevelSetFunctionSharedData.h,v $
  Language:  C++
  Date:      $Date: 2009-05-14 21:46:49 $
  Version:   $Revision: 1.1 $

  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 __itkRegionBasedLevelSetFunctionSharedData_h
#define __itkRegionBasedLevelSetFunctionSharedData_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 TSingleData >
class RegionBasedLevelSetFunctionSharedData : public LightObject
{
public:

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

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

  itkTypeMacro(RegionBasedLevelSetFunctionSharedData, 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;

  typedef std::list< unsigned int >                     ListPixelType;
  typedef Image< ListPixelType, itkGetStaticConstMacro(ImageDimension) >
                                                        ListImageType;
  typedef typename ListImageType::Pointer               ListImagePointer;
  typedef typename ListImageType::ConstPointer          ListImageConstPointer;
  typedef typename ListImageType::RegionType            ListRegionType;
  typedef typename ListImageType::SizeType              ListSizeType;
  typedef typename ListSizeType::SizeValueType          ListSizeValueType;
  typedef typename ListImageType::SpacingType           ListSpacingType;
  typedef typename ListImageType::IndexType             ListIndexType;
  typedef typename ListIndexType::IndexValueType        ListIndexValueType;
  typedef typename ListImageType::PointType             ListPointType;
  typedef ImageRegionIteratorWithIndex< ListImageType > ListIteratorType;

  typedef Vector< float, itkGetStaticConstMacro(ImageDimension) >
                                                        CentroidVectorType;
  typedef itk::Statistics::ListSample< CentroidVectorType >   SampleType;
  typedef itk::Statistics::KdTreeGenerator< SampleType >      TreeGeneratorType;
  typedef typename TreeGeneratorType::Pointer                 TreePointer;
  typedef typename TreeGeneratorType::KdTreeType              TreeType;
  typedef typename TreeType::Pointer                          KdTreePointer;

  typedef TSingleData                                   LevelSetDataType;
  typedef typename LevelSetDataType::Pointer            LevelSetDataPointer;
  typedef std::vector< LevelSetDataPointer >            LevelSetDataPointerVector;
  typedef typename LevelSetDataPointerVector::iterator  LevelSetDataPointerVectorIterator;

  void SetFunctionCount( const unsigned int& n )
    {
    this->m_FunctionCount = n;
    this->m_LevelSetDataPointerVector.resize( n, 0 );

    LevelSetDataPointerVectorIterator it = m_LevelSetDataPointerVector.begin();
    LevelSetDataPointerVectorIterator end = m_LevelSetDataPointerVector.end();
    while( it != end )
      {
      (*it) = LevelSetDataType::New();
      it++;
      }
    }

  void SetNumberOfNeighbors( const unsigned int& n )
    {
    this->m_NumberOfNeighbors = n;
    }

  void CreateHeavisideFunctionOfLevelSetImage( const unsigned int& j, const InputImageType * image )
    {
    m_LevelSetDataPointerVector[j]->CreateHeavisideFunctionOfLevelSetImage( image );
    }

  void SetKdTree( KdTreePointer kdtree )
    {
    this->m_KdTree = kdtree;
    }

  void AllocateListImage( const FeatureImageType * featureImage )
    {
    this->m_NearestNeighborListImage = ListImageType::New();
    this->m_NearestNeighborListImage->CopyInformation( featureImage );
    this->m_NearestNeighborListImage->SetRegions( featureImage->GetLargestPossibleRegion() );
    this->m_NearestNeighborListImage->Allocate();
    }

  virtual void PopulateListImage() = 0;

  LevelSetDataPointerVector         m_LevelSetDataPointerVector;

  unsigned int                      m_FunctionCount;
  unsigned int                      m_NumberOfNeighbors;
  ListImagePointer                  m_NearestNeighborListImage;
  KdTreePointer                     m_KdTree;

protected:
  RegionBasedLevelSetFunctionSharedData() : m_NumberOfNeighbors( 6 ), m_KdTree( 0 ){}
  ~RegionBasedLevelSetFunctionSharedData(){}

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

} //end namespace itk

#endif