#if SK_SUPPORT_GPU
#include "GrMemoryPool.h"
#include "SkBenchmark.h"
#include "SkRandom.h"
#include "SkTDArray.h"
#include "SkTemplates.h"
#define OVERRIDE_NEW 1
struct A {
int gStuff[10];
#if OVERRIDE_NEW
void* operator new (size_t size) { return gBenchPool.allocate(size); }
void operator delete (void* mem) { if (mem) { return gBenchPool.release(mem); } }
#endif
static GrMemoryPool gBenchPool;
};
GrMemoryPool A::gBenchPool(10 * (1 << 10), 10 * (1 << 10));
class GrMemoryPoolBenchStack : public SkBenchmark {
public:
virtual bool isSuitableFor(Backend backend) SK_OVERRIDE {
return backend == kNonRendering_Backend;
}
protected:
virtual const char* onGetName() {
return "grmemorypool_stack";
}
virtual void onDraw(const int loops, SkCanvas*) {
SkRandom r;
enum {
kMaxObjects = 4 * (1 << 10),
};
A* objects[kMaxObjects];
SkFixed delThresh = -SK_FixedHalf;
const int kSwitchThreshPeriod = loops / (2 * kMaxObjects);
int s = 0;
int count = 0;
for (int i = 0; i < loops; i++, ++s) {
if (kSwitchThreshPeriod == s) {
delThresh = -delThresh;
s = 0;
}
SkFixed del = r.nextSFixed1();
if (count &&
(kMaxObjects == count || del < delThresh)) {
delete objects[count-1];
--count;
} else {
objects[count] = new A;
++count;
}
}
for (int i = 0; i < count; ++i) {
delete objects[i];
}
}
private:
typedef SkBenchmark INHERITED;
};
struct B {
int gStuff[10];
#if OVERRIDE_NEW
void* operator new (size_t size) { return gBenchPool.allocate(size); }
void operator delete (void* mem) { if (mem) { return gBenchPool.release(mem); } }
#endif
static GrMemoryPool gBenchPool;
};
GrMemoryPool B::gBenchPool(10 * (1 << 10), 10 * (1 << 10));
class GrMemoryPoolBenchRandom : public SkBenchmark {
public:
virtual bool isSuitableFor(Backend backend) SK_OVERRIDE {
return backend == kNonRendering_Backend;
}
protected:
virtual const char* onGetName() {
return "grmemorypool_random";
}
virtual void onDraw(const int loops, SkCanvas*) {
SkRandom r;
enum {
kMaxObjects = 4 * (1 << 10),
};
SkAutoTDelete<B> objects[kMaxObjects];
for (int i = 0; i < loops; i++) {
uint32_t idx = r.nextRangeU(0, kMaxObjects-1);
if (NULL == objects[idx].get()) {
objects[idx].reset(new B);
} else {
objects[idx].free();
}
}
}
private:
typedef SkBenchmark INHERITED;
};
struct C {
int gStuff[10];
#if OVERRIDE_NEW
void* operator new (size_t size) { return gBenchPool.allocate(size); }
void operator delete (void* mem) { if (mem) { return gBenchPool.release(mem); } }
#endif
static GrMemoryPool gBenchPool;
};
GrMemoryPool C::gBenchPool(10 * (1 << 10), 10 * (1 << 10));
class GrMemoryPoolBenchQueue : public SkBenchmark {
enum {
M = 4 * (1 << 10),
};
public:
virtual bool isSuitableFor(Backend backend) SK_OVERRIDE {
return backend == kNonRendering_Backend;
}
protected:
virtual const char* onGetName() {
return "grmemorypool_queue";
}
virtual void onDraw(const int loops, SkCanvas*) {
SkRandom r;
C* objects[M];
for (int i = 0; i < loops; i++) {
uint32_t count = r.nextRangeU(0, M-1);
for (uint32_t i = 0; i < count; i++) {
objects[i] = new C;
}
for (uint32_t i = 0; i < count; i++) {
delete objects[i];
}
}
}
private:
typedef SkBenchmark INHERITED;
};
DEF_BENCH( return new GrMemoryPoolBenchStack(); )
DEF_BENCH( return new GrMemoryPoolBenchRandom(); )
DEF_BENCH( return new GrMemoryPoolBenchQueue(); )
#endif