mitkImagePixelAccessor.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkImagePixelAccessor_h
#define mitkImagePixelAccessor_h
 
#include "mitkImage.h"
#include "mitkImageDataItem.h"
 
#include <typeinfo>
 
namespace mitk
{
  template <class TPixel, unsigned int VDimension = 3>
  class ImagePixelAccessor
  {
  public:
    typedef itk::Index<VDimension> IndexType;
    typedef ImagePixelAccessor<TPixel, VDimension> ImagePixelAccessorType;
    typedef Image::ConstPointer ImageConstPointer;
 
    int GetDimension(int i) const { return m_ImageDataItem->GetDimension(i); }
 
  private:
  friend class Image;
 
  protected:
    ImagePixelAccessor(ImageConstPointer iP, const mitk::ImageDataItem *iDI) : m_ImageDataItem(iDI)
    {
      if (iDI == nullptr)
      {
        m_ImageDataItem = iP->GetChannelData();
      }
      CheckData(iP.GetPointer());
    }
 
    virtual ~ImagePixelAccessor() {}
 
    void CheckData( const Image *image )
    {
      // Check if Dimensions are correct
      if ( m_ImageDataItem == nullptr )
      {
        if ( image->GetDimension() != VDimension )
        {
          mitkThrow() << "Invalid ImageAccessor: The Dimensions of ImageAccessor and Image are not equal."
          << " They have to be equal if an entire image is requested."
          << " image->GetDimension(): " << image->GetDimension() << " , VDimension: " << VDimension;
        }
      }
      else
      {
        if ( m_ImageDataItem->GetDimension() != VDimension )
        {
          mitkThrow() << "Invalid ImageAccessor: The Dimensions of ImageAccessor and ImageDataItem are not equal."
          << " m_ImageDataItem->GetDimension(): " << m_ImageDataItem->GetDimension() << " , VDimension: " << VDimension;
        }
      }
 
      if (!( image->GetPixelType() == mitk::MakePixelType< itk::Image< TPixel, VDimension > >() ||
             image->GetPixelType() == mitk::MakePixelType< itk::VectorImage< TPixel, VDimension > >
               ( image->GetPixelType().GetNumberOfComponents() )
         ) )
      {
        mitkThrow() << "Invalid ImageAccessor: PixelTypes of Image and ImageAccessor are not equal."
        << " image->GetPixelType(): " << typeid(image->GetPixelType()).name()
        << "\n m_ImageDataItem->GetDimension(): " << m_ImageDataItem->GetDimension()
        << " , VDimension: " << VDimension
        << " , TPixel: " << typeid(TPixel).name()
        << " , NumberOfComponents: " << image->GetPixelType().GetNumberOfComponents() << std::endl;
      }
    }
 
    const ImageDataItem *m_ImageDataItem;
 
    BaseGeometry::Pointer m_Geometry;
 
    itk::ImageRegion<VDimension> *m_SubRegion;
 
    int m_Options;
 
    unsigned int GetOffset(const IndexType &idx) const
    {
      const unsigned int *imageDims = m_ImageDataItem->m_Dimensions;
 
      unsigned int offset = 0;
      switch (VDimension)
      {
        case 4:
          offset += idx[3] * imageDims[0] * imageDims[1] * imageDims[2];
          /* FALLTHRU */
        case 3:
          offset += idx[2] * imageDims[0] * imageDims[1];
          /* FALLTHRU */
        case 2:
          offset += idx[0] + idx[1] * imageDims[0];
          /* FALLTHRU */
        default:
          break;
      }
      return offset;
    }
  };
}
 
#endif