pool_allocator.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2019 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner
 
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, 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
// Lesser General Public License for more details.
 
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
#ifndef LIBMESH_POOL_ALLOCATOR_H
#define LIBMESH_POOL_ALLOCATOR_H
 
#include "libmesh/libmesh_config.h"
 
#ifdef LIBMESH_HAVE_BOOST
// See: http://stackoverflow.com/questions/17000542/boost-pool-can-i-wean-it-from-boost-system
#define BOOST_POOL_NO_MT       // disable multi-threading
#define BOOST_THREAD_MUTEX_HPP // define the #include-guard to disable the header
#  include <boost/pool/pool.hpp>
#  include <boost/pool/object_pool.hpp>
#  include <boost/pool/pool_alloc.hpp>
#endif
 
#include <memory> // std::allocator
 
namespace libMesh
{
// If Boost is enabled, wrappers to use their allocators.
#ifdef LIBMESH_HAVE_BOOST
 
template <typename T>
class PoolAllocator : public boost::pool_allocator<T>
{
public:
 
  template<typename U>
  struct rebind {
    typedef PoolAllocator<U> other;
  };
 
 
  PoolAllocator() :
    boost::pool_allocator<T>()
  {}
 
  explicit PoolAllocator(const PoolAllocator & o) :
    boost::pool_allocator<T>(o)
  {}
 
  static bool release_memory ()
  {
    return boost::singleton_pool<boost::pool_allocator_tag, sizeof(T)>::release_memory();
  }
 
  static bool purge_memory ()
  {
    return boost::singleton_pool<boost::pool_allocator_tag, sizeof(T)>::purge_memory();
  }
};
 
 
 
template <typename T>
class FastPoolAllocator : public boost::fast_pool_allocator<T>
{
public:
 
  template<typename U>
  struct rebind {
    typedef FastPoolAllocator<U> other;
  };
 
 
  FastPoolAllocator() :
    boost::fast_pool_allocator<T>()
  {}
 
  explicit FastPoolAllocator(const FastPoolAllocator & o) :
    boost::fast_pool_allocator<T>(o)
  {}
 
 
  static bool release_memory ()
  {
    return boost::singleton_pool<boost::fast_pool_allocator_tag, sizeof(T)>::release_memory();
  }
 
  static bool purge_memory ()
  {
    return boost::singleton_pool<boost::fast_pool_allocator_tag, sizeof(T)>::purge_memory();
  }
};
 
// Otherwise fall back to std::allocator<>.
#else
 
template <typename T>
class PoolAllocator : public std::allocator<T>
{
public:
 
  template<typename U>
  struct rebind {
    typedef PoolAllocator<U> other;
  };
 
  PoolAllocator() :
    std::allocator<T>()
  {}
 
  explicit PoolAllocator(const PoolAllocator & o) :
    std::allocator<T>(o)
  {}
 
  static bool release_memory () { /* no-op for std::allocator<> - already freed. */ return false; }
 
  static bool purge_memory ()   { /* no-op for std::allocator<> - already freed. */ return false; }
};
 
 
 
template <typename T>
class FastPoolAllocator : public std::allocator<T>
{
public:
 
  template<typename U>
  struct rebind {
    typedef FastPoolAllocator<U> other;
  };
 
  FastPoolAllocator() :
    std::allocator<T>()
  {}
 
  explicit FastPoolAllocator(const FastPoolAllocator & o) :
    std::allocator<T>(o)
  {}
 
  static bool release_memory () { /* no-op for std::allocator<> - already freed. */ return false; }
 
  static bool purge_memory ()   { /* no-op for std::allocator<> - already freed. */ return false; }
};
 
#endif
 
} // end namespace libMesh
 
 
#endif // LIBMESH_POOL_ALLOCATOR_H