HR-e{cddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlZddlmZddlmZmZmZmZddlmZddlmZddlmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(erddl)Z)d d d d d d dZ*dddddddZ+ej,dZ-ej.ej.ej/ej.ej0dZ1dZ2dZ3dZ4dZ5dZ6GddZ7dS)N)reduce)HAS_BZ2)_is_url_requires_fsspec download_fileget_readable_fileobj) classproperty)AstropyUserWarning)NOT_OVERWRITING_MSG) _array_from_file_array_to_file _write_stringfileobj_closed fileobj_mode fileobj_nameisfile isreadable iswritable path_likerbrb+ab+wb)readonly copyonwriteupdateappendostream denywriterrrr)rrrzwb+abrz^[rwa]((t?\+?)|(\+?t?))$)rrrrr ssPKsBZcHtrt|tjSdSNF)r isinstancebz2BZ2File)fileobjs 4lib/python3.11/site-packages/astropy/io/fits/file.py _is_bz2filer)gs!'3;///uc|n|tvret|rtd|dt|}|tvrtd|d|}|S)Nz Text mode 'z4' not supported: files must be opened in binary modeMode '' not recognized) IO_FITS_MODESTEXT_REmatch ValueError FILE_MODESget)modenew_modes r(_normalize_fits_moder6ns D 55 ==   XdXXX >>$'' = ( (j j!r>|d vrt%d |d tEj#|$}nHt|tJrtj&'|}tQ||_||_d|_)t'|r|*|||nKt|t4t6fr|+|||n|,|||t+|j|_t|tZj.rd|_n8t|t^j0rd|_ntc|rd|_|dvs |jr|dkrd|_n|dks |jr |dkrd|_ |dks|jste|jdsd|_3nl|j4} |j5dd|j4|_3|j5| |jr4t'|js d|_dS|js|j6s d|_dSdSdSdS)NTFr,r-zGRequested FITS mode '{}' not compatible with open file handle mode '{}'rrbinary)encodingr9r: close_files)rrr)cachezMode z not supported for HTTPResponsegzipzipbzip2)rrr rrseekr)7bool strict_memmap_fileclosedr<r4memmap compressionr writeonlyclose_on_error_mmap simulateonlyr$osPathLikefspathr.r1rr6rformatrr __enter__strbytesrrnamehttpclient HTTPResponseioBytesIOreadrpath expanduserr file_like _open_fileobj_open_filename_open_fileliker@GzipFilezipfileZipFiler)hasattrsizetellrC_mmap_available) selfr'r4rI overwriter?r9r:objmodeposs r(__init__z_File.__init__s"&\\V      $ ? $D  F %D '2;// -)G,,  M 9 9) __module__ __class____name__rGrjs r(__repr__z_File.__repr__s,L4?LLT^%<LLtzLLLLr*c|SNrws r(rSz_File.__enter__s r*c.|dSrz)close)rjtypevalue tracebacks r(__exit__z_File.__exit__ s r*c<|jrdSt|jSr#)rKrrGrws r(readablez_File.readable s! > 5$*%%%r*ct|jdst |j|S#t$r|jdkrYdSwxYw)Nr\r@)rfrGEOFErrorr\OSErrorrJrjrgs r(r\z _File.readsitz6** N :??4(( (   6))rr   s8AArc~t|jdstt|tjst j|}|r#||jzdkrtd|d|t|tr|f}|s|stj dtd}|r |s ||jzf}|rZ|rXt j ||jz}||krtd|d|d|||krtd|d |d||j } |jrL|jt |j}|jdd t'j|jd|d |_n#t*$r}|jt,jkre|jd krZtj d tt'j|jdt dd |_nYd }~nd }~wwxYwt j||||j|j |j|St3t4j|} |j|t9|j|| } || _| |j |j|S#|j |j|wxYw)a* Similar to file.read(), but returns the contents of the underlying file as a numpy array (or mmap'd array if memmap=True) rather than a string. Usually it's best not to use the `size` argument with this method, but it's provided for compatibility. r\rzsize z not a multiple of z?No size or shape given to readarray(); assuming a shape of (1,))r z is too few bytes for a z array of z is too many bytes for a N)accessoffsetrzCould not memory map array with mode='readonly', falling back to mode='denywrite', which means that the array will be read-onlyr )shapedtyperbuffer)rfrGrr$npritemsizer1intwarningswarnr prodrhrIrM MEMMAP_MODESr4rCmmapfilenorerrnoENOMEMndarrayroperatormulr r) rjrgrrr actualsizefilepos access_modeexccountdatas r( readarrayz_File.readarraystz6** N%** $HUOOE  GD5>)Q..ETEEeEEFF F eS ! ! HE   MQ"   E  . .U^+-E  E %.8JD   RDRR%RR5RRd"" SDSS5SSESS*//##? %{6 :%#/ty"9KJOOAq)))"%)Y J--//;q&&& #"""9 44j9P9P$M!>!3 *. $ 1 1 3 3 !'3K'@'( ***DJJ"'JJJJ#"4zuVDJ JOO    JOOG $ $ $ $x|U33 '''' E5AA"  JOO    JOOG $ $ $ $ JOO    JOOG $ $ $ $s>->DfTYPS>T>T"       r*c|jrtt|jdst|jS)Nrh)rNrrfrGrrhrws r(rhz _File.tells=   Mtz6** Nz   r*cht|jdr|j|dSdS)Ntruncate)rfrGrrs r(rz_File.truncates< 4:z * * & J   % % % % % & &r*ct|jdr|j|d|_d|_d|_dS)z1 Close the 'physical' FITS file. r}NTF)rfrGr}_maybe_close_mmaprMrHrLrws r(r}z _File.closes\ 4:w ' '  J           #r*c|jWttdrDtj|jd|zkr&|jd|_dSdSdSdS)z When mmap is in use these objects hold a reference to the mmap of the file (so there is only one, shared by all HDUs that reference this file). This will close the mmap if there are no arrays referencing it. N getrefcountrD)rMrfsysrr})rjrefcount_deltas r(rz_File._maybe_close_mmapsm J "]++ # ++q>/AAA J     DJJJ # " " "AAr*c|jrt|dr |jdksLtj|jrtj|jdkr|r_|jr't|dr|ddS|s| tj |jdSttj |jdSdS)zOverwrite an existing file if ``overwrite`` is ``True``, otherwise raise an OSError. The exact behavior of this method depends on the _File object state and is only meant for use within the ``_open_*`` internal methods. lenrrN)r_rfrrOr]existsrVgetsizerr}removerr rR)rjrkr'rHs r(_overwrite_existingz_File._overwrite_existings N Eww66 E7;?? GNN49 % %)ggz&B&B)$$Q'''''!( Idi(((((18CCDDD DJJ G__ >dit)<==DJ ,,, """JOOA&&E JOOA        FF  !!'5$77777s,AB?? C  C cd|_||_t|rtd|dt |t jr|||dSt|jdrt|jds dx|_ }|dkr| ||d|j d vr>$) $ $ di&& "!q  " " " " " " " " " " " " " " "Egty))!,((E4S(II 'dit)<==DJ"&D  DJ'' DI,=,= JOOA     ->,=sA99A=A=)lazyctj\}} tj|dtj| t j|dt j}n_#t$rR}tj d|dt~Yd}~tj |tj |dSd}~wwxYw | ng#t$rZtj dtY| tj |tj |dSwxYw | n#| wxYw tj |tj |n-#tj |tj |wxYwd S) aTests that mmap, and specifically mmap.flush works. This may be the case on some uncommon platforms (see https://github.com/astropy/astropy/issues/968). If mmap.flush is found not to work, ``self.memmap = False`` is set and a warning is issued.  r )rzFailed to create mmap: z; mmap use will be disabledNFzYmmap.flush is unavailable on this platform; using mmap in writeable mode will be disabledT)tempfilemkstemprOrfsyncr ACCESS_WRITErrrr r}rr)clstmpfdtmpnamemmrs r(riz_File._mmap_available{s")++w  HUD ! ! ! HUOOO Yua0ABBB    NcNNN&uuu HUOOO Ig     )       D&  HUOOO Ig        HUOOO Ig     HUOOO Ig    tsq)F!A$#F$ C.B; F;CFCE$D==E>F<D==EFE,,F*Gc|dvrtdt|tjstj|}d}n|}d}|}t |dkrtdt jd|_|j | |d |r| |j d d S) zLimited support for zipfile.ZipFile objects containing a single a file. Allows reading only for now by extracting the file to a tempfile. rz7Writing to zipped fits files is not currently supportedTFr z2Zip files with multiple members are not supported.z.fits)suffixrN) rr$rdrenamelistrrNamedTemporaryFilerGrr\r}rC)rjr'r4zfiler}rs r(rz_File._open_zipfiles ' ' 'STT T'7?33 OG,,EEEEE>>## x==A  NOO O0@@@  HQK00111   KKMMM r*)NNNFTrz)r)r)rvrt __qualname____doc__rnrxrSrrr\ruint8rrrrrrCrhrr}rrrr`rbrar rirr{r*r(r8r8{s   @$@$@$@$@$@$DMMM&&&    "!284l%l%l%l%\&&& ... . . .    !!!&&&& $ $ $"EEE(#,#,#,#,J888B(((T.]%%%Nr*r8)8rr@ http.clientrWrZrrrOrerrrrd functoolsrnumpyr"astropy.utils.compat.optional_depsrastropy.utils.datarrrrastropy.utils.decoratorsr astropy.utils.exceptionsr astropy.utils.miscr utilr rrrrrrrrrr%r.r2compiler/ ACCESS_COPYr ACCESS_READrrrrr)r6r8r{r*r(rs   766666 322222777777222222                         JJJ           "*0 1 1 #!   !     B B B B B B B B B B r*