IR-eRdZddlZddlmZmZgdZd)dZd*dZGd d eZ Gd d e Z Gd de Z Gdde Z Gdde Z Gdde ZGdde ZGdde ZGdde ZGdde ZGdde ZGdd e ZGd!d"e ZGd#d$e ZGd%d&e ZGd'd(ee ZdS)+z Classes that deal with stretching, i.e. mapping a range of [0:1] values onto another set of [0:1] values with a transformation. N) BaseTransformCompositeTransform) BaseStretch LinearStretch SqrtStretch PowerStretchPowerDistStretchSquaredStretch LogStretch AsinhStretch SinhStretch HistEqStretchContrastBiasStretchCompositeStretchc|)tj|tj|z Stj||tj|tj|||S)zCalculate the log base n of x.Nout)nplog true_divide)nxrs =lib/python3.11/site-packages/astropy/visualization/stretch.py_lognrs] {vayy26!99$$ qc sBF1II3//// Tc|rtj|dd|S|tj|dStj||dd<|S)z Prepare the data by optionally clipping and copying, and return the array that should be subsequently used for in-place calculations. ?rNT)copy)rcliparrayasarray)valuesr!rs r_preparer%'s[  wvsCS1111 ;8F... .Z''CFJrcLeZdZdZedZdZddZedZdS) rz Base class for the stretch classes, which, when called with an array of values in the range [0:1], return an transformed array of values, also in the range [0:1]. cdS)NFselfs r_supports_invalid_kwz BaseStretch._supports_invalid_kw=surc"t||SN)r)r*others r__add__zBaseStretch.__add__Ast,,,rTNcdS)ac Transform values using this stretch. Parameters ---------- values : array-like The input values, which should already be normalized to the [0:1] range. clip : bool, optional If `True` (default), values outside the [0:1] range are clipped to the [0:1] range. out : ndarray, optional If specified, the output values will be placed in this array (typically used for in-place calculations). Returns ------- result : ndarray The transformed values. Nr(r*r$r!rs r__call__zBaseStretch.__call__DrcdS)5A stretch object that performs the inverse operation.Nr(r)s rinversezBaseStretch.inverseZr3rTN) __name__ __module__ __qualname____doc__propertyr+r/r2r6r(rrrr6s| X---    ,DDXDDDrrcBeZdZdZd fd Zd dZedZxZS) ras A linear stretch with a slope and offset. The stretch is given by: .. math:: y = slope x + intercept Parameters ---------- slope : float, optional The ``slope`` parameter used in the above formula. Default is 1. intercept : float, optional The ``intercept`` parameter used in the above formula. Default is 0. rrcdt||_||_dSr-)super__init__slope intercept)r*rArB __class__s rr@zLinearStretch.__init__ps+  "rTNct|||}|jdkrtj||j||jdkrtj||j||S)Nr!rrrr)r%rArmultiplyrBaddr1s rr2zLinearStretch.__call__use&t555 :?? K  7 7 7 7 >Q   F64>v 6 6 6 6 rcNtd|jz |j |jz Sr5r)rrArBr)s rr6zLinearStretch.inverse}s&S4:-/KLLLr)rrr7 r8r9r:r;r@r2r<r6 __classcell__rCs@rrr_s} ###### MMXMMMMMrrcFeZdZdZedZddZedZdS)rzb A square root stretch. The stretch is given by: .. math:: y = \sqrt{x} cdSNTr(r)s rr+z SqrtStretch._supports_invalid_kwtrTNct|||}| o|du}tjd5|r|dk}tj||dddn #1swxYwY|r|||<|S)Z Transform values using this stretch. Parameters ---------- values : array-like The input values, which should already be normalized to the [0:1] range. clip : bool, optional If `True` (default), values outside the [0:1] range are clipped to the [0:1] range. out : ndarray, optional If specified, the output values will be placed in this array (typically used for in-place calculations). invalid : None or float, optional Value to assign NaN values generated by this class. NaNs in the input ``values`` array are not changed. This option is generally used with matplotlib normalization classes, where the ``invalid`` value should map to the matplotlib colormap "under" value (i.e., any finite value < 0). If `None`, then NaN values are not replaced. This keyword has no effect if ``clip=True``. Returns ------- result : ndarray The transformed values. rENignoreinvalidrr)r%rerrstatesqrtr*r$r!rrUreplace_invalididxs rr2zSqrtStretch.__call__s:&t555"(:wd': [ * * * ( ( !qj GF ' ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (  ""F3K sAA!Ac tdS)r5)r r)s rr6zSqrtStretch.inversesArTNN)r8r9r:r;r<r+r2r6r(rrrrsiX))))VXrrcVeZdZdZedZfdZddZedZxZ S) r z A power stretch. The stretch is given by: .. math:: y = x^a Parameters ---------- a : float The power index (see the above formula). ``a`` must be greater than 0. cdSrOr(r)s rr+z!PowerStretch._supports_invalid_kwrPrct|dkrtd||_dSNrz a must be > 0)r?r@ ValueErrorpowerr*arCs rr@zPowerStretch.__init__s;  66_-- - rTNc<t|||}| o-|duo)d|jcxkodkncpd|jcxkodknc}tjd5|r|dk}tj||j|dddn #1swxYwY|r|||<|S) rRrENrrrSrTr)r%rcrrVrXs rr2zPowerStretch.__call__s(:&t555H @t# @tz%%%%A%%%%>1tz+=+=+=+=A+=+=+=+=  [ * * * 5 5 !qj HVTZV 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5  ""F3K s%B  BBc0td|jz SrI)r rcr)s rr6zPowerStretch.inverse sC$*,---rr] r8r9r:r;r<r+r@r2r6rKrLs@rr r s  X ----^..X.....rr cBeZdZdZdfd Zd dZedZxZS) r aI An alternative power stretch. The stretch is given by: .. math:: y = \frac{a^x - 1}{a - 1} Parameters ---------- a : float, optional The ``a`` parameter used in the above formula. ``a`` must be greater than or equal to 0, but cannot be set to 1. Default is 1000. @@c|dks|dkrtdt||_dSNrrz&a must be >= 0, but cannot be set to 1rbr?r@exprds rr@zPowerDistStretch.__init__"D q55AFFEFF F rTNct|||}tj|j||tj|d|tj||jdz ||S)NrErrr)r%rrcrosubtractrr1s rr2zPowerDistStretch.__call__(sh&t555 6v.... FA6**** vtx#~6:::: rc,t|jSr5re)InvertedPowerDistStretchror)s rr6zPowerDistStretch.inverse/s($(3333rrkr7rJrLs@rr r sv  44X44444rr cBeZdZdZdfd Zd dZedZxZS) rva Inverse transformation for `~astropy.image.scaling.PowerDistStretch`. The stretch is given by: .. math:: y = \frac{\log(y (a-1) + 1)}{\log a} Parameters ---------- a : float, optional The ``a`` parameter used in the above formula. ``a`` must be greater than or equal to 0, but cannot be set to 1. Default is 1000. rkc|dks|dkrtdt||_dSrmrnrds rr@z!InvertedPowerDistStretch.__init__GrprTNct|||}tj||jdz |tj|d|t |j|||S)NrErrr)r%rrFrorGrr1s rr2z!InvertedPowerDistStretch.__call__Msf&t555 FDHsN7777 vqf%%%% dhF++++ rc,t|jSrt)r ror)s rr6z InvertedPowerDistStretch.inverseTs $(++++rrwr7rJrLs@rrvrv5sv" ,,X,,,,,rrvc8eZdZdZfdZedZxZS)r zt A convenience class for a power stretch of 2. The stretch is given by: .. math:: y = x^2 cJtddS)Nr\)r?r@)r*rCs rr@zSquaredStretch.__init__ds! rctSr5)rr)s rr6zSquaredStretch.inversegs}}r)r8r9r:r;r@r<r6rKrLs@rr r Zs]Xrr cXeZdZdZedZd fd Zd dZedZxZ S) r a A log stretch. The stretch is given by: .. math:: y = \frac{\log{(a x + 1)}}{\log{(a + 1)}} Parameters ---------- a : float The ``a`` parameter used in the above formula. ``a`` must be greater than 0. Default is 1000. cdSrOr(r)s rr+zLogStretch._supports_invalid_kw}rPrrkct|dkrtd||_dSrar?r@rbrords rr@zLogStretch.__init__;  66_-- -rTNct|||}| o|du}tjd5|r|dk}tj||j|tj|d|tj||tj|tj|jdz|dddn #1swxYwY|r|||<|S)rRrENrSrTrrr)r%rrVrFrorGrrrXs rr2zLogStretch.__call__s9:&t555"(:wd': [ * * * G G !qj Kf 5 5 5 5 F63F + + + + F6v & & & & N626$(S.#9#9v F F F F  G G G G G G G G G G G G G G G  ""F3K sBB>>CCc*t|jSr)InvertedLogStretchror)s rr6zLogStretch.inverses"$(+++rrwr]rirLs@rr r ms  X ,,,,\,,X,,,,,rr c@eZdZdZfdZddZedZxZS)rat Inverse transformation for `~astropy.image.scaling.LogStretch`. The stretch is given by: .. math:: y = \frac{e^{y \log{a + 1}} - 1}{a} \\ y = \frac{e^{y} (a + 1) - 1}{a} Parameters ---------- a : float, optional The ``a`` parameter used in the above formula. ``a`` must be greater than 0. Default is 1000. ct|dkrtd||_dSrarrds rr@zInvertedLogStretch.__init__rrTNct|||}tj|tj|jdz|tj||tj|d|tj||j||S)NrErr)r%rrFrrorrrr1s rr2zInvertedLogStretch.__call__s&t555 FBF48c>22???? v6"""" FCV,,,, vtxV4444 rc*t|jSr)r ror)s rr6zInvertedLogStretch.inverses$(###rr7rJrLs@rrrsq  $$X$$$$$rrcBeZdZdZdfd Zd dZedZxZS) r a An asinh stretch. The stretch is given by: .. math:: y = \frac{{\rm asinh}(x / a)}{{\rm asinh}(1 / a)}. Parameters ---------- a : float, optional The ``a`` parameter used in the above formula. The value of this parameter is where the asinh curve transitions from linear to logarithmic behavior, expressed as a fraction of the normalized image. ``a`` must be greater than 0 and less than or equal to 1 (0 < a <= 1). Default is 0.1. 皙?ct|dks|dkrtd||_dSNrrza must be > 0 and <= 1r?r@rbrerds rr@zAsinhStretch.__init__D  66QUU566 6rTNct|||}tj||j|tj||tj|tjd|jz ||SNrErr)r%rrrearcsinhr1s rr2zAsinhStretch.__call__sp&t555 vtv62222 6v&&&& vrz#,77VDDDD rc\tdtjd|jz z Sr5rru)rrrrer)s rr6zAsinhStretch.inverses*S2:cDFl#;#;;<<<+c3DI??DKKK DKKKrTctt|||}tj||j|j|dd<|SNrE)r%rinterprr$r1s rr2zHistEqStretch.__call__Es:&t555Ifdi==qqq  rc8t|j|jSr5)r$)InvertedHistEqStretchrr$r)s rr6zHistEqStretch.inverseJs%TYt{CCCCrr-r7r8r9r:r;r@r2r<r6r(rrrr*sf   ! ! ! ! DDXDDDrrc8eZdZdZddZddZedZdS) ra. Inverse transformation for `~astropy.image.scaling.HistEqStretch`. Parameters ---------- data : array-like The data defining the equalization. values : array-like, optional The input image values, which should already be normalized to the [0:1] range. Nc|tj||_|/tjddt |j|_dS||_dSr)rrrrrr$)r*rr$s rr@zInvertedHistEqStretch.__init__]sIT**+ >+c3DI??DKKK DKKKrTctt|||}tj||j|j|dd<|Sr)r%rrr$rr1s rr2zInvertedHistEqStretch.__call__ds:&t555Ifdk49==qqq  rc8t|j|jSr)rrr$r)s rr6zInvertedHistEqStretch.inverseisTYt{;;;;rr-r7rr(rrrrPsa  !!!! <<X<<rsX 88888888        &D&D&D&D&D-&D&D&DR!M!M!M!M!MK!M!M!MH<<<<<+<<<~L.L.L.L.L.;L.L.L.^!4!4!4!4!4{!4!4!4H",",",",",{",",",J\&K,K,K,K,K,K,K,K,\"$"$"$"$"$"$"$"$J#=#=#=#=#=;#=#=#=L!;!;!;!;!;+!;!;!;H#D#D#D#D#DK#D#D#DL<<<<*E*E*E*E*E+*E*E*EZ$=$=$=$=$=+$=$=$=N     );     r