[c@s9ddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZddlmZdd lmZdd lmZmZdd lmZmZd efd YZdefdYZdefdYZdeefdYZdefdYZdS(iNi(tDELAYEDt _ValidHDUt ExtensionHDUt BITPIX2DTYPEt DTYPE2BITPIXi(tHeader(t_is_pseudo_unsignedt_unsigned_zerot_is_int(t VerifyWarningi(t string_types(trangetzip(t isiterablet lazypropertyt _ImageBaseHDUcBszeZdZidd6dd6dd6dd6d d 6d d 6d d6Zd(d(eeeedZedZ e dZ e dZ e dZ e dZejdZedZejdZdZd(dZd(dd(d(dZd(dd(d(ddZddZd Zeed!Zd"Zd#Zd$Zd%Zd&Zd'ZRS()s{FITS image HDU base class. Attributes ---------- header image header data image data sconforms to FITS standardtSIMPLEsImage extensiontXTENSIONsarray data typetBITPIXsnumber of array dimensionstNAXISs has groupstGROUPSsnumber of parameterstPCOUNTsnumber of groupstGCOUNTcKs(ddlm}tt|jd|d||dk r\t|ts\tdq\n|t kr|dkrtdq||_ nIt|t rdd|j df} nd t |j d f} | d d |j d fd d |j d fg} t||r+| jdt |j dfnt|t |fr}| jdd |j df| jdd|j dfn|dk r|j} t| }|j| dt dt dt n t| }||_ ||_||_||_d|j k} d|j k} |j jdd |_|j jdd|_gt|j jd d D])}|j jd t|dd ^q]|_|j jd |_|j jdd|_|j jdd |_|rdn|j jd|_|j|j|_|j jd|_ |j|_!|j|_"d|kr^|dr^|d|_#nt$|_%|t kr| r|jdks|jd krt |_&ndS||_'|j(|j jd |_|j jd|_|j jd|_| s|j|_!n| s$|j|_"ndS(Ni(t GroupsHDUtdatatheadersheader must be a Header objectsNo header to setup HDU.RtIMAGERRiRiRRRtstriptupdatetendtBZEROtBSCALEtBLANKtname()tgroupsRtsuperRt__init__tNonet isinstanceRt ValueErrorRt_headerRtstandard_keyword_commentstTruetappendtcopytextendt_do_not_scale_image_datat_uintt _scale_backtgett_bzerot_bscaleR tstrt_axest_bitpixt_gcountt_pcountt_blankt _verify_blankt _orig_bitpixt _orig_blankt _orig_bzerot _orig_bscaleR!tFalset _modifiedt_data_needs_rescaleRt update_header(tselfRRtdo_not_scale_image_datatuintt scale_backt ignore_blanktkwargsRtc0tcardstorigtbzero_in_headertbscale_in_headertaxis((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR$(s          "     K!         cCs tdS(s _ImageBaseHDU is sort of an abstract class for HDUs containing image data (as opposed to table data) and should never be used directly. N(tNotImplementedError(tclsR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyt match_headerscCstS(N(R*(RC((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pytis_imagescCs t|S(sB Access a section of the image array without loading the entire array into memory. The :class:`Section` object returned by this attribute is not meant to be used directly by itself. Rather, slices of the section return the appropriate slice of the data, and loads *only* that section into memory. Sections are mostly obsoleted by memmap support, but should still be used to deal with very large scaled images. See the :ref:`data-sections` section of the Astropy documentation for more details. (tSection(RC((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pytsectionscCstt|jS(sX Shape of the image array--should be equivalent to ``self.data.shape``. (ttupletreversedR5(RC((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pytshapescCs|jS(N(R((RC((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRscCs ||_t|_|jdS(N(R(R*R@RB(RCR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs  cCsEt|jdkrdS|j|j|j}|j|j|S(s Image/array data as a `~numpy.ndarray`. Please remember that the order of axes on an Numpy array are opposite of the order specified in the FITS file. For example for a 2D image the "rows" or y-axis are the first dimension, and the "columns" or x-axis are the second dimension. If the data is scaled using the BZERO and BSCALE parameters, this attribute returns the data scaled to its physical values unless the file was opened with ``do_not_scale_image_data=True``. iN(tlenR5t_get_scaled_image_datat _data_offsetRWt_update_header_scale_infotdtype(RCR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs cCsd|jkr[|jddk r[|jd|kr9dSt|_t|jdj}nt|_t}|dk rt|tj  rytj |}Wqt k rt dj |qXn||jdR=(RCRt was_unsigned((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs@"               cCsR|jp0|jjp0|jo0|j|jjks7dS|jjdd}d|jkrk|jd}n|jjd}|jjd|j |dd|jr|j|jjkrt |jj|_ |j j nd|jkr|jjd}n |jd}|jjdt |j |ddxt|j D]}\}}dt|d}||jkr||j|R0R(R+RmRR!R2R3R6tpopR9(RCR\toriginal_was_unsignedtkeyword((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR[ys2        toldc Cs,|jd|d|d|d|dddS(so Scale image data by using ``BSCALE``/``BZERO``. Call to this method will scale `data` and update the keywords of ``BSCALE`` and ``BZERO`` in the HDU's header. This method should only be used right before writing to the output file, as the data will be scaled and is therefore not very usable after the call. Parameters ---------- type : str, optional destination data type, use a string representing a numpy dtype name, (e.g. ``'uint8'``, ``'int16'``, ``'float32'`` etc.). If is `None`, use the current data type. option : str, optional How to scale the data: ``"old"`` uses the original ``BSCALE`` and ``BZERO`` values from when the data was read/created (defaulting to 1 and 0 if they don't exist). For integer data only, ``"minmax"`` uses the minimum and maximum of the data to scale. User-specified ``bscale``/``bzero`` values always take precedence. bscale, bzero : int, optional User-specified ``BSCALE`` and ``BZERO`` values ttypetoptiontbscaletbzerotblankN(t_scale_internalR%(RCR{R|R}R~((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pytscalesic Cs]|jdkrdS|dkr/t|j}ntt|}|dk re|dk re|}|}n-|dk r|}d}n|dk rd}|}n|dkr|jdk r|jdk r|j}|j}n|dkrt|tj  rtj j |jj } tj j |jj } |tjkrO| }| | d}q| | d}d|j} | | d| d}n d}d}|dkrtj|j| d |jd d ||jd R=t issubclasstfloatingtminimumtreducetflattmaximumtuint8titemsizetaddR(RmtintegertisnanR\R{RataroundRR!R1R2R3R9R;R<( RCR{R|R}R~Rt_typet_scalet_zerotmintmaxtnbytes((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRsr         #  '    twarncCs*|j|jtt|j|S(N(RBR:R#Rt_verify(RCR|((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR6s  cCs|jdkrdSg}t|jsM|jdj|jd|_n|jdksu|jdd|_nx|D]}tj|tq|WdS(NsInvalid value for 'BLANK' keyword in header: {0!r} The 'BLANK' keyword must be an integer. It will be ignored in the meantime.is|Invalid 'BLANK' keyword in header. The 'BLANK' keyword is only applicable to integer data, and will be ignored in this HDU.( R9R%RR+RdR6twarningsRR (RCtmessagestmsg((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR:>s    cCsh|jr)|jt|jd|jn|j| rO|jrO|j}ntt |j ||S(NR( R0RRR;R<RBRARR#Rt _prewriteto(RCtchecksumtinplacet_((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRXs    cCs>d}|jdk r:tjdkr-d}nd}t|jjrtj|jt|jjddj |jjj }t }n%|j}|jj d}||k}|j s#|r|jjr|jtz|j|Wd|jtXq |j|jt q#|j|n||j|j 7}n|S( Nitlittleti{}(RR(R(RR%tsyst byteorderRR\R_RaRRdRR?R4t simulateonlytflagst writeabletbyteswapR*t writearraytsize(RCtfileobjRt swap_typestoutputt should_swapR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyt_writedata_internales0       cCs|j}|jr|jdkrxtdtjdfdtjdfdtjdffD]3\}}||kr^|jd|d>kr^|Sq^Wn|dkrtjdS|d krtjd Sd S( s Determine the dtype that the data should be converted to depending on the BITPIX value in the header, and possibly on the BSCALE value as well. Returns None if there should not be any change. iituint16i tuint32i@tuint64tfloat64itfloat32N(R;R/R>R_R\R=(RCtbitpixtbitsR\((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRus "#    cCsl|j}|dk rh|jdkrh|jd}tj|d|}|tjd|d>8}|SdS(sW Handle "pseudo-unsigned" integers, if the user requested it. Returns the converted data array if so; otherwise returns None. In this case case, we don't need to handle BLANK to convert it to NAN, since we can't do NaNs with integers, anyway, i.e. the user is responsible for managing blanks. RtiR\iN(RuR%RvRR_RaR(RCRR\R((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyt_convert_pseudo_unsigneds  cCs5t|j}|j|||}|jjd|_|jsm|jdkrq|jdkrq|jdkrq|Sy|j j rt dnWnt k rnXd}|jdko|jdks|j|}n|dkr1d}|jdk rM|jdkrM|j|jk}|jt|dkrMtj|}qMn|j}|dk r}tj|d|}nK|j dk r|j jr|j}n!|jjs|j}n|}~|jdkrtj||j|n|jdkr||j7}n|jr1tj|j|iis`Cannot load a memory-mapped image: BZERO/BSCALE/BLANK header keywords present. Set memmap=False.iR\N(RR;t _get_raw_dataR\t newbyteorderR.R=R>R9R%t_filet strict_memmapR'tAttributeErrorRR6RtsumRXR_twhereRuRatmemmapR,RRtmultiplytnan(RCtoffsetRWtcodetraw_dataRtblankst new_dtype((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRYsJ         cCs#|jj}|jrV|jdkr-d}q|jjj}||jdd}n|jr~t |jr~t |j }nd}|r|j r|j dks|jdkr|j}|dk r|dj|j7}qntt|j}|j|j|t|j||dfS(sC Summarize the HDU: name, dimensions, and formats. tt.iis (rescales to {0})N(t __class__t__name__t _data_loadedRR%R\R!trfindRWtallRR6R.R>R=RuRdRURVtverRXR((RCt class_nameRdRRW((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyt_summarys      cCsi|jrL|j}t|jjratj|jt|jjddj|jjj}n|jj ddkr|j j rt }|j t }|jjd|_q|j t}|jjd|_t}nt}|j|jjtjd|}|rHt|jj rH|j t |jjd|_n|Stt|jd|SdS(sJ Calculate the value for the ``DATASUM`` card in the HDU. R\si{}iRtblockingRN(RiRRR\R_RaRRdRR4RRR*RRR?t_compute_checksumtflattentviewRR#Rt_calculate_datasum(RCRtdt byteswappedtcs((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR&s,      N(Rt __module__t__doc__R)R%R?R*R$t classmethodRQtpropertyRRRTRWRtsetterRRRBR[RRRR:RRRuRRYRR(((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRsB     > C 5  h   *   I !RScBs)eZdZdZdZdZRS(s Image section. Slices of this object load the corresponding section of an image array from the underlying FITS file on disk, and applies any BSCALE/BZERO factors. Section slices cannot be assigned to, and modifications to a section are not saved back to the underlying file. See the :ref:`data-sections` section of the Astropy documentation for more details. cCs ||_dS(N(thdu(RCR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR$gscCst|ts|f}nt|jj}td|DoRt||k}td|Ds{|tf7}ntg|D]}|tkr|^q}t|||ks|dkrtdnt dt |tfD}|| t df|t|d||d}td|DoTt||k}g}d} xlt |D]^}|jj|} t||| } | | | j} t||sp|j| jPqpqpW| j} xyt |d|D]d} |jj| } t|| | } |j| j| j| krM| jrM| | 9} qt} qW| rt|pld }|jj}|jj| t|d} |jj| |}n|j|}|r|j}n|r|j}n|S( Ncss'|]}t|ttjfVqdS(N(R&tintR_R(t.0tk((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pys nscss|]}|tkVqdS(N(tEllipsis(RR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pys psistoo many indices for arraycss'|]\}}|tkr|VqdS(N(R(RtiR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pys xscss'|]}t|ttjfVqdS(N(R&RR_R(RR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pys zsii(i(R&RURXRRWRtanyRt IndexErrortnextRltsliceR%R t _IndexInfoRRR+tnptst contiguousR?R;RZtabsRYt_getdatatitemtsqueeze(RCtkeytnaxist return_scalarRtellipsis_countRrt return_0dimtdimsRRNtindxt is_contiguoustjdxRR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyt __getitem__jsT +"#5    c Csxtt||jjD]c\}\}}t|trVt|j|}Pqt|rt j |dt |}PqqWg|D]'}||| |f||d^q}t d||dDrt j |St jg|D]}t j|^qSdS(NR\icss*|] }t|tp!t|VqdS(N(R&RR (RR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pys s(RlR RRWR&RR tindicesR R_tarangeRRRat concatenatet atleast_1d( RCtkeysRrRRNtksRRRa((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs. 4  (RRRR$RR(((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRSYs   7t PrimaryHDUcBsSeZdZdZddeeeddZedZ dZ ddZ RS(s! FITS primary HDU class. tPRIMARYc Cstt|jd|d|d|d|d|d||d kr|jd}|dkred }n|jjd td dt|nd S( s Construct a primary HDU. Parameters ---------- data : array or DELAYED, optional The data in the HDU. header : Header instance, optional The header to be used (as a template). If ``header`` is `None`, a minimal header will be provided. do_not_scale_image_data : bool, optional If `True`, image data is not scaled using BSCALE/BZERO values when read. (default: False) ignore_blank : bool, optional If `True`, the BLANK header keyword will be ignored if present. Otherwise, pixels equal to this value will be replaced with NaNs. (default: False) uint : bool, optional Interpret signed integer data where ``BZERO`` is the central value and ``BSCALE == 1`` as unsigned integer data. For example, ``int16`` data with ``BZERO = 32768`` and ``BSCALE = 1`` would be treated as ``uint16`` data. (default: True) scale_back : bool, optional If `True`, when saving changes to a file that contained scaled image data, restore the data to the original type and reapply the original BSCALE/BZERO values. This could lead to loss of accuracy if scaling back to integer values after performing floating point operations on the data. Pseudo-unsigned integers are automatically rescaled unless scale_back is explicitly set to `False`. (default: None) RRRDRERGRFRiRtEXTENDRhN(R#RR$R%R(RkR*R4(RCRRRDRGRERFtdim((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR$s)      cCs?|jd}|jdko>d|ks8|dtko>|jS(NiRR(RJRyR*tvalue(RPRtcard((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRQs cCsptt|jd|jkrlt|jrMdtt|j}nd}|jjdd|ndS(NRRRh(R#RRBR(RXR5R4Rk(RCRh((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRBs RcCsitt|jd|}d|jkre|jjdd}|jd|ddt||n|S(NR|RRiicSs t|tS(N(R&tbool(tv((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyts(R#RRR(R1t req_cardsR*(RCR|terrsR((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs N( RRRt _default_nameR%R?R*R$RRQRBR(((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs 4 tImageHDUcBsJeZdZdZdddeeddZedZ ddZ RS(s) FITS image extension HDU class. Rc Cs;tt|jd|d|d|d|d|d|dS(s Construct an image HDU. Parameters ---------- data : array The data in the HDU. header : Header instance The header to be used (as a template). If ``header`` is `None`, a minimal header will be provided. name : str, optional The name of the HDU, will be the value of the keyword ``EXTNAME``. do_not_scale_image_data : bool, optional If `True`, image data is not scaled using BSCALE/BZERO values when read. (default: False) uint : bool, optional Interpret signed integer data where ``BZERO`` is the central value and ``BSCALE == 1`` as unsigned integer data. For example, ``int16`` data with ``BZERO = 32768`` and ``BSCALE = 1`` would be treated as ``uint16`` data. (default: True) scale_back : bool, optional If `True`, when saving changes to a file that contained scaled image data, restore the data to the original type and reapply the original BSCALE/BZERO values. This could lead to loss of accuracy if scaling back to integer values after performing floating point operations on the data. Pseudo-unsigned integers are automatically rescaled unless scale_back is explicitly set to `False`. (default: None) RRR!RDRERFN(R#RR$(RCRRR!RDRERF((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR$s* cCsP|jd}|j}t|tr4|j}n|jdkoO||jkS(NiR(RJRR&R trstripRyt _extension(RPRRtxtension((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRQLs   RcCsWtt|jd|}|jjdd}|jd|ddd|||S(s) ImageHDU verify method. R|RiRicSst|o|dkS(Ni(R(R ((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR ^s(R#RRR(R1R (RCR|R R((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRTs  N( RRRRR%R?R*R$RRQR(((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRs  .RcBseZdZRS(cCst|r^d|ko#|knrFd|_||_t|_qtdj|nt|tr|j |\}}}||||_||_|dk|_nEt |rt ||_d|_t |_ntdj|dS(NiisIndex {} out of range.sIllegal index {}( RRRR*RRRdR&RRR RXR?(RCRRtstarttstoptstep((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyR$ds         (RRR$(((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyRcs( RRtnumpyR_tbaseRRRRRRRtutilRRRtverifyR t extern.sixR textern.six.movesR R tutilsR RRtobjectRSRRR(((s8lib/python2.7/site-packages/astropy/io/fits/hdu/image.pyts"   (J^_M