some work on assetmgr

2026-03-04 14:45:34 +02:00
parent 9884006888
commit ddbdef713b
20 changed files with 282 additions and 153 deletions
--- a/game/client/assetmgr.cpp
+++ b/game/client/assetmgr.cpp
@@ -3,91 +3,152 @@
 #include "tier2/fileformats/json.h"
 #include "game.h"

-struct FunnyShader_t
-{
-	CUtlString m_szName;
-	IShader *m_pShader;
-};
 #define MAX_MODEL_COUNT 2048
+#define MAX_MESH_COUNT 2048
 #define MAX_MATERIAL_COUNT 2048
 #define MAX_TEXTURE_COUNT 4096
 #define MAX_SHADER_COUNT 1024

+#define MAX_PHYSICAL_MESH_COUNT 1024
+#define MAX_PHYSICAL_MATERIAL_COUNT 512
+
+template<typename T, uint32_t nCount>
+class CAssetArc
+{
+	struct ObjectHandle_t
+	{
+		CUtlString m_szName;
+		uint32_t m_uUsageCount;
+		T *m_pObject;
+	};
+	ObjectHandle_t m_objects[nCount] = {};
+public:
+	uint32_t GetOrCreateObject( const char *szName, bool *pbHasBeenCreated )
+	{
+		uint32_t uFoundIndex = 1;
+		for ( uint32_t u = 1; u < nCount; u++ )
+		{
+			if (m_objects[u].m_pObject == NULL)
+			{
+				continue;
+			}
+			if (m_objects[u].m_szName == szName)
+			{
+				m_objects[u].m_uUsageCount++;
+				if (pbHasBeenCreated)
+					*pbHasBeenCreated = false;
+				return u;
+			}
+		}
+		for ( auto &m: m_objects)
+		{
+			if (m.m_pObject == NULL)
+				break;
+			uFoundIndex++;
+		}
+		m_objects[uFoundIndex].m_szName = szName;
+		m_objects[uFoundIndex].m_uUsageCount++;
+		m_objects[uFoundIndex].m_pObject = (T*)V_malloc(sizeof(T));
+		if (pbHasBeenCreated)
+			*pbHasBeenCreated = true;
+		return uFoundIndex;
+	}
+	T *GetObjectPtr( uint32_t uIndex )
+	{
+		if (uIndex >= nCount)
+			return 0;
+		return m_objects[uIndex].m_pObject;
+	}
+	void UnrefObject( uint32_t uIndex )
+	{
+		if (uIndex >= nCount)
+			return;
+
+		m_objects[uIndex].m_uUsageCount--;
+		if (m_objects[uIndex].m_uUsageCount == 0)
+		{
+			V_free(m_objects[uIndex].m_pObject);
+			m_objects[uIndex].m_pObject = NULL;
+			m_objects[uIndex].m_szName = NULL;
+		}
+	}
+};
+
 class CAssetManager: public IAssetManager
 {
 public:
-	virtual uint32_t LoadModel( const char *szName ) override;
-	virtual uint32_t LoadMaterial( const char *szName ) override;
-	virtual FunnyModel_t *GetModelByIndex( uint32_t uIndex ) override;
-	virtual FunnyMaterial_t *GetMaterialByIndex( uint32_t uIndex ) override;
-	virtual void UnrefModel( uint32_t uIndex ) override;
-	virtual void UnrefMaterial( uint32_t uIndex ) override;
-	
-	FunnyModel_t *LoadModelFromParams( const char *szMesh, const char *szMaterial );
+	virtual HFunnyModel LoadModel( const char *szName ) override;
+	virtual FunnyModel_t *GetModelByIndex( HFunnyModel hModel ) override;
+	virtual void UnrefModel( HFunnyModel uIndex ) override;
+
+	virtual HFunnyMesh LoadMesh( const char *szName ) override;
+	virtual FunnyMesh_t *GetMeshByIndex( HFunnyMesh hMesh ) override;
+	virtual void UnrefMesh( HFunnyMesh hMesh ) override;
+
+	virtual HFunnyMaterial LoadMaterial( const char *szName ) override;
+	virtual FunnyMaterial_t *GetMaterialByIndex( HFunnyMaterial hMat ) override;
+	virtual void UnrefMaterial( HFunnyMaterial hMat ) override;
+
+	virtual HFunnyPhysicalMesh LoadPhysicalMesh( const char *szName ) override;
+	virtual FunnyPhysicalMesh_t *GetPhysicalMeshByIndex( HFunnyPhysicalMesh hPhysicalMesh ) override;
+	virtual void UnrefPhysicalMesh( HFunnyPhysicalMesh hPhysicalMesh ) override;
 	
+	virtual HFunnyPhysicalMaterial LoadPhysicalMaterial( const char *szName ) override;
+	virtual FunnyPhysicalMaterial_t *GetPhysicalMaterialByIndex( HFunnyPhysicalMaterial hPhysicalMaterial ) override;
+	virtual void UnrefPhysicalMaterial( HFunnyPhysicalMaterial hPhysicalMaterial ) override;
+
 	uint32_t LoadShader( const char *szName );
-	FunnyShader_t *GetShaderByIndex( uint32_t uIndex );
-	void UnrefShader( uint32_t uIndex );
+	IShader **GetShaderByIndex( uint32_t hShader );
+	void UnrefShader( uint32_t hShader );

-	FunnyModel_t *m_models[MAX_MODEL_COUNT] = {};
-	uint32_t m_modelUsages[MAX_MODEL_COUNT] = {};
-
-	FunnyMaterial_t *m_materials[MAX_MATERIAL_COUNT] = {};
-	uint32_t m_materialsUsages[MAX_MATERIAL_COUNT] = {};
-	
-	FunnyMaterial_t *m_textures[MAX_TEXTURE_COUNT] = {};
-	uint32_t m_textureUsages[MAX_TEXTURE_COUNT] = {};
-
-	FunnyShader_t *m_shaders[MAX_SHADER_COUNT] = {};
-	uint32_t m_shaderUsages[MAX_SHADER_COUNT] = {};
+	CAssetArc<FunnyModel_t, MAX_MODEL_COUNT> m_models = {};
+	CAssetArc<FunnyMesh_t, MAX_MESH_COUNT> m_meshes = {};
+	CAssetArc<FunnyMaterial_t, MAX_MATERIAL_COUNT> m_materials = {};
+	CAssetArc<IShader*, MAX_SHADER_COUNT> m_shaders = {};

 };

 FunnyModel_t *CAssetManager::GetModelByIndex( uint32_t uIndex )
 {
-	return m_models[uIndex];
+	return m_models.GetObjectPtr(uIndex);
 }

 FunnyMaterial_t *CAssetManager::GetMaterialByIndex( uint32_t uIndex )
 {
-	return m_materials[uIndex];
+	return m_materials.GetObjectPtr(uIndex);
 }
