diff -r 000000000000 -r 4f2f89ce4247 JavaScriptCore/wtf/RetainPtr.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/JavaScriptCore/wtf/RetainPtr.h Fri Sep 17 09:02:29 2010 +0300 @@ -0,0 +1,227 @@ +/* + * Copyright (C) 2005, 2006, 2007, 2008 Apple Inc. All rights reserved. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Library General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Library General Public License for more details. + * + * You should have received a copy of the GNU Library General Public License + * along with this library; see the file COPYING.LIB. If not, write to + * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + * Boston, MA 02110-1301, USA. + * + */ + +#ifndef RetainPtr_h +#define RetainPtr_h + +#include "HashTraits.h" +#include "TypeTraits.h" +#include +#include + +#ifdef __OBJC__ +#import +#endif + +namespace WTF { + + // Unlike most most of our smart pointers, RetainPtr can take either the pointer type or the pointed-to type, + // so both RetainPtr and RetainPtr will work. + + enum AdoptCFTag { AdoptCF }; + enum AdoptNSTag { AdoptNS }; + +#ifdef __OBJC__ + inline void adoptNSReference(id ptr) + { + if (ptr) { + CFRetain(ptr); + [ptr release]; + } + } +#endif + + template class RetainPtr { + public: + typedef typename RemovePointer::Type ValueType; + typedef ValueType* PtrType; + + RetainPtr() : m_ptr(0) {} + RetainPtr(PtrType ptr) : m_ptr(ptr) { if (ptr) CFRetain(ptr); } + + RetainPtr(AdoptCFTag, PtrType ptr) : m_ptr(ptr) { } + RetainPtr(AdoptNSTag, PtrType ptr) : m_ptr(ptr) { adoptNSReference(ptr); } + + RetainPtr(const RetainPtr& o) : m_ptr(o.m_ptr) { if (PtrType ptr = m_ptr) CFRetain(ptr); } + + // Hash table deleted values, which are only constructed and never copied or destroyed. + RetainPtr(HashTableDeletedValueType) : m_ptr(hashTableDeletedValue()) { } + bool isHashTableDeletedValue() const { return m_ptr == hashTableDeletedValue(); } + + ~RetainPtr() { if (PtrType ptr = m_ptr) CFRelease(ptr); } + + template RetainPtr(const RetainPtr& o) : m_ptr(o.get()) { if (PtrType ptr = m_ptr) CFRetain(ptr); } + + PtrType get() const { return m_ptr; } + + PtrType releaseRef() { PtrType tmp = m_ptr; m_ptr = 0; return tmp; } + + PtrType operator->() const { return m_ptr; } + + bool operator!() const { return !m_ptr; } + + // This conversion operator allows implicit conversion to bool but not to other integer types. + typedef PtrType RetainPtr::*UnspecifiedBoolType; + operator UnspecifiedBoolType() const { return m_ptr ? &RetainPtr::m_ptr : 0; } + + RetainPtr& operator=(const RetainPtr&); + template RetainPtr& operator=(const RetainPtr&); + RetainPtr& operator=(PtrType); + template RetainPtr& operator=(U*); + + void adoptCF(PtrType); + void adoptNS(PtrType); + + void swap(RetainPtr&); + + private: + static T* hashTableDeletedValue() { return reinterpret_cast(-1); } + + PtrType m_ptr; + }; + + template inline RetainPtr& RetainPtr::operator=(const RetainPtr& o) + { + PtrType optr = o.get(); + if (optr) + CFRetain(optr); + PtrType ptr = m_ptr; + m_ptr = optr; + if (ptr) + CFRelease(ptr); + return *this; + } + + template template inline RetainPtr& RetainPtr::operator=(const RetainPtr& o) + { + PtrType optr = o.get(); + if (optr) + CFRetain(optr); + PtrType ptr = m_ptr; + m_ptr = optr; + if (ptr) + CFRelease(ptr); + return *this; + } + + template inline RetainPtr& RetainPtr::operator=(PtrType optr) + { + if (optr) + CFRetain(optr); + PtrType ptr = m_ptr; + m_ptr = optr; + if (ptr) + CFRelease(ptr); + return *this; + } + + template inline void RetainPtr::adoptCF(PtrType optr) + { + PtrType ptr = m_ptr; + m_ptr = optr; + if (ptr) + CFRelease(ptr); + } + + template inline void RetainPtr::adoptNS(PtrType optr) + { + adoptNSReference(optr); + + PtrType ptr = m_ptr; + m_ptr = optr; + if (ptr) + CFRelease(ptr); + } + + template template inline RetainPtr& RetainPtr::operator=(U* optr) + { + if (optr) + CFRetain(optr); + PtrType ptr = m_ptr; + m_ptr = optr; + if (ptr) + CFRelease(ptr); + return *this; + } + + template inline void RetainPtr::swap(RetainPtr& o) + { + std::swap(m_ptr, o.m_ptr); + } + + template inline void swap(RetainPtr& a, RetainPtr& b) + { + a.swap(b); + } + + template inline bool operator==(const RetainPtr& a, const RetainPtr& b) + { + return a.get() == b.get(); + } + + template inline bool operator==(const RetainPtr& a, U* b) + { + return a.get() == b; + } + + template inline bool operator==(T* a, const RetainPtr& b) + { + return a == b.get(); + } + + template inline bool operator!=(const RetainPtr& a, const RetainPtr& b) + { + return a.get() != b.get(); + } + + template inline bool operator!=(const RetainPtr& a, U* b) + { + return a.get() != b; + } + + template inline bool operator!=(T* a, const RetainPtr& b) + { + return a != b.get(); + } + + template struct HashTraits > : GenericHashTraits > { + static const bool emptyValueIsZero = true; + static void constructDeletedValue(RetainPtr

& slot) { new (&slot) RetainPtr

(HashTableDeletedValue); } + static bool isDeletedValue(const RetainPtr

& value) { return value == reinterpret_cast(-1); } + }; + + template struct PtrHash > : PtrHash { + using PtrHash::hash; + static unsigned hash(const RetainPtr

& key) { return hash(key.get()); } + using PtrHash::equal; + static bool equal(const RetainPtr

& a, const RetainPtr

& b) { return a == b; } + static bool equal(P* a, const RetainPtr

& b) { return a == b; } + static bool equal(const RetainPtr

& a, P* b) { return a == b; } + }; + + template struct DefaultHash > { typedef PtrHash > Hash; }; + +} // namespace WTF + +using WTF::AdoptCF; +using WTF::AdoptNS; +using WTF::RetainPtr; + +#endif // WTF_RetainPtr_h