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
|
/*=========================================================================
*
* Copyright UMC Utrecht and contributors
*
* 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
*
* http://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.
*
*=========================================================================*/
/*=========================================================================
*
* Copyright Insight Software Consortium
*
* 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
*
* http://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 itkGPUUnaryFunctorImageFilter_h
#define itkGPUUnaryFunctorImageFilter_h
#include "itkGPUFunctorBase.h"
#include "itkGPUInPlaceImageFilter.h"
#include "itkUnaryFunctorImageFilter.h"
namespace itk
{
/** \class GPUUnaryFunctorImageFilter
* \brief Implements pixel-wise generic operation on one image using the GPU.
*
* GPU version of unary functor image filter.
* GPU Functor handles parameter setup for the GPU kernel.
*
* \note This file was taken from ITK 4.1.0.
* It was modified by Denis P. Shamonin and Marius Staring.
* Division of Image Processing,
* Department of Radiology, Leiden, The Netherlands.
* Added functionality is described in the Insight Journal paper:
* http://hdl.handle.net/10380/3393
*
* \ingroup ITKGPUCommon
*/
template <typename TInputImage,
typename TOutputImage,
typename TFunction,
typename TParentImageFilter = InPlaceImageFilter<TInputImage, TOutputImage>>
class ITK_TEMPLATE_EXPORT ITKOpenCL_EXPORT GPUUnaryFunctorImageFilter
: public GPUInPlaceImageFilter<TInputImage, TOutputImage, TParentImageFilter>
{
public:
ITK_DISALLOW_COPY_AND_MOVE(GPUUnaryFunctorImageFilter);
/** Standard class typedefs. */
using Self = GPUUnaryFunctorImageFilter;
using CPUSuperclass = TParentImageFilter;
using GPUSuperclass = GPUInPlaceImageFilter<TInputImage, TOutputImage>;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Run-time type information (and related methods). */
itkTypeMacro(GPUUnaryFunctorImageFilter, GPUInPlaceImageFilter);
/** Some typedefs. */
using FunctorType = TFunction;
using InputImageType = TInputImage;
using InputImagePointer = typename InputImageType::ConstPointer;
using InputImageRegionType = typename InputImageType::RegionType;
using InputImagePixelType = typename InputImageType::PixelType;
using OutputImageType = TOutputImage;
using OutputImagePointer = typename OutputImageType::Pointer;
using OutputImageRegionType = typename OutputImageType::RegionType;
using OutputImagePixelType = typename OutputImageType::PixelType;
/** ImageDimension constants */
itkStaticConstMacro(InputImageDimension, unsigned int, TInputImage::ImageDimension);
itkStaticConstMacro(OutputImageDimension, unsigned int, TOutputImage::ImageDimension);
FunctorType &
GetFunctor()
{
return m_Functor;
}
const FunctorType &
GetFunctor() const
{
return m_Functor;
}
/** Set the functor object. */
void
SetFunctor(const FunctorType & functor)
{
if (m_Functor != functor)
{
m_Functor = functor;
this->Modified();
}
}
protected:
GPUUnaryFunctorImageFilter() = default;
~GPUUnaryFunctorImageFilter() override = default;
void
GenerateOutputInformation() override;
void
GPUGenerateData() override;
/** GPU kernel handle is defined here instead of in the child class
* because GPUGenerateData() in this base class is used. */
int m_UnaryFunctorImageFilterGPUKernelHandle{};
private:
FunctorType m_Functor{};
};
} // end of namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkGPUUnaryFunctorImageFilter.hxx"
#endif
#endif
|