;ò U‹P@c@ssdZdkZdkZdkZdkZdkZdk Z dk Z dk Z dk Z dk Z dklZdklZdklZdklZdkZdZd„Ze ie iee iƒeZdd Zd „Zd „Zd „Zd „Zd„Zd„Z d„Z!d„Z"d„Z#de#_$defd„ƒYZ%d„Z&d„Z'd„Z(d„Z)ed„Z*dd„Z+dd„Z,dd„Z-dd„Z.d „Z/eeed!„Z0d"„Z1d#„Z2dd$„Z3dk4Z5dk6Z6d%k6l7Z7l8Z8l9Z9l:Z:l;Z;dk<Z<d&„Z=dd'„Z>d(„Z?dd)„Z@ed*„ZAd+„ZBd,„ZCd-„ZDdS(.søndarray: Base multidimensional array class This is the base multidimensional array class which implements all structural array operations but treats the array contents as opaque objects $Id: generic.py,v 1.43 2004/02/26 19:29:28 jaytmiller Exp $ N(s_ndarray(s_isIntegerSequence(sproduct(s _alignmenticCstit|ƒffSdS(s9Converts a "buffer" object to a pickling reduction tuple.N(smemorysmemory_from_stringsstrsb(sb((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_buffer_reducesiicCs)d}x|D]}||9}q W|SdS(Ni(snsshapess(sshapesssn((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_product+s cGs©yƒ|fjo| ofSnt|ƒtitigjo|f|}n|o t‚nt|ƒ}t|ƒo|SnWntj onXtdƒ‚dS(sÐVerifies that this is a legal shape specification and returns tuple Shape can be an integer or a sequence of integers. Also can pass several integer arguments. Raises an exception on problems. s"Shape must be sequence of integersN( sshapesargsstypes_typessIntTypesLongTypes TypeErrorstuples_isIntegerSequence(sshapesargs((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysgetShape1s    cCs6|i}||jo|Snt|ƒ}t|ƒ}||jotSn||jot|i ƒ}n4dg||t|i ƒ}|d||!|}||jo_x\t |ƒD]J}||||jo/||djod|| dimension of current view strides[0] --> bytestride for k index dimension element_byte_offset = byteoffset + ( i*strides[2] + j*strides[1] + k*strides[0]) where 0 <= i < shape[2], 0 <= j < shape[1], 0 <= k < shape[0] For contiguous numarray strides[i] = shape[i+1]*strides[i+1] Summary of attribute meanings: _data buffer with data for the array _shape dimensions of the array _byteoffset The byte offset of the first element from the beginning of the buffer _bytestride The separation between items in bytes. _itemsize The size of items in bytes iic Csºti|ƒ||_||_t|ƒ|_ |t jo ||_ n'||jot dƒ‚n ||_ |o ||_n&|i |iƒ}ti|ƒ|_|iƒ|_dS(Nsbytestride must be >= itemsize(s_ndarrays__init__sselfsitemsizes _itemsizes byteoffsets _byteoffsetsgetShapesshapes_shapes bytestridesNones _bytestrides ValueErrorsbuffers_datas nelementsssizesmemorys new_memorys_stridesFromShapes_strides( sselfsshapesitemsizesbuffers byteoffsets bytestrides byteordersalignedssize((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys__init__Ås        cCs5t|iƒot|idƒSn tdƒ‚dS(NisRank-0 array has no length.(slensselfs_shapesints ValueError(sself((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys__len__ÛscCsüt|tƒot|iƒdjo¢|djo||id7}nd|jo|idjn otdƒ‚n|id||i}|t jo|i |ƒSqÈ|i ||ƒSnt|ttfƒo|i|ƒ}n|}t|tititttifƒo |f}nKt|tƒo4t|dtiƒot|ƒ}qv|f}n|}t|tƒ otdƒ‚nt|ƒ}t|ƒo|i||ƒSn5|i|ƒo|i||ƒSn|i||ƒSdS(s@Handles both getting (value == None) and setting (value != None)iisIndex out of ranges Illegal indexN(s isinstanceskeysintslensselfs_shapes IndexErrors_stridess _byteoffsetsoffsetsvaluesNones_getitems_setitemsliststuplesfactorysfvalues_typess SliceTypes EllipsisTypeslongs_ncsNumArraystkeystkey2s_isIntegerSequences_simpleIndexings_isSlices_slicedIndexings_arrayIndexing(sselfskeysvaluesoffsetstkey2stkeysfvalue((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_universalIndexingás8& & (    cCsðt|ƒt|iƒjotdƒ‚n|i|ƒ}t|ƒt|iƒjo2|tjo|i |ƒSqì|i ||ƒnd|i ƒ}|it|ƒ|_|i t|ƒ|_ ||_|tjo|Sn|i|ƒdS(NsToo many indices(slenskeysselfs_shapes IndexErrors_getByteOffsetsoffsetsvaluesNones_getitems_setitemsviewsretarrs_stridess _byteoffsets _copyFrom(sselfskeysvaluesretarrsoffset((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_simpleIndexing s    cCs|i|i|if\}}}|tjo d}n|djotdƒ‚n|djoÄ|tjo d}nJ||jo |}n3|djo%||7}|djo d}qÂn|tjo |}qò||jo |}qò|djo%||7}|djo d}q"qònÍ|tjo|d}nN||jo|d}n3|djo%||7}|djo d}qŽn|tjo d}nN||jo|d}n3|djo%||7}|djo d}qònt|ƒt|ƒt|ƒfSdS(Niisslice step of zero not allowediÿÿÿÿ(sslicesstartsstopsstepsNones IndexErrorsshapesint(sselfsslicesshapesstopsstartsstep((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys _fixSlicesN!                             c Cs"t|ƒ oH|tjo|Sq|ifjo||f  cGsð|iƒ otdƒ‚ntt||Œƒ}|iƒ}|idƒ}|djot dƒ‚n;|djo-t t |ƒƒ}||||idƒt |i|ƒD])}||g}|i||||ƒqâWdS(ssWrite the array to a file repeatedly in blocks This is done similarly to ufunc._doOverDimensions iN(slensdimsslevels indexlevelsblockingparameterssnregShapeIterssshapesleftovers leftoverShapes currentIndexstshapesxrangesistdimss_bytess copyToStringsselfs_datas_getByteOffsets_stridess _itemsizesssfileswrites_shapes_tofileByBlocks(sselfsfilesdimss indexlevelsblockingparametersslevelsistshapesshapesss currentIndexs leftoverShapesleftoversnregShapeItersstdims((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_tofileByBlocksÕs(        cCst|iƒ}|tjot|ƒ}|iƒnt|ƒ|jotdƒ‚nt |ƒ}|i ƒ|t|ƒjotdƒ‚nggf\}}x6|D].}|i|i|ƒ|i|i|ƒq¬Wt|ƒ|_t|ƒ|_dS(s¼transpose() re-shapes the array by permuting it's dimensions as specified by 'axes'. If 'axes' is none, transpose returns the array with it's dimensions reversed. s Wrong number of axes in tranposes#Duplicate or missing transpose axesN(slensselfs_shapesslensaxessNonesrangesreverses ValueErrorsliststaxssortsnshapesnstridessisappends_stridesstuple(sselfsaxessslensnstridessistaxsnshape((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys transposeïs$    cCsT|iƒ}||_t|ƒ|_d|_ti t |ƒ|i ƒ|_ |SdS(sEreturns an empty array identical to 'self' but with 'shape'. iN( sselfsviewscsshapes_shapes_stridesFromShapes_stridess _byteoffsetsmemorys new_memorysproducts _itemsizes_data(sselfsshapesc((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_clones   cCs!|iƒ}|i|ƒ|SdS(N(sselfscopyscsresizesshape(sselfsshapesc((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_clones  cCs_ti|dtiƒ}tt|iƒƒ}t ||d|dd!ƒ}|i ƒ}|SdS(Nstypeii( s_ncsarraysindicess_nts MaybeLongsrangeslens_shapesdts transposescopy(sselfsindicessdt((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_fix_pt_indicess  cKsZt|ƒ}t||ƒ\}}|id|ƒ}|i|ƒ}|i ||ƒ|SdS(Nsshape( slistsindicess _takeShapesselfs impliedShapesNs_clonesresults_fix_pt_indicess_taker(sselfsindicesskeywdss impliedShapesNsresult((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_takes  cKszt|ƒ}t||ƒ\}}t||iƒ o|i |ƒ}nt ||ƒ}|i |ƒ}|i ||ƒdS(N( slistsindicess _takeShapesselfs impliedShapesNs isinstancesvaluess __class__sfactorys _broadcasts_fix_pt_indicess_putter(sselfsindicessvaluesskeywdss impliedShapesN((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_put#s cOst|||ŽSdS(N(stakesselfsindicesskeywords(sselfsindicesskeywords((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pystake,scOst|||ŽSdS(N(sputsselfsindicesskeywords(sselfsindicesskeywords((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysput/scCsti|ƒSdS(N(sufuncsnonzerosself(sself((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysnonzero2scGsÊt||Œ}t|ƒ}|djotdƒ‚n|iƒ}t |i t i ƒpt |i t iƒoŒ|iƒo|iƒ}n |iƒ}t i||iƒ|_ |i|_d|_t||ƒ}|iƒ|iƒ|| ||*n|i i||iƒ|f|_|iƒ|_|on||jo]|}x8|||jo&|d|!||||+||7}qKW|d||!|||+qªnd|(||_|iƒ|_|SdS(sp resize() shrinks/grows 'self' to new 'shape', possibly replacing the underlying buffer object. is*Negative shape dims don't work with resizeN(sgetShapesshapesargssproductsnlens ValueErrorsselfs nelementssolens isinstances_datas_typess BufferTypesmemorys MemoryTypes iscontiguoussviewsoselfscopys new_memorys _itemsizes _bytestrides _byteoffsetsminsblensravelsresizes_shapes_stridesFromShapes_stridessoffset(sselfsshapesargssblensoffsetsnlensoselfsolen((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysresize5s@   ,          cCst|||ƒSdS(sVrepeat() returns a new array with each element 'a[i]' repeated 'r[i]' times. N(srepeatsselfsrepeatssaxis(sselfsrepeatssaxis((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysrepeat`scCs|i|iƒfƒdS(sBravel(self) setshapes 'self' into an equivalent 1D array. N(sselfssetshapes nelements(sself((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysravelescOs'ti|ii}|i||ŽSdS(s‡factory(...) calls the array(...) factory function defined in the source module where the class of 'self' was defined. N( ssyssmodulessselfs __class__s __module__smodulesarraysargsskeys(sselfsargsskeyssmodule((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysfactoryjscCs|iƒSdS(N(sselfscopy(sselfstype((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysastypeqscCs|dG|iGHdG|iGHdG|iGHdG|iGHdG|iGHdG|iGHdG|iƒGHdG|iƒGHd Gt |i ƒGHd S( s1info() prints out the key attributes of an array.sclass:sshape:sstrides:s byteoffset:s bytestride:s itemsize:saligned:s contiguous:sdata:N( sselfs __class__s_shapes_stridess _byteoffsets _bytestrides _itemsizes isaligneds iscontiguoussreprs_data(sself((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysinfots      (<s__name__s __module__s__doc__s _PROTOTYPEsNonessyss byteorders__init__s__len__s_universalIndexings_simpleIndexings _fixSlices_slicedIndexing0s_slicedIndexings_takers_putters_viewsswapaxess __getstate__s __setstate__s __nonzero__s__copy__s __deepcopy__s __reduce__s__repr__s__str__sitemsizes is_c_arrays_stridesFromShapes_arrayIndexings_isSlices _broadcasts_dualbroadcasts _copyFromssetshapesgetshapespropertysshapesgetranksgetflatssetflatsflatscopystostringstofiles_tofileByBlockss transposes_clones_fix_pt_indicess_takes_putstakesputsnonzerosresizesrepeatsravelsfactorysastypesinfo(((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysNDArray¤sn !  )  0 ' "               +        "         +    cGsPti|ƒ}|iƒo|iƒ}n |iƒ}|i||Œ|SdS(s|Returns a reshaped *view* of array if possible, else a *copy*. Call either as reshape(i,j,k) or reshape((i,j,k)). N( s_ncsasarraysarrsvs iscontiguoussviewscopyssetshapesshapesargs(sarrsshapesargssv((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysreshape€s  cCs)ti|ƒ}t||iƒfƒSdS(s)Returns a *view* of array reshaped as 1D.N(s_ncsasarraysarraysreshapes nelements(sarray((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysravelscCsdS(N((s datastring((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys fromstring’scCs$ti|ƒ}|i|ƒ|SdS(sFReturns a *copy* of array, replicated or truncated to match new shape.N(s_ncsarraysresizesshape(sarraysshape((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysresize”s cCs*ti|ƒiƒ}|i|ƒ|SdS(s*Returns the transpose of a *view* of arrayN(s_ncsasarraysarraysviewsvs transposesaxes(sarraysaxessv((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys transposešs iÿÿÿÿcCs$ti|ƒ}|i|ƒ|SdS(s!Returns a sorted *copy* of array.N(s_ncsarrayssortsaxis(sarraysaxis((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pyssort s cCs ti|ƒ}|i|ƒSdS(sXReturns an array of indices which, if "taken" from 'array', would sort 'array'. N(s_ncsasarraysarraysargsortsaxis(sarraysaxis((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysargsort¦scCs ti|ƒ}|i|ƒSdS(s_Returns an array of the indices of the minumum elements of 'array' taken along 'axis'. N(s_ncsasarraysarraysargminsaxis(sarraysaxis((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysargmin­scCs ti|ƒ}|i|ƒSdS(s_Returns an array of the indices of the maximum elements of 'array' taken along 'axis'. N(s_ncsasarraysarraysargmaxsaxis(sarraysaxis((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysargmax´scCs\|tjo|Snti|ƒiƒ}y|i||ƒWnt j onX|SdS(s<Returns a *view* of array with axis1 and axis2 interchanged.N( sarraysNones_ncsasarraysviewsvsswapaxessaxis1saxis2s ValueError(sarraysaxis1saxis2sv((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysswapaxes»s cCs˜|tjo |tjo.|tjoti|ƒSq”tdƒ‚nM|tjp |tjotdƒ‚ntti |dƒ||f|ƒSdS(söwhere() returns an array shaped like 'condition' with elements selected from 'x' or 'y' by the 1 or 0 value of each condition element, respectively. If neither 'x' nor 'y' is specified, where acts as a synonym for nonzero(). s6single parameter where() does not support output arraysInvalid parameter listiN( sxsNonesysoutsufuncsnonzeros conditions ValueErrorschooses not_equal(s conditionsxsysout((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pyswhereÆs cCs@ti||ƒdti||ƒ}t||||fƒSdS(sœclip() returns a new array with every entry in m that is less than m_min replaced by m_min, and every entry greater than m_max replaced by m_max. iN(sufuncslesssmsm_minsgreatersm_maxsselectorschoose(smsm_minsm_maxsselector((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pysclipØs&c Cs,ttigi}|D]}||idƒq~ƒ}|did}|ft |ƒ}yt i |ƒ}Wn)tj o|did|ƒ}nX|did|d|ƒ}d}xb|D]Z}|id|jotdƒ‚n||||id!i|ƒ||id7}qÆW|SdS(s\_concat handles the simplest case of concatenating numarray along the zero-th axis. iisshapestypes4_concat array shapes must match except 1st dimensionN(sreducesoperatorsaddsappends_[1]sarrssas_shapescombinedLengthsrShapestuples destShapesufuncs _maxPopTypesconvTypes TypeErrors_clonesdests __class__sixs ValueErrors _copyFrom( sarrssasixsrShapesdests destShapes_[1]scombinedLengthsconvType((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys_concatßs":c Csttti|ƒ}|djot|ƒSnDttgi}|D]}|t||dƒƒqA~ƒ|dƒSdS(sqconcatenate() joins the sequence of numarray 'arrs' in a into a single array along the specified 'axis'. iN( smaps_ncsasarraysarrssaxiss_concatsswapaxessappends_[1]sm(sarrssaxissms_[1]((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys concatenateös  (schooses_takestakes_putsputcCst|ti|ƒƒSdS(spcompress selects members of array 'a' along 'axis' which correspond to non-zero members of 'condition'. N(s_takesasufuncsnonzeros condition(s conditionsa((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pys _compress scCsG|djot||ƒSn&tt|t|d|ƒƒd|ƒSdS(Ni(saxiss _compresss conditionsasswapaxes(s conditionsasaxis((sI/mit/python/arch/i386_rh9/lib/python2.3/site-packages/numarray/generic.pyscompresss c Csèti|ƒo|ft|ƒ}nti|dtiƒ}ti i |ƒ}|f|i d}|id|d|iƒ}d}xWtt|ƒƒD]C}||}x&t|ƒD]}|||||