YiAssetLoader.h

Go to the documentation of this file.

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

#include "asset/YiAssetLocator.h"
#include "binding/YiScriptableObject.h"
#include "signal/YiSignalHandler.h"
#include "thread/YiMutex.h"

class CYIDecoderThread;
class CYIAsset;
class CYIAssetDecoder;
class CYIAssetLoadParams;
class CYILoadThreadManager;

class CYIAssetLoader : public CYISignalHandler, public CYIScriptableObject
{
    friend class CYIDecoderThread;

public:

    class DecodeListener
    {
    public:
        virtual ~DecodeListener() {}

        virtual void OnDecodeComplete(int32_t nJobID, std::shared_ptr<CYIAsset> pAsset, int32_t nLoadStatus, void *pListenerPrivate) = 0;
    };

    CYIAssetLoader();

    virtual ~CYIAssetLoader();

    void AddDecoder(std::unique_ptr<CYIAssetDecoder> pDecoder);

    std::unique_ptr<CYIAssetDecoder> RemoveDecoder(const CYIAssetDecoder *pDecoder);

    static void SetAssetLocator(const CYIAssetLocator &rLocator);

    static const CYIAssetLocator &GetAssetLocator();


    uint32_t GetDecoderCount();

    std::shared_ptr<CYIAsset> Load(const CYIRuntimeTypeInfo &assetType, const CYIString &filename, CYIAssetLoadParams *pLoadParams = nullptr);

    std::shared_ptr<CYIAsset> LoadFromPath(const CYIRuntimeTypeInfo &assetType, const CYIString &path, CYIAssetLoadParams *pLoadParams = nullptr);

    std::shared_ptr<CYIAsset> LoadFromPath(const CYIString &path, CYIAssetLoadParams *pLoadParams = nullptr);

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

    std::shared_ptr<CYIAsset> Load(const CYIRuntimeTypeInfo &assetType, const uint8_t *pData, uint32_t nDataSize, CYIAssetLoadParams *pLoadParams = nullptr);

    int32_t LoadAsync(const CYIRuntimeTypeInfo &assetType, const CYIString &filename, CYIAssetLoadParams *pLoadParams, DecodeListener *pListener, void *pListenerPrivate);

    int32_t LoadAsyncFromPath(const CYIRuntimeTypeInfo &assetType, const CYIString &path, CYIAssetLoadParams *pLoadParams, DecodeListener *pListener, void *pListenerPrivate);

    int32_t LoadAsyncFromPath(const CYIString &path, CYIAssetLoadParams *pLoadParams, DecodeListener *pListener, void *pListenerPrivate);

    int32_t LoadAsync(const CYIRuntimeTypeInfo &assetType, const uint8_t *pData, uint32_t nDataSize, CYIAssetLoadParams *pLoadParams, DecodeListener *pListener, void *pListenerPrivate);

    void AbortAsyncLoad(int32_t nDecodeTaskID);

    uint32_t GetAsyncJobsCount();

    CYIString ProbeExtension(const CYIRuntimeTypeInfo &assetType, const uint8_t *pData, uint32_t nDataSize) const;

    CYIString ProbeExtension(const uint8_t *pData, uint32_t nDataSize) const;

private:
    CYIAssetDecoder *GetDecoder(const CYIRuntimeTypeInfo &assetType, const CYIString &path, const CYIAssetLoadParams *pLoadParams = nullptr);
    CYIAssetDecoder *GetDecoder(const CYIRuntimeTypeInfo &assetType, const uint8_t *pData, uint32_t nDataSize, const CYIAssetLoadParams *pLoadParams = nullptr);
    CYIAssetDecoder *GetDecoder(const std::shared_ptr<CYIAsset> &pAsset);

    int32_t StartAsyncLoad(const CYIRuntimeTypeInfo &assetType,
                           const CYIString &path,
                           const uint8_t *pData,
                           uint32_t nDataSize,
                           CYIAssetLoadParams *pLoadParams,
                           DecodeListener *pListener,
                           void *pListenerPrivate);

    void OnJobComplete(int32_t nJobID);

    static CYIAssetLoader *m_pSingleton;

    CYIMutex m_jobsMutex;
    std::vector<std::unique_ptr<CYIAssetDecoder>> m_decoders;
    std::map<int32_t, CYIDecoderThread *> m_decodingJobs;

    static CYIAssetLocator m_AssetLocator;

    int32_t m_nNextDecodingJobID;

    std::unique_ptr<CYILoadThreadManager> m_pLoadManager;

    YI_DISALLOW_COPY_AND_ASSIGN(CYIAssetLoader)
};

#endif // _YI_ASSET_LOADER_H_