ParaVertexBufferPool.h

#pragma once
#include "AssetManager.h"
#include <unordered_set>

namespace ParaEngine
{
    class ParaVertexBuffer;

    class ParaVertexBufferPool : public AssetEntity
    {
    public:
        ParaVertexBufferPool(const AssetKey& key);
        ParaVertexBufferPool();
        virtual ~ParaVertexBufferPool();

        ATTRIBUTE_DEFINE_CLASS(ParaVertexBufferPool);
        virtual int InstallFields(CAttributeClass* pClass, bool bOverride);
        ATTRIBUTE_METHOD1(ParaVertexBufferPool, GetFullSizedBufferSize_s, int*)     { *p1 = cls->GetFullSizedBufferSize(); return S_OK; }
        ATTRIBUTE_METHOD1(ParaVertexBufferPool, SetFullSizedBufferSize_s, int)  { cls->SetFullSizedBufferSize((uint32)p1); return S_OK; }

        ATTRIBUTE_METHOD1(ParaVertexBufferPool, GetMaxPooledCount_s, int*)      { *p1 = cls->GetMaxPooledCount(); return S_OK; }
        ATTRIBUTE_METHOD1(ParaVertexBufferPool, SetMaxPooledCount_s, int)   { cls->SetMaxPooledCount((uint32)p1); return S_OK; }

        ATTRIBUTE_METHOD1(ParaVertexBufferPool, GetActiveBufferCount_s, int*)       { *p1 = cls->GetActiveBufferCount(); return S_OK; }
        ATTRIBUTE_METHOD1(ParaVertexBufferPool, GetTotalBufferCount_s, int*)        { *p1 = cls->GetTotalBufferCount(); return S_OK; }
        ATTRIBUTE_METHOD1(ParaVertexBufferPool, GetTotalBufferBytes_s, int*)        { *p1 = cls->GetTotalBufferBytes(); return S_OK; }

    public:

        uint32 GetFullSizedBufferSize() const;
        void SetFullSizedBufferSize(uint32 val);
        uint32 GetMaxPooledCount() const;
        void SetMaxPooledCount(uint32 val);
        size_t GetTotalBufferBytes(bool bRecalculate=false);

        int GetActiveBufferCount() const;
        int GetTotalBufferCount() const;

        ParaVertexBuffer* CreateBuffer(uint32 nBufferSize, DWORD dwFormat = 0, DWORD dwUsage = 0, D3DPOOL dwPool = D3DPOOL_MANAGED);

        void ReleaseBuffer(ParaVertexBuffer* pBuffer);

        void TickCache();

        virtual void Cleanup();
    
        virtual HRESULT RendererRecreated();
        
        
    protected:
        std::unordered_set<ParaVertexBuffer*> m_activeBuffers;
        std::unordered_set<ParaVertexBuffer*> m_unusedFullSizedBuffers;

        uint32 m_nFullSizedBufferSize;

        uint32 m_nMaxPooledCount;

        size_t m_nTotalBufferBytes;
        
    };


    class CVertexBufferPoolManager : public AssetManager < ParaVertexBufferPool, ParaVertexBufferPool, ParaVertexBufferPool >
    {
    public:
        CVertexBufferPoolManager(){};

        ParaVertexBufferPool* CreateGetPool(const std::string& name);

        static CVertexBufferPoolManager& GetInstance();
    public:
        void TickCache();

        size_t GetVertexBufferPoolTotalBytes();

    };

}