V]c@@sddlmZddlmZddlmZddlmZddlmZddl mZm Z ddl m Z dd l m Z dd l mZdd l mZdd l mZdd ZdefdYZd@Zidd6dd6ed6ed6ZdZdZdZdZdZdZdZdZdZd?edZ!de"fd YZ#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-Z0d.Z1d/Z2d0e"fd1YZ3d2Z4d?d3Z5d?d4Z6e7d5e j8Z9e6d6d7e9Z:e6d8Z;e6d9Z<e6d:Z=e6d;Z>e6d<d7e9Z?e6d=d7e9Z@e6d>ZAd?S(Ai(tabsolute_importi(tCythonTransform(t ModuleNode(t CompileError(tCythonUtilityCode(t UtilityCodetTempitaUtilityCode(tOptions(t Interpreter(t PyrexTypes(tNaming(tSymtabcC@shddlm}||}|dkrdd|}djg|jdD]}||^qH}n|S(Ni(tdedentt s (ttextwrapR tjointsplit(ttexttreindentR tindenttx((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyR s    2tIntroduceBufferAuxiliaryVarscB@s8eZeZeZdZdZdZdZRS(cC@s`t|tstd|_tt|j|}|jr\t|j t |j n|S(Ni( t isinstanceRtAssertionErrortmax_ndimtsuperRt__call__tbuffers_existstuse_bufstruct_declare_codetscopetuse_py2_buffer_functions(tselftnodetresult((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyR s   c @sZjj}g|D]\jjr^q}t|dkrk|jddt|_ng|D]\jjrr^qr}t|dkrt|_nx?|D]7\dkrt j t rt|_ PqqW~t t r/t|dkr/tjdnx|D] jjjr`tjdnjj}|jtjkrtjdtjn|j|jkr|j|_nfd}tjtjftjtjff}g|D]\} } || | ^q\} } tj| | _q6W|_ |_!dS( NitkeycS@s|jS(N(tname(tentry((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt5tt memoryviews'Buffer vars not allowed in module scopes-Buffers with pointer types not yet supported.s1Buffer ndims exceeds Options.buffer_max_dims = %dc @sRj|}jddd|d|dj}jrNt|_n|S(NR#tcnamettypetpos(tmanglet declare_vartNoneR*tis_argtTruetused(R)tprefixR(taux_var(R$R#R R(s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytdecvarSs   ("tentriestitemsR)t is_buffertlentsortR/Rtis_memoryviewsliceRtutility_code_definitionRtusing_memoryviewRRR*tdtypetis_ptrR#tndimRtbuffer_max_dimsRR tc_pyx_buffer_nd_typeR tpybuffernd_prefixtc_pyx_buffer_typetpybufferstruct_prefixR t BufferAuxt buffer_auxtbuffer_entriesR( RR Rt scope_itemstbufvarstmemviewslicevarstbuftypeR3tauxvarsR)R1t pybufferndtrcbuffer((R$R#R Rs5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt handle_scope.s@+ +  !     . cC@s$|j||j|j||S(N(RNRt visitchildren(RR ((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytvisit_ModuleNodees cC@s$|j||j|j||S(N(RNt local_scopeRO(RR ((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytvisit_FuncDefNodejs ( t__name__t __module__tFalseRR;RRNRPRR(((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRs   7 R<R>tmodetnegative_indicestcasttfulls"%s" is not a buffer optionsToo many buffer optionss#"%s" buffer option already supplieds "%s" missings[Only allowed buffer modes are: "c", "fortran", "full", "strided" (as a compile-time string)s#ndim must be a non-negative integers0dtype must be "object", numeric type or a structs"%s" must be a booleanc @s`|dkrt}ntj||d|dd\}}t|tkret|ddtnixL|jD]>\}\}}|t krt|t |n||RVRYtstridedtctfortranc@s8j|}t|ts4tt|ndS(N(tgetRtboolRt ERR_BUF_BOOL(R#R(t globalpostoptions(s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt assert_boolsRWRXN(iR<(RYR\R]R^(R-tbuffer_defaultsRtinterpret_compiletime_optionsR7tbuffer_positional_options_countRtERR_BUF_TOO_MANYR5tbuffer_optionstERR_BUF_OPTION_UNKNOWNtzipt ERR_BUF_DUPtKeyErrortERR_BUF_MISSINGR_tis_extension_typet ERR_BUF_DTYPERtintt ERR_BUF_NDIMt ERR_BUF_MODE( Rbtenvtposargstdictargstdefaultst need_completeR#tvalueR*R<R>RVRd((RbRcs5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytanalyse_buffer_optionssJ   "     !"  t BufferEntrycB@sGeZdZdZdZdZdZdZdZRS(cC@sT||_|j|_|jjj|_d|j|_|jj|_|jdS(Ns%s.rcbuffer->pybuffer.buf( R$R)REtbuflocal_nd_varR(tbuf_ptrtbuffer_ptr_typet buf_ptr_typetinit_attributes(RR$((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt__init__s   cC@s1|j|_|j|_|j|_dS(N(tget_buf_shapevarstshapetget_buf_stridevarststridestget_buf_suboffsetvarst suboffsets(R((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRscC@s |jdS(Ns%s.diminfo[%d].suboffsets(t _for_all_ndim(R((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRscC@s |jdS(Ns%s.diminfo[%d].strides(R(R((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRscC@s |jdS(Ns%s.diminfo[%d].shape(R(R((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRscC@s0gt|jjD]}||j|f^qS(N(trangeR)R>R((Rtsti((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRscC@sg}|jj}|jj}|dkrxVt||j|jD]6\}}}|j||j||j|qIWd|} t} n|dkrd|} t} nJ|dkrd|} t } n+|dkrd|} t } n t st x=t||jD]&\}}|j||j|qW| |j jkr|j jj| |j d } |j d } | | | d | d |n|jj} d | | |jdj|f}|S(NRYs__Pyx_BufPtrFull%ddR\s__Pyx_BufPtrStrided%ddR]s__Pyx_BufPtrCContig%ddR^s__Pyx_BufPtrFortranContig%ddtutility_code_prototutility_code_defR#tnds%s(%s, %s, %s)s, (R)R>RVRkRRtappendtbuf_lookup_full_codetbuf_lookup_strided_codetbuf_lookup_c_codetbuf_lookup_fortran_codeRURt globalstatet utility_codestaddRtempty_declaration_codeR}R(Rtcodet index_cnamestparamsRRVRRtotfuncnametfuncgent protocodetdefcodetbuf_ptr_type_codetptrcode((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytgenerate_buffer_lookup_codesD                   "   ( RSRTRRRRRRR(((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyR{s      cC@sd}|j}|dkr(|d7}nW|dkrA|d7}n>|dkrZ|d7}n%|dkrs|d 7}n tst|jr|d 7}n|S( Nt PyBUF_FORMATRYs| PyBUF_INDIRECTR\s| PyBUF_STRIDESR]s| PyBUF_C_CONTIGUOUSR^s| PyBUF_F_CONTIGUOUSs| PyBUF_WRITABLE(RVRURtwritable_needed(REt buffer_typetflagsRV((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt get_flags s           cC@s%|j}t|j_t|j_dS(N(RER/R|R0t rcbuf_var(R$RE((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytused_buffer_aux_varss  c C@s|j|jj}}|jj}ddg}|dkrJ|jdng}xNt|jjD]:}x1|D])}|jd||||||fqpWqcW|jdj |dS(NRRRYRs1%s.diminfo[%d].%s = %s.rcbuffer->pybuffer.%s[%d];R ( RER)RVR|R(RRR>tputlnR( t buf_entryRRERVtpybuffernd_structtfldnamestlnRtfldname((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt put_unpack_buffer_aux_into_scope s      cC@so|j}|jj}|jj}|jd||jd||jd||jd||fdS(Ns%s.pybuffer.buf = NULL;s%s.refcount = 0;s%s.data = NULL;s%s.rcbuffer = &%s;(RER|R(RR(R$RtbufauxRtpybuffer_struct((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt put_init_vars2s   cC@s|j}t||j||j}|jd|jd|jjj|j|jd|||jdt||dS(Nt{s'__Pyx_BufFmt_StackElem __pyx_stack[%d];s%s == -1t}( REtget_getbuffer_callR(R)RR<tstruct_nesting_deptht error_goto_ifR(R$RR*REt getbuffer((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytput_acquire_arg_buffer@s   cC@s.|jjt|jd|jjjdS(Ns0__Pyx_SafeReleaseBuffer(&%s.rcbuffer->pybuffer);(Rtuse_utility_codetacquire_utility_codeRRER|R((RR$((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytput_release_buffer_codeNsc C@s`|j}t|j}t||}|jj}t||j}|jj t dt S(Ns__Pyx_GetBufferAndValidate(&%(pybuffernd_struct)s.rcbuffer->pybuffer, (PyObject*)%(obj_cname)s, &%(dtype_typeinfo)s, %(flags)s, %(ndim)d, %(cast)d, __pyx_stack)( R>RqRXRR|R(tget_type_information_cnameR<RRRtlocals( Rt obj_cnameRERR>RXRRtdtype_typeinfo((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRSs  c@s|j|j}}|jj}t||} jdjd|jjtd||} |rjd|j j t j dt } jd| | |fjdjd| tfd td D} jd | jdjd | |jd | jjtjdjdjd| jdjd| x| D]} j j| qWjdt|jj| |j j| njdjd | |jd|t j|t jd|fjj|jdt|jdjddS(sJ 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). Rs'__Pyx_BufFmt_StackElem __pyx_stack[%d];s%ss0__Pyx_SafeReleaseBuffer(&%s.rcbuffer->pybuffer);t manage_refs%s = %s;s if (%s) {s%s < 0c3@s*|] }jjtjdtVqdS(RN(t funcstatet allocate_tempR tpy_object_typeRU(t.0t_(R(s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pys sisPyErr_Fetch(&%s, &%s, &%s);s%s == -1s/Py_XDECREF(%s); Py_XDECREF(%s); Py_XDECREF(%s);s!__Pyx_RaiseBufferFallbackError();s} else {sPyErr_Restore(%s, %s, %s);Rs%s = %s = %s = 0;sA%s = %s; __Pyx_INCREF(Py_None); %s.rcbuffer->pybuffer.buf = NULL;tPy_NoneN(RER)R|R(RRR<RRRRR t c_int_typeRUtunlikelyttupleRRRtraise_buffer_fallback_codet release_tempRterror_goto_if_negttypecastRt error_gototput(t lhs_cnamet rhs_cnameRtis_initializedR*RRERRRRt retcode_cnamet exc_tempstt((Rs5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytput_assign_to_bufferasL  "      "     cC@sb|do |}|dr|jjtjdt}|jd|x tt|||jD]\} \} } } | dkr|jd| |r|jd| | f|jd|j d | || fn|jd || f|j d n| dkrd } nd } |jd|j d| | | f|| fqeW|rs|j j t d}n|j j td}|jd|j d||jd||f|j|j||jd|jj|n^|rRxUt|||jD]8\} } } | dkr|jd| | | fqqWn|j||S(se 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 t wraparoundt boundscheckRs%s = -1;is if (%s < 0) {s %s += %s;sif (%s) %s = %d;s%s < 0s%s = %d;s} else R&s(size_t)s %s >= %s%st __Pyx_RaiseBufferIndexErrorNogilt__Pyx_RaiseBufferIndexErrors if (%s) {s%s != -1s%s(%s);Rsif (%s < 0) %s += %s;(RRR RRURt enumerateRkRRRRRtraise_indexerror_nogiltraise_indexerror_codeRRR(R$t index_signedsRt directivesR*RRWtin_nogil_contexttfailed_dim_temptdimtsignedR(RRXtfunc((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytput_buffer_lookup_codesF 4       ( $cC@s|jtdS(N(Rtbuffer_struct_declare_code(Rt((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRsc C@sdjgt|D]}d|||f^q}|jd||||fdjgt|D]}d|||f^qe}|jd||f|jtd||fdjgt|D]"}td||||f^qd d S( s Generates a buffer lookup function for the right number of dimensions. The function gives back a void* at the right location. s, s i%d, s%d, o%ds1#define %s(type, buf, %s) (type)(%s_imp(buf, %s))s.Py_ssize_t i%d, Py_ssize_t s%d, Py_ssize_t o%ds1static CYTHON_INLINE void* %s_imp(void* buf, %s);sf static CYTHON_INLINE void* %s_imp(void* buf, %s) { char* ptr = (char*)buf; R&sX ptr += s%d * i%d; if (o%d >= 0) ptr = *((char**)ptr) + o%d; s return ptr; }N(RRRR (tprototdefinR#RRt macroargstfuncargs((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRs 55cC@sdjgt|D]}d||f^q}djgt|D]}d||f^qE}|jd|||fdS(s Generates a buffer lookup function for the right number of dimensions. The function gives back a void* at the right location. s, si%d, s%ds + s i%d * s%ds1#define %s(type, buf, %s) (type)((char*)buf + %s)N(RRR(RRR#RRtargstoffset((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRs22cC@s|dkr |jd|ndjgt|D]}d||f^q3}djgt|dD]}d||f^qi}|jd||||dfdS( s 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. is.#define %s(type, buf, i0, s0) ((type)buf + i0)s, si%d, s%ds + s i%d * s%ds9#define %s(type, buf, %s) ((type)((char*)buf + %s) + i%d)N(RRR(RRR#RRRR((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRs  26cC@s|dkr |jd|ndjgt|D]}d||f^q3}djgtd|D]}d||f^qh}|jd|||dfd S( sB Like C lookup, but the first index is optimized instead. is.#define %s(type, buf, i0, s0) ((type)buf + i0)s, si%d, s%ds + s i%d * s%ds9#define %s(type, buf, %s) ((type)((char*)buf + %s) + i%d)iN(RRR(RRR#RRRR((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRs  25cC@s|jtdS(N(RtGetAndReleaseBufferUtilityCode(Rt((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyR)sRcB@sAeZdZeZdZdZdZdZ dZ RS(cC@sdS(N((R((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyR5scC@s t|tS(N(RR(Rtother((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt__eq__8scC@sdS(NiFos((R((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyt__hash__;scK@sdS(N((Rtkwargs((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytget_tree>R&c@s|d}|d}|jj}|jjgtfd|tjddddtd}|j|j }|j|j |j |d }|j ||j |dS( NRRc@s|krdSj|x|jD]}|q'Wx|jD]}t|jtrcqEn|j}|jrE|kr|j rqEnd}}xJ|j j D]<}|j dkr|j }q|j dkr|j }qqW|r j|j||fq qEqEWdS(Nu __getbuffer__u__releasebuffer__(Rtcimported_modulest type_entriesRR:RR)RoR0R-Rtpyfunc_entriesR#t func_cnameRt typeptr_cname(RtmteRtreleaseR_R(t cython_scopetfind_buffer_typesttypestvisited_scopes(s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRIs(      tGetAndReleaseBuffert from_filesBuffer.ctcontextRi( t module_nodeRRRtsetRtloadtdictt format_codeRtinject_string_constantstimplR(RtoutputRt proto_codeRtt util_codeRR ((RRRRs5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytput_code@s         N( RSRTR-trequiresRUtis_cython_utilityRRRRR(((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyR-s    cC@sM|jr dS|jrdS|js,|jr5d}nd}||jSdS(Ntobjecttptrtnn_R&(t is_pyobjectR=t is_typedeftis_struct_or_uniontspecialization_name(R<R1((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytmangle_dtype_nameks   c C@s9t|}d|}d|}|jr-dS|dkrH|j}n|dkrctsctn||jjkr5|jjj||jd}g}|j rx)|j r|j |j |j }qWn|j o|j}|j} |jrd}n|jr|jj} t| dks3tg| D]} t|| j|d^q:} |jd|d txLt| | D];\} } |jd | | j|j| jfd tqW|jd d t|jd d tn tstt|}d }d }|tjkr4d| }d}n|jrTd| }d|}nr|sc|j rld}nZ|j!r~d}nH|jrd}|j"rd}qn$|j#rd}ntst|d} |||| dj$g|D]}t|^qpd t||||f }|j| |d tn|S(sr 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. s__Pyx_TypeInfo_%ss__Pyx_StructFields_%ssittypeinfotNULLis!static __Pyx_StructField %s[] = {tsafes {&%s, "%s", offsetof(%s, %s)},s {NULL, NULL, 0}s};t0sIS_UNSIGNED(%s)s'H's%s ? 'U' : 'I's'C's'R's'S't__PYX_BUF_FLAGS_PACKED_STRUCTs'O'sLstatic __Pyx_TypeInfo %s = { "%s", %s, sizeof(%s), { %s }, %s, %s, %s, %s };s, N(%Rtis_errorR-RRURRRRtis_arrayRtsizet base_typeRtcan_be_complexRtis_simple_buffer_dtypet is_structRt var_entriesR7RR)RR/RkR#R(tstrR t c_char_typetis_intt is_complextis_floattpackedRR(RR<tmaxdeptht namesuffixR#tstructinfo_namettypecodet arraysizestcomplex_possibletdeclcodetfieldstfRRtrepRt is_unsignedt typegroupRttup((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pyRysv              ) *                4cK@s<|dkrtj|d|Stj|dd||SdS(NsBuffer.cR(R-RRR(tutil_code_nameRR((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytload_buffer_utilitys tmax_dimstBufferStructDeclareRtBufferFormatStructstBufferIndexErrortBufferIndexErrorNogiltBufferFallbackErrortBufferGetAndValidatetBufferFormatChecktTypeInfoToFormatN(R<R>RVRWRX(Bt __future__RtVisitorRRtErrorsRRRtCodeRR&RRR R R R RRiR/RUReRgRjRhRlRnRsRrRpRaR-RzRR{RRRRRRRRRRRRRRRRRRR:RR?RRtbuffer_formats_declare_codeRRRRtbuffer_format_check_codet_typeinfo_to_format_code(((s5lib/python2.7/site-packages/Cython/Compiler/Buffer.pytsj ["EG        H B    >  U