1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
|
#ifndef RLEImageIterator_h
#define RLEImageIterator_h
#include "RLEImageConstIterator.h"
#include "itkImageIteratorWithIndex.h"
namespace itk
{
/**
* \class ImageIterator
* \brief A multi-dimensional iterator templated over image type.
*
* This is a base class of ImageConstIterator that adds write-access
* functionality. Please see ImageConstIterator for more information.
*
*/
template< typename TPixel, unsigned int VImageDimension, typename CounterType >
class ImageIterator<RLEImage<TPixel, VImageDimension, CounterType> >
:public ImageConstIterator<RLEImage<TPixel, VImageDimension, CounterType> >
{
public:
/** Standard class typedefs. */
typedef ImageIterator Self;
/** Dimension of the image the iterator walks. This constant is needed so
* functions that are templated over image iterator type (as opposed to
* being templated over pixel type and dimension) can have compile time
* access to the dimension of the image that the iterator walks. */
itkStaticConstMacro(ImageIteratorDimension, unsigned int, VImageDimension);
/** Define the superclass */
typedef ImageConstIterator< RLEImage<TPixel, VImageDimension, CounterType> > Superclass;
/** Inherit types from the superclass */
typedef typename Superclass::IndexType IndexType;
typedef typename Superclass::SizeType SizeType;
typedef typename Superclass::OffsetType OffsetType;
typedef typename Superclass::RegionType RegionType;
typedef typename Superclass::ImageType ImageType;
typedef typename Superclass::InternalPixelType InternalPixelType;
typedef typename Superclass::PixelType PixelType;
/** Default Constructor. Need to provide a default constructor since we
* provide a copy constructor. */
ImageIterator(){}
/** Default Destructor */
~ImageIterator() {}
/** Copy Constructor. The copy constructor is provided to make sure the
* handle to the image is properly reference counted. */
ImageIterator(const Self & it) :
ImageConstIterator<ImageType>(it) {}
/** Constructor establishes an iterator to walk a particular image and a
* particular region of that image. */
ImageIterator(ImageType *ptr, const RegionType & region):
ImageConstIterator<ImageType>(ptr, region){}
/** operator= is provided to make sure the handle to the image is properly
* reference counted. */
Self & operator=(const Self & it)
{
this->ImageConstIterator<ImageType>::operator=(it);
return *this;
}
/** Set the pixel value.
* Changing the RLE structure invalidates all other iterators (except this one). */
void Set(const PixelType & value) const
{
const_cast<ImageType *>(this->m_Image.GetPointer())->
SetPixel(*const_cast<typename ImageType::RLLine *>(this->rlLine),
this->segmentRemainder, this->realIndex, value);
}
///** Return a reference to the pixel
// * Setting this value would change value of the whole run-length segment.
// * If we wanted to safely enable it,
// * we would isolate this pixel into its own segment. */
//PixelType & Value(void)
//{
// return myBuffer[m_Index[2]][m_Index[1]][realIndex].second;
//}
/** Get the image that this iterator walks. */
ImageType * GetImage() const
{
// const_cast is needed here because m_Image is declared as a const pointer
// in the base class which is the ConstIterator.
return const_cast< ImageType * >( this->m_Image.GetPointer() );
}
protected:
/** This constructor is declared protected in order to enforce
const-correctness */
ImageIterator(const ImageConstIterator< ImageType > & it) :
ImageConstIterator<ImageType>(it) {}
Self & operator=(const ImageConstIterator< ImageType > & it)
{
this->ImageConstIterator< ImageType>::operator=(it);
return *this;
}
};
template< typename TPixel, unsigned int VImageDimension, typename CounterType >
class ImageIteratorWithIndex<RLEImage<TPixel, VImageDimension, CounterType> >
:public ImageConstIteratorWithIndex<RLEImage<TPixel, VImageDimension, CounterType> >
{
public:
typedef RLEImage<TPixel, VImageDimension, CounterType> ImageType;
typedef typename itk::ImageConstIterator<RLEImage<TPixel, VImageDimension, CounterType> >::RegionType RegionType;
/** Default Constructor. Need to provide a default constructor since we
* provide a copy constructor. */
ImageIteratorWithIndex() :ImageConstIteratorWithIndex< ImageType >(){ }
/** Copy Constructor. The copy constructor is provided to make sure the
* handle to the image is properly reference counted. */
ImageIteratorWithIndex(const ImageIteratorWithIndex & it)
{
this->ImageIterator< ImageType >::operator=(it);
}
/** Constructor establishes an iterator to walk a particular image and a
* particular region of that image. */
ImageIteratorWithIndex(const ImageType *ptr, const RegionType & region)
:ImageConstIteratorWithIndex< ImageType >(ptr, region) { }
/** Set the pixel value.
* Changing the RLE structure invalidates all other iterators (except this one). */
void Set(const TPixel & value) const
{
const_cast<ImageType *>(this->m_Image.GetPointer())->
SetPixel(*const_cast<typename ImageType::RLLine *>(this->rlLine),
this->segmentRemainder, this->realIndex, value);
}
/** Get the image that this iterator walks. */
ImageType * GetImage() const
{
// const_cast is needed here because m_Image is declared as a const pointer
// in the base class which is the ConstIterator.
return const_cast< ImageType * >(this->m_Image.GetPointer());
}
}; //no additional implementation required
} // end namespace itk
#endif //RLEImageIterator_h
|