ó  ‰\c@s˜ddgZddljZyddlmZWnAek rryddlZWqsek rnedƒ‚qsXnXdd„Ze d„Z d„Z dS( timreadtimread_collectioniÿÿÿÿN(tfitss|PyFITS could not be found. Please refer to http://www.stsci.edu/resources/software_hardware/pyfits for further instructions.cCswtj|ƒ}d}xQ|D]I}t|tjƒsFt|tjƒr|jdk re|j}PqeqqW|jƒ|S(s•Load an image from a FITS file. Parameters ---------- fname : string Image file name, e.g. ``test.fits``. dtype : dtype, optional For FITS, this argument is ignored because Stefan is planning on removing the dtype argument from imread anyway. Returns ------- img_array : ndarray Unlike plugins such as PIL, where different color bands/channels are stored in the third dimension, FITS images are greyscale-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(tpyfitstopentNonet isinstancetImageHDUt PrimaryHDUtdatatclose(tfnametdtypethdulistt img_arraythdu((s>lib/python2.7/site-packages/skimage/io/_plugins/fits_plugin.pyRs    c Cs6t|ƒ}|tk r3|tk r3tdƒ‚n|tk rK|g}ng}xÉ|D]Á}tj|ƒ}xŸttt|ƒƒ|ƒD]‚\}}t |tj ƒs¹t |tj ƒr‰y|j ƒ}Wntk rå|j }nX|dkr |j ||fƒq q‰q‰W|jƒqXWtj|dtd|ƒS(sêLoad 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. converve_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. s-Input must be a filename or list of filenamesit load_functconserve_memory(ttypetlisttstrt TypeErrorRRtziptrangetlenRRRtsizetappendR tiotImageCollectiont FITSFactory( t load_patternRtintypetext_listtfilenameR tnRt data_size((s>lib/python2.7/site-packages/skimage/io/_plugins/fits_plugin.pyR?s(    (   cCsÕt|tƒstdƒ‚nt|ƒdkr?tdƒ‚n|d}|d}t|ƒtk swt|ƒtk r†tdƒ‚ntj |ƒ}||j }|j ƒ|dkrÑt d||fƒ‚n|S( sªLoad 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. sExpected a tupleisExpected a tuple of length 2iis&Expected a (filename, extension) tuplesExtension %d of %s has no dataN(RttupleRRt ValueErrorRRtintRRR R Rt RuntimeError(t image_extR!textnumR R ((s>lib/python2.7/site-packages/skimage/io/_plugins/fits_plugin.pyRqs  $   ( t__all__t skimage.ioRt astropy.ioRRt ImportErrorRRtTrueRR(((s>lib/python2.7/site-packages/skimage/io/_plugins/fits_plugin.pyts    . 2