o i@sdZddlmZddlmZer"ddlmZddlmZddlm Z ddl m Z ddl mZdd l mZGd d d e Zed Gd ddeZedGdddeZdS)zAccess path classes.) annotations) TYPE_CHECKING)Sequence)Object)Any)IntEnum)register_objectc@s(eZdZdZdZdZdZdZdZdZ dS) AccessKindz(Kinds of access steps in an access path.rrN) __name__ __module__ __qualname____doc__ATTR ARRAY_ITEMMAP_ITEM ATTR_MISSINGARRAY_ITEM_MISSINGMAP_ITEM_MISSINGrrP/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/tvm_ffi/access_path.pyr &sr zffi.reflection.AccessStepc@s"eZdZUdZded<ded<dS) AccessStepzAccess step container.intkindrkeyN)rrrr__annotations__rrrrr1s  rzffi.reflection.AccessPathcs*eZdZUdZded<ded<ded<erNedBd d ZdCd dZdDddZ dEddZ dFddZ dDddZ dEddZ dFddZdGddZdHd!d"ZdGd#d$ZdIfd&d' ZedBd(d)ZdJd+d,ZdJd-d.ZdKd/d0ZdLd2d3ZdLd4d5ZdMd7d8ZdMd9d:ZdNdd?ZdHd@dAZejZZ 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 | NoneparentzAccessStep | NonesteprdepthreturncCdSNrrrrr_root[szAccessPath._root_1rcCr%r&rselfr(rrr_extend]zAccessPath._extendstrcCr%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_missingboolcCr%r&rr)rrr _is_prefix_ofdr,zAccessPath._is_prefix_ofSequence[AccessStep]cCr%r&rr*rrr _to_stepser,zAccessPath._to_stepscCr%r&rr)rrr _path_equalfr,zAccessPath._path_equalNonecsttd)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__cCstS)zCreate a root access path. Returns ------- AccessPath A path representing the root of an object graph. )r r'rrrrrootrs zAccessPath.rootothercCst|tsdS||S)z*Return whether two access paths are equal.F isinstancer r9r*rArrr__eq__~s  zAccessPath.__eq__cCst|tsdS|| 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_ofattr_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*rJrrrattrrIzAccessPath.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_missingrIzAccessPath.attr_missingindexcCrG)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 )r/r*rNrrr array_itemrIzAccessPath.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 )r2rOrrrarray_item_missingrIzAccessPath.array_item_missingrcCrG)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 )r0r*rrrrmap_itemrIzAccessPath.map_itemcCrG)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 )r3rRrrrmap_item_missingrIzAccessPath.map_item_missingcCs|S)zConvert the access path to a list of access steps. Returns ------- access_steps The list of access steps )r8r7rrrto_stepss 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__rtypingrcollections.abcrtvm_ffirrenumrcoreregistryr r rr rrrrs