o Y۷i@sdZddlmZddlmZmZddlmZmZm Z m Z zddlm Z Wne y7GdddZ e Z Ynwe dZee ZGd d d Zeed ed dddZdddZdS) zFPublic helpers for describing dataclass-style defaults on FFI proxies.) annotations) _MISSING_TYPEMISSING)AnyCallableTypeVarcast)KW_ONLYc@seZdZdZdS)_KW_ONLY_SentinelN)__name__ __module__ __qualname__ __slots__r r r O/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/dataclasses/field.pyr sr _FieldValuec@s*eZdZdZdZdededdddZdS)Fielda](Experimental) Descriptor placeholder returned by :func:`tvm_ffi.dataclasses.field`. A ``Field`` mirrors the object returned by :func:`dataclasses.field`, but it is understood by :func:`tvm_ffi.dataclasses.c_class`. The decorator inspects the ``Field`` instances, records the ``default_factory`` and later replaces the field with a property that forwards to the underlying C++ attribute. Users should not instantiate ``Field`` directly - use :func:`field` instead, which guarantees that ``name`` and ``default_factory`` are populated in a way the decorator understands. )default_factoryinitkw_onlynameNTrrrrr str | Noner)Callable[[], _FieldValue] | _MISSING_TYPErboolrbool | _MISSING_TYPEreturnNonecCs||_||_||_||_dS)z0Do not call directly; use :func:`field` instead.Nr)selfrrrrr r r__init__4s  zField.__init__) rrrrrrrrrr)r r r__doc__rrrr r r rr%s rT)defaultrrrr!_FieldValue | _MISSING_TYPErrrrrrrcCsx|tur |tur tdt|tstd|tur(t|ts(tdt|j|tur0t|}t|||d}t t |S)a (Experimental) Declare a dataclass-style field on a :func:`c_class` proxy. Use this helper exactly like :func:`dataclasses.field` when defining the Python side of a C++ class. When :func:`c_class` processes the class body it replaces the placeholder with a property and arranges for ``default`` or ``default_factory`` to be respected by the synthesized ``__init__``. Parameters ---------- default A literal default value that populates the field when no argument is given. At most one of ``default`` or ``default_factory`` may be given. default_factory A zero-argument callable that produces the default. This matches the semantics of :func:`dataclasses.field` and is useful for mutable defaults such as ``list`` or ``dict``. init If ``True`` the field is included in the generated ``__init__``. If ``False`` the field is omitted from input arguments of ``__init__``. kw_only If ``True``, the field is a keyword-only argument in ``__init__``. If ``MISSING``, inherits from the class-level ``kw_only`` setting or from a preceding ``KW_ONLY`` sentinel annotation. Note ---- The decision to forward a field to the C++ ``__ffi_init__`` constructor depends on its configuration: * If ``init=True``, the field's value (from user input or defaults) is forwarded. * If ``init=False``: - With a ``default`` or ``default_factory``, its computed value is forwarded. The user cannot provide this value via Python ``__init__``. - Without a ``default`` or ``default_factory``, the field is *not* forwarded to C++ ``__ffi_init__`` and must be initialized by the C++ constructor. Returns ------- Field A placeholder object that :func:`c_class` will consume during class registration. Examples -------- ``field`` integrates with :func:`c_class` to express defaults the same way a Python ``dataclass`` would: .. code-block:: python @c_class("testing.TestCxxClassBase") class PyBase: v_i64: int v_i32: int = field(default=16) obj = PyBase(v_i64=4) obj.v_i32 # -> 16 Use ``kw_only=True`` to make a field keyword-only: .. code-block:: python @c_class("testing.TestCxxClassBase") class PyBase: v_i64: int v_i32: int = field(kw_only=True) obj = PyBase(4, v_i32=8) # v_i32 must be keyword z3Cannot specify both `default` and `default_factory`z`init` must be a boolz`kw_only` must be a bool, got )rrr) r ValueError isinstancer TypeErrortyper _make_default_factoryrrr)r!rrrretr r rfieldCsT  r)valuerCallable[[], Any]csdfdd }|S)z4Make a default factory that returns the given value.rrcsS)Nr r r*r rfactorysz&_make_default_factory..factoryN)rrr )r*r-r r,rr'sr'N) r!r"rrrrrrrr)r*rrr+)r __future__r dataclassesrrtypingrrrrr ImportErrorr rr& _KW_ONLY_TYPErr)r'r r r rs&    `