-PhJiFddlmZddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl m Z dd l m Z dd l mZd;d ZGd deZdZdddddZdZdZdZdZdZdZdZdZdZd)CythonTransform) ModuleNode) CompileError)CythonUtilityCode) UtilityCodeTempitaUtilityCode)Options) Interpreter) PyrexTypes)Naming)Symtabcddlm}||}|dkr9d|zdfd|dD}|S)Nrdedent  cg|]}|zSr).0xindents V/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/Cython/Compiler/Buffer.py zdedent..s???&1*???)textwraprjoinsplit)textreindentrrs @rrr sk 6$<.>???@@ Krc8eZdZdZdZfdZdZdZdZxZ S)IntroduceBufferAuxiliaryVarsFct|tsJd|_t|}|jrt |j|S)Nr) isinstancermax_ndimsuper__call__buffers_existsuse_bufstruct_declare_codescope)selfnoderesult __class__s rr'z%IntroduceBufferAuxiliaryVars.__call__sW$ +++++ !!$''   3 &tz 2 2 2 rc j}d|D}t|dkr|dd|_d|D}t|dkrd|_|D].\ dkr#t jtr d|_n/~t tr(t|dkrtj d|D] j j jrtj d  j j }|jt"jkr"tj d t"jz|j|jkr |j|_ fd  t(jt,jft(jt,jff} fd |D\}} t5j||  _|_|_dS) Nc0g|]\}}|jj|Sr)type is_bufferrnameentrys rrz=IntroduceBufferAuxiliaryVars.handle_scope..0s&PPP[T55:;OP5PPPrrc|jSN)r4)r5s rz;IntroduceBufferAuxiliaryVars.handle_scope..2s5:r)keyTc0g|]\}}|jj|Sr)r1is_memoryviewslicer3s rrz=IntroduceBufferAuxiliaryVars.handle_scope..5s&bbbkdEEJDabEbbbr memoryviewz'Buffer vars not allowed in module scopez-Buffers with pointer types not yet supported.z1Buffer ndims exceeds Options.buffer_max_dims = %dc|}d||j}jrd|_|S)N)r4cnamer1posT)mangle declare_varr?is_argused)r1prefixr>aux_varr5r4r,r*s rdecvarz9IntroduceBufferAuxiliaryVars.handle_scope..decvarPsS VT22++U1548,EE<(#'GLrc.g|]\}}||Srr)rr1rDrFs rrz=IntroduceBufferAuxiliaryVars.handle_scope..[s)#W#W#W^dFFF4$8$8#W#W#Wr)entriesitemslensortr(r$utility_code_definitionrusing_memoryviewrrr?r1dtypeis_ptrr4ndimr buffer_max_dimsr%r c_pyx_buffer_nd_typer pybuffernd_prefixc_pyx_buffer_typepybufferstruct_prefixr BufferAux buffer_auxbuffer_entriesr*) r+r,r* scope_itemsbufvarsmemviewslicevarsbuftypeauxvars pybufferndrcbufferrFr5r4s `` @@@r handle_scopez)IntroduceBufferAuxiliaryVars.handle_scope+s0m))++ PPKPPP w< d} | r| jrt t  d} | r0t#| t$r| dkrt t& d } | r| d vrt t( fd } | d | d  S)aO Must be called during type analysis, as analyse is called on the dtype argument. posargs and dictargs should consist of a list and a dict of tuples (value, pos). Defaults should be a dict of values. Returns a dict containing all the options a buffer can have and its value (with the positions stripped). N)rrN)type_env type_argsrrNrPrrl)rostridedcfortranc|}t|tstt|zdSr7)getr$boolr ERR_BUF_BOOL)r4r globalposoptionss r assert_boolz+analyse_buffer_options..assert_boolsE KK  !T"" ?y,*=>> > ? ?rrmrn)buffer_defaultsr interpret_compiletime_optionsrJbuffer_positional_options_countrERR_BUF_TOO_MANYrIbuffer_optionsERR_BUF_OPTION_UNKNOWNzip ERR_BUF_DUPKeyErrorERR_BUF_MISSINGrxis_extension_type ERR_BUF_DTYPEr$int ERR_BUF_NDIM ERR_BUF_MODE)r{envposargsdictargsdefaults need_completer4valuer?rNrPrlr}r|s` @ranalyse_buffer_optionsr|s"#AC<AAAGX 7||55572;q>+;<<<G&nn..luc ~ % %s$:T$ABB B !.'::luc ~ % %s$:T$ABB B 7??sK$$677 7 JJ w   J (  J J J J&y/D2HIIIJJ J  KK E 5(59m444 ;;v  D 4Zc**4dQhh9l333 ;;v  D 4T@@@9l333?????? K"###K Ns D$EEc8eZdZdZdZdZdZdZdZdZ dS) BufferEntryc||_|j|_|jjj|_d|jz|_|jj|_|dS)Nz%s.rcbuffer->pybuffer.buf) r5r1rWbuflocal_nd_varr>buf_ptrbuffer_ptr_type buf_ptr_typeinit_attributes)r+r5s r__init__zBufferEntry.__init__sU J %5; 2TZ? !J6 rc||_||_||_dSr7)get_buf_shapevarsshapeget_buf_stridevarsstridesget_buf_suboffsetvars suboffsetsr+s rrzBufferEntry.init_attributess?++-- ..00 4466rc,|dS)Nz%s.diminfo[%d].suboffsets _for_all_ndimrs rrz!BufferEntry.get_buf_suboffsetvarss!!"=>>>rc,|dS)Nz%s.diminfo[%d].stridesrrs rrzBufferEntry.get_buf_stridevarss!!":;;;rc,|dS)Nz%s.diminfo[%d].shaperrs rrzBufferEntry.get_buf_shapevarss!!"8999rcRfdtjjDS)Nc(g|]}j|fzSr)r>)risr+s rrz-BufferEntry._for_all_ndim..s$CCCTZO#CCCr)ranger1rP)r+rs``rrzBufferEntry._for_all_ndims-CCCCCU49>-B-BCCCCrc g}|jj}|jj}|dkrt|||D]E\}}}||||||Fd|z} t} n|dkr d|z} t} n(|dkr d|z} t} n|dkr d|z} t} nJt||D]/\}}||||0| |j j vrH|j j | |j d } |j d } | | | | | |j} | d | d|jdd|d}|S)Nroz__Pyx_BufPtrFull%ddrtz__Pyx_BufPtrStrided%ddruz__Pyx_BufPtrCContig%ddrvz__Pyx_BufPtrFortranContig%ddFutility_code_protoutility_code_def)r4nd(, ))r1rPrlrrrappendbuf_lookup_full_codebuf_lookup_strided_codebuf_lookup_c_codebuf_lookup_fortran_code globalstate utility_codesaddrempty_declaration_coderr)r+code index_cnamesparamsrrlrrofuncnamefuncgen protocodedefcodebuf_ptr_type_codeptrcodes rgenerate_buffer_lookup_codez'BufferEntry.generate_buffer_lookup_codes Y^y~ 6>>|#6688#99;;== ! !1a a    a    a    ,r1H*GGy  3b813b8+""9B>1 L$*A*A*C*CDD ! !1 a    a     4+9 9 9   * . .x 8 8 8()=>I&'9:G GIwX" = = = = -DDFF&.hh0A0A0A4<<<&*ii&7&7&7&79rN) rhrirjrrrrrrrrrrrrs777 ???<<<:::DDD*****rrcd}|j}|dkr|dz }n&|dkr|dz }n|dkr|dz }n|dkr|d z }nJ|jr|d z }|S) N PyBUF_FORMATroz| PyBUF_INDIRECTrtz| PyBUF_STRIDESruz| PyBUF_C_CONTIGUOUSrvz| PyBUF_F_CONTIGUOUSFz| PyBUF_WRITABLE)rlwritable_needed)rW buffer_typeflagsrls r get_flagsrs E  D v~~ ##    ""  ''    '' !>5,>#>5 LrcD|j}d|j_d|j_dSNT)rWrrC rcbuf_var)r5rWs rused_buffer_aux_varsrs&!J&*J# $Jrc T|j|jj}}|jj}ddg}|dkr|dg}t |jjD]%}|D] }|d||||||fz!&|d |dS)Nrrrorz1%s.diminfo[%d].%s = %s.rcbuffer->pybuffer.%s[%d];r) rWr1rlrr>rrrPputlnr) buf_entryrrWrlpybuffernd_structfldnameslnrfldnames r put_unpack_buffer_aux_into_scopers!+Y^-@J"287#H v~~ %%% B 9>& ' '  G III!1g!7AM       JJsxx||rc |j}|jj}|jj}|d|z|d|z|d|z||d|ddS)Nz%s.pybuffer.buf = NULL;z%s.refcount = 0;z%s.data = NULL;z .rcbuffer = &;)rWrr>rr)r5rbufauxrpybuffer_structs r put_init_varsr0s  F.4&,OJJ(?:;;;JJ!O3444 JJ #44555JJ'8'8'8///JKKKKKrc|j}t||j||j}|d|d|jjz||d|z||dt||dS)N{'__Pyx_BufFmt_StackElem __pyx_stack[%d];%s == -1}) rWget_getbuffer_callr>r1rrNstruct_nesting_depth error_goto_ifr)r5rr?rW getbuffers rput_acquire_arg_bufferr>s!J"4j%*MMI JJsOOOJJ85:;K;`;`;b;bbcccJJt!!*y"8#>>???JJsOOO%UD11111rc|jt|d|jjjzdS)N0__Pyx_SafeReleaseBuffer(&%s.rcbuffer->pybuffer);)ruse_utility_codeacquire_utility_coderrWrr>)rr5s rput_release_buffer_coderLsA%%&:;;;JJAEDTDdDjjkkkkkrc|j}t|j}t||}|jj}t ||j}|j tdtzS)Nz__Pyx_GetBufferAndValidate(&%(pybuffernd_struct)s.rcbuffer->pybuffer, (PyObject*)%(obj_cname)s, &%(dtype_typeinfo)s, %(flags)s, %(ndim)d, %(cast)d, __pyx_stack)) rPrrnrrr>get_type_information_cnamerNrrrlocals) r obj_cnamerWrrPrnrrdtype_typeinfos rrrQsu  D { D j+ . .E"28/k6GHHN%%&:;;; %'-xx 01rc (|j|j}}|jj}t ||} dd|jztd||} |r)d|zj tj d} | d| |zdd  d | zztfd td D} d | zd  d| |zzzd| zjt$ddd| zdd| z| D]} j | dt)|| |j | nωd  d| |zzz|dtj|tjdd|d|dt)|dddS)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 < 0c3dK|]*}jtjdV+dS)FrN) funcstate allocate_tempr py_object_type)r_rs r z'put_assign_to_buffer..sP--.66z7P]b6cc------rzPyErr_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)rWr1rr>rrrNrrrrr c_int_typeunlikelytuplerrrraise_buffer_fallback_code release_temprerror_goto_if_negtypecastr error_gotoput) lhs_cname rhs_cnameris_initializedr?rrWrrrr retcode_cname exc_tempsts ` rput_assign_to_bufferr_sv (2IN J"28 j+ . .EJJsOOOJJ8;;L;a;a;c;ccddd"4z;GGI* EHYYZZZ44Z5JW\4]]   I0E0E0EFGGG ;$--=0H"I"IJKKK ----#(88-----  09<=== ;zY=R/S!T!TTUUU DyPQQQ ))*DEEE 6777 : /);<<< 3 &2333 + +A N ' ' * * * * 3(D999 4))-==>>> ##M2222 ;zY=R/S!T!TTUUU II' Z5NPYZZZZ%%%' ( ( ( 4??3''((( (D999 3JJsOOOOOrc j|do|}|dr:|jtjd}|d|zt t |||D]\} \} } } | dkr|d| z|rM|| d| d |d |d | z|| fzn|d || fz| d | dkrd} nd} |d || d| | || fz|r"|j td}n!|j td}|d|d|zz||d|d||||d|j|nQ|rOt |||D]+\} } } | dkr|d| d| d| d ,|||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 enumeraterrrr rrraise_indexerror_nogilraise_indexerror_coder rr)r5 index_signedsr directivesr?rrmin_nogil_contextfailed_dim_tempdimsignedr>rrnfuncs rput_buffer_lookup_coder#s",/D4D- ,L .66z7LY^6__ :/000+4S V[VmVmVoVo5p5p+q+q ' ' 'C'&%{{ ?U2333#DJJeeeUUU;<<<JJ1 h&677'5. .////JJz_c,BBCCC###{{! JJ) eeeTT55ABB-&& ' ' ' '  1   - -.D E E E5DD   - -.C D D D0D ;zO/K!L!LLMMM ooo6777 4??3''((( 3 ##O4444 L$' |UE\E\E^E^$_$_ L L FE5{{ eeeUUUEEEJKKK  , ,T< @ @@rc:|tdSr7)rbuffer_struct_declare_code)rs rr)r)s344444rc ddt|D}|d|d|d|d|d ddt|D}|d |d |d |td ||fzd dt|DzdzdS) Generates a buffer lookup function for the right number of dimensions. The function gives back a void* at the right location. rcg|] }d|||fz S)z i%d, s%d, o%drrrs rrz(buf_lookup_full_code..s#JJJ1?aAY6JJJr#define (type, buf, z ) (type)(z _imp(buf, z))cg|] }d|||fz S)z.Py_ssize_t i%d, Py_ssize_t s%d, Py_ssize_t o%drr)s rrz(buf_lookup_full_code..s'jjj[\JaQRTUYVjjjrzstatic CYTHON_INLINE void* z_imp(void* buf, rzf static CYTHON_INLINE void* %s_imp(void* buf, %s) { char* ptr = (char*)buf; rc:g|]}td||||fzS)zX ptr += s%d * i%d; if (o%d >= 0) ptr = *((char**)ptr) + o%d; rr)s rrz(buf_lookup_full_code..sK+0+0+0!",23 , , 1a|,+0+0+0rz return ptr; }N)rrrr)protodefinr4r macroargsfuncargss rrrs.  JJb JJJKKI KKKtttYYYX\X\X\^g^g^ghiiiyyjj`efh`i`ijjjkkH KKKtttXXXVWWW KK   h #%''+0+0&+2YY+0+0+0# #        rc ddt|D}ddt|D}|d|d|d|dd S) r'rcg|] }d||fz Szi%d, s%drr)s rrz+buf_lookup_strided_code..s!===ajAq6)===r + cg|] }d||fz Sz i%d * s%drr)s rrz+buf_lookup_strided_code.. s!AAA!1v-AAArr*r+z) (type)((char*)buf + rN)rrrr.r/r4rargsoffsets rrrs 99==599=== > >D ZZAAuRyyAAA B BF KKKtttTTTSYSYSYZ[[[[[rc:|dkr|d|zdSddt|D}ddt|dz D}|d||||dz fzdS) 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)rcg|] }d||fz Sr4rr)s rrz%buf_lookup_c_code..!AAA!*1v-AAArr5cg|] }d||fz Sr7rr)s rrz%buf_lookup_c_code..s!IIIa[Aq61IIIr9#define %s(type, buf, %s) ((type)((char*)buf + %s) + i%d)Nrrrr8s rrr s  Qww DtKLLLLLyyAAuRyyAAABBII5a==IIIJJ OSWY]_egilmgmRnnooooorc0|dkr|d|zd Sddt|D}ddtd|D}|d|||dfzd S) zB Like C lookup, but the first index is optimized instead. rr<rcg|] }d||fz Sr4rr)s rrz+buf_lookup_fortran_code.."r>rr5cg|] }d||fz Sr7rr)s rrz+buf_lookup_fortran_code..#s!HHHa[Aq61HHHrr@rNrAr8s rrrs Qww DtKLLLLLyyAAuRyyAAABBHH5B<<HHHII OSWY]_eghRiijjjjjrcz|jrdS|jrdS|js|jrd}nd}||zS)Nobjectptrnn_r) is_pyobjectrO is_typedefis_struct_or_unionspecialization_name)rNrDs rmangle_dtype_namerM's[  4x 4u   u7 FFF113333rc t|}d|z}d|z}|jrdS|dkrJ|jjvrjj|jd}g}|jr/|jr(||j|j }|j(|j o| }| } | rd}n |jr|j} |jr| j} | j} t'| dksJfd | D} |d |zd |jr|j } n| } t-| | D]3\}}|d |d|jd| d|jd d 4|dd |dd nJt3|}d}d}|t4jurd| z}d}nO|jr d| z}d|z}n=|s|jrd}n1|jrd}n'|jr d}|jrd}n|j rd}n J|d}|||| d!d|Dpdt'||||f }|||zd |S)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_%szNrFtypeinfoNULLcBg|]}t|jdz S)r)rr1)rfrmaxdepths rrz.get_type_information_cname..as;&&&0afhlKK&&&rz'static const __Pyx_StructField %s[] = {T)safez {&z, "z ", offsetof(rz)},z {NULL, NULL, 0}z};0z__PYX_IS_UNSIGNED(%s)z'H'z%s ? 'U' : 'I'z'C'z'R'z'S'__PYX_BUF_FLAGS_PACKED_STRUCTz'O'zRstatic const __Pyx_TypeInfo %s = { "%s", %s, sizeof(%s), { %s }, %s, %s, %s, %s };c,g|]}t|Sr)str)rrs rrz.get_type_information_cname..s555Q#a&&555r)"rMis_errorrrrris_arrayrsize base_typerKcan_be_complexris_simple_buffer_dtype is_structr*is_cv_qualifiedbase_type_scope var_entriesrJr cv_base_typerr4r>rXr c_char_typeis_int is_complexis_floatpackedrIr)rrNrS namesuffixr4structinfo_nametypecode arraysizescomplex_possibledeclcode struct_scopefieldstypes struct_typerRrOrepr is_unsigned typegrouptups` ` rrr5s#5))J  +D- :O ~'iiE$>$>$@$@1}}  4#111 &**4000#J/ > (. (!!%*---. (!3N8L8L8N8N//11  ' ' ) ) $OO _  ;L$ <+; !-Fv;;????&&&&&$&&&E NNDV]aN b b b$ =#0GGII #::<< "6511 T T 8 (!&&&+++qwww HNRTTTT NN.TN : : : NN4dN + + + + L%jj J* * *1Hrs$$$$$$"""""" ******11111111UUUUU?UUUtGf$PUVV"#6,3  l 4 B ' >>>>JDDDDDDDDN %%% & L L L 2 2 2lll 1 1 1EEEP?A?A?AD555   *\\\ p p p k k k 4 4 4]]]]~^^^^ $/ 0 0 0001FPWXXX112GHH,+,>??,,-DEE001FGG**+A7SSS../BGTTT/./ABBr