ultimatepp/uppsrc/Core/Other.h
cxl 93fc976688 Improved LRUCache (Key value now template parameter)
git-svn-id: svn://ultimatepp.org/upp/trunk@722 f0d560ea-af0d-0410-9eb7-867de7ffcac7
2009-01-02 21:51:01 +00:00

432 lines
11 KiB
C++

template <class T>
T& Single() {
static T *p;
ONCELOCK {
static T o;
p = &o;
}
return *p;
}
template <class T>
class One : MoveableAndDeepCopyOption< One<T> > {
mutable T *ptr;
void Free() { if(ptr && ptr != (T*)1) delete ptr; }
void Chk() const { ASSERT(ptr != (T*)1); }
void ChkP() const { Chk(); ASSERT(ptr); }
void Pick(pick_ One<T>& data) { T *p = data.ptr; data.ptr = (T*)1; ptr = p; }
public:
void Attach(T *data) { Free(); ptr = data; }
T *Detach() pick_ { ChkP(); T *t = ptr; ptr = NULL; return t; }
T *operator-() pick_ { return Detach(); }
void Clear() { Free(); ptr = NULL; }
void operator=(T *data) { Attach(data); }
void operator=(pick_ One<T>& d) { Free(); Pick(d); }
const T *operator->() const { ChkP(); return ptr; }
T *operator->() { ChkP(); return ptr; }
const T *operator~() const { Chk(); return ptr; }
T *operator~() { Chk(); return ptr; }
const T& operator*() const { ChkP(); return *ptr; }
T& operator*() { ChkP(); return *ptr; }
template <class TT>
TT& Create() { TT *q = new TT; Attach(q); return *q; }
T& Create() { T *q = new T; Attach(q); return *q; }
bool IsPicked() const { return ptr == (T*)1; }
bool IsEmpty() const { Chk(); return !ptr; }
operator bool() const { return ptr; }
One() { ptr = NULL; }
One(T *newt) { ptr = newt; }
One(pick_ One<T>& p) { Pick(p); }
One(const One<T>& p, int) { ptr = p.IsEmpty() ? NULL : DeepCopyNew(*p); }
~One() { Free(); }
};
class Any : Moveable<Any> {
struct BaseData {
int typeno;
virtual ~BaseData() {}
};
template <class T>
struct Data : BaseData {
T data;
Data() { typeno = StaticTypeNo<T>(); }
};
BaseData *ptr;
void Chk() const { ASSERT(ptr != (void *)1); }
void Pick(pick_ Any& s) { ptr = s.ptr; const_cast<Any&>(s).ptr = (BaseData *)1; }
public:
template <class T> T& Create() { Clear(); Data<T> *x = new Data<T>; ptr = x; return x->data; }
template <class T> bool Is() const { return ptr && ptr->typeno == StaticTypeNo<T>(); }
template <class T> T& Get() { ASSERT(Is<T>()); Chk(); return ((Data<T>*)ptr)->data; }
template <class T> const T& Get() const { ASSERT(Is<T>()); Chk(); return ((Data<T>*)ptr)->data; }
void Clear() { if(ptr && !IsPicked()) delete ptr; ptr = NULL; }
bool IsEmpty() const { return ptr == NULL; }
bool IsPicked() const { return ptr == (void *)1; }
void operator=(pick_ Any& s) { Clear(); Pick(s); }
Any(pick_ Any& s) { Pick(s); }
Any() { ptr = NULL; }
~Any() { Clear(); }
};
template <class T>
class Buffer : Moveable< Buffer<T> > {
mutable T *ptr;
public:
operator T*() { return ptr; }
operator const T*() const { return ptr; }
T *operator~() { return ptr; }
const T *operator~() const { return ptr; }
void Alloc(int size) { Clear(); ptr = new T[size]; }
void Alloc(int size, const T& in) { Clear(); ptr = new T[size];
Fill(ptr, ptr + size, in); }
void Clear() { if(ptr) delete[] ptr; ptr = NULL; }
Buffer() { ptr = NULL; }
Buffer(int size) { ptr = new T[size]; }
Buffer(int size, const T& init) { ptr = new T[size]; Fill(ptr, ptr + size, init); }
~Buffer() { if(ptr) delete[] ptr; }
void operator=(pick_ Buffer& v) { if(ptr) delete[] ptr; ptr = v.ptr; v.ptr = NULL; }
Buffer(pick_ Buffer& v) { ptr = v.ptr; v.ptr = NULL; }
};
class Bits : Moveable<Bits> {
mutable int alloc;
dword *bp;
public:
void Clear();
void Set(int i, bool b = true);
void Set(int i, bool b, int count);
bool Get(int i) const { ASSERT(i >= 0 && alloc >= 0); int q = i >> 5;
return q < alloc ? bp[q] & (1 << (i & 31)) : false; }
bool operator[](int i) const { return Get(i); }
Bits() { bp = NULL; alloc = 0; }
~Bits() { Clear(); }
Bits(pick_ Bits& b) { alloc = b.alloc; bp = b.bp; b.alloc = -1; }
void operator=(pick_ Bits& b) { Clear(); alloc = b.alloc; bp = b.bp; b.alloc = -1; }
};
//# System dependent
template <class T>
class Mitor : Moveable< Mitor<T> > {
union {
mutable unsigned count;
mutable Vector<T> *vector;
};
byte elem0[sizeof(T)];
T& Get(int i) const;
void Pick(pick_ Mitor& m);
void Copy(const Mitor& m);
void Chk() const { ASSERT(count != 2); }
public:
T& operator[](int i) { return Get(i); }
const T& operator[](int i) const { return Get(i); }
int GetCount() const;
T& Add();
void Add(const T& x);
void Clear();
void Shrink();
Mitor(pick_ Mitor& m) { Pick(m); }
void operator=(pick_ Mitor& m) { Clear(); Pick(m); }
Mitor(Mitor& m, int) { Copy(m); }
void operator<<=(const Mitor& m) { Clear(); Copy(m); }
Mitor() { count = 0; }
~Mitor() { Clear(); }
};
template <class T>
T& Mitor<T>::Get(int i) const
{
ASSERT(i >= 0 && i < GetCount());
return i == 0 ? *(T*)elem0 : (*const_cast<Vector<T>*>(vector))[i - 1];
}
template <class T>
void Mitor<T>::Pick(pick_ Mitor& m)
{
m.Chk();
vector = m.vector;
memcpy(&elem0, &m.elem0, sizeof(T));
m.count = 2;
}
template <class T>
void Mitor<T>::Copy(const Mitor& m)
{
m.Chk();
if(m.count > 0)
DeepCopyConstruct((T*)elem0, m.elem0);
if(m.count > 1)
vector = new Vector<T>(m.vector, 1);
}
template <class T>
int Mitor<T>::GetCount() const
{
Chk();
return count > 1 ? vector->GetCount() + 1 : count;
}
template <class T>
T& Mitor<T>::Add()
{
Chk();
if(count == 0) {
count = 1;
return *new(elem0) T;
}
if(count == 1)
vector = new Vector<T>;
return vector->Add();
}
template <class T>
void Mitor<T>::Add(const T& x)
{
Chk();
if(count == 0) {
count = 1;
new((T*) elem0) T(x);
}
else {
if(count == 1)
vector = new Vector<T>;
vector->Add(x);
}
}
template <class T>
void Mitor<T>::Clear()
{
if(count > 2)
delete vector;
if(count && count != 2)
((T*)elem0)->T::~T();
count = 0;
}
template <class T>
void Mitor<T>::Shrink()
{
if(count > 2)
vector->Shrink();
}
//#
template <class T, int N = 1>
class Link {
protected:
T *prev[N];
T *next[N];
void LPN(int i) { prev[i]->next[i] = next[i]->prev[i] = (T *)this; }
public:
T *GetPtr() { return (T *) this; }
const T *GetPtr() const { return (const T *) this; }
T *GetNext(int i = 0) { return next[i]; }
T *GetPrev(int i = 0) { return prev[i]; }
const T *GetNext(int i = 0) const { return next[i]; }
const T *GetPrev(int i = 0) const { return prev[i]; }
void LinkSelf(int i = 0) { next[i] = prev[i] = (T *)this; }
void LinkSelfAll() { for(int i = 0; i < N; i++) LinkSelf(i); }
void Unlink(int i = 0) { next[i]->prev[i] = prev[i]; prev[i]->next[i] = next[i];
LinkSelf(i); }
void UnlinkAll() { for(int i = 0; i < N; i++) Unlink(i); }
void LinkBefore(Link *n, int i = 0) { next[i] = (T *)n; prev[i] = next[i]->prev[i]; LPN(i); }
void LinkAfter(Link *p, int i = 0) { prev[i] = (T *)p; next[i] = prev[i]->next[i]; LPN(i); }
T *InsertNext(int i = 0) { T *x = new T; x->LinkAfter(this, i); return x; }
T *InsertPrev(int i = 0) { T *x = new T; x->LinkBefore(this, i); return x; }
void DeleteList(int i = 0) { while(next[i] != GetPtr()) delete next[i]; }
bool InList(int i = 0) const { return next[i] != GetPtr(); }
bool IsEmpty(int i = 0) const { return !InList(i); }
Link() { LinkSelfAll(); }
~Link() { UnlinkAll(); }
private:
Link(const Link&);
void operator=(const Link&);
public:
#ifdef _DEBUG
void Dump() {
for(T *t = GetNext(); t != this; t = t->GetNext())
LOG(t);
LOG("-------------------------------------");
}
#endif
};
template <class T, int N = 1>
class LinkOwner : public Link<T, N> {
public:
~LinkOwner() { Link<T, N>::DeleteList(); }
};
template <class T, class K = String>
struct LRUCache {
struct Maker {
virtual K Key() const = 0;
virtual int Make(T& object) const = 0;
virtual ~Maker() {}
};
private:
struct Item : Moveable<Item> {
int16 prev, next;
int size;
One<T> data;
bool flag;
};
struct Key : Moveable<Key> {
K key;
const void *type;
bool operator==(const Key& b) const { return key == b.key && type == b.type; }
unsigned GetHashValue() const { return CombineHash(key, (uintptr_t)type); }
};
Index<Key> key;
Vector<Item> data;
int head;
int size;
int count;
int foundsize;
int newsize;
bool flag;
void Unlink(int i);
void LinkHead(int i);
public:
int GetSize() const { return size; }
void Shrink(int maxsize);
const T& Get(const Maker& m);
void ClearCounters();
int GetFoundSize() const { return foundsize; }
int GetNewSize() const { return newsize; }
LRUCache() { head = -1; size = 0; count = 0; ClearCounters(); }
};
template <class T, class K>
void LRUCache<T, K>::LinkHead(int i)
{
Item& m = data[i];
if(head >= 0) {
int tail = data[head].prev;
m.next = head;
m.prev = tail;
data[head].prev = i;
data[tail].next = i;
}
else
m.prev = m.next = i;
head = i;
count++;
}
template <class T, class K>
void LRUCache<T, K>::Unlink(int i)
{
Item& m = data[i];
if(m.prev == i)
head = -1;
else {
if(head == i)
head = m.next;
data[(int)m.next].prev = m.prev;
data[(int)m.prev].next = m.next;
}
count--;
}
template <class T, class K>
void LRUCache<T, K>::Shrink(int maxsize)
{
if(maxsize < 0)
return;
while((count > 30000 || size > maxsize) && head >= 0) {
int tail = data[head].prev;
size -= data[tail].size;
data[tail].data.Clear();
Unlink(tail);
key.Unlink(tail);
}
}
template <class T, class K>
void LRUCache<T, K>::ClearCounters()
{
flag = !flag;
newsize = foundsize = 0;
}
template <class T, class K>
const T& LRUCache<T, K>::Get(const Maker& m)
{
Key k;
k.key = m.Key();
k.type = &typeid(m);
int q = key.Find(k);
if(q < 0) {
q = key.Put(k);
Item& t = data.At(q);
t.size = m.Make(t.data.Create());
size += t.size;
newsize += t.size;
t.flag = flag;
}
else {
Item& t = data[q];
Unlink(q);
if(t.flag != flag) {
t.flag = flag;
foundsize += t.size;
}
}
LinkHead(q);
return *data[q].data;
}