// **************************************************************************
// * This file is part of the FreeFileSync project. It is distributed under *
// * GNU General Public License: http://www.gnu.org/licenses/gpl.html       *
// * Copyright (C) 2008-2011 ZenJu (zhnmju123 AT gmx.de)                    *
// **************************************************************************

#ifndef SMART_COM_PTR_H
#define SMART_COM_PTR_H

#include <algorithm>
#include <Objbase.h>

namespace zen
{
/*
ComPtr: RAII class handling COM objects

Example:
   --------
    ComPtr<IUPnPDeviceFinder> devFinder;
	if (FAILED(::CoCreateInstance(CLSID_UPnPDeviceFinder,
                          NULL,
                          CLSCTX_ALL,
                          IID_PPV_ARGS(devFinder.init()))))
					  return -1;

	ComPtr<IEnumUnknown> devEnum = com_dynamic_cast<IEnumUnknown>(devColl);
	if (!devEnum)
		return -1;
*/

template <class T>
class ComPtr
{
public:
    ComPtr() : ptr(NULL) {}

    ComPtr(const ComPtr&  other) : ptr(other.ptr) { if (ptr) ptr->AddRef(); }
    ComPtr(      ComPtr&& other) : ptr(other.release()) {}

    ComPtr& operator=(const ComPtr&  other) { ComPtr(other).swap(*this); return *this; }
    ComPtr& operator=(      ComPtr&& other) { swap(other); return *this; }

    ~ComPtr() { if (ptr) ptr->Release(); }

    T** init() //get pointer for use with ::CoCreateInstance()
    {
        ComPtr<T>().swap(*this);
        return &ptr;
    }

    T* get() const { return ptr; }

    T* release() //throw()
    {
        T* tmp = ptr;
        ptr = NULL;
        return tmp;
    }

    void swap(ComPtr& rhs) { std::swap(ptr, rhs.ptr); } //throw()

    T* operator->() const { return ptr; }

private:
    T* ptr;

    struct ConversionToBool { int dummy; };
public:
    //use member pointer as implicit conversion to bool (C++ Templates - Vandevoorde/Josuttis; chapter 20)
    operator int ConversionToBool::* () const { return ptr != NULL ? &ConversionToBool::dummy : NULL; }
};


template <class S, class T>
ComPtr<S> com_dynamic_cast(const ComPtr<T>& other); //throw()

























//################# Inline Implementation #############################

//we cannot specialize std::swap() for a class template and are not allowed to overload it => offer swap in own namespace
template <class T> inline
void swap(zen::ComPtr<T>& lhs, zen::ComPtr<T>& rhs)
{
    lhs.swap(rhs);
}


template <class S, class T> inline
ComPtr<S> com_dynamic_cast(const ComPtr<T>& other) //throw()
{
    ComPtr<S> outPtr;
    if (other)
        other->QueryInterface(IID_PPV_ARGS(outPtr.init()));
    return outPtr;
}
}


#endif //SMART_COM_PTR_H