stl_pool.h

Go to the documentation of this file.

/* -*- mode:C++; c-basic-offset:4 -*-
     Shore-kits -- Benchmark implementations for Shore-MT

                       Copyright (c) 2007-2009
      Data Intensive Applications and Systems Labaratory (DIAS)
               Ecole Polytechnique Federale de Lausanne

                         All Rights Reserved.

   Permission to use, copy, modify and distribute this software and
   its documentation is hereby granted, provided that both the
   copyright notice and this permission notice appear in all copies of
   the software, derivative works or modified versions, and any
   portions thereof, and that both notices appear in supporting
   documentation.

   This code 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. THE AUTHORS
   DISCLAIM ANY LIABILITY OF ANY KIND FOR ANY DAMAGES WHATSOEVER
   RESULTING FROM THE USE OF THIS SOFTWARE.
*/

#ifndef __STL_POOL
#define __STL_POOL

#include <iostream>
#include "util/guard.h"

class Pool {
public:
    Pool(size_t granularity, size_t size);

    ~Pool();

    size_t GetGranularity() const {
        return m_granularity;
    }

    size_t GetSize() const {
        return m_size;
    }

    size_t GetUsed() const {
        return m_used;
    }

    size_t GetOverflow() const {
        return m_overflow;
    }

    void* Allocate();

    void Deallocate(void* block);

    template<typename T>
    T* Construct() {
        assert(sizeof(T) <= m_granularity);
        T* block = reinterpret_cast<T*>( Allocate());
        return new(block) T;
    }

    template<typename T>
    void Destroy(T* instance) {
        assert(sizeof(T) <= m_granularity);
        instance->~T();
        Deallocate(instance);
    }

private:
    bool IsFromPool(void const* instance) const {
        char const* block = reinterpret_cast<char const*>( instance );
        return m_storage.get() <= block && block < (m_storage.get() + m_size * m_granularity);
    }

    size_t m_granularity;    
    size_t m_size;            
    size_t volatile m_used;            
    size_t volatile m_overflow;        

    array_guard_t<char> m_storage;    
    array_guard_t<void*> m_slots;        
};

#endif // __STL_POOL