ó šßÈ[c@ sFddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlZddlmZddlmZmZmZmZmZmZmZmZmZmZddlmZmZdd l m!Z!m"Z"dd l#m$Z$m%Z%dd l&m'Z'id d 6d d6dd6dd6dd6d d6Z(id d 6dd6dd6dd6dd6dd6Z)idd 6dd6dd6dd6dd6Z*edƒZ+edƒZ,edƒZ-yddl.Z.Wne/k re0Z1nXe2Z1de3fd„ƒYZ4d „Z5dS(!iÿÿÿÿ(tdivisiontwith_statementN(treduce(tmemmapi( t isreadablet iswritabletisfilet fileobj_opent fileobj_nametfileobj_closedt fileobj_modet_array_from_filet_array_to_filet _write_stringi(tbt string_types(t download_filet_is_url(t classpropertytdeprecated_renamed_argument(tAstropyUserWarningtrbtreadonlyt copyonwritesrb+tupdatesab+tappendtwbtostreamt denywriteswb+tabtcsr+trs‹sPKtBZt_FilecB seZdZedddƒdddeed„ƒZd„Zd„Z d„Z d„Z dd „Z dd e jdd „Zd „Zd „Zd„Zd„Zd d„Zd„Zdd„Zd„Zd d„Zd„Zd„Zd„Zd„Zedeƒd„ƒZd„ZRS(sM Represents a FITS file on disk (or in some other file-like object). tclobbert overwrites2.0cC s¯t|ƒ|_|dkr!tn|}|dkr‘d|_t|_t|_||_||_ d|_ t|_ t|_ t|_ t|_dSt|_ trÁt|tjƒrÁt|ƒ}nd|_|dkr t|ƒrt|ƒ}tj|ƒ}q d}n|tkr-tdj|ƒƒ‚nt|tƒrl|dkrlt|ƒrlt|d|ƒ|_nt |ƒ|_t|_t|_||_||_ t|_!t|_d|_ t|_ t|_ t|ƒrî|j"|||ƒn8t|tƒr|j#|||ƒn|j$|||ƒt|jƒ|_t|t%j&ƒrVd|_ n<t|t'j(ƒrtd|_ nt|t)j*ƒr’d|_ n|dks³|j r¿|d kr¿t|_ n-|dksà|j rì|dkrìt|_ n|dks|j st+|jd ƒ r d |_,nD|jj-ƒ}|jj.d dƒ|jj-ƒ|_,|jj.|ƒ|j r«t/|jƒsˆt|_ q«|j r«|j0 r«t|_ q«ndS(NRsMode '{}' not recognizedRRtcachetgziptziptbzip2RRRtseekii(RR(RRR(1tboolt strict_memmaptNonetTruet_filetFalsetclosedtbinarytmodeRt compressionRt writeonlyt simulateonlytclose_on_errort HAS_PATHLIBt isinstancetpathlibtPathtstrt_mmapt_is_random_access_file_backedR t FILE_MODEStgett IO_FITS_MODESt ValueErrortformatRRRtnameRt file_liket _open_fileobjt_open_filenamet_open_filelikeR%tGzipFiletzipfiletZipFiletbz2tBZ2FilethasattrtsizettellR(Rt_mmap_available(tselftfileobjR1RR#R$tfmodetpos((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt__init__SsŠ                                        cC sdj|j|jj|jƒS(Ns <{}.{} {}>(RAt __module__t __class__t__name__R-(RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt__repr__ÀscC s|S(N((RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt __enter__ÅscC s|jƒdS(N(tclose(RPttypetvaluet traceback((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt__exit__ÈscC s|jr tSt|jƒS(N(R3R.RR-(RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pytreadableËs cC s]t|jdƒst‚ny|jj|ƒSWn'tk rX|jdkrRdS‚nXdS(NtreadR%t(RLR-tEOFErrorR`tIOErrorR2(RPRM((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR`Ðs  ic C s8t|jdƒst‚nt|tjƒs?tj|ƒ}n|rs||jdkrstdj||ƒƒ‚nt|t ƒrŽ|f}n|p—|s³t j dt ƒd}n|rÓ| rÓ||jf}n|rL|rLtj |ƒ|j}||krtdj|||ƒƒ‚qL||krLtdj|||ƒƒ‚qLn|jjƒ}zÅ|jrÛ|jd kr¶t|jdt|jd tjƒ}|j|_d |_~ntjd |d |d |d |jƒSttj|ƒ}|jj|ƒt|j||ƒ} || _| SWd |jj|ƒXd S(s* 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`issize {} not a multiple of {}s?No size or shape given to readarray(); assuming a shape of (1,)is-size {} is too few bytes for a {} array of {}s.size {} is too many bytes for a {} array of {}R1tdtypetshapetoffsettbufferN(i(RLR-RbR7tnpRdtitemsizeR@RAtinttwarningstwarnRtprodRNRR;R+tMemmapt MEMMAP_MODESR1tuint8tbasetndarrayRtoperatortmulR(R Re( RPRMRfRdRet actualsizetfileposRtcounttdata((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt readarrayÝsN                  cC s|jr tSt|jƒS(N(RR.RR-(RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pytwritable*s cC s)t|jdƒr%t|j|ƒndS(Ntwrite(RLR-R (RPtstring((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR{/scC s)t|jdƒr%t||jƒndS(s× 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. R{N(RLR-R (RPtarray((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt writearray3scC s&t|jdƒr"|jjƒndS(Ntflush(RLR-R(RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR>scC sÒt|jdƒsdSt|jtjƒrr|r_|dkrP|jj|}q_tdƒ‚n|jj|ƒn|jj||ƒ|jjƒ}|j rÎ||j krÎt j dj |j |ƒt ƒndS(NR(isSeek from end not supporteds\File may have been truncated: actual file length ({}) is smaller than the expected size ({})(RLR-R7R%RGRfR@R(RNRMRkRlRAR(RPRftwhenceRS((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR(Bs  cC s(t|jdƒst‚n|jjƒS(NRN(RLR-RbRN(RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pyRNXs cC s)t|jdƒr%|jj|ƒndS(Nttruncate(RLR-R(RPRM((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR]scC sKt|jdƒr"|jjƒn|jƒd|_t|_t|_ dS(s1 Close the 'physical' FITS file. RZN( RLR-RZt_maybe_close_mmapR+R;R,R/R.R5(RP((s3lib/python2.7/site-packages/astropy/io/fits/file.pyRZas    cC sH|jdk rDtj|jƒd|krD|jjƒd|_ndS(sú 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. iN(R;R+tsyst getrefcountRZ(RPtrefcount_delta((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR‚qs  cC sÊ|jr't|dƒr'|jdksWtjj|jƒrÆtjj|jƒdkrÆ|r«|jr…t|dƒr…|jdƒqÃ|s˜|j ƒntj |jƒqÆt dj |jƒƒ‚ndS(sýOverwrite an existing file if ``overwrite`` is ``True``, otherwise raise an IOError. The exact behavior of this method depends on the _File object state and is only meant for use within the ``_open_*`` internal methods. tleniRsFile {!r} already exists.N( RCRLR†tostpathtexistsRBtgetsizeRRZtremoveRcRA(RPR#RQR/((s3lib/python2.7/site-packages/astropy/io/fits/file.pyt_overwrite_existings'0 cC s^t|ƒ}t|ƒpt|}|dkrD|j|||ƒn|s÷|dkrod|kpkd|k sÐ|dkr‡|dksÐ|dkr¸d |kp´d |kp´d|k sÐ|d krë|dkrëtdj||ƒƒ‚n||_nDt|ƒrt|j t|ƒ|_nt j |j t|ƒ|_|dkrZ|jj dƒndS(s9Open a FITS file from a file object or a GzipFile object.RRRRRt+Rsab+srb+twtaRswb+s>Mode argument '{}' does not match mode of the input file ({}).iN(RRR(sab+srb+(srb+swb+( R R R?RŒR@RAR-RRRBR%topenR((RPRQR1R#R/RR((s3lib/python2.7/site-packages/astropy/io/fits/file.pyRD“s(    %   cC sBt|_||_t|ƒr6tdj|ƒƒ‚nt|tjƒrl|j ||ƒ|jj dƒdSt |jdƒ s’t |jdƒ r¢d|_ }n|dkrÄ|j ||tƒn|j d krt |jd ƒ rtd j|j ƒƒ‚n|j dkr>t |jd ƒ r>td j|j ƒƒ‚ndS(sgOpen a FITS file from a file-like object, i.e. one that has read and/or write methods. s;Cannot read from/write to a closed file-like object ({!r}).iNR(RNRRRR{sHFile-like object does not have a 'write' method, required for mode '{}'.R`sGFile-like object does not have a 'read' method, required for mode {!r}.(RRR(R,RCR-R RcRAR7RHRIt _open_zipfileR(RLR1RŒR.(RPRQR1R#((s3lib/python2.7/site-packages/astropy/io/fits/file.pyRFµs*      " cC sÀ|dkr"|j|dtƒntjj|jƒrdt|jdƒ}|jdƒ}WdQXn t dƒ}tjj |jƒd}|dks¤|j t ƒrÌt j|jt|ƒ|_d|_n¼|d ksç|j tƒrý|j|j|ƒn‹|d ks|j tƒrf|dkr3td ƒ‚n|dkrEdnd}tj|j|ƒ|_n"t|jt|ƒ|_t|_t|jtjƒr¬|dkr¬n|jjdƒdS(s(Open a FITS file from a filename string.RRiNRais.gzR%s.zips.bz2RRs:update and append modes are not supported with bzip2 filesRŽRi(RR(RŒR+R,R‡RˆR‰RBRR`Rtsplitextt startswitht GZIP_MAGICR%RR?R-R2t PKZIP_MAGICR‘t BZIP2_MAGICRcRJRKR5R7R((RPtfilenameR1R#tftmagictextt bzip2_mode((s3lib/python2.7/site-packages/astropy/io/fits/file.pyREÜs,     !tlazycC søtjƒ\}}zÄtj|dƒtj|ƒytj|ddtjƒ}Wn9tjk rŒ}tj dj t |ƒƒt ƒ~t SXz:y|jƒWn%tjk rÅtj dt ƒt SXWd|jƒXWdtj|ƒtj|ƒXtS(sTests 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. t itaccesss4Failed to create mmap: {}; mmap use will be disabledsYmmap.flush is unavailable on this platform; using mmap in writeable mode will be disabledN(ttempfiletmkstempR‡R{tfsynctmmapt ACCESS_WRITEterrorRkRlRAR:RR.RRZR‹R,(tclsttmpfdttmpnametmmtexc((s3lib/python2.7/site-packages/astropy/io/fits/file.pyROs,      cC sÐ|d krtdƒ‚nt|tjƒsEtj|ƒ}t}n |}t}|jƒ}t|ƒdkr~tdƒ‚ntj ddƒ|_ |j j |j |dƒƒ|rÃ|j ƒnd |_d S( s£Limited support for zipfile.ZipFile objects containing a single a file. Allows reading only for now by extracting the file to a tempfile. RRs7Writing to zipped fits files is not currently supportedis2Zip files with multiple members are not supported.tsuffixs.fitsiR&N(RR(RcR7RHRIR,R.tnamelistR†RŸtNamedTemporaryFileR-R{R`RZR2(RPRQR1tzfileRZR«((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR‘*s"      N( RWRUt__doc__RR+R.R,RTRXRYR^R_R`RhRpRyRzR{R~RR(RNRRZR‚RŒRDRFRERROR‘(((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR!Ns2 k     M          " ' )%cC st|ƒpt|tjƒS(s*Returns `True` if fileobj is a `file` or `io.FileIO` object or a `gzip.GzipFile` object. Although reading from a zip file is supported, this does not include support for random access, and we do not yet support reading directly from an already opened `zipfile.ZipFile` object. (RR7R%RG(RQ((s3lib/python2.7/site-packages/astropy/io/fits/file.pyR<Hs (6t __future__RRRJR%R¢RsR‡RƒRŸRkRHt functoolsRtnumpyRhRRntutilRRRRRR R R R R t extern.sixRRt utils.dataRRtutils.decoratorsRRtutils.exceptionsRR?R=RoR”R•R–R8t ImportErrorR.R6R,tobjectR!R<(((s3lib/python2.7/site-packages/astropy/io/fits/file.pytsR          F       ÿû