U ?hw@s`ddgZddlmZzddlmZWnek r@edYnXddZd 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.c CsTt|@}d}|D]0}t|tjs0t|tjr|jdk r|j}qFqW5QRX|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 Q/var/www/html/venv/lib/python3.8/site-packages/skimage/io/_plugins/fits_plugin.pyrs    Tc Cst|}|tk r |tk r td|tk r.|g}g}|D]}t|x}ttt||D]^\}}t |tj sxt |tj rXz |j }Wntk r| }YnX|dkrX| ||fqXW5QRXq6tj|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 filenamesr)Z load_funcconserve_memory)typeliststr TypeErrorrrziprangelenrrrsizeappendioZImageCollection FITSFactory) Z load_patternrZintypeZext_listfilenamer nr data_sizer r rr6s*    c Cst|tstdt|dkr&td|d}|d}t|tk sNt|tk rVtdt |}||j }W5QRX|dkrt 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