U e{jC@sdZddlZddlZddlZddlmZmZddlm Z d Z Gddde Z Gdd d ZGd d d eZd d ZddZddZdddZdS)a ``fitsheader`` is a command line script based on astropy.io.fits for printing the header(s) of one or more FITS file(s) to the standard output in a human- readable format. Example uses of fitsheader: 1. Print the header of all the HDUs of a .fits file:: $ fitsheader filename.fits 2. Print the header of the third and fifth HDU extension:: $ fitsheader --extension 3 --extension 5 filename.fits 3. Print the header of a named extension, e.g. select the HDU containing keywords EXTNAME='SCI' and EXTVER='2':: $ fitsheader --extension "SCI,2" filename.fits 4. Print only specific keywords:: $ fitsheader --keyword BITPIX --keyword NAXIS filename.fits 5. Print keywords NAXIS, NAXIS1, NAXIS2, etc using a wildcard:: $ fitsheader --keyword NAXIS* filename.fits 6. Dump the header keywords of all the files in the current directory into a machine-readable csv file:: $ fitsheader --table ascii.csv *.fits > keywords.csv 7. Specify hierarchical keywords with the dotted or spaced notation:: $ fitsheader --keyword ESO.INS.ID filename.fits $ fitsheader --keyword "ESO INS ID" filename.fits 8. Compare the headers of different fits files, following ESO's ``fitsort`` format:: $ fitsheader --fitsort --extension 0 --keyword ESO.INS.ID *.fits 9. Same as above, sorting the output along a specified keyword:: $ fitsheader -f -s DATE-OBS -e 0 -k DATE-OBS -k ESO.INS.ID *.fits 10. Sort first by OBJECT, then DATE-OBS:: $ fitsheader -f -s OBJECT -s DATE-OBS *.fits Note that compressed images (HDUs of type :class:`~astropy.io.fits.CompImageHDU`) really have two headers: a real BINTABLE header to describe the compressed data, and a fake IMAGE header representing the image that was compressed. Astropy returns the latter by default. You must supply the ``--compressed`` option if you require the real header that describes the compression. With Astropy installed, please run ``fitsheader --help`` to see the full usage documentation. N) __version__log)fitsa Print the header(s) of a FITS file. Optional arguments allow the desired extension(s), keyword(s), and output format to be specified. Note that in the case of a compressed image, the decompressed header is shown by default. This script is part of the Astropy package. See https://docs.astropy.org/en/latest/io/fits/usage/scripts.html#module-astropy.io.fits.scripts.fitsheader for further documentation. c@seZdZdZdS)ExtensionNotFoundExceptionz@Raised if an HDU extension requested by the user does not exist.N)__name__ __module__ __qualname____doc__r r a/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/fits/scripts/fitsheader.pyrTsrc@s<eZdZdZdddZdddZd d Zd d Zd dZdS)HeaderFormattera:Class to format the header(s) of a FITS file for display by the `fitsheader` tool; essentially a wrapper around a `HDUList` object. Example usage: fmt = HeaderFormatter('/path/to/file.fits') print(fmt.parse(extensions=[0, 3], keywords=['NAXIS', 'BITPIX'])) Parameters ---------- filename : str Path to a single FITS file. verbose : bool Verbose flag, to show more information about missing extensions, keywords, etc. Raises ------ OSError If `filename` does not exist or cannot be read. TcCs||_||_t||_dSN)filenameverboseropen_hdulist)selfrrr r r __init__pszHeaderFormatter.__init__NFc Cs|dkrtt|j}ng}|D]z}z|t|Wq tk r|d}t|dkrd|dd}t|d}|||fn ||Yq Xq ||||S)aReturns the FITS file header(s) in a readable format. Parameters ---------- extensions : list of int or str, optional Format only specific HDU(s), identified by number or name. The name can be composed of the "EXTNAME" or "EXTNAME,EXTVER" keywords. keywords : list of str, optional Keywords for which the value(s) should be returned. If not specified, then the entire header is returned. compressed : bool, optional If True, shows the header describing the compression, rather than the header obtained after decompression. (Affects FITS files containing `CompImageHDU` extensions only.) Returns ------- formatted_header : str or astropy.table.Table Traditional 80-char wide format in the case of `HeaderFormatter`; an Astropy Table object in the case of `TableHeaderFormatter`. N,r) rangelenrappendint ValueErrorsplitjoin_parse_internal) r extensionskeywords compressedhdukeysextpartsZextnameZextverr r r parseus   zHeaderFormatter.parsec Csg}t|D]z\}}z||||}Wntk r>Yq YnX|dkrR|d|d|d|jd|D]}||dqpq d|S)zEThe meat of the formatting; in a separate method to allow overriding.r z# HDU z in z: ) enumerate _get_cardsrrrr) rr"r r!resultidxhducardscr r r rs  zHeaderFormatter._parse_internalc Csz"|r|j|j}n |j|j}WnBttfk rd|jd|d}|jrXt|t |YnX|sr|j }ntg}|D]j}z2|j |}t |t j jr||n ||Wqztk r|jrtdj|j||dYqzXqz|S)aReturns a list of `astropy.io.fits.card.Card` objects. This function will return the desired header cards, taking into account the user's preference to see the compressed or uncompressed version. Parameters ---------- hdukey : int or str Key of a single HDU in the HDUList. keywords : list of str, optional Keywords for which the cards should be returned. compressed : bool, optional If True, shows the header describing the compression. Raises ------ ExtensionNotFoundException If the hdukey does not correspond to an extension. z : Extension z not found.z2{filename} (HDU {hdukey}): Keyword {kw} not found.)rhdukeykw)rZ_headerheader IndexErrorKeyErrorrrrwarningrr- isinstancercardZCardrextendformat) rr/r r!r1messager-r0Zcrdr r r r)s8    zHeaderFormatter._get_cardscCs|jdSr )rclose)rr r r r:szHeaderFormatter.close)T)NNF) rrrr rr%rr)r:r r r r r Zs   /9r c@seZdZdZddZdS)TableHeaderFormatteraClass to convert the header(s) of a FITS file into a Table object. The table returned by the `parse` method will contain four columns: filename, hdu, keyword, and value. Subclassed from HeaderFormatter, which contains the meat of the formatting. c Csxg}|D]P}z6||||D]"}||j||jt|jdqWqtk rVYqXq|rtddlm}| |SdS)z6Method called by the parse method in the parent class.)rr,keywordvaluertableN) r)rrr<strr=rastropyr?Table)rr"r r!Z tablerowsr,r6r?r r r rs"   z$TableHeaderFormatter._parse_internalN)rrrr rr r r r r;sr;c Cst|jD]\}}|dkr&|js&td}z^z(t|}t||j|j|jddWn0t k r}zt t |W5d}~XYnXW5|r|Xq dS)zPrints FITS header(s) using the traditional 80-char format. Parameters ---------- args : argparse.Namespace Arguments passed from the command-line as defined below. rNr')end) r(rr<printr:r r%rr!OSErrorrerrorr@)argsr+r formatterer r r print_headers_traditionals $rJc Csg}|jD]|}d}zdz.t|}||j|j|j}|r@||Wn0tk rr}zt t |W5d}~XYnXW5|r|Xq t |dkrdSt |dkr|d}nddl m}||}|jtj|jddS)zPrints FITS header(s) in a machine-readable table format. Parameters ---------- args : argparse.Namespace Arguments passed from the command-line as defined below. NrFrr>)r8)rr:r;r%rr<r!rrErrFr@rrAr?vstackwritesysstdout)rGtablesrrHtblrIresulting_tabler?r r r print_headers_as_table*s& $      rRc Csddlm}g}|jD]}d}zz\t|dd}||j|j|j}|r^|t |ddk}n|j |ggdd }| |Wn0t k r}ztt|W5d}~XYnXW5|r|Xqt|dkrdSt|d kr|d}n ||}|d }t j|r|}tt |d krX|D](} |d d | D} | j| ddq(d} nd} g} |D]x} |j| ddgddg} d| jkr| D]6}|ddkrq| |j|dg|| dq| | | qd|| } |jr| |j| jddddS)zPrints FITS header(s) with keywords as columns. This follows the dfits+fitsort format. Parameters ---------- args : argparse.Namespace Arguments passed from the command-line as defined below. rr>NF)rr<r')r)namesrr,cSs"g|]}|dd|dqS)r,:r<r ).0rowr r r {sz/print_headers_as_comparison..z hdu+keyword)namerr=)COMMENTZHISTORYr) max_lines max_width)rAr?rr:r;r%rr<r!npwhererBrrErrFr@rrKmaZ isMaskedArrayuniqueColumn add_columnZcolnamessortpprint)rGr?rOrrHrPrIrQZhdustabZ new_columnZkeyword_column_nameZ final_tablesZ final_tablerVr r r print_headers_as_comparisonNsX   $        recCsptjttjd}|jdddtd|jdddd d d d |jd ddd tdd|}|jddddddd|jddddd|jddd d td!d|jd"d#dd$d|jd%d&d'd(||}|j d)krd*|_ |j r d+d,|j D|_ |j s t d-td.|jr$d/d,|jD|_z0|j r8t|n|j rJt|nt|Wntk rjYnXd)S)0z.zISorting with -s/--sort is only supported in conjunction with -f/--fitsortcSsg|]}|ddqSrsrvrxr r r rWs)argparseArgumentParser DESCRIPTIONRawDescriptionHelpFormatter add_argumentrr@add_mutually_exclusive_group parse_argsr?rbZfitsortrrFrMexitr<rRrerJrE)rGparserZ mode_groupr r r mains            r)N)r r{rMnumpyr\rArrZ astropy.iorstripr} Exceptionrr r;rJrRrerr r r r s>   $G