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
|
/*=========================================================================
*
* 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 itkFiniteDifferenceSparseImageFunction_h
#define itkFiniteDifferenceSparseImageFunction_h
#include "itkFiniteDifferenceFunction.h"
namespace itk
{
/**
* \class FiniteDifferenceSparseImageFunction
*
* \brief This is the base class for function classes that can be used with
* filters derived from FiniteDifferenceSparseImageFilter.
*
* \par This class is derived from FiniteDifferenceFunction. It is designed to
*work with neighborhoods of pointers instead of actual data. Towards this
*purpose ComputeUpdate method is no longer used and is replaced by the
*ComputeSparseUpdate method. ComputeSparseUpdate assumes that the pointers are
*associated with structures that have the member variable m_Data. The pointers
*would normally originate from a SparseImage object.
*
* \par The PrecomputeSparseUpdate method can be defined to do a first pass of
*computations which then can be used by ComputeSparseUpdate. If
*PrecomputeSparseUpdate is used then the m_PrecomputeFlag should be set in the
*FiniteDifferenceSparseImageFilter class. The precomputation step can be used
*to speed up computation by avoiding repetitions. See the
*NormalVectorDiffusionFunction for an example.
* \ingroup ITKFiniteDifference
*/
template <typename TSparseImageType>
class ITK_TEMPLATE_EXPORT FiniteDifferenceSparseImageFunction : public FiniteDifferenceFunction<TSparseImageType>
{
public:
ITK_DISALLOW_COPY_AND_MOVE(FiniteDifferenceSparseImageFunction);
/** Standard class type alias. */
using Self = FiniteDifferenceSparseImageFunction;
using Superclass = FiniteDifferenceFunction<TSparseImageType>;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(FiniteDifferenceSparseImageFunction);
/** The image dimension. */
static constexpr unsigned int ImageDimension = Superclass::ImageDimension;
/** Typedefs from the superclass. */
using typename Superclass::PixelType;
using typename Superclass::TimeStepType;
using typename Superclass::RadiusType;
using typename Superclass::NeighborhoodType;
using typename Superclass::FloatOffsetType;
using SparseImageType = typename Superclass::ImageType;
/** The index type for the sparse image. */
using IndexType = typename SparseImageType::IndexType;
/** The actual type of nodes stored the sparse image. */
using NodeType = typename SparseImageType::NodeType;
/** The type for the variables of NodeType. Scalar or vector. */
using NodeDataType = typename NodeType::NodeDataType;
/** This function is not called from the FiniteDifferenceSparseImageFilter
* class because we need to work with neighborhoods of pointers to data
* variables instead of neighborhoods of data directly. This function is
* replaced by the ComputeSparseUpdate function. */
PixelType
ComputeUpdate(const NeighborhoodType &, void *, const FloatOffsetType &) override
{
return static_cast<PixelType>(nullptr);
}
/** The update called from the FiniteDifferenceSparseImageFilter. This
function replaces the ComputeUpdate function. */
virtual NodeDataType
ComputeSparseUpdate(NeighborhoodType & neighborhood,
void * globalData,
const FloatOffsetType & offset = FloatOffsetType(0.0)) const = 0;
/** This function provides support for a 2 step update computation that
* avoids repetitive computation. FiniteDifferenceSparseImageFilter first
* calls this function for all active pixels in the SparseImage class, then
* calls ComputeSparseUpdate for the same set of pixels. This is used in
* NormalVectorDiffusionFunction to calculate flux variables which are then
* used to compute the updates. Intermediate variables such as the flux in
* the above examples are stored in the nodes of the SparseImage
* itself. Therefore, this function will have to know about the NodeType it
* is dealing with. This function does nothing by default. */
virtual void
PrecomputeSparseUpdate(NeighborhoodType &) const
{}
protected:
FiniteDifferenceSparseImageFunction() = default;
~FiniteDifferenceSparseImageFunction() override = default;
void
PrintSelf(std::ostream & os, Indent indent) const override;
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkFiniteDifferenceSparseImageFunction.hxx"
#endif
#endif
|