ó ʽ÷Xc@sdZddlZddlZddlmZdZdZdZdZdefd efd efd efgZ d:d;gZ d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLgZ d2Z d3Z d4Zd5„Zd6„Zd7„Zd8„Zed9kreƒndS(MsZ Generate manual wrappers for C++ sparsetools code. Type codes used: 'i': integer scalar 'I': integer array 'T': data array 'B': boolean array 'V': std::vector* 'W': std::vector* '*': indicates that the next argument is an output argument 'v': void See sparsetools.cxx for more details. iÿÿÿÿN(tnewers¾ bsr_diagonal v iiiiIIT*T bsr_scale_rows v iiiiII*TT bsr_scale_columns v iiiiII*TT bsr_sort_indices v iiii*I*I*T bsr_transpose v iiiiIIT*I*I*T bsr_matmat_pass2 v iiiiiIITIIT*I*I*T bsr_matvec v iiiiIITT*T bsr_matvecs v iiiiiIITT*T bsr_elmul_bsr v iiiiIITIIT*I*I*T bsr_eldiv_bsr v iiiiIITIIT*I*I*T bsr_plus_bsr v iiiiIITIIT*I*I*T bsr_minus_bsr v iiiiIITIIT*I*I*T bsr_maximum_bsr v iiiiIITIIT*I*I*T bsr_minimum_bsr v iiiiIITIIT*I*I*T bsr_ne_bsr v iiiiIITIIT*I*I*B bsr_lt_bsr v iiiiIITIIT*I*I*B bsr_gt_bsr v iiiiIITIIT*I*I*B bsr_le_bsr v iiiiIITIIT*I*I*B bsr_ge_bsr v iiiiIITIIT*I*I*B s[ csc_diagonal v iiIIT*T csc_tocsr v iiIIT*I*I*T csc_matmat_pass1 v iiIIII*I csc_matmat_pass2 v iiIITIIT*I*I*T csc_matvec v iiIITT*T csc_matvecs v iiiIITT*T csc_elmul_csc v iiIITIIT*I*I*T csc_eldiv_csc v iiIITIIT*I*I*T csc_plus_csc v iiIITIIT*I*I*T csc_minus_csc v iiIITIIT*I*I*T csc_maximum_csc v iiIITIIT*I*I*T csc_minimum_csc v iiIITIIT*I*I*T csc_ne_csc v iiIITIIT*I*I*B csc_lt_csc v iiIITIIT*I*I*B csc_gt_csc v iiIITIIT*I*I*B csc_le_csc v iiIITIIT*I*I*B csc_ge_csc v iiIITIIT*I*I*B s  csr_matmat_pass1 v iiIIII*I csr_matmat_pass2 v iiIITIIT*I*I*T csr_diagonal v iiIIT*T csr_tocsc v iiIIT*I*I*T csr_tobsr v iiiiIIT*I*I*T csr_matvec v iiIITT*T csr_matvecs v iiiIITT*T csr_elmul_csr v iiIITIIT*I*I*T csr_eldiv_csr v iiIITIIT*I*I*T csr_plus_csr v iiIITIIT*I*I*T csr_minus_csr v iiIITIIT*I*I*T csr_maximum_csr v iiIITIIT*I*I*T csr_minimum_csr v iiIITIIT*I*I*T csr_ne_csr v iiIITIIT*I*I*B csr_lt_csr v iiIITIIT*I*I*B csr_gt_csr v iiIITIIT*I*I*B csr_le_csr v iiIITIIT*I*I*B csr_ge_csr v iiIITIIT*I*I*B csr_scale_rows v iiII*TT csr_scale_columns v iiII*TT csr_sort_indices v iI*I*T csr_eliminate_zeros v ii*I*I*T csr_sum_duplicates v ii*I*I*T get_csr_submatrix v iiIITiiii*V*V*W csr_sample_values v iiIITiII*T csr_count_blocks i iiiiII csr_sample_offsets i iiIIiII*I expandptr v iI*I test_throw_error i csr_has_sorted_indices i iII csr_has_canonical_format i iII sž coo_tocsr v iiiIIT*I*I*T coo_todense v iiiIIT*Ti coo_matvec v iIITT*T dia_matvec v iiiiITT*T cs_graph_components i iII*I tbsrtcsrtcsctothert NPY_INT32t npy_int32t NPY_INT64t npy_int64tNPY_BOOLtnpy_bool_wrappertNPY_BYTEtnpy_bytet NPY_UBYTEt npy_ubytet NPY_SHORTt npy_shortt NPY_USHORTt npy_ushorttNPY_INTtnpy_inttNPY_UINTtnpy_uinttNPY_LONGtnpy_longt NPY_ULONGt npy_ulongt NPY_LONGLONGt npy_longlongt NPY_ULONGLONGt npy_ulonglongt NPY_FLOATt npy_floatt NPY_DOUBLEt npy_doubletNPY_LONGDOUBLEtnpy_longdoublet NPY_CFLOATtnpy_cfloat_wrappert NPY_CDOUBLEtnpy_cdouble_wrappertNPY_CLONGDOUBLEtnpy_clongdouble_wrappersd static Py_ssize_t %(name)s_thunk(int I_typenum, int T_typenum, void **a) { %(thunk_content)s } s£ NPY_VISIBILITY_HIDDEN PyObject * %(name)s_method(PyObject *self, PyObject *args) { return call_thunk('%(ret_spec)s', "%(arg_spec)s", %(name)s_thunk, args); } s] static int get_thunk_case(int I_typenum, int T_typenum) { %(content)s; return -1; } c Csüg}g}d}d}xÇtD]¿\}}d}||td|d|ƒ7}|j||d |d fƒ|d7}x]tD]U\}}d}||td|d|ƒ7}|j|||||fƒ|d7}q{W|d 7}qW||ttd |ƒfS( s¶ Get a list containing cartesian product of data types, plus a getter routine. Returns ------- i_types : list [(j, I_typenum, None, I_type, None), ...] Pairing of index type numbers and the corresponding C++ types, and an unique index `j`. This is for routines that are parameterized only by I but not by T. it_types : list [(j, I_typenum, T_typenum, I_type, T_type), ...] Same as `i_types`, but for routines parameterized both by T and I. getter_code : str C++ code for a function that takes I_typenum, T_typenum and returns the unique index corresponding to the lists, or -1 if no match was found. is if (0) {}sb else if (I_typenum == %(I_typenum)s) { if (T_typenum == -1) { return %(j)s; }t I_typenumtjisC else if (T_typenum == %(T_typenum)s) { return %(j)s; }t T_typenums }tcontentN(tI_TYPEStdicttappendtNonetT_TYPEStGET_THUNK_CASE_TEMPLATE( tit_typesti_typesR,t getter_codeR+tI_typetpieceR-tT_type((s@/tmp/pip-build-7oUkmx/scipy/scipy/sparse/generate_sparsetools.pytget_thunk_type_set¼s"  csS|d}|d‰‡fd†}d}xÛ|D]Ó\}}}} } || | ƒ} | dkrpd| f} nd| | f} dˆkr™| d7} nd } |d kr¸| d 7} n | d 7} || td |d| d| d|d|d| d|d| ƒ7}q0W|d7}ttd|d|ƒ}ttd|d|dˆƒ}||fS(s3 Generate thunk and method code for a given routine. Parameters ---------- name : str Name of the C++ routine args : str Argument list specification (in format explained above) types : list List of types to instantiate, as returned `get_thunk_type_set` iics”g}t}d}xrˆD]j}|r+dnd}t}|dkrOt}qn*|dkry|jd|||fƒn|dkr£|jd|||fƒnÖ|d krÍ|jd|||fƒn¬|d krð|jd |fƒn‰|d kr+|rtd ƒ‚n|jd||fƒnN|dkrf|rLtdƒ‚n|jd||fƒntd|fƒ‚|d7}qWdj|ƒS(sE Generate argument list for calling the C++ function itsconst t*tis *(%s*)a[%d]tIs (%s*)a[%d]tTtBs(npy_bool_wrapper*)a[%d]tVs"'V' argument must be an output args(std::vector<%s>*)a[%d]tWs"'W' argument must be an output argsInvalid spec character %ris, (tFalsetTrueR1t ValueErrortjoin(R8R:targstnext_is_writeableR,tttconst(targ_spec(s@/tmp/pip-build-7oUkmx/scipy/scipy/sparse/generate_sparsetools.pyt get_arglistþs8        s>int j = get_thunk_case(I_typenum, T_typenum); switch (j) {s%ss%s,%sRAs,npy_bool_wrappers case %(j)s:tvsM (void)%(name)s<%(dispatch)s>(%(arglist)s); return 0;s8 return %(name)s<%(dispatch)s>(%(arglist)s);R,R8R:R+R-targlisttnametdispatchsb default: throw std::runtime_error("internal error: invalid argument typenums"); }t thunk_contenttret_specRLN(R2R0tTHUNK_TEMPLATEtMETHOD_TEMPLATE(RPRHttypesRSRMRRR,R+R-R8R:RORQR9t thunk_codet method_code((RLs@/tmp/pip-build-7oUkmx/scipy/scipy/sparse/generate_sparsetools.pyt parse_routineìs:  %         c Csltjdtjƒƒ}|jddddddtƒ|jƒ\}}g}tƒ\}}}xtD] \}}g} g} x(|j ƒD]} | jƒ} | s| j dƒr»qny| j dd ƒ\} }Wn$t k rýt d | fƒ‚nXd j|j ƒƒ}d |ks+d |krFt| ||ƒ\} }nt| ||ƒ\} }| |kr€t d| fƒ‚n|j| ƒ| j| ƒ| j|ƒqWtjjtjjtƒd|dƒ}tt|ƒsë|jred|fGHt|dƒZ}t|ƒ|j|ƒx| D]} |j| ƒq'Wx| D]}|j|ƒqEWWdQXqhd|fGHqhWd }x|D]} |d| f7}q‚Wd}x%|D]} |dtd| ƒ7}qªW|d7}tjjtjjtƒddƒ}tt|ƒs|jr\d|fGHt|dƒ+}t|ƒ|j|ƒ|j|ƒWdQXn d|fGHdS(Ntusages --no-forcetactiont store_falsetdesttforcetdefaultt#isMalformed line: %rR<RJR@sDuplicate routine %rt sparsetoolss_impl.hs$[generate_sparsetools] generating %rtws,[generate_sparsetools] %r already up-to-datesCNPY_VISIBILITY_HIDDEN PyObject *%s_method(PyObject *, PyObject *); s4 static struct PyMethodDef sparsetools_methods[] = {sH {"%(name)s", (PyCFunction)%(name)s_method, METH_VARARGS, NULL},RPs% {NULL, NULL, 0, NULL} };ssparsetools_impl.h(toptparset OptionParsert__doc__tstript add_optionREt parse_argsR;tCOMPILATION_UNITSt splitlinest startswithtsplitR2RFRGRYR1tostpathtdirnamet__file__RR^topentwrite_autogen_blurbtwriteR0(tptoptionsRHtnamesR6R5R7t unit_nametroutinestthunkstmethodstlineRPtthunktmethodtdsttft method_defst method_struct((s@/tmp/pip-build-7oUkmx/scipy/scipy/sparse/generate_sparsetools.pytmainKst                  cCs|jdƒdS(Nsh/* This file is autogenerated by generate_sparsetools.py * Do not edit manually or check into VCS. */ (Rs(tstream((s@/tmp/pip-build-7oUkmx/scipy/scipy/sparse/generate_sparsetools.pyRržst__main__(RR(RR(R R (R R (R R(RR(RR(RR(RR(RR(RR(RR(RR(RR (R!R"(R#R$(R%R&(R'R((R)R*(ReRcRmtdistutils.dep_utilRt BSR_ROUTINESt CSC_ROUTINESt CSR_ROUTINEStOTHER_ROUTINESRiR/R3RTRUR4R;RYR‚Rrt__name__(((s@/tmp/pip-build-7oUkmx/scipy/scipy/sparse/generate_sparsetools.pytsN  #       0 _ S