/* Used in Boost.MultiIndex tests.
 *
 * Copyright 2003-2020 Joaquin M Lopez Munoz.
 * Distributed under the Boost Software License, Version 1.0.
 * (See accompanying file LICENSE_1_0.txt or copy at
 * http://www.boost.org/LICENSE_1_0.txt)
 *
 * See http://www.boost.org/libs/multi_index for library home page.
 */

#ifndef BOOST_MULTI_INDEX_TEST_COUNT_ALLOCATOR_HPP
#define BOOST_MULTI_INDEX_TEST_COUNT_ALLOCATOR_HPP

#include <boost/config.hpp> /* keep it first to prevent nasty warns in MSVC */
#include <memory>

template<typename T>
struct count_allocator:std::allocator<T>
{
  typedef std::allocator<T> super;
  template<class U>
  struct rebind{typedef count_allocator<U> other;};

  count_allocator(std::size_t& element_count,std::size_t& allocator_count):
    pelement_count(&element_count),pallocator_count(&allocator_count)
    {++(*pallocator_count);}
  count_allocator(const count_allocator<T>& x):
    super(x),
    pelement_count(x.pelement_count),pallocator_count(x.pallocator_count)
    {++(*pallocator_count);}
  template<class U>count_allocator(const count_allocator<U>& x):
    super(x),
    pelement_count(x.pelement_count),pallocator_count(x.pallocator_count)
    {++(*pallocator_count);}
  ~count_allocator()
    {--(*pallocator_count);}

  count_allocator& operator=(const count_allocator<T>& x)
  {
    pelement_count=x.pelement_count;
    pallocator_count=x.pallocator_count;
    return *this;
  }

  T* allocate(std::size_t n)
  {
    *pelement_count+=n;
    return super::allocate(n);
  }

  void deallocate(T* p,std::size_t n)
  {
    super::deallocate(p,n);
    *pelement_count-=n;
  }

  std::size_t* pelement_count;
  std::size_t* pallocator_count;
};

#endif