IR-e`ddlZddlmZddlmZddlmZddgZ d d Zdd Z dS)N)log)getdata) simple_norm fits2bitmapmainlinear?皙?Greys_rc ddl} ddlm} ddlm} t |}n#t $rYnwxYw t||}n,#t$r}tj |Yd}~dSd}~wwxYw|j dkrtj d|d|dtj |d}|dr%tj |d}|d z }tj |ddd} || d r| j| ndd lm}|| n/#t t&f$rtj | d YdSwxYwt)|||||||| | }| |||| d|tjd|ddS)a Create a bitmap file from a FITS image, applying a stretching transform between minimum and maximum cut levels and a matplotlib colormap. Parameters ---------- filename : str The filename of the FITS file. ext : int FITS extension name or number of the image to convert. The default is 0. out_fn : str The filename of the output bitmap image. The type of bitmap is determined by the filename extension (e.g. '.jpg', '.png'). The default is a PNG file with the same name as the FITS file. stretch : {'linear', 'sqrt', 'power', log', 'asinh'} The stretching function to apply to the image. The default is 'linear'. power : float, optional The power index for ``stretch='power'``. The default is 1.0. asinh_a : float, optional For ``stretch='asinh'``, the value where the asinh curve transitions from linear to logarithmic behavior, expressed as a fraction of the normalized image. Must be in the range between 0 and 1. The default is 0.1. min_cut : float, optional The pixel value of the minimum cut level. Data values less than ``min_cut`` will set to ``min_cut`` before stretching the image. The default is the image minimum. ``min_cut`` overrides ``min_percent``. max_cut : float, optional The pixel value of the maximum cut level. Data values greater than ``min_cut`` will set to ``min_cut`` before stretching the image. The default is the image maximum. ``max_cut`` overrides ``max_percent``. min_percent : float, optional The percentile value used to determine the pixel value of minimum cut level. The default is 0.0. ``min_percent`` overrides ``percent``. max_percent : float, optional The percentile value used to determine the pixel value of maximum cut level. The default is 100.0. ``max_percent`` overrides ``percent``. percent : float, optional The percentage of the image values used to determine the pixel values of the minimum and maximum cut levels. The lower cut level will set at the ``(100 - percent) / 2`` percentile, while the upper cut level will be set at the ``(100 + percent) / 2`` percentile. The default is 100.0. ``percent`` is ignored if either ``min_percent`` or ``max_percent`` is input. cmap : str The matplotlib color map name. The default is 'Greys_r'. rN) minversionzdata in FITS extension z is not a 2D arrayz.fitsz.pngz3.5)cmz) is not a valid matplotlib colormap name.)stretchpowerasinh_amin_cutmax_cut min_percent max_percentpercentlower)cmaporiginformatzSaved file to .) matplotlibmatplotlib.imageimageastropy.utils.introspectionr int ValueErrorr Exceptionrcriticalndimospathsplitextendswith colormapsrget_cmapKeyErrorrimsaveinfo)filenameextout_fnrrrrrrrrrrmimgr r e out_formatrnorms Ilib/python3.11/site-packages/astropy/visualization/scripts/fits2bitmap.pyrr scH######666666 #hh      #&&  Qqqqqq zQ FsFFFGGG ~!!(++A. ??7 # # 1W%%f--a0F&!!&))!,QRR0J  :j% ( (    & & & % % % % % % KK     ! GGGHHHqq     D KKU $wzKRRRH 'f ' ' '(((((s5" //A A-A((A-'5E(F F cddl}|d}|ddddd|d d tdd |d tdd|dtdd|dtdd|dtdd|dtdd|dtdd|dtdd|dtdd |d!d"td#d$ |d d%d&'||}|jD]T}t||j|j |j |j |j |j |j|j|j|j|j( UdS))Nrz'Create a bitmap file from a FITS image.) descriptionz-ez--exthduz8Specify the HDU extension number or name (Default is 0).)metavardefaulthelpz-or0zZFilename for the output image (Default is a PNG file with the same name as the FITS file).)r;typer<r=z --stretchrz^Type of image stretching ("linear", "sqrt", "power", "log", or "asinh") (Default is "linear").)r>r<r=z--powerr z4Power index for "power" stretching (Default is 1.0).z --asinh_ar zThe value in normalized image where the asinh curve transitions from linear to logarithmic behavior (used only for "asinh" stretch) (Default is 0.1).z --min_cutzHThe pixel value of the minimum cut level (Default is the image minimum).z --max_cutzHThe pixel value of the maximum cut level (Default is the image maximum).z --min_percentzLThe percentile value used to determine the minimum cut level (Default is 0).z --max_percentzNThe percentile value used to determine the maximum cut level (Default is 100).z --percentz}The percentage of the image values used to determine the pixel values of the minimum and maximum cut levels (Default is 100).z--cmap colormap_namer z1matplotlib color map name (Default is "Greys_r").+z)Path to one or more FITS files to convert)nargsr=) r1r2rrrrrrrrr)argparseArgumentParser add_argumentstrfloat parse_argsr0rr1orrrrrrrrr)argsrBparserr0s r7rrsOOO  $ $=%F   G     =       A    C           W     W     0    2    3       @  #$O   T " "DM   6LLL((L*L   ) rNrr r NNNNNr )N) r'astropyrastropy.io.fitsr#astropy.visualization.mpl_normalizer__all__rrrKr7rQs ######;;;;;; & !       y)y)y)y)xs s s s s s rK