gecko/xpcom/ds/nsFixedSizeAllocator.h
Ehsan Akhgari 8c296bbcd4 Bug 579517 - Part 1: Automated conversion of NSPR numeric types to stdint types in Gecko; r=bsmedberg
This patch was generated by a script.  Here's the source of the script for
future reference:

function convert() {
echo "Converting $1 to $2..."
find . ! -wholename "*nsprpub*" \
       ! -wholename "*security/nss*" \
       ! -wholename "*/.hg*" \
       ! -wholename "obj-ff-dbg*" \
       ! -name nsXPCOMCID.h \
       ! -name prtypes.h \
         -type f \
      \( -iname "*.cpp" \
         -o -iname "*.h" \
         -o -iname "*.c" \
         -o -iname "*.cc" \
         -o -iname "*.idl" \
         -o -iname "*.ipdl" \
         -o -iname "*.ipdlh" \
         -o -iname "*.mm" \) | \
    xargs -n 1 sed -i -e "s/\b$1\b/$2/g"
}

convert PRInt8 int8_t
convert PRUint8 uint8_t
convert PRInt16 int16_t
convert PRUint16 uint16_t
convert PRInt32 int32_t
convert PRUint32 uint32_t
convert PRInt64 int64_t
convert PRUint64 uint64_t

convert PRIntn int
convert PRUintn unsigned

convert PRSize size_t

convert PROffset32 int32_t
convert PROffset64 int64_t

convert PRPtrdiff ptrdiff_t

convert PRFloat64 double
2012-08-22 11:56:38 -04:00

166 lines
4.4 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
/*
A simple fixed-size allocator that allocates its memory from an
arena.
Although the allocator can handle blocks of any size, its
preformance will degrade rapidly if used to allocate blocks of
arbitrary size. Ideally, it should be used to allocate and recycle a
large number of fixed-size blocks.
Here is a typical usage pattern:
#include NEW_H // You'll need this!
#include "nsFixedSizeAllocator.h"
// Say this is the object you want to allocate a ton of
class Foo {
public:
// Implement placement new & delete operators that will
// use the fixed size allocator.
static Foo *
Create(nsFixedSizeAllocator &aAllocator)
{
void *place = aAllocator.Alloc(sizeof(Foo));
return place ? ::new (place) Foo() : nullptr;
}
static void
Destroy(nsFixedSizeAllocator &aAllocator, Foo *aFoo)
{
aFoo->~Foo();
aAllocator.Free(aFoo, sizeof(Foo));
}
// ctor & dtor
Foo() {}
~Foo() {}
};
int main(int argc, char* argv[])
{
// Somewhere in your code, you'll need to create an
// nsFixedSizeAllocator object and initialize it:
nsFixedSizeAllocator pool;
// The fixed size allocator will support multiple fixed sizes.
// This array lists an initial set of sizes that the allocator
// should be prepared to support. In our case, there's just one,
// which is Foo.
static const size_t kBucketSizes[]
= { sizeof(Foo) }
// This is the number of different "buckets" you'll need for
// fixed size objects. In our example, this will be "1".
static const int32_t kNumBuckets
= sizeof(kBucketSizes) / sizeof(size_t);
// This is the intial size of the allocator, in bytes. We'll
// assume that we want to start with space for 1024 Foo objects.
static const int32_t kInitialPoolSize = sizeof(Foo) * 1024;
// Initialize (or re-initialize) the pool
pool.Init("TheFooPool", kBucketSizes, kNumBuckets, kInitialPoolSize);
// Now we can use the pool.
// Create a new Foo object using the pool:
Foo* foo = Foo::Create(pool);
if (! foo) {
// uh oh, out of memory!
}
// Delete the object. The memory used by `foo' is recycled in
// the pool, and placed in a freelist
Foo::Destroy(foo);
// Create another foo: this one will be allocated from the
// free-list
foo = Foo::Create(pool);
// When pool is destroyed, all of its memory is automatically
// freed. N.B. it will *not* call your objects' destructors! In
// this case, foo's ~Foo() method would never be called.
}
*/
#ifndef nsFixedSizeAllocator_h__
#define nsFixedSizeAllocator_h__
#include "nscore.h"
#include "nsError.h"
#include "plarena.h"
class nsFixedSizeAllocator
{
protected:
PLArenaPool mPool;
struct Bucket;
struct FreeEntry;
friend struct Bucket;
friend struct FreeEntry;
struct FreeEntry {
FreeEntry* mNext;
};
struct Bucket {
size_t mSize;
FreeEntry* mFirst;
Bucket* mNext;
};
Bucket* mBuckets;
Bucket *
AddBucket(size_t aSize);
Bucket *
FindBucket(size_t aSize);
public:
nsFixedSizeAllocator() : mBuckets(nullptr) {}
~nsFixedSizeAllocator() {
if (mBuckets)
PL_FinishArenaPool(&mPool);
}
/**
* Initialize the fixed size allocator. 'aName' is used to tag
* the underlying PLArena object for debugging and measurement
* purposes. 'aNumBuckets' specifies the number of elements in
* 'aBucketSizes', which is an array of integral block sizes
* that this allocator should be prepared to handle.
*/
nsresult
Init(const char* aName,
const size_t* aBucketSizes,
int32_t aNumBuckets,
int32_t aInitialSize,
int32_t aAlign = 0);
/**
* Allocate a block of memory 'aSize' bytes big.
*/
void* Alloc(size_t aSize);
/**
* Free a pointer allocated using a fixed-size allocator
*/
void Free(void* aPtr, size_t aSize);
};
#endif // nsFixedSizeAllocator_h__