Chris@16: /*
Chris@101:  * Copyright (c) 2012-2014 Glen Joseph Fernandes
Chris@16:  * glenfe at live dot com
Chris@16:  *
Chris@16:  * Distributed under the Boost Software License,
Chris@16:  * Version 1.0. (See accompanying file LICENSE_1_0.txt
Chris@16:  * or copy at http://boost.org/LICENSE_1_0.txt)
Chris@16:  */
Chris@16: #ifndef BOOST_SMART_PTR_ALLOCATE_SHARED_ARRAY_HPP
Chris@16: #define BOOST_SMART_PTR_ALLOCATE_SHARED_ARRAY_HPP
Chris@16: 
Chris@101: #include <boost/smart_ptr/detail/array_count_impl.hpp>
Chris@16: #include <boost/smart_ptr/detail/sp_if_array.hpp>
Chris@16: 
Chris@16: namespace boost {
Chris@101:     template<class T, class A>
Chris@16:     inline typename boost::detail::sp_if_array<T>::type
Chris@16:     allocate_shared(const A& allocator, std::size_t size) {
Chris@16:         typedef typename boost::detail::array_inner<T>::type T1;
Chris@16:         typedef typename boost::detail::array_base<T1>::type T2;
Chris@101:         typedef boost::detail::ms_init_tag R1;
Chris@101:         typedef boost::detail::as_allocator<A, T, R1> A1;
Chris@101:         typedef boost::detail::ms_in_allocator_tag D1;
Chris@101:         std::size_t n1 = size * boost::detail::array_total<T1>::size;
Chris@16:         T1* p1 = 0;
Chris@16:         T2* p2 = 0;
Chris@101:         D1 d1;
Chris@101:         A1 a1(allocator, size, &p2);
Chris@101:         shared_ptr<T> s1(p1, d1, a1);
Chris@101:         A1* a2 = static_cast<A1*>(s1._internal_get_untyped_deleter());
Chris@101:         a2->set(0);
Chris@101: #if !defined(BOOST_NO_CXX11_ALLOCATOR)
Chris@101:         boost::detail::as_init(allocator, p2, n1);
Chris@101: #else
Chris@101:         boost::detail::ms_init(p2, n1);
Chris@101: #endif
Chris@101:         a2->set(p2);
Chris@16:         p1 = reinterpret_cast<T1*>(p2);
Chris@101:         return shared_ptr<T>(s1, p1);
Chris@16:     }
Chris@101: 
Chris@101:     template<class T, class A>
Chris@101:     inline typename boost::detail::sp_if_size_array<T>::type
Chris@101:     allocate_shared(const A& allocator) {
Chris@16:         typedef typename boost::detail::array_inner<T>::type T1;
Chris@16:         typedef typename boost::detail::array_base<T1>::type T2;
Chris@101:         typedef boost::detail::ms_init_tag R1;
Chris@101:         typedef boost::detail::as_allocator<A, T, R1> A1;
Chris@101:         typedef boost::detail::ms_in_allocator_tag D1;
Chris@16:         enum {
Chris@16:             N = boost::detail::array_total<T>::size
Chris@16:         };
Chris@16:         T1* p1 = 0;
Chris@16:         T2* p2 = 0;
Chris@101:         D1 d1;
Chris@101:         A1 a1(allocator, &p2);
Chris@101:         shared_ptr<T> s1(p1, d1, a1);
Chris@101:         A1* a2 = static_cast<A1*>(s1._internal_get_untyped_deleter());
Chris@101:         a2->set(0);
Chris@101: #if !defined(BOOST_NO_CXX11_ALLOCATOR)
Chris@101:         boost::detail::as_init(allocator, p2, N);
Chris@101: #else
Chris@101:         boost::detail::ms_init(p2, N);
Chris@101: #endif
Chris@101:         a2->set(p2);
Chris@16:         p1 = reinterpret_cast<T1*>(p2);
Chris@101:         return shared_ptr<T>(s1, p1);
Chris@16:     }
Chris@101: 
Chris@101:     template<class T, class A>
Chris@101:     inline typename boost::detail::sp_if_array<T>::type
Chris@101:     allocate_shared(const A& allocator, std::size_t size,
Chris@101:         const typename boost::detail::array_inner<T>::type& value) {
Chris@16:         typedef typename boost::detail::array_inner<T>::type T1;
Chris@16:         typedef typename boost::detail::array_base<T1>::type T2;
Chris@16:         typedef const T2 T3;
Chris@101:         typedef boost::detail::ms_init_tag R1;
Chris@101:         typedef boost::detail::as_allocator<A, T, R1> A1;
Chris@101:         typedef boost::detail::ms_in_allocator_tag D1;
Chris@101:         enum {
Chris@101:             M = boost::detail::array_total<T1>::size
Chris@101:         };
Chris@101:         std::size_t n1 = M * size;
Chris@101:         T1* p1 = 0;
Chris@101:         T2* p2 = 0;
Chris@101:         T3* p3 = reinterpret_cast<T3*>(&value);
Chris@101:         D1 d1;
Chris@101:         A1 a1(allocator, size, &p2);
Chris@101:         shared_ptr<T> s1(p1, d1, a1);
Chris@101:         A1* a2 = static_cast<A1*>(s1._internal_get_untyped_deleter());
Chris@101:         a2->set(0);
Chris@101: #if !defined(BOOST_NO_CXX11_ALLOCATOR)
Chris@101:         boost::detail::as_init<T2, A, M>(allocator, p2, n1, p3);
Chris@101: #else
Chris@101:         boost::detail::ms_init<T2, M>(p2, n1, p3);
Chris@101: #endif
Chris@101:         a2->set(p2);
Chris@101:         p1 = reinterpret_cast<T1*>(p2);
Chris@101:         return shared_ptr<T>(s1, p1);
Chris@101:     }
Chris@101: 
Chris@101:     template<class T, class A>
Chris@101:     inline typename boost::detail::sp_if_size_array<T>::type
Chris@101:     allocate_shared(const A& allocator, 
Chris@101:         const typename boost::detail::array_inner<T>::type& value) {
Chris@101:         typedef typename boost::detail::array_inner<T>::type T1;
Chris@101:         typedef typename boost::detail::array_base<T1>::type T2;
Chris@101:         typedef const T2 T3;
Chris@101:         typedef boost::detail::ms_init_tag R1;
Chris@101:         typedef boost::detail::as_allocator<A, T, R1> A1;
Chris@101:         typedef boost::detail::ms_in_allocator_tag D1;
Chris@101:         enum {
Chris@101:             N = boost::detail::array_total<T>::size,
Chris@101:             M = boost::detail::array_total<T1>::size
Chris@101:         };
Chris@101:         T1* p1 = 0;
Chris@101:         T2* p2 = 0;
Chris@101:         T3* p3 = reinterpret_cast<T3*>(&value);
Chris@101:         D1 d1;
Chris@101:         A1 a1(allocator, &p2);
Chris@101:         shared_ptr<T> s1(p1, d1, a1);
Chris@101:         A1* a2 = static_cast<A1*>(s1._internal_get_untyped_deleter());
Chris@101:         a2->set(0);
Chris@101: #if !defined(BOOST_NO_CXX11_ALLOCATOR)
Chris@101:         boost::detail::as_init<T2, A, M>(allocator, p2, N, p3);
Chris@101: #else
Chris@101:         boost::detail::ms_init<T2,  M>(p2, N, p3);
Chris@101: #endif
Chris@101:         a2->set(p2);
Chris@101:         p1 = reinterpret_cast<T1*>(p2);
Chris@101:         return shared_ptr<T>(s1, p1);
Chris@101:     }
Chris@101: 
Chris@101:     template<class T, class A>
Chris@101:     inline typename boost::detail::sp_if_array<T>::type
Chris@101:     allocate_shared_noinit(const A& allocator, std::size_t size) {
Chris@101:         typedef typename boost::detail::array_inner<T>::type T1;
Chris@101:         typedef typename boost::detail::array_base<T1>::type T2;
Chris@101:         typedef boost::detail::ms_noinit_tag R1;
Chris@101:         typedef boost::detail::as_allocator<A, T, R1> A1;
Chris@101:         typedef boost::detail::ms_in_allocator_tag D1;
Chris@101:         std::size_t n1 = size * boost::detail::array_total<T1>::size;
Chris@101:         T1* p1 = 0;
Chris@101:         T2* p2 = 0;
Chris@101:         D1 d1;
Chris@101:         A1 a1(allocator, size, &p2);
Chris@101:         shared_ptr<T> s1(p1, d1, a1);
Chris@101:         A1* a2 = static_cast<A1*>(s1._internal_get_untyped_deleter());
Chris@101:         a2->set(0);
Chris@101:         boost::detail::ms_noinit(p2, n1);
Chris@101:         a2->set(p2);
Chris@101:         p1 = reinterpret_cast<T1*>(p2);
Chris@101:         return shared_ptr<T>(s1, p1);
Chris@101:     }
Chris@101: 
Chris@101:     template<class T, class A>
Chris@101:     inline typename boost::detail::sp_if_size_array<T>::type
Chris@101:     allocate_shared_noinit(const A& allocator) {
Chris@101:         typedef typename boost::detail::array_inner<T>::type T1;
Chris@101:         typedef typename boost::detail::array_base<T1>::type T2;
Chris@101:         typedef boost::detail::ms_noinit_tag R1;
Chris@101:         typedef boost::detail::as_allocator<A, T, R1> A1;
Chris@101:         typedef boost::detail::ms_in_allocator_tag D1;
Chris@16:         enum {
Chris@16:             N = boost::detail::array_total<T>::size
Chris@16:         };
Chris@16:         T1* p1 = 0;
Chris@16:         T2* p2 = 0;
Chris@101:         D1 d1;
Chris@101:         A1 a1(allocator, &p2);
Chris@101:         shared_ptr<T> s1(p1, d1, a1);
Chris@101:         A1* a2 = static_cast<A1*>(s1._internal_get_untyped_deleter());
Chris@101:         a2->set(0);
Chris@101:         boost::detail::ms_noinit(p2, N);
Chris@101:         a2->set(p2);
Chris@16:         p1 = reinterpret_cast<T1*>(p2);
Chris@101:         return shared_ptr<T>(s1, p1);
Chris@16:     }
Chris@16: }
Chris@16: 
Chris@16: #endif