B \}@sddlmZmZddlmZddlmZddlmZm Z m Z ddl m Z Gddde Ze e jGd d d eZe e jGd d d eZGd ddeZGdddeZe e je e jGdddeZe e je e je e je e je e je e je e je e je e j e e j!e e j"e e j#e e j$e e j%e e j&e e j'e e j(e e j)e e j*e e j+e e j,e e j-e e j.e e j/e e j0e e j1GdddeZ2e e j3Gddde2Z4e e j5e e j6GdddeZ7e e j8GdddeZ9e e j:GdddeZ;e e j<Gddde;Z=e e j>Gdd d e;Z?e e j@Gd!d"d"eZAe e jBe e jCGd#d$d$eZDGd%d&d&eZEGd'd(d(eEZFe e jGGd)d*d*eFZHe e jIe e jJGd+d,d,eFZKe e jLGd-d.d.eFZMe e jNGd/d0d0eFZOe e jPGd1d2d2eFZQe e jRGd3d4d4eFZSe e jTGd5d6d6eFZUe e jVGd7d8d8eFZWe e jXGd9d:d:eFZYe e jZGd;d<dd>eFZbe e jcGd?d@d@eFZde e jeGdAdBdBeFZfe e jgGdCdDdDebZhe e jiGdEdFdFeFZje e jkGdGdHdHeEZle e jmGdIdJdJeZme e jnGdKdLdLeZnGdMdNdNeFZoGdOdPdPeFZpe e jqGdQdRdReFZqe e jrGdSdTdTeFZse e jte e juGdUdVdVeZve e jwGdWdXdXeFZwe e jxGdYdZdZeFZxe e jyGd[d\d\eFZye e jzGd]d^d^eFZze e j{Gd_d`d`eEZ|e e j}GdadbdbeFZ~e e jGdcddddeFZe e jGdedfdfeFZe e jdgdhZe e jdidjZe e jdkdlZe e jGdmdndneFZe e jGdodpdpeEZdqS)r)print_functionabsolute_import)partial)ir)cgutilstypes numpy_support)register_defaultc@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||_||_dS)N)_dmm_fe_type)selfdmmfe_typer5lib/python3.7/site-packages/numba/datamodel/models.py__init__"szDataModel.__init__cCs|jS)N)r )rrrrr&szDataModel.fe_typecCstdS)N)NotImplementedError)rrrrget_value_type*szDataModel.get_value_typecCs|S)N)r)rrrr get_data_type-szDataModel.get_data_typecCs|S)z8Return a LLVM type or nested tuple of LLVM type )r)rrrrget_argument_type0szDataModel.get_argument_typecCs|S)N)r)rrrrget_return_type5szDataModel.get_return_typecCstdS)N)r)rbuildervaluerrras_data8szDataModel.as_datacCs t|dS)z` Takes one LLVM value Return a LLVM value or nested tuple of LLVM value N)r)rrrrrr as_argument;szDataModel.as_argumentcCs t|dS)N)r)rrrrrr as_returnBszDataModel.as_returncCs t|dS)N)r)rrrrrr from_dataEszDataModel.from_datacCs t|dS)za Takes a LLVM value or nested tuple of LLVM value Returns one LLVM value N)r)rrrrrr from_argumentHszDataModel.from_argumentcCstdS)N)r)rrrrrr from_returnOszDataModel.from_returnNcCs|||j||dS)z~ Load value from a pointer to data. This is the default implementation, sufficient for most purposes. )align)rload)rrptrr!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. r)rrrrrtraverseYszDataModel.traversecsfddDS)zE Recursively list all models involved in this model. csg|]}j|qSr)r ).0t)rrr esz-DataModel.traverse_models..)traverse_types)rr)rrtraverse_modelsaszDataModel.traverse_modelscCs|jg|S)zM Recursively list all frontend types involved in this model. )r inner_types)rrrrr)gszDataModel.traverse_typescCsgS)z2 List all *inner* frontend types. r)rrrrr+mszDataModel.inner_typescCsdS)z Returns the MemInfo object or None if it is not tracked. It is only defined for types.meminfo_pointer Nr)rrrrrrget_nrt_meminfosszDataModel.get_nrt_meminfocCsdS)NFr)rrrrhas_nrt_meminfozszDataModel.has_nrt_meminfocCstdd|DS)zQ Recursively check all contained types for need for NRT meminfo. css|]}|VqdS)N)r-)r&modelrrr sz1DataModel.contains_nrt_meminfo..)anyr*)rrrrcontains_nrt_meminfo}szDataModel.contains_nrt_meminfocCst||jfS)N)typer )rrrr_compared_fieldsszDataModel._compared_fieldscCstt|S)N)hashtupler3)rrrr__hash__szDataModel.__hash__cCs(t|t|kr ||kSdSdS)NF)r2r3)rotherrrr__eq__szDataModel.__eq__cCs || S)N)r8)rr7rrr__ne__szDataModel.__ne__)N)__name__ __module__ __qualname____doc__rpropertyrrrrrrrrrrr r$r%r*r)r+r,r-r1r3r6r8r9rrrrr s2  r c@s(eZdZdZddZddZddZdS) OmittedArgDataModelz A data model for omitted arguments. Only the "argument" representation is defined, other representations raise a NotImplementedError. cCsdS)Nrr)rrrrrsz%OmittedArgDataModel.get_argument_typecCsdS)Nrr)rrvalrrrrszOmittedArgDataModel.as_argumentcCs|dkst|dS)Nr)AssertionError)rrr@rrrrsz!OmittedArgDataModel.from_argumentN)r:r;r<r=rrrrrrrr?sr?c@speZdZedZedZddZddZddZ d d Z d d Z d dZ ddZ ddZddZddZdS) BooleanModelr cCs|jS)N) _bit_type)rrrrrszBooleanModel.get_value_typecCs|jS)N) _byte_type)rrrrrszBooleanModel.get_data_typecCs|S)N)r)rrrrrszBooleanModel.get_return_typecCs|S)N)r)rrrrrszBooleanModel.get_argument_typecCs|||S)N)Zzextr)rrrrrrrszBooleanModel.as_datacCs |||S)N)r)rrrrrrrszBooleanModel.as_argumentcCs |||S)N)r)rrrrrrrszBooleanModel.as_returncCs|||S)N)Ztruncr)rrrrrrrszBooleanModel.from_datacCs |||S)N)r)rrrrrrrszBooleanModel.from_argumentcCs |||S)N)r)rrrrrrr szBooleanModel.from_returnN)r:r;r<rIntTyperDrErrrrrrrrrr rrrrrBs  rBcsXeZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ Z S)PrimitiveModelzZA primitive type can be represented natively in the target in all usage contexts. cstt|||||_dS)N)superrGrbe_type)rrrrI) __class__rrrszPrimitiveModel.__init__cCs|jS)N)rI)rrrrrszPrimitiveModel.get_value_typecCs|S)Nr)rrrrrrrszPrimitiveModel.as_datacCs|S)Nr)rrrrrrrszPrimitiveModel.as_argumentcCs|S)Nr)rrrrrrrszPrimitiveModel.as_returncCs|S)Nr)rrrrrrrszPrimitiveModel.from_datacCs|S)Nr)rrrrrrrszPrimitiveModel.from_argumentcCs|S)Nr)rrrrrrr szPrimitiveModel.from_return) r:r;r<r=rrrrrrrr __classcell__rr)rJrrGs rGc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS) ProxyModelzB Helper class for models which delegate to another model. cCs |jS)N)_proxied_modelr)rrrrrszProxyModel.get_value_typecCs |jS)N)rMr)rrrrrszProxyModel.get_data_typecCs |jS)N)rMr)rrrrrszProxyModel.get_return_typecCs |jS)N)rMr)rrrrrszProxyModel.get_argument_typecCs|j||S)N)rMr)rrrrrrrszProxyModel.as_datacCs|j||S)N)rMr)rrrrrrrszProxyModel.as_argumentcCs|j||S)N)rMr)rrrrrrrszProxyModel.as_returncCs|j||S)N)rMr)rrrrrrrszProxyModel.from_datacCs|j||S)N)rMr)rrrrrrrszProxyModel.from_argumentcCs|j||S)N)rMr )rrrrrrr szProxyModel.from_returnN)r:r;r<r=rrrrrrrrrr rrrrrLsrLcs eZdZdZfddZZS) EnumModelzA Enum members are represented exactly like their values. cs$tt|||||j|_dS)N)rHrNrlookupdtyperM)rrr)rJrrrszEnumModel.__init__)r:r;r<r=rrKrr)rJrrN srNcs.eZdZdZedZfddZZ S) OpaqueModelz# Passed as opaque pointers rCcs|j}tt||||dS)N) _ptr_typerHrQr)rrrrI)rJrrr8szOpaqueModel.__init__) r:r;r<r=rrF as_pointerrRrrKrr)rJrrQsrQc@s$eZdZddZddZddZdS) MemInfoModelcCs|j|jjS)N)r rOr rPr))rrrrr+@szMemInfoModel.inner_typescCsdS)NTr)rrrrr-CszMemInfoModel.has_nrt_meminfocCs|S)Nr)rrrrrrr,FszMemInfoModel.get_nrt_meminfoN)r:r;r<r+r-r,rrrrrT=srTcseZdZfddZZS) IntegerModelcs$t|j}tt||||dS)N)rrFZbitwidthrHrUr)rrrrI)rJrrrMs zIntegerModel.__init__)r:r;r<rrKrr)rJrrUJsrUcseZdZfddZZS) FloatModelcsH|tjkrt}n|tjkr(t}nt|tt| |||dS)N) rZfloat32rZ FloatTypeZfloat64Z DoubleTyperrHrVr)rrrrI)rJrrrTs     zFloatModel.__init__)r:r;r<rrKrr)rJrrVRsrVcseZdZfddZZS) PointerModelcs<||j|_|j|_|j}tt||||dS)N) rOrP_pointee_modelr_pointee_be_typerSrHrWr)rrrrI)rJrrr`s  zPointerModel.__init__)r:r;r<rrKrr)rJrrW^srWc@s.eZdZddZddZddZd dd ZdS) EphemeralPointerModelcCs|jS)N)rY)rrrrrjsz#EphemeralPointerModel.get_data_typecCs||}|j||S)N)r"rXr)rrrrrrrms zEphemeralPointerModel.as_datacCs tddS)Nz$use load_from_data_pointer() instead)r)rrrrrrrqszEphemeralPointerModel.from_dataNcCs|||S)N)bitcastr)rrr#r!rrrr$tsz,EphemeralPointerModel.load_from_data_pointer)N)r:r;r<rrrr$rrrrrZgsrZcs>eZdZfddZddZddZddZd d d ZZS) EphemeralArrayModelcs*tt|||t|j|jj|_dS)N) rHr\rr ArrayTyperYr count _data_type)rrr)rJrrr{szEphemeralArrayModel.__init__cCs|jS)N)r_)rrrrrsz!EphemeralArrayModel.get_data_typecs(fddt|jjD}t|S)Nc s g|]}t|qSr)r"r gep_inbounds)r&i)rrrrr(sz/EphemeralArrayModel.as_data..)ranger r^rZ pack_array)rrrvaluesr)rrrrs zEphemeralArrayModel.as_datacCs tddS)Nz$use load_from_data_pointer() instead)r)rrrrrrrszEphemeralArrayModel.from_dataNcCs|||S)N)r[r)rrr#r!rrrr$sz*EphemeralArrayModel.load_from_data_pointer)N) r:r;r<rrrrr$rKrr)rJrr\xs  r\cseZdZfddZZS)ExternalFuncPointerModelcsT|j}|j}fdd|jD}tt||}tt | ||dS)Ncsg|]}|qSr)rOr)r&r')rrrr(sz5ExternalFuncPointerModel.__init__..) sigrOZ return_typerargsrZ PointerType FunctionTyperHrdr)rrrreZrettyrfrI)rJ)rrrs z!ExternalFuncPointerModel.__init__)r:r;r<rrKrr)rJrrdsrdcs|eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZZS) UniTupleModelcsZtt|||||j|_t||_t |j |j|_ t |j |j|_ dS)N)rHrhrrOrP _elem_modellen_countrr]r _value_typerr_)rrr)rJrrrs    zUniTupleModel.__init__cCs|jS)N)rl)rrrrrszUniTupleModel.get_value_typecCs|jS)N)r_)rrrrrszUniTupleModel.get_data_typecCs|S)N)r)rrrrrszUniTupleModel.get_return_typecCs|jf|jS)N)rirrk)rrrrrszUniTupleModel.get_argument_typecCsBg}x8t|jD]*}|||g}|j||}||qW|S)N)rbrk extract_valuerirappend)rrroutravrrrrs zUniTupleModel.as_argumentcCsJt|tj}x2t|D]&\}}|j||}||||g}qW|S)N)rConstantr Undefined enumeraterir insert_value)rrrrorarprrrrs zUniTupleModel.from_argumentcCsVt|tj}x>t|jD]0}|||g}|j||}| |||g}qW|S)N) rrqrrrrbrkrmrirrt)rrrrorar@dvalrrrrs zUniTupleModel.as_datacCsVt|tj}x>t|jD]0}|||g}|j||}| |||g}qW|S)N) rrqrrrrbrkrmrirrt)rrrrorar@rurrrrs zUniTupleModel.from_datacCs|S)Nr)rrrrrrrszUniTupleModel.as_returncCs|S)Nr)rrrrrrr szUniTupleModel.from_returncs&fddfddtjDS)Ncs ||S)N)rm)rar)rrrgettersz&UniTupleModel.traverse..gettercsg|]}jjt|fqSr)r rPr)r&ra)rvrrrr(sz*UniTupleModel.traverse..)rbrk)rrr)rrvrrr%s  zUniTupleModel.traversecCs |jS)N)rir))rrrrr+szUniTupleModel.inner_types)r:r;r<rrrrrrrrrrr r%r+rKrr)rJrrhs rhc@seZdZdZdS)CompositeModelzWAny model that is composed of multiple other models should subclass from this. N)r:r;r<r=rrrrrwsrwcseZdZdZdZfddZddZddZdd 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|qSr)r rO)r&r')rrrr(sz(StructModel.__init__..)rHrxrzip_fields_membersr5_models)rrrmembers)rJ)rrrs  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_typecCs(|jdkr"tdd|jD|_|jS)NcSsg|] }|qSr)r)r&r'rrrr(sz.StructModel.get_value_type..)rlrLiteralStructTyper|)rrrrrs  zStructModel.get_value_typecCs(|jdkr"tdd|jD|_|jS)NcSsg|] }|qSr)r)r&r'rrrr(sz-StructModel.get_data_type..)r_rrr|)rrrrrs  zStructModel.get_data_typecCstdd|jDS)NcSsg|] }|qSr)r)r&r'rrrr( sz1StructModel.get_argument_type..)r5r|)rrrrrszStructModel.get_argument_typecCs|S)N)r)rrrrr szStructModel.get_return_typec CsDg}x6t|jD](\}}|t|||||||qWt|S)N)rsr|rngetattrgetr5)rmethnamerrZ extractedradmrrr_ass zStructModel._asc CsXt|tj}x@tt|j|D],\}\}}t||||}|||||}q$W|S)N) rrqrrrrsryr|rset) rrrrstructrarr@rprrr_froms zStructModel._fromcCsJ|d||}t|tj}x$t|D]\}}||||g}q*W|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)rrrqrrrrsrt)rrrelemsrraelrrrrs 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|]}|gqSr)rm)r&ra)rrrrr(;sz)StructModel.from_data..r)rbrjr{r)rrrvalsr)rrrr4s zStructModel.from_datac Csg}x>t|jD]0\}}t||d|}||||}||qWt|tj } x$t|D]\}}| || ||} q`W| S)Nr) rsr|rr`r$rnrrqrrrr) rrr#r!rcrar.Zelem_ptrr@rrrrr$?sz"StructModel.load_from_data_pointercCs|d||S)Nr)r)rrrrrrrKszStructModel.as_argumentcCs|d||S)Nr)r)rrrrrrrNszStructModel.from_argumentcCsJ|d||}t|tj}x$t|D]\}}||||g}q*W|S)Nr)rrrqrrrrsrt)rrrrrrarrrrrQs zStructModel.as_returncs,fddtt|jD}|d|S)Ncsg|]}|gqSr)rm)r&ra)rrrrr(Ysz+StructModel.from_return..r)rbrjr{r)rrrrr)rrrr Xs 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) isinstancestrr~rmrz)rrr@rrrrr]s   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)rrr~rtrz)rrZstvalr@rrrrrrs   zStructModel.setcCs:y |j|Stk r4td|jj|fYnXdS)Nz!%s does not have a field named %r)rzindex ValueErrorKeyErrorrJr:)rZfieldrrrr~s  zStructModel.get_field_positioncCs t|jS)N)rjrz)rrrr 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 )rrr~r{)rrrrrget_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 )r|)rrrrr get_models zStructModel.get_modelcs$fddfddjDS)Ncs8|jkr*|jf}tdj|||S)Nzexpecting {0} but got {1})r2r TypeErrorformatr)krrf)rrrrrvsz$StructModel.traverse..gettercs g|]}|t|fqSr)rr)r&r)rvrrrr(sz(StructModel.traverse..)rz)rrr)rrvrrr%szStructModel.traversecCs$g}x|jD]}||7}q W|S)N)r|r))rrrrrrr+s zStructModel.inner_types)N)r:r;r<rlr_rrrrrrrrrrr$rrrr rrr~r>rrrr%r+rKrr)rJrrxs2       rxcs eZdZeZfddZZS) ComplexModelcs,d|jfd|jfg}tt||||dS)Nrealimag)Zunderlying_floatrHrr)rrrr})rJrrrs zComplexModel.__init__)r:r;r<NotImplementedZ _element_typerrKrr)rJrrsrcseZdZfddZZS) TupleModelcs*ddt|D}tt||||dS)NcSs g|]\}}dt||fqS)f)r)r&rar'rrrr(sz'TupleModel.__init__..)rsrHrr)rrrr})rJrrrszTupleModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) PairModelcs,d|jfd|jfg}tt||||dS)Nfirstsecond)Z first_typeZ second_typerHrr)rrrr})rJrrrs zPairModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS)ListPayloadModelcs>dtjfdtjfdtjfd|jjfg}tt||||dS)NsizeZ allocateddirtydata)rintpboolean containerrPrHrr)rrrr})rJrrrs zListPayloadModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) ListModelcs:t|}dt|fdtjfg}tt||||dS)Nmeminfoparent)r ListPayloadMemInfoPointerpyobjectrHrr)rrr payload_typer})rJrrrs   zListModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) ListIterModelcsBt|j}dt|fdttjfg}tt||||dS)Nrr) rrrrEphemeralPointerrrHrr)rrrrr})rJrrrs  zListIterModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) SetEntryModelcs2|jj}dtjfd|fg}tt||||dS)Nr4key)Zset_typerPrrrHrr)rrrrPr})rJrrrs zSetEntryModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS)SetPayloadModelcsVt|j}dtjfdtjfdtjfdtjfdtjfd|fg}tt||||dS)NZfillZusedmaskZfingerrentries)rSetEntryrrrrHrr)rrrZ entry_typer})rJrrrs  zSetPayloadModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS)SetModelcs:t|}dt|fdtjfg}tt||||dS)Nrr)r SetPayloadrrrHrr)rrrrr})rJrrr%s   zSetModel.__init__)r:r;r<rrKrr)rJrr#srcseZdZfddZZS) SetIterModelcsBt|j}dt|fdttjfg}tt||||dS)Nrr) rrrrrrrHrr)rrrrr})rJrrr1s  zSetIterModel.__init__)r:r;r<rrKrr)rJrr/srcseZdZfddZZS) ArrayModelc sv|j}dt|jfdtjfdtjfdtjfdt|jfdttj|fdttj|fg}tt | |||dS)Nrrnitemsitemsizershapestrides) ndimrrrPrrCPointerUniTuplerHrr)rrrrr})rJrrrDszArrayModel.__init__)r:r;r<rrKrr)rJrr=srcseZdZfddZZS)ArrayFlagsModelcs$d|jfg}tt||||dS)Nr) array_typerHrr)rrrr})rJrrrUs zArrayFlagsModel.__init__)r:r;r<rrKrr)rJrrSsrcseZdZfddZZS)SmartArrayModelcs,dtjfd|jfg}tt||||dS)Nrr)rrZas_arrayrHrr)rrrr})rJrrr^s zSmartArrayModel.__init__)r:r;r<rrKrr)rJrr\srcs$eZdZfddZddZZS)NestedArrayModelcs(||j|_tt|||dS)N)rOrPr_be_typerHrr)rrr)rJrrrhszNestedArrayModel.__init__cCst|j|jj}|S)N)rr]rr r)rZretrrrrlszNestedArrayModel.get_data_type)r:r;r<rrrKrr)rJrrfs rcs<eZdZfddZddZddZddZd d ZZS) OptionalModelcs:d|jfdtjfg}||j|_tt||||dS)Nrvalid)r2rrrO _value_modelrHrr)rrrr})rJrrrss zOptionalModel.__init__cCs |jS)N)rr)rrrrr{szOptionalModel.get_return_typecCstdS)N)r)rrrrrrr~szOptionalModel.as_returncCs|j||S)N)rr )rrrrrrr szOptionalModel.from_returncs:fdd}fddd|fdfgS)Ncs.|}|d}||t|jdS)Nr)rZselectrrqr2)rrr)r get_validrrrget_datasz(OptionalModel.traverse..get_datacs|dS)Nr)r)r)rrrrrsz)OptionalModel.traverse..get_validrr)r)rrrr)rrrrr%s zOptionalModel.traverse) r:r;r<rrrr r%rKrr)rJrrqs  rcsveZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dddZZS) RecordModelcsNtt||fdd|jD_ttd|j_ j _ dS)Ncsg|]\}}j|qSr)r rO)r&_r')rrrr(sz(RecordModel.__init__..rC) rHrrr}r|rr]rFrrrS _be_ptr_type)rrr)rJ)rrrszRecordModel.__init__cCs|jS)z6Passed around as reference to underlying data )r)rrrrrszRecordModel.get_value_typecCs|jS)N)r)rrrrrszRecordModel.get_argument_typecCs|jS)N)r)rrrrrszRecordModel.get_return_typecCs|jS)N)r)rrrrrszRecordModel.get_data_typecCs ||S)N)r")rrrrrrrszRecordModel.as_datacCs tddS)Nz$use load_from_data_pointer() instead)r)rrrrrrrszRecordModel.from_datacCs|S)Nr)rrrrrrrszRecordModel.as_argumentcCs|S)Nr)rrrrrrrszRecordModel.from_argumentcCs|S)Nr)rrrrrrrszRecordModel.as_returncCs|S)Nr)rrrrrrr szRecordModel.from_returnNcCs|||S)N)r[r)rrr#r!rrrr$sz"RecordModel.load_from_data_pointer)N)r:r;r<rrrrrrrrrrr r$rKrr)rJrrs rcs,eZdZfddZddZddZZS)UnicodeCharSeqcs6tt|||ttjd}t||j|_ dS)NrC) rHrrrrFrZsizeof_unicode_charr]r^r)rrrcharty)rJrrrszUnicodeCharSeq.__init__cCs|jS)N)r)rrrrrszUnicodeCharSeq.get_value_typecCs|jS)N)r)rrrrrszUnicodeCharSeq.get_data_type)r:r;r<rrrrKrr)rJrrs rcs\eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ Z S)CharSeqcs0tt|||td}t||j|_dS)NrC)rHrrrrFr]r^r)rrrr)rJrrrs zCharSeq.__init__cCs|jS)N)r)rrrrrszCharSeq.get_value_typecCs|jS)N)r)rrrrrszCharSeq.get_data_typecCs|S)Nr)rrrrrrrszCharSeq.as_datacCs|S)Nr)rrrrrrrszCharSeq.from_datacCs|S)Nr)rrrrrrrszCharSeq.as_returncCs|S)Nr)rrrrrrr szCharSeq.from_returncCs|S)Nr)rrrrrrrszCharSeq.as_argumentcCs|S)Nr)rrrrrrrszCharSeq.from_argument) r:r;r<rrrrrrr rrrKrr)rJrrs rcseZdZfddZZS)CContiguousFlatItercsv|jjdkst|j}|j}|j}d|fdtjfdttjfg}|r^|dt tj|ft t | |||dS)NCarrayZstriderindices) rlayoutrArPrrrrrnEphemeralArrayrHrr)rrr need_indicesrrPrr})rJrrrszCContiguousFlatIter.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS)FlatItercsf|j}|j}|j}d|fdtt||fdttj|fdttjfg}t t | |||dS)NrZpointersr exhausted) rrPrrrrrrrrHrr)rrrrrPrr})rJrrrszFlatIter.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) UniTupleItercs2dttjfd|jfg}tt||||dS)Nrr5)rrrrrHrr)rrrr})rJrrr s zUniTupleIter.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) SliceModelcs4dtjfdtjfdtjfg}tt||||dS)Nstartstopstep)rrrHrr)rrrr})rJrrrs zSliceModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS)NPDatetimeModelcs"td}tt||||dS)N@)rrFrHrr)rrrrI)rJrrrs zNPDatetimeModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) ArrayIteratorcs2dttjfd|jfg}tt||||dS)Nrr)rrZuintprrHrr)rrrr})rJrrr#s zArrayIterator.__init__)r:r;r<rrKrr)rJrr!srcseZdZfddZZS) EnumerateTypecs2dttjfd|jfg}tt||||dS)Nr^iter)rrr source_typerHrr)rrrr})rJrrr,s zEnumerateType.__init__)r:r;r<rrKrr)rJrr*srcseZdZfddZZS)ZipTypecs,ddt|jD}tt||||dS)NcSsg|]\}}d||jfqS)ziter%d) iterator_type)r&rarrrrr(6sz$ZipType.__init__..)rsZ source_typesrHrr)rrrr})rJrrr5szZipType.__init__)r:r;r<rrKrr)rJrr3srcseZdZfddZZS)RangeIteratorTypecsF|j}dt|fd|fd|fdt|fg}tt||||dS)Nrrrr^) yield_typerrrHrr)rrrint_typer})rJrrr=s  zRangeIteratorType.__init__)r:r;r<rrKrr)rJrr;srcsleZdZfddZddZddZddZd 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 rO)r&r')rrrr(Ksz+GeneratorModel.__init__..csg|]}j|qSr)r rO)r&r')rrrr(MscSsg|] }|qSr)r)r&r'rrrr(PscSsg|] }|qSr)r)r&r'rrrr(Rs)rHrrZ arg_typesZ _arg_modelsZ state_typesZ _state_modelsrrZ _args_be_typeZ_state_be_typer rOrZint32rrrSr)rrr)rJ)rrrHszGeneratorModel.__init__cCs|jS)zH The generator closure is passed around as a reference. )r)rrrrrYszGeneratorModel.get_value_typecCs|jS)N)r)rrrrr_sz GeneratorModel.get_argument_typecCs|jS)N)r)rrrrrbszGeneratorModel.get_return_typecCs|jS)N)r)rrrrreszGeneratorModel.get_data_typecCs|S)Nr)rrrrrrrhszGeneratorModel.as_argumentcCs|S)Nr)rrrrrrrkszGeneratorModel.from_argumentcCs |||S)N)r)rrrrrrrnszGeneratorModel.as_returncCs |||S)N)r)rrrrrrr qszGeneratorModel.from_returncCs ||S)N)r")rrrrrrrtszGeneratorModel.as_datacCst||j}||||S)N)rZ alloca_oncer2Zstore)rrrstackrrrrws zGeneratorModel.from_data)r:r;r<rrrrrrrrr rrrKrr)rJrrFs rcseZdZfddZZS)ArrayCTypesModelcs8dt|jfdt|jfg}tt||||dS)Nrr)rrrPrrHrr)rrrr})rJrrrszArrayCTypesModel.__init__)r:r;r<rrKrr)rJrr}srcseZdZfddZZS) RangeModelcs6|jj}d|fd|fd|fg}tt||||dS)Nrrr)rrrHrr)rrrrr})rJrrrs  zRangeModel.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZZS) NdIndexModelcsP|j}dttj|fdttj|fdttjfg}tt| |||dS)Nrrr) rrrrrrrrHrr)rrrrr})rJrrrs zNdIndexModel.__init__)r:r;r<rrKrr)rJrrsrcCs(|jjdkrt||ddSt||SdS)NrF)r)rrrr)rtyrrrhandle_numpy_flat_types rcCs(|jjdkrt||ddSt||SdS)NrT)r)rrrr)rrrrrhandle_numpy_ndenumerate_types rcCs ||jS)N)this)rrrrrhandle_bound_functionsrcseZdZfddZZS)NdItercs|j}|j}|jr|nd}dttjfdt|fdttj|fdt tj|fg}x\t |j D]N\}}|\} } } } d|} | dkr| | ttjfqd| dkrqdd sdt qdWx>t |jD]0\}}t|tjsd |} | | t|fqWtt||||dS) Nr rarraysrrzindex%dZflat)ZscalarZindexedZ0drzscalar%d)rrZneed_shaped_indexingrrrTuplerrrrsZindexersrnrArArrayrHrr)rrrZ array_typesrZ shape_lenr}rasubkindZ start_dimZend_dimr member_namer)rJrrrs(    zNdIter.__init__)r:r;r<rrKrr)rJrrsrcseZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZd&ddZddZddZd d!Zed"d#Zd$d%ZZS)'DeferredStructModelcs0tt|||dt||_||_dS)Nz deferred.{0})rHrrridtypenameractual_fe_type)rrr)rJrrrszDeferredStructModel.__init__cCstj|jdS)Nz.value)rglobal_contextget_identified_typer)rrrrrsz"DeferredStructModel.get_value_typecCstj|jdS)Nz.data)rrrr)rrrrrsz!DeferredStructModel.get_data_typecCs |jS)N) _actual_modelr)rrrrrsz%DeferredStructModel.get_argument_typecCs|||}|j||S)N)rrr)rrrinnerrrrrs zDeferredStructModel.as_argumentcCs |j||}||||S)N)rrrmake_uninitialized)rrrZresrrrrsz!DeferredStructModel.from_argumentcCs8||||}|j||}|}||||S)N)_definerrrrr)rrrelemrorrrrs  zDeferredStructModel.from_datacCs<||||}|j||}|jdd}||||S)Nr)r)rrrrrr)rrrrrorrrrs   zDeferredStructModel.as_datacCs|S)Nr)rrrrrrr szDeferredStructModel.from_returncCs|S)Nr)rrrrrrrszDeferredStructModel.as_returncCs||dgS)Nr)rm)rrrrrrrszDeferredStructModel.getcCs|||dgS)Nr)rt)rrrZcontentrrrrszDeferredStructModel.setrcCs0||dkr|}n|}t|tjS)Nr)rrrrrqrr)rrrrrrrs  z&DeferredStructModel.make_uninitializedcCs(|}|||}||dS)N)r_define_value_typer_define_data_type)rZvaltyZdattyrrrrs zDeferredStructModel._definecCs|jr||jdS)N) is_opaqueset_bodyrr)rZ value_typerrrr sz&DeferredStructModel._define_value_typecCs|jr||jdS)N)r r rr)rZ data_typerrrr sz%DeferredStructModel._define_data_typecCs|j|jS)N)r rOr)rrrrrsz!DeferredStructModel._actual_modelcs|jfddfgS)Ncs|dgS)Nr)rm)r)rrr"sz.DeferredStructModel.traverse..)r)rrr)rrr% szDeferredStructModel.traverse)r)r:r;r<rrrrrrrrr rrrrrr r r>rr%rKrr)rJrrs$   rN)Z __future__rr functoolsrZllvmliterZnumbarrrregistryr objectr rr?ZBooleanrBrGrLZ EnumMemberZ IntEnumMemberrNZOpaqueZPyObjectZ RawPointerZNoneTypeZ StringLiteralZ EllipsisTypeZFunctionZTypeZObjectZModuleZPhantomZContextManagerZ DispatcherZObjModeDispatcherZExceptionClassZDummyZExceptionInstanceZExternalFunctionZMacroZ EnumClassZ IntEnumClassZ NumberClassZTypeRefZNamedTupleClassZDTypeZ RecursiveCallrQrrTZIntegerZIntegerLiteralrUZFloatrVrrWrrZrr\ZExternalFunctionPointerrdrZ NamedUniTuplerhrwrxZComplexrrZ NamedTuplerZPairrrrZListrZListIterrrrrrSetrZSetIterrrZBufferZ ByteArrayZBytesZ MemoryViewZPyArrayrZ ArrayFlagsrZSmartArrayTyperZ NestedArrayrZOptionalrZRecordrrrrrrZ SliceTyperZ NPDatetimeZ NPTimedeltarrrrr GeneratorrZ ArrayCTypesrZ RangeTyperZNumpyNdIndexTyperZ NumpyFlatTyperZNumpyNdEnumerateTyperZ BoundFunctionrZNumpyNdIterTyperZ DeferredTyperrrrrs    #$ :      ET         +     6  "