ó  ‰\c@sddlmZddlmZdgZyddljZeZ Wne k r[e Z nXd„Z ddddid„ZdS( iÿÿÿÿ(tceil(t cpu_counttapply_parallelNc Cs¾g}tt|dt|ƒƒƒ}d}x…|D]}}||kr||}|||}|dkru|f}q™|f|d|f}n |f}|j|ƒ||9}q3Wt|ƒS(sàSplit the array into equal sized chunks based on the number of available processors. The last chunk in each dimension absorbs the remainder array elements if the number of CPUs does not divide evenly into the number of array elements. Examples -------- >>> _get_chunks((4, 4), 4) ((2, 2), (2, 2)) >>> _get_chunks((4, 4), 2) ((2, 2), (4,)) >>> _get_chunks((5, 5), 2) ((2, 3), (5,)) >>> _get_chunks((2, 4), 2) ((1, 1), (4,)) gð?ii(tintRtlentappendttuple( tshapetncputchunkstnchunks_per_dimt used_chunkstit regular_chunktremainder_chunkt chunk_lens((s:lib/python2.7/site-packages/skimage/util/apply_parallel.pyt _get_chunkss         ic sètstdƒ‚n|d krc|j}y tƒ}Wntk rPd}nXt||ƒ}n|dkrxd}n*|dkrd}n|dkr¢d}n‡‡‡fd †} tj|d |ƒ} | j | |d |ƒj ƒS( sÏMap a function in parallel across an array. Split an array into possibly overlapping chunks of a given depth and boundary type, call the given function in parallel on the chunks, combine the chunks and return the resulting array. Parameters ---------- function : function Function to be mapped which takes an array as an argument. array : numpy array Array which the function will be applied to. chunks : int, tuple, or tuple of tuples, optional A single integer is interpreted as the length of one side of a square chunk that should be tiled across the array. One tuple of length ``array.ndim`` represents the shape of a chunk, and it is tiled across the array. A list of tuples of length ``ndim``, where each sub-tuple is a sequence of chunk sizes along the corresponding dimension. If None, the array is broken up into chunks based on the number of available cpus. More information about chunks is in the documentation `here `_. depth : int, optional Integer equal to the depth of the added boundary cells. Defaults to zero. mode : {'reflect', 'symmetric', 'periodic', 'wrap', 'nearest', 'edge'}, optional type of external boundary padding. extra_arguments : tuple, optional Tuple of arguments to be passed to the function. extra_keywords : dictionary, optional Dictionary of keyword arguments to be passed to the function. Notes ----- Numpy edge modes 'symmetric', 'wrap', and 'edge' are converted to the equivalent `dask` boundary modes 'reflect', 'periodic' and 'nearest', respectively. sACould not import 'dask'. Please install using 'pip install dask'itwraptperiodict symmetrictreflecttedgetnearestcsˆ|ˆˆŽS(N((tarr(textra_argumentstextra_keywordstfunction(s:lib/python2.7/site-packages/skimage/util/apply_parallel.pyt wrapped_funcosR tboundaryN( tdask_availablet RuntimeErrortNoneRRtNotImplementedErrorRtdat from_arrayt map_overlaptcompute( RtarrayR tdepthtmodeRRRRRtdarr((RRRs:lib/python2.7/site-packages/skimage/util/apply_parallel.pyR5s$'           ((tmathRtmultiprocessingRt__all__t dask.arrayR%R!tTrueRt ImportErrortFalseRRR(((s:lib/python2.7/site-packages/skimage/util/apply_parallel.pyts     '