U e{jR @sRdZddlZddlmZmZddddd d d d d dddg Zd*ddZd+ddZGdddeZ Gddde Z Gddde Z Gddde Z Gdd d e Z Gddde ZGdd d e ZGdd d e ZGdd d e ZGd!d d e ZGd"d d e ZGd#dde ZGd$d%d%e ZGd&dde ZGd'd(d(e ZGd)ddee 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 HistEqStretchContrastBiasStretchCompositeStretchcCsH|dkrt|t|Stj||dtj|t||d|SdS)zCalculate the log base n of x.Nout)nplog true_divide)nxrr\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/visualization/stretch.py_logns rTcCsF|rtj|dd|dS|dkr,tj|ddSt||dd<|SdS)z Prepare the data by optionally clipping and copying, and return the array that should be subsequently used for in-place calculations. ?rNT)copy)rcliparrayZasarray)valuesrrrrr_prepare's r!c@s:eZdZdZeddZddZd dd Zed d ZdS) 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]. cCsdS)NFrselfrrr_supports_invalid_kw>sz BaseStretch._supports_invalid_kwcCs t||SN)r)r#otherrrr__add__BszBaseStretch.__add__TNcCsdS)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. Nrr#r rrrrr__call__EszBaseStretch.__call__cCsdS)5A stretch object that performs the inverse operation.Nrr"rrrinverse[szBaseStretch.inverse)TN) __name__ __module__ __qualname____doc__propertyr$r'r)r+rrrrr7s  cs8eZdZdZd fdd Zd dd Zed d ZZS)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. rrcst||_||_dSr%)super__init__slope intercept)r#r3r4 __class__rrr2qs zLinearStretch.__init__TNcCsJt|||d}|jdkr*tj||j|d|jdkrFtj||j|d|S)Nrrrrr)r!r3rmultiplyr4addr(rrrr)vs   zLinearStretch.__call__cCstd|j|j |jSr*r)rr3r4r"rrrr+~szLinearStretch.inverse)rr)TN r,r-r.r/r2r)r0r+ __classcell__rrr5rr`s  c@s2eZdZdZeddZd ddZedd ZdS) rzb A square root stretch. The stretch is given by: .. math:: y = \sqrt{x} cCsdSNTrr"rrrr$sz SqrtStretch._supports_invalid_kwTNc Cs^t|||d}| o|dk }tjdd |r6|dk}tj||dW5QRX|rZ|||<|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. r7Nignoreinvalidrr)r!rerrstatesqrtr#r rrrAZreplace_invalididxrrrr)szSqrtStretch.__call__cCstdS)r*)rr"rrrr+szSqrtStretch.inverse)TNN)r,r-r.r/r0r$r)r+rrrrrs    ,csBeZdZdZeddZfddZd dd Zed d ZZ S) rz 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. cCsdSr=rr"rrrr$sz!PowerStretch._supports_invalid_kwcs$t|dkrtd||_dSNrz a must be > 0)r1r2 ValueErrorpowerr#ar5rrr2s zPowerStretch.__init__TNc Cst|||d}| oN|dk oNd|jko0dknpNd|jkoJdkn}tjdd$|rj|dk}tj||j|dW5QRX|r|||<|S) r>r7Nrrr?r@r)r!rIrrBrDrrrr)s2zPowerStretch.__call__cCstd|jSr:)rrIr"rrrr+szPowerStretch.inverse)TNN r,r-r.r/r0r$r2r)r+r<rrr5rrs   0cs8eZdZdZd fdd Zd ddZed d ZZS) 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. @@cs,|dks|dkrtdt||_dSNrrz&a must be >= 0, but cannot be set to 1rHr1r2exprJr5rrr2%s zPowerDistStretch.__init__TNcCsJt|||d}tj|j||dtj|d|dtj||jd|d|S)Nr7rrr)r!rrIrQsubtractrr(rrrr)+s zPowerDistStretch.__call__cCs t|jdSr*rK)InvertedPowerDistStretchrQr"rrrr+2szPowerDistStretch.inverse)rN)TNr;rrr5rr s  cs8eZdZdZd fdd Zd ddZed d ZZS) rUa 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. rNcs,|dks|dkrtdt||_dSrOrPrJr5rrr2Js z!InvertedPowerDistStretch.__init__TNcCsHt|||d}tj||jd|dtj|d|dt|j||d|S)Nr7rrr)r!rr8rQr9rr(rrrr)Ps z!InvertedPowerDistStretch.__call__cCs t|jdSrS)r rQr"rrrr+Wsz InvertedPowerDistStretch.inverse)rN)TNr;rrr5rrU8s  rUcs,eZdZdZfddZeddZZS)r zt A convenience class for a power stretch of 2. The stretch is given by: .. math:: y = x^2 cstddS)NrF)r1r2r"r5rrr2gszSquaredStretch.__init__cCstSr*)rr"rrrr+jszSquaredStretch.inverse)r,r-r.r/r2r0r+r<rrr5rr ]s csDeZdZdZeddZd fdd Zdd d Zed d ZZ 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. cCsdSr=rr"rrrr$szLogStretch._supports_invalid_kwrNcs$t|dkrtd||_dSrGr1r2rHrQrJr5rrr2s zLogStretch.__init__TNc Cst|||d}| o|dk }tjdd^|r6|dk}tj||j|dtj|d|dtj||dtj|t|jd|dW5QRX|r|||<|S)r>r7Nr?r@rrr)r!rrBr8rQr9rrrDrrrr)s&zLogStretch.__call__cCs t|jSrV)InvertedLogStretchrQr"rrrr+szLogStretch.inverse)rN)TNNrMrrr5rr ps  /cs6eZdZdZfddZd ddZedd ZZS) rXat 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. cs$t|dkrtd||_dSrGrWrJr5rrr2s zInvertedLogStretch.__init__TNcCs^t|||d}tj|t|jd|dtj||dtj|d|dtj||j|d|S)Nr7rr)r!rr8rrQrRrr(rrrr)s zInvertedLogStretch.__call__cCs t|jSrV)r rQr"rrrr+szInvertedLogStretch.inverse)TNr;rrr5rrXs   rXcs8eZdZdZd fdd Zd ddZed d ZZS) 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. 皙?cs,t|dks|dkr"td||_dSNrrza must be > 0 and <= 1r1r2rHrKrJr5rrr2s zAsinhStretch.__init__TNcCsNt|||d}tj||j|dtj||dtj|td|j|d|SNr7rr)r!rrrKarcsinhr(rrrr)s zAsinhStretch.__call__cCstdtd|jdSr*rrT)r rr]rKr"rrrr+szAsinhStretch.inverse)rY)TNr;rrr5rr s  cs8eZdZdZd fdd Zd ddZed d ZZS) r aT A sinh stretch. The stretch is given by: .. math:: y = \frac{{\rm sinh}(x / a)}{{\rm sinh}(1 / a)} Parameters ---------- a : float, optional The ``a`` parameter used in the above formula. ``a`` must be greater than 0 and less than or equal to 1 (0 < a <= 1). Default is 1/3. UUUUUU?cs,t|dks|dkr"td||_dSrZr[rJr5rrr2s zSinhStretch.__init__TNcCsNt|||d}tj||j|dtj||dtj|td|j|d|Sr\)r!rrrKsinhr(rrrr)!s zSinhStretch.__call__cCstdtd|jdSr^)r rr`rKr"rrrr+(szSinhStretch.inverse)r_)TNr;rrr5rr s  c@s0eZdZdZd ddZd ddZedd ZdS) ra  A histogram equalization stretch. 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. NcCsvt||_|jt|j|_|j}|j}|j||||_|dkrltddt|j|_ n||_ dSNrr) rsortZraveldataisfiniteminmaxlinspacelenr )r#rcr ZvminZvmaxrrrr2;s  zHistEqStretch.__init__TcCs,t|||d}t||j|j|dd<|SNr7)r!rinterprcr r(rrrr)IszHistEqStretch.__call__cCst|j|jdSr*)r )InvertedHistEqStretchrcr r"rrrr+NszHistEqStretch.inverse)N)TNr,r-r.r/r2r)r0r+rrrrr.s   c@s0eZdZdZd ddZd ddZedd ZdS) rla. 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. NcCs:|t||_|dkr0tddt|j|_n||_dSra)rrdrcrgrhr )r#rcr rrrr2aszInvertedHistEqStretch.__init__TcCs,t|||d}t||j|j|dd<|Sri)r!rrjr rcr(rrrr)hszInvertedHistEqStretch.__call__cCst|j|jdSrk)rrcr r"rrrr+mszInvertedHistEqStretch.inverse)N)TNrmrrrrrlTs   rlcs6eZdZdZfddZd ddZedd ZZS) ra A stretch that takes into account contrast and bias. The stretch is given by: .. math:: y = (x - {\rm bias}) * {\rm contrast} + 0.5 and the output values are clipped to the [0:1] range. Parameters ---------- contrast : float The contrast parameter (see the above formula). bias : float The bias parameter (see the above formula). cst||_||_dSr%r1r2contrastbiasr#rorpr5rrr2s zContrastBiasStretch.__init__TNcCs\t|d|d}tj||j|dtj||j|dtj|d|d|rXtj|dd|d|S)NFr7r?rr)r!rrRrpr8ror9rr(rrrr)szContrastBiasStretch.__call__cCst|j|jSrV)InvertedContrastBiasStretchrorpr"rrrr+szContrastBiasStretch.inverse)TNr;rrr5rrss   cs6eZdZdZfddZd ddZedd ZZS) rsa1 Inverse transformation for ContrastBiasStretch. Parameters ---------- contrast : float The contrast parameter (see `~astropy.visualization.ConstrastBiasStretch). bias : float The bias parameter (see `~astropy.visualization.ConstrastBiasStretch). cst||_||_dSr%rnrqr5rrr2s z$InvertedContrastBiasStretch.__init__TNcCs\t|d|d}tj|d|dtj||j|dtj||j|d|rXtj|dd|d|S)NFr7rrrrr)r!rrRrror9rprr(rrrr)sz$InvertedContrastBiasStretch.__call__cCst|j|jSrV)rrorpr"rrrr+sz#InvertedContrastBiasStretch.inverse)TNr;rrr5rrss   rsc@seZdZdZdddZdS)ra A combination of two stretches. Parameters ---------- stretch_1 : :class:`astropy.visualization.BaseStretch` The first stretch to apply. stretch_2 : :class:`astropy.visualization.BaseStretch` The second stretch to apply. TNcCs|j|j|||d||dSri)Z transform_2Z transform_1r(rrrr)s zCompositeStretch.__call__)TN)r,r-r.r/r)rrrrrs )N)TN)r/numpyrZ transformrr__all__rr!rrrrr rUr r rXr r rrlrrsrrrrrsB )$@P$%O%&$&-'