U e{j @sLddlZddlmZddlmZddlmZddgZd d dZdd dZ dS)N)log)getdata) simple_norm fits2bitmapmainlinear?皙?Greys_rc  Csddl} ddlm} ddlm}z t|}Wntk r@YnXzt||}Wn2tk r}zt |WYdSd}~XYnX|j dkrt d|d|dkrt j |d}|drt j |d}|d 7}t j |ddd}z2|| d r | j| ndd lm}|| Wn,ttfk rPt | d YdSXt|||||||| | d }| j|||| d|dt d|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 .) matplotlibZmatplotlib.imageimageZastropy.utils.introspectionr int ValueErrorr ExceptionrcriticalndimospathsplitextendswithZ colormapsrZget_cmapKeyErrorrZimsaveinfo)filenameextout_fnrrrrrrrrrrZmimgr reZ out_formatrZnormr-h/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/visualization/scripts/fits2bitmap.pyr sRD          cCsBddl}|jdd}|jdddddd|jd d tdd d |jd tddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtdd d|jd!d"td#d$d |jd d%d&d'||}|jD]<}t||j|j |j |j |j |j |j|j|j|j|jd( qdS))Nrz'Create a bitmap file from a FITS image.) descriptionz-ez--extZhduz8Specify the HDU extension number or name (Default is 0).)metavardefaulthelpz-or)zZFilename for the output image (Default is a PNG file with the same name as the FITS file).)r0typer1r2z --stretchrz^Type of image stretching ("linear", "sqrt", "power", "log", or "asinh") (Default is "linear").)r3r1r2z--powerrz4Power 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--cmapZ colormap_namer z1matplotlib color map name (Default is "Greys_r").+z)Path to one or more FITS files to convert)nargsr2) r*r+rrrrrrrrr)argparseArgumentParser add_argumentstrfloat parse_argsr)rr*orrrrrrrrr)argsr6parserr)r-r-r.rs        ) rNrrr NNNNNr )N) r#ZastropyrZastropy.io.fitsrZ#astropy.visualization.mpl_normalizer__all__rrr-r-r-r.s"    |