resolverContextCache.h

Go to the documentation of this file.

#ifndef AR_AYONUSDRESOLVER_RESOLVER_CONTEXT_CACHE_H
#define AR_AYONUSDRESOLVER_RESOLVER_CONTEXT_CACHE_H
 
#include "../cache/assetIdentifierDef.h"
#include "AyonCppApi.h"
 
#include <nlohmann/json.hpp>
 
#include <filesystem>
#include <memory>
#include <optional>
#include <shared_mutex>
#include <string>
#include <unordered_map>
 
PXR_NAMESPACE_USING_DIRECTIVE
 
enum class CacheName { AYONCACHE, COMMONCACHE };
 
class PinningFileHandler {
    public:
        PinningFileHandler(const std::string &pinningFilePath,
                           const std::unordered_map<std::string, std::string> &rootReplaceData);
        ~PinningFileHandler() = default;
 
        AssetIdentifier getAssetData(const std::string &resolveKey);
 
    private:
        std::filesystem::path m_pinningFilePath;
        nlohmann::json m_pinningFileData;
 
        std::unordered_map<std::string, std::string> m_rootReplaceData;
};
 
 
class ResolverContextCache {
    public:
        ResolverContextCache();
        ~ResolverContextCache();
 
        void insert(AssetIdentifier &sourceAssetIdent);
 
        void migratePreCacheIntoAyonCache();
 
        AssetIdentifier getAsset(const std::string &assetIdentifier, const CacheName selectedCache, const bool isAyonPath);
 
        void setCacheFromPinningFile(const std::string &pinningFilePath);
 
        void removeCachedObject(const std::string &key);
 
        void removeCachedObject(const std::string &key, const CacheName selectedCache);
 
        void ClearCache();
 
        void printCache() const;
 
        bool isCacheStatic() const;
 
    private:
        std::unordered_set<AssetIdentifier, AssetIdentifierHash> m_PreCache;
        std::unordered_set<AssetIdentifier, AssetIdentifierHash> m_AyonCache;
        std::unordered_set<AssetIdentifier, AssetIdentifierHash> m_CommonCache;
 
        mutable std::shared_mutex m_PreCacheSharedMutex;
        mutable std::shared_mutex m_AyonCacheSharedMutex;
        mutable std::shared_mutex m_CommonCacheSharedMutex;
 
        std::optional<std::unique_ptr<AyonApi>> m_ayon;
        bool m_staticCache;
 
        std::optional<PinningFileHandler> m_pinningFileHandler;
};
 
#endif   // AR_AYONUSDRESOLVER_RESOLVER_CONTEXT_CACHE_H