itkTreeIteratorClone.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkTreeIteratorClone.h,v $
  Language:  C++
  Date:      $Date: 2009-03-03 15:10:40 $
  Version:   $Revision: 1.3 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkTreeIteratorClone_h
#define __itkTreeIteratorClone_h

#include "itkMacro.h"
#include <iostream>

namespace itk
{

template <class TObjectType>
class ITK_EXPORT TreeIteratorClone 
{
public:
  
  typedef TreeIteratorClone<TObjectType> Self;
  typedef TObjectType                    ObjectType;

  TreeIteratorClone () 
    { 
    m_Pointer = 0; 
    }

  TreeIteratorClone (const TreeIteratorClone<ObjectType> &p)
    { 
    m_Pointer = 0;
    if(p.m_Pointer != NULL)
      {
      m_Pointer = p.m_Pointer->Clone();
      }
    }

  TreeIteratorClone (ObjectType *p)
    { 
    m_Pointer = 0;
    if(p!=NULL) 
      {
      m_Pointer=p->Clone();
      }
    }

  TreeIteratorClone (const ObjectType &p)
    { 
    m_Pointer = 0;
    m_Pointer=const_cast<ObjectType*>(&p)->Clone();
    }

  ~TreeIteratorClone ()
    {
    delete m_Pointer;
    m_Pointer = 0;
    }

  ObjectType *operator -> () const
    { return m_Pointer; }

  bool IsNotNull() const
    { return m_Pointer != 0; }
  bool IsNull() const
    { return m_Pointer == 0; }

  template <typename TR>
  bool operator == ( TR r ) const
    { return (m_Pointer == (ObjectType*)(r) ); }

  template <typename TR>
  bool operator != ( TR r ) const
    { return (m_Pointer != (ObjectType*)(r) ); }
    
  ObjectType *GetPointer () const 
    { return m_Pointer; }

  bool operator < (const TreeIteratorClone &r) const
    { return (void*)m_Pointer < (void*) r.m_Pointer; }

  bool operator > (const TreeIteratorClone &r) const
    { return (void*)m_Pointer > (void*) r.m_Pointer; }

  bool operator <= (const TreeIteratorClone &r) const
    { return (void*)m_Pointer <= (void*) r.m_Pointer; }

  bool operator >= (const TreeIteratorClone &r) const
    { return (void*)m_Pointer >= (void*) r.m_Pointer; }

  TreeIteratorClone &operator = (const TreeIteratorClone &r)
    { return this->operator = (r.GetPointer()); }

  TreeIteratorClone &operator = (const ObjectType *r)
    {
    if (m_Pointer != r)
      {
      delete m_Pointer;
      m_Pointer = 0;  
      if(r!=NULL) 
        m_Pointer=const_cast<ObjectType*>(r)->Clone();
      }
    return *this;
    }

  Self &
  operator++()
    {
    if(m_Pointer)
      {
      ++(*m_Pointer);
      }
    return *this;
    }

  const Self
  operator++(int)
    {
    if(m_Pointer)
      {
      const Self oldValue( m_Pointer ); // create a copy of the iterator behind the pointer (Clone())
      ++(*m_Pointer);
      return oldValue;
      }
    }
   
  ObjectType *Print (std::ostream& os) const 
    { 
    // This prints the object pointed to by the pointer  
    (*m_Pointer).Print(os);  
    return m_Pointer;
    } 

private:
  ObjectType* m_Pointer;
};  


template <typename T>
std::ostream& operator<< (std::ostream& os, TreeIteratorClone<T> p) 
{
  p.Print(os); 
  return os;
}

} // end namespace itk
  
#endif