Aekq6ddlmZddlmZddlmZddlmZddlmZddl mZm Z ddl m Z dd l m Z dd l mZdd l mZdd l mZd?d ZGddeZdZdddddZdZdZdZdZdZdZdZdZdZd@dZGdd eZ d!Z!d"Z"d#Z#d$Z$d%Z%d&Z&d'Z'd(Z(d)Z)d*Z*d+Z+d,Z,d-Z-d.Z.d/Z/Gd0d1eZ0d2Z1dAd3Z2dAd4Z3e4e jj5Z6e3d6e67Z7e3d8Z8e3d9Z9e3d:Z:e3d;Z;e3dZ>y)B)absolute_import)CythonTransform) ModuleNode) CompileError)CythonUtilityCode) UtilityCodeTempitaUtilityCode)Options) Interpreter) PyrexTypes)Naming)Symtabcddlm}||}|dkDr6d|z}dj|jdDcgc]}||z c}}|Scc}w)Nr)dedent  )textwraprjoinsplit)textreindentrindentxs 6lib/python3.12/site-packages/Cython/Compiler/Buffer.pyrrsO $?&1*?@ K@s A c8eZdZdZdZfdZdZdZdZxZ S)IntroduceBufferAuxiliaryVarsFct|tsJd|_tt||}|j r*t|jt|j|S)Nr) isinstancermax_ndimsuperr__call__buffers_existsuse_bufstruct_declare_codescopeuse_py2_buffer_functions)selfnoderesult __class__s rr"z%IntroduceBufferAuxiliaryVars.__call__ sO$ ++ 3TCDI    &tzz 2 $TZZ 0 c jj}|Dcgc]\}}|jjs| }}}t |dkDr|j dd|_|Dcgc]\}}|jjs| }}}t |dkDrd|_|D].\dk(s tjts'd|_ n~ttr$t |dkDrtjd|D]Ajjj rtjdj"j}|j$t&j(kDr'tjdt&j(z|j$|j*kDr|j$|_fd } t,j.t0j2ft,j4t0j6ff} | D cgc]\} } | | | c} } \} }t9j:| |_D|_|_ ycc}}wcc}}wcc} } w) Nrc|jSN)name)entrys rz;IntroduceBufferAuxiliaryVars.handle_scope..5s 5::r+)keyT memoryviewz'Buffer vars not allowed in module scopez-Buffers with pointer types not yet supported.z1Buffer ndims exceeds Options.buffer_max_dims = %dcj|}jd||j}jrd|_|S)N)r/cnametypeposT)mangle declare_varr7is_argused)r6prefixr5aux_varr0r/r(r%s rdecvarz9IntroduceBufferAuxiliaryVars.handle_scope..decvarSsI VT2++U15488,E<<#'GLr+)!entriesitemsr6 is_bufferlensortr#is_memoryviewslicerutility_code_definitionrusing_memoryviewrrr7dtypeis_ptrr/ndimr buffer_max_dimsr r c_pyx_buffer_nd_typerpybuffernd_prefixc_pyx_buffer_typepybufferstruct_prefixr BufferAux buffer_auxbuffer_entriesr%)r'r(r% scope_itemsr/r0bufvarsmemviewslicevarsbuftyper>auxvarsr6r< pybufferndrcbuffers `` `` r handle_scopez)IntroduceBufferAuxiliaryVars.handle_scope.s$mm))+ ,7P[T55::;O;O5PP w.assert_bools2 KK !T"y,*=> >#r+rfrg)buffer_defaultsr interpret_compiletime_optionsrBbuffer_positional_options_countrERR_BUF_TOO_MANYr@buffer_optionsERR_BUF_OPTION_UNKNOWNzip ERR_BUF_DUPKeyErrorERR_BUF_MISSINGrqis_extension_type ERR_BUF_DTYPErint ERR_BUF_NDIM ERR_BUF_MODE)rtenvposargsdictargsdefaults need_completer/valuer7rGrIrervrus` @ranalyse_buffer_optionsrs"#AAC<AGX 7|5572;q>+;<<G&nn.luc~%s$:T$AB B  ".':luc~%s$:T$AB B 7?sK$$67 7 Jw J ( J KK E ((9m44 ;;v D Zc*dQh9l33 ;;v D T@@9l33? "# N1 J &y/D2HII! Js-FF>=F>c6eZdZdZdZdZdZdZdZdZ y) BufferEntryc||_|j|_|jjj|_d|jz|_|jj |_|jy)Nz%s.rcbuffer->pybuffer.buf) r0r6rPbuflocal_nd_varr5buf_ptrbuffer_ptr_type buf_ptr_typeinit_attributes)r'r0s r__init__zBufferEntry.__init__s[ JJ %%55;; 2TZZ? !JJ66 r+c|j|_|j|_|j |_yr.)get_buf_shapevarsshapeget_buf_stridevarsstridesget_buf_suboffsetvars suboffsetsr's rrzBufferEntry.init_attributess2++- ..0 446r+c$|jdS)Nz%s.diminfo[%d].suboffsets _for_all_ndimrs rrz!BufferEntry.get_buf_suboffsetvarss!!"=>>r+c$|jdS)Nz%s.diminfo[%d].stridesrrs rrzBufferEntry.get_buf_stridevarss!!":;;r+c$|jdS)Nz%s.diminfo[%d].shaperrs rrzBufferEntry.get_buf_shapevarss!!"899r+ct|jjDcgc]}||j|fzc}Scc}wr.)ranger6rIr5)r'sis rrzBufferEntry._for_all_ndims0-2499>>-BCTZZO#CCCs=c g}|jj}|jj}|dk(rqt||j |j D]9\}}}|j ||j ||j |;d|z} t} ny|dk(r d|z} t} n$|dk(r d|z} t} n|dk(r d|z} t} nJt||j D]'\}}|j ||j |)| |jjvrO|jjj| |jd } |jd } | | | | | |jj} | d | d |j d d j#|d}|S)Nrhz__Pyx_BufPtrFull%ddrmz__Pyx_BufPtrStrided%ddrnz__Pyx_BufPtrCContig%ddroz__Pyx_BufPtrFortranContig%ddutility_code_protoutility_code_def)r/nd(, ))r6rIrer}rrappendbuf_lookup_full_codebuf_lookup_strided_codebuf_lookup_c_codebuf_lookup_fortran_code globalstate utility_codesaddrempty_declaration_coderr)r'code index_cnamesparamsrrerrofuncnamefuncgen protocodedefcodebuf_ptr_type_codeptrcodes rgenerate_buffer_lookup_codez'BufferEntry.generate_buffer_lookup_codes YY^^yy~~ 6>|#668#99;= !1a a  a  a  ! -r1H*Gy 3b813b8+"9B>1L$*A*A*CD !1 a  a  ! 4++99 9    * * . .x 8(()=>I&&'9:G IwX" = --DDF&.0A4<<&*ii&79r+N) rarbrcrrrrrrrr+rrrs&7 ?<:D*r+rcd}|j}|dk(r|dz }n#|dk(r|dz }n|dk(r|dz }n |dk(r|d z }nJ|jr|d z }|S) N PyBUF_FORMATrhz| PyBUF_INDIRECTrmz| PyBUF_STRIDESrnz| PyBUF_C_CONTIGUOUSroz| PyBUF_F_CONTIGUOUSz| PyBUF_WRITABLE)rewritable_needed)rP buffer_typeflagsres r get_flagsr sw E   D v~ ##   ""  ''   ''!!5,>#>5 Lr+c`|j}d|j_d|j_yNT)rPrr; rcbuf_var)r0rPs rused_buffer_aux_varsrs)!!J&*J# $Jr+c v|j|jj}}|jj}ddg}|dk(r|j dg}t |jjD]#}|D]}|j d||||||fz%|jdj|y)Nrrrhrz1%s.diminfo[%d].%s = %s.rcbuffer->pybuffer.%s[%d];r) rPr6rerr5rrrIputlnr) buf_entryrrPrepybuffernd_structfldnameslnrfldnames r put_unpack_buffer_aux_into_scoper s!++Y^^-@-@J"22887#H v~ % B 9>>&& '5 5G III&7&44 5 55  JJsxx|r+c|j}|jj}|jj}|j d|z|j d|z|j d|z|j |d|dy)Nz%s.pybuffer.buf = NULL;z%s.refcount = 0;z%s.data = NULL;z .rcbuffer = &;)rPrr5rr)r0rbufauxrpybuffer_structs r put_init_varsr2s{   F..44&&,,OJJ(?:;JJ!O34 JJ #445JJ'8/JKr+cp|j}t||j||j}|j d|j d|jj j z|j |jd|z||j dt||y)N{'__Pyx_BufFmt_StackElem __pyx_stack[%d];%s == -1}) rPget_getbuffer_callr5r6rrGstruct_nesting_depth error_goto_ifr)r0rr7rP getbuffers rput_acquire_arg_bufferr@s!!J"4j%**MI JJsOJJ85::;K;K;`;`;bbcJJt!!*y"8#>?JJsO%UD1r+c|jjt|jd|jj j zy)N0__Pyx_SafeReleaseBuffer(&%s.rcbuffer->pybuffer);)ruse_utility_codeacquire_utility_coderrPrr5)rr0s rput_release_buffer_coderNs;%%&:;JJAEDTDTDdDdDjDjjkr+c |j}t|j}t||}|jj }t ||j}|jjtdtzS)Nz__Pyx_GetBufferAndValidate(&%(pybuffernd_struct)s.rcbuffer->pybuffer, (PyObject*)%(obj_cname)s, &%(dtype_typeinfo)s, %(flags)s, %(ndim)d, %(cast)d, __pyx_stack)) rIrrgrrr5get_type_information_cnamerGrrrlocals) r obj_cnamerPrrIrgrrdtype_typeinfos rrrSsv   D { D j+ .E"2288/k6G6GHN%%&:; %'-x 01r+c f|j|j}}|jj}t ||} j dj d|j jztd||} |rj d|zjjtjd} j | d| |zdj d jd | zztfd td D} j d | zj d jd| |zzzj d| zj j#t$j dj dj d| zj dj d| z| D]} jj'| j dt)|j j+| |jj'| nj d jd| |zzzj |dtj,|tj.dd|dj j1|j3dt)|j dj dy)aJ Generate code for reassigning a buffer variables. This only deals with getting the buffer auxiliary structure and variables set up correctly, the assignment itself and refcounting is the responsibility of the caller. However, the assignment operation may throw an exception so that the reassignment never happens. Depending on the circumstances there are two possible outcomes: - Old buffer released, new acquired, rhs assigned to lhs - Old buffer released, new acquired which fails, reaqcuire old lhs buffer (which may or may not succeed). rrz%srF manage_refz = r if (%s) {%s < 0c3tK|]/}jjtjd1yw)FrN) funcstate allocate_tempr py_object_type).0_rs r z'put_assign_to_buffer..s4-..66z7P7P]b6c-s58zPyErr_Fetch(&%s, &%s, &%s);rz/Py_XDECREF(%s); Py_XDECREF(%s); Py_XDECREF(%s);z!__Pyx_RaiseBufferFallbackError();z} else {zPyErr_Restore(%s, %s, %s);rz%s = %s = %s = 0;Py_Nonez; __Pyx_INCREF(Py_None); z.rcbuffer->pybuffer.buf = NULL;N)rPr6rr5rrrGrrrrr c_int_typeunlikelytuplerrrraise_buffer_fallback_code release_temprerror_goto_if_negtypecastr error_gotoput) lhs_cname rhs_cnameris_initializedr7rrPrrrr retcode_cname exc_tempsts ` rput_assign_to_bufferr as (22INN J"2288 j+ .EJJsOJJ8;;L;L;a;a;ccd"4z;GI EHYYZ44Z5J5JW\4]   I0EFG ;$--=0H"IJK -#(8--  09<= ;zY=R/S!TTU DyPQ ))*DE 67 : /);< 3 &23 +A NN ' ' * + 3(D9 4))-=> ##M2 ;zY=R/S!TTU '' Z5N5NPYZ%' ( 4??3'( (D9 3JJsOr+c |dxr|}|dr|jjtjd}|j d|zt t |||jD]\} \} } } | dk7r~|j d| z|rA|j | d| d |j d |jd | z|| fzn|j d || fz|jd | dk7rd} nd} |j d |j| d| | || fz|r"|jjtd}n!|jjtd}|j d|jd|zz|j |d|d|j |j||j d|jj|nG|rEt |||jD]'\} } } | dk7s |j d| d| d| d )|j!||S)ae Generates code to process indices and calculate an offset into a buffer. Returns a C string which gives a pointer which can be read from or written to at will (it is an expression so caller should store it in a temporary if it is used more than once). As the bounds checking can have any number of combinations of unsigned arguments, smart optimizations etc. we insert it directly in the function body. The lookup however is delegated to a inline function that is instantiated once per ndim (lookup with suboffsets tend to get quite complicated). entry is a BufferEntry wraparound boundscheckFrz%s = -1;rz if (%s < 0) {z += rzif (%s) %s = %d;rz%s = %d;z} else z(size_t)z >= __Pyx_RaiseBufferIndexErrorNogil__Pyx_RaiseBufferIndexErrorrz%s != -1r);rzif (z < 0) )rrr rr enumerater}rrrrrraise_indexerror_nogilraise_indexerror_coderrr)r0 index_signedsr directivesr7rrfin_nogil_contextfailed_dim_tempdimsignedr5rrgfuncs rput_buffer_lookup_coders;",/D4D- ..66z7L7LY^6_ :/0+4S V[VmVmVo5p+q ' 'C'&%{ ?U23#JJeU;<JJ1 h&67'5. ./JJz_c,BBC#{! JJ) eT5AB-&& '# '*     - -.D E5D    - -.C D0D ;zO/K!LLM o67 4??3'( 3 ##O4 $' |UE\E\E^$_ L FE5{ eUEJK L  , ,T< @@r+c.|jtyr.)rbuffer_struct_declare_coders rr$r$s34r+c djt|Dcgc] }d|||fz c}}|jd|d|d|d|d djt|Dcgc] }d|||fz c}}|jd |d |d |jtd ||fzd jt|Dcgc]}td||||fzc}zdzycc}wcc}wcc}w) Generates a buffer lookup function for the right number of dimensions. The function gives back a void* at the right location. rz i%d, s%d, o%d#define (type, buf, z ) (type)(z _imp(buf, z))z.Py_ssize_t i%d, Py_ssize_t s%d, Py_ssize_t o%dzstatic CYTHON_INLINE void* z_imp(void* buf, rzf static CYTHON_INLINE void* %s_imp(void* buf, %s) { char* ptr = (char*)buf; rzX ptr += s%d * i%d; if (o%d >= 0) ptr = *((char**)ptr) + o%d; z return ptr; }N)rrrr)protodefinr/rr macroargsfuncargss rrrs  b J1?aAY6JKI KKtYX\^ghiyy`efh`ij[\JaQRTUYVjkH KKtXVW KK  h #%''&+2Y+0!",23 , 1a|,+0#    Kk +0sC/%C4C9c djt|Dcgc] }d||fz c}}djt|Dcgc] }d||fz c}}|jd|d|d|dy cc}wcc}w) r"ri%d, s%d + i%d * s%dr#r$z) (type)((char*)buf + rN)rrrr%r&r/rrargsoffsets rrrsl 9959=ajAq6)= >D ZZuRyA!1v-A BF KKtTSYZ[>As A7A<c<|dk(r|jd|zydjt|Dcgc] }d||fz c}}djt|dz Dcgc] }d||fz c}}|jd||||dz fzycc}wcc}w) z Similar to strided lookup, but can assume that the last dimension doesn't need a multiplication as long as. Still we keep the same signature for now. r.#define %s(type, buf, i0, s0) ((type)buf + i0)rr*r+r,9#define %s(type, buf, %s) ((type)((char*)buf + %s) + i%d)Nrrrr-s rrrs  Qw DtKLyyuRyA!*1v-AB5a=Ia[Aq61IJ OSWY]_egilmgmRnnoBIs B#Bc2|dk(r|jd|zy djt|Dcgc] }d||fz c}}djtd|Dcgc] }d||fz c}}|jd|||dfzy cc}wcc}w) zB Like C lookup, but the first index is optimized instead. rr1rr*r+r,r2rNr3r-s rrrs Qw DtKLyyuRyA!*1v-AB5B<Ha[Aq61HI OSWY]_eghRiijBHs B!Bc6|jtyr.)rGetAndReleaseBufferUtilityCoder s rr&r&)s79:r+c2eZdZdZdZdZdZdZdZdZ y)r6NFcyr.rrs rrz'GetAndReleaseBufferUtilityCode.__init__5s r+c"t|tSr.)rr6)r'others r__eq__z%GetAndReleaseBufferUtilityCode.__eq__8s%!?@@r+cy)NiFosrrs r__hash__z'GetAndReleaseBufferUtilityCode.__hash__;sr+c yr.r)r'kwargss rget_treez'GetAndReleaseBufferUtilityCode.get_tree>sr+c |d}|d}|jj}|jjg t  fd  |t j ddt }|j|j}|j|j|j|d}|j||j|y) Nrrc| vry j||jD] }| |jD]}t|jt r|j }|js7|ur |jsHdx}}|jjD]:}|jdk(r |j}|jdk(s/|j}<|s j|j||fy)N __getbuffer____releasebuffer__)rcimported_modules type_entriesrrErr6rr;r%pyfunc_entriesr/ func_cnamer typeptr_cname) r%mer releaserqr cython_scopefind_buffer_typestypesvisited_scopess rrNzBGetAndReleaseBufferUtilityCode.put_code..find_buffer_typesIs&   u %,, %!!$ %'' Fa779JKFF&& ,QVV $((GcWW33T66%55Q\\sVV';;q||WT aoosG%DE Fr+GetAndReleaseBufferBuffer.c)rO) from_filecontextr) module_noder%rTrMsetr loaddict format_coder%inject_string_constantsimplr) r'outputr proto_coder util_coder%r[rMrNrOrPs @@@@rput_codez'GetAndReleaseBufferUtilityCode.put_code@s()01   &&{{//  F( #&++ !Zu%' %%ioo6$$  - -innf Ea HJ  4r+) rarbrcrequiresis_cython_utilityrr;r=r@r_rr+rr6r6-s) H A'(r+r6c|jry|jry|js |jrd}nd}||j zS)Nobjectptrnn_r) is_pyobjectrH is_typedefis_struct_or_unionspecialization_name)rGr<s rmangle_dtype_namerjksI     u77FF11333r+c ,t|}d|z}d|z}|jry||j}|dkrJ||jjvr|jjj ||jd}g}|j r@|j r4|j|j|j}|j r4|jxr|j}|j} |jrd}n|jr|j} |j r | j"} | j$} t'| dkDsJ| D cgc]} t)|| j*|dz } } |j-d|zd t/| | D]G\} }|j-d |d | j0d |jd| j2d d I|j-dd |j-dd nJt5|}d}d}|t6j8urd| z}d}nj|j:r d| z}d|z}nS|s |j<rd}nB|j>rd}n3|jrd}|j@rd}n|jBrd}nJ|d}|||| djE|Dcgc] }t5|c}xsdt'||||f }|j-||zd |Scc} wcc}w)ar Output the run-time type information (__Pyx_TypeInfo) for given dtype, and return the name of the type info struct. Structs with two floats of the same size are encoded as complex numbers. One can separate between complex numbers declared as struct or with native encoding by inspecting to see if the fields field of the type is filled in. z__Pyx_TypeInfo_%sz__Pyx_StructFields_%szrtypeinfoNULLrz!static __Pyx_StructField %s[] = {T)safez {&z, "z ", offsetof(rz)},z {NULL, NULL, 0}z};0zIS_UNSIGNED(%s)z'H'z%s ? 'U' : 'I'z'C'z'R'z'S'__PYX_BUF_FLAGS_PACKED_STRUCTz'O'zLstatic __Pyx_TypeInfo %s = { "%s", %s, sizeof(%s), { %s }, %s, %s, %s, %s };)#rjis_errorrrrris_arrayrsize base_typerhcan_be_complexris_simple_buffer_dtype is_structr%is_constconst_base_type_scope var_entriesrBrr6rr}r/r5strr c_char_typeis_int is_complexis_floatpackedrfr)rrGmaxdepth namesuffixr/structinfo_nametypecode arraysizescomplex_possibledeclcode struct_scopefieldsfrOrlrepr is_unsigned typegrouprtups rrrys#5)J  +D- :O ~~iE$>$>$@1} 4##111 &&**40##J/ >>..!!%**-..!33N8L8L8N//1  ' ' )$O __ ;;L~~+AA !--Fv;? "$&0affhlK&E& NN>PW[N \"651 c 8$affe.J.J.Lagg W]ac c NN.TN : NN4dN + %j J** *+h6KI \\+h6K(;6I !1!1I ^^I __I||7   I % LS/8yy*5Q#a&56=#s:+u. x#~D1 KS&J6s )!L  Lc f|tj|dfi|Stj|dfd|i|S)NrRrT)r rWr )util_code_namerTr?s rload_buffer_utilityrs= EfEE!&&~z]7]V\]]r+)max_dimsBufferStructDeclare)rTBufferFormatStructsBufferIndexErrorBufferIndexErrorNogilBufferFallbackErrorBufferGetAndValidateBufferFormatCheckTypeInfoToFormat)rrr.)? __future__rVisitorrrErrorsrr rCoder rr r r rrrrr{rwryr|rzr~rrrrrsrrcrrrrrrrrr rr$rrrrr&r6rjrrrXrJrTrbuffer_formats_declare_coderrrrbuffer_format_check_code_typeinfo_to_format_coderr+rrs&$" *1V?VvGf$PUV"#6,3  l 4 B ' >JD&DN % $ L 2l 1EP?AD5 *\ p k;;V;| 4Un^ // 001FPWX12GH,,>?,-DE01FG*+A7S./BGT//ABr+