itkLevelSet.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkLevelSet.h,v $
  Language:  C++
  Date:      $Date: 2003/09/10 14:29:14 $
  Version:   $Revision: 1.5 $

  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 _itkLevelSet_h
#define _itkLevelSet_h

#include "itkIndex.h"
#include "itkImage.h"
#include "itkVectorContainer.h"
#include "itkVector.h"

namespace itk
{

template<class TPixel, unsigned int VSetDimension = 2>
class ITK_EXPORT LevelSetNode
{
public:
  typedef LevelSetNode Self;

  typedef TPixel PixelType;

  itkStaticConstMacro(SetDimension, unsigned int, VSetDimension);

  typedef Index<VSetDimension> IndexType;

  bool operator> ( const Self& node ) const
    { return m_Value > node.m_Value; }

  bool operator< ( const Self& node ) const
    { return m_Value < node.m_Value; }

  bool operator<= ( const Self& node ) const
    { return m_Value <= node.m_Value; }

  bool operator>= ( const Self& node ) const
    { return m_Value >= node.m_Value; }

  Self& operator= ( const Self& rhs )
    {
      if( this == &rhs ) {return *this;}
  
      m_Value = rhs.m_Value;
      m_Index = rhs.m_Index;
      return *this;
    }

  PixelType& GetValue()
    { return m_Value; };
  const PixelType& GetValue() const
    { return m_Value; };
  void SetValue( const PixelType& input )
    { m_Value = input; };

  IndexType& GetIndex()
    { return m_Index; }
  const IndexType& GetIndex() const
    { return m_Index; }
  void SetIndex( const IndexType& input )
    { m_Index = input; };

  LevelSetNode() : m_Value( NumericTraits<PixelType>::Zero ) {
    m_Index.Fill( 0 );
    };

  LevelSetNode(const Self &node) : m_Value( node.m_Value ), m_Index( node.m_Index ) {};
  
private:
  PixelType       m_Value;
  IndexType       m_Index;
  
};

template<class TLevelSet>
class ITK_EXPORT LevelSetTypeDefault
{
 public:
  typedef LevelSetTypeDefault Self;
  typedef TLevelSet LevelSetImageType;
  
  
  itkStaticConstMacro(SetDimension, unsigned int, TLevelSet::ImageDimension);

  typedef typename TLevelSet::Pointer LevelSetPointer;
  typedef typename TLevelSet::ConstPointer LevelSetConstPointer;

  typedef typename TLevelSet::PixelType PixelType;
  
  typedef
      LevelSetNode<PixelType, itkGetStaticConstMacro(SetDimension)> NodeType;

  typedef VectorContainer<unsigned int,NodeType> NodeContainer;

  typedef typename NodeContainer::Pointer NodeContainerPointer;
};


template < 
class TPixel,
unsigned int VAuxDimension = 1,
unsigned int VSetDimension = 2
>
class ITK_EXPORT AuxVarTypeDefault
{
 public:
  typedef AuxVarTypeDefault Self;
  
  typedef TPixel AuxValueType;

  itkStaticConstMacro(AuxDimension, unsigned int, VAuxDimension);

  itkStaticConstMacro(SetDimension, unsigned int, VSetDimension);

  typedef Vector<TPixel,VAuxDimension> AuxValueVectorType;

  typedef VectorContainer<unsigned int,AuxValueVectorType> AuxValueContainer;

  typedef Image<AuxValueType, VSetDimension> AuxImageType;

  typedef typename AuxImageType::Pointer AuxImagePointer;
  typedef typename AuxImageType::ConstPointer AuxImageConstPointer;
};


} // namespace itk

#endif

---