Ñò pXJc @sÒdZddkTddkZeiiddƒddkiiZddkii Z e i Z ddk ii Z d„Zd„Zd„ZddkZddkZd „Zed joed d d dƒndS(t iÿÿÿÿ(t*Nis..c Cst|ƒ}t|ƒ}t|ƒ}t|ƒ}ti|iƒ}d}dtƒ}ti|ddgdtdddd ƒ\}}||fS( Nt PyArray_FLOATs‚ #line 37 "vq.py" // Use tensor notation. blitz::Array<%(type)s,2> dist_sq(Ncode_book[0],Nobs[0]); blitz::firstIndex i; blitz::secondIndex j; blitz::thirdIndex k; dist_sq = sum(pow2(obs(j,k) - code_book(i,k)),k); // Surely there is a better way to do this... PyArrayObject* py_code = (PyArrayObject*) PyArray_FromDims(1,&Nobs[0],PyArray_LONG); blitz::Array code((int*)(py_code->data), blitz::shape(Nobs[0]), blitz::neverDeleteData); code = minIndex(dist_sq(j,i),j); PyArrayObject* py_min_dist = (PyArrayObject*) PyArray_FromDims(1,&Nobs[0],PyArray_FLOAT); blitz::Array min_dist((float*)(py_min_dist->data), blitz::shape(Nobs[0]), blitz::neverDeleteData); min_dist = sqrt(min(dist_sq(j,i),j)); py::tuple results(2); results[0] = py_code; results[1] = py_min_dist; return_val = results; tobst code_bookttype_converterstcompilertgcctverbosei( tasarraytshapetc_spectnum_to_c_typesttypecodetlocalst inline_toolstinlinetblitz_type_converters(RRtobs_sht code_book_shttypetar_typetcodet distortion((sGC:\graphics\Tools\Python26\Lib\site-packages\scipy\weave\examples\vq.pytvqs     c Cst|ƒ}t|ƒ}t|ƒ}t|ƒ}ti|iƒ}d}dtƒ}ti|ddgdtdddd ƒ\}}||fS( sd doesn't use blitz (except in conversion) ALSO DOES NOT HANDLE STRIDED ARRAYS CORRECTLY Rs½ #line 83 "vq.py" // THIS DOES NOT HANDLE STRIDED ARRAYS CORRECTLY // Surely there is a better way to do this... PyArrayObject* py_code = (PyArrayObject*) PyArray_FromDims(1,&Nobs[0],PyArray_LONG); PyArrayObject* py_min_dist = (PyArrayObject*) PyArray_FromDims(1,&Nobs[0],PyArray_FLOAT); int* raw_code = (int*)(py_code->data); float* raw_min_dist = (float*)(py_min_dist->data); %(type)s* raw_obs = obs.data(); %(type)s* raw_code_book = code_book.data(); %(type)s* this_obs = NULL; %(type)s* this_code = NULL; int Nfeatures = Nobs[1]; float diff,dist; for(int i=0; i < Nobs[0]; i++) { this_obs = &raw_obs[i*Nfeatures]; raw_min_dist[i] = (%(type)s)10000000.; // big number for(int j=0; j < Ncode_book[0]; j++) { this_code = &raw_code_book[j*Nfeatures]; dist = 0; for(int k=0; k < Nfeatures; k++) { diff = this_obs[k] - this_code[k]; dist += diff*diff; } dist = dist; if (dist < raw_min_dist[i]) { raw_code[i] = j; raw_min_dist[i] = dist; } } raw_min_dist[i] = sqrt(raw_min_dist[i]); } py::tuple results(2); results[0] = py_code; results[1] = py_min_dist; return_val = results; RRRRRRi( R R R R R RRRR(RRRRRRRR((sGC:\graphics\Tools\Python26\Lib\site-packages\scipy\weave\examples\vq.pytvq2Bs    * cCs„t|ƒ}t|ƒ}t|ƒ}t|ƒ}ti|iƒ}dtƒ}ddk}ti|ddgƒ\}}||fS(sw Uses standard array conversion completely bi-passing blitz. THIS DOES NOT HANDLE STRIDED ARRAYS CORRECTLY s #line 139 "vq.py" // Surely there is a better way to do this... PyArrayObject* py_code = (PyArrayObject*) PyArray_FromDims(1,&Nobs[0],PyArray_LONG); PyArrayObject* py_min_dist = (PyArrayObject*) PyArray_FromDims(1,&Nobs[0],PyArray_FLOAT); int* code_data = (int*)(py_code->data); float* min_dist_data = (float*)(py_min_dist->data); %(type)s* this_obs = NULL; %(type)s* this_code = NULL; int Nfeatures = Nobs[1]; float diff,dist; for(int i=0; i < Nobs[0]; i++) { this_obs = &obs_data[i*Nfeatures]; min_dist_data[i] = (float)10000000.; // big number for(int j=0; j < Ncode_book[0]; j++) { this_code = &code_book_data[j*Nfeatures]; dist = 0; for(int k=0; k < Nfeatures; k++) { diff = this_obs[k] - this_code[k]; dist += diff*diff; } if (dist < min_dist_data[i]) { code_data[i] = j; min_dist_data[i] = dist; } } min_dist_data[i] = sqrt(min_dist_data[i]); } py::tuple results(2); results[0] = py_code; results[1] = py_min_dist; return_val = results; iÿÿÿÿNRR( R R R R R Rt ext_toolsRR(RRRRRRRR((sGC:\graphics\Tools\Python26\Lib\site-packages\scipy\weave\examples\vq.pytvq3ƒs    '  cCsðtidd||fƒ}tidd||fƒ}ddk}|iid||||fGHtiƒ}x2t|ƒD]$}|iii||ƒ\} } qzWtiƒ} | |} dG| ||GH| d G| d GHtiƒ}x2t|ƒD]$}|iii||ƒ\} } qóWtiƒ} dG| ||GH| d G| d GHd| | |GHt||ƒ} tiƒ}x)t|ƒD]}t||ƒ\} } q‚Wtiƒ} d G| ||GH| d G| d GHd| | |GHt||ƒ} tiƒ}x)t|ƒD]}t||ƒ\} } qWtiƒ} d G| ||GH| d G| d GHd| | |GHt ||ƒ} tiƒ}x)t|ƒD]}t ||ƒ\} } qŽWtiƒ} d G| ||GH| d G| d GHd| | |GHdS( Nggð?iÿÿÿÿsBvq with %d observation, %d features and %d codes for %d iterationss speed in python:is speed in standard c:s speed up: %3.2fs speed inline/blitz:s speed inline/blitz2:s speed using C arrays:( t RandomArraytnormaltscipy.cluster.vqtclusterRttimetrangetpy_vqRR(tmtNobstNcodest NfeaturesRtcodestscipytt1tiRtdisttt2tpytb((sGC:\graphics\Tools\Python26\Lib\site-packages\scipy\weave\examples\vq.pytcompare¿sb    "    "          t__main__idièii (t__doc__tnumpytsystpathtinserttscipy.weave.inline_toolstweaveRtscipy.weave.converterst converterstblitzRtscipy.weave.c_specR RRRR RR/t__name__(((sGC:\graphics\Tools\Python26\Lib\site-packages\scipy\weave\examples\vq.pyts    ) A :   5