// Copyright 2022 The Dawn Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "dawn/native/BlobCache.h"

#include "dawn/common/Assert.h"
#include "dawn/native/CacheKey.h"
#include "dawn/native/Instance.h"
#include "dawn/platform/DawnPlatform.h"

namespace dawn::native {

    CachedBlob::CachedBlob(size_t size) {
        if (size != 0) {
            Reset(size);
        }
    }

    bool CachedBlob::Empty() const {
        return mSize == 0;
    }

    const uint8_t* CachedBlob::Data() const {
        return mData.get();
    }

    uint8_t* CachedBlob::Data() {
        return mData.get();
    }

    size_t CachedBlob::Size() const {
        return mSize;
    }

    void CachedBlob::Reset(size_t size) {
        mSize = size;
        mData = std::make_unique<uint8_t[]>(size);
    }

    BlobCache::BlobCache(dawn::platform::CachingInterface* cachingInterface)
        : mCache(cachingInterface) {
    }

    CachedBlob BlobCache::Load(const CacheKey& key) {
        std::lock_guard<std::mutex> lock(mMutex);
        return LoadInternal(key);
    }

    void BlobCache::Store(const CacheKey& key, size_t valueSize, const void* value) {
        std::lock_guard<std::mutex> lock(mMutex);
        StoreInternal(key, valueSize, value);
    }

    void BlobCache::Store(const CacheKey& key, const CachedBlob& value) {
        Store(key, value.Size(), value.Data());
    }

    CachedBlob BlobCache::LoadInternal(const CacheKey& key) {
        CachedBlob result;
        if (mCache == nullptr) {
            return result;
        }
        const size_t expectedSize = mCache->LoadData(nullptr, key.data(), key.size(), nullptr, 0);
        if (expectedSize > 0) {
            result.Reset(expectedSize);
            const size_t actualSize =
                mCache->LoadData(nullptr, key.data(), key.size(), result.Data(), expectedSize);
            ASSERT(expectedSize == actualSize);
        }
        return result;
    }

    void BlobCache::StoreInternal(const CacheKey& key, size_t valueSize, const void* value) {
        ASSERT(value != nullptr);
        ASSERT(valueSize > 0);
        if (mCache == nullptr) {
            return;
        }
        mCache->StoreData(nullptr, key.data(), key.size(), value, valueSize);
    }

}  // namespace dawn::native