rtkDePierroRegularizationImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright RTK 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
 *
 *         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 rtkDePierroRegularizationImageFilter_h
#define rtkDePierroRegularizationImageFilter_h

#include <itkMultiplyImageFilter.h>
#include <itkSubtractImageFilter.h>
#include <itkDivideImageFilter.h>
#include <itkImageKernelOperator.h>
#include <itkNeighborhoodOperatorImageFilter.h>
#include <itkConstantBoundaryCondition.h>
#include "rtkConstantImageSource.h"

namespace rtk
{

template <class TInputImage, class TOutputImage = TInputImage>
class ITK_TEMPLATE_EXPORT DePierroRegularizationImageFilter : public itk::ImageToImageFilter<TInputImage, TOutputImage>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(DePierroRegularizationImageFilter);

  using Self = DePierroRegularizationImageFilter;
  using Superclass = itk::ImageToImageFilter<TOutputImage, TOutputImage>;
  using Pointer = itk::SmartPointer<Self>;
  using ConstPointer = itk::SmartPointer<const Self>;

  using InputImageType = TInputImage;
  using InputImagePointerType = typename TInputImage::Pointer;
  using OutputImageType = TOutputImage;
  using InputPixelType = typename TInputImage::PixelType;

  static constexpr unsigned int InputImageDimension = TInputImage::ImageDimension;

  using MultiplyImageFilterType = itk::MultiplyImageFilter<InputImageType, InputImageType>;
  using MultpiplyImageFilterPointerType = typename MultiplyImageFilterType::Pointer;
  using ConstantVolumeSourceType = rtk::ConstantImageSource<InputImageType>;
  using ConstantVolumeSourcePointerType = typename ConstantVolumeSourceType::Pointer;
  using SubtractImageFilterType = itk::SubtractImageFilter<InputImageType, InputImageType>;
  using SubtractImageFilterPointerType = typename SubtractImageFilterType::Pointer;
  using ImageKernelOperatorType = itk::ImageKernelOperator<InputPixelType, InputImageDimension>;
  using NOIFType = itk::NeighborhoodOperatorImageFilter<InputImageType, InputImageType>;
  using NOIFPointerType = typename NOIFType::Pointer;
  using CustomBinaryFilterType = itk::BinaryGeneratorImageFilter<InputImageType, InputImageType, OutputImageType>;
  using CustomBinaryFilterPointerType = typename CustomBinaryFilterType::Pointer;

  using BoundaryCondition = itk::ConstantBoundaryCondition<InputImageType>;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(DePierroRegularizationImageFilter);

  itkGetMacro(Beta, double);
  itkSetMacro(Beta, double);

protected:
  DePierroRegularizationImageFilter();
  ~DePierroRegularizationImageFilter() override = default;

  void
  GenerateInputRequestedRegion() override;

  void
  GenerateOutputInformation() override;

  void
  GenerateData() override;

  MultpiplyImageFilterPointerType m_MultiplyConstant1ImageFilter;
  MultpiplyImageFilterPointerType m_MultiplyConstant2ImageFilter;
  ConstantVolumeSourcePointerType m_KernelImage;
  ConstantVolumeSourcePointerType m_DefaultNormalizationVolume;
  SubtractImageFilterPointerType  m_SubtractImageFilter;
  BoundaryCondition               m_BoundsCondition;
  ImageKernelOperatorType         m_KernelOperator;
  NOIFPointerType                 m_ConvolutionFilter;
  CustomBinaryFilterPointerType   m_CustomBinaryFilter;

private:
  double m_Beta{ 0.01 };

}; // end of class

} // end namespace rtk

#ifndef ITK_MANUAL_INSTANTIATION
#  include "rtkDePierroRegularizationImageFilter.hxx"
#endif

#endif