CandFitTrackSA/Handle.h

Go to the documentation of this file.

#ifndef HANDLE_H
#define HANDLE_H


#include <iostream>

template <typename T>
class Handle {

public:
    Handle();
    
    Handle(T* ptr);
    
    Handle(const Handle& rhs);
    
    ~Handle();

    Handle& operator=(const Handle& rhs);
    
    T* operator->() const;
    
    T& operator*() const;

    void release();

    bool operator==(const Handle& rhs);
    
    bool operator!=(const Handle& rhs);

private:
    T* p;
    
    void init();
};


template <typename T>
Handle<T>::Handle() : p(0)
{}

template <typename T>
Handle<T>::Handle(T* realPtr) : p(realPtr)
{
    init();
}

template <typename T>
Handle<T>::Handle(const Handle& rhs) : p(rhs.p)
{
    init();
}


template <typename T>
Handle<T>& Handle<T>::operator=(const Handle& rhs)
{
    if (p != rhs.p) {
        if (p) {
            p->release();
        }

        p = rhs.p;
        init();
    }
    return *this;
}


template <typename T>
void Handle<T>::init()
{
    if (p == 0) {
        return;
    }

    p->add_ref();
}

template <typename T>
Handle<T>::~Handle()
{
    if (p) p->release();
}


template <typename T>
T* Handle<T>::operator->() const 
{ 
    return p; 
}


template <typename T>
T& Handle<T>::operator*() const 
{ 
    return *p; 
}

template <typename T>
void Handle<T>::release()
{
    if (p) p->release();
    p = 0;
}
 

template <typename T>
bool Handle<T>::operator==(const Handle& rhs)
{
    if (p == rhs.p)
       return true;

    return false;
}

template <typename T>
bool Handle<T>::operator!=(const Handle& rhs)
{
    if (p != rhs.p)
       return true;

    return false;
}
   
template <typename T>
std::ostream& operator<<(std::ostream& o, const Handle<T>& handle)
{
    o << *handle;
    return o;
}

#endif  // HANDLE_H

---