rtkForwardWarpImageFilter.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 rtkForwardWarpImageFilter_h
#define rtkForwardWarpImageFilter_h

#include <itkWarpImageFilter.h>

#include "rtkMacro.h"

namespace rtk
{

template <class TInputImage,
          class TOutputImage = TInputImage,
          class TDVF = itk::Image<itk::CovariantVector<typename TInputImage::PixelType, TInputImage::ImageDimension>,
                                  TInputImage::ImageDimension>>
class ITK_TEMPLATE_EXPORT ForwardWarpImageFilter : public itk::WarpImageFilter<TInputImage, TOutputImage, TDVF>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(ForwardWarpImageFilter);

  using Self = ForwardWarpImageFilter;
  using Superclass = itk::WarpImageFilter<TInputImage, TOutputImage, TDVF>;
  using Pointer = itk::SmartPointer<Self>;
  using ConstPointer = itk::SmartPointer<const Self>;

  using DisplacementFieldType = TDVF;
  using DisplacementFieldPointer = typename DisplacementFieldType::Pointer;
  using DisplacementFieldConstPointer = typename DisplacementFieldType::ConstPointer;
  using DisplacementType = typename DisplacementFieldType::PixelType;

  using CoordinateType = double;
  using PointType = itk::Point<CoordinateType, Self::ImageDimension>;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(ForwardWarpImageFilter);

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

  void
  GenerateData() override;

  // Redefine stuff that is private in the Superclass
  void
  Protected_EvaluateDisplacementAtPhysicalPoint(const PointType & point, DisplacementType & output);

  bool m_Protected_DefFieldSizeSame;

  typename TOutputImage::IndexType m_Protected_StartIndex;
  typename TOutputImage::IndexType m_Protected_EndIndex;
};

} // end namespace rtk

#ifndef rtk_MANUAL_INSTANTIATION
#  include "rtkForwardWarpImageFilter.hxx"
#endif

#endif