include/stg/ref.h

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  19  */
  20
  21 namespace STG
  22 {
  23
  24 // The implementation is derived from
  25 // http://en.cppreference.com/w/cpp/utility/functional/reference_wrapper
  26 // and
  27 // http://en.cppreference.com/w/cpp/memory/addressof
  28
  29 template< class T >
  30 T* AddressOf(T& arg)
  31 {
  32     return reinterpret_cast<T*>(&const_cast<char&>(reinterpret_cast<const volatile char&>(arg)));
  33 }
  34
  35 template <class T>
  36 class RefWrapper {
  37     public:
  38           typedef T type;
  39           RefWrapper(T& ref) throw() : m_ptr(AddressOf(ref)) {}
  40           RefWrapper(const RefWrapper& rhs) : m_ptr(rhs.m_ptr) {}
  41           RefWrapper& operator=(const RefWrapper& rhs) throw() { m_ptr = rhs.m_ptr; }
  42           operator T& () const throw() { return *m_ptr; }
  43           T& get() const throw() { return *m_ptr; }
  44
  45           void operator()() { (*m_ptr)(); }
  46           template <typename A>
  47           void operator()(A a) { (*m_ptr)(a); }
  48           template <typename A1, typename A2>
  49           void operator()(A1 a1, A2 a2) { (*m_ptr)(a1, a2); }
  50     private:
  51           T* m_ptr;
  52 };
  53
  54 } // namespace STG