itkWeakPointer.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 __itkWeakPointer_h
#define __itkWeakPointer_h

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

namespace itk
{
template< typename TObjectType >
class WeakPointer
{
public:
  typedef TObjectType ObjectType;

  WeakPointer ()
  { m_Pointer = ITK_NULLPTR; }

  WeakPointer (const WeakPointer< ObjectType > & p):m_Pointer(p.m_Pointer) {}

  WeakPointer (ObjectType *p):m_Pointer(p) {}

  ~WeakPointer ()
  { m_Pointer = ITK_NULLPTR; }

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

  operator ObjectType *() const
        { return m_Pointer; }

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

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

  ObjectType * GetPointer() const
  { return m_Pointer; }

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

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

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

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

  // cppcheck-suppress operatorEqVarError
  WeakPointer & operator=(const WeakPointer & r)
  { return this->operator=( r.GetPointer() ); }

  WeakPointer & operator=(ObjectType *r)
  {
    m_Pointer = r;
    return *this;
  }

  ObjectType * Print(std::ostream & os) const
  {
    if( this->IsNull() )
      {
      os << "(null)";
      }
    else
      {
      // 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, WeakPointer< T > p)
{
  p.Print(os);
  return os;
}
} // end namespace itk

#endif