asset manager

game/client/assetmgr.cpp

@@ -1,27 +1,80 @@
 #include "assetmgr.h"
 #include "tier2/ifilesystem.h"
 #include "tier2/fileformats/json.h"
+#include "game.h"
+
+struct FunnyShader_t
+{
+	CUtlString m_szName;
+	IShader *m_pShader;
+};
+#define MAX_MODEL_COUNT 2048
+#define MAX_MATERIAL_COUNT 2048
+#define MAX_SHADER_COUNT 1024
 
 class CAssetManager: public IAssetManager
 {
 public:
-	virtual FunnyModel_t *LoadModel( const char *szName ) override;
-	virtual FunnyMaterial_t *LoadMaterial( const char *szName ) override;
-	virtual void UnrefModel( FunnyModel_t *pModel ) override;
-	virtual void UnrefMaterial( FunnyMaterial_t *pModel ) override;
-	CUtlVector<FunnyModel_t> m_models;
-	CUtlVector<FunnyModel_t*> m_modelUsages;
-	CUtlVector<FunnyMaterial_t> m_materials;
-	CUtlVector<FunnyMaterial_t*> m_materialUsages;
-
+	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 );
+	
+	uint32_t LoadShader( const char *szName );
+	void UnrefShader( uint32_t uIndex );
+
+	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] = {};
+
+	FunnyShader_t *m_shaders[MAX_SHADER_COUNT] = {};
+	uint32_t m_shaderUsages[MAX_SHADER_COUNT] = {};
+
 };
 
-FunnyModel_t *CAssetManager::LoadModel( const char *szName )
+FunnyModel_t *CAssetManager::GetModelByIndex( uint32_t uIndex )
 {
+	return m_models[uIndex];
+}
+
+FunnyMaterial_t *CAssetManager::GetMaterialByIndex( uint32_t uIndex )
+{
+	return m_materials[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_szName == szName)
+		{
+			m_modelUsages[uFoundIndex]++;
+			return u;
+		}
+		u++;
+	}
+	for ( auto &m: m_models)
+	{
+		if (m == NULL)
+			break;
+		uFoundIndex++;
+	}
+
 	IFileHandle *pHandle = filesystem->Open(szName, FILEMODE_READ);
 	CUtlString szProperties = filesystem->ReadString(pHandle);
-	V_printf("%s\n", szProperties.GetString());
 	IJSONValue *pRoot = JSONManager()->ReadString(szProperties);
 	IJSONObject *pMainObject;
 	CUtlString szMeshData;
@@ -34,7 +87,7 @@ FunnyModel_t *CAssetManager::LoadModel( const char *szName )
 			if (!pMainObject)
 			{
 				V_printf("Failed to load properties\n");
-				return NULL;
+				return 0;
 
 			}
 			IJSONValue *pMesh = pMainObject->GetValue("mesh");
@@ -42,31 +95,119 @@ FunnyModel_t *CAssetManager::LoadModel( const char *szName )
 			CUtlString szMesh = pMesh->GetStringValue();
 			CUtlString szMaterial = pMaterial->GetStringValue();
 
-			return NULL;
+			m_models[uFoundIndex] = LoadModelFromParams(szMesh, szMaterial);
+			m_modelUsages[uFoundIndex]++;
+
+			return uFoundIndex;
 		}	
 		break;
 	default:
-		return NULL;
+		return 0;
 	}
 }
 
-void CAssetManager::UnrefModel( FunnyModel_t *pModel )
+void CAssetManager::UnrefModel( uint32_t uIndex )
 {
 
 }
 
-FunnyMaterial_t *CAssetManager::LoadMaterial( const char *szName )
+
+uint32_t CAssetManager::LoadMaterial( const char *szName )
 {
-	FunnyMaterial_t stMaterial = {};
+	uint32_t u = 0;
+	uint32_t uFoundIndex = 1;
+	for ( auto &m: m_models)
+	{
+		if (m == NULL)
+		{
+			u++;
+			continue;
+		}
+		if (m->m_szName == szName)
+		{
+			m_modelUsages[uFoundIndex]++;
+			return u;
+		}
+		u++;
+	}
+	for ( auto &m: m_models)
+	{
+		if (m == NULL)
+			break;
+		uFoundIndex++;
+	}
 }
 
-void CAssetManager::UnrefMaterial( FunnyMaterial_t *pModel )
+void CAssetManager::UnrefMaterial( uint32_t uIndex )
 {
 
 }
+
 FunnyModel_t *CAssetManager::LoadModelFromParams( const char *szMesh, const char *szMaterial )
 {
+	IFileHandle *hMesh = filesystem->Open(szMesh, FILEMODE_READ);
+	if ( hMesh == NULL )
+	{
+		V_printf("Failed to load %s\n", szMesh);
+		return NULL;
+	}
+
+	IVertexBuffer *pVertexBuffer = g_pRenderContext->CreateVertexBuffer(filesystem->Size(hMesh));
+	pVertexBuffer->Lock();
+	void *pMapped = pVertexBuffer->Map();
+	filesystem->Read(hMesh, pMapped, pVertexBuffer->GetSize());
+	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);
+	return pModel;
 }
 
+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++;
+	}
+
+
+	IShader *pShader = g_pRenderContext->CreateShader(szName);
+	if ( pShader == NULL )
+	{
+		V_printf("Failed to load %s\n", szName);
+		return 0;
+	}
+	FunnyShader_t *pFunnyShader = new FunnyShader_t;
+	pFunnyShader->m_szName = szName;
+	pFunnyShader->m_pShader = pShader;
+	return uFoundIndex;
+}
+
+void CAssetManager::UnrefShader( uint32_t uIndex )
+{
+
+}
+
+
 static CAssetManager s_assetmgr;
 IAssetManager *g_pAssetManager = &s_assetmgr;