U ?h@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 j3GdddeZ4ee j5Gddde4Z6ee j7ee j8GdddeZ9ee j:GdddeZ;ee j<GdddeZ=ee j>Gddde=Z?ee j@Gdd d e=ZAee jBGd!d"d"eZCee jDee jEee jFGd#d$d$eZGGd%d&d&eZHGd'd(d(eHZIee jJGd)d*d*eIZKee jLee jMee jNee jOee jPGd+d,d,eIZQee jRGd-d.d.eIZSee jTGd/d0d0eIZUee jVGd1d2d2eIZWee jXGd3d4d4eIZYee jZGd5d6d6eIZ[ee j\Gd7d8d8eIZ]ee j^Gd9d:d:eIZ_ee j`Gd;d<dd>eIZcee jdee jeee jfee jgee jhee jiGd?d@d@eIZjee jkGdAdBdBeIZlee jmGdCdDdDejZnee joGdEdFdFeIZpee jqGdGdHdHeHZree jsGdIdJdJeZsee jtGdKdLdLeZtGdMdNdNeIZuGdOdPdPeIZvee jwGdQdRdReIZwee jxjyee jzGdSdTdTeIZ{ee j|ee j}GdUdVdVeZ~ee jGdWdXdXeIZee jGdYdZdZeIZee jGd[d\d\eIZee jGd]d^d^eIZee jGd_d`d`eHZee jGdadbdbeIZee jGdcddddeIZee jGdedfdfeIZee jdgdhZee jdidjZee jdkdlZee jGdmdndneIZee jGdodpdpeHZee jGdqdrdreIZGdsdtdteIZduS)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_typerM/var/www/html/venv/lib/python3.8/site-packages/numba/core/datamodel/models.py__init__"szDataModel.__init__cCs|jSr )r rrrrr&szDataModel.fe_typecCs t|dSr NotImplementedErrorrrrrget_value_type*szDataModel.get_value_typecCs|Sr rrrrr get_data_type-szDataModel.get_data_typecCs|S)z8Return a LLVM type or nested tuple of LLVM type rrrrrget_argument_type0szDataModel.get_argument_typecCs|Sr rrrrrget_return_type5szDataModel.get_return_typecCs t|dSr rrbuildervaluerrras_data8szDataModel.as_datacCs t|dS)z` Takes one LLVM value Return a LLVM value or nested tuple of LLVM value Nrrrrr as_argument;szDataModel.as_argumentcCs t|dSr rrrrr as_returnBszDataModel.as_returncCs t|dSr rrrrr from_dataEszDataModel.from_datacCs t|dS)za Takes a LLVM value or nested tuple of LLVM value Returns one LLVM value Nrrrrr from_argumentHszDataModel.from_argumentcCs t|dSr rrrrr from_returnOszDataModel.from_returnNcCs|||j||dS)z~ Load value from a pointer to data. This is the default implementation, sufficient for most purposes. )align)r"loadrrptrr%rrrload_from_data_pointerRsz DataModel.load_from_data_pointercCsgS)z Traverse contained members. Returns a iterable of contained (types, getters). Each getter is a one-argument function accepting a LLVM value. rrrrrrtraverseYszDataModel.traversecsfddDS)zE Recursively list all models involved in this model. csg|]}j|qSr)r .0trrr esz-DataModel.traverse_models..)traverse_typesrrrrtraverse_modelsaszDataModel.traverse_modelscCsZ|jg}t|g}t|dkrV|}|D]$}|j|kr.||||jq.q|S)zM Recursively list all frontend types involved in this model. r)r rlenpopleft inner_modelsappend)rrqueuedmZi_dmrrrr0gs     zDataModel.traverse_typescCsgS)z* List all *inner* models. rrrrrr4wszDataModel.inner_modelscCsdS)z Returns the MemInfo object or None if it is not tracked. It is only defined for types.meminfo_pointer Nrrrrrget_nrt_meminfo}szDataModel.get_nrt_meminfocCsdSNFrrrrrhas_nrt_meminfoszDataModel.has_nrt_meminfocCstdd|DS)zQ Recursively check all contained types for need for NRT meminfo. css|]}|VqdSr )r:)r-modelrrr sz1DataModel.contains_nrt_meminfo..)anyr1rrrrcontains_nrt_meminfoszDataModel.contains_nrt_meminfocCst||jfSr )typer rrrr_compared_fieldsszDataModel._compared_fieldscCstt|Sr )hashtupler@rrrr__hash__szDataModel.__hash__cCs(t|t|kr ||kSdSdSr9)r?r@rotherrrr__eq__szDataModel.__eq__cCs || Sr )rFrDrrr__ne__szDataModel.__ne__)N)__name__ __module__ __qualname____doc__rpropertyrrrrrrr r!r"r#r$r)r+r1r0r4r8r:r>r@rCrFrGrrrrr s4  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 )rLiteralStructTyperrrrrsz"OmittedArgDataModel.get_value_typecCsdSNrrrrrrrsz%OmittedArgDataModel.get_argument_typecCsdSrOrrrvalrrrr szOmittedArgDataModel.as_argumentcCs|dkst|dSrO)AssertionErrorrPrrrr#sz!OmittedArgDataModel.from_argumentN)rHrIrJrKrrr r#rrrrrMs rMc@speZdZedZedZddZddZddZ d d Z d d Z d dZ ddZ ddZddZddZdS) BooleanModelcCs|jSr ) _bit_typerrrrrszBooleanModel.get_value_typecCs|jSr ) _byte_typerrrrrszBooleanModel.get_data_typecCs|Sr rrrrrrszBooleanModel.get_return_typecCs|Sr rXrrrrrszBooleanModel.get_argument_typecCs|||Sr )ZzextrrrrrrszBooleanModel.as_datacCs |||Sr rrrrrr szBooleanModel.as_argumentcCs |||Sr rYrrrrr!szBooleanModel.as_returnc Cs|}t||}|d||d}||J\}}|||d|W5QRX|||d|W5QRXW5QRX||S)Nz==rrT)rr alloca_onceZ icmp_unsignedr?Zif_elsestorer&)rrrtyZ resallocaZcondZthenZ otherwiserrrr"s $zBooleanModel.from_datacCs |||Sr r"rrrrr#szBooleanModel.from_argumentcCs |||Sr r]rrrrr$szBooleanModel.from_returnN)rHrIrJrIntTyperVrWrrrrrr r!r"r#r$rrrrrSs   rScsXeZdZdZfddZddZddZdd 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 )superr_rbe_typerrrra __class__rrrszPrimitiveModel.__init__cCs|jSr )rarrrrrszPrimitiveModel.get_value_typecCs|Sr rrrrrrszPrimitiveModel.as_datacCs|Sr rrrrrr szPrimitiveModel.as_argumentcCs|Sr rrrrrr!szPrimitiveModel.as_returncCs|Sr rrrrrr"szPrimitiveModel.from_datacCs|Sr rrrrrr#szPrimitiveModel.from_argumentcCs|Sr rrrrrr$szPrimitiveModel.from_return) rHrIrJrKrrrr r!r"r#r$ __classcell__rrrcrr_s r_c@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS) ProxyModelzB Helper class for models which delegate to another model. cCs |jSr )_proxied_modelrrrrrrszProxyModel.get_value_typecCs |jSr )rgrrrrrrszProxyModel.get_data_typecCs |jSr )rgrrrrrr szProxyModel.get_return_typecCs |jSr )rgrrrrrr szProxyModel.get_argument_typecCs|j||Sr )rgrrrrrrszProxyModel.as_datacCs|j||Sr )rgr rrrrr szProxyModel.as_argumentcCs|j||Sr )rgr!rrrrr!szProxyModel.as_returncCs|j||Sr )rgr"rrrrr"szProxyModel.from_datacCs|j||Sr )rgr#rrrrr#szProxyModel.from_argumentcCs|j||Sr )rgr$rrrrr$szProxyModel.from_returnN)rHrIrJrKrrrrrr r!r"r#r$rrrrrfsrfcs eZdZdZfddZZS) EnumModelzA Enum members are represented exactly like their values. cs$tt|||||j|_dSr )r`rhrlookupdtypergr rcrrr)szEnumModel.__init__rHrIrJrKrrerrrcrrh#srhcs.eZdZdZedZfddZZ S) OpaqueModelz# Passed as opaque pointers rUcs|j}tt||||dSr ) _ptr_typer`rlrrbrcrrrOszOpaqueModel.__init__) rHrIrJrKrr^ as_pointerrmrrerrrcrrl.srlc@s$eZdZddZddZddZdS) MemInfoModelcCs|j|jjgSr )r rir rjrrrrr4WszMemInfoModel.inner_modelscCsdS)NTrrrrrr:ZszMemInfoModel.has_nrt_meminfocCs|Sr rrrrrr8]szMemInfoModel.get_nrt_meminfoN)rHrIrJr4r:r8rrrrroTsrocseZdZfddZZS) IntegerModelcs$t|j}tt||||dSr )rr^Zbitwidthr`rprrbrcrrrds zIntegerModel.__init__rHrIrJrrerrrcrrpasrpcseZdZfddZZS) FloatModelcsH|tjkrt}n|tjkr(t}nt|tt| |||dSr ) rZfloat32rZ FloatTypeZfloat64Z DoubleTyperr`rrrrbrcrrrks     zFloatModel.__init__rqrrrcrrrisrrcseZdZfddZZS) PointerModelcs<||j|_|j|_|j}tt||||dSr ) rirj_pointee_modelr_pointee_be_typernr`rsrrbrcrrrws  zPointerModel.__init__rqrrrcrrsusrsc@s.eZdZddZddZddZd dd ZdS) EphemeralPointerModelcCs|jSr )rurrrrrsz#EphemeralPointerModel.get_data_typecCs||}|j||Sr )r&rtrrrrrrs zEphemeralPointerModel.as_datacCs tddSNz$use load_from_data_pointer() insteadrrrrrr"szEphemeralPointerModel.from_dataNcCs|||Sr Zbitcastrr'rrrr)sz,EphemeralPointerModel.load_from_data_pointer)N)rHrIrJrrr"r)rrrrrv~srvcs>eZdZfddZddZddZddZd d d ZZS) EphemeralArrayModelcs*tt|||t|j|jj|_dSr ) r`ryrr ArrayTyperur count _data_typer rcrrrszEphemeralArrayModel.__init__cCs|jSr r|rrrrrsz!EphemeralArrayModel.get_data_typecs(fddt|jjD}t|S)Nc s g|]}t|qSr)r&r gep_inboundsr-irrrrr/sz/EphemeralArrayModel.as_data..)ranger r{rZ pack_array)rrrvaluesrrrrs  zEphemeralArrayModel.as_datacCs tddSrwrrrrrr"szEphemeralArrayModel.from_dataNcCs|||Sr rxr'rrrr)sz*EphemeralArrayModel.load_from_data_pointer)N) rHrIrJrrrr"r)rerrrcrrys  rycseZdZfddZZS)ExternalFuncPointerModelcsT|j}|j}fdd|jD}tt||}tt | ||dS)Ncsg|]}|qSr)rirr,rrrr/sz5ExternalFuncPointerModel.__init__..) sigri return_typerargsrZ PointerType FunctionTyper`rr)rrrrZrettyrrarcrrrs z!ExternalFuncPointerModel.__init__rqrrrcrrsrcs|eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZZS) UniTupleModelcsZtt|||||j|_t||_t |j |j|_ t |j |j|_ dSr )r`rrrirj _elem_modelr2_countrrzr _value_typerr|r rcrrrs   zUniTupleModel.__init__cCs|jSr )rrrrrrszUniTupleModel.get_value_typecCs|jSr r}rrrrrszUniTupleModel.get_data_typecCs|Sr rrrrrrszUniTupleModel.get_return_typecCs|jf|jSr )rrrrrrrrszUniTupleModel.get_argument_typecCs>g}t|jD]*}|||g}|j||}||q|Sr )rr extract_valuerr r5rrroutrvrrrr s  zUniTupleModel.as_argumentcCsFt|tj}t|D]&\}}|j||}||||g}q|Sr )rConstantr Undefined enumeraterr# insert_valuerrrrr#s zUniTupleModel.from_argumentcCsRt|tj}t|jD]0}|||g}|j||}| |||g}q|Sr ) rrrrrrrrrrrrrrrrQdvalrrrrs zUniTupleModel.as_datacCsRt|tj}t|jD]0}|||g}|j||}| |||g}q|Sr ) rrrrrrrrr"rrrrrr"s zUniTupleModel.from_datacCs|Sr rrrrrr!szUniTupleModel.as_returncCs|Sr rrrrrr$szUniTupleModel.from_returncs&fddfddtjDS)Ncs ||Sr r)rrrrrgettersz&UniTupleModel.traverse..gettercsg|]}jjt|fqSr)r rjrrrrrrr/sz*UniTupleModel.traverse..)rrr*rrrrrr+s  zUniTupleModel.traversecCs|jgSr )rrrrrr4szUniTupleModel.inner_models)rHrIrJrrrrrr r#rr"r!r$r+r4rerrrcrrs rc@seZdZdZdS)CompositeModelzWAny model that is composed of multiple other models should subclass from this. N)rHrIrJrKrrrrrsrcseZdZdZdZfddZddZddZdd 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|||r(t|\__n d__tfddjD_dS)Nrcsg|]}j|qSrr rir,rrrr/ sz(StructModel.__init__..)r`rrzip_fields_membersrB_modelsrrrmembersrcrrrs  zStructModel.__init__cCs||}|j|S)zU StructModel-specific: get the Numba type of the field named *name*. )get_field_positionr)rnameposrrrget_member_fe_type s zStructModel.get_member_fe_typecCs(|jdkr"tdd|jD|_|jS)NcSsg|] }|qSrrr,rrrr/sz.StructModel.get_value_type..)rrrNrrrrrrs    zStructModel.get_value_typecCs(|jdkr"tdd|jD|_|jS)NcSsg|] }|qSrrXr,rrrr/sz-StructModel.get_data_type..)r|rrNrrrrrrs    zStructModel.get_data_typecCstdd|jDS)NcSsg|] }|qSr)rr,rrrr/!sz1StructModel.get_argument_type..)rBrrrrrr szStructModel.get_argument_typecCs|Sr rXrrrrr#szStructModel.get_return_typec Cs@g}t|jD](\}}|t|||||||qt|Sr )rrr5getattrgetrB)rmethnamerrZ extractedrr7rrr_as&s  zStructModel._asc CsTt|tj}tt|j|D],\}\}}t||||}|||||}q"|Sr ) rrrrrrrrset) rrrrstructrr7rQrrrr_from-s zStructModel._fromcCsF|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. rrrrrrrrrrrZelemsrrelrrrr6s zStructModel.as_datacs,fddtt|jD}|d|S)z Convert from "data" representation back into "value" representation. Usually invoked when loading from array. See notes in `as_data()` csg|]}|gqSrrrrrrr/Ssz)StructModel.from_data..r"rr2rrrrrvalsrrrr"Ls  zStructModel.from_datac Csxg}t|jD]0\}}t||d|}||||}||qt|tj } t|D]\}}| || ||} qZ| SNr) rrrr~r)r5rrrrr) rrr(r%rrr;Zelem_ptrrQrrrrr)Ws z"StructModel.load_from_data_pointercCs|d||S)Nr )rrrrrr cszStructModel.as_argumentcCs|d||S)Nr#)rrrrrr#fszStructModel.from_argumentcCsF|d||}t|tj}t|D]\}}||||g}q(|S)Nrrrrrrr!is zStructModel.as_returncs,fddtt|jD}|d|S)Ncsg|]}|gqSrrrrrrr/qsz+StructModel.from_return..r"rrrrrr$ps  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)rrrQrrrrrus     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)rrZstvalrQrrrrrs     zStructModel.setcCs<z|j|WStk r6td|jj|fYnXdS)Nz!%s does not have a field named %r)rindex ValueErrorKeyErrorrdrH)rfieldrrrrs  zStructModel.get_field_positioncCs t|jSr )r2rrrrr field_countszStructModel.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})r?r TypeErrorformatr)krrrrrrrsz$StructModel.traverse..gettercs g|]}|t|fqSr)rr)r-rrrrr/sz(StructModel.traverse..)rr*rrrr+szStructModel.traversecCs|jSr rrrrrr4szStructModel.inner_models)N)rHrIrJrr|rrrrrrrrrr"r)r r#r!r$rrrrLrrrr+r4rerrrcrrs4       rcs eZdZeZfddZZS) ComplexModelcs,d|jfd|jfg}tt||||dS)Nrealimag)Zunderlying_floatr`rrrrcrrrszComplexModel.__init__)rHrIrJNotImplementedZ _element_typerrerrrcrrsrcseZdZfddZZS) TupleModelcs*ddt|D}tt||||dS)NcSs g|]\}}dt||fqS)f)r)r-rr.rrrr/sz'TupleModel.__init__..)rr`rrrrcrrrszTupleModel.__init__rqrrrcrrsrcseZdZfddZZS) UnionModelcs4dtjfdtj|jfg}tt||||dS)Ntagpayload)ruintpTupleZ from_typesr`rrrrcrrrszUnionModel.__init__rqrrrcrrsrcseZdZfddZZS) PairModelcs,d|jfd|jfg}tt||||dS)Nfirstsecond)Z first_typeZ second_typer`rrrrcrrrszPairModel.__init__rqrrrcrrsrcseZdZfddZZS)ListPayloadModelcs>dtjfdtjfdtjfd|jjfg}tt||||dS)NsizeZ allocateddirtydata)rintpboolean containerrjr`rrrrcrrrs  zListPayloadModel.__init__rqrrrcrrsrcseZdZfddZZS) ListModelcs:t|}dt|fdtjfg}tt||||dSNmeminfoparent)r ListPayloadMemInfoPointerpyobjectr`rrrrrZ payload_typerrcrrrs   zListModel.__init__rqrrrcrr srcseZdZfddZZS) ListIterModelcsBt|j}dt|fdttjfg}tt||||dSNrr) rrrrEphemeralPointerrr`rrrrcrrrs   zListIterModel.__init__rqrrrcrrsrcseZdZfddZZS) SetEntryModelcs2|jj}dtjfd|fg}tt||||dS)NrAkey)set_typerjrrr`rr)rrrrjrrcrrr)s zSetEntryModel.__init__rqrrrcrr'srcseZdZfddZZS)SetPayloadModelcsVt|j}dtjfdtjfdtjfdtjfdtjfd|fg}tt||||dS)NfillusedmaskZfingerrentries)rSetEntryrrrr`rr)rrrZ entry_typerrcrrr5s zSetPayloadModel.__init__rqrrrcrr3srcseZdZfddZZS)SetModelcs:t|}dt|fdtjfg}tt||||dSr)r SetPayloadrrr`rrrrcrrrIs   zSetModel.__init__rqrrrcrrGsrcseZdZfddZZS) SetIterModelcsBt|j}dt|fdttjfg}tt||||dSr) rrrrrrr`rrrrcrrrUs   zSetIterModel.__init__rqrrrcrrSsrcseZdZfddZZS) ArrayModelc sv|j}dt|jfdtjfdtjfdtjfdt|jfdttj|fdttj|fg}tt | |||dS)Nrrnitemsitemsizershapestrides) ndimrrrjrrCPointerUniTupler`rrrrrrrrcrrrhs zArrayModel.__init__rqrrrcrrasrcseZdZfddZZS)ArrayFlagsModelcs$d|jfg}tt||||dS)Nr) array_typer`rrrrcrrryszArrayFlagsModel.__init__rqrrrcrrwsrcs$eZdZfddZddZZS)NestedArrayModelcs(||j|_tt|||dSr )rirjr_be_typer`rrr rcrrrszNestedArrayModel.__init__cCst|j|jj}|S)zXReturn the LLVM type representation for the storage of the nestedarray. )rrzrr r)rretrrras_storage_typesz NestedArrayModel.as_storage_type)rHrIrJrrrerrrcrrs rcs<eZdZfddZddZddZddZd d ZZS) OptionalModelcs:d|jfdtjfg}||j|_tt||||dS)Nrvalid)r?rrri _value_modelr`rrrrcrrrs zOptionalModel.__init__cCs |jSr )rrrrrrrszOptionalModel.get_return_typecCstdSr rrrrrr!szOptionalModel.as_returncCs|j||Sr )rr$rrrrr$szOptionalModel.from_returncs:fdd}fddd|fdfgS)Ncs.|}|d}||t|jdS)Nr)rselectrrr?)rrrr get_validrrrget_datasz(OptionalModel.traverse..get_datacs|dS)Nr)rrrrrrsz)OptionalModel.traverse..get_validrr)r)rrr rrrr+s   zOptionalModel.traverse) rHrIrJrrr!r$r+rerrrcrrs  rcsveZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dddZZS) RecordModelcsNtt||fdd|jD_ttd|j_ j _ dS)Ncsg|]\}}j|qSrr)r-_r.rrrr/sz(RecordModel.__init__..rU) r`r rrrrrzr^rrrn _be_ptr_typer rcrrrszRecordModel.__init__cCs|jS)z6Passed around as reference to underlying data r rrrrrszRecordModel.get_value_typecCs|jSr rrrrrrszRecordModel.get_argument_typecCs|jSr rrrrrrszRecordModel.get_return_typecCs|jSr rrrrrrszRecordModel.get_data_typecCs ||Sr r&rrrrrszRecordModel.as_datacCs tddSrwrrrrrr"szRecordModel.from_datacCs|Sr rrrrrr szRecordModel.as_argumentcCs|Sr rrrrrr#szRecordModel.from_argumentcCs|Sr rrrrrr!szRecordModel.as_returncCs|Sr rrrrrr$szRecordModel.from_returnNcCs|||Sr rxr'rrrr)sz"RecordModel.load_from_data_pointer)N)rHrIrJrrrrrrr"r r#r!r$r)rerrrcrr s r cs\eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ Z S)UnicodeCharSeqcs6tt|||ttjd}t||j|_ dSNrU) r`rrrr^rZsizeof_unicode_charrzr{rrrrZchartyrcrrrszUnicodeCharSeq.__init__cCs|jSr rrrrrrszUnicodeCharSeq.get_value_typecCs|jSr rrrrrrszUnicodeCharSeq.get_data_typecCs|Sr rrrrrrszUnicodeCharSeq.as_datacCs|Sr rrrrrr"szUnicodeCharSeq.from_datacCs|Sr rrrrrr!szUnicodeCharSeq.as_returncCs|Sr rrrrrr$szUnicodeCharSeq.from_returncCs|Sr rrrrrr szUnicodeCharSeq.as_argumentcCs|Sr rrrrrr#szUnicodeCharSeq.from_argument rHrIrJrrrrr"r!r$r r#rerrrcrrs rcs\eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ Z S)CharSeqcs0tt|||td}t||j|_dSr)r`rrrr^rzr{rrrcrrrs zCharSeq.__init__cCs|jSr rrrrrrszCharSeq.get_value_typecCs|jSr rrrrrrszCharSeq.get_data_typecCs|Sr rrrrrrszCharSeq.as_datacCs|Sr rrrrrr" szCharSeq.from_datacCs|Sr rrrrrr! szCharSeq.as_returncCs|Sr rrrrrr$szCharSeq.from_returncCs|Sr rrrrrr szCharSeq.as_argumentcCs|Sr rrrrrr#szCharSeq.from_argumentrrrrcrrs rcseZdZfddZZS)CContiguousFlatItercsv|jjdkst|j}|j}|j}d|fdtjfdttjfg}|r^|dt tj|ft t | |||dS)NCarrayZstriderindices) rlayoutrRrjrrrrr5EphemeralArrayr`rr)rrr need_indicesrrjrrrcrrrszCContiguousFlatIter.__init__rqrrrcrrsrcseZdZfddZZS)FlatItercsf|j}|j}|j}d|fdtt||fdttj|fdttjfg}t t | |||dS)NrZpointersr exhausted) rrjrrrrrrrr`rr)rrrrrjrrrcrrr*szFlatIter.__init__rqrrrcrr)srcseZdZfddZZS) UniTupleItercs2dttjfd|jfg}tt||||dS)NrrB)rrrrr`rrrrcrrr8szUniTupleIter.__init__rqrrrcrr6srcseZdZfddZZS) SliceModelcs4dtjfdtjfdtjfg}tt||||dSNstartstopstep)rrr`r rrrcrrrAs zSliceModel.__init__rqrrrcrr >sr cseZdZfddZZS)NPDatetimeModelcs"td}tt||||dS)N@)rr^r`r%rrbrcrrrLs zNPDatetimeModel.__init__rqrrrcrr%Isr%cseZdZfddZZS) ArrayIteratorcs2dttjfd|jfg}tt||||dS)Nrr)rrrrr`r'rrrcrrrSszArrayIterator.__init__rqrrrcrr'Qsr'cseZdZfddZZS) EnumerateTypecs2dttjfd|jfg}tt||||dS)Nr{iter)rrr source_typer`r(rrrcrrr\szEnumerateType.__init__rqrrrcrr(Zsr(cseZdZfddZZS)ZipTypecs,ddt|jD}tt||||dS)NcSsg|]\}}d||jfqS)ziter%d) iterator_type)r-rr*rrrr/fsz$ZipType.__init__..)rZ source_typesr`r+rrrcrrreszZipType.__init__rqrrrcrr+csr+cseZdZfddZZS)RangeIteratorTypecsF|j}dt|fd|fd|fdt|fg}tt||||dS)Nr)r#r$r{) yield_typerrr`r-rrrrZint_typerrcrrrms  zRangeIteratorType.__init__rqrrrcrr-ksr-csleZdZfddZddZddZddZd 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 rir,rrrr/{s z+GeneratorModel.__init__..csg|]}j|qSrrr,rrrr/}scSsg|] }|qSrrXr,rrrr/scSsg|] }|qSrrXr,rrrr/s)r`r0rZ arg_typesZ _arg_modelsZ state_typesZ _state_modelsrrNZ _args_be_typeZ_state_be_typer rirZint32rrrnr r rcrrrxs zGeneratorModel.__init__cCs|jS)zH The generator closure is passed around as a reference. rrrrrrszGeneratorModel.get_value_typecCs|jSr rrrrrrsz GeneratorModel.get_argument_typecCs|jSr rrrrrrszGeneratorModel.get_return_typecCs|jSr rrrrrrszGeneratorModel.get_data_typecCs|Sr rrrrrr szGeneratorModel.as_argumentcCs|Sr rrrrrr#szGeneratorModel.from_argumentcCs |||Sr rYrrrrr!szGeneratorModel.as_returncCs |||Sr r]rrrrr$szGeneratorModel.from_returncCs ||Sr rrrrrrszGeneratorModel.as_datacCst||j}||||Sr )rrZr?r[)rrrstackrrrr"s zGeneratorModel.from_data)rHrIrJrrrrrr r#r!r$rr"rerrrcrr0vs r0cseZdZfddZZS)ArrayCTypesModelcs8dt|jfdt|jfg}tt||||dS)Nrr)rrrjrr`r3rrrcrrrszArrayCTypesModel.__init__rqrrrcrr3sr3cseZdZfddZZS) RangeModelcs6|jj}d|fd|fd|fg}tt||||dSr!)r,r.r`r4rr/rcrrrs zRangeModel.__init__rqrrrcrr4sr4cseZdZfddZZS) NdIndexModelcsP|j}dttj|fdttj|fdttjfg}tt| |||dS)Nrrr) rrrrrrrr`r5rrrcrrrs zNdIndexModel.__init__rqrrrcrr5sr5cCs(|jjdkrt||ddSt||SdS)NrFrrrrrrr\rrrhandle_numpy_flat_types r9cCs(|jjdkrt||ddSt||SdS)NrTr6r7r8rrrhandle_numpy_ndenumerate_types r:cCs ||jSr )thisr8rrrhandle_bound_functionsr<cseZdZfddZZS)NdItercs|j}|j}|jr|nd}dttjfdt|fdttj|fdt tj|fg}t |j D]N\}}|\} } } } d|} | dkr| | ttjfqb| dkrqbd sbt qbt |jD]0\}}t|tjsd |} | | t|fqtt||||dS) NrTrarraysrrzindex%dZflat)ZscalarZindexedZ0drzscalar%d)r>rZneed_shaped_indexingrrrrrrrrZindexersr5rRrArrayr`r=r)rrrZ array_typesrZ shape_lenrrsubkindZ start_dimZend_dimr  member_namer\rcrrrs*    zNdIter.__init__rqrrrcrr=sr=cseZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZd&ddZddZddZd d!Zed"d#Zd$d%ZZS)'DeferredStructModelcs0tt|||dt||_||_dS)Nz deferred.{0})r`rCrridtypenameractual_fe_typer rcrrrszDeferredStructModel.__init__cCstj|jdS)Nz.valuerZglobal_contextZget_identified_typerErrrrr sz"DeferredStructModel.get_value_typecCstj|jdS)Nz.datarGrrrrrsz!DeferredStructModel.get_data_typecCs |jSr ) _actual_modelrrrrrrsz%DeferredStructModel.get_argument_typecCs|||}|j||Sr )rrHr )rrrinnerrrrr s zDeferredStructModel.as_argumentcCs |j||}||||Sr )rHr#rmake_uninitialized)rrrresrrrr#sz!DeferredStructModel.from_argumentcCs8||||}|j||}|}||||Sr )_definerrHr"rJrrrrelemrrrrr"s  zDeferredStructModel.from_datacCs<||||}|j||}|jdd}||||S)Nr)rA)rLrrHrrJrrMrrrr#s   zDeferredStructModel.as_datacCs|Sr rrrrrr$*szDeferredStructModel.from_returncCs|Sr rrrrrr!-szDeferredStructModel.as_returncCs||dgSrrrrrrr0szDeferredStructModel.getcCs|||dgSr)r)rrrcontentrrrr3szDeferredStructModel.setrcCs0||dkr|}n|}t|tjS)Nr)rLrrrrr)rrAr\rrrrJ6s  z&DeferredStructModel.make_uninitializedcCs(|}|||}||dSr )r_define_value_typer_define_data_type)rZvaltyZdattyrrrrL>s zDeferredStructModel._definecCs|jr||jdSr ) is_opaqueset_bodyrHr)rZ value_typerrrrPDsz&DeferredStructModel._define_value_typecCs|jr||jdSr )rRrSrHr)rZ data_typerrrrQHsz%DeferredStructModel._define_data_typecCs|j|jSr )r rirFrrrrrHLsz!DeferredStructModel._actual_modelcs|jfddfgS)Ncs|dgSrrr rrrRz.DeferredStructModel.traverse..)rFr*rrrr+Ps zDeferredStructModel.traverse)r)rHrIrJrrrrr r#r"rr$r!rrrJrLrPrQrLrHr+rerrrcrrCs&   rCcs eZdZdZfddZZS)StructPayloadModelz.Model for the payload of a mutable struct cs"t|j}t|||dSr )rBZ field_dictitemsr`r)rrfe_typrrcrrrYszStructPayloadModel.__init__rkrrrcrrVUsrVcs eZdZdZfddZZS)StructRefModelz?Model for a mutable struct. A reference to the payload cs,|}dt|fg}t|||dS)Nr)rrrr`r)rrrXrjrrcrrrbs zStructRefModel.__init__rkrrrcrrY^srYN) functoolsr collectionsrZllvmliterZnumba.core.datamodel.registryrZ numba.corerrZnumba.nprobjectr r1rMBooleanZBooleanLiteralrSr_rfZ EnumMemberZ IntEnumMemberrhZOpaqueZPyObjectZ RawPointerZNoneTypeZ StringLiteralZ EllipsisTypeFunctionTypeZObjectModuleZPhantomContextManagerZ DispatcherZObjModeDispatcherZExceptionClassDummyZExceptionInstanceZExternalFunctionZ EnumClassZ IntEnumClassZ NumberClassZTypeRefZNamedTupleClassZDTypeZ RecursiveCallZMakeFunctionLiteralZPoisonrlrroIntegerZIntegerLiteralrpFloatrrrrsrrvrryZExternalFunctionPointerrrZ NamedUniTupleZStarArgUniTuplerrrComplexrZ LiteralListZLiteralStrKeyDictr NamedTupleZ StarArgTupler UnionTyperZPairrrrListrZListIterrrrrrSetrZSetIterrr?BufferZ ByteArrayBytesZ MemoryViewZPyArrayrZ ArrayFlagsrZ NestedArrayrOptionalrZRecordr rrrrrmiscZ SliceLiteralZ SliceTyper Z NPDatetimeZ NPTimedeltar%r'r(r+r- Generatorr0Z ArrayCTypesr3Z RangeTyper4ZNumpyNdIndexTyper5Z NumpyFlatTyper9ZNumpyNdEnumerateTyper:Z BoundFunctionr<ZNumpyNdIterTyper=Z DeferredTyperCZStructRefPayloadrVrYrrrrs.     +$ F    EQ        +    6     "P