diff -r 000000000000 -r 6474c204b198 gfx/skia/trunk/include/utils/win/SkTScopedComPtr.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/gfx/skia/trunk/include/utils/win/SkTScopedComPtr.h Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,76 @@ + +/* + * Copyright 2011 Google Inc. + * + * Use of this source code is governed by a BSD-style license that can be + * found in the LICENSE file. + */ + +#ifndef SkTScopedComPtr_DEFINED +#define SkTScopedComPtr_DEFINED + +#include "SkTypes.h" +#include "SkTemplates.h" + +template +class SkBlockComRef : public T { +private: + virtual ULONG STDMETHODCALLTYPE AddRef(void) = 0; + virtual ULONG STDMETHODCALLTYPE Release(void) = 0; +}; + +template T* SkRefComPtr(T* ptr) { + ptr->AddRef(); + return ptr; +} + +template T* SkSafeRefComPtr(T* ptr) { + if (ptr) { + ptr->AddRef(); + } + return ptr; +} + +template +class SkTScopedComPtr : SkNoncopyable { +private: + T *fPtr; + +public: + explicit SkTScopedComPtr(T *ptr = NULL) : fPtr(ptr) { } + ~SkTScopedComPtr() { + this->reset(); + } + T &operator*() const { SkASSERT(fPtr != NULL); return *fPtr; } + SkBlockComRef *operator->() const { + return static_cast*>(fPtr); + } + /** + * Returns the address of the underlying pointer. + * This is dangerous -- it breaks encapsulation and the reference escapes. + * Must only be used on instances currently pointing to NULL, + * and only to initialize the instance. + */ + T **operator&() { SkASSERT(fPtr == NULL); return &fPtr; } + T *get() const { return fPtr; } + void reset() { + if (NULL != this->fPtr) { + this->fPtr->Release(); + this->fPtr = NULL; + } + } + + void swap(SkTScopedComPtr& that) { + T* temp = this->fPtr; + this->fPtr = that.fPtr; + that.fPtr = temp; + } + + T* release() { + T* temp = this->fPtr; + this->fPtr = NULL; + return temp; + } +}; + +#endif