B z‹dYÉã!@sâdZddddddddd d g Zd d lmZd dlZd dlZd dlm Z d dlm Z ej Z e dZ d+dd„Zd,d d„Zd-d"d„Zd.d$d„Zd/d%d„Zd0d&d„Zd'd„Zd(d„Zejjje_d)d „Zejjje_d*d „Zejjje_dS)1a³ ===================================================== visufunc.py : Healpix visualization functions ===================================================== This module provides functions to display Healpix map. Map projections --------------- - :func:`mollview` displays a map using Mollweide projection (full sky) - :func:`gnomview` displays a map using Gnomonic projection (local map) - :func:`cartview` displays a map using Cartesian projection - :func:`orthview` displays a map using Orthographic projection (full or half sky) - :func:`azeqview` displays a map using Azimuthal equidistant projection Graticules ---------- - :func:`graticule` adds a graticule to the current map - :func:`delgraticules` deletes all graticules of a map Tracing lines or points ----------------------- - :func:`projplot` plots data points on the current map - :func:`projscatter` displays scatter points - :func:`projtext` display a text on the current map ÚmollviewÚgnomviewÚcartviewÚorthviewÚazeqviewÚ graticuleÚ delgraticulesÚprojplotÚ projscatterÚprojtexté)ÚprojaxeséN)Ú pixelfuncg€f@Úé úMollweide viewFÚastroú%gTÚgrayÚwhiteéc2 Csâddl}|dkr$t d¡tj}d}t |¡}tj||d|sf|sf|sf|j|dd}|s^d}d } n|r¼| ¡}t  |  ¡  ¡¡  ¡\}!}"}#}$|s˜d }|!|"|#|!|$|"f} |  |  ¡¡nÆ|rÊ| ¡}n¸| ¡}t|d ƒrè|\}%}&}'n |d |d d |d }%}&}'|'dks |'|&|%kr2td|%|&|'fƒ‚|'d|&|'d|&}(})|sVd}|(d|&d|)dd|%d|&d|%f} |sÔ| d|d| d|d| d|d|d| d|d|df} | ¡}*| ¡zÔt |¡}|dk rt |¡}|r|  ¡}+n tj|| |||| d}+| |+¡| rPtj|| |dd}n| rhtj|| |dd}|+j|||||| |||||d },|r^|+ ¡d}-|-j t dd|-jjd¡¡}.t |-jj|-jj |-jj¡}/t!j"j#t$j%j&|-jj|-jj d|d}0t$j'dkr,|j(|0|+dddt )||¡dd|.|/|d }1n&|j(|0dddt )||¡dd|.|/|d! }1|1j* +d¡|+ ,|¡|sŠ|+j-d"d|+j.j/d#d$|+j0d%|r®|1j1j-dd&|d#|1j1j0d'd'd(| 2|+¡Wd| 3¡|*rÒ| 4¡X|rÞ|,SdS))a.Plot a healpix map (given as an array) in Mollweide projection. Parameters ---------- map : float, array-like or None An array containing the map, supports masked maps, see the `ma` function. If None, will display a blank map, useful for overplotting. fig : int or None, optional The figure number to use. Default: create a new figure rot : scalar or sequence, optional Describe the rotation to apply. In the form (lon, lat, psi) (unit: degrees) : the point at longitude *lon* and latitude *lat* will be at the center. An additional rotation of angle *psi* around this direction is applied. coord : sequence of character, optional Either one of 'G', 'E' or 'C' to describe the coordinate system of the map, or a sequence of 2 of these to rotate the map from the first to the second coordinate system. unit : str, optional A text describing the unit of the data. Default: '' xsize : int, optional The size of the image. Default: 800 title : str, optional The title of the plot. Default: 'Mollweide view' nest : bool, optional If True, ordering scheme is NESTED. Default: False (RING) min : float, optional The minimum range value max : float, optional The maximum range value flip : {'astro', 'geo'}, optional Defines the convention of projection : 'astro' (default, east towards left, west towards right) or 'geo' (east towards right, west towards left) remove_dip : bool, optional If :const:`True`, remove the dipole+monopole remove_mono : bool, optional If :const:`True`, remove the monopole gal_cut : float, scalar, optional Symmetric galactic cut for the dipole/monopole fit. Removes points in latitude range [-gal_cut, +gal_cut] format : str, optional The format of the scale label. Default: '%g' format2 : str, optional Format of the pixel value under mouse. Default: '%g' cbar : bool, optional Display the colorbar. Default: True notext : bool, optional If True, no text is printed around the map norm : {'hist', 'log', None} Color normalization, hist= histogram equalized color mapping, log= logarithmic color mapping, default: None (linear color mapping) cmap : a color map The colormap to use (see matplotlib.cm) badcolor : str Color to use to plot bad values bgcolor : str Color to use for background hold : bool, optional If True, replace the current Axes by a MollweideAxes. use this if you want to have multiple maps on the same figure. Default: False sub : int, scalar or sequence, optional Use only a zone of the current figure (same syntax as subplot). Default: None reuse_axes : bool, optional If True, reuse the current Axes (should be a MollweideAxes). This is useful if you want to overplot with a partially transparent colormap, such as for plotting a line integral convolution. Default: False margins : None or sequence, optional Either None, or a sequence (left,bottom,right,top) giving the margins on left,bottom,right and top of the axes. Values are relative to figure (0-1). Default: None return_projected_map : bool if True returns the projected map in a 2d numpy array alpha : float, array-like or None An array containing the alpha channel, supports masked maps, see the `ma` function. If None, no transparency will be applied. See Also -------- gnomview, cartview, orthview, azeqview r Né F)Únest)g!@gš™™™™™@)Úfigsize)g{®Gáz”?gš™™™™™©?g{®Gáz”?gš™™™™™©?)gggð?gð?)ggggÚ__len__édé r z Wrong values for sub: %d, %d, %d)g{®Gáz„?ggg{®Gáz”?gð?ré)ÚcoordÚrotÚformatÚflipconvT)Úgal_cutrÚcopy) rÚxsizerÚvminÚvmaxÚcmapÚbadcolorÚbgcolorÚnormÚalpha)r%r&)r*r'z0.91.0Ú horizontalgà?égš™™™™™©?gš™™™™™¹?) ÚaxÚ orientationÚshrinkÚaspectÚticksÚpadÚfractionÚ boundariesÚvaluesr ) r/r0r1r2r3r4r5r6r g…ëQ¸…ë?éÚbold)ÚfontsizeÚ fontweightÚ transformgð¿Úcenter)r9r;ÚhaÚva)5ÚpylabÚnpÚzerosÚinfrÚ get_nsideÚ check_nsideÚfigureÚgcfÚarrayÚgcaÚ get_positionÚravelÚdelaxesÚhasattrÚ ValueErrorÚ isinteractiveÚioffÚ ma_to_arrayÚPAÚHpxMollweideAxesÚadd_axesÚ remove_dipoleÚremove_monopoleÚprojmapÚ get_imagesr*ÚinverseÚlinspacer'ÚNr%r&ÚpltÚcmÚScalarMappableÚ matplotlibÚcolorsÚ NormalizeÚ __version__ÚcolorbarÚBoundaryLocatorÚsolidsÚset_rasterizedÚ set_titleÚtextÚprojÚ coordsysstrÚ transAxesr.ÚscaÚdrawÚion)2ÚmapÚfigrrÚunitr$ÚtitlerÚminÚmaxÚflipÚ remove_dipÚ remove_monor"r Úformat2Úcbarr'r(r)Únotextr*ÚholdÚ reuse_axesÚmarginsÚsubZnlocsÚreturn_projected_mapr+r?ÚnsideÚfÚextentÚleftÚbottomÚrightÚtopÚnrowsÚncolsÚidxÚcÚrÚwasinteractiver.ÚimgÚimÚbÚvÚmappableÚcb©r’ú\/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/healpy/visufunc.pyrJsös                      éÈçø?ú Gnomonic viewú%.3gc5Csfddl}|dkr$t d¡tj}d}t |¡}tj|| d|sf|sf|sf|j|dd} |s^d}d }!n|r¼| ¡} t  |  ¡  ¡¡  ¡\}"}#}$}%|s˜d }|"|#|$|"|%|#f}!|   |  ¡¡nÆ|rÊ| ¡} n¸| ¡} t|d ƒrè|\}&}'}(n |d |d d |d }&}'}(|(dks |(|'|&kr2td|&|'|(fƒ‚|(d|'|(d|'})}*|sVd}|)d|'d|*dd|&d|'d|&f}!|sÔ|!d|d|!d|d|!d|d|d|!d|d|df}!| ¡}+| ¡z8t |¡}|r|   ¡},n tj| |!||||d},|  |,¡| r))7r?r@rArBrrCrDrErFrGrHrIrJrKrLrMrNrOrPZfloat64ÚresizerQZHpxCartesianAxesrSrTrUrVrWr*rXrYr'rZr%r&r[r\r]r^r_r`rarbrcrdrerfrgrhrirjr.rkrmrl)5rnrorÚzatrrpr$r˜rŸr rqrrurvr"rrrsrtr rxr'r(r)r*r1rzr}r{r|ryr~r+r?rr€rr‚rƒr„r…r†r‡rˆr‰rŠr‹r.rŒrrŽrrr‘r’r’r“rysx                    úOrthographic viewc2Cs¾ddl}|dkr$t d¡tj}d}t |¡}tj||d|sf|sf|sf|j|dd}|s^d}d } n|r¼| ¡}t  |  ¡  ¡¡  ¡\}!}"}#}$|s˜d }|!|"|#|!|$|"f} |  |  ¡¡nÆ|rÊ| ¡}n¸| ¡}t|d ƒrè|\}%}&}'n |d |d d |d }%}&}'|'dks |'|&|%kr2td|%|&|'fƒ‚|'d|&|'d|&}(})|sVd}|(d|&d|)dd|%d|&d|%f} |sÔ| d|d| d|d| d|d|d| d|d|df} | ¡}*| ¡z°|rø|  ¡}+n tj|| |||| d}+| |+¡| r2tj|||dd}n| rJtj|||dd}|+j|||||| | |||||d },|r:|+ ¡d}-|-j t dd|-jjd¡¡}.t |-jj|-jj|-jj¡}/t j!j"t#j$j%|-jj|-jjd|d}0t#j&dkr |j'|0|+dddt (¡dd|.|/|d }1n"|j'|0dddt (¡dd|.|/|d! }1|1j) *d¡|+ +|¡|sf|+j,d"d|+j-j.d#d$|+j/d%|rŠ|1j0j,dd&|d#|1j0j/d'd'd(| 1|+¡Wd| 2¡|*r®| 3¡X|rº|,SdS))acPlot a healpix map (given as an array) in Orthographic projection. Parameters ---------- map : float, array-like or None An array containing the map. If None, will display a blank map, useful for overplotting. fig : int or None, optional The figure number to use. Default: create a new figure rot : scalar or sequence, optional Describe the rotation to apply. In the form (lon, lat, psi) (unit: degrees) : the point at longitude *lon* and latitude *lat* will be at the center. An additional rotation of angle *psi* around this direction is applied. coord : sequence of character, optional Either one of 'G', 'E' or 'C' to describe the coordinate system of the map, or a sequence of 2 of these to rotate the map from the first to the second coordinate system. half_sky : bool, optional Plot only one side of the sphere. Default: False unit : str, optional A text describing the unit of the data. Default: '' xsize : int, optional The size of the image. Default: 800 title : str, optional The title of the plot. Default: 'Orthographic view' nest : bool, optional If True, ordering scheme is NESTED. Default: False (RING) min : float, optional The minimum range value max : float, optional The maximum range value flip : {'astro', 'geo'}, optional Defines the convention of projection : 'astro' (default, east towards left, west towards right) or 'geo' (east towards roght, west towards left) remove_dip : bool, optional If :const:`True`, remove the dipole+monopole remove_mono : bool, optional If :const:`True`, remove the monopole gal_cut : float, scalar, optional Symmetric galactic cut for the dipole/monopole fit. Removes points in latitude range [-gal_cut, +gal_cut] format : str, optional The format of the scale label. Default: '%g' format2 : str, optional Format of the pixel value under mouse. Default: '%g' cbar : bool, optional Display the colorbar. Default: True notext : bool, optional If True, no text is printed around the map norm : {'hist', 'log', None} Color normalization, hist= histogram equalized color mapping, log= logarithmic color mapping, default: None (linear color mapping) cmap : a color map The colormap to use (see matplotlib.cm) badcolor : str Color to use to plot bad values bgcolor : str Color to use for background hold : bool, optional If True, replace the current Axes by an OrthographicAxes. use this if you want to have multiple maps on the same figure. Default: False sub : int, scalar or sequence, optional Use only a zone of the current figure (same syntax as subplot). Default: None reuse_axes : bool, optional If True, reuse the current Axes (should be a OrthographicAxes). This is useful if you want to overplot with a partially transparent colormap, such as for plotting a line integral convolution. Default: False margins : None or sequence, optional Either None, or a sequence (left,bottom,right,top) giving the margins on left,bottom,right and top of the axes. Values are relative to figure (0-1). Default: None return_projected_map : bool if True returns the projected map in a 2d numpy array alpha : float, array-like or None An array containing the alpha channel, supports masked maps, see the `ma` function. If None, no transparency will be applied. See Also -------- mollview, gnomview, cartview, azeqview r NrF)r)g!@gš™™™™™@)r)g{®Gáz”?gš™™™™™©?g{®Gáz”?gš™™™™™©?)gggð?gð?)ggggrrrr z Wrong values for sub: %d, %d, %d)g{®Gáz„?ggg{®Gáz”?gð?rr)rrr r!T)r"rr#) rr$Úhalf_skyrr%r&r'r(r)r*r+)r%r&)r*r'z0.91.0r,gà?r-gš™™™™™©?gš™™™™™¹?) r.r/r0r1r2r3r4r5r6r ) r/r0r1r2r3r4r5r6r g…ëQ¸…ë?r7r8)r9r:r;gð¿r<)r9r;r=r>)4r?r@rArBrrCrDrErFrGrHrIrJrKrLrMrNrOrQZHpxOrthographicAxesrSrTrUrVrWr*rXrYr'rZr%r&r[r\r]r^r_r`rarbrcrdrerfrgrhrirjr.rkrlrm)2rnrorrrpr$r¤rqrrrrsrtrurvr"r rwrxr'r(r)ryr*rzr|r}r{r~r+r?rr€rr‚rƒr„r…r†r‡rˆr‰rŠr‹r.rŒrrŽrrr‘r’r’r“rŽsòu                 c!6CsÞddl}!|dkr$t d¡tj}d}t |¡}"tj|"| d|sf|sf|sf|!j|dd}#|s^d}d }$n|r¼|! ¡}#t  |#  ¡  ¡¡  ¡\}%}&}'}(|s˜d }|%|&|'|%|(|&f}$|#  |#  ¡¡nÆ|rÊ|! ¡}#n¸|! ¡}#t|d ƒrè|\})}*}+n |d |d d |d })}*}+|+dks |+|*|)kr2td|)|*|+fƒ‚|+d|*|+d|*},}-|sVd}|,d|*d|-dd|)d|*d|)f}$|sÔ|$d|d|$d|d|$d|d|d|$d|d|df}$|! ¡}.|! ¡zÐ|rø|#  ¡}/n tj|#|$||||d}/|# |/¡| r2tj||| dd}n|rJtj||| dd}|/j|| |||| | |||||||| d}0|r@|/ ¡d}1|1j t dd|1jjd¡¡}2t |1jj|1jj|1jj¡}3t j!j"t#j$j%|1jj|1jjd|d}4t#j&dkr|#j'|4|/dddt (¡dd|2|3|d }5n"|#j'|4dddt (¡dd|2|3|d! }5|5j) *d¡| dkrZ| rVd"} nd#} |/ +| ¡|s†|/j,d$d|/j-j.d%d&|/j/d'|rª|5j0j,dd(|d%|5j0j/d)d)d*|# 1|/¡Wd|! 2¡|.rÎ|! 3¡X|rÚ|0SdS)+aùPlot a healpix map (given as an array) in Azimuthal equidistant projection or Lambert azimuthal equal-area projection. Parameters ---------- map : float, array-like or None An array containing the map, supports masked maps, see the `ma` function. If None, will display a blank map, useful for overplotting. fig : int or None, optional The figure number to use. Default: create a new figure rot : scalar or sequence, optional Describe the rotation to apply. In the form (lon, lat, psi) (unit: degrees) : the point at longitude *lon* and latitude *lat* will be at the center. An additional rotation of angle *psi* around this direction is applied. coord : sequence of character, optional Either one of 'G', 'E' or 'C' to describe the coordinate system of the map, or a sequence of 2 of these to rotate the map from the first to the second coordinate system. unit : str, optional A text describing the unit of the data. Default: '' xsize : int, optional The size of the image. Default: 800 ysize : None or int, optional The size of the image. Default: None= xsize reso : float, optional Resolution (in arcmin). Default: 1.5 arcmin lamb : bool, optional If True, plot Lambert azimuthal equal area instead of azimuthal equidistant. Default: False (az equidistant) half_sky : bool, optional Plot only one side of the sphere. Default: False title : str, optional The title of the plot. Default: 'Azimuthal equidistant view' or 'Lambert azimuthal equal-area view' (if lamb is True) nest : bool, optional If True, ordering scheme is NESTED. Default: False (RING) min : float, optional The minimum range value max : float, optional The maximum range value flip : {'astro', 'geo'}, optional Defines the convention of projection : 'astro' (default, east towards left, west towards right) or 'geo' (east towards roght, west towards left) remove_dip : bool, optional If :const:`True`, remove the dipole+monopole remove_mono : bool, optional If :const:`True`, remove the monopole gal_cut : float, scalar, optional Symmetric galactic cut for the dipole/monopole fit. Removes points in latitude range [-gal_cut, +gal_cut] format : str, optional The format of the scale label. Default: '%g' cbar : bool, optional Display the colorbar. Default: True notext : bool, optional If True, no text is printed around the map norm : {'hist', 'log', None} Color normalization, hist= histogram equalized color mapping, log= logarithmic color mapping, default: None (linear color mapping) cmap : a color map The colormap to use (see matplotlib.cm) badcolor : str Color to use to plot bad values bgcolor : str Color to use for background hold : bool, optional If True, replace the current Axes by an Equidistant AzimuthalAxes. use this if you want to have multiple maps on the same figure. Default: False sub : int, scalar or sequence, optional Use only a zone of the current figure (same syntax as subplot). Default: None reuse_axes : bool, optional If True, reuse the current Axes (should be a AzimuthalAxes). This is useful if you want to overplot with a partially transparent colormap, such as for plotting a line integral convolution. Default: False margins : None or sequence, optional Either None, or a sequence (left,bottom,right,top) giving the margins on left,bottom,right and top of the axes. Values are relative to figure (0-1). Default: None return_projected_map : bool if True returns the projected map in a 2d numpy array alpha : float, array-like or None An array containing the alpha channel, supports masked maps, see the `ma` function. If None, no transparency will be applied. See Also -------- mollview, gnomview, cartview, orthview r NrF)r)g!@gš™™™™™@)r)g{®Gáz”?gš™™™™™©?g{®Gáz”?gš™™™™™©?)gggð?gð?)ggggrrrr z Wrong values for sub: %d, %d, %d)g{®Gáz„?ggg{®Gáz”?gð?rr)rrr r!T)r"rr#)rr$r˜r™Úlambr¤rr%r&r'r(r)r*r+)r%r&)r*r'z0.91.0r,gà?r-gš™™™™™©?gš™™™™™¹?) r.r/r0r1r2r3r4r5r6r ) r/r0r1r2r3r4r5r6r z!Lambert azimuthal equal-area viewzAzimuthal equidistant viewg…ëQ¸…ë?r7r8)r9r:r;gð¿r<)r9r;r=r>)4r?r@rArBrrCrDrErFrGrHrIrJrKrLrMrNrOrQZHpxAzimuthalAxesrSrTrUrVrWr*rXrYr'rZr%r&r[r\r]r^r_r`rarbrcrdrerfrgrhrirjr.rkrlrm)6rnrorr¢rrpr$r˜r™r¥r¤rqrrurvr"rrrsrtr rxr'r(r)r*r1rzr}r{r|ryr~r+r?rr€rr‚rƒr„r…r†r‡rˆr‰rŠr‹r.rŒrrŽrrr‘r’r’r“r˜s                  c Ks¾ddl}| ¡}| ¡}| ¡z‚t| ¡ƒdkrhtj|d|d}| |¡|j dd|j j dd|j d x6| ¡D]*}t |tjƒrr|jf||||d œ|—ŽqrWWd| ¡|r¸| ¡XdS) aXDraw a graticule on the current Axes. Parameters ---------- dpar, dmer : float, scalars Interval in degrees between meridians and between parallels coord : {'E', 'G', 'C'} The coordinate system of the graticule (make rotation if needed, using coordinate system of the map if it is defined). local : bool If True, draw a local graticule (no rotation is performed, useful for a gnomonic view, for example) Notes ----- Other keyword parameters will be transmitted to the projplot function. See Also -------- delgraticules r N)g{®Gáz”?gš™™™™™©?g¸…ëQ¸î?gÍÌÌÌÌÌì?)rg…ëQ¸…ë?gš™™™™™©?r7r8)r9r:r;)ÚdparÚdmerrÚlocal)r?rFrNrOÚlenÚget_axesrQrRrSrgrhrirjÚ isinstanceÚSphericalProjAxesrrlrm) r¦r§rr¨Úkwdsr?r€r‹r.r’r’r“r¶s*   "cCsfddl}| ¡}| ¡}| ¡z*x$| ¡D]}t|tjƒr,| ¡q,WWd|  ¡|r`|  ¡XdS)zcDelete all graticules previously created on the Axes. See Also -------- graticule r N) r?rFrNrOrªr«rQr¬rrlrm)r?r€r‹r.r’r’r“ræs cOsnddl}| ¡}| ¡}| ¡d}z.x(| ¡D]}t|tjƒr0|j||Ž}q0WWd|  ¡|rh|  ¡X|S)Nr ) r?rFrNrOrªr«rQr¬rrlrm)Úargsr­r?r€r‹Úretr.r’r’r“rýs  cOsnddl}| ¡}| ¡}| ¡d}z.x(| ¡D]}t|tjƒr0|j||Ž}q0WWd|  ¡|rh|  ¡X|S)Nr ) r?rFrNrOrªr«rQr¬r rlrm)r®r­r?r€r‹r¯r.r’r’r“r s  cOsnddl}| ¡}| ¡}| ¡d}z.x(| ¡D]}t|tjƒr0|j||Ž}q0WWd|  ¡|rh|  ¡X|S)Nr ) r?rFrNrOrªr«rQr¬r rlrm)r®r­r?r€r‹r¯r.r’r’r“r )s  )NNNNrrrFNNrFFr rrTNrrFNFFNNrFN)NNNNrr”Nr•r–FFFr NNrr—TNrrNFNFNFFFN) NNNNNrrNNNržFFFr NNrr—TNrrNNFNFNFFN)NNNNrrFr£FNNrFFr rrTNrrFNFNNFFN)!NNNNNrrNr•FFNFFFr NNrr—TNrrNNFNFNFFN)NNNN)Ú__doc__Ú__all__rr rQÚnumpyr@r^Zmatplotlib.pyplotZpyplotr[rÚpiZdtorrrrrrrrrr¬r r r’r’r’r“Ú1sp    o   w o ~ 0