itkAdaptImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  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 itkAdaptImageFilter_h
#define itkAdaptImageFilter_h
 
#include "itkUnaryFunctorImageFilter.h"
 
namespace itk
{
namespace Functor
{
template <typename TInput, typename TAccessor>
class AccessorFunctor
{
public:
  using Self = AccessorFunctor;
  using AccessorType = TAccessor;

  using OutputType = typename TAccessor::ExternalType;
  inline OutputType
  operator()(const TInput & A) const
  {
    return m_Accessor.Get(A);
  }

  AccessorType &
  GetAccessor()
  {
    return m_Accessor;
  }

  void
  SetAccessor(AccessorType & accessor)
  {
    m_Accessor = accessor;
  }

  bool
  operator==(const Self & other) const
  {
    return m_Accessor == other.m_Accessor;
  }
 
  ITK_UNEQUAL_OPERATOR_MEMBER_FUNCTION(Self);
 
private:
  AccessorType m_Accessor{};
};
} // namespace Functor

template <typename TInputImage, typename TOutputImage, typename TAccessor>
class AdaptImageFilter
  : public UnaryFunctorImageFilter<TInputImage,
                                   TOutputImage,
                                   Functor::AccessorFunctor<typename TInputImage::PixelType, TAccessor>>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(AdaptImageFilter);

  using Self = AdaptImageFilter;
 
  using Superclass = UnaryFunctorImageFilter<TInputImage,
                                             TOutputImage,
                                             Functor::AccessorFunctor<typename TInputImage::PixelType, TAccessor>>;
 
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
  using typename Superclass::FunctorType;

  itkNewMacro(Self);

  using AccessorType = TAccessor;

  itkOverrideGetNameOfClassMacro(AdaptImageFilter);

  AccessorType &
  GetAccessor()
  {
    return this->GetFunctor().GetAccessor();
  }

  void
  SetAccessor(AccessorType & accessor)
  {
    FunctorType functor = this->GetFunctor();
    if (accessor != functor.GetAccessor())
    {
      functor.SetAccessor(accessor);
      this->SetFunctor(functor);
      this->Modified();
    }
  }
 
protected:
  AdaptImageFilter() = default;
  ~AdaptImageFilter() override = default;
};
} // end namespace itk
 
#endif