mirror of https://github.com/PrimeDecomp/prime.git
87 lines
2.0 KiB
C++
87 lines
2.0 KiB
C++
#ifndef _TTOKEN
|
|
#define _TTOKEN
|
|
|
|
#include "types.h"
|
|
|
|
#include "Kyoto/CSimplePool.hpp"
|
|
#include "Kyoto/CToken.hpp"
|
|
#include "Kyoto/IObj.hpp"
|
|
|
|
#include "rstl/auto_ptr.hpp"
|
|
|
|
template < typename T >
|
|
class TToken : public CToken {
|
|
public:
|
|
TToken() {}
|
|
TToken(const CToken& token) : CToken(token) {}
|
|
TToken(T* obj) : CToken(TObjOwnerDerivedFromIObj< T >::GetNewDerivedObject(obj).release()) {}
|
|
TToken(const rstl::auto_ptr< T >& obj) : CToken(GetIObjObjectFor(obj).release()) {}
|
|
|
|
T* GetT() { return reinterpret_cast< T* >(CToken::GetObj()->GetContents()); }
|
|
T* operator*() { return GetT(); }
|
|
T* operator->() { return GetT(); }
|
|
|
|
static rstl::auto_ptr< TObjOwnerDerivedFromIObj< T > >
|
|
GetIObjObjectFor(const rstl::auto_ptr< T >& obj) {
|
|
return TObjOwnerDerivedFromIObj< T >::GetNewDerivedObject(obj);
|
|
}
|
|
|
|
TToken< T > NonConstCopy() const { return *const_cast< TToken< T >* >(this); }
|
|
};
|
|
|
|
template < typename T >
|
|
class TCachedToken : public TToken< T > {
|
|
public:
|
|
TCachedToken() {}
|
|
TCachedToken(const CToken& token) : TToken< T >(token), x8_item(nullptr) {}
|
|
|
|
T* operator*() { return x8_item; }
|
|
|
|
bool IsLoaded() {
|
|
if (x8_item != nullptr) {
|
|
return true;
|
|
}
|
|
if (CToken::HasLock() && CToken::IsLoaded()) {
|
|
x8_item = TToken< T >::GetT();
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void Unlock() {
|
|
x8_item = nullptr;
|
|
TToken< T >::Unlock();
|
|
}
|
|
|
|
private:
|
|
T* x8_item;
|
|
};
|
|
|
|
template < typename T >
|
|
class TLockedToken {
|
|
public:
|
|
TLockedToken() {}
|
|
TLockedToken(const CToken& token) : x0_token(token), x8_item(*x0_token) {}
|
|
TLockedToken(const TLockedToken< T >& token) : x0_token(token), x8_item(*token) {
|
|
x0_token.Lock();
|
|
}
|
|
|
|
TLockedToken& operator=(const TLockedToken< T >& token) {
|
|
x0_token = token;
|
|
x8_item = *token;
|
|
return *this;
|
|
}
|
|
|
|
operator const TToken< T >&() const { return x0_token; }
|
|
T* operator*() const { return x8_item; }
|
|
T* operator->() const { return x8_item; }
|
|
bool IsLoaded() const { return x0_token.IsLoaded(); }
|
|
|
|
private:
|
|
TToken< T > x0_token;
|
|
T* x8_item;
|
|
};
|
|
|
|
#endif // _TTOKEN
|