itkAdaptImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  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 itkAdaptImageFilter_h
#define itkAdaptImageFilter_h

#include "itkUnaryFunctorImageFilter.h"

namespace itk
{
namespace Functor
{
template< typename TInput, typename TAccessor >
class AccessorFunctor
{
public:
  typedef AccessorFunctor Self;
  typedef TAccessor       AccessorType;

  AccessorFunctor():m_Accessor() {}
  ~AccessorFunctor() {}

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

  AccessorType & GetAccessor()
  {
    return m_Accessor;
  }

  AccessorFunctor & operator=(const AccessorFunctor & functor)
  {
    m_Accessor = functor.m_Accessor;
    return *this;
  }

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

  bool operator!=(const Self & functor) const
  {
    return ( m_Accessor != functor.m_Accessor );
  }

  bool operator==(const Self & other) const
  {
    return !( *this != other );
  }

private:
  AccessorType m_Accessor;
};
}

template< typename TInputImage, typename TOutputImage, typename TAccessor >
class AdaptImageFilter:
  public UnaryFunctorImageFilter< TInputImage, TOutputImage,
                                  Functor::AccessorFunctor< typename TInputImage::PixelType, TAccessor > >
{
public:
  typedef AdaptImageFilter Self;

  typedef UnaryFunctorImageFilter< TInputImage,
                                   TOutputImage,
                                   Functor::AccessorFunctor<
                                     typename TInputImage::PixelType,
                                     TAccessor > >  Superclass;

  typedef SmartPointer< Self >             Pointer;
  typedef SmartPointer< const Self >       ConstPointer;
  typedef typename Superclass::FunctorType FunctorType;

  itkNewMacro(Self);

  typedef TAccessor AccessorType;

  itkTypeMacro(AdaptImageFilter, UnaryFunctorImageFilter);

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

  void SetAccessor(AccessorType & accessor)
  {
    FunctorType functor;

    functor = this->GetFunctor();
    if ( accessor != functor.GetAccessor() )
      {
      functor.SetAccessor(accessor);
      this->SetFunctor(functor);
      this->Modified();
      }
  }

protected:
  AdaptImageFilter() {}
  virtual ~AdaptImageFilter() ITK_OVERRIDE {}

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(AdaptImageFilter);
};
} // end namespace itk

#endif