#ifndef TRITON_TVM_FFI_VALUE_H_
#define TRITON_TVM_FFI_VALUE_H_

#include "macro.h"
#include "type.h"
#include <tvm/ffi/any.h>
#include <tvm/ffi/object.h>

namespace triton_tvm_ffi {

class TypedValueObj : public tvm::ffi::Object {
public:
  TypedValueObj(Type type, const tvm::ffi::Any &value);
  TypedValueObj(Type type, tvm::ffi::Any &&value);
  TypedValueObj(const TypedValueObj &other) = default;
  TypedValueObj(TypedValueObj &&other) = default;
  TypedValueObj &operator=(const TypedValueObj &other) = default;
  TypedValueObj &operator=(TypedValueObj &&other) = default;
  TRITON_TVM_FFI_INLINE Type GetType() const { return type_; }
  TRITON_TVM_FFI_INLINE const tvm::ffi::Any &GetValue() const { return value_; }
  TVM_FFI_DECLARE_OBJECT_INFO_FINAL("triton_tvm_ffi.TypedValue", TypedValueObj,
                                    tvm::ffi::Object);

private:
  Type type_;
  tvm::ffi::Any value_;
};

class TypedValue : public tvm::ffi::ObjectRef {
public:
  TypedValue(Type type, const tvm::ffi::Any &value);
  TypedValue(Type type, tvm::ffi::Any &&value);
  using tvm::ffi::ObjectRef::ObjectRef;
  using tvm::ffi::ObjectRef::operator=;
  TRITON_TVM_FFI_INLINE Type GetType() const { return get()->GetType(); }
  TRITON_TVM_FFI_INLINE const tvm::ffi::Any &GetValue() const {
    return get()->GetValue();
  }
  TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE(TypedValue, tvm::ffi::ObjectRef,
                                                TypedValueObj);
};

tvm::ffi::Optional<TypedValue> MakeTypedValue(const tvm::ffi::String &type,
                                              const tvm::ffi::Any &value);

tvm::ffi::Array<TypedValue>
MakeTypedValues(const tvm::ffi::Array<tvm::ffi::String> &types,
                const tvm::ffi::Array<tvm::ffi::Any> &values);

} // namespace triton_tvm_ffi

#endif