Sto lavorando su un allocatore mmap che consente ai vettori di utilizzare la memoria da un file mappato in memoria. L'obiettivo è disporre di vettori che utilizzano l'archiviazione direttamente nella memoria virtuale mappata da mmap. Il nostro problema è migliorare la lettura di file molto grandi (> 10 GB) in memoria senza overhead di copia, quindi ho bisogno di questo allocatore personalizzato.
Finora ho lo scheletro di un allocatore personalizzato (che deriva da std :: allocator), penso che sia un buon punto di partenza per scrivere i propri allocatori. Sentiti libero di usare questo codice nel modo che preferisci:
#include <memory>
#include <stdio.h>
namespace mmap_allocator_namespace
{
// See StackOverflow replies to this answer for important commentary about inheriting from std::allocator before replicating this code.
template <typename T>
class mmap_allocator: public std::allocator<T>
{
public:
typedef size_t size_type;
typedef T* pointer;
typedef const T* const_pointer;
template<typename _Tp1>
struct rebind
{
typedef mmap_allocator<_Tp1> other;
};
pointer allocate(size_type n, const void *hint=0)
{
fprintf(stderr, "Alloc %d bytes.\n", n*sizeof(T));
return std::allocator<T>::allocate(n, hint);
}
void deallocate(pointer p, size_type n)
{
fprintf(stderr, "Dealloc %d bytes (%p).\n", n*sizeof(T), p);
return std::allocator<T>::deallocate(p, n);
}
mmap_allocator() throw(): std::allocator<T>() { fprintf(stderr, "Hello allocator!\n"); }
mmap_allocator(const mmap_allocator &a) throw(): std::allocator<T>(a) { }
template <class U>
mmap_allocator(const mmap_allocator<U> &a) throw(): std::allocator<T>(a) { }
~mmap_allocator() throw() { }
};
}
Per utilizzare questo, dichiarare un contenitore STL come segue:
using namespace std;
using namespace mmap_allocator_namespace;
vector<int, mmap_allocator<int> > int_vec(1024, 0, mmap_allocator<int>());
Può essere utilizzato ad esempio per registrare ogni volta che viene allocata la memoria. Ciò che è necessario è la struttura di rebind, altrimenti il contenitore vettoriale utilizza i metodi di allocazione / deallocazione delle superclassi.
Aggiornamento: l'allocatore di mappatura della memoria è ora disponibile su https://github.com/johannesthoma/mmap_allocator ed è LGPL. Sentiti libero di usarlo per i tuoi progetti.