o giѭ @s.ddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z Gddde Zee jGd d d eZee jee jGd d d eZGd ddeZGdddeZee jee jGdddeZee jee jee jee jee jee jee jee j ee j!ee j"ee j#ee j$ee j%ee j&ee j'ee j(ee j)ee j*ee j+ee j,ee j-ee j.ee j/ee j0ee j1ee j2ee j3ee j4GdddeZ5ee j6Gddde5Z7ee j8ee j9GdddeZ:ee j;GdddeZee j?Gddde>Z@ee jAGdd d e>ZBee jCGd!d"d"eZDee jEee jFee jGGd#d$d$eZHGd%d&d&eZIGd'd(d(eIZJee jKGd)d*d*eJZLee jMee jNee jOee jPee jQGd+d,d,eJZRee jSGd-d.d.eJZTee jUGd/d0d0eJZVee jWGd1d2d2eJZXee jYGd3d4d4eJZZee j[Gd5d6d6eJZ\ee j]Gd7d8d8eJZ^ee j_Gd9d:d:eJZ`ee jaGd;d<dd>eJZdee jeee jfee jgee jhee jiee jjGd?d@d@eJZkee jlGdAdBdBeJZmee jnGdCdDdDekZoee jpGdEdFdFeJZqee jrGdGdHdHeIZsee jtGdIdJdJeZtee juGdKdLdLeZuGdMdNdNeJZvGdOdPdPeJZwee jxGdQdRdReJZxee jyjzee j{GdSdTdTeJZ|ee j}ee j~GdUdVdVeZee jGdWdXdXeJZee jGdYdZdZeJZee jGd[d\d\eJZee jGd]d^d^eJZee jGd_d`d`eIZee jGdadbdbeJZee jGdcddddeJZee jGdedfdfeJZee jdgdhZee jdidjZee jdkdlZee jGdmdndneJZee jGdodpdpeIZee jGdqdrdreJZGdsdtdteJZduS)v)partial)deque)ir)register_default)typescgutils) numpy_supportc@seZdZdZddZeddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZd3ddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2ZdS)4 DataModela DataModel describe how a FE type is represented in the LLVM IR at different contexts. Contexts are: - value: representation inside function body. Maybe stored in stack. The representation here are flexible. - data: representation used when storing into containers (e.g. arrays). - argument: representation used for function argument. All composite types are unflattened into multiple primitive types. - return: representation used for return argument. Throughput the compiler pipeline, a LLVM value is usually passed around in the "value" representation. All "as_" prefix function converts from "value" representation. All "from_" prefix function converts to the "value" representation. cCs||_||_dSN)_dmm_fe_typeselfdmmfe_typerN/home/kuhnn/.local/lib/python3.10/site-packages/numba/core/datamodel/models.py__init__"s zDataModel.__init__cC|jSr )r rrrrr&szDataModel.fe_typecCt|r NotImplementedErrorrrrrget_value_type*zDataModel.get_value_typecC|Sr rrrrr get_data_type-rzDataModel.get_data_typecCr)z8Return a LLVM type or nested tuple of LLVM type rrrrrget_argument_type0szDataModel.get_argument_typecCrr rrrrrget_return_type5rzDataModel.get_return_typecCrr rrbuildervaluerrras_data8rzDataModel.as_datacCr)z` Takes one LLVM value Return a LLVM value or nested tuple of LLVM value rr rrr as_argument;zDataModel.as_argumentcCrr rr rrr as_returnBrzDataModel.as_returncCrr rr rrr from_dataErzDataModel.from_datacCr)za Takes a LLVM value or nested tuple of LLVM value Returns one LLVM value rr rrr from_argumentHr%zDataModel.from_argumentcCrr rr rrr from_returnOrzDataModel.from_returnNcCs|||j||dS)z~ Load value from a pointer to data. This is the default implementation, sufficient for most purposes. )align)r'loadrr!ptrr*rrrload_from_data_pointerRsz DataModel.load_from_data_pointercCgS)z Traverse contained members. Returns a iterable of contained (types, getters). Each getter is a one-argument function accepting a LLVM value. rrr!rrrtraverseYszDataModel.traversecsfddDS)zE Recursively list all models involved in this model. csg|]}j|qSr)r .0trrr esz-DataModel.traverse_models..)traverse_typesrrrrtraverse_modelsazDataModel.traverse_modelscCsd|jg}t|g}t|dkr0|}|D]}|j|vr)||||jqt|dks|S)zM Recursively list all frontend types involved in this model. r)r rlenpopleft inner_modelsappend)rrqueuedmi_dmrrrr6gs       zDataModel.traverse_typescCr/)z* List all *inner* models. rrrrrr;wszDataModel.inner_modelscCdS)z Returns the MemInfo object or None if it is not tracked. It is only defined for types.meminfo_pointer Nrr rrrget_nrt_meminfo}szDataModel.get_nrt_meminfocCr@NFrrrrrhas_nrt_meminfozDataModel.has_nrt_meminfocCstdd|DS)zQ Recursively check all contained types for need for NRT meminfo. css|]}|VqdSr )rC)r3modelrrr sz1DataModel.contains_nrt_meminfo..)anyr7rrrrcontains_nrt_meminfor8zDataModel.contains_nrt_meminfocCst||jfSr )typer rrrr_compared_fieldszDataModel._compared_fieldscCstt|Sr )hashtuplerJrrrr__hash__zDataModel.__hash__cCs$t|t|ur||kSdSrB)rIrJrotherrrr__eq__szDataModel.__eq__cCs || Sr )rRrPrrr__ne__ zDataModel.__ne__r )__name__ __module__ __qualname____doc__rpropertyrrrrrr#r$r&r'r(r)r.r1r7r6r;rArCrHrJrNrRrSrrrrr s6   r c@s0eZdZdZddZddZddZdd Zd S) OmittedArgDataModelz A data model for omitted arguments. Only the "argument" representation is defined, other representations raise a NotImplementedError. cCs tgSr )rLiteralStructTyperrrrr z"OmittedArgDataModel.get_value_typecCr@NrrrrrrrrDz%OmittedArgDataModel.get_argument_typecCr@r]rrr!valrrrr$rDzOmittedArgDataModel.as_argumentcCs|dksJ|dSr]rr^rrrr(sz!OmittedArgDataModel.from_argumentN)rUrVrWrXrrr$r(rrrrrZs  rZc@speZdZedZedZddZddZddZ d d Z d d Z d dZ ddZ ddZddZddZdS) BooleanModelcCrr ) _bit_typerrrrrzBooleanModel.get_value_typecCrr ) _byte_typerrrrrrdzBooleanModel.get_data_typecCrr rrrrrrrzBooleanModel.get_return_typecCrr rfrrrrrrzBooleanModel.get_argument_typecC|||Sr )zextrr rrrr#rOzBooleanModel.as_datacC |||Sr r#r rrrr$rTzBooleanModel.as_argumentcCrir rjr rrrr&rTzBooleanModel.as_returnc Cs|}t||}|d||d}||>\}}|||d|Wdn1s1wY|||d|Wdn1sKwYWdn1sZwY||S)Nz==rra)rr alloca_once icmp_unsignedrIif_elsestorer+)rr!r"ty resallocacondthen otherwiserrrr's  zBooleanModel.from_datacCrir r'r rrrr(rTzBooleanModel.from_argumentcCrir rtr rrrr)rTzBooleanModel.from_returnN)rUrVrWrIntTypercrerrrrr#r$r&r'r(r)rrrrr`s   r`csXeZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ Z S)PrimitiveModelzZA primitive type can be represented natively in the target in all usage contexts. cstt|||||_dSr )superrvrbe_typerrrrx __class__rrrs zPrimitiveModel.__init__cCrr )rxrrrrrrdzPrimitiveModel.get_value_typecC|Sr rr rrrr#rDzPrimitiveModel.as_datacCr|r rr rrrr$rDzPrimitiveModel.as_argumentcCr|r rr rrrr&rDzPrimitiveModel.as_returncCr|r rr rrrr'rDzPrimitiveModel.from_datacCr|r rr rrrr(rDzPrimitiveModel.from_argumentcCr|r rr rrrr)rDzPrimitiveModel.from_return) rUrVrWrXrrr#r$r&r'r(r) __classcell__rrrzrrvs rvc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS) ProxyModelzB Helper class for models which delegate to another model. cC |jSr )_proxied_modelrrrrrrr\zProxyModel.get_value_typecCrr )rrrrrrrr\zProxyModel.get_data_typecCrr )rrrrrrr r\zProxyModel.get_return_typecCrr )rrrrrrr r\zProxyModel.get_argument_typecC|j||Sr )rr#r rrrr#rKzProxyModel.as_datacCrr )rr$r rrrr$rKzProxyModel.as_argumentcCrr )rr&r rrrr&rKzProxyModel.as_returncCrr )rr'r rrrr'rKzProxyModel.from_datacCrr )rr(r rrrr(rKzProxyModel.from_argumentcCrr )rr)r rrrr)rKzProxyModel.from_returnN)rUrVrWrXrrrrr#r$r&r'r(r)rrrrr~s r~c eZdZdZfddZZS) EnumModelzA Enum members are represented exactly like their values. cs$tt|||||j|_dSr )rwrrlookupdtyperr rzrrr)szEnumModel.__init__rUrVrWrXrr}rrrzrr#srcs.eZdZdZedZfddZZ S) OpaqueModelz# Passed as opaque pointers rbcs|j}tt||||dSr ) _ptr_typerwrrryrzrrrPszOpaqueModel.__init__) rUrVrWrXrru as_pointerrrr}rrrzrr.src@s$eZdZddZddZddZdS) MemInfoModelcCs|j|jjgSr )r rr rrrrrr;XzMemInfoModel.inner_modelscCr@)NTrrrrrrC[rDzMemInfoModel.has_nrt_meminfocCr|r rr rrrrA^rDzMemInfoModel.get_nrt_meminfoN)rUrVrWr;rCrArrrrrUs rceZdZfddZZS) IntegerModelcs$t|j}tt||||dSr )rrubitwidthrwrrryrzrrres zIntegerModel.__init__rUrVrWrr}rrrzrrbrcr) FloatModelcsH|tjkr t}n|tjkrt}nt|tt| |||dSr ) rfloat32r FloatTypefloat64 DoubleTyperrwrrryrzrrrls    zFloatModel.__init__rrrrzrrjrcr) PointerModelcs<||j|_|j|_|j}tt||||dSr ) rr_pointee_modelr_pointee_be_typerrwrrryrzrrrxs  zPointerModel.__init__rrrrzrrvrrc@s.eZdZddZddZddZd dd ZdS) EphemeralPointerModelcCrr )rrrrrrrdz#EphemeralPointerModel.get_data_typecCs||}|j||Sr )r+rr#r rrrr#s zEphemeralPointerModel.as_datacCtdNz$use load_from_data_pointer() insteadrr rrrr'rzEphemeralPointerModel.from_dataNcCrgr bitcastrr,rrrr.rOz,EphemeralPointerModel.load_from_data_pointerr )rUrVrWrr#r'r.rrrrrs rcs>eZdZfddZddZddZddZd d d ZZS) EphemeralArrayModelcs*tt|||t|j|jj|_dSr ) rwrrr ArrayTyperr count _data_typer rzrrrs zEphemeralArrayModel.__init__cCrr rrrrrrrdz!EphemeralArrayModel.get_data_typecs(fddt|jjD}t|S)Nc s g|] }t|qSr)r+r gep_inboundsr3ir!r"rrr5sz/EphemeralArrayModel.as_data..)ranger rr pack_array)rr!r"valuesrrrr#s   zEphemeralArrayModel.as_datacCrrrr rrrr'rzEphemeralArrayModel.from_dataNcCrgr rr,rrrr.rOz*EphemeralArrayModel.load_from_data_pointerr ) rUrVrWrrr#r'r.r}rrrzrrs  rcr)ExternalFuncPointerModelcsT|j}|j}fdd|jD}tt||}tt | ||dS)Ncsg|] }|qSr)rrr2rrrr5sz5ExternalFuncPointerModel.__init__..) sigr return_typerargsr PointerType FunctionTyperwrr)rrrrrettyrrxrzrrrs z!ExternalFuncPointerModel.__init__rrrrzrrrrcs|eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZZS) UniTupleModelcsZtt|||||j|_t||_t |j |j|_ t |j |j|_ dSr )rwrrrr _elem_modelr9_countrrr _value_typerrr rzrrrs    zUniTupleModel.__init__cCrr )rrrrrrrdzUniTupleModel.get_value_typecCrr rrrrrrrdzUniTupleModel.get_data_typecCrr rrrrrrrzUniTupleModel.get_return_typecCs|jf|jSr )rrrrrrrrrzUniTupleModel.get_argument_typecCs>g}t|jD]}|||g}|j||}||q|Sr )rr extract_valuerr$r<rr!r"outrvrrrr$s  zUniTupleModel.as_argumentcCsFt|tj}t|D]\}}|j||}||||g}q |Sr )rConstantr Undefined enumeraterr( insert_valuerrrrr(s zUniTupleModel.from_argumentcCRt|tj}t|jD]}|||g}|j||}| |||g}q|Sr ) rrrrrrrrr#rrr!r"rrr_dvalrrrr# zUniTupleModel.as_datacCrr ) rrrrrrrrr'rrrrrr'rzUniTupleModel.from_datacCr|r rr rrrr&rDzUniTupleModel.as_returncCr|r rr rrrr)rDzUniTupleModel.from_returncs&fddfddtjDS)Ncs ||Sr r)rr"r!rrgetterrTz&UniTupleModel.traverse..gettercsg|] }jjt|fqSr)r rrrrrrrr5sz*UniTupleModel.traverse..)rrr0rr!rrrr1s  zUniTupleModel.traversecCs|jgSr )rrrrrr;rzUniTupleModel.inner_models)rUrVrWrrrrrr$r(r#r'r&r)r1r;r}rrrzrrs  rc@seZdZdZdS)CompositeModelzWAny model that is composed of multiple other models should subclass from this. N)rUrVrWrXrrrrrsrcseZdZdZdZfddZddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZd0ddZddZddZddZddZd d!Zd"d#Zd$d%Zed&d'Zd(d)Zd*d+Zd,d-Zd.d/ZZS)1 StructModelNcsRtt|||rt|\__nd__tfddjD_dS)Nrcg|]}j|qSrr rr2rrrr5 z(StructModel.__init__..)rwrrzip_fields_membersrM_modelsrrrmembersrzrrrs  zStructModel.__init__cCs||}|j|S)zU StructModel-specific: get the Numba type of the field named *name*. )get_field_positionr)rnameposrrrget_member_fe_types  zStructModel.get_member_fe_typecC(|jdurtdd|jD|_|jS)NcSg|]}|qSrrr2rrrr5 z.StructModel.get_value_type..)rrr[rrrrrr   zStructModel.get_value_typecCr)NcSrrrfr2rrrr5rz-StructModel.get_data_type..)rrr[rrrrrrrzStructModel.get_data_typecCstdd|jDS)NcSrr)rr2rrrr5"z1StructModel.get_argument_type..)rMrrrrrr!szStructModel.get_argument_typecCrr rfrrrrr$rzStructModel.get_return_typec Cs@g}t|jD]\}}|t|||||||qt|Sr )rrr<getattrgetrM)rmethnamer!r" extractedrr>rrr_as's  zStructModel._asc CsTt|tj}tt|j|D]\}\}}t||||}|||||}q|Sr ) rrrrrrrrset) rrr!r"structrr>r_rrrr_from.s zStructModel._fromcCF|d||}t|tj}t|D] \}}||||g}q|S)a, Converts the LLVM struct in `value` into a representation suited for storing into arrays. Note ---- Current implementation rarely changes how types are represented for "value" and "data". This is usually a pointless rebuild of the immutable LLVM struct value. Luckily, LLVM optimization removes all redundancy. Sample usecase: Structures nested with pointers to other structures that can be serialized into a flat representation when storing into array. r#rrrrrrrrr!r"elemsrrelrrrr#7s zStructModel.as_datac,fddtt|jD}|d|S)z Convert from "data" representation back into "value" representation. Usually invoked when loading from array. See notes in `as_data()` cg|] }|gqSrrrrrrr5Tz)StructModel.from_data..r'rr9rrrr!r"valsrrrr'Ms  zStructModel.from_datac Csxg}t|jD]\}}t||d|}||||}||qt|tj } t|D] \}}| || ||} q-| SNr) rrrrr.r<rrrrr) rr!r-r*rrrEelem_ptrr_rrrrr.Xs z"StructModel.load_from_data_pointercC|d||S)Nr$)rr rrrr$drKzStructModel.as_argumentcCr)Nr()rr rrrr(grKzStructModel.from_argumentcCr)Nr#rrrrrr&js zStructModel.as_returncr)Ncrrrrrrrr5rrz+StructModel.from_return..r'rrrrrr)qs  zStructModel.from_returncCs0t|tr ||}|j||gd|j|dS)a'Get a field at the given position or the fieldname Args ---- builder: LLVM IRBuilder val: value to be inserted pos: int or str field index or field name Returns ------- Extracted value z extracted.r) isinstancestrrrr)rr!r_rrrrrvs    zStructModel.getcCs2t|tr ||}|j|||gd|j|dS)anSet a field at the given position or the fieldname Args ---- builder: LLVM IRBuilder stval: LLVM struct value val: value to be inserted pos: int or str field index or field name Returns ------- A new LLVM struct with the value inserted z inserted.r)rrrrr)rr!stvalr_rrrrrs    zStructModel.setcCs2z|j|WStytd|jj|fw)Nz!%s does not have a field named %r)rindex ValueErrorKeyErrorr{rU)rfieldrrrrs  zStructModel.get_field_positioncCs t|jSr )r9rrrrr field_counts zStructModel.field_countcCst|tr ||}|j|S)zGet the frontend type (numba type) of a field given the position or the fieldname Args ---- pos: int or str field index or field name )rrrrrrrrrget_types  zStructModel.get_typecCs |j|S)z Get the datamodel of a field given the position or the fieldname. Args ---- pos: int or str field index or field name rrrrr get_models zStructModel.get_modelcs$fddfddjDS)Ncs8|jkr|jf}tdj|||S)Nzexpecting {0} but got {1})rIr TypeErrorformatr)kr"rr!rrrrsz$StructModel.traverse..gettercs g|] }|t|fqSr)rr)r3rrrrr5 z(StructModel.traverse..)rr0rrrr1szStructModel.traversecCrr rrrrrr;rdzStructModel.inner_modelsr )rUrVrWrrrrrrrrrrr#r'r.r$r(r&r)rrrrYrrrr1r;r}rrrzrrs6        rcs eZdZeZfddZZS) ComplexModelcs,d|jfd|jfg}tt||||dS)Nrealimag)underlying_floatrwrrrrzrrrszComplexModel.__init__)rUrVrWNotImplemented _element_typerr}rrrzrrrcr) TupleModelcs*ddt|D}tt||||dS)NcSs g|] \}}dt||fqS)f)r)r3rr4rrrr5rz'TupleModel.__init__..)rrwr rrrzrrrszTupleModel.__init__rrrrzrr sr cr) UnionModelcs4dtjfdtj|jfg}tt||||dS)Ntagpayload)ruintpTuple from_typesrwr rrrzrrrszUnionModel.__init__rrrrzrr rr cr) PairModelcs,d|jfd|jfg}tt||||dS)Nfirstsecond) first_type second_typerwrrrrzrrrszPairModel.__init__rrrrzrrrrcr)ListPayloadModelcs>dtjfdtjfdtjfd|jjfg}tt||||dS)Nsize allocateddirtydata)rintpboolean containerrrwrrrrzrrrs  zListPayloadModel.__init__rrrrzrrrrcr) ListModelc:t|}dt|fdtjfg}tt||||dSNmeminfoparent)r ListPayloadMemInfoPointerpyobjectrwr rrrr payload_typerrzrrr  zListModel.__init__rrrrzrr rr cr) ListIterModelcBt|j}dt|fdttjfg}tt||||dSNr#r) rr%rr&EphemeralPointerrrwr+rr(rzrrrs  zListIterModel.__init__rrrrzrr+rr+cr) SetEntryModelcs2|jj}dtjfd|fg}tt||||dS)NrLkey)set_typerrrrwr/r)rrrrrrzrrr*s zSetEntryModel.__init__rrrrzrr/(rr/cr)SetPayloadModelcsVt|j}dtjfdtjfdtjfdtjfdtjfd|fg}tt||||dS)Nfillusedmaskfingerrentries)rSetEntryrrrrwr2r)rrr entry_typerrzrrr6s zSetPayloadModel.__init__rrrrzrr24rr2cr)SetModelcr!r")r SetPayloadr&r'rwr:rr(rzrrrJr*zSetModel.__init__rrrrzrr:Hrr:cr) SetIterModelcr,r-) rr;rr&r.rrwr<rr(rzrrrVs  zSetIterModel.__init__rrrrzrr<Trr<cr) ArrayModelc sv|j}dt|jfdtjfdtjfdtjfdt|jfdttj|fdttj|fg}tt | |||dS)Nr#r$nitemsitemsizershapestrides) ndimrr&rr'rCPointerUniTuplerwr=rrrrrBrrzrrris zArrayModel.__init__rrrrzrr=bsr=cr)ArrayFlagsModelcs$d|jfg}tt||||dS)Nr$) array_typerwrFrrrzrrrzszArrayFlagsModel.__init__rrrrzrrFxrrFcs$eZdZfddZddZZS)NestedArrayModelcs(||j|_tt|||dSr )rrr_be_typerwrHrr rzrrrszNestedArrayModel.__init__cCst|j|jj}|S)zXReturn the LLVM type representation for the storage of the nestedarray. )rrrIr r>)rretrrras_storage_typesz NestedArrayModel.as_storage_type)rUrVrWrrKr}rrrzrrHs rHcs<eZdZfddZddZddZddZd d ZZS) OptionalModelcs:d|jfdtjfg}||j|_tt||||dS)Nrvalid)rIrrr _value_modelrwrLrrrzrrrs zOptionalModel.__init__cCrr )rNrrrrrrr\zOptionalModel.get_return_typecCstr rr rrrr&rDzOptionalModel.as_returncCrr )rNr)r rrrr)rKzOptionalModel.from_returncs:fdd}fddd|fdfgS)Ncs.|}|d}||t|jdS)Nr)rselectrrrI)r"rMrr! get_validrrrget_datasz(OptionalModel.traverse..get_datacs|dS)NrM)rr"rrrrQrKz)OptionalModel.traverse..get_validrrM)r)rr!rRrrPrr1s   zOptionalModel.traverse) rUrVrWrrr&r)r1r}rrrzrrLs  rLcsveZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dddZZS) RecordModelcsNtt||fdd|jD_ttd|j_ j _ dS)Ncsg|] \}}j|qSrr)r3_r4rrrr5sz(RecordModel.__init__..rb) rwrTrrrrrrurrIr _be_ptr_typer rzrrrszRecordModel.__init__cCr)z6Passed around as reference to underlying data rVrrrrrszRecordModel.get_value_typecCrr rWrrrrrrdzRecordModel.get_argument_typecCrr rWrrrrrrdzRecordModel.get_return_typecCrr rIrrrrrrdzRecordModel.get_data_typecC ||Sr r+r rrrr#r\zRecordModel.as_datacCrrrr rrrr'rzRecordModel.from_datacCr|r rr rrrr$rDzRecordModel.as_argumentcCr|r rr rrrr(rDzRecordModel.from_argumentcCr|r rr rrrr&rDzRecordModel.as_returncCr|r rr rrrr)rDzRecordModel.from_returnNcCrgr rr,rrrr.rOz"RecordModel.load_from_data_pointerr )rUrVrWrrrrrr#r'r$r(r&r)r.r}rrrzrrTs rTc\eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ Z S)UnicodeCharSeqcs6tt|||ttjd}t||j|_ dSNrb) rwr\rrrursizeof_unicode_charrrrIrrrchartyrzrrrszUnicodeCharSeq.__init__cCrr rXrrrrrrdzUnicodeCharSeq.get_value_typecCrr rXrrrrrrdzUnicodeCharSeq.get_data_typecCr|r rr rrrr#rDzUnicodeCharSeq.as_datacCr|r rr rrrr'rDzUnicodeCharSeq.from_datacCr|r rr rrrr&rDzUnicodeCharSeq.as_returncCr|r rr rrrr)rDzUnicodeCharSeq.from_returncCr|r rr rrrr$rDzUnicodeCharSeq.as_argumentcCr|r rr rrrr(rDzUnicodeCharSeq.from_argument rUrVrWrrrr#r'r&r)r$r(r}rrrzrr\ r\cr[)CharSeqcs0tt|||td}t||j|_dSr])rwrcrrrurrrIr_rzrrrs zCharSeq.__init__cCrr rXrrrrrrdzCharSeq.get_value_typecCrr rXrrrrrrdzCharSeq.get_data_typecCr|r rr rrrr#rDzCharSeq.as_datacCr|r rr rrrr' rDzCharSeq.from_datacCr|r rr rrrr& rDzCharSeq.as_returncCr|r rr rrrr)rDzCharSeq.from_returncCr|r rr rrrr$rDzCharSeq.as_argumentcCr|r rr rrrr(rDzCharSeq.from_argumentrarrrzrrcrbrccr)CContiguousFlatItercsv|jjdksJ|j}|j}|j}d|fdtjfdttjfg}|r/|dttj|ft t | |||dS)NCarraystriderindices) rGlayoutrrBrrr.r<EphemeralArrayrwrdr)rrr need_indicesrGrrBrrzrrrszCContiguousFlatIter.__init__rrrrzrrdrdcr)FlatItercsf|j}|j}|j}d|fdtt||fdttj|fdttjfg}t t | |||dS)Nrfpointersrh exhausted) rGrrBrrjrCrr.rrwrmr)rrrrGrrBrrzrrr+szFlatIter.__init__rrrrzrrm*rlrmcr) UniTupleIterc2dttjfd|jfg}tt||||dS)NrrM)rr.rrrwrprrrzrrr9szUniTupleIter.__init__rrrrzrrp7rrpcr) SliceModelcs4dtjfdtjfdtjfg}tt||||dSNstartstopstep)rrrwrrrrrzrrrBs zSliceModel.__init__rrrrzrrr?rrrcr)NPDatetimeModelcs"td}tt||||dS)N@)rrurwrwrryrzrrrMs zNPDatetimeModel.__init__rrrrzrrwJrrwcr) ArrayIteratorcrq)Nrrf)rr.rrGrwryrrrzrrrTszArrayIterator.__init__rrrrzrryRrrycr) EnumerateTypecrq)Nriter)rr.r source_typerwrzrrrzrrr]szEnumerateType.__init__rrrrzrrz[rrzcr)ZipTypecs,ddt|jD}tt||||dS)NcSsg|] \}}d||jfqS)ziter%d) iterator_type)r3rr|rrrr5gsz$ZipType.__init__..)r source_typesrwr}rrrzrrrfszZipType.__init__rrrrzrr}drr}cr)RangeIteratorTypecsF|j}dt|fd|fd|fdt|fg}tt||||dS)Nr{rurvr) yield_typerr.rwrrrrrint_typerrzrrrns  zRangeIteratorType.__init__rrrrzrrlrrcsleZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ZS)GeneratorModelcstt||fdd|jD_fdd|jD_tddjD_ tddjD_ tj t jj j g_j_dS)Ncs$g|]}t|tjsj|qSr)rrOmittedr rr2rrrr5|s z+GeneratorModel.__init__..crrrr2rrrr5~rcSrrrfr2rrrr5rcSrrrfr2rrrr5r)rwrr arg_types _arg_models state_types _state_modelsrr[ _args_be_type_state_be_typer rrint32rrIrrVr rzrrryszGeneratorModel.__init__cCr)zH The generator closure is passed around as a reference. rWrrrrrszGeneratorModel.get_value_typecCrr rWrrrrrrdz GeneratorModel.get_argument_typecCrr rXrrrrrrdzGeneratorModel.get_return_typecCrr rXrrrrrrdzGeneratorModel.get_data_typecCr|r rr rrrr$rDzGeneratorModel.as_argumentcCr|r rr rrrr(rDzGeneratorModel.from_argumentcCrir rjr rrrr&rTzGeneratorModel.as_returncCrir rtr rrrr)rTzGeneratorModel.from_returncCrYr rZr rrrr#r\zGeneratorModel.as_datacCst||j}||||Sr )rrkrIrn)rr!r"stackrrrr's zGeneratorModel.from_data)rUrVrWrrrrrr$r(r&r)r#r'r}rrrzrrws rcr)ArrayCTypesModelcs8dt|jfdt|jfg}tt||||dS)Nrr#)rrCrr&rwrrrrzrrrszArrayCTypesModel.__init__rrrrzrrrrcr) RangeModelcs6|jj}d|fd|fd|fg}tt||||dSrs)r~rrwrrrrzrrrs zRangeModel.__init__rrrrzrrrrcr) NdIndexModelcsP|j}dttj|fdttj|fdttjfg}tt| |||dS)Nr@rhro) rBrrDrrjr.rrwrrrErzrrrs zNdIndexModel.__init__rrrrzrrrrcC$|jjdkr t||ddSt||S)NreFrkrGrirdrmrrorrrhandle_numpy_flat_type  rcCr)NreTrrrrrrhandle_numpy_ndenumerate_typerrcCs ||jSr )thisrrrrhandle_bound_functions rcr)NdItercs|j}|j}|jr |nd}dttjfdt|fdttj|fdt tj|fg}t |j D]$\}}|\} } } } d|} | dkrO| | ttjfq1| dvrTq1Jt |jD]\}}t |tjssd |} | | t|fq[tt||||dS) Nraroarraysr@rhzindex%dflat)scalarindexed0drzscalar%d)rrBneed_shaped_indexingrr.rrrDrrjrindexersr<rArrayrwrr)rrr array_typesrB shape_lenrrsubkind start_dimend_dimrU member_namerorzrrrs,   zNdIter.__init__rrrrzrrrrcseZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZd&ddZddZddZd d!Zed"d#Zd$d%ZZS)'DeferredStructModelcs0tt|||dt||_||_dS)Nz deferred.{0})rwrrridtypenameractual_fe_typer rzrrrszDeferredStructModel.__init__cCtj|jdS)Nz.valuerglobal_contextget_identified_typerrrrrr rz"DeferredStructModel.get_value_typecCr)Nz.datarrrrrrrz!DeferredStructModel.get_data_typecCrr ) _actual_modelrrrrrrr\z%DeferredStructModel.get_argument_typecCs|||}|j||Sr )rrr$)rr!r"innerrrrr$s zDeferredStructModel.as_argumentcCs |j||}||||Sr )rr(rmake_uninitialized)rr!r"resrrrr(sz!DeferredStructModel.from_argumentcCs8||||}|j||}|}||||Sr )_definerrr'rrrr!r"elemrrrrr's  zDeferredStructModel.from_datacCs<||||}|j||}|jdd}||||S)Nr)r)rrrr#rrrrrrr#$s   zDeferredStructModel.as_datacCr|r rr rrrr)+rDzDeferredStructModel.from_returncCr|r rr rrrr&.rDzDeferredStructModel.as_returncCs||dgSrrr rrrr1rKzDeferredStructModel.getcCs|||dgSr)r)rr!r"contentrrrr4rOzDeferredStructModel.setr"cCs0||dkr |}n|}t|tjS)Nr")rrrrrr)rrrorrrr7s  z&DeferredStructModel.make_uninitializedcCs(|}|||}||dSr )r_define_value_typer_define_data_type)rvaltydattyrrrr?s zDeferredStructModel._definecC|jr ||jdSdSr ) is_opaqueset_bodyrr)r value_typerrrrEz&DeferredStructModel._define_value_typecCrr )rrrr)r data_typerrrrIrz%DeferredStructModel._define_data_typecCs|j|jSr )r rrrrrrrMsz!DeferredStructModel._actual_modelcs|jfddfgS)Ncs|dgSrrrSrrrSsz.DeferredStructModel.traverse..)rr0rrrr1Qs zDeferredStructModel.traverserS)rUrVrWrrrrr$r(r'r#r)r&rrrrrrrYrr1r}rrrzrrs(   rcr)StructPayloadModelz.Model for the payload of a mutable struct cs"t|j}t|||dSr )rM field_dictitemsrwr)rrfe_typrrzrrrZszStructPayloadModel.__init__rrrrzrrVr rcr)StructRefModelz?Model for a mutable struct. A reference to the payload cs,|}dt|fg}t|||dS)Nr#)rrr&rwr)rrrrrrzrrrcs zStructRefModel.__init__rrrrzrr_srN) functoolsr collectionsrllvmliternumba.core.datamodel.registryr numba.corerrnumba.nprobjectr rrZBooleanBooleanLiteralr`rvr~ EnumMember IntEnumMemberrOpaquePyObject RawPointerNoneType StringLiteral EllipsisTypeFunctionTypeObjectModulePhantomUndefVarContextManager DispatcherObjModeDispatcherExceptionClassDummyExceptionInstanceExternalFunction EnumClass IntEnumClass NumberClassTypeRefNamedTupleClassDType RecursiveCallMakeFunctionLiteralPoisonrr&rIntegerIntegerLiteralrFloatrrCrr.rrjrExternalFunctionPointerrrD NamedUniTupleStarArgUniTuplerrrComplexr LiteralListLiteralStrKeyDictr NamedTuple StarArgTupler  UnionTyper Pairrr%rListr ListIterr+r8r/r;r2Setr:SetIterr<rBuffer ByteArrayBytes MemoryViewPyArrayr= ArrayFlagsrF NestedArrayrHOptionalrLRecordrTr\rcrdrmrpmisc SliceLiteral SliceTyperr NPDatetime NPTimedeltarwryrzr}r Generatorr ArrayCTypesr RangeTyperNumpyNdIndexTyper NumpyFlatTyperNumpyNdEnumerateTyper BoundFunctionrNumpyNdIterTyper DeferredTyperStructRefPayloadrrrrrrs2    +$ H    EQ        +    6    "P