o Ù°“iƒã@s¬dZddlmZddlmZer"ddlmZddlmZddlm Z ddl m Z ddl mZdd l mZGd d „d e ƒZed ƒGd d„deƒƒZedƒGdd„deƒƒZdS)zAccess path classes.é)Ú annotations)Ú TYPE_CHECKING)ÚSequence)ÚObject)ÚAny)ÚIntEnumé)Úregister_objectc@s(eZdZdZdZdZdZdZdZdZ dS) Ú AccessKindz(Kinds of access steps in an access path.rrééééN) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚATTRÚ ARRAY_ITEMÚMAP_ITEMÚ ATTR_MISSINGÚARRAY_ITEM_MISSINGÚMAP_ITEM_MISSING©rrúG/home/ubuntu/.local/lib/python3.10/site-packages/tvm_ffi/access_path.pyr &sr zffi.reflection.AccessStepc@s"eZdZUdZded<ded<dS)Ú AccessStepzAccess step container.ÚintÚkindrÚkeyN)rrrrÚ__annotations__rrrrr1s  rzffi.reflection.AccessPathcs*eZdZUdZded<ded<ded<erNedBd d „ƒZdCd d„ZdDdd„Z dEdd„Z dFdd„Z dDdd„Z dEdd„Z dFdd„ZdGdd„ZdHd!d"„ZdGd#d$„ZdI‡fd&d'„ ZedBd(d)„ƒZdJd+d,„ZdJd-d.„ZdKd/d0„ZdLd2d3„ZdLd4d5„ZdMd7d8„ZdMd9d:„ZdNdd?„ZdHd@dA„ZejZ‡Z S)OÚ AccessPatha6Access path container. It describes how to reach a nested attribute or item inside a complex FFI object by recording a sequence of steps (attribute, array index, or map key). It is primarily used by diagnostics to pinpoint structural mismatches. Examples -------- .. code-block:: python from tvm_ffi.access_path import AccessPath root = AccessPath.root() # Build a path equivalent to obj.layer.weight[2] p = root.attr("layer").attr("weight").array_item(2) assert isinstance(p, AccessPath) z Object | NoneÚparentzAccessStep | NoneÚsteprÚdepthÚreturncCódS©NrrrrrÚ_root[szAccessPath._rootÚ_1rcCr%r&r©Úselfr(rrrÚ_extend]ózAccessPath._extendÚstrcCr%r&rr)rrrÚ_attr^r,zAccessPath._attrcCr%r&rr)rrrÚ _array_item_r,zAccessPath._array_itemrcCr%r&rr)rrrÚ _map_item`r,zAccessPath._map_itemcCr%r&rr)rrrÚ _attr_missingar,zAccessPath._attr_missingcCr%r&rr)rrrÚ_array_item_missingbr,zAccessPath._array_item_missingcCr%r&rr)rrrÚ_map_item_missingcr,zAccessPath._map_item_missingÚboolcCr%r&rr)rrrÚ _is_prefix_ofdr,zAccessPath._is_prefix_ofúSequence[AccessStep]cCr%r&r©r*rrrÚ _to_stepser,zAccessPath._to_stepscCr%r&rr)rrrÚ _path_equalfr,zAccessPath._path_equalÚNonecstƒ ¡tdƒ‚)z>Disallow direct construction; use `AccessPath.root()` instead.zcAccessPath can't be initialized directly. Use AccessPath.root() to create a path to the root object)ÚsuperÚ__init__Ú ValueErrorr7©Ú __class__rrr<js ÿzAccessPath.__init__cCst ¡S)z•Create a root access path. Returns ------- AccessPath A path representing the root of an object graph. )r r'rrrrÚrootrs zAccessPath.rootÚothercCst|tƒsdS| |¡S)z*Return whether two access paths are equal.F©Ú isinstancer r9©r*rArrrÚ__eq__~s  zAccessPath.__eq__cCst|tƒsdS| |¡ S)z.Return whether two access paths are not equal.TrBrDrrrÚ__ne__„s  zAccessPath.__ne__cCó | |¡S)aPCheck if this access path is a prefix of another access path. Parameters ---------- other The access path to check if it is a prefix of this access path Returns ------- bool True if this access path is a prefix of the other access path, False otherwise )r5rDrrrÚ is_prefix_ofŠó zAccessPath.is_prefix_ofÚattr_keycCrG)aCreate an access path to the attribute of the current object. Parameters ---------- attr_key The key of the attribute to access Returns ------- AccessPath The extended access path )r.©r*rJrrrÚattršrIzAccessPath.attrcCrG)aCreate an access path that indicate an attribute is missing. Parameters ---------- attr_key The key of the attribute to access Returns ------- AccessPath The extended access path )r1rKrrrÚ attr_missingªrIzAccessPath.attr_missingÚindexcCrG)zûCreate an access path to the item of the current array. Parameters ---------- index The index of the item to access Returns ------- AccessPath The extended access path )r/©r*rNrrrÚ array_itemºrIzAccessPath.array_itemcCrG)aCreate an access path that indicate an array item is missing. Parameters ---------- index The index of the item to access Returns ------- AccessPath The extended access path )r2rOrrrÚarray_item_missingÊrIzAccessPath.array_item_missingrcCrG)zõCreate an access path to the item of the current map. Parameters ---------- key The key of the item to access Returns ------- AccessPath The extended access path )r0©r*rrrrÚmap_itemÚrIzAccessPath.map_itemcCrG)zúCreate an access path that indicate a map item is missing. Parameters ---------- key The key of the item to access Returns ------- AccessPath The extended access path )r3rRrrrÚmap_item_missingêrIzAccessPath.map_item_missingcCs| ¡S)z—Convert the access path to a list of access steps. Returns ------- access_steps The list of access steps )r8r7rrrÚto_stepsús zAccessPath.to_steps)r$r )r(rr$r )r(r-r$r )r(rr$r )r(rr$r )r(r r$r4)r$r6)r$r:)rArr$r4)rAr r$r4)rJr-r$r )rNrr$r )rrr$r )!rrrrrrÚ staticmethodr'r+r.r/r0r1r2r3r5r8r9r<r@rErFrHrLrMrPrQrSrTrUrÚ__hash__Ú __classcell__rrr>rr >s@                       r N)rÚ __future__rÚtypingrÚcollections.abcrÚtvm_ffirrÚenumrÚcoreÚregistryr r rr rrrrÚs