/*
 * 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 PANDA_RUNTIME_INCLUDE_OBJECT_ACCESSOR_H_
#define PANDA_RUNTIME_INCLUDE_OBJECT_ACCESSOR_H_

#include <cstddef>

#include "runtime/mem/gc/gc_barrier_set.h"

namespace panda {

class ObjectHeader;
class Field;
class ManagedThread;

class ObjectAccessor {
public:
    template <class T, bool is_volatile = false>
    static T GetPrimitive(const void *obj, size_t offset)
    {
        return Get<T, is_volatile>(obj, offset);
    }

    template <class T, bool is_volatile = false>
    static void SetPrimitive(void *obj, size_t offset, T value)
    {
        Set<T, is_volatile>(obj, offset, value);
    }

    template <bool is_volatile = false, bool need_read_barrier = true, bool is_dyn = false>
    static ObjectHeader *GetObject(const void *obj, size_t offset);

    template <bool is_volatile = false, bool need_write_barrier = true, bool is_dyn = false>
    static void SetObject(void *obj, size_t offset, ObjectHeader *value);

    template <class T>
    static T GetFieldPrimitive(const void *obj, const Field &field);

    template <class T>
    static void SetFieldPrimitive(void *obj, const Field &field, T value);

    template <bool need_read_barrier = true, bool is_dyn = false>
    static ObjectHeader *GetFieldObject(const void *obj, const Field &field);

    template <bool need_write_barrier = true, bool is_dyn = false>
    static void SetFieldObject(void *obj, const Field &field, ObjectHeader *value);

    // Pass thread parameter to speed up interpreter
    template <bool is_volatile = false, bool need_read_barrier = true, bool is_dyn = false>
    static ObjectHeader *GetObject(const ManagedThread *thread, const void *obj, size_t offset);

    template <bool is_volatile = false, bool need_write_barrier = true, bool is_dyn = false>
    static void SetObject(const ManagedThread *thread, void *obj, size_t offset, ObjectHeader *value);

    template <bool need_read_barrier = true, bool is_dyn = false>
    static ObjectHeader *GetFieldObject(ManagedThread *thread, const void *obj, const Field &field);

    template <bool need_write_barrier = true, bool is_dyn = false>
    static void SetFieldObject(ManagedThread *thread, void *obj, const Field &field, ObjectHeader *value);

    template <class T>
    static T GetFieldPrimitive(const void *obj, size_t offset, std::memory_order memory_order);

    template <class T>
    static void SetFieldPrimitive(void *obj, size_t offset, T value, std::memory_order memory_order);

    template <bool need_read_barrier = true, bool is_dyn = false>
    static ObjectHeader *GetFieldObject(const void *obj, int offset, std::memory_order memory_order);

    template <bool need_write_barrier = true, bool is_dyn = false>
    static void SetFieldObject(void *obj, size_t offset, ObjectHeader *value, std::memory_order memory_order);

    template <typename T>
    static std::pair<bool, T> CompareAndSetFieldPrimitive(void *obj, size_t offset, T old_value, T new_value,
                                                          std::memory_order memory_order, bool strong);

    template <bool need_write_barrier = true, bool is_dyn = false>
    static std::pair<bool, ObjectHeader *> CompareAndSetFieldObject(void *obj, size_t offset, ObjectHeader *old_value,
                                                                    ObjectHeader *new_value,
                                                                    std::memory_order memory_order, bool strong);

    template <typename T>
    static T GetAndSetFieldPrimitive(void *obj, size_t offset, T value, std::memory_order memory_order);

    template <bool need_write_barrier = true, bool is_dyn = false>
    static ObjectHeader *GetAndSetFieldObject(void *obj, size_t offset, ObjectHeader *value,
                                              std::memory_order memory_order);

    template <typename T>
    static T GetAndAddFieldPrimitive(void *obj, size_t offset, T value, std::memory_order memory_order);

    template <typename T>
    static T GetAndBitwiseOrFieldPrimitive(void *obj, size_t offset, T value, std::memory_order memory_order);

    template <typename T>
    static T GetAndBitwiseAndFieldPrimitive(void *obj, size_t offset, T value, std::memory_order memory_order);

    template <typename T>
    static T GetAndBitwiseXorFieldPrimitive(void *obj, size_t offset, T value, std::memory_order memory_order);

    template <class T>
    static inline void SetDynPrimitive(void *obj, size_t offset, T value)
    {
        auto *addr = reinterpret_cast<T *>(ToUintPtr(obj) + offset);
        ASSERT(IsInObjectsAddressSpace(ToUintPtr(addr)));
        *addr = value;
    }

    template <bool need_write_barrier = true>
    static inline void SetDynObject(const ManagedThread *thread, void *obj, size_t offset, ObjectHeader *value);

    template <class T>
    static inline T GetDynValue(const void *obj, size_t offset)
    {
        auto *addr = reinterpret_cast<T *>(ToUintPtr(obj) + offset);
        ASSERT(IsInObjectsAddressSpace(ToUintPtr(addr)));
        return *addr;
    }

private:
    template <class T, bool is_volatile>
    static T Get(const void *obj, size_t offset)
    {
        auto *addr = reinterpret_cast<T *>(reinterpret_cast<uintptr_t>(obj) + offset);
        ASSERT(IsInObjectsAddressSpace(ToUintPtr(addr)));
        if (is_volatile) {
            return reinterpret_cast<const std::atomic<T> *>(addr)->load(std::memory_order_seq_cst);
        }
        return reinterpret_cast<const std::atomic<T> *>(addr)->load(std::memory_order_relaxed);
    }

    template <class T, bool is_volatile>
    static void Set(void *obj, size_t offset, T value)
    {
        auto *addr = reinterpret_cast<T *>(reinterpret_cast<uintptr_t>(obj) + offset);
        ASSERT(IsInObjectsAddressSpace(ToUintPtr(addr)));
        if (is_volatile) {
            return reinterpret_cast<std::atomic<T> *>(addr)->store(value, std::memory_order_seq_cst);
        }
        return reinterpret_cast<std::atomic<T> *>(addr)->store(value, std::memory_order_relaxed);
    }

    template <class T>
    static T Get(const void *obj, size_t offset, std::memory_order memory_order)
    {
        auto *addr = reinterpret_cast<T *>(reinterpret_cast<uintptr_t>(obj) + offset);
        ASSERT(IsInObjectsAddressSpace(ToUintPtr(addr)));
        return reinterpret_cast<const std::atomic<T> *>(addr)->load(memory_order);
    }

    template <class T>
    static void Set(void *obj, size_t offset, T value, std::memory_order memory_order)
    {
        auto *addr = reinterpret_cast<T *>(reinterpret_cast<uintptr_t>(obj) + offset);
        ASSERT(IsInObjectsAddressSpace(ToUintPtr(addr)));
        return reinterpret_cast<std::atomic<T> *>(addr)->store(value, memory_order);
    }

    static mem::GCBarrierSet *GetBarrierSet();

    static mem::GCBarrierSet *GetBarrierSet(const ManagedThread *thread);

    static mem::BarrierType GetPreBarrierType(const ManagedThread *thread);

    static mem::BarrierType GetPostBarrierType(const ManagedThread *thread);
};

}  // namespace panda

#endif  // PANDA_RUNTIME_INCLUDE_OBJECT_ACCESSOR_H_