B q\a @sPddlZddlmZddlmZddlmZddlmZdd d Z dd d Z dS)N) LooseVersion) simple_norm)log)getdatalinear?皙?Greys_rc  Csddl} ddlm} ddlm}y t|}Wntk r@YnXyt||}Wn,tk r|}zt |dSd}~XYnX|j dkrt d ||dkrt j|d}|drt j|d}|d7}t j|ddd}|dkrt| jtd kr|ddd }| | jkr:t d | dSt|||||||| | d }|j|||| d |dt d |dS)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'. rNz,data in FITS extension {0} is not a 2D arrayz.fitsz.pngZpngz2.0.0z,{0} is not a valid matplotlib colormap name.)stretchpowerasinh_amin_cutmax_cut min_percent max_percentpercentlower)cmaporiginformatzSaved file to {0}.) matplotlibZ matplotlib.cmcmZmatplotlib.imageimageint ValueErrorr ExceptionrZcriticalndimrospathsplitextendswithrr __version__ZdatadrZimsaveinfo)filenameextout_fnr rrrrrrrrrrZmimgreZ out_formatZnormr*Hlib/python3.7/site-packages/astropy/visualization/scripts/fits2bitmap.py fits2bitmap sF;        r,cCsFddl}|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'||}xF|jD]<}t||j|j |j |j |j |j |j|j|j|j|jd( qWdS))Nrz'Create a bitmap file from a FITS image.)Z 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).)r-typer.r/z --stretchrz^Type of image stretching ("linear", "sqrt", "power", "log", or "asinh") (Default is "linear").)r0r.r/z--powerg?z4Power index for "power" stretching (Default is 1.0).z --asinh_ag?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)nargsr/) r'r(r rrrrrrrr)argparseArgumentParser add_argumentstrfloat parse_argsr&r,r'or rrrrrrrr)argsr3parserr&r*r*r+mainysD               r<) rNrrrNNNNNr )N) r Zdistutils.versionrZ#astropy.visualization.mpl_normalizerZastropyrZastropy.io.fitsrr,r<r*r*r*r+s     k