+Unity Runtime codeHEADmaster

1 files changed, 70 insertions, 0 deletions

diff --git a/Runtime/Allocator/UnityDefaultAllocator.h b/Runtime/Allocator/UnityDefaultAllocator.h
new file mode 100644
index 0000000..b26ce6b
--- /dev/null
+++ b/Runtime/Allocator/UnityDefaultAllocator.h
@@ -0,0 +1,70 @@
+#ifndef UNITY_DEFAULT_ALLOCATOR_H_
+#define UNITY_DEFAULT_ALLOCATOR_H_
+
+#if ENABLE_MEMORY_MANAGER
+
+#include "BaseAllocator.h"
+#include "Runtime/Threads/Mutex.h"
+#include "Runtime/Allocator/LowLevelDefaultAllocator.h"
+#include "Runtime/Utilities/BitUtility.h"
+
+enum RequestType
+{
+	kRegister,
+	kUnregister,
+	kTest
+};
+
+template<class LLAlloctor>
+class UnityDefaultAllocator : public BaseAllocator
+{
+public:
+	UnityDefaultAllocator(const char* name);
+
+	virtual void* Allocate (size_t size, int align);
+	virtual void* Reallocate (void* p, size_t size, int align);
+	virtual void Deallocate (void* p);
+	virtual bool Contains (const void* p);
+
+	virtual size_t GetPtrSize(const void* ptr) const;
+
+	virtual ProfilerAllocationHeader* GetProfilerHeader(const void* ptr) const;
+
+	static int GetOverheadSize(void* ptr);
+private:
+	// needs 30 bit (4byte aligned allocs and packed as bitarray) ( 1 byte -> 32Bytes, 4bytes rep 128Bytes(7bit))
+	enum
+	{
+		kTargetBitsRepresentedPerBit = StaticLog2<kDefaultMemoryAlignment>::value,
+		kTargetBitsRepresentedPerByte = 5 + kTargetBitsRepresentedPerBit,
+		kPage1Bits = 7, // 128*4Bytes = 512Bytes (page: 4GB/128 = 32MB per pointer)
+		kPage2Bits = 7, // 128*4Bytes = 512Bytes (page: 32MB/128 = 256KB per pointer)
+		kPage3Bits = 5, // 32*4Bytes = 128Bytes (page: 256K/32 = 8K per pointer)
+		kPage4Bits = 32-kPage1Bits-kPage2Bits-kPage3Bits-kTargetBitsRepresentedPerByte
+	};
+
+	struct PageAllocationElement
+	{
+		PageAllocationElement() : m_HighBits(0), m_PageAllocations(NULL){}
+		UInt32 m_HighBits;
+		int**** m_PageAllocations;
+	};
+
+	static const int kNumPageAllocationBlocks = 5; //each block represents 4gb of memory;
+	PageAllocationElement m_PageAllocationList[kNumPageAllocationBlocks];
+
+	template<RequestType requestType>
+	bool AllocationPage(const void* p);
+
+	Mutex m_AllocLock;
+
+	void RegisterAllocation(const void* p);
+	void RegisterDeallocation(const void* p);
+
+	void* AddHeaderAndFooter( void* ptr, size_t size, int align ) const;
+};
+
+#endif
+
+#endif
+