ImageContainerByITKImage.h

#pragma once

#if defined(ImageContainerByITKImage_RECURSES)
#error Recursive header files inclusion detected in ImageContainerByITKImage.h
#else // defined(ImageContainerByITKImage_RECURSES)

#define ImageContainerByITKImage_RECURSES

#if !defined ImageContainerByITKImage_h

#define ImageContainerByITKImage_h

// Inclusions
#include "DGtal/base/Common.h"
#include "DGtal/base/CLabel.h"
#include "DGtal/kernel/domains/CDomain.h"

#include <itkImage.h>
#include <itkImageRegionConstIterator.h>
#include <itkImageRegionIterator.h>
#include <iostream>


namespace DGtal
{
  namespace experimental
  {

    // template class ImageContainerByITKImage
    template <typename TDomain, typename TValue>
    class ImageContainerByITKImage
    {
      // ----------------------- Standard services ------------------------------
    public:

      BOOST_CONCEPT_ASSERT(( CLabel<TValue> ));
      BOOST_CONCEPT_ASSERT(( CDomain<TDomain> ));

      typedef TValue Value;
      typedef TDomain Domain;

      // static constants
      static const typename Domain::Dimension dimension = Domain::dimension;

      typedef typename Domain::Point Point;
      typedef typename Domain::Vector Vector;
      typedef typename Domain::Dimension Dimension;
      typedef typename Domain::Integer Integer;
      typedef typename Domain::Size Size;
      typedef Point Vertex;

      typedef typename itk::Image< TValue, dimension> ITKImage;
      typedef typename ITKImage::Pointer ITKImagePointer;
      typedef typename itk::ImageRegionConstIterator< ITKImage > ConstIterator;
      typedef typename itk::ImageRegionIterator< ITKImage > Iterator;


      ImageContainerByITKImage(const Point &aPointA,
             const Point &aPointB );

      ImageContainerByITKImage(const Point &aPointA,
             const Point &aPointB,
             ITKImagePointer &aRef);

      ~ImageContainerByITKImage();

      // ----------------------- Interface --------------------------------------
    public:


      Value operator()(const Point &aPoint) const;

      Value operator()(const ConstIterator &it) const;

      Value operator()(const Iterator &it) const;


      void setValue(const Point &aPoint, const Value &aValue);

      void setValue(Iterator &it, const Value &V);

      // ------------------------- methods ------------------------------


      Domain domain() const
      {
  return Domain(myLowerBound, myUpperBound);
      }
    
      Point extent() const
      {
  return myUpperBound - myLowerBound;
      }


      ITKImagePointer getImagePointer() const
      {
  return myITKImagePointer;
      }

      // ------------------------- stream ------------------------------

      void selfDisplay ( std::ostream & out ) const;

      bool isValid() const;

      // ------------------------- Iterators ------------------------------
      ConstIterator begin() const
      {
  return myConstItBegin;
      }

      Iterator begin()
      {
  return myItBegin;
      }

      ConstIterator begin ( const Point &aPoint ) const;

      const ConstIterator end() const
      {
  return myConstItEnd;
      }

      Iterator end()
      {
  return myItEnd;
      }

      ConstIterator end(const Point &aPoint) const;

      // ------------------------- Private Datas --------------------------------
    private:

      // ------------------------- Hidden services ------------------------------
    protected:

      ImageContainerByITKImage();

    private:

      ImageContainerByITKImage ( const ImageContainerByITKImage & other );

      ImageContainerByITKImage & operator= ( const ImageContainerByITKImage & other );

      // ------------------------- Internals ------------------------------------
    private:

      Point myLowerBound;
      Point myUpperBound;
      ITKImagePointer myITKImagePointer;
      typename ITKImage::RegionType myRegion;
      ConstIterator myConstItBegin;
      Iterator myItBegin;
      ConstIterator myConstItEnd;
      Iterator myItEnd;

    }; // end of class ImageContainerByITKImage

  }

  template <typename T, typename TV>
  std::ostream&
  operator<< ( std::ostream & out, const experimental::ImageContainerByITKImage<T, TV> & object );

}
// Includes inline functions.
#include "DGtal/images/ImageContainerByITKImage.ih"

//

#endif // !defined ImageContainerByITKImage_h

#undef ImageContainerByITKImage_RECURSES
#endif // else defined(ImageContainerByITKImage_RECURSES)