o i@spdZddlmZddlmZmZddlmZmZm Z m Z e dZ GdddZ eedd dddZ dddZdS)zFPublic helpers for describing dataclass-style defaults on FFI proxies.) annotations) _MISSING_TYPEMISSING)AnyCallableTypeVarcast _FieldValuec@s(eZdZdZdZdedddddZdS)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_factoryinitnameNTr r r r str | Noner )Callable[[], _FieldValue] | _MISSING_TYPEr boolreturnNonecCs||_||_||_dS)z0Do not call directly; use :func:`field` instead.Nr)selfr r r rV/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/tvm_ffi/dataclasses/field.py__init__*s zField.__init__)r rr rr rrr)__name__ __module__ __qualname____doc__ __slots__rrrrrrr s r T)defaultr r r_FieldValue | _MISSING_TYPEr rr rrcCsP|tur |tur tdt|tstd|turt|}t||d}tt|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__``. 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 z3Cannot specify both `default` and `default_factory`z`init` must be a bool)r r ) r ValueError isinstancer TypeError_make_default_factoryr rr )rr r retrrrfield7sC   r$valuerCallable[[], Any]csdfdd }|S)z4Make a default factory that returns the given value.rrcsS)Nrrr%rrfactorysz&_make_default_factory..factoryN)rrr)r%r(rr'rr"sr"N)rrr rr rrr )r%rrr&)r __future__r dataclassesrrtypingrrrrr r r$r"rrrrs M