YiAssetManager.h

Go to the documentation of this file.

// © You i Labs Inc. 2000-2017. All rights reserved.
#ifndef _YI_ASSET_MANAGER_H_
#define _YI_ASSET_MANAGER_H_

#include "asset/YiAsset.h"
#include "asset/YiAssetCacheStrategy.h"
#include "asset/YiAssetShaderObject.h"
#include "asset/YiAssetShaderProgram.h"
#include "renderer/YiGPUObject.h"
#include "thread/YiRecursiveMutex.h"
#include "utility/YiString.h"
#include "utility/YiStringUtilities.h"

class CYIRemoteAssetCatalog;
struct CYIEmbeddedShader;

#ifndef YI_USE_CASE_INSENSITIVE_ASSET_LOCATING

    #define YI_USE_CASE_INSENSITIVE_ASSET_LOCATING YI_UWP && YI_XBOX_ONE
#endif //YI_USE_CASE_INSENSITIVE_ASSET_LOCATING

class CYIAssetManager
{
public:
    static const CYIString YI_NORMALIZED_QUAD_VBO;          
    static const CYIString YI_NORMALIZED_QUAD_YFLIP_VBO;    
    static const CYIString YI_TEXT_ATLAS_IBO;               
    static const CYIString YI_VSHADER_3D;                   
    static const CYIString YI_FSHADER_3D_BGR;               
    static const CYIString YI_PROGRAM_3D_BGR;               
    static const CYIString YI_FSHADER_3D_RGB;               
    static const CYIString YI_PROGRAM_3D_RGB;               
    static const CYIString YI_VSHADER_3D_YUV;               
    static const CYIString YI_FSHADER_3D_YUV;               
    static const CYIString YI_PROGRAM_3D_YUV;               
    static const CYIString YI_VSHADER_3D_YUV_SEPARATE;      
    static const CYIString YI_FSHADER_3D_YUV_SEPARATE;      
    static const CYIString YI_PROGRAM_3D_YUV_SEPARATE;      
    static const CYIString YI_CHECKERBOARD_TEXTURE;         
    static const CYIString YI_FSHADER_3D_SOLID;             
    static const CYIString YI_PROGRAM_3D_SOLID;             
    static const CYIString YI_VSHADER_3D_TEXT;              
    static const CYIString YI_FSHADER_3D_TEXT;              
    static const CYIString YI_PROGRAM_3D_TEXT;              
    static const CYIString YI_VSHADER_SDF_TEXT;             
    static const CYIString YI_FSHADER_SDF_TEXT;             
    static const CYIString YI_PROGRAM_SDF_TEXT;             
    static const CYIString YI_VSHADER_SDF_APPROX_TEXT;             
    static const CYIString YI_FSHADER_SDF_APPROX_TEXT;             
    static const CYIString YI_PROGRAM_SDF_APPROX_TEXT;             
    static const CYIString YI_VSHADER_VERTEX_COLOR;         
    static const CYIString YI_FSHADER_VERTEX_COLOR;         
    static const CYIString YI_PROGRAM_VERTEX_COLOR;         
    static const CYIString YI_CROSSHAIR_TEXTURE;            
    static const CYIString YI_SMALL_CROSSHAIR_TEXTURE;      
    static const CYIString YI_VSHADER_3D_EFFECT_BLUR_H;     
    static const CYIString YI_FSHADER_3D_EFFECT_BLUR_H;     
    static const CYIString YI_PROGRAM_3D_EFFECT_BLUR_H;     
    static const CYIString YI_VSHADER_3D_EFFECT_BLUR_V;     
    static const CYIString YI_FSHADER_3D_EFFECT_BLUR_V;     
    static const CYIString YI_PROGRAM_3D_EFFECT_BLUR_V;     
    static const CYIString YI_VSHADER_3D_EFFECT_FLIP_COPY;  
    static const CYIString YI_FSHADER_3D_EFFECT_FLIP_COPY;  
    static const CYIString YI_PROGRAM_3D_EFFECT_FLIP_COPY;  
    static const CYIString YI_VSHADER_3D_EFFECT_TINT;       
    static const CYIString YI_FSHADER_3D_EFFECT_TINT;       
    static const CYIString YI_PROGRAM_3D_EFFECT_TINT;       
    static const CYIString YI_VSHADER_3D_EFFECT_TINT_FLIP_COPY;       
    static const CYIString YI_FSHADER_3D_EFFECT_TINT_FLIP_COPY;       
    static const CYIString YI_PROGRAM_3D_EFFECT_TINT_FLIP_COPY;       
    static const CYIString YI_VSHADER_2D_GEOMETRY_BATCH;    
    static const CYIString YI_FSHADER_2D_GEOMETRY_BATCH;    
    static const CYIString YI_PROGRAM_2D_GEOMETRY_BATCH;    
    static const CYIString YI_VSHADER_2D_NANOVG;            
    static const CYIString YI_FSHADER_2D_NANOVG;            
    static const CYIString YI_PROGRAM_2D_NANOVG;            
    friend class CYIAsset;

    CYIAssetManager();
    virtual ~CYIAssetManager();

    void AddCacheStrategy(std::unique_ptr<CYIAssetCacheStrategy> pStrategy);

    bool AddAsset(const std::shared_ptr<CYIAsset> &pAsset);

    void ExecuteCacheStrategies();

    void DumpAssetList() const;
    
    std::vector<std::shared_ptr<CYIAsset>> GetAssets() const;

    std::shared_ptr<CYIAsset> GetAsset(const CYIString &name);

    std::shared_ptr<CYIAsset> GetAssetFromPath(const CYIString &path);

    std::shared_ptr<CYIRemoteAssetCatalog> GetRemoteAssetCatalog(uint64_t uCatalogID) const;

    std::vector<std::shared_ptr<CYIRemoteAssetCatalog>> GetRemoteAssetCatalogs() const;

    void InvalidateHardwareAssets();

    void LoadShaders();

    void RemoveAllAssets();

    void RemoveAsset(const std::shared_ptr<CYIAsset> &pAsset);

    void RemoveAssetsOfType(const CYIRuntimeTypeInfo &assetType);

    std::unique_ptr<CYIAssetCacheStrategy> RemoveCacheStrategy(const CYIAssetCacheStrategy *pStrategy);

    void RestoreHardwareAssets();
    
    bool IsSystemAsset(const std::shared_ptr<const CYIAsset> &pAsset);

private:
    typedef std::map<ssize_t, std::weak_ptr<CYIAsset>> AssetMap;
    typedef AssetMap::iterator                          AssetMapIterator;
    typedef AssetMap::const_iterator                    AssetMapConstIterator;

    typedef std::map<ssize_t, std::shared_ptr<CYIAsset>> SystemAssetMap;
    typedef SystemAssetMap::iterator                    SystemAssetMapIterator;
    typedef SystemAssetMap::const_iterator              SystemAssetMapConstIterator;

#if YI_USE_CASE_INSENSITIVE_ASSET_LOCATING
    typedef std::map<CYIString, std::weak_ptr<CYIAsset>, CYICaseInsensitiveLessThanComparator> AssetStringMap;
#else
    typedef std::map<CYIString, std::weak_ptr<CYIAsset>> AssetStringMap;
#endif

    typedef AssetStringMap::iterator        AssetStringMapIterator;
    typedef AssetStringMap::const_iterator  AssetStringMapConstIterator;

    ssize_t GenerateNewID();

    bool RenameAssetInNamedAssetsMap(const CYIString &oldName, const CYIString &newName, ssize_t nID);

    void AddAssetToMap(const std::shared_ptr<CYIAsset> &pAsset);

    // utility functions for shader program assets created during AssetManager construction
    std::shared_ptr<CYIAssetShaderProgram> AddAssetShaderProgram(const CYIString &shaderProgramName,
                                                                 const CYIString &vertexShaderName,
                                                                 const CYIEmbeddedShader &vertexShaderSource,
                                                                 const CYIString &fragmentShaderName,
                                                                 const CYIEmbeddedShader &fragmentShaderSource);

    std::shared_ptr<CYIAssetShaderObject> AddAssetShaderObject(CYIAssetShaderObject::SHADER_TYPE shaderType,
                                                               const CYIString &shaderName,
                                                               const CYIEmbeddedShader &shaderSource);

    void AddBuiltInTextureAssets();
    void AddAssetTexturesFromRawImages();

    mutable CYIRecursiveMutex           m_Mutex;
    AssetMap                            m_Assets;
    SystemAssetMap                      m_SystemAssets;
    AssetStringMap                      m_NamedAssets;
    AssetStringMap                      m_AssetLocations;
    std::list<std::unique_ptr<CYIAssetCacheStrategy>> m_CacheStrategyList;
    std::map<uint64_t, std::weak_ptr<CYIRemoteAssetCatalog>> m_AssetCatalogs;
    ssize_t m_NextID;

    YI_DISALLOW_COPY_AND_ASSIGN(CYIAssetManager)
};

#endif // _YI_ASSET_MANAGER_H_