-FunnyShader_t *CAssetManager::GetShaderByIndex( uint32_t uIndex )
+
+IShader **CAssetManager::GetShaderByIndex( uint32_t uIndex )
 {
-	return m_shaders[uIndex];
+	return m_shaders.GetObjectPtr(uIndex);
+}
+
+FunnyMesh_t *CAssetManager::GetMeshByIndex( uint32_t uIndex )
+{
+	return m_meshes.GetObjectPtr(uIndex);
 }


 uint32_t CAssetManager::LoadModel( const char *szName )
 {
-	uint32_t u = 0;
-	uint32_t uFoundIndex = 1;
-	for ( auto &m: m_models)
-	{
-		if (m == NULL)
-		{
-			u++;
-			continue;
-		}
-		if (m->m_szParentName == szName)
-		{
-			m_modelUsages[uFoundIndex]++;
-			return u;
-		}
-		u++;
-	}
-	for ( auto &m: m_models)
-	{
-		if (m == NULL)
-			break;
-		uFoundIndex++;
-	}
+	bool bHasBeenCreated = false;
+	HFunnyModel hModel = m_models.GetOrCreateObject(szName, &bHasBeenCreated);
+	if (!bHasBeenCreated)
+		return hModel;
+	
+	FunnyModel_t *pModel = m_models.GetObjectPtr(hModel);	

 	IFileHandle *pHandle = filesystem->Open(szName, FILEMODE_READ);
+	if (!pHandle)
+		return 0;
+
 	CUtlString szProperties = filesystem->ReadString(pHandle);
 	IJSONValue *pRoot = JSONManager()->ReadString(szProperties);
 	filesystem->Close(pHandle);

+	if (!pRoot)
+		return 0;
+
 	IJSONObject *pMainObject;
 	CUtlString szMeshData;
 	IVertexBuffer *pVertices;
@@ -104,16 +165,14 @@ uint32_t CAssetManager::LoadModel( const char *szName )
 			}
 			IJSONValue *pMesh = pMainObject->GetValue("mesh");
 			IJSONValue *pMaterial = pMainObject->GetValue("material");
+			if (pMesh)
+				pModel->m_hMesh = LoadMesh(pMesh->GetStringValue());
+			if (pMaterial)
+				pModel->m_hMaterial = LoadMaterial(pMaterial->GetStringValue());


-			CUtlString szMesh = pMesh->GetStringValue();
-			CUtlString szMaterial = pMaterial->GetStringValue();

-			m_models[uFoundIndex] = LoadModelFromParams(szMesh, szMaterial);
-			m_models[uFoundIndex]->m_szParentName = szName;
-			m_modelUsages[uFoundIndex]++;
-
-			return uFoundIndex;
+			return hModel;
 		}	
 		break;
 	default:
@@ -129,31 +188,20 @@ void CAssetManager::UnrefModel( uint32_t uIndex )

 uint32_t CAssetManager::LoadMaterial( const char *szName )
 {
-	uint32_t u = 0;
-	uint32_t uFoundIndex = 1;
-	for ( auto &m: m_materials)
-	{
-		if (m == NULL)
-		{
-			u++;
-			continue;
-		}
-		if (m->m_szName == szName)
-		{
-			m_materialsUsages[uFoundIndex]++;
-			return u;
-		}
-		u++;
-	}
-	for ( auto &m: m_materials)
-	{
-		if (m == NULL)
-			break;
-		uFoundIndex++;
-	}
+	bool bHasBeenCreated = false;
+	HFunnyMaterial hMaterial = m_materials.GetOrCreateObject(szName, &bHasBeenCreated);
+	if (!bHasBeenCreated)
+		return hMaterial;
+	FunnyMaterial_t *pMaterial = m_materials.GetObjectPtr(hMaterial);	
+
 	IFileHandle *pHandle = filesystem->Open(szName, FILEMODE_READ);
+	if (!pHandle)
+		return 0;
 	CUtlString szProperties = filesystem->ReadString(pHandle);
 	IJSONValue *pRoot = JSONManager()->ReadString(szProperties);
+	if (!pRoot)
+		return 0;
+
 	IJSONObject *pMainObject;
 	switch (pRoot->GetType())
 	{
@@ -168,23 +216,25 @@ uint32_t CAssetManager::LoadMaterial( const char *szName )
 			}
 			IJSONValue *pShaderValue = pMainObject->GetValue("shader");
 			CUtlString szShader = pShaderValue->GetStringValue();
-			CBaseMaterial *pMaterial = CreateMaterial(szShader);
+			CBaseMaterial *pRenderMaterial = CreateMaterial(szShader);

-			uint32_t uShaderId = LoadShader(pMaterial->GetShaderPath());
-			FunnyShader_t *pShader = GetShaderByIndex(uShaderId);
-			m_materials[uFoundIndex] = new FunnyMaterial_t;
-			m_materials[uFoundIndex]->m_pShaders = pShader->m_pShader;
-			m_materials[uFoundIndex]->m_pMaterial = g_pRenderContext->CreateMaterial(pShader->m_pShader);
-			m_materials[uFoundIndex]->m_pLayout = pMaterial;
-			m_materialsUsages[uFoundIndex]++;
+			uint32_t uShaderId = LoadShader(pRenderMaterial->GetShaderPath());
+			if (!uShaderId)
+				return 0;

-			return uFoundIndex;
+			IShader **ppShader = GetShaderByIndex(uShaderId);
+
+			pMaterial->m_pShaders = *ppShader; 
+			pMaterial->m_pMaterial = g_pRenderContext->CreateMaterial(*ppShader); 
+			pMaterial->m_pLayout = pRenderMaterial; 
+
+			return hMaterial;
 		}	
 		break;
 	default:
 		return 0;
 	}
-	return uFoundIndex;
+	return hMaterial;
 }

 void CAssetManager::UnrefMaterial( uint32_t uIndex )
@@ -192,13 +242,21 @@ void CAssetManager::UnrefMaterial( uint32_t uIndex )

 }

-FunnyModel_t *CAssetManager::LoadModelFromParams( const char *szMesh, const char *szMaterial )
+HFunnyMesh CAssetManager::LoadMesh( const char *szName )
 {
-	IFileHandle *hMesh = filesystem->Open(szMesh, FILEMODE_READ);
+
+	bool bHasBeenCreated = false;
+	HFunnyMesh hAsset = m_meshes.GetOrCreateObject(szName, &bHasBeenCreated);
+	if (!bHasBeenCreated)
+		return hAsset;
+	FunnyMesh_t *pMesh = m_meshes.GetObjectPtr(hAsset);
+
+	IFileHandle *hMesh = filesystem->Open(szName, FILEMODE_READ);
 	if ( hMesh == NULL )
 	{
-		V_printf("Failed to load %s\n", szMesh);
-		return NULL;
+		V_printf("Failed to load %s\n", szName);
+		m_meshes.UnrefObject(hAsset);
+		return 0;
 	}

 	IVertexBuffer *pVertexBuffer = g_pRenderContext->CreateVertexBuffer(filesystem->Size(hMesh));
@@ -208,38 +266,24 @@ FunnyModel_t *CAssetManager::LoadModelFromParams( const char *szMesh, const char
 	pVertexBuffer->Unmap();
 	pVertexBuffer->Unlock();

-	FunnyModel_t *pModel = new FunnyModel_t;
-	pModel->m_szName = szMesh;
-	pModel->m_pMesh = g_pWorldRenderer->CreateMesh(szMesh);
-	pModel->m_pMesh->SetVertices(pVertexBuffer);
-	uint32_t uMaterial = LoadMaterial(szMaterial);
-	pModel->m_pFunnyMaterial = GetMaterialByIndex(uMaterial);
-	return pModel;
+	pMesh->m_pMesh = g_pWorldRenderer->CreateMesh(szName);
+	pMesh->m_pMesh->SetVertices(pVertexBuffer);
+
+	return hAsset;
+
+}
+void CAssetManager::UnrefMesh( uint32_t uIndex )
+{
+	m_meshes.UnrefObject(uIndex);
 }

 uint32_t CAssetManager::LoadShader( const char *szName )
 {
-	uint32_t u = 0;
-	uint32_t uFoundIndex = 1;
-	for ( auto &s: m_shaders)
-	{
-		if (s == NULL)
-		{
-			u++;
-			continue;
-		}
-		if ( s->m_szName == szName )
-		{
-			return u;
-		}
-		u++;
-	}	
-	for ( auto &m: m_models)
-	{
-		if (m == NULL)
-			break;
-		uFoundIndex++;
-	}
+	bool bHasBeenCreated = false;
+	uint32_t hShader = m_shaders.GetOrCreateObject(szName, &bHasBeenCreated);
+	if (!bHasBeenCreated)
+		return hShader;
+	IShader **ppShader = m_shaders.GetObjectPtr(hShader);	


 	IShader *pShader = g_pRenderContext->CreateShader(szName);
@@ -248,11 +292,8 @@ uint32_t CAssetManager::LoadShader( const char *szName )
 		V_printf("Failed to load %s\n", szName);
 		return 0;
 	}
-	FunnyShader_t *pFunnyShader = new FunnyShader_t;
-	pFunnyShader->m_szName = szName;
-	pFunnyShader->m_pShader = pShader;
-	m_shaders[uFoundIndex] = pFunnyShader;
-	return uFoundIndex;
+	*ppShader = pShader;
+	return hShader;
 }

 void CAssetManager::UnrefShader( uint32_t uIndex )
@@ -260,6 +301,38 @@ void CAssetManager::UnrefShader( uint32_t uIndex )

 }

+HFunnyPhysicalMesh CAssetManager::LoadPhysicalMesh( const char *szName )
+{
+
+}
+
+FunnyPhysicalMesh_t *CAssetManager::GetPhysicalMeshByIndex( HFunnyPhysicalMesh hPhysicalMesh )
+{
+
+}
+
+void CAssetManager::UnrefPhysicalMesh( HFunnyPhysicalMesh hPhysicalMesh )
+{
+
+}
+
+
+HFunnyPhysicalMaterial CAssetManager::LoadPhysicalMaterial( const char *szName )
+{
+
+}
+
+FunnyPhysicalMaterial_t *CAssetManager::GetPhysicalMaterialByIndex( HFunnyPhysicalMaterial hPhysicalMaterial )
+{
+
+}
+
+void CAssetManager::UnrefPhysicalMaterial( HFunnyPhysicalMaterial hPhysicalMaterial )
+{
+
+}
+
+

 static CAssetManager s_assetmgr;
 IAssetManager *g_pAssetManager = &s_assetmgr;