B q\`@sNddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlZddlmZmZmZmZmZmZmZmZmZmZddlmZm Z ddl!m"Z"m#Z#ddl$m%Z%ddd d d dd Z&d ddddddZ'e(dZ)ej*ej*ej+ej*ej,dZ-dZ.dZ/dZ0ddZ1GdddZ2dS)N)reduce) isreadable iswritableisfile fileobj_open fileobj_namefileobj_closed fileobj_mode_array_from_file_array_to_file _write_string) download_file_is_url) classpropertydeprecated_renamed_argument)AstropyUserWarningrbzrb+zab+wb)readonly copyonwriteupdateappendostream denywriterrrr)rzrb+rzwb+Zabzab+z^[rwa]((t?\+?)|(\+?t?))$)rrrrrssPKsBZcCsP|dk rL|tkrLt|r(td|t|}|tkrHtd||}|S)NzAText mode '{}' not supported: files must be opened in binary modezMode '{}' not recognized) IO_FITS_MODESTEXT_REmatch ValueErrorformat FILE_MODESget)modeZnew_moder#3lib/python3.7/site-packages/astropy/io/fits/file.py_normalize_fits_modeNs  r%c@seZdZdZedddd9dd Zd d Zd d ZddZddZ d:ddZ dde j dfddZ ddZddZddZddZd;dd Zd!d"Zdd,d-Zd.d/Zd0d1Zd2d3Zedd4d5d6Zd7d8ZdS)?_FilezM Represents a FITS file on disk (or in some other file-like object). Zclobber overwritez2.0NFTcCst||_|dkrdn|}|dkrbd|_d|_d|_||_||_d|_d|_d|_ d|_ d|_ dSd|_ t |t jr~t|}nt |trtdd|_|dk r|tkrtd|t|rtt|}|dk r||krtd|||}|dkrd}t |tr(|dkr(t|r(t||d|_nBt |tjjr`|dkrPtd |t |!}n t"||_d|_d|_||_||_d|_#d|_ d|_d|_d|_ t|r|$|||n*t |tr|%|||n|&|||t|j|_t |t'j(rd |_n*t |t)j*rd |_nt |t+j,r0d |_|d ksL|jrT|dkrTd|_n"|dksp|jrv|dkrvd|_ |dks|jst-|jdsd|_.n0|j/}|j0dd|j/|_.|j0||jrt|jsd|_n|js|j1sd|_dS)NTFz"names should be `str` not `bytes`.zMode '{}' not recognizedzGRequested FITS mode '{}' not compatible with open file handle mode '{}'r)rrr)cachez&Mode {} not supported for HTTPResponsegzipzipbzip2)rrrrrrseekr)2boolZ strict_memmap_fileclosedZbinaryr"memmap compressionr writeonlyZ simulateonlyclose_on_error isinstancepathlibZPathstrbytes TypeError_mmaprrrrr%r rrnamehttpZclientZ HTTPResponseioBytesIOreadr file_like _open_fileobj_open_filename_open_fileliker)GzipFilezipfileZipFilebz2BZ2Filehasattrsizetellr,_mmap_available)selffileobjr"r1r'r(Zobjmodeposr#r#r$__init___s                   z_File.__init__cCsd|j|jj|jS)Nz <{}.{} {}>)r __module__ __class____name__r/)rMr#r#r$__repr__sz_File.__repr__cCs|S)Nr#)rMr#r#r$ __enter__sz_File.__enter__cCs |dS)N)close)rMtypevalue tracebackr#r#r$__exit__sz_File.__exit__cCs|jr dSt|jS)NF)r3rr/)rMr#r#r$readablesz_File.readablecCsFt|jdsty |j|Stk r@|jdkr:dSYnXdS)Nr?r))rIr/EOFErrorr?OSErrorr2)rMrJr#r#r$r?s   z _File.readrc Cst|jdstt|tjs&t|}|rH||jdkrHtd||t|t rX|f}|sp|spt dt d}|r|s||jf}|r|rt ||j}||krtd|||n||krtd||||j}z|jr|jdkrt|j}|jddytj|jd|dd |_Wnltk r}zL|jtjkr|jd krt d t tj|jdtd dd |_nWdd}~XYnXtj||||jd Sttj|} |j|t|j|| } || _| SWd|j|XdS)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 {} not a multiple of {}z?No size or shape given to readarray(); assuming a shape of (1,))rz-size {} is too few bytes for a {} array of {}z.size {} is too many bytes for a {} array of {}N)accessoffsetrzCould not memory map array with mode='readonly', falling back to mode='denywrite', which means that the array will be read-onlyr)shapedtyper`buffer)rIr/r]r5nprbitemsizerrintwarningswarnrZprodrKr1r: MEMMAP_MODESr"r,mmapfilenor^errnoZENOMEMZndarrayroperatormulr ra) rMrJr`rbraZ actualsizeZfileposZ access_modeexccountdatar#r#r$ readarrays`               z_File.readarraycCs|jr dSt|jS)NF)rrr/)rMr#r#r$writable[sz_File.writablecCst|jdrt|j|dS)Nwrite)rIr/r )rMstringr#r#r$rt`s z _File.writecCst|jdrt||jdS)z Similar to file.write(), but writes a numpy array instead of a string. Also like file.write(), a flush() or close() may be needed before the file on disk reflects the data written. rtN)rIr/r )rMZarrayr#r#r$ writearrayds z_File.writearraycCst|jdr|jdS)Nflush)rIr/rw)rMr#r#r$rwos z _File.flushcCsRt|jdsdS|j|||j}|jrN||jkrNtd|j|tdS)Nr,z\File may have been truncated: actual file length ({}) is smaller than the expected size ({})) rIr/r,rKrJrgrhrr)rMr`whencerOr#r#r$r,ss  z _File.seekcCst|jdst|jS)NrK)rIr/r]rK)rMr#r#r$rK}s z _File.tellcCst|jdr|j|dS)Ntruncate)rIr/ry)rMrJr#r#r$rys z_File.truncatecCs4t|jdr|j|d|_d|_d|_dS)z1 Close the 'physical' FITS file. rVNTF)rIr/rV_maybe_close_mmapr:r0r4)rMr#r#r$rVs   z _File.closecCs2|jdk r.t|jd|kr.|jd|_dS)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. Nr-)r:sys getrefcountrV)rMZrefcount_deltar#r#r$rzs  z_File._maybe_close_mmapcCs|jrt|dr|jdks:tj|jrtj|jdkr|rt|jrZt|drZ|dq|sf| t |jnt d |jdS)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. lenrryzFile {!r} already exists.N) r@rIr}ospathexistsr;getsizeryrVremover^r)rMr'rNr0r#r#r$_overwrite_existings  z_File._overwrite_existingr\cCs|dks|trb|dkr"tdtt|d}t|trD||d<n||d<tjf||_ d|_ np|dkst|t r| |j |d |_ nH|d ks|tr|d krtd |d krdnd}tj||d|_ d|_ |j dk S)z4Attempt to determine if the given file is compressedz.gzrzH'append' mode is not supported with gzip files.Use 'update' mode instead)r"filenamerNr)z.zipr*z.bz2)rrz:update and append modes are not supported with bzip2 filesrwrr+N) startswith GZIP_MAGICr^dictrr5r7r)rDr/r2 PKZIP_MAGIC _open_zipfiler; BZIP2_MAGICrGrH)rMZ obj_or_namemagicr"extkwargsZ bzip2_moder#r#r$_try_read_compresseds&  z_File._try_read_compressedc Cst|}t|pt|}|dkr.|||||s:||_nt|rTt|jt||_y0|dkrj|jd|j d}|jdWnt t fk rdSX| |||dS)zAOpen a FITS file from a file object (including compressed files).r)rrrN) r r rrr/rrr;r,r?r^r)rMrNr"r'r0Zfmoderr#r#r$rAs   z_File._open_fileobjcCsd|_||_t|r"td|t|tjr>|||dSt |jdrVt |jds`d|_ }|dkrv| ||d|j dkrt |jd std |j |j dkrt |jd std |j dS) zgOpen a FITS file from a file-like object, i.e. one that has read and/or write methods. Tz;Cannot read from/write to a closed file-like object ({!r}).Nr,rKrF)rrrrtzHFile-like object does not have a 'write' method, required for mode '{}'.r?zGFile-like object does not have a 'read' method, required for mode {!r}.) r@r/r r^rr5rErFrrIr"r)rMrNr"r'r#r#r$rCs(        z_File._open_filelikec Cs|dkr||ddtj|jrHt|jd}|d}WdQRXnd}tj|jd}|j|j|||dst|jt ||_ d|_ t |j t jr|dks|j d dS) z(Open a FITS file from a filename string.rNTrrr)rr)rr~rrr;rr?splitextrrr/r4r5rGrHr,)rMrr"r'frrr#r#r$rBsz_File._open_filename)Zlazyc Cst\}}zt|dt|ytj|dtjd}Wn:tk rt}zt d t |t ~dSd}~XYnXz2y | Wn tk rt dt dSXWd|XWdt|t|XdS) 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)r_z4Failed to create mmap: {}; mmap use will be disabledFNzYmmap.flush is unavailable on this platform; using mmap in writeable mode will be disabledT)tempfileZmkstempr~rtfsyncrj ACCESS_WRITEr^rgrhrr7rrwrVr)clsZtmpfdZtmpnameZmmror#r#r$rL4s*       z_File._mmap_availablecCs|dkrtdt|tjs,t|}d}n|}d}|}t|dkrPtdtjdd|_|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. )rrz7Writing to zipped fits files is not currently supportedTFrz2Zip files with multiple members are not supported.z.fits)suffixrN) r^r5rErFnamelistr}rZNamedTemporaryFiler/rtr?rVr,)rMrNr"ZzfilerVrr#r#r$rYs"   z_File._open_zipfile)NNNFT)N)r)N)r)r\)rSrQ __qualname____doc__rrPrTrUrZr[r?rdZuint8rrrsrtrvrwr,rKryrVrzrrrArCrBrrLrr#r#r#r$r&Zs4  { b     !&%r&)3rGr)rlZ http.clientr<rjrmr6r=r~r{rrgrEre functoolsrZnumpyrdutilrrrrrr r r r r Zastropy.utils.datarrZastropy.utils.decoratorsrrZastropy.utils.exceptionsrrr compilerZ ACCESS_COPYrZ ACCESS_READrirrrr%r&r#r#r#r$sN 0