More TVertexBlockResource progress

Author: AdventureT

Toshi/Include/TKernel/TMemory.h

@@ -11,9 +11,10 @@ TOSHI_NAMESPACE_BEGIN
 
 class TKERNELINTERFACE_EXPORTS TMemory
 {
-
+public:
 	struct HALMemInfo
 	{
+		TINT m_iMemUsage; // 0x40
 		// sizeof 0x54
 	};
 
@@ -57,6 +58,7 @@ class TKERNELINTERFACE_EXPORTS TMemory
 	static void TOSHI_API      DebugPrintHALMemInfo(TCHAR const *a_pPrint){};
 	static TMemory &TOSHI_API  GetMemMangager();
 	static MemBlock *TOSHI_API GetGlobalBlock();
+	static void                GetHALMemInfo(HALMemInfo &a_rInfo);
 	static MemNode *TOSHI_API  GetMemNodeFromAddress(TPVOID a_pAddr);
 	static void TOSHI_API      ExtendNodeSize(MemNode *a_pMemNode, TUINT a_iuSize);
 

Toshi/Include/TRenderD3D/TD3DVertexBlockResource.h

@@ -43,8 +43,17 @@ class TRENDERINTERFACED3D_EXPORTS TVertexBlockResource : public TResource
 		m_Unk1           = 0;
 	}
 
+	virtual TBOOL Validate();
+
 	TBOOL Create(TVertexFactoryResource *a_pFactory, TUSHORT a_uiMaxVertices, TUINT a_uiFlags);
 
+protected:
+
+	TBOOL CreateHAL();
+	void  DestroyHAL();
+
+public:
+
 	TBOOL AttachPool(TVertexPoolResource *a_pPool);
 	TBOOL CanFit(TVertexPoolResource *a_pPoolResource);
 
@@ -65,6 +74,14 @@ class TRENDERINTERFACED3D_EXPORTS TVertexBlockResource : public TResource
 	// $TRenderD3DInterface: FUNCTION 100094e0
 	TUINT                            GetFlags() { return m_uiFlags; }
 
+	inline static TINT s_iCurrentNumHALCreated;
+	inline static TINT s_iTotalNumHALCreated;
+	inline static TINT s_iTotalNumHALDestroyed;
+	inline static TINT s_iCurrentVertexBufferBytesAllocated;
+	inline static TINT s_iTotalVertexBufferBytesAllocated;
+	inline static TINT s_iHALMemoryUsage;
+	inline static TINT s_iWastedVertexBufferBytesAllocated;
+
 private:
 	TVertexFactoryResourceInterface *m_pFactory; // 0x30
 	TUINT                            m_uiFlags;  // 0x34

Toshi/Source/TKernel/TMemory.cpp

@@ -1,4 +1,5 @@
 #include "TMemory.h"
+#include "TSystemTools.h"
 #include <new>
 
 #include "TMemoryDebug.h"
@@ -78,6 +79,12 @@ TMemory::MemBlock *TOSHI_API TMemory::GetGlobalBlock()
 	return g_oMemManager.m_pMemBlock;
 }
 
+// $TKernelInterface: FUNCTION 10025d70
+void TMemory::GetHALMemInfo(HALMemInfo &a_rInfo)
+{
+	TSystem::MemSet(&a_rInfo, 0, sizeof(HALMemInfo));
+}
+
 // $TKernelInterface: FUNCTION 100245a0
 TMemory::MemNode *TOSHI_API TMemory::GetMemNodeFromAddress(TPVOID a_pAddr)
 {

Toshi/Source/TRenderD3D/TD3DVertexBlockResource.cpp

@@ -93,6 +93,22 @@ TBOOL TVertexBlockResource::AttachPool(TVertexPoolResource *a_pPool)
 	return TTRUE;
 }
 
+// $TRenderD3DInterface: FUNCTION 10008960
+TBOOL TVertexBlockResource::Validate()
+{
+	if (IsValid()) {
+		return TTRUE;
+	}
+	TASSERT(TTRUE==Parent()->GetClass().IsExactly(TGetClass(TVertexFactoryResource)));
+	if (!CreateHAL()) {
+		return TFALSE;
+	}
+	if (GetFlags() & 1) {
+
+	}
+	TIMPLEMENT();
+}
+
 // $TRenderD3DInterface: FUNCTION 100088c0
 TBOOL TVertexBlockResource::Create(TVertexFactoryResource *a_pFactory, TUSHORT a_uiMaxVertices, TUINT a_uiFlags)
 {
@@ -103,3 +119,66 @@ TBOOL TVertexBlockResource::Create(TVertexFactoryResource *a_pFactory, TUSHORT a
 	m_uiFlags       = a_uiFlags;
 	return TResource::Create();
 }
+
+// $TRenderD3DInterface: FUNCTION 100091a0
+TBOOL TVertexBlockResource::CreateHAL()
+{
+	DestroyHAL();
+	TRenderD3DInterface  *pRenderer    = static_cast<TRenderD3DInterface *>(GetRenderer());
+	TVertexFactoryFormat *vertexFormat = GetFactory()->GetVertexFormat();
+	m_HALBuffer.uiNumStreams           = vertexFormat->m_uiNumStreams;
+	for (TUINT i = 0; i < m_HALBuffer.uiNumStreams; i++) {
+		UINT  length = vertexFormat->m_aStreamFormats[i].m_uiVertexSize * m_uiMaxVertices;
+		DWORD usage  = D3DUSAGE_WRITEONLY;
+		if (GetFlags() & 1) {
+			usage      = D3DUSAGE_DYNAMIC | D3DUSAGE_WRITEONLY;
+			m_uiOffset = 0;
+		}
+		if (!pRenderer->GetCurrentDevice()->SupportsHardwareTransfomations()) {
+			usage |= D3DUSAGE_SOFTWAREPROCESSING;
+		}
+		HRESULT hRes = pRenderer->GetD3DDevice()->CreateVertexBuffer(
+			length,
+			usage,
+			0,
+			D3DPOOL_DEFAULT,
+			&m_HALBuffer.apVertexBuffers[i]);
+		TRenderD3DInterface::TD3DAssert(hRes, "Unable to create a new vertex buffer!");
+		
+		TMemory::HALMemInfo memInfoHAL;
+		TMemory::GetHALMemInfo(memInfoHAL);
+		s_iHALMemoryUsage += memInfoHAL.m_iMemUsage;
+		s_iCurrentNumHALCreated += 1;
+		s_iTotalNumHALCreated += 1;
+		s_iTotalVertexBufferBytesAllocated += length;
+		s_iCurrentVertexBufferBytesAllocated += length;
+
+		if (s_iTotalVertexBufferBytesAllocated < s_iHALMemoryUsage) {
+			s_iWastedVertexBufferBytesAllocated = s_iHALMemoryUsage - s_iTotalVertexBufferBytesAllocated;
+		}
+	}
+
+	return TTRUE;
+}
+
+// $TRenderD3DInterface: FUNCTION 10009300
+void TVertexBlockResource::DestroyHAL()
+{
+	TMemory::HALMemInfo memInfoHAL;
+	TMemory::GetHALMemInfo(memInfoHAL);
+	for (TUINT i = 0; i < m_HALBuffer.uiNumStreams; i++) {
+		if (m_HALBuffer.apVertexBuffers[i]) {
+			TD3DRELEASE(m_HALBuffer.apVertexBuffers[i]);
+			m_HALBuffer.apVertexBuffers[i] = TNULL;
+			s_iCurrentNumHALCreated--;
+			s_iTotalNumHALDestroyed++;
+			s_iCurrentVertexBufferBytesAllocated -= GetFactory()->GetVertexFormat()->m_aStreamFormats[i].m_uiVertexSize * m_uiMaxVertices;
+		}
+	}
+	m_HALBuffer.uiNumStreams = 0;
+	TMemory::GetHALMemInfo(memInfoHAL);
+	s_iHALMemoryUsage = s_iHALMemoryUsage - memInfoHAL.m_iMemUsage;
+	if (s_iTotalVertexBufferBytesAllocated < s_iHALMemoryUsage) {
+		s_iWastedVertexBufferBytesAllocated = s_iHALMemoryUsage - s_iTotalVertexBufferBytesAllocated;
+	}
+}