// Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
//
// 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.

#pragma once

#include "paddle/ap/include/axpr/ordered_dict.h"
#include "paddle/ap/include/memory/circlable_ref.h"

namespace ap::axpr {

template <typename ValueT>
using MutableOrderedDictImpl =
    OrderedDictImpl<ValueT, ValueT, axpr::Hash<ValueT>>;

template <typename ValueT>
struct MutableOrderedDict
    : public memory::CirclableRef<MutableOrderedDict<ValueT>,
                                  MutableOrderedDictImpl<ValueT>> {
  using Base = memory::CirclableRef<MutableOrderedDict<ValueT>,
                                    MutableOrderedDictImpl<ValueT>>;
  using Base::Base;
};

template <typename ValueT>
struct TypeImpl<MutableOrderedDict<ValueT>> : public std::monostate {
  using value_type = MutableOrderedDict<ValueT>;

  const char* Name() const { return "MutableOrderedDict"; }
};

};  // namespace ap::axpr