xref: /aosp_15_r20/external/android-nn-driver/CacheDataHandler.cpp (revision 3e777be0405cee09af5d5785ff37f7cfb5bee59a)

//
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "CacheDataHandler.hpp"

#include <log/log.h>

namespace armnn_driver
{

CacheDataHandler& CacheDataHandlerInstance()
{
    static CacheDataHandler instance;
    return instance;
}

void CacheDataHandler::Register(const HidlToken token, const size_t hashValue, const size_t cacheSize)
{
    if (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end()
                        && m_CacheDataMap.at(hashValue).GetToken() == token
                        && m_CacheDataMap.at(hashValue).GetCacheSize() == cacheSize)
    {
        ALOGV("CacheHandler::Register() Hash value has already registered.");
        return;
    }
    CacheHandle cacheHandle(token, cacheSize);
    m_CacheDataMap.insert({hashValue, cacheHandle});
}

bool CacheDataHandler::Validate(const HidlToken token, const size_t hashValue, const size_t cacheSize) const
{
    return (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end()
                             && m_CacheDataMap.at(hashValue).GetToken() == token
                             && m_CacheDataMap.at(hashValue).GetCacheSize() == cacheSize);
}

size_t CacheDataHandler::Hash(std::vector<uint8_t>& cacheData)
{
    std::size_t hash = cacheData.size();
    for (auto& i : cacheData)
    {
        hash = ((hash << 5) - hash) + i;
    }
    return hash;
}

size_t CacheDataHandler::GetCacheSize(HidlToken token)
{
    for (auto i = m_CacheDataMap.begin(); i != m_CacheDataMap.end(); ++i)
    {
        if (i->second.GetToken() == token)
        {
            return i->second.GetCacheSize();
        }
    }
    return 0;
}

void CacheDataHandler::Clear()
{
    m_CacheDataMap.clear();
}

} // armnn_driver