o fxr@sddlmZddlmZddlmZddlmZddlmZddl mZm Z ddl m Z dd l m Z dd l mZdd l mZdd l mZdSd dZGdddeZdZdddddZdZdZdZdZdZdZdZdZdZdTdd ZGd!d"d"eZ d#d$Z!d%d&Z"d'd(Z#d)d*Z$d+d,Z%d-d.Z&d/d0Z'd1d2Z(d3d4Z)d5d6Z*d7d8Z+d9d:Z,d;d<Z-d=d>Z.d?d@Z/GdAdBdBeZ0dCdDZ1dUdEdFZ2dUdGdHZ3e4e j5dIZ6e3dJe6dKZ7e3dLZ8e3dMZ9e3dNZ:e3dOZ;e3dPe6dKZdS)V)absolute_import)CythonTransform) ModuleNode) CompileError)CythonUtilityCode) UtilityCodeTempitaUtilityCode)Options) Interpreter) PyrexTypes)Naming)SymtabcsFddlm}||}|dkr!d|dfdd|dD}|S)Nrdedent  csg|]}|qSr.0xindentrR/var/www/html/mig_web/myenv/lib/python3.10/site-packages/Cython/Compiler/Buffer.py zdedent..)textwraprjoinsplit)textreindentrrrrrs rcs<eZdZdZdZfddZddZddZdd ZZ S) IntroduceBufferAuxiliaryVarsFcsBt|tsJd|_tt||}|jrt|jt |j|S)Nr) isinstancermax_ndimsuperr!__call__buffers_existsuse_bufstruct_declare_codescopeuse_py2_buffer_functions)selfnoderesult __class__rrr% s  z%IntroduceBufferAuxiliaryVars.__call__c sfj}dd|D}t|dkr|jdddd|_dd|D}t|dkr-d|_|D]\d krBtjtrBd|_nq/~tt rUt|dkrUt j d |D]Sj j jrdt j d jj }|jtjkryt j d tj|j|jkr|j|_fd dtjtjftjtjff}fdd|D\}} t|| _qW|_|_dS)NcSg|] \}}|jjr|qSr)type is_bufferrnameentryrrrr3z=IntroduceBufferAuxiliaryVars.handle_scope..rcSs|jSN)r3)r4rrr5sz;IntroduceBufferAuxiliaryVars.handle_scope..)keyTcSr/r)r0is_memoryviewslicer2rrrr8r5 memoryviewz'Buffer vars not allowed in module scopez-Buffers with pointer types not yet supported.z1Buffer ndims exceeds Options.buffer_max_dims = %dcs0|}jd||jd}jrd|_|S)N)r3cnamer0posT)mangle declare_varr<is_argused)r0prefixr;aux_var)r4r3r+r(rrdecvarSs z9IntroduceBufferAuxiliaryVars.handle_scope..decvarcsg|] \}}||qSrr)rr0rA)rCrrr^)entriesitemslensortr&r"utility_code_definitionrusing_memoryviewrrr<r0dtypeis_ptrr3ndimr buffer_max_dimsr#r c_pyx_buffer_nd_typer pybuffernd_prefixc_pyx_buffer_typepybufferstruct_prefixr BufferAux buffer_auxbuffer_entriesr() r*r+r( scope_itemsbufvarsmemviewslicevarsbuftypeauxvars pybufferndrcbufferr)rCr4r3r+r(r handle_scope.sF           z)IntroduceBufferAuxiliaryVars.handle_scopecC|||j|||Sr6)r]r( visitchildrenr*r+rrrvisit_ModuleNodee z-IntroduceBufferAuxiliaryVars.visit_ModuleNodecCr^r6)r] local_scoper_r`rrrvisit_FuncDefNodejrbz.IntroduceBufferAuxiliaryVars.visit_FuncDefNode) __name__ __module__ __qualname__r&rJr%r]rard __classcell__rrr-rr!s 7r!)rKrMmodenegative_indicescastfullTF)rMrirjrkz"%s" is not a buffer optionzToo many buffer optionsz#"%s" buffer option already suppliedz "%s" missingz[Only allowed buffer modes are: "c", "fortran", "full", "strided" (as a compile-time string)z#ndim must be a non-negative integerz0dtype must be "object", numeric type or a structz"%s" must be a booleanNc s|durt}tj|||dd\}}t|tkr t|ddti|D]\}\}}|tvr7t|t |||<q&t t|D] \}\}}|tvrRt|t ||vr]t|t |||<qAtD]!}|vrz|||<Wqdt y|rtt |Yqdwqdd} | r| jrttd} | rt| tr| dkrttd } | r| d vrttfd d } | d | dS)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)rrK)type_env type_argsrrKrMrri)rlstridedcfortrancs&|}t|tstt|dSr6)getr"boolr ERR_BUF_BOOL)r3r globalposoptionsrr assert_bools  z+analyse_buffer_options..assert_boolrjrk)buffer_defaultsr interpret_compiletime_optionsrGbuffer_positional_options_countrERR_BUF_TOO_MANYrFbuffer_optionsERR_BUF_OPTION_UNKNOWNzip ERR_BUF_DUPKeyErrorERR_BUF_MISSINGrsis_extension_type ERR_BUF_DTYPEr"int ERR_BUF_NDIM ERR_BUF_MODE) rwenvposargsdictargsdefaults need_completer3valuer<rKrMriryrrvranalyse_buffer_optionssR             rc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) BufferEntrycCs<||_|j|_|jjj|_d|j|_|jj|_|dS)Nz%s.rcbuffer->pybuffer.buf) r4r0rTbuflocal_nd_varr;buf_ptrbuffer_ptr_type buf_ptr_typeinit_attributes)r*r4rrr__init__s     zBufferEntry.__init__cCs"||_||_||_dSr6)get_buf_shapevarsshapeget_buf_stridevarsstridesget_buf_suboffsetvars suboffsetsr*rrrrs  zBufferEntry.init_attributescC |dS)Nz%s.diminfo[%d].suboffsets _for_all_ndimrrrrr z!BufferEntry.get_buf_suboffsetvarscCr)Nz%s.diminfo[%d].stridesrrrrrrrzBufferEntry.get_buf_stridevarscCr)Nz%s.diminfo[%d].shaperrrrrrrzBufferEntry.get_buf_shapevarscsfddtjjDS)Ncsg|] }j|fqSr)r;risr*rrrrDz-BufferEntry._for_all_ndim..)ranger0rM)r*rrrrrszBufferEntry._for_all_ndimcCs>g}|jj}|jj}|dkr4t|||D]\}}}||||||qd|} t} n9|dkr?d|} t} n|dkrJd|} t } n |dkrUd|} t } nJt||D]\}}||||q^| |j j vr|j j | |j d } |j d } | | | | |d |j} d | | |jd|f}|S)Nrlz__Pyx_BufPtrFull%ddrpz__Pyx_BufPtrStrided%ddrqz__Pyx_BufPtrCContig%ddrrz__Pyx_BufPtrFortranContig%ddFutility_code_protoutility_code_def)r3ndz%s(%s, %s, %s), )r0rMrirrrappendbuf_lookup_full_codebuf_lookup_strided_codebuf_lookup_c_codebuf_lookup_fortran_code globalstate utility_codesaddrempty_declaration_coderr)r*code index_cnamesparamsrrirrofuncnamefuncgen protocodedefcodebuf_ptr_type_codeptrcoderrrgenerate_buffer_lookup_codesH          z'BufferEntry.generate_buffer_lookup_codeN) rerfrgrrrrrrrrrrrrs rcCshd}|j}|dkr|d7}n|dkr|d7}n|dkr |d7}n |dkr)|d 7}nJ|jr2|d 7}|S) N PyBUF_FORMATrlz| PyBUF_INDIRECTrpz| PyBUF_STRIDESrqz| PyBUF_C_CONTIGUOUSrrz| PyBUF_F_CONTIGUOUSFz| PyBUF_WRITABLE)riwritable_needed)rT buffer_typeflagsrirrr get_flags s    rcCs|j}d|j_d|j_dSNT)rTrr@ rcbuf_var)r4rTrrrused_buffer_aux_varss rc Cs|j|jj}}|jj}ddg}|dkr|dg}t|jjD]}|D]}|d||||||fq%q!|d |dS)Nrrrlrz1%s.diminfo[%d].%s = %s.rcbuffer->pybuffer.%s[%d];r) rTr0rirr;rrrMputlnr) buf_entryrrTripybuffernd_structfldnameslnrfldnamerrr put_unpack_buffer_aux_into_scope s  rcCsV|j}|jj}|jj}|d||d||d||d||fdS)Nz%s.pybuffer.buf = NULL;z%s.refcount = 0;z%s.data = NULL;z%s.rcbuffer = &%s;)rTrr;rr)r4rbufauxrpybuffer_structrrr put_init_vars3srcCsf|j}t||j||j}|d|d|jj||d|||dt||dS)N{'__Pyx_BufFmt_StackElem __pyx_stack[%d];%s == -1}) rTget_getbuffer_callr;r0rrKstruct_nesting_depth error_goto_ifr)r4rr<rT getbufferrrrput_acquire_arg_bufferAs  rcCs$|jt|d|jjjdS)N0__Pyx_SafeReleaseBuffer(&%s.rcbuffer->pybuffer);)ruse_utility_codeacquire_utility_coderrTrr;)rr4rrrput_release_buffer_codeOs rc CsD|j}t|j}t||}|jj}t||j}|j t dt S)Nz__Pyx_GetBufferAndValidate(&%(pybuffernd_struct)s.rcbuffer->pybuffer, (PyObject*)%(obj_cname)s, &%(dtype_typeinfo)s, %(flags)s, %(ndim)d, %(cast)d, __pyx_stack)) rMrrkrrr;get_type_information_cnamerKrrrlocals) r obj_cnamerTrrMrkrrdtype_typeinforrrrTs    rc s|j|j}}|jj}t||} dd|jtd||} |rd|j j t j dd} d| | |fd d | tfd d td D} d | d d| |d| jtddd| dd| | D]} j | qdt|| |j | n5d d| |d|t |t jd|f|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%s = %s; if (%s) {%s < 0c3s"|] }jjtjddVqdS)FrN) funcstate allocate_tempr py_object_type)r_rrr sz'put_assign_to_buffer..zPyErr_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;zA%s = %s; __Pyx_INCREF(Py_None); %s.rcbuffer->pybuffer.buf = NULL;Py_NoneN)rTr0rr;rrrKrrrrr c_int_typeunlikelytuplerrrraise_buffer_fallback_code release_temprerror_goto_if_negtypecastr error_gotoput) lhs_cname rhs_cnameris_initializedr<rrTrrrr retcode_cname exc_tempstrrrput_assign_to_bufferbsR            rc Cs|do|}|dr|jjtjdd}|d|tt|||D]V\} \} } } | dkr_|d| |rQ|d| | f|d |d | || fn |d || f| d | dkrfd } nd} |d |d| | | f|| fq$|r|j t d}n|j t d}|d|d||d||f||||d|j|n|rt|||D]\} } } | dkr|d| | | fq|||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 %s += %s;zif (%s) %s = %d;rz%s = %d;z} else z(size_t)z %s >= %s%s __Pyx_RaiseBufferIndexErrorNogil__Pyx_RaiseBufferIndexErrorrz%s != -1z%s(%s);rzif (%s < 0) %s += %s;)rrr rr enumeraterrrrrrraise_indexerror_nogilraise_indexerror_coderrr)r4 index_signedsr directivesr<rrjin_nogil_contextfailed_dim_tempdimsignedr;rrkfuncrrrput_buffer_lookup_codesL "        r cCs|tdSr6)rbuffer_struct_declare_coderrrrr'sr'cCsdddt|D}|d||||fdddt|D}|d||f|td||fdd dt|Dd d S) Generates a buffer lookup function for the right number of dimensions. The function gives back a void* at the right location. rcSg|] }d|||fqS)z i%d, s%d, o%drrrrrrrDz(buf_lookup_full_code..z1#define %s(type, buf, %s) (type)(%s_imp(buf, %s))cSr)z.Py_ssize_t i%d, Py_ssize_t s%d, Py_ssize_t o%drrrrrrrDz1static CYTHON_INLINE void* %s_imp(void* buf, %s);zf static CYTHON_INLINE void* %s_imp(void* buf, %s) { char* ptr = (char*)buf; rcSs g|] }td||||fqS)zX ptr += s%d * i%d; if (o%d >= 0) ptr = *((char**)ptr) + o%d; rrrrrrs  z return ptr; }N)rrrr)protodefinr3r macroargsfuncargsrrrrs   rcCsHdddt|D}dddt|D}|d|||fdS)rrcSg|]}d||fqSzi%d, s%drrrrrr z+buf_lookup_strided_code.. + cSrz i%d * s%drrrrrr rz1#define %s(type, buf, %s) (type)((char*)buf + %s)N)rrrrrr3rargsoffsetrrrrsrcCsl|dkr |d|d Sdddt|D}dddt|dD}|d||||dfd S) 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)rcSrrrrrrrrrz%buf_lookup_c_code..rcSrrrrrrrrr9#define %s(type, buf, %s) ((type)((char*)buf + %s) + i%d)Nrrrrrrrrs rcCsf|dkr |d|d Sdddt|D}dddtd|D}|d|||d fd S) zB Like C lookup, but the first index is optimized instead. rrrcSrrrrrrrr%rz+buf_lookup_fortran_code..rcSrrrrrrrr&rrrNr rrrrrs rcCs|tdSr6)rGetAndReleaseBufferUtilityCoderrrrr)*sr)c@s<eZdZdZdZddZddZddZd d Zd d Z dS) r!NFcCdSr6rrrrrr6z'GetAndReleaseBufferUtilityCode.__init__cCs t|tSr6)r"r!)r*otherrrr__eq__9rz%GetAndReleaseBufferUtilityCode.__eq__cCsdS)NiFosrrrrr__hash__<r#z'GetAndReleaseBufferUtilityCode.__hash__cKr"r6r)r*kwargsrrrget_tree?sz'GetAndReleaseBufferUtilityCode.get_treecs|d}|d}|jj}|jjgtfdd|tjddtdd}||j }|| |j |d }| || |dS) Nrrcs|vrdS||jD]}|q|jD]>}t|jtr!q|j}|jrV|ur/|js/qd}}|j j D]}|j dkrB|j }q7|j dkrJ|j }q7|rV |j||fqdS)N __getbuffer____releasebuffer__)rcimported_modules type_entriesr"rIrr0rr@r(pyfunc_entriesr3 func_cnamer typeptr_cname)r(merreleasersr cython_scopefind_buffer_typestypesvisited_scopesrrr5Js*      zBGetAndReleaseBufferUtilityCode.put_code..find_buffer_typesGetAndReleaseBufferBuffer.c)r6) from_filecontextr) module_noder(r;r4setr loaddict format_coderinject_string_constantsimplr)r*outputr proto_coder util_coderrBrr3rput_codeAs$  z'GetAndReleaseBufferUtilityCode.put_code) rerfrgrequiresis_cython_utilityrr%r&r(rFrrrrr!.s r!cCs6|jrdS|jr dS|js|jrd}nd}||S)Nobjectptrnn_r) is_pyobjectrL is_typedefis_struct_or_unionspecialization_name)rKrArrrmangle_dtype_namels  rPc sBt|}d|}d|}|jrdSdur|dkrJ|jjvrjj|jd}g}|jrF|jrF||j|j }|js:|j oL| }| } | rXd}na|jr|j} |jrd| j} | j} t| dksoJfd d | D} |jd |d d |jr|j } n| } t| | D]\}}|jd||j| |jfd d q|jdd d |jdd d nJt|}d}d}|tjurd| }d}n/|jrd| }d|}n#|s|jrd}n|jrd}n|jrd}|jrd}n |j rd}nJ|d}|||| d!dd |Dpdt||||f }|j||d d |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_%szNrFtypeinfoNULLcsg|] }t|jdqS)r)rr0)rfrmaxdepthrrrsz.get_type_information_cname..z!static __Pyx_StructField %s[] = {T)safez {&%s, "%s", offsetof(%s, %s)},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'zLstatic __Pyx_TypeInfo %s = { "%s", %s, sizeof(%s), { %s }, %s, %s, %s, %s };rcSsg|]}t|qSr)strrrrrrr)"rPis_errorrrrris_arrayrsize base_typerNcan_be_complexris_simple_buffer_dtype is_structr(is_cv_qualifiedbase_type_scope var_entriesrGr cv_base_typerr3r;rYr c_char_typeis_int is_complexis_floatpackedrLr)rrKrU namesuffixr3structinfo_nametypecode arraysizescomplex_possibledeclcode struct_scopefieldsr6 struct_typerSrQrepr is_unsigned typegrouptuprrTrrzs         rcKs4|durtj|dfi|Stj|dfd|i|S)Nr9r;)rr>r )util_code_namer;r'rrrload_buffer_utilitysrx)max_dimsBufferStructDeclare)r;BufferFormatStructsBufferIndexErrorBufferIndexErrorNogilBufferFallbackErrorBufferGetAndValidateBufferFormatCheckTypeInfoToFormat)rrr6)? __future__rVisitorrrErrorsrrrCoder rr r r r rrr!r~rzr|rr}rrrrrrurrIrrrrrrrrrr r'rrrrr)r!rPrrxr?rNr;rbuffer_formats_declare_coderrrrbuffer_format_check_code_typeinfo_to_format_coderrrrsl          [ EGHB  >  _