`idZddlmZddlmZerddlmZddlmZddlm Z ddl m Z ddl mZdd l mZGd d e Zed Gd deZedGddeZdS)zAccess path classes.) annotations) TYPE_CHECKING)Sequence)Object)Any)IntEnum)register_objectc*eZdZdZdZdZdZdZdZdZ dS) AccessKindz(Kinds of access steps in an access path.rr N) __name__ __module__ __qualname____doc__ATTR ARRAY_ITEMMAP_ITEM ATTR_MISSINGARRAY_ITEM_MISSINGMAP_ITEM_MISSINGg/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/tvm_ffi/access_path.pyr r &s622 DJHLrr zffi.reflection.AccessStepc(eZdZUdZded<ded<dS) AccessStepzAccess step container.intkindrkeyN)rrrr__annotations__rrrrr1s*  III HHHHHrrzffi.reflection.AccessPathc&eZdZUdZded<ded<ded<er4ed+d Zd,d Zd-dZ d.dZ d/dZ d-dZ d.dZ d/dZd0dZd1dZd0dZd2fd Zed+dZd3dZd3dZd4d Zd5d"Zd5d#Zd6d%Zd6d&Zd7d(Zd7d)Zd1d*ZejZxZ S)8 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 | NoneparentzAccessStep | Nonestepr depthreturncdSNrrrr_rootzAccessPath._root[s#&3r_1rcdSr+rselfr-s r_extendzAccessPath._extend]rstrcdSr+rr/s r_attrzAccessPath._attr^r2rcdSr+rr/s r _array_itemzAccessPath._array_item_r2rrcdSr+rr/s r _map_itemzAccessPath._map_item`r2rcdSr+rr/s r _attr_missingzAccessPath._attr_missingar2rcdSr+rr/s r_array_item_missingzAccessPath._array_item_missingbr2rcdSr+rr/s r_map_item_missingzAccessPath._map_item_missingcr2rboolcdSr+rr/s r _is_prefix_ofzAccessPath._is_prefix_ofdr2rSequence[AccessStep]cdSr+rr0s r _to_stepszAccessPath._to_stepser2rcdSr+rr/s r _path_equalzAccessPath._path_equalfr2rNonecbttd)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__ ValueError)r0 __class__s rrLzAccessPath.__init__js1  H   rc4tS)zCreate a root access path. Returns ------- AccessPath A path representing the root of an object graph. )r%r,rrrrootzAccessPath.rootrs!!!rothercZt|tsdS||S)z*Return whether two access paths are equal.F isinstancer%rHr0rQs r__eq__zAccessPath.__eq__~s-%,, 5&&&rc\t|tsdS|| S)z.Return whether two access paths are not equal.TrSrUs r__ne__zAccessPath.__ne__s0%,, 4##E****rc,||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 )rBrUs r is_prefix_ofzAccessPath.is_prefix_ofs!!%(((rattr_keyc,||S)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 )r5r0r[s rattrzAccessPath.attrszz(###rc,||S)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 )r;r]s r attr_missingzAccessPath.attr_missings!!(+++rindexc,||S)zCreate 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 )r7r0ras r array_itemzAccessPath.array_items&&&rc,||S)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 )r=rcs rarray_item_missingzAccessPath.array_item_missings''...rr"c,||S)zCreate 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 )r9r0r"s rmap_itemzAccessPath.map_items~~c"""rc,||S)zCreate 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 )r?rhs rmap_item_missingzAccessPath.map_item_missings%%c***rc*|S)zConvert the access path to a list of access steps. Returns ------- access_steps The list of access steps )rFrEs rto_stepszAccessPath.to_stepss~~r)r)r%)r-rr)r%)r-r3r)r%)r-r r)r%)r-rr)r%)r-r%r)r@)r)rC)r)rI)rQrr)r@)rQr%r)r@)r[r3r)r%)rar r)r%)r"rr)r%)!rrrrr#r staticmethodr,r1r5r7r9r;r=r?rBrFrHrLrPrVrXrZr^r`rdrfrirkrmr__hash__ __classcell__)rNs@rr%r%>s.JJJ > &&& &????6666<<<<::::>>>>DDDDBBBB????;;;;====       " " "\ "'''' ++++ )))) $$$$ ,,,, '''' //// #### ++++     HHHHHrr%N)r __future__rtypingrcollections.abcrtvm_ffirrenumrcoreregistryr r rr%rrrrxsk$ #""""" ((((((%%%%%%,--       .- ,--FFFFFFF.-FFFr