Ticket #7106: multi_array-alloc.cpp

#include <boost/multi_array.hpp>
#include <iostream>
#include <string>
#include <boost/array.hpp>

template<typename T>
class my_allocator : public std::allocator<T>
{
private:
    std::string name_;
    template<typename B> friend class my_allocator;
public:
    typedef std::allocator<T> base_type;

    template<typename B>
    my_allocator(const my_allocator<B> &b)
        : name_(b.name_) {}

    explicit my_allocator(const std::string &name = "default", const base_type &base = base_type())
        : base_type(base), name_(name) {}

    template<typename U> struct rebind
    {
        typedef my_allocator<U> other;
    };


    typename base_type::pointer allocate(typename base_type::size_type n)
    {
        typename base_type::pointer ans = base_type::allocate(n);
        if (ans)
            std::cout << "Allocated " << n << " bytes [name = " << name_ << ']' << std::endl;
        return ans;
    }

    typename base_type::pointer allocate(typename base_type::size_type n,
                                         typename base_type::const_pointer hint)
    {
        typename base_type::pointer ans = base_type::allocate(n, hint);
        if (ans)
            std::cout << "Allocated " << n << " bytes [name = " << name_ << ']' << std::endl;
        return ans;
    }
};

int main()
{
    int dummy = 0;
    std::vector<int, my_allocator<int> > v(&dummy, &dummy, my_allocator<int>("vector"));
    v.resize(17);

    boost::array<int, 2> zero = {{0, 0}};
    boost::multi_array<int, 2, my_allocator<int> > m(zero, boost::c_storage_order(), my_allocator<int>("multi_array"));

    boost::array<int, 2> dims = {{5, 3}};
    m.resize(dims);
    return 0;
}