/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * 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.
 */

#ifndef ECMASCRIPT_IC_PROPERTIES_CACHE_H
#define ECMASCRIPT_IC_PROPERTIES_CACHE_H

#include <array>

#include "ecmascript/js_hclass.h"
#include "ecmascript/js_tagged_value.h"
#include "ecmascript/ecma_macros.h"

namespace panda::ecmascript {
class EcmaVM;
class PropertiesCache {
public:
    inline int Get(JSHClass *jsHclass, JSTaggedValue key)
    {
        int hash = Hash(jsHclass, key);
        PropertyKey &prop = keys_[hash];
        if ((prop.hclass_ == jsHclass) && (prop.key_ == key)) {
            return keys_[hash].results_;
        }
        return NOT_FOUND;
    }
    inline void Set(JSHClass *jsHclass, JSTaggedValue key, int index)
    {
        int hash = Hash(jsHclass, key);
        PropertyKey &prop = keys_[hash];
        prop.hclass_ = jsHclass;
        prop.key_ = key;
        keys_[hash].results_ = index;
    }
    inline void Clear()
    {
        for (auto &key : keys_) {
            key.hclass_ = nullptr;
        }
    }

    static const int NOT_FOUND = -1;
    static const uint32_t CACHE_LENGTH_BIT = 10;
    static const uint32_t CACHE_LENGTH = (1U << CACHE_LENGTH_BIT);
    static const uint32_t CACHE_LENGTH_MASK = CACHE_LENGTH - 1;

    struct PropertyKey : public base::AlignedStruct<JSTaggedValue::TaggedTypeSize(),
                                                    base::AlignedPointer,
                                                    JSTaggedValue,
                                                    base::AlignedUint32> {
        enum class Index : size_t {
            HclassIndex = 0,
            KeyIndex,
            ResultsIndex,
            NumOfMembers
        };
        static size_t GetHclassOffset(bool isArch32 = false)
        {
            return GetOffset<static_cast<size_t>(Index::HclassIndex)>(isArch32);
        }

        static size_t GetKeyOffset(bool isArch32 = false)
        {
            return GetOffset<static_cast<size_t>(Index::KeyIndex)>(isArch32);
        }

        static size_t GetResultsOffset(bool isArch32 = false)
        {
            return GetOffset<static_cast<size_t>(Index::ResultsIndex)>(isArch32);
        }

        static size_t GetPropertyKeySize()
        {
            return static_cast<size_t>(Index::NumOfMembers) * static_cast<size_t>(JSTaggedValue::TaggedTypeSize());
        }
        
        static_assert(static_cast<size_t>(Index::NumOfMembers) == NumOfTypes);
        alignas(EAS) JSHClass *hclass_ {nullptr};
        alignas(EAS) JSTaggedValue key_ {JSTaggedValue::Hole()};
        alignas(EAS) int results_ {NOT_FOUND};
    };

private:
    PropertiesCache()
    {
        for (uint32_t i = 0; i < CACHE_LENGTH; ++i) {
            keys_[i].hclass_ = nullptr;
            keys_[i].key_ = JSTaggedValue::Hole();
            keys_[i].results_ = NOT_FOUND;
        }
    }
    ~PropertiesCache() = default;

    // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)

    static inline int Hash(JSHClass *cls, JSTaggedValue key)
    {
        uint32_t clsHash = static_cast<uint32_t>(reinterpret_cast<uintptr_t>(cls)) >> 3U;  // skip 8bytes
        uint32_t keyHash = key.GetKeyHashCode();
        return static_cast<int>((clsHash ^ keyHash) & CACHE_LENGTH_MASK);
    }

    std::array<PropertyKey, CACHE_LENGTH> keys_{};

    friend class EcmaContext;
};
}  // namespace panda::ecmascript
#endif  // ECMASCRIPT_IC_PROPERTIES_CACHE_H