a dw@s^ddgZddlmZzddlmZWney>edYn0ddZd ddZd d ZdS) imreadimread_collectionN)fitszAstropy could not be found. It is needed to read FITS files. Please refer to https://www.astropy.org for installation instructions.cCsht|J}d}|D]0}t|tjs0t|tjr|jdur|j}qFqWdn1sZ0Y|S)aLoad an image from a FITS file. Parameters ---------- fname : string Image file name, e.g. ``test.fits``. Returns ------- img_array : ndarray Unlike plugins such as PIL, where different color bands/channels are stored in the third dimension, FITS images are grayscale-only and can be N-dimensional, so an array of the native FITS dimensionality is returned, without color channels. Currently if no image is found in the file, None will be returned Notes ----- Currently FITS ``imread()`` always returns the first image extension when given a Multi-Extension FITS file; use ``imread_collection()`` (which does lazy loading) to get all the extensions at once. N)ropen isinstanceImageHDU PrimaryHDUdata)fnamehdulistZ img_arrayhdur h/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/io/_plugins/fits_plugin.pyrs    $Tc Cst|}|tur |tur td|tur.|g}g}|D]}t|}ttt||D]\\}}t |tj sxt |tj rXz |j }Wnty| }Yn0|dkrX| ||fqXWdq61s0Yq6tj|t|dS)aLoad a collection of images from one or more FITS files Parameters ---------- load_pattern : str or list List of extensions to load. Filename globbing is currently unsupported. conserve_memory : bool If True, never keep more than one in memory at a specific time. Otherwise, images will be cached once they are loaded. Returns ------- ic : ImageCollection Collection of images. z-Input must be a filename or list of filenamesrN)Z load_funcconserve_memory)typeliststr TypeErrorrrziprangelenrrrsizeappendioZImageCollection FITSFactory) Z load_patternrZintypeZext_listfilenamer nr data_sizer r rr6s*     0cCst|tstdt|dkr&td|d}|d}t|tusNt|turVtdt |}||j }Wdn1s0Y|durt d|d |d |S) aLoad an image extension from a FITS file and return a NumPy array Parameters ---------- image_ext : tuple FITS extension to load, in the format ``(filename, ext_num)``. The FITS ``(extname, extver)`` format is unsupported, since this function is not called directly by the user and ``imread_collection()`` does the work of figuring out which extensions need loading. zExpected a tuplezExpected a tuple of length 2rz&Expected a (filename, extension) tupleNz Extension z of z has no data) rtuplerr ValueErrorrrintrrr RuntimeError)Z image_extrZextnumr r r r rrfs   (r)T) __all__Z skimage.iorZ astropy.ior ImportErrorrrrr r r rs   ( 0