IR-es dZddlZddlmZddlmZmZmZgdZ e Z iZ iZ e Z e dejjj DzZ e hejejejejejejejejejejejejejejejej ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4ej5ej6ej7ej8ej9ej:ej;ej<ej=ej>ej?ej@ejAejBejCejDejEejFejGzZ ejHejIejJejKejLejMejNejOejPejQejRejSejTejUejVhZWeWejXejYejZej[ej\ej]ej^ej_ej`ejaejbejcejdejeejfejgejhejiejjejkejlejmhzZWeWdejnjoj DzZWereWejpejqhzZWeejrejsejthzZeeWzZee Zuee ZvdZwevd Zxevd Zyevd Zzevd Z{evdLdZ|euej}ej~ejejejejhdZeuejejejejejejhdZeuejejhdZeuejhdZeudMdZevdNdZevdOdZevdOdZevdOdZevdOdZevdPdZevdZevd ZevdQd"Zevd#Zevd$ZevdRd%Zer evd&Zevd'ZevdSd(ZeudNd)Zevd*Zevd d+d,ZeudNd-ZevdNdd.d/Zd0ZdTd1ZevdUd2Zd3ZdUd4ZevdUd5Zevd6ZevdMd7Zevd8Zevd9ZevdVd:ZeudWd;Zevd<Zevd=ZevdXd?Zevd@ZGdAdBZdYdEZevddddCdDejddddddDf dFZevdZdGZdddHdHdIZdJZejnjj D]-Zeveeeeeee<.e edKe e e e zDz Z dS)[aHelpers for letting numpy functions interact with Masked arrays. The module supplies helper routines for numpy functions that propagate masks appropriately., for use in the ``__array_function__`` implementation of `~astropy.utils.masked.MaskedNDArray`. They are not very useful on their own, but the ones with docstrings are included in the documentation so that there is a place to find out how the mask is interpreted. N)FunctionAssigner) NUMPY_LT_1_23 NUMPY_LT_1_24 NUMPY_LT_2_0)MASKED_SAFE_FUNCTIONSAPPLY_TO_BOTH_FUNCTIONSDISPATCHED_FUNCTIONSUNSUPPORTED_FUNCTIONSc@h|]}|dvtt|S)>putalenwherechooseresize searchsortedgetattrnp).0names Elib/python3.11/site-packages/astropy/utils/masked/function_helpers.py rOs8  OOO BOOOc8h|]}tt|Sr)r setopsnames rrrs/ *GB rc~ddlm}|j|\}}tdt ||D}||fS)zSeparate out arguments into tuples of data and masks. An all-False mask is created if an argument does not have a mask. Maskedc3|K|]7\}}||n+tjtj|tV8dSN)rzerosshapebool)rdms r z&_get_data_and_masks..sS@D1Q]!d ; ;r)corer _get_data_and_maskstuplezip)argsr datamaskss rr*r*sk ,&,d3KD% HKDRWHXHX  E ;rcjtj|jg|Ri||jdfSr")rdatetime_as_stringunmaskedmaskcopy)arrr-kwargss rr1r1s7  !#, @ @ @ @ @ @#(--//SW XXrchtj|j|jdfSr")rsincr2r3r4xs rr8r8s% 71:   t 33rchtj|j|jdfSr")r iscomplexr2r3r4r9s rr<r<s% < # #QV[[]]D 88rcjtj|jg|Ri||jdfSr")runwrapr2r3r4)pr-r6s rr>r>s5 9QZ 1$ 1 1 1& 1 116;;==$ FFrTc~tj|j||||}|r||jdfn|S)N)r4nanposinfneginf)r nan_to_numr2r3r4)r:r4rBrCrDr.s rrErEs? =$Cv V V VD*. 5D!&++-- & &A5r)helpsc>t|\}}||z||z|dfSr"r*)ar-r6r.r3s rmasked_a_helperrJs,%Q''JD$ $;t VT 11rc>t|\}}||z||z|dfSr"rH)r'r-r6r.r3s rmasked_m_helperrL*$Q''JD$ $;t VT 11rc>t|\}}||z||z|dfSr"rH)vr-r6r.r3s rmasked_v_helperrPrMrc>t|\}}||z||z|dfSr"rH)arrayr-r6r.r3s rmasked_arr_helperrSs*$U++JD$ $;t VT 11rFcPt|\}}||t||dfS)a?Broadcast array to the given shape. Like `numpy.broadcast_to`, and applied to both unmasked data and mask. Note that ``subok`` is taken to mean whether or not subclasses of the unmasked data and mask are allowed, i.e., for ``subok=False``, a `~astropy.utils.masked.MaskedNDArray` will be returned. )r$subokNr*dict)rRr$rUr.r3s r broadcast_torXs1%U++JD$ t%u555t ;;rctjtj|tj||S)N)out)rmultiplyouterravel)rIbrZs rr\r\s/ ;  RXa[["(1++3  ? ??rKctj|j||||}|D|jjr$tj|jntjd}tj|j||||}||dfS)zReturn a new array with the same shape and type as a given array. Like `numpy.empty_like`, but will add an empty mask. dtypeorderrUr$N?)r empty_liker2rbnamesmamake_mask_descrr3) prototyperbrcrUr$r2r3s rreres }%uEH ~# BE ! !(. 1 1 1#  =e5U   D T4 rcHtj|j||||}|ddfS)zReturn an array of zeros with the same shape and type as a given array. Like `numpy.zeros_like`, but will add an all-false mask. raFN)r zeros_liker2rIrbrcrUr$r2s rrkrks6 } %uEH UD  rcHtj|j||||}|ddfS)zReturn an array of ones with the same shape and type as a given array. Like `numpy.ones_like`, but will add an all-false mask. raFN)r ones_liker2rls rrnrns6 | %uEH UD  rcBtj|||||}||d<|S)zReturn a full array with the same shape and type as a given array. Like `numpy.full_like`, but with a mask that is also set. If ``fill_value`` is `numpy.ma.masked`, the data will be left unset (i.e., as created by `numpy.empty_like`). ra.)rre)rI fill_valuerbrcrUr$results r full_likerr*s,]1Ee5 Q Q QFF3K Mrraisecddlm}t||st||st||\}}|t j|j|||t j|j|||dS)zReplaces specified elements of an array with given values. Like `numpy.put`, but for masked array ``a`` and possibly masked value ``v``. Masked indices ``ind`` are not supported. rrN)mode) astropy.utils.maskedr isinstanceNotImplementedError_get_data_and_maskrr r2r3)rIindrOrur v_datav_masks rr r 7s,+++++#v"jF&;&;"!!))!,,NFF  qz3T2222F163T******rcddlm}t||st||st||\}}|t j|j||t j|j||dS)zChanges elements of an array based on conditional and input values. Like `numpy.putmask`, but for masked array ``a`` and possibly masked ``values``. Masked ``mask`` is not supported. rrN) rvr rwrxryrputmaskr2r3)rIr3valuesr values_data values_masks rr~r~Js,+++++$"z!V'<'<"!! 33F;;K 1:t[111Jqvt[)))))rcddlm}t||st||st||\}}|t j|j||t j|j||dS)zChange elements of an array based on conditional and input values. Like `numpy.place`, but for masked array ``a`` and possibly masked ``values``. Masked ``mask`` is not supported. rrN) rvr rwrxryrplacer2r3)r5r3valsr vals_data vals_masks rrr\s,+++++$"z#v'>'>"!!11$77Iy tY///HSXtY'''''r same_kindcddlm}t||rt||rt||\}}|t j|j||||t j|j||dSdS)zCopies values from one array to another, broadcasting as necessary. Like `numpy.copyto`, but for masked destination ``dst`` and possibly masked source ``src``. rrN)castingr)r) rvr rwrxryrcopytor2r3)dstsrcrrr src_datasrc_masks rrrns,+++++ c6 " ""j&?&?"!!//44Hh #,'GGGG #(HE222222rctj|jg|Ri|}tj|jg|Ri|t }||dfSr")rpackbitsr2r3astyper%rIr-r6rqr3s rrrs_ [ 5d 5 5 5f 5 5F ;qv / / / / / / 6 6t < |jdkr3||j|jkr||j}|j|j}d}|a||\}}|Gtj || t| t}tj ||d}||dfS)zCount number of occurrences of each value in array of non-negative ints. Like `numpy.bincount`, but masked entries in ``x`` will be skipped. Any masked entries in ``weights`` will lead the corresponding bin to be masked. rrNr) minlength) rvr r asanyarrayrwndimr3r2rybincountrintr%)r:weightsrr r3w_maskrqs rrrs,+++++-((!V 1  7<16#9#9qvg&G Jw  D 33G<<  ;q&--"4"4 JJJQQRVWWD [Gq 1 1 1F 4 rcZ|}|d|S)Nraxis)r4sort)rIrqs rmsortrs(   rcX|}|t|jjt js]|jjdvr|dS|jjdkr|dS|dS|S)NbhBHFgGD) r4r issubclassrbtypercomplexfloatingcharr)rIr^s r sort_complexrs AFFHHH aglB$6 7 7 7<6 ! !88C== W\S 88C== 88C== rc.t|\}}|/tj||||tj||dfSddlm}t ||st tj||j|tj||j||||S)N)rrbrrrr)rZr)rZrrbr) r*r concatenatervr rwrxr3r2)arraysrrZrbrr.r/r s rrrs%v.KD% { N4d% I I I N5t , , ,   0/////#v&& &% % u#(6666 tDwWWWW rcPt||\}}||t|dfS)NrrV)r5rrr.r/s rappendrs,%c622KD% $ --rcddlm}tjj|\}}}}tjj|||\}}}tjd|D}td|D}td|D} |r| sdnd} |tj ||| } t||D]\} } | | tf| z<| S) Nrrcg|] }|j Srrrr5s r zblock..s9993SY999rc30K|]}|jdVdS) F_CONTIGUOUSNflagsrs rr(zblock..)>>#)N+>>>>>>rc30K|]}|jdVdS) C_CONTIGUOUSNrrs rr(zblock..rrrC)r$rbrc) rvr rr) shape_base _block_setup_block_info_recursion result_typeallemptyr,Ellipsis)rr list_ndim result_ndim final_sizer$slicesrbF_orderC_orderrcrq the_slicer5s rblockrs,+++++131C1P1PQW1X1X.FI{JG.DD ;E66 N99&999 :E>>v>>>>>G>>v>>>>>G 37 3CCE VBH5UCCC D DFff--.. 3*-{Y&'' MrrUcddlmfd|D}dt||D}tj|di}t |t r |djn|jfdt||D}fdt||D}t|dkr|n|dS) aIBroadcast arrays to a common shape. Like `numpy.broadcast_arrays`, applied to both unmasked data and masks. Note that ``subok`` is taken to mean whether or not subclasses of the unmasked data and masks are allowed, i.e., for ``subok=False``, `~astropy.utils.masked.MaskedNDArray` instances will be returned. rrc0g|]}t|Sr)rw)rargr s rrz$broadcast_arrays..s#:::c*S&)):::rc*g|]\}}|r|jn|Sr)r2)rr is_maskeds rrz$broadcast_arrays..s4   1?i +   rrUrcVg|]%\}}|rtj|jnd&S)rN)rrXr3)rrrr$rUs rrz$broadcast_arrays.. sK    C;D M5 6 6 6 6   rc6g|]\}}| ||n|Sr"r)rrqr3r s rrz$broadcast_arrays.. sC VT"&!1   vr) r)r r,rbroadcast_arraysrwlistr$len)rUr- are_maskedr.resultsr/r r$s` @@rrrs::::T:::J  CFtZCXCX   D!45u55G *7D 9 9 LGAJ  w}E     !$ 33   E!'511G'llQ&&77GAJ6rcddlm}t||st||stt ||\\}}\}}||||f||||fidfS)zInsert values along the given axis before the given indices. Like `numpy.insert` but for possibly masked ``arr`` and ``values``. Masked ``obj`` is not supported. rrN)rvr rwrxr*) r5objrrr arr_dataval_dataarr_maskval_masks rinsertrs,+++++#v"jf&=&="!!1DS&1Q1Q.Xx.8X sHd +hXt-LbRV WWrkeepdimsc|tjd|j}tj|||S)zCounts the number of non-zero values in the array ``a``. Like `numpy.count_nonzero`, with masked values counted as 0 or `False`. rr)filledrr#rb count_nonzero)rIrrrs rrr$s; XXbhr17++ , ,F FD8 < < <++++++!V" 38O8O!!q A aA 6  2 21 5 5B@AAA  ::j%00v#,  $#  AG  1?Bk(9 !'A+&&&sRv#,  $#  AG   rcNtj|d}t||g|Ri|S)Nd)r true_divider)rIrr-r6s r percentiler s3 q#A Aq *4 * * *6 * **rct||\\}}\}}|j|jkrdS||k}|r,|tj|tj|zz}t ||z|zS)NF)r*r$risnanr%r)a1a2 equal_nana1da2da1ma2mequals r array_equalrs0R88JS# c yCIu 3JE/ ##.. s"'')) * **rcLt||kSr")r%r)rrs r array_equivrs r  ! !!rcddlm}|s|ddfS||\}}t |\}}t j|g|R}t j|g|R}|||z}|||S)Nrrr)rvr nonzeroryr*rr) conditionr-r c_maskr.r/r2r3s rrrs++++++ /  ""D$..11)<<Iv%t,KD%x )D)))H 8I & & & &D   6( & & &&rcrddlm}||\}}||dkr|d}d|i}|!t ||st |j|d<t|\}} tj ||fi|} | |j |d<tj || fi|} || |z} | || | n|S)aAConstruct an array from an index array and a set of arrays to choose from. Like `numpy.choose`. Masked indices in ``a`` will lead to masked output values and underlying data values are ignored if out of bounds (for ``mode='raise'``). Any values masked in ``choices`` will be propagated if chosen. rrNrs)rprurZ) rvr ryrrwrxr2r*rrr3) rIchoicesrZrur a_dataa_maskr6r.r/ data_chosen mask_chosens rrrs,+++++..q11NFF dgooQ''d^F #v&& &% % u %w/KD%)FD33F33K u )FE44V44K v /2{66+{ + + +Crcddlmfd|D}t|\}}|tjjur |n dd}|||jf|||jfidfS)zReturn an array drawn from elements in choicelist, depending on conditions. Like `numpy.select`, with masks in ``choicelist`` are propagated. Any masks in ``condlist`` are ignored. rrcBg|]}t|r|jn|Sr)rwr2)rcr s rrzselect..s.MMMqjF33: MMMrTrN)rvr r*rrgmaskedr2r3)condlist choicelistdefault data_list mask_listr s @rselectr-s,+++++MMMMHMMMH. ;Iy!( !8X"6Q??? Q b RA R RQ R RAE R R R   aA E D 1XX 5 5{~~ 5 LL! % % % KK    Xa[>Dy1}} Xa[))) DD3333334445$'' e$ Hrcddlm}||\}}t||st||rJt ||\\}}\}} |j|jcxkrdkrnn|| z} || }|| }t j|||g|Ri|} || n|| |S)zOne-dimensional linear interpolation. Like `numpy.interp`, but any masked points in ``xp`` and ``fp`` are ignored. Any masked values in ``x`` will still be evaluated, but masked on output. rrr) rvr ryrwr*rrinterpr4) r:xpfpr-r6r xdxmxpmfpmr'rqs rr?r?#s,+++++  & &q ) )FB"fB!7!722r::R*3 7bg " " " " " " " " "c AQBBQBB Yr2r 3D 3 3 3F 3 3FZ66VVFBGGII%>%>>rcbddlm}g}|D]}t||rh|j}|rB|jr)|}|jd||j<|||jgz| |tj ||S)zPerform an indirect stable sort using a sequence of keys. Like `numpy.lexsort` but for possibly masked ``keys``. Masked values are sorted towards the end for each key. rrrr) r)r rwr2r3r1r4flatextendrrlexsort)keysrr new_keyskeynew_keys rrJrJ:sH ! ! c6 " " !lG 4CHLLNN 4!,,..$+LO! OOWch/ 0 0 0 0 OOC :hT * * **rc|}|j}tj|jdkr|f}|D]_}|dkr||z}||f}||}|j|jkr|})tj||}|j|jkr|}Qt d|S)Nrz7function is not returning an array of the correct shape)rrrR expand_dimsr)rrIaxesvalNrr-ress rapply_over_axesrUTs C A x~~aw !88t8DT{dDk 8sx  CC.d++Cx38## M Jrc4eZdZdZdZdZedZdS) MaskedFormatzFormatter for masked array scalars. For use in `numpy.array2string`, wrapping the regular formatters such that if a value is masked, its formatted string is replaced. Typically initialized using the ``from_data`` class method. c||_t|dd}|rd|D|j_t|dd}|rt||j_dSdS)Nformat_functionsc,g|]}t|Sr)rW)rffs rrz)MaskedFormat.__init__..s 4T4T4T"\"5E5E4T4T4Trformat_function)r\rrYrW)selfr\ffsr[s r__init__zMaskedFormat.__init__zs.o'94@@  U4T4TPS4T4T4TD 1 _&7 > >  D3?3C3CD 0 0 0 D Drc jjr+|fdjjDSjr|S|jd}jrFt dtdt|}dt||z zd|zzS|S)Nc g|] }| Srr)rfieldr:s rrz)MaskedFormat.__call__..s(M(M(Me5(M(M(Mrrr u—) rbrfr\r$r2r3minmaxr)r]r:stringr7s ` r__call__zMaskedFormat.__call__s 7= O''(M(M(M(Mqw}(M(M(MNN N 7 +''** *%%ajn55 6  As1c&kk**++A#f++/*X\9 9Mrc 2ddlm}|||fi|S)Nr)_get_format_function)numpy.core.arrayprintrj)clsr.optionsrjs r from_datazMaskedFormat.from_datas8>>>>>>s''8888999rN)__name__ __module__ __qualname____doc__r_rh classmethodrnrrrrWrWqsZ D D D2::[:::rrWrdc (ddlm}m}tj|}|j|dkrd}|||d}nd}t j|fi|}d} | dt|zz } ||||d| ||d||d } | S) Nr) _formatArray_leading_trailing thresholdz... edgeitemsrtrd linewidthlegacy) rkrvrwrasarrayrrWrnr) rIrm separatorprefixrvrwr.summary_insertr\next_line_prefixlsts r _array2stringrsFEEEEEEE :a==Dv $$$  w{';<<#,T==W==Oc&kk)) ,     C Jrc ddlm} m}|||| ||dd| || }| }|||dxxt | zcc<|jdkrdSt||||S)Nr)_format_options_make_options_dictrzz[])rkrrr4updaterrr)rImax_line_width precisionsuppress_smallr}r~style formatterrxrysign floatmodesuffixrr overridesrms r array2stringrs"JIIIIIII""     I""$$G NN9 KCKK' v{{t GY 7 77rc*t||||ddS)Nrdrt)r)rIrrrs r array_strrs >9nc2 N NNrr)nansum nancumsumnanprod nancumprodctt|ddt|dfd}d|d} |ddz }nd|vr|dz }n|d z }||_||_|S) Nrcc$ddlm}||\}}t|jjt jrt j|}||n||z}|#|||}| }|g|Ri|S)Nrr) rvr ryrrbrrinexactrr)rIr-r6r r3nansrpnp_funcs rnanfunczmasked_nanfunc..nanfuncs//////++A..4 aglBJ / / ;8A;;D<44dTkD  q$A%HHZ((wq*4***6***rz Like `numpy.z$`, skipping masked values as well. z$Masked/NaN values are replaced with z. The output is not masked.rzRNo exceptions are raised for fully masked/NaN slices. Instead, these give index 0.zZNo warnings are given for fully masked/NaN slices. Instead, they are masked in the output.)rr_nanfunc_fill_valuesgetrrro) nanfuncnamerdocrprs @@rmasked_nanfuncrsb+abb/**G%))+t<rEr4asfarrayrrrollaxisrollrJflipflipudfliplrrot90triutrilrLdiagdiagflatrPdeleterSrXr\rerkrnrrr r~rrrrrrrrrrrrrrrrrrrr rrrrr-r=r?rJrUrWr_NoValuerrrr nanfunctionsrrglobalssortedrrrrrs   KKKKKKKKKKKKKKKK     . #+ &(&8:<. M K!lJ  ^.0[M=#%-24<>9GIiJ&([35(=?K.130D !$O%$\%$$&9%&N'&H'&!i'&*,'&57I'(N)(')(*,)(35')*+*)+.O/.\/.$&7/.-/I/.8:w/010n14F54 K5458H98e9@GRXrz2#6GRZRZRZJ BJ") FBOR[L"/2#3FBKL"+L"."."2HFBGRXr|RXIr~  .0f.@.H   F**! !899 &&';<<   YYY444999GGG6666 7BKBKbg N222 bgry")RXrwPQQQ22RQ2 bgr{+,,,22-,2 bi[!!!22"!2  < < < <@@@@    *!!!!!!!!    ++++$***"((("3333& 2    $.... 0"&777778 X X X X=E=====TTT        * O O O$    .+++ ++++""" ' ' ' DDDDD*. . . b???,++++285:5:5:5:5:5:5:5:p    F +  '8'8'8'8TOOOO #$!QRSS(((V6&.K00{##772{+C+CGGIIk66  # * * , ,--4H4O4O4Q4Q0R0RR  r