Initial checkin

BeefRT/rt/ThreadLocalStorage.h

@@ -0,0 +1,150 @@
+#pragma once
+
+#include "BeefySysLib/util/CritSect.h"
+#include "BeefySysLib/util/Array.h"
+#include "BeefySysLib/util/Dictionary.h"
+
+namespace bf
+{
+	namespace System
+	{
+		namespace Threading
+		{
+			class Thread;
+		}
+	}
+}
+
+bf::System::Threading::Thread* BfGetCurrentThread();
+
+class BfTLSEntry
+{
+public:
+    BfpTLS* mThreadKey;
+};
+
+class BfTLSDatumRoot
+{
+public:
+	virtual void Finalize() = 0;
+};
+
+class BfTLSManager
+{
+public:
+    static BfpTLS* sInternalThreadKey;
+	Beefy::CritSect mCritSect;
+
+    BfpTLS** mAllocatedKeys;
+	BfTLSDatumRoot** mAssociatedTLSDatums;
+	int mAllocSize;
+	int mAllocIdx;
+    Beefy::Array<int> mFreeIndices;
+
+	BfTLSManager()
+	{
+        sInternalThreadKey = BfpTLS_Create();
+		mAssociatedTLSDatums = NULL;
+		mAllocSize = 0;
+		mAllocIdx = 1;
+		mAllocatedKeys = NULL;
+	}
+
+	uint32 Alloc();
+	void RegisterTlsDatum(uint32 key, BfTLSDatumRoot* tlsDatum);
+	void Free(uint32 idx);
+
+    BfpTLS* Lookup(uint32 idx)
+	{
+		return mAllocatedKeys[idx];
+	}
+};
+
+extern BfTLSManager gBfTLSManager;
+
+//#define DECLARE_THREAD_LOCAL_STORAGE(memberType) __thread memberType
+#define DECLARE_THREAD_LOCAL_STORAGE(memberType) BfTLSStatic<memberType>
+
+class BfTLSStaticBase
+{
+};
+
+template <class T>
+class BFMemClear
+{
+public:
+	static void Clear(T& valPtr)
+	{
+		memset(&valPtr, 0, sizeof(T));
+	}
+};
+
+template <class T>
+class BFNoMemClear
+{
+public:
+	static void Clear(T& valPtr)
+	{
+	}
+};
+
+template <class T, class TClear = BFMemClear<T> >
+class BfTLSStatic : public BfTLSStaticBase
+{
+public:
+	typedef BfTLSStatic<T, TClear> TLSType;
+	typedef T TLSInnerType;
+	typedef T InnerType;
+    typedef Beefy::Dictionary<bf::System::Threading::Thread*, T*> ValueMapType;
+	typedef void BFBoxedType;
+	mutable Beefy::CritSect mCritSect;
+	mutable uint32 mKey;
+	mutable ValueMapType mValueMap;
+
+public:
+	BfTLSStatic()
+	{
+        mKey = BfpTLS_Create();
+	}
+
+	T* getSafe() const
+	{
+        T* valPtr = (T*) BfpTLS_GetValue(mKey);
+		if (valPtr == NULL)
+		{
+			Beefy::AutoCrit autoCrit(mCritSect);
+			valPtr = new T();
+			TClear::Clear(*valPtr);
+			mValueMap[::BfGetCurrentThread()] = valPtr;
+            BfpTLS_SetValue(mKey, valPtr);
+		}
+		return valPtr;
+	}
+
+	const BfTLSStatic<T>& operator=(const T& rhs) const
+	{
+		*getSafe() = rhs;
+		return *this;
+	}
+
+	T& operator->() const
+	{
+		return *getSafe();
+	}
+
+	T& operator*() const
+	{
+		return *getSafe();
+	}
+
+	operator T&() const
+	{
+		return *getSafe();
+	}
+
+	ValueMapType* GetValueMap()
+	{
+		return &mValueMap;
+	}
+};
+