o 1&iZ@s>dZddlZddlmZdZeddgZd dd Zd d ZdS) z The same algorithm as translated from numpy. See numpy/core/src/npysort/mergesort.c.src. The high-level numba code is adding a little overhead comparing to the pure-C implementation in numpy. N) namedtupleMergesortImplementation run_mergesortFcstddd}dur|di|ddn |di||r/|di|fddn |di|fdd|di|fd d |dd fd d }|dd fdd}t|rg|dS|dS)NTF)no_cpython_wrapperZ_nrtcSs||kSN)abrr?C:\wamp64\www\opt\env\Lib\site-packages\numba/misc/mergesort.pyltszmake_mergesort_impl..ltcs||||Srrr r valsr rr lessthansz%make_mergesort_impl..lessthancs ||Srrr rrr r#s c s|jtkr|jd}|d|||||d||t|D]}||||<q"|d|}||d}|}d}}} ||jkrv||jkrv|||||s^|||| <|d7}n |||| <|d7}| d7} ||jkrv||jksI||jkr|||| <|d7}| d7} ||jks{||jkr|||| <|d7}| d7} ||jksdSdSd}||jkr|}|dkr||||d|r||||d||d<||<|d8}|dkr||||d|s|d7}||jksdSdS)aThe actual mergesort function Parameters ---------- arr : array [read+write] The values being sorted inplace. For argsort, this is the indices. vals : array [readonly] ``None`` for normal sort. In argsort, this is the actual array values. ws : array [write] The workspace. Must be of size ``arr.size // 2`` Nr)sizeSMALL_MERGESORTrange) arrrwsmidileftrightoutjk)argmergesort_innerrrr r'sJ              " z/make_mergesort_impl..argmergesort_inner)rcs&tj|jd|jd}|d||S)ZInplacerdtypeN)npemptyrr!)rrrrr mergesortes z&make_mergesort_impl..mergesortcs2t|j}tj|jd|jd}||||S)z Out-of-placerr )r"Zarangerr#r!)rZidxsrr$rr argmergesortms  z)make_mergesort_impl..argmergesort)rr)dictr)wrapr Z is_argsortZ kwargs_liter%r&r)rrr r make_mergesort_impls*      =r)cOs"ddlm}t|g|Ri|S)Nr)njit)Znumbar*r))argskwargsr*rrr make_jit_mergesortzs r-)NF) __doc__numpyr" collectionsrrrr)r-rrrr s   g