HR-eMBdZddlZddlZddlZddlmZmZddlm Z d Z Gdde Z Gdd ZGd d eZd Zd ZdZddZdS)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. ceZdZdZdS)ExtensionNotFoundExceptionz@Raised if an HDU extension requested by the user does not exist.N)__name__ __module__ __qualname____doc__Blib/python3.11/site-packages/astropy/io/fits/scripts/fitsheader.pyrrTsJJDr rc4eZdZdZd dZd dZdZdZd ZdS) 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. TcT||_||_tj||_dSN)filenameverboseropen_hdulist)selfrrs r__init__zHeaderFormatter.__init__ps%    (++ r NFc|"tt|j}ng}|D]} |t |&#t $r|d}t|dkrJd|dd}t |d}|||fn||YwxYw||||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 compressedhdukeysextpartsextnameextvers rparsezHeaderFormatter.parseus4  C ..//GGG! , , ,NN3s88,,,,!,,,IIcNNE5zzA~~"%((52;"7"7!$U2Y'89999s+++,##GXzBBBs"ABC#"C#c^g}t|D]\}} ||||}n#t$rY*wxYw|dkr|d|d|d|jd|D]}||dd|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'resultidxhducardscs rr$zHeaderFormatter._parse_internals!'** ( (HC XzBB-    Qww d### MM>3>>DM>>> ? ? ? ( ( hhh'''' (wwvs 0 ==c 4 |r|j|j}n|j|j}nL#ttf$r8|jd|d}|jrtj|t|wxYw|s|j }ng}|D]} |j |}t|tj jr||n||[#t$r,|jr"tj|jd|d|dYwxYw|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.z (HDU z ): Keyword )r_headerheader IndexErrorKeyErrorrrrwarningrr6 isinstancercardCardrextend) rhdukeyr&r'r:messager6kwcrds rr2zHeaderFormatter._get_cardssh0 6 6v.6v.5H% 6 6 6GGFGGGG| % G$$$,W55 5  6  LEEE    ,r*C!#ty~66* S)))) S)))| #}VVFVVrVVV  s'*A A3AC3DDc8|jdSr)rclose)rs rrGzHeaderFormatter.closes r )T)NNF) rr r r rr-r$r2rGr r rrrZsy*,,,, -C-C-C-C^ 555nr rceZdZdZdZdS)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 g}|D]c} ||||D]8}||j||jt |jd9T#t $rY`wxYw|rddlm}| |SdS)z6Method called by the parse method in the parent class.)rr5keywordvaluertableN) r2rrrKstrrLrastropyrNTable)rr(r&r' tablerowsr5r?rNs rr$z$TableHeaderFormatter._parse_internals   C  OOC:FFD$$(, #&'+|%(__ .      * % % % % % %;;y)) )tsAA A&%A&N)rr r r r$r r rrIrIs-r rIct|jD]\}}|dkr|jstd} t |}t||j|j|jdn8#t$r+}tj t|Yd}~nd}~wwxYw|r| #|r| wwxYwdS)zPrints FITS header(s) using the traditional 80-char format. Parameters ---------- args : argparse.Namespace Arguments passed from the command-line as defined below. rNr0)end) r1rrKprintrr-r%r'OSErrorrerrorrOrG)argsr4r formatteres rprint_headers_traditionalr[s#4=11"" X 774<7 GGG  "'11I tOOUW         Ic!ff          "!!! "!!!! """s0AA=<C = B2!B-(C -B22C  C%cvg}|jD]}d} t|}||j|j|j}|r||n8#t$r+}tj t|Yd}~nd}~wwxYw|r| #|r| wwxYwt|dkrdSt|dkr |d}nddl m}||}|t"j|jdS)zPrints FITS header(s) in a machine-readable table format. Parameters ---------- args : argparse.Namespace Arguments passed from the command-line as defined below. NrFrrM)format)rrIr-r%rKr'rrVrrWrOrGrrPrNvstackwritesysstdout)rXtablesrrYtblrZresulting_tablerNs rprint_headers_as_tablere(sfFM " "  ",X66I//$/4<QQC # c"""    Ic!ff          "!!! "!!!! " 6{{au V   )!!!!!!,,v..#*TZ88888s0A AB+ B&!B B+ BB++Cc ddlm}g}|jD]}d} t|d}||j|j|j}|r%|tj |ddk}n| |ggd }| |n8#t$r+}tjt|Yd}~nd}~wwxYw|r|#|r|wwxYwt#|dkrdSt#|d kr |d}n||}|d }tj|r|}t#tj|d kr>|D]8} |d | D} | | d 9d } nd} g} |D]} || ddgdg} d| jvrG| D]D}|ddvr | ||dg|| E| | | || } |jr| |j| dddS)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. rrMNF)rrKr0)r)namesrr5c4g|]}|dd|dS)r5:rKr ).0rows r z/print_headers_as_comparison..ys.&W&W&WC#e*'G'Gs9~'G'G&W&W&Wr z hdu+keyword)namerrL)COMMENTHISTORYr) max_lines max_width)rPrNrrIr-r%rKr'npwhererQrrVrrWrOrGrr^ma isMaskedArrayuniqueColumn add_columncolnamessortpprint)rXrNrbrrYrcrZrdhdustab new_columnkeyword_column_name final_tables final_tablerks rprint_headers_as_comparisonrLsC FM""  ",XuEEEI//$/4<QQC E"(3y>R#7889kkH:,mkDD MM#        Ic!ff          "!!! "!!!! " 6{{au V   ),,v.. 5 !D u4  !   29T??a ; ;C&W&WSV&W&W&WXXJ NN:MN : : : :+'L 6 6||S_Q%7$8z|JJK cl " "  y>%;;;""LL#g,c:M6NLOO EKK 445555,,|,,K y$###r22222s0B B"!C1" C,!C C1CC11D ctjttj}|dddt |dddd d d |d ddd t d|}|dddddd|dddd|ddd d t d!|d"d#dd$|d%d&d'(||}|j d*|_ |j rEd+|j D|_ |j s(tj d,tjd-|jrd.|jD|_ |j rt#|d)S|j rt%|d)St'|d)S#t($rYd)SwxYw)/z.s&@@@sS[[c**@@@r zISorting with -s/--sort is only supported in conjunction with -f/--fitsortc:g|]}|ddSrrrs rrlzmain..s&FFF# C--FFFr )argparseArgumentParser DESCRIPTIONRawDescriptionHelpFormatter add_argumentrrOadd_mutually_exclusive_group parse_argsrNrzfitsortrrWr`exitrKrerr[rV)rXparser mode_groups rmainrs  $1UF I/H;/H/H    8       9   4466J  8     :     J      3   A   T " "D z(  y@@di@@@ |  I     HQKKK |GFFFFF   : , "4 ( ( ( ( ( \ , ' - - - - - %d + + + + +      s'G(?G(G(( G65G6r)r rr`numpyrrrPrr astropy.iorstripr ExceptionrrrIr[rerrr r rrsK<<| $$$$$$$$  EGG         RRRRRRRRj?@"""2!9!9!9HD3D3D3Nn n n n n n r