o ٰi@sdZddlmZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZdd lmZmZmZmZd d lmZd d lmZe d Zeefd dd ddZd!ddZdS)"aHelpers for mirroring registered C++ FFI types with Python dataclass syntax. The :func:`c_class` decorator is the primary entry point. It inspects the reflection metadata that the C++ runtime exposes via the TVM FFI registry and turns it into Python ``dataclass``-style descriptors: annotated attributes become properties that forward to the underlying C++ object, while an ``__init__`` method is synthesized to call the FFI constructor when requested. ) annotationsN)Callable)InitVar)ClassVarTypeTypeVar get_originget_type_hints)dataclass_transform) TypeFieldTypeInfo+_lookup_or_register_type_info_from_type_key _set_type_cls)_utils)field _InputClsType)field_specifiersTtype_keystrinitboolreturn4Callable[[Type[_InputClsType]], Type[_InputClsType]]csdfdd }|S)a (Experimental) Create a dataclass-like proxy for a C++ class registered with TVM FFI. The decorator reads the reflection metadata that was registered on the C++ side using ``tvm::ffi::reflection::ObjectDef`` and binds it to the annotated attributes in the decorated Python class. Each field defined in C++ becomes a property on the Python class, and optional default values can be provided with :func:`tvm_ffi.dataclasses.field` in the same way as Python's native ``dataclasses.field``. The intent is to offer a familiar dataclass authoring experience while still exposing the underlying C++ object. The ``type_key`` of the C++ class must match the string passed to :func:`c_class`, and inheritance relationships are preserved-subclasses registered in C++ can subclass the Python proxy defined for their parent. Parameters ---------- type_key The reflection key that identifies the C++ type in the FFI registry, e.g. ``"testing.MyClass"`` as registered in ``src/ffi/extra/testing.cc``. init If ``True`` and the Python class does not define ``__init__``, an initializer is auto-generated that mirrors the reflected constructor signature. The generated initializer calls the C++ ``__init__`` function registered with ``ObjectDef`` and invokes ``__post_init__`` if it exists on the Python class. Returns ------- Callable[[type], type] A class decorator that materializes the final proxy class. Examples -------- Register the C++ type and its fields with TVM FFI: .. code-block:: c++ TVM_FFI_STATIC_INIT_BLOCK() { namespace refl = tvm::ffi::reflection; refl::ObjectDef() .def_static("__init__", [](int64_t v_i64, int32_t v_i32, double v_f64, float v_f32) -> Any { return ObjectRef(ffi::make_object( v_i64, v_i32, v_f64, v_f32)); }) .def_rw("v_i64", &MyClass::v_i64) .def_rw("v_i32", &MyClass::v_i32) .def_rw("v_f64", &MyClass::v_f64) .def_rw("v_f32", &MyClass::v_f32); } Mirror the same structure in Python using dataclass-style annotations: .. code-block:: python from tvm_ffi.dataclasses import c_class, field @c_class("example.MyClass") class MyClass: v_i64: int v_i32: int v_f64: float = field(default=0.0) v_f32: float = field(default_factory=lambda: 1.0) obj = MyClass(v_i64=4, v_i32=8) obj.v_f64 = 3.14 # transparently forwards to the underlying C++ object super_type_clsType[_InputClsType]rcs~od|jvt}|jdusJt|||_|jD]}t||qr,t||nd}tj||d|id}t |||S)N__init__) type_infoclsmethods) __dict__rparent_type_info_inspect_c_class_fieldsfieldsrfill_dataclass_field method_inittype_info_to_clsr)rr type_fieldfn_inittype_clsrrO/home/ubuntu/.local/lib/python3.10/site-packages/tvm_ffi/dataclasses/c_class.py decoratorxs   zc_class..decoratorN)rrrrr,)rrr.r,r+r-c_class*sNr/r*typerr list[TypeField]c stjdkr t|ddnt|fddt|diD}dd|jD}g}|D]$\}}|dr8q.||d}|durMt d |d |d | |q.|rld d d| D}t d|d|d|S)N) T)include_extrascs*i|]}t|ttfvr||qSr,)rrr).0nametype_hints_resolvedr,r- s z+_inspect_c_class_fields..__annotations__cSsi|]}|j|qSr,r6r5fr,r,r-r9s __tvm_ffizExtraneous field `.z#`. Defined in Python but not in C++z, css|] }d|jdVqdS)`Nr;r<r,r,r- sz*_inspect_c_class_fields..zMissing fields in `z`: z". Defined in C++ but not in Python) sys version_infor getattrkeysr$items startswithpop ValueErrorappendjoinvalues) r*r type_hints_pytype_fields_cxx type_fields field_name _field_ty_pyr( extra_fieldsr,r7r-r#s0      r#)T)rrrrrr)r*r0rr rr1)__doc__ __future__rrBcollections.abcr dataclassesrtypingrrrrr typing_extensionsr corer r rrrrrr/r#r,r,r,r-s       d