Source/WTF/wtf/GetPtr.h

   1 /*
   2  *  Copyright (C) 2006 Apple Inc.
   3  *
   4  *  This library is free software; you can redistribute it and/or
   5  *  modify it under the terms of the GNU Library General Public
   6  *  License as published by the Free Software Foundation; either
   7  *  version 2 of the License, or (at your option) any later version.
   8  *
   9  *  This library is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  *  Library General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU Library General Public License
  15  *  along with this library; see the file COPYING.LIB.  If not, write to
  16  *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  17  *  Boston, MA 02110-1301, USA.
  18  *
  19  */
  20
  21 #ifndef WTF_GetPtr_h
  22 #define WTF_GetPtr_h
  23
  24 #include <memory>
  25
  26 namespace WTF {
  27
  28 template <typename T> inline T* getPtr(T* p) { return p; }
  29
  30 template <typename T> struct IsSmartPtr {
  31     static const bool value = false;
  32 };
  33
  34 template <typename T, bool isSmartPtr>
  35 struct GetPtrHelperBase;
  36
  37 template <typename T>
  38 struct GetPtrHelperBase<T, false /* isSmartPtr */> {
  39     typedef T* PtrType;
  40     static T* getPtr(T& p) { return &p; }
  41 };
  42
  43 template <typename T>
  44 struct GetPtrHelperBase<T, true /* isSmartPtr */> {
  45     typedef typename T::PtrType PtrType;
  46     static PtrType getPtr(const T& p) { return p.get(); }
  47 };
  48
  49 template <typename T>
  50 struct GetPtrHelper : GetPtrHelperBase<T, IsSmartPtr<T>::value> {
  51 };
  52
  53 template <typename T>
  54 inline typename GetPtrHelper<T>::PtrType getPtr(T& p)
  55 {
  56     return GetPtrHelper<T>::getPtr(p);
  57 }
  58
  59 template <typename T>
  60 inline typename GetPtrHelper<T>::PtrType getPtr(const T& p)
  61 {
  62     return GetPtrHelper<T>::getPtr(p);
  63 }
  64
  65 // Explicit specialization for C++ standard library types.
  66
  67 template <typename T, typename Deleter> struct IsSmartPtr<std::unique_ptr<T, Deleter>> {
  68     static const bool value = true;
  69 };
  70
  71 template <typename T, typename Deleter>
  72 struct GetPtrHelper<std::unique_ptr<T, Deleter>> {
  73     typedef T* PtrType;
  74     static T* getPtr(const std::unique_ptr<T, Deleter>& p) { return const_cast<T*>(p.get()); }
  75 };
  76
  77 } // namespace WTF
  78
  79 #endif // WTF_GetPtr_h