/*
(c) 2014-2015 Glen Joseph Fernandes
glenjofe at gmail dot com

Distributed under the Boost Software
License, Version 1.0.
http://boost.org/LICENSE_1_0.txt
*/
#ifndef BOOST_ALIGN_ALIGNED_ALLOCATOR_ADAPTOR_HPP
#define BOOST_ALIGN_ALIGNED_ALLOCATOR_ADAPTOR_HPP

#include <boost/config.hpp>
#include <boost/static_assert.hpp>
#include <boost/align/align.hpp>
#include <boost/align/aligned_allocator_adaptor_forward.hpp>
#include <boost/align/alignment_of.hpp>
#include <boost/align/detail/addressof.hpp>
#include <boost/align/detail/is_alignment_constant.hpp>
#include <boost/align/detail/max_align.hpp>
#include <new>

#if !defined(BOOST_NO_CXX11_ALLOCATOR)
#include <memory>
#endif

#if !defined(BOOST_NO_CXX11_RVALUE_REFERENCES)
#include <utility>
#endif

namespace boost {
namespace alignment {

template<class Allocator, std::size_t Alignment>
class aligned_allocator_adaptor
    : public Allocator {
    BOOST_STATIC_ASSERT(detail::
        is_alignment_constant<Alignment>::value);

#if !defined(BOOST_NO_CXX11_ALLOCATOR)
    typedef std::allocator_traits<Allocator> traits;

    typedef typename traits::
        template rebind_alloc<char> char_alloc;

    typedef typename traits::
        template rebind_traits<char> char_traits;

    typedef typename char_traits::pointer char_ptr;
#else
    typedef typename Allocator::
        template rebind<char>::other char_alloc;

    typedef typename char_alloc::pointer char_ptr;
#endif

    enum {
        ptr_align = alignment_of<char_ptr>::value
    };

public:
#if !defined(BOOST_NO_CXX11_ALLOCATOR)
    typedef typename traits::value_type value_type;
    typedef typename traits::size_type size_type;
#else
    typedef typename Allocator::value_type value_type;
    typedef typename Allocator::size_type size_type;
#endif

    typedef value_type* pointer;
    typedef const value_type* const_pointer;
    typedef void* void_pointer;
    typedef const void* const_void_pointer;
    typedef std::ptrdiff_t difference_type;

private:
    enum {
        min_align = detail::max_align<Alignment,
            detail::max_align<alignment_of<value_type>::value,
                alignment_of<char_ptr>::value>::value>::value
    };

public:
    template<class U>
    struct rebind {
#if !defined(BOOST_NO_CXX11_ALLOCATOR)
        typedef aligned_allocator_adaptor<typename traits::
            template rebind_alloc<U>, Alignment> other;
#else
        typedef aligned_allocator_adaptor<typename Allocator::
            template rebind<U>::other, Alignment> other;
#endif
    };

#if !defined(BOOST_NO_CXX11_DEFAULTED_FUNCTIONS)
    aligned_allocator_adaptor() = default;
#else
    aligned_allocator_adaptor()
        : Allocator() {
    }
#endif

#if !defined(BOOST_NO_CXX11_RVALUE_REFERENCES)
    template<class A>
    explicit aligned_allocator_adaptor(A&& alloc) BOOST_NOEXCEPT
        : Allocator(std::forward<A>(alloc)) {
    }
#else
    template<class A>
    explicit aligned_allocator_adaptor(const A& alloc)
        BOOST_NOEXCEPT
        : Allocator(alloc) {
    }
#endif

    template<class U>
    aligned_allocator_adaptor(const aligned_allocator_adaptor<U,
        Alignment>& other) BOOST_NOEXCEPT
        : Allocator(other.base()) {
    }

    Allocator& base() BOOST_NOEXCEPT {
        return static_cast<Allocator&>(*this);
    }

    const Allocator& base() const BOOST_NOEXCEPT {
        return static_cast<const Allocator&>(*this);
    }

    pointer allocate(size_type size) {
        std::size_t n1 = size * sizeof(value_type);
        std::size_t n2 = n1 + min_align - ptr_align;
        char_alloc a(base());
        char_ptr p1 = a.allocate(sizeof p1 + n2);
        void* p2 = detail::addressof(*p1) + sizeof p1;
        (void)align(min_align, n1, p2, n2);
        void* p3 = static_cast<char_ptr*>(p2) - 1;
        ::new(p3) char_ptr(p1);
        return static_cast<pointer>(p2);
    }

    pointer allocate(size_type size, const_void_pointer hint) {
        std::size_t n1 = size * sizeof(value_type);
        std::size_t n2 = n1 + min_align - ptr_align;
        char_ptr h = char_ptr();
        if (hint) {
            h = *(static_cast<const char_ptr*>(hint) - 1);
        }
        char_alloc a(base());
#if !defined(BOOST_NO_CXX11_ALLOCATOR)
        char_ptr p1 = char_traits::allocate(a, sizeof p1 + n2, h);
#else
        char_ptr p1 = a.allocate(sizeof p1 + n2, h);
#endif
        void* p2 = detail::addressof(*p1) + sizeof p1;
        (void)align(min_align, n1, p2, n2);
        void* p3 = static_cast<char_ptr*>(p2) - 1;
        ::new(p3) char_ptr(p1);
        return static_cast<pointer>(p2);
    }

    void deallocate(pointer ptr, size_type size) {
        char_ptr* p1 = reinterpret_cast<char_ptr*>(ptr) - 1;
        char_ptr p2 = *p1;
        p1->~char_ptr();
        char_alloc a(base());
        a.deallocate(p2, size * sizeof(value_type) +
            min_align - ptr_align + sizeof p2);
    }
};

template<class A1, class A2, std::size_t Alignment>
inline bool operator==(const aligned_allocator_adaptor<A1,
    Alignment>& a, const aligned_allocator_adaptor<A2,
    Alignment>& b) BOOST_NOEXCEPT
{
    return a.base() == b.base();
}

template<class A1, class A2, std::size_t Alignment>
inline bool operator!=(const aligned_allocator_adaptor<A1,
    Alignment>& a, const aligned_allocator_adaptor<A2,
    Alignment>& b) BOOST_NOEXCEPT
{
    return !(a == b);
}

} /* :alignment */
} /* :boost */

#endif