/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2026.
 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
 * CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */
#pragma once
#include "macros.h"
#include "probing_scheme.h"
#include "pair.h"
#include "detail/open_addressing/open_addressing_ref_impl.h"

namespace aclco {
template <typename Key,
          typename KeyEqual,
          typename ProbingScheme,
          typename StorageRef>

class StaticMapRef {
  using ImplType = OpenAddressingRefImpl<Key, KeyEqual, ProbingScheme, StorageRef>;

 public:
  using KeyType = Key;
  using ValueType = typename ImplType::ValueType;
  using PayloadType = typename ValueType::SecondType;
  using SizeType = typename ImplType::SizeType;
  // using hasher = typename ProbingScheme::hasher;

  static_assert(sizeof(ValueType) >= 4, "sizeof(Pair<Key, T>) must be at least 4 bytes.");

  COLLECTION_HOST_DEVICE explicit constexpr StaticMapRef(ValueType emptyValue,
                                                         KeyEqual const& predicate,
                                                         ProbingScheme const& probingScheme,
                                                         StorageRef storageRef) noexcept;

  template <typename Value>
  COLLECTION_SIMT_DEVICE bool Insert(Value value) noexcept;

  template <typename Value>
  COLLECTION_SIMT_DEVICE bool InsertOrAssign(Value value) noexcept;

  /** @brief 三态返回：SUCCESS=插入，DUPLICATE=assign，FAILED=表满。 */
  template <typename Value>
  COLLECTION_SIMT_DEVICE InsertResult InsertOrAssignVerdict(Value value) noexcept;

  /** @brief assign-only：key 已存在则更新 value 并返回 true；不存在返回 false（不插入）。 */
  template <typename Value>
  COLLECTION_SIMT_DEVICE bool Assign(Value value) noexcept;

  template <typename Value>
  COLLECTION_SIMT_DEVICE Pair<PayloadType, bool> InsertAndFind(Value value) noexcept;  

  template <typename ProbeKey>
  COLLECTION_SIMT_DEVICE bool Erase(ProbeKey key) noexcept;

  /** @brief 墓碑删除重载：删除时写入 erasedSlotValue 而非空值。 */
  template <typename ProbeKey>
  COLLECTION_SIMT_DEVICE bool Erase(ProbeKey key, ValueType erasedSlotValue) noexcept;

  template <typename ProbeKey>
  COLLECTION_SIMT_DEVICE PayloadType Find(ProbeKey key) noexcept;

  template <typename ProbeKey>
  COLLECTION_SIMT_DEVICE bool Contains(ProbeKey key) noexcept;
  
  template <typename ProbeKey>
  COLLECTION_SIMT_DEVICE SizeType Count(ProbeKey key) noexcept;

  template <typename ProbeKey, typename CallbackOp>
  COLLECTION_SIMT_DEVICE void ForEach(ProbeKey key, CallbackOp& callback_op) noexcept;

 private:
  ImplType impl_;
};
}

#include "detail/static_map/static_map_ref.inl"