itkVectorImageToImageAdaptor.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 itkVectorImageToImageAdaptor_h
#define itkVectorImageToImageAdaptor_h

#include "itkImageAdaptor.h"
#include "itkVectorImage.h"

namespace itk
{
namespace Accessor
{
template< typename TType >
class VectorImageToImagePixelAccessor
  : private DefaultVectorPixelAccessor< TType >
{
public:

  typedef unsigned int VectorLengthType;

  typedef  TType ExternalType;

  typedef TType InternalType;

  typedef VariableLengthVector< TType > ActualPixelType;

  inline void Set(ActualPixelType output, const ExternalType & input) const
  {
    output[m_ComponentIdx] = input;
  }

  inline void Set(InternalType &output, const ExternalType & input,
                  const unsigned long offset) const
  {
    return Set( Superclass::Get( output, offset ), input );
  }

  inline ExternalType Get(const ActualPixelType & input) const
  {
    ExternalType output;

    output = input[m_ComponentIdx];
    return output;
  }

  inline ExternalType Get(const InternalType &input, const SizeValueType offset) const
  {
    return Get( Superclass::Get(input, offset) );
  }

  void SetExtractComponentIdx(VectorLengthType idx)
  {
    m_ComponentIdx = idx;
  }

  VectorLengthType GetExtractComponentIdx() const
  {
    return m_ComponentIdx;
  }

  void SetVectorLength(VectorLengthType l)
  {
    Superclass::SetVectorLength( l );
  }

  VectorLengthType GetVectorLength() const { return Superclass::GetVectorLength(); }

  VectorImageToImagePixelAccessor( unsigned int length = 1)
    :m_ComponentIdx(0)
    {
    Superclass::SetVectorLength( length );
    }

protected:
  typedef DefaultVectorPixelAccessor< TType > Superclass;

private:
  VectorLengthType m_ComponentIdx;
};
} // end namespace Accessor

template< typename TPixelType, unsigned int Dimension >
class VectorImageToImageAdaptor:public
  ImageAdaptor< VectorImage< TPixelType, Dimension >,
                Accessor::VectorImageToImagePixelAccessor< TPixelType > >
{
public:
  typedef VectorImageToImageAdaptor            Self;
  typedef VectorImage< TPixelType, Dimension > VectorImageType;
  typedef ImageAdaptor< VectorImageType,
                        Accessor::VectorImageToImagePixelAccessor< TPixelType >  > Superclass;

  typedef SmartPointer< Self >       Pointer;
  typedef SmartPointer< const Self > ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(VectorImageToImageAdaptor, ImageAdaptor);

  typedef typename Superclass::PixelContainer             PixelContainer;
  typedef typename Superclass::PixelContainerPointer      PixelContainerPointer;
  typedef typename Superclass::PixelContainerConstPointer PixelContainerConstPointer;
  typedef typename Superclass::IOPixelType                IOPixelType;

  typedef typename VectorImageType::VectorLengthType VectorLengthType;

  // Set/GetMethods to set the component to be extracted.
  void SetExtractComponentIndex(VectorLengthType componentIdx)
  {
    this->GetPixelAccessor().SetExtractComponentIdx(componentIdx);
  }

  // Set/GetMethods to set the component to be extracted.
  VectorLengthType GetExtractComponentIndex() const
  {
    return this->GetPixelAccessor().GetExtractComponentIdx();
  }

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

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

#endif