OpenCores
URL https://opencores.org/ocsvn/openrisc/openrisc/trunk

Subversion Repositories openrisc

[/] [openrisc/] [trunk/] [gnu-stable/] [gcc-4.5.1/] [libstdc++-v3/] [include/] [std/] [future] - Rev 866

Go to most recent revision | Compare with Previous | Blame | View Log

// <future> -*- C++ -*-

// Copyright (C) 2009, 2010 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library.  This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 3, 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 General Public License for more details.

// Under Section 7 of GPL version 3, you are granted additional
// permissions described in the GCC Runtime Library Exception, version
// 3.1, as published by the Free Software Foundation.

// You should have received a copy of the GNU General Public License and
// a copy of the GCC Runtime Library Exception along with this program;
// see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
// <http://www.gnu.org/licenses/>.

/** @file future
 *  This is a Standard C++ Library header.
 */

#ifndef _GLIBCXX_FUTURE
#define _GLIBCXX_FUTURE 1

#pragma GCC system_header

#ifndef __GXX_EXPERIMENTAL_CXX0X__
# include <bits/c++0x_warning.h>
#else

#include <functional>
#include <memory>
#include <mutex>
#include <thread>
#include <condition_variable>
#include <system_error>
#include <exception>
#include <atomic>
#include <bits/functexcept.h>

namespace std
{
  /**
   * @defgroup futures Futures
   * @ingroup concurrency
   *
   * Classes for futures support.
   * @{
   */

  /// Error code for futures
  enum class future_errc
  {
    broken_promise,
    future_already_retrieved,
    promise_already_satisfied,
    no_state
  };

  template<>
    struct is_error_code_enum<future_errc> : public true_type { };

  /// Points to a statically-allocated object derived from error_category.
  extern const error_category* const future_category;

  // TODO: requires constexpr
  inline error_code make_error_code(future_errc __errc)
  { return error_code(static_cast<int>(__errc), *future_category); }

  // TODO: requires constexpr
  inline error_condition make_error_condition(future_errc __errc)
  { return error_condition(static_cast<int>(__errc), *future_category); }

  /**
   *  @brief Exception type thrown by futures.
   *  @ingroup exceptions
   */
  class future_error : public logic_error
  {
    error_code                  _M_code;

  public:
    explicit future_error(error_code __ec)
    : logic_error("std::future_error"), _M_code(__ec)
    { }

    virtual ~future_error() throw();

    virtual const char* 
    what() const throw();

    const error_code& 
    code() const throw() { return _M_code; }
  };

  // Forward declarations.
  template<typename _Res>
    class future;

  template<typename _Res>
    class shared_future;

  template<typename _Res>
    class atomic_future;

  template<typename _Signature> 
    class packaged_task;

  template<typename _Res>
    class promise;

  enum class launch { any, async, sync };

  template<typename _Fn, typename... _Args>
    future<typename result_of<_Fn(_Args...)>::type>
    async(launch __policy, _Fn&& __fn, _Args&&... __args);

  template<typename _Fn, typename... _Args>
    typename
    enable_if<!is_same<typename decay<_Fn>::type, launch>::value,
              future<decltype(std::declval<_Fn>()(std::declval<_Args>()...))>
             >::type
    async(_Fn&& __fn, _Args&&... __args);

#if defined(_GLIBCXX_HAS_GTHREADS) && defined(_GLIBCXX_USE_C99_STDINT_TR1) \
  && defined(_GLIBCXX_ATOMIC_BUILTINS_4)

  /// Base class and enclosing scope.
  struct __future_base
  {
    /// Base class for results.
    struct _Result_base
    {
      exception_ptr             _M_error;

      _Result_base() = default;
      _Result_base(const _Result_base&) = delete;
      _Result_base& operator=(const _Result_base&) = delete;

      // _M_destroy() allows derived classes to control deallocation
      virtual void _M_destroy() = 0;

      struct _Deleter
      {
        void operator()(_Result_base* __fr) const { __fr->_M_destroy(); }
      };

    protected:
      ~_Result_base();
    };

    /// Result.
    template<typename _Res>
      struct _Result : _Result_base
      {
      private:
        typedef alignment_of<_Res>                              __a_of;
        typedef aligned_storage<sizeof(_Res), __a_of::value>    __align_storage;
        typedef typename __align_storage::type                  __align_type;

        __align_type            _M_storage;
        bool                    _M_initialized;

      public:
        _Result() : _M_initialized() { }
        
        ~_Result()
        {
          if (_M_initialized)
            _M_value().~_Res();
        }

        // Return lvalue, future will add const or rvalue-reference
        _Res& 
        _M_value() { return *static_cast<_Res*>(_M_addr()); }

        void
        _M_set(const _Res& __res)
        {
          ::new (_M_addr()) _Res(__res);
          _M_initialized = true;
        }

        void
        _M_set(_Res&& __res)
        {
          ::new (_M_addr()) _Res(std::move(__res));
          _M_initialized = true;
        }

      private:
        void _M_destroy() { delete this; }

        void* _M_addr() { return static_cast<void*>(&_M_storage); }
    };

    // TODO: use template alias when available
    /*
      template<typename _Res>
      using _Ptr = unique_ptr<_Res, _Result_base::_Deleter>;
    */
    /// A unique_ptr based on the instantiating type.
    template<typename _Res>
      struct _Ptr
      {
        typedef unique_ptr<_Res, _Result_base::_Deleter> type;
      };

    // TODO: use when allocator_arg_t available
    /*
    /// Result_alloc.
    template<typename _Res, typename _Alloc>
      struct _Result_alloc : _Result<_Res>
      {
        typedef typename _Alloc::template rebind<_Result_alloc>::other
          __allocator_type;

        explicit
        _Result_alloc(const _Alloc& __a) : _Result<_Res>(), _M_alloc(__a)
        { }
        
      private:
        void _M_destroy()
        {
          __allocator_type __a(_M_alloc);
          __a.destroy(this);
          __a.deallocate(this, 1);
        }

        __allocator_type _M_alloc;
    };

    template<typename _Res, typename _Allocator>
      static typename _Ptr<_Result_alloc<_Res, _Allocator>>::type
      _S_allocate_result(const _Allocator& __a)
      {
        typedef _Result_alloc<_Res, _Allocator> __result_type;
        typename __result_type::__allocator_type __a2(__a);
        __result_type* __p = __a2.allocate(1);
        __try
        {
          __a2.construct(__p, __a);
        }
        __catch(...)
        {
          __a2.deallocate(__p, 1);
          __throw_exception_again;
        }
        return typename _Ptr<__result_type>::type(__p);
      }
    */


    /// Shared state between a promise and one or more associated futures.
    class _State
    {
      typedef _Ptr<_Result_base>::type _Ptr_type;

      _Ptr_type                 _M_result;
      mutex                     _M_mutex;
      condition_variable        _M_cond;
      atomic_flag               _M_retrieved;
      once_flag                 _M_once;

    public:
      _State() : _M_result(), _M_retrieved(ATOMIC_FLAG_INIT) { }

      _State(const _State&) = delete;
      _State& operator=(const _State&) = delete;

      _Result_base&
      wait()
      {
        _M_run_deferred();
        unique_lock<mutex> __lock(_M_mutex);
        if (!_M_ready())
          _M_cond.wait(__lock, std::bind<bool>(&_State::_M_ready, this));
        return *_M_result;
      }

      template<typename _Rep, typename _Period>
        bool
        wait_for(const chrono::duration<_Rep, _Period>& __rel)
        {
          unique_lock<mutex> __lock(_M_mutex);
          auto __bound = std::bind<bool>(&_State::_M_ready, this);
          return _M_ready() || _M_cond.wait_for(__lock, __rel, __bound);
        }

      template<typename _Clock, typename _Duration>
        bool
        wait_until(const chrono::time_point<_Clock, _Duration>& __abs)
        {
          unique_lock<mutex> __lock(_M_mutex);
          auto __bound = std::bind<bool>(&_State::_M_ready, this);
          return _M_ready() || _M_cond.wait_until(__lock, __abs, __bound);
        }

      void
      _M_set_result(function<_Ptr_type()> __res, bool __ignore_failure = false)
      {
        bool __set = __ignore_failure;
        // all calls to this function are serialized,
        // side-effects of invoking __res only happen once
        call_once(_M_once, mem_fn(&_State::_M_do_set), this, ref(__res),
            ref(__set));
        if (!__set)
          __throw_future_error(int(future_errc::promise_already_satisfied));
      }

      void
      _M_break_promise(_Ptr_type __res)
      {
        if (static_cast<bool>(__res))
          {
            error_code __ec(make_error_code(future_errc::broken_promise));
            __res->_M_error = copy_exception(future_error(__ec));
            {
              lock_guard<mutex> __lock(_M_mutex);
              _M_result.swap(__res);
            }
            _M_cond.notify_all();
          }
      }

      // Called when this object is passed to a future.
      void
      _M_set_retrieved_flag()
      {
        if (_M_retrieved.test_and_set())
          __throw_future_error(int(future_errc::future_already_retrieved));
      }

      template<typename _Res, typename _Arg>
        struct _Setter;

      // set lvalues
      template<typename _Res, typename _Arg>
        struct _Setter<_Res, _Arg&>
        {
          // check this is only used by promise<R>::set_value(const R&)
          // or promise<R>::set_value(R&)
          static_assert(is_same<_Res, _Arg&>::value  // promise<R&>
              || is_same<const _Res, _Arg>::value,  // promise<R>
              "Invalid specialisation");

          typename promise<_Res>::_Ptr_type operator()()
          {
            _State::_S_check(_M_promise->_M_future);
            _M_promise->_M_storage->_M_set(_M_arg);
            return std::move(_M_promise->_M_storage);
          }
          promise<_Res>*    _M_promise;
          _Arg&             _M_arg;
        };

      // set rvalues
      template<typename _Res>
        struct _Setter<_Res, _Res&&>
        {
          typename promise<_Res>::_Ptr_type operator()()
          {
            _State::_S_check(_M_promise->_M_future);
            _M_promise->_M_storage->_M_set(std::move(_M_arg));
            return std::move(_M_promise->_M_storage);
          }
          promise<_Res>*    _M_promise;
          _Res&             _M_arg;
        };

      struct __exception_ptr_tag { };

      // set exceptions
      template<typename _Res>
        struct _Setter<_Res, __exception_ptr_tag>
        {
          typename promise<_Res>::_Ptr_type operator()()
          {
            _State::_S_check(_M_promise->_M_future);
            _M_promise->_M_storage->_M_error = _M_ex;
            return std::move(_M_promise->_M_storage);
          }

          promise<_Res>*   _M_promise;
          exception_ptr&    _M_ex;
        };

      template<typename _Res, typename _Arg>
        static _Setter<_Res, _Arg&&>
        __setter(promise<_Res>* __prom, _Arg&& __arg)
        {
          return _Setter<_Res, _Arg&&>{ __prom, __arg };
        }

      template<typename _Res>
        static _Setter<_Res, __exception_ptr_tag>
        __setter(exception_ptr& __ex, promise<_Res>* __prom)
        {
          return _Setter<_Res, __exception_ptr_tag>{ __prom, __ex };
        }

      static _Setter<void, void>
      __setter(promise<void>* __prom);

      template<typename _Tp>
        static bool
        _S_check(const shared_ptr<_Tp>& __p)
        {
          if (!static_cast<bool>(__p))
            __throw_future_error((int)future_errc::no_state);
        }

    private:
      void
      _M_do_set(function<_Ptr_type()>& __f, bool& __set)
      {
        _Ptr_type __res = __f();
        {
          lock_guard<mutex> __lock(_M_mutex);
          _M_result.swap(__res);
        }
        _M_cond.notify_all();
        __set = true;
      }

      bool _M_ready() const { return static_cast<bool>(_M_result); }

      virtual void _M_run_deferred() { }
    };

    template<typename _Res>
      class _Deferred_state;

    template<typename _Res>
      class _Async_state;

    template<typename _Signature>
      class _Task_state;

    template<typename _StateT, typename _Res = typename _StateT::_Res_type>
      struct _Task_setter;
  };

  inline __future_base::_Result_base::~_Result_base() = default;

  /// Partial specialization for reference types.
  template<typename _Res>
    struct __future_base::_Result<_Res&> : __future_base::_Result_base
    {
      _Result() : _M_value_ptr() { }

      void _M_set(_Res& __res) { _M_value_ptr = &__res; }

      _Res& _M_get() { return *_M_value_ptr; }

    private:
      _Res*                     _M_value_ptr;
      
      void _M_destroy() { delete this; }
    };

  /// Explicit specialization for void.
  template<>
    struct __future_base::_Result<void> : __future_base::_Result_base
    {
    private:
      void _M_destroy() { delete this; }
    };


  /// Common implementation for future and shared_future.
  template<typename _Res>
    class __basic_future : public __future_base
    {
    protected:
      typedef shared_ptr<_State>                __state_type;
      typedef __future_base::_Result<_Res>&     __result_type;

    private:
      __state_type              _M_state;

    public:
      // Disable copying.
      __basic_future(const __basic_future&) = delete;
      __basic_future& operator=(const __basic_future&) = delete;

      bool 
      valid() const { return static_cast<bool>(_M_state); }

      void 
      wait() const { _M_state->wait(); }

      template<typename _Rep, typename _Period>
        bool
        wait_for(const chrono::duration<_Rep, _Period>& __rel) const
        { return _M_state->wait_for(__rel); }

      template<typename _Clock, typename _Duration>
        bool
        wait_until(const chrono::time_point<_Clock, _Duration>& __abs) const
        { return _M_state->wait_until(__abs); }

    protected:
      /// Wait for the state to be ready and rethrow any stored exception
      __result_type
      _M_get_result()
      {
        _Result_base& __res = _M_state->wait();
        if (!(__res._M_error == 0))
          rethrow_exception(__res._M_error);
        return static_cast<__result_type>(__res);
      }

      void _M_swap(__basic_future& __that)
      {
        _M_state.swap(__that._M_state);
      }

      // Construction of a future by promise::get_future()
      explicit
      __basic_future(const __state_type& __state) : _M_state(__state)
      {
        _State::_S_check(_M_state);
        _M_state->_M_set_retrieved_flag();
      }

      // Copy construction from a shared_future
      explicit
      __basic_future(const shared_future<_Res>&);

      // Move construction from a shared_future
      explicit
      __basic_future(shared_future<_Res>&&);

      // Move construction from a future
      explicit
      __basic_future(future<_Res>&&);

      __basic_future() { }

      struct _Reset
      {
        explicit _Reset(__basic_future& __fut) : _M_fut(__fut) { }
        ~_Reset() { _M_fut._M_state.reset(); }
        __basic_future& _M_fut;
      };
    };


  /// Primary template for future.
  template<typename _Res>
    class future : public __basic_future<_Res>
    {
      friend class promise<_Res>;
      template<typename> friend class packaged_task;
      template<typename _Fn, typename... _Args>
        friend future<typename result_of<_Fn(_Args...)>::type>
        async(launch, _Fn&&, _Args&&...);

      typedef __basic_future<_Res> _Base_type;
      typedef typename _Base_type::__state_type __state_type;

      explicit
      future(const __state_type& __state) : _Base_type(__state) { }

    public:
      future() : _Base_type() { }

      /// Move constructor
      future(future&& __uf) : _Base_type(std::move(__uf)) { }

      // Disable copying
      future(const future&) = delete;
      future& operator=(const future&) = delete;

      future& operator=(future&& __fut)
      {
        future(std::move(__fut))._M_swap(*this);
        return *this;
      }

      /// Retrieving the value
      _Res
      get()
      {
        typename _Base_type::_Reset __reset(*this);
        return std::move(this->_M_get_result()._M_value());
      }
    };
 
  /// Partial specialization for future<R&>
  template<typename _Res>
    class future<_Res&> : public __basic_future<_Res&>
    {
      friend class promise<_Res&>;
      template<typename> friend class packaged_task;
      template<typename _Fn, typename... _Args>
        friend future<typename result_of<_Fn(_Args...)>::type>
        async(launch, _Fn&&, _Args&&...);

      typedef __basic_future<_Res&> _Base_type;
      typedef typename _Base_type::__state_type __state_type;

      explicit
      future(const __state_type& __state) : _Base_type(__state) { }

    public:
      future() : _Base_type() { }

      /// Move constructor
      future(future&& __uf) : _Base_type(std::move(__uf)) { }

      // Disable copying
      future(const future&) = delete;
      future& operator=(const future&) = delete;

      future& operator=(future&& __fut)
      {
        future(std::move(__fut))._M_swap(*this);
        return *this;
      }

      /// Retrieving the value
      _Res& 
      get()
      {
        typename _Base_type::_Reset __reset(*this);
        return this->_M_get_result()._M_get();
      }
    };

  /// Explicit specialization for future<void>
  template<>
    class future<void> : public __basic_future<void>
    {
      friend class promise<void>;
      template<typename> friend class packaged_task;
      template<typename _Fn, typename... _Args>
        friend future<typename result_of<_Fn(_Args...)>::type>
        async(launch, _Fn&&, _Args&&...);

      typedef __basic_future<void> _Base_type;
      typedef typename _Base_type::__state_type __state_type;

      explicit
      future(const __state_type& __state) : _Base_type(__state) { }

    public:
      future() : _Base_type() { }

      /// Move constructor
      future(future&& __uf) : _Base_type(std::move(__uf)) { }

      // Disable copying
      future(const future&) = delete;
      future& operator=(const future&) = delete;

      future& operator=(future&& __fut)
      {
        future(std::move(__fut))._M_swap(*this);
        return *this;
      }

      /// Retrieving the value
      void 
      get()
      {
        typename _Base_type::_Reset __reset(*this);
        this->_M_get_result();
      }
    };


  /// Primary template for shared_future.
  template<typename _Res>
    class shared_future : public __basic_future<_Res>
    {
      typedef __basic_future<_Res> _Base_type;

    public:
      shared_future() : _Base_type() { }

      /// Copy constructor
      shared_future(const shared_future& __sf) : _Base_type(__sf) { }

      /// Construct from a future rvalue
      shared_future(future<_Res>&& __uf)
      : _Base_type(std::move(__uf))
      { }

      /// Construct from a shared_future rvalue
      shared_future(shared_future&& __sf)
      : _Base_type(std::move(__sf))
      { }

      shared_future& operator=(const shared_future& __sf)
      {
        shared_future(__sf)._M_swap(*this);
        return *this;
      }

      shared_future& operator=(shared_future&& __sf)
      {
        shared_future(std::move(__sf))._M_swap(*this);
        return *this;
      }

      /// Retrieving the value
      const _Res&
      get()
      {
        typename _Base_type::__result_type __r = this->_M_get_result();
        _Res& __rs(__r._M_value());
        return __rs;
      }
    };
 
  /// Partial specialization for shared_future<R&>
  template<typename _Res>
    class shared_future<_Res&> : public __basic_future<_Res&>
    {
      typedef __basic_future<_Res&>           _Base_type;

    public:
      shared_future() : _Base_type() { }

      /// Copy constructor
      shared_future(const shared_future& __sf) : _Base_type(__sf) { }

      /// Construct from a future rvalue
      shared_future(future<_Res&>&& __uf)
      : _Base_type(std::move(__uf))
      { }

      /// Construct from a shared_future rvalue
      shared_future(shared_future&& __sf)
      : _Base_type(std::move(__sf))
      { }

      shared_future& operator=(const shared_future& __sf)
      {
        shared_future(__sf)._M_swap(*this);
        return *this;
      }

      shared_future& operator=(shared_future&& __sf)
      {
        shared_future(std::move(__sf))._M_swap(*this);
        return *this;
      }

      /// Retrieving the value
      _Res& 
      get() { return this->_M_get_result()._M_get(); }
    };

  /// Explicit specialization for shared_future<void>
  template<>
    class shared_future<void> : public __basic_future<void>
    {
      typedef __basic_future<void> _Base_type;

    public:
      shared_future() : _Base_type() { }

      /// Copy constructor
      shared_future(const shared_future& __sf) : _Base_type(__sf) { }

      /// Construct from a future rvalue
      shared_future(future<void>&& __uf)
      : _Base_type(std::move(__uf))
      { }

      /// Construct from a shared_future rvalue
      shared_future(shared_future&& __sf)
      : _Base_type(std::move(__sf))
      { }

      shared_future& operator=(const shared_future& __sf)
      {
        shared_future(__sf)._M_swap(*this);
        return *this;
      }

      shared_future& operator=(shared_future&& __sf)
      {
        shared_future(std::move(__sf))._M_swap(*this);
        return *this;
      }

      // Retrieving the value
      void 
      get() { this->_M_get_result(); }
    };

  // Now we can define the protected __basic_future constructors.
  template<typename _Res>
    inline __basic_future<_Res>::
    __basic_future(const shared_future<_Res>& __sf)
    : _M_state(__sf._M_state)
    { }

  template<typename _Res>
    inline __basic_future<_Res>::
    __basic_future(shared_future<_Res>&& __sf)
    : _M_state(std::move(__sf._M_state))
    { }

  template<typename _Res>
    inline __basic_future<_Res>::
    __basic_future(future<_Res>&& __uf)
    : _M_state(std::move(__uf._M_state))
    { }


  /// Primary template for promise
  template<typename _Res>
    class promise
    {
      typedef __future_base::_State             _State;
      typedef __future_base::_Result<_Res>      _Res_type;
      typedef typename __future_base::_Ptr<_Res_type>::type _Ptr_type;
      template<typename, typename> friend class _State::_Setter;
      
      shared_ptr<_State>                        _M_future;
      _Ptr_type                                 _M_storage;

    public:
      promise()
      : _M_future(std::make_shared<_State>()),
        _M_storage(new _Res_type())
      { }

      promise(promise&& __rhs)
      : _M_future(std::move(__rhs._M_future)),
        _M_storage(std::move(__rhs._M_storage))
      { }

      // TODO: needs allocator_arg_t
      /*
      template<typename _Allocator>
        promise(allocator_arg_t, const _Allocator& __a)
        : _M_future(std::allocate_shared<_State>(__a)),
        _M_storage(__future_base::_S_allocate_result<_Res>(__a))
        { }
      */

      promise(const promise&) = delete;

      ~promise()
      {
        if (static_cast<bool>(_M_future) && !_M_future.unique())
          _M_future->_M_break_promise(std::move(_M_storage));
      }

      // Assignment
      promise&
      operator=(promise&& __rhs)
      {
        promise(std::move(__rhs)).swap(*this);
        return *this;
      }

      promise& operator=(const promise&) = delete;

      void
      swap(promise& __rhs)
      {
        _M_future.swap(__rhs._M_future);
        _M_storage.swap(__rhs._M_storage);
      }

      // Retrieving the result
      future<_Res>
      get_future()
      { return future<_Res>(_M_future); }

      // Setting the result
      void
      set_value(const _Res& __r)
      {
        auto __setter = _State::__setter(this, __r);
        _M_future->_M_set_result(std::move(__setter));
      }

      void
      set_value(_Res&& __r)
      {
        auto __setter = _State::__setter(this, std::move(__r));
        _M_future->_M_set_result(std::move(__setter));
      }

      void
      set_exception(exception_ptr __p)
      {
        auto __setter = _State::__setter(__p, this);
        _M_future->_M_set_result(std::move(__setter));
      }
    };

  template<typename _Res>
    inline void
    swap(promise<_Res>& __x, promise<_Res>& __y)
    { __x.swap(__y); }

  /// Partial specialization for promise<R&>
  template<typename _Res>
    class promise<_Res&>
    {
      typedef __future_base::_State             _State;
      typedef __future_base::_Result<_Res&>     _Res_type;
      typedef typename __future_base::_Ptr<_Res_type>::type _Ptr_type;
      template<typename, typename> friend class _State::_Setter;

      shared_ptr<_State>                        _M_future;
      _Ptr_type                                 _M_storage;

    public:
      promise()
      : _M_future(std::make_shared<_State>()),
        _M_storage(new _Res_type())
      { }

      promise(promise&& __rhs)
      : _M_future(std::move(__rhs._M_future)), 
        _M_storage(std::move(__rhs._M_storage))
      { }

      // TODO: needs allocator_arg_t
      /*
      template<typename _Allocator>
        promise(allocator_arg_t, const _Allocator& __a)
        : _M_future(std::allocate_shared<_State>(__a)),
        _M_storage(__future_base::_S_allocate_result<_Res&>(__a))
        { }
      */

      promise(const promise&) = delete;

      ~promise()
      {
        if (static_cast<bool>(_M_future) && !_M_future.unique())
          _M_future->_M_break_promise(std::move(_M_storage));
      }

      // Assignment
      promise&
      operator=(promise&& __rhs)
      {
        promise(std::move(__rhs)).swap(*this);
        return *this;
      }

      promise& operator=(const promise&) = delete;

      void
      swap(promise& __rhs)
      {
        _M_future.swap(__rhs._M_future);
        _M_storage.swap(__rhs._M_storage);
      }

      // Retrieving the result
      future<_Res&>
      get_future()
      { return future<_Res&>(_M_future); }

      // Setting the result
      void
      set_value(_Res& __r)
      {
        auto __setter = _State::__setter(this, __r);
        _M_future->_M_set_result(std::move(__setter));
      }

      void
      set_exception(exception_ptr __p)
      {
        auto __setter = _State::__setter(__p, this);
        _M_future->_M_set_result(std::move(__setter));
      }
    };

  /// Explicit specialization for promise<void>
  template<>
    class promise<void>
    {
      typedef __future_base::_State             _State;
      typedef __future_base::_Result<void>      _Res_type;
      typedef typename __future_base::_Ptr<_Res_type>::type _Ptr_type;
      template<typename, typename> friend class _State::_Setter;

      shared_ptr<_State>                        _M_future;
      _Ptr_type                                 _M_storage;

    public:
      promise()
      : _M_future(std::make_shared<_State>()),
        _M_storage(new _Res_type())
      { }

      promise(promise&& __rhs)
      : _M_future(std::move(__rhs._M_future)),
        _M_storage(std::move(__rhs._M_storage))
      { }


      // TODO: needs allocator_arg_t
      /*
      template<typename _Allocator>
        promise(allocator_arg_t, const _Allocator& __a)
        : _M_future(std::allocate_shared<_State>(__a)),
        _M_storage(__future_base::_S_allocate_result<void>(__a))
        { }
      */

      promise(const promise&) = delete;

      ~promise()
      {
        if (static_cast<bool>(_M_future) && !_M_future.unique())
          _M_future->_M_break_promise(std::move(_M_storage));
      }

      // Assignment
      promise&
      operator=(promise&& __rhs)
      {
        promise(std::move(__rhs)).swap(*this);
        return *this;
      }

      promise& operator=(const promise&) = delete;

      void
      swap(promise& __rhs)
      {
        _M_future.swap(__rhs._M_future);
        _M_storage.swap(__rhs._M_storage);
      }

      // Retrieving the result
      future<void>
      get_future()
      { return future<void>(_M_future); }

      // Setting the result
      void set_value();

      void
      set_exception(exception_ptr __p)
      {
        auto __setter = _State::__setter(__p, this);
        _M_future->_M_set_result(std::move(__setter));
      }
    };

  // set void
  template<>
    struct __future_base::_State::_Setter<void, void>
    {
      promise<void>::_Ptr_type operator()()
      {
        _State::_S_check(_M_promise->_M_future);
        return std::move(_M_promise->_M_storage);
      }

      promise<void>*    _M_promise;
    };

  inline __future_base::_State::_Setter<void, void>
  __future_base::_State::__setter(promise<void>* __prom)
  {
    return _Setter<void, void>{ __prom };
  }

  inline void
  promise<void>::set_value()
  {
    auto __setter = _State::__setter(this);
    _M_future->_M_set_result(std::move(__setter));
  }

  // TODO: needs allocators
  /*
  template<typename _Res, class Alloc>
    struct uses_allocator<promise<_Res>, Alloc> : true_type  { };
  */


  template<typename _StateT, typename _Res>
    struct __future_base::_Task_setter
    {
      typename _StateT::_Ptr_type operator()()
      {
        __try
          {
            _M_state->_M_result->_M_set(_M_fn());
          }
        __catch(...)
          {
            _M_state->_M_result->_M_error = current_exception();
          }
        return std::move(_M_state->_M_result);
      }
      _StateT*                  _M_state;
      std::function<_Res()>     _M_fn;
    };

  template<typename _StateT>
    struct __future_base::_Task_setter<_StateT, void>
    {
      typename _StateT::_Ptr_type operator()()
      {
        __try
          {
            _M_fn();
          }
        __catch(...)
          {
            _M_state->_M_result->_M_error = current_exception();
          }
        return std::move(_M_state->_M_result);
      }
      _StateT*                  _M_state;
      std::function<void()>     _M_fn;
    };

  template<typename _Res, typename... _Args>
    struct __future_base::_Task_state<_Res(_Args...)> : __future_base::_State
    {
      typedef _Res _Res_type;

      _Task_state(std::function<_Res(_Args...)> __task)
      : _M_result(new _Result<_Res>()), _M_task(std::move(__task))
      { }

      // TODO: needs allocator_arg_t
      /*
      template<typename _Func, typename _Alloc>
        _Task_state(_Func&& __task, const _Alloc& __a)
        : _M_result(_S_allocate_result<_Res>(__a))
        , _M_task(allocator_arg, __a, std::move(__task))
        { }
      */

      void
      _M_run(_Args... __args)
      {
        // bound arguments decay so wrap lvalue references
        auto __bound = std::bind<_Res>(_M_task,
            _S_maybe_wrap_ref(std::forward<_Args>(__args))...);
        _Task_setter<_Task_state> __setter{ this, std::move(__bound) };
        _M_set_result(std::move(__setter));
      }

      template<typename, typename> friend class _Task_setter;
      typedef typename __future_base::_Ptr<_Result<_Res>>::type _Ptr_type;
      _Ptr_type _M_result;
      std::function<_Res(_Args...)> _M_task;

      template<typename _Tp>
        static reference_wrapper<_Tp>
        _S_maybe_wrap_ref(_Tp& __t)
        { return std::ref(__t); }

      template<typename _Tp>
        static typename enable_if<!is_lvalue_reference<_Tp>::value,
                        _Tp>::type&&
        _S_maybe_wrap_ref(_Tp&& __t)
        { return std::forward<_Tp>(__t); }
    };

  /// packaged_task
  template<typename _Res, typename... _ArgTypes>
    class packaged_task<_Res(_ArgTypes...)>
    {
      typedef __future_base::_Task_state<_Res(_ArgTypes...)>  _State_type;
      shared_ptr<_State_type>                   _M_state;

    public:
      typedef _Res result_type;

      // Construction and destruction
      packaged_task() { }

      template<typename _Fn>
        explicit
        packaged_task(const _Fn& __fn)
        : _M_state(std::make_shared<_State_type>(__fn))
        { }

      template<typename _Fn>
        explicit
        packaged_task(_Fn&& __fn)
        : _M_state(std::make_shared<_State_type>(std::move(__fn)))
        { }

      explicit
      packaged_task(_Res(*__fn)(_ArgTypes...))
      : _M_state(std::make_shared<_State_type>(__fn))
      { }

      // TODO: needs allocator_arg_t
      /*
      template<typename _Fn, typename _Allocator>
        explicit
        packaged_task(allocator_arg_t __tag, const _Allocator& __a, _Fn __fn)
        : _M_state(std::allocate_shared<_State_type>(__a, std::move(__fn)))
        { }
      */

      ~packaged_task()
      {
        if (static_cast<bool>(_M_state) && !_M_state.unique())
          _M_state->_M_break_promise(std::move(_M_state->_M_result));
      }

      // No copy
      packaged_task(packaged_task&) = delete;
      packaged_task& operator=(packaged_task&) = delete;

      // Move support
      packaged_task(packaged_task&& __other)
      { this->swap(__other); }

      packaged_task& operator=(packaged_task&& __other)
      {
        packaged_task(std::move(__other)).swap(*this);
        return *this;
      }

      void
      swap(packaged_task& __other)
      { _M_state.swap(__other._M_state); }

      explicit operator bool() const { return static_cast<bool>(_M_state); }

      // Result retrieval
      future<_Res>
      get_future()
      { return future<_Res>(_M_state); }

      // Execution
      void
      operator()(_ArgTypes... __args)
      {
        __future_base::_State::_S_check(_M_state);
        _M_state->_M_run(std::forward<_ArgTypes>(__args)...);
      }

      void
      reset()
      {
        __future_base::_State::_S_check(_M_state);
        packaged_task(std::move(_M_state->_M_task)).swap(*this);
      }
    };

  template<typename _Res, typename... _ArgTypes>
    inline void
    swap(packaged_task<_Res(_ArgTypes...)>& __x,
         packaged_task<_Res(_ArgTypes...)>& __y)
    { __x.swap(__y); }
 
  template<typename _Res>
    class __future_base::_Deferred_state : public __future_base::_State
    {
    public:
      typedef _Res _Res_type;

      explicit
      _Deferred_state(std::function<_Res()>&& __fn)
      : _M_result(new _Result<_Res>()), _M_fn(std::move(__fn))
      { }

    private:
      template<typename, typename> friend class _Task_setter;
      typedef typename __future_base::_Ptr<_Result<_Res>>::type _Ptr_type;
      _Ptr_type _M_result;
      std::function<_Res()> _M_fn;

      virtual void
      _M_run_deferred()
      {
        _Task_setter<_Deferred_state> __setter{ this, _M_fn };
        // safe to call multiple times so ignore failure
        _M_set_result(std::move(__setter), true);
      }
    };

  template<typename _Res>
    class __future_base::_Async_state : public __future_base::_State
    {
    public:
      typedef _Res _Res_type;

      explicit 
      _Async_state(std::function<_Res()>&& __fn)
      : _M_result(new _Result<_Res>()), _M_fn(std::move(__fn)),
        _M_thread(mem_fn(&_Async_state::_M_do_run), this)
      { }

      ~_Async_state() { _M_thread.join(); }

    private:
      void _M_do_run()
      {
        _Task_setter<_Async_state> __setter{ this, std::move(_M_fn) };
        _M_set_result(std::move(__setter));
      }

      template<typename, typename> friend class _Task_setter;
      typedef typename __future_base::_Ptr<_Result<_Res>>::type _Ptr_type;
      _Ptr_type _M_result;
      std::function<_Res()> _M_fn;
      thread _M_thread;
    };

  template<typename _Fn, typename... _Args>
    future<typename result_of<_Fn(_Args...)>::type>
    async(launch __policy, _Fn&& __fn, _Args&&... __args)
    {
      typedef typename result_of<_Fn(_Args...)>::type result_type;
      std::shared_ptr<__future_base::_State> __state;
      if (__policy == launch::async)
        {
          typedef typename __future_base::_Async_state<result_type> _State;
          __state = std::make_shared<_State>(std::bind<result_type>(
              std::forward<_Fn>(__fn), std::forward<_Args>(__args)...));
        }
      else
        {
          typedef typename __future_base::_Deferred_state<result_type> _State;
          __state = std::make_shared<_State>(std::bind<result_type>(
              std::forward<_Fn>(__fn), std::forward<_Args>(__args)...));
        }
      return future<result_type>(__state);
    }

  template<typename _Fn, typename... _Args>
    inline typename
    enable_if<!is_same<typename decay<_Fn>::type, launch>::value,
              future<decltype(std::declval<_Fn>()(std::declval<_Args>()...))>
             >::type
    async(_Fn&& __fn, _Args&&... __args)
    {
      return async(launch::any, std::forward<_Fn>(__fn),
                   std::forward<_Args>(__args)...);
    }

#endif // _GLIBCXX_HAS_GTHREADS && _GLIBCXX_USE_C99_STDINT_TR1
       // && _GLIBCXX_ATOMIC_BUILTINS_4

  // @} group futures
}

#endif // __GXX_EXPERIMENTAL_CXX0X__

#endif // _GLIBCXX_FUTURE

Go to most recent revision | Compare with Previous | Blame | View Log

powered by: WebSVN 2.1.0

© copyright 1999-2024 OpenCores.org, equivalent to Oliscience, all rights reserved. OpenCores®, registered trademark.