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
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#ifndef itkFloodFilledImageFunctionConditionalIterator_h
#define itkFloodFilledImageFunctionConditionalIterator_h
#include "itkFloodFilledImageFunctionConditionalConstIterator.h"
namespace itk
{
/**
* \class FloodFilledImageFunctionConditionalIterator
* \brief Iterates over a flood-filled image function with write access to pixels.
*
* \ingroup ImageIterators
*
* \ingroup ITKCommon
*
* \sphinx
* \sphinxexample{Core/Common/IterateImageStartingAtSeed,Iterate Image Starting At Seed}
* \endsphinx
*/
template <typename TImage, typename TFunction>
class FloodFilledImageFunctionConditionalIterator
: public FloodFilledImageFunctionConditionalConstIterator<TImage, TFunction>
{
public:
/** Standard class type aliases. */
using Self = FloodFilledImageFunctionConditionalIterator;
using Superclass = FloodFilledImageFunctionConditionalConstIterator<TImage, TFunction>;
/** Type of function */
using typename Superclass::FunctionType;
/** Type of vector used to store location info in the spatial function */
using typename Superclass::FunctionInputType;
/** Index type alias support. */
using typename Superclass::IndexType;
/** Index ContainerType. */
using typename Superclass::SeedsContainerType;
/** Size type alias support. */
using typename Superclass::SizeType;
/** Region type alias support */
using typename Superclass::RegionType;
/** Image type alias support. */
using typename Superclass::ImageType;
/** Internal Pixel Type */
using typename Superclass::InternalPixelType;
/** External Pixel Type */
using typename Superclass::PixelType;
/** 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. */
static constexpr unsigned int NDimensions = Superclass::NDimensions;
/** Constructor establishes an iterator to walk a particular image and a
* particular region of that image. This version of the constructor uses
* an explicit seed pixel for the flood fill, the "startIndex" */
FloodFilledImageFunctionConditionalIterator(ImageType * imagePtr, FunctionType * fnPtr, IndexType startIndex)
: Superclass(imagePtr, fnPtr, startIndex)
{}
/** Constructor establishes an iterator to walk a particular image and a
* particular region of that image. This version of the constructor uses
* an explicit list of seed pixels for the flood fill, the "startIndex" */
FloodFilledImageFunctionConditionalIterator(ImageType * imagePtr,
FunctionType * fnPtr,
std::vector<IndexType> & startIndex)
: Superclass(imagePtr, fnPtr, startIndex)
{}
/** Constructor establishes an iterator to walk a particular image and a
* particular region of that image. This version of the constructor
* should be used when the seed pixel is unknown. */
FloodFilledImageFunctionConditionalIterator(ImageType * imagePtr, FunctionType * fnPtr)
: Superclass(imagePtr, fnPtr)
{}
/** Get the pixel value */
const PixelType
Get() const override
{
return const_cast<ImageType *>(this->m_Image.GetPointer())->GetPixel(this->m_IndexStack.front());
}
/** Set the pixel value */
void
Set(const PixelType & value)
{
const_cast<ImageType *>(this->m_Image.GetPointer())->GetPixel(this->m_IndexStack.front()) = value;
}
/** Default Destructor. */
~FloodFilledImageFunctionConditionalIterator() override = default;
};
} // end namespace itk
#endif
|