B z‹d&wã3@sêddgZddlZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZddlmZdd lmZddlZGd d „d ejƒZGd d „d ejƒZGdd„dejƒZGdd„dejƒZdd„Zdd„Zd(d$d„Zd)d%d„Zd*d&d'„ZdS)+ÚprojviewÚ newprojplotéNé)Úang2pixÚ npix2nsideÚ remove_dipoleÚremove_monopole)ÚRotatorÚcoordsys2euler_zyz)Úget_color_table)ÚGeoAxes)ÚMultipleLocatorcs"eZdZdZd‡fdd„ Z‡ZS)Ú!ThetaFormatterCounterclockwisePhiz~Convert tick labels from rads to degs and shifts labelling from -180|-90|0|90|180 to conterclockwise periodic 180|90|0|270|180Ncs8|dkr|d9}|dkr&|dtj7}tt|ƒ ||¡S)Nréÿÿÿÿé)ÚnpÚpiÚsuperrÚ__call__)ÚselfÚxÚpos)Ú __class__©ú_/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/healpy/newvisufunc.pyrs z*ThetaFormatterCounterclockwisePhi.__call__)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ __classcell__rr)rrr srcs"eZdZdZd‡fdd„ Z‡ZS)ÚThetaFormatterClockwisePhizxConvert tick labels from rads to degs and shifts labelling from -180|-90|0|90|180 to clockwise periodic 180|270|0|90|180Ncs(|dkr|dtj7}tt|ƒ ||¡S)Nrr)rrrr r)rrr)rrrrsz#ThetaFormatterClockwisePhi.__call__)N)rrrrrrrr)rrr sr cs"eZdZdZd‡fdd„ Z‡ZS)ÚThetaFormatterSymmetricPhizKJust convert phi ticks from rad to degs and keep the true -180|-90|0|90|180Ncstt|ƒ ||¡S)N)rr!r)rrr)rrrr&sz#ThetaFormatterSymmetricPhi.__call__)N)rrrrrrrr)rrr!#sr!cs"eZdZdZd‡fdd„ Z‡ZS)ÚThetaFormatterThetaz%Convert theta ticks from rads to degsNcstt|ƒ ||¡S)N)rr"r)rrr)rrrr-szThetaFormatterTheta.__call__)N)rrrrrrrr)rrr"*sr"cCs&t |¡}tjdt |¡}||fS)z0Converts theta and phi to longitude and latituder)rZasarrayr)ÚthetaÚphiÚ longitudeÚlatituderrrÚlonlat1s r'cCs.x(| ¡D]\}}||kr ||||<q W|S)N)Úitems)Z main_dictZ update_dictÚkeyZkey_valrrrÚupdate_dictionary9sr*ÚéèFÚastroú%gTÚviridisÚ mollweideÚ horizontalé<éÚblackÚcounterclockwiseéoc3d s² ddddg}4|dkr&ˆdk r&t ˆ¡}| dkr@ˆdk r@t ˆ¡} dddd œ}5|dk r^t|5|ƒ}5|0rtt||2|d d }n|1rˆt||2|d d }d d „}6ddddddddœ}7|$dk r¶t|7|$ƒ}7t |6|ƒ|6| ƒg¡}8|8dkrÚd}9nd|8}9d}:|};|dkr$|dkr d}|dk r t d0t|>ƒd1¡t|>|ƒ}>t d2t|>ƒd1¡|"dkr®d3n|"}?d4|?d5d5||d6œ}@|dk rt d7t|@ƒd1¡t|@|ƒ}@t d8t|@ƒd1¡|spt |,d9ƒr"|,\}A}B}Cn |,d:|,d:d|,d}A}B}C|CdksZ|C|B|Akrlt d;|A|B|Cfƒ‚|/s¨|-s¨|,dd=|>d=|>d>fd?}d@}Dnœ|/røt   ¡}t | ¡ ¡¡ ¡\}E}F}G}H|E|F|G|E|H|Ff}D| | ¡¡nL|-rt   ¡}nd=|>d=|>d>|A|Bfd?}nt   ¡}|d#kr^| |A|B|C¡}In|j|A|B|C|dA}Id}EdB}GdC}Hd }F|r”|r”|Ed7}E|dD}Jt tjd|J¡}Kt tj tj|¡}Lt t dEdF|¡¡}M| dGkrì|MdddH…}M|s t j|E|G|H|FdI| |I_t t dJd+|J¡¡}Nt |L|K¡\}O}P|s:|rvt|||+dK}Q|Q|P ¡|O ¡ƒ\}P}O|P |J|¡}P|O |J|¡}Ott|ƒƒ}R|dk r†t  |¡t !|¡B}S|dk rØ|dkrÂ||S ¡}| dkrØ||S ¡} t"|| ||Sf| |dLœ|5—Ž\}T}Ut#|R|P|O|dM}V||V}W|r"|M|N|WgS|dkrNt j$|M|N|Wf|Ud |TdNdOœ|3—Ž}Xn8|dkr†t |M|N¡\}Y}Z|Ij%|Y|Z|Wf|T|Ud dPœ|3—Ž}X|s–|dk ršdQ}|sª|dkrÎ|"dkrÀt  &|¡nt j&||"dR|r6||4kr|}|I '|¡|I (|¡|I )d+¡n4|}|}|Ij* +t,t -|¡ƒ¡|Ij. +t,t -|¡ƒ¡||@r.|%dSkrTt/|ƒ}[n&|%dTkrht0|ƒ}[n|%dUkrzt1|ƒ}[|Ij* 2|[¡|Ij. 2t3|ƒ¡|&dk rây|Ij*j4|&|#dVWn,t dWtt|Ij* 5¡ƒƒdX¡YnX|'dk r.y|Ij.j4|'|#dVWn,t dWtt|Ij. 5¡ƒƒdY¡YnX|r:|sb|Ij* 4g¡|Ij. 4g¡|Ij6dZdZdd[|Ij7||7d\|#d]|Ij6d^|7d_| d`|Ij6da|7db|!d`|d#kr²|I 8d¡| r˜ˆdkrÊ|| g‰|*dk r$dc}*|t |¡krìdd}*| t |¡k rde}*|t |¡k r$| t |¡k r$dZ}*|) r.dnˆ}\|j9|X||>df|>dg|\|*dh}]|) r¼t:t;|] <¡ƒt;ˆƒBƒ}\t =|\¡}\|\|\|k}\|\|\| k}\‡‡fdidj„|\Dƒ}^|] >|\|^¡|] 4|^¡n‡fdkdj„ˆDƒ}^|dk r(|]j?j@|^|#dV|]j?j*jA||7dl|#d]|]j?j6d^|7dm|>dndo|>dp|]j?j*_B|dk rŒ|]j?jC|^|>d*dq|#dr|]j?j.jA||7dld+|#ds|]j?j6da|7dm|>dndo|>dp|]j?j._B|]jD Edt¡|IjF||7du|#d]|IjG||7dv|#d]| sÐ|dk rR|dk rÞdw}tH||@dx|@dydz\}_}`xXtI|_ƒD]L\}a}b|a|`k r"|@d{d|}cn|@d{}ct jJ|b|c|@d}|@d~|@dd€œŽ qW|dk r|t jKdd‚|dƒdq|7dl|#|IjLd„|dk r¦t jKd…d‚|d†dq|7dl|#|IjLd„t  M¡|XS)‡a‚Plot a healpix map (given as an array) in the chosen projection. See examples of using this function in the documentation under "Other tutorials". Overplot points or lines using :func:`newprojplot`. .. warning:: this function is work in progress, the aim is to reimplement the healpy plot functions using the new features of matplotlib and remove most of the custom projection code. Please report bugs or submit feature requests via Github. The interface will change in future releases. Parameters ---------- m : 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. default: 'G' unit : str, optional A text describing the unit of the data. Default: '' xsize : int, optional The size of the image. Default: 800 width : float, optional Sets the width of the figure. Use override_plot_properties for more. Overrides the default width of the figure 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) It creates the `healpy_flip` attribute on the Axes to save the convention in the figure. format : str, optional The format of the scale label. Default: '%g' cbar : bool, optional Display the colorbar. Default: True cmap : str, optional Specify the colormap. default: Viridis norm : {'hist', 'log', 'symlog', 'symlog2', None} Color normalization: hist = histogram equalized color mapping. log = logarithmic color mapping. symlog = symmetric logarithmic, linear between -linthresh and linthresh. symlog2 = similar to symlog, used for plack log colormap. default: None (linear color mapping) norm_dict : dict, optional Parameters for normalization: default is set to {"linthresh": 1, "base": 10, "linscale": 0.1} where linthresh determines the linear regime of symlog norm, and linscale sets the size of the linear regime on the cbar. default: None graticule : bool add graticule graticule_labels : bool longitude and latitude labels rot_graticule : bool rotate also the graticule when rotating the map graticule_coord : str Either one of 'G', 'E' or 'C' to describe the coordinate system of the graticule override_rot_graticule_properties : dict Override the following rotated graticule properties: "g_linestyle", "g_linewidth", "g_color", "g_alpha", "t_step", "p_step". return_only_data : bool, optional Return array of data projection_type : {'aitoff', 'hammer', 'lambert', 'mollweide', 'cart', '3d', 'polar'} type of the plot cb_orientation : {'horizontal', 'vertical'} color bar orientation xlabel : str set x axis label ylabel : str set y axis label longitude_grid_spacing : float set x axis grid spacing latitude_grid_spacing : float set y axis grid spacing override_plot_properties : dict Override the following plot properties: "cbar_shrink", "cbar_pad", "cbar_label_pad", "cbar_tick_direction", "vertical_tick_rotation" "figure_width": width, "figure_size_ratio": ratio. title : str set title of the plot rlabel : str set label at top right corner of axis llabel : str set label at top left corner of axis xtick_label_color : str Change the color of the graticule xticks ytick_label_color : str Change the color of the graticule yticks graticule_color : str Change the color of the graticule fontname : str Change the fontname of the text fontsize: dict Override fontsize of labels: "xlabel", "ylabel", "title", "xtick_label", "ytick_label", "cbar_label", "cbar_tick_label". phi_convention : string convention on x-axis (phi), 'counterclockwise' (default), 'clockwise', 'symmetrical' (phi as it is truly given) if `flip` is "geo", `phi_convention` should be set to 'clockwise'. custom_xtick_labels : list override x-axis tick labels custom_ytick_labels : list override y-axis tick labels cbar_ticks : list custom ticks on the colorbar show_tickmarkers : bool, optional Preserve tickmarkers for the full bar with labels specified by ticks default: None extend : str, optional Whether to extend the colorbar to mark where min or max tick is less than the min or max of the data. Options are "min", "max", "neither", or "both" invRot : bool invert rotation sub : int, scalar or sequence, optional Use only a zone of the current figure (same syntax as subplot). Default: 111 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 hold : bool, optional If True, replace the current Axes by new axis. use this if you want to have multiple maps on the same figure. Default: False 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] kwargs : dict any leftover arguments will be passed to pcolormesh ZaitoffÚhammerZlambertr0Nré gš™™™™™¹?)Z linthreshÚbaseZlinscaleT)Úgal_cutÚnestÚcopycSs&ytt|ƒ d¡dƒSdSdS)NÚ.rr)ÚlenÚstrÚsplit)ÚnumberrrrÚfind_number_of_decimals%sz)projview..find_number_of_decimalsé é)ÚxlabelÚylabelÚtitleÚ xtick_labelÚ ytick_labelÚ cbar_labelÚcbar_tick_labeléiåÿÿÿi÷ÿÿÿg)\Âõ(ä?Z3dÚverticalgš™™™™™á?g{®Gáz”?g'@r1gš™™™™™É?riöÿÿÿg333333ã?g{®Gáz„?gš™™™™™©?iøÿÿÿg!@Zcartg333333#@gáz®GáÚ?gš™™™™™Ù?iôÿÿÿgš™™™™™!@ZpolarÚvertical_tick_rotationéZééÚout)Ú cbar_shrinkÚcbar_padÚcbar_label_padÚcbar_tick_directionrNÚ figure_widthÚfigure_size_ratioz) *** Overriding default plot properies: z ***z *** New plot properies: Zgreyú-gè?)Ú g_linestyleÚg_colorÚg_alphaÚ g_linewidthÚt_stepÚp_stepz. *** Overriding rotated graticule properies: z' *** New rotated graticule properies: Ú__len__édz Wrong values for sub: %d, %d, %dr6rWrX)Zfigsize)g{®Gáz”?gš™™™™™©?g¸…ëQ¸î?gÍÌÌÌÌÌì?)Z projectiong\Âõ(\ï?gffffffî?riLÿÿÿé´r-r)ÚleftÚrightÚtopÚbottomi¦ÿÿÿ)ÚcoordÚrotÚinv)ÚcmapÚnorm)r;Úauto)rkÚ rasterizedrjZshading)rjrkrmF)Úcolorr5Z clockwiseZ symmetrical)ÚfontnamezPut names for all z& x-tick labels!. No re-labelling done.z& y-tick labels!. No re-labelling done.Zboth)ÚaxisÚwhichÚlengthrG)ÚfontsizerorrH)rpÚ labelsizeÚcolorsÚyrIZneitherÚminÚmaxrSrT)Z orientationÚshrinkÚpadÚticksÚextendcs g|]}|ˆkrˆ|nd‘qS)r+r)Ú.0Útick)Ú cbar_ticksÚformatrrú §szprojview..csg|] }ˆ|‘qSrr)r}r~)r€rrr¬srJrKrV)rprtÚ directionrUÚcenter)ÚrotationÚvaro)rsr„roZfacerErFÚGr^r_)r^r_r]g@rZr[r\)Ú linewidthZ linestylernÚalphag333333ï?gš™™™™™í?rd)Zhar…rsroZ transformgš™™™™™™?rc)Nrrwrxr*rrÚwarningsÚwarnr?ÚhasattrÚ ValueErrorÚpltZfigureZgcfÚarrayÚgcaZ get_positionZravelZdelaxesZ get_fignumsZ add_subplotÚlinspacerÚradiansZsubplots_adjustÚ healpy_flipZmeshgridr ÚflattenZreshaperr>ÚisnanÚisinfr rZ pcolormeshZ plot_surfaceÚgridZset_longitude_gridZset_latitude_gridZset_longitude_grid_endsZxaxisZset_major_locatorr Údeg2radZyaxisrr r!Zset_major_formatterr"Zset_ticklabelsZget_ticklabelsZ tick_paramsÚ set_titleZ set_aspectZcolorbarÚlistÚsetZ get_ticksÚsortZ set_ticksÚaxZset_xticklabelsZset_label_textZlabelpadZset_yticklabelsZsolidsZ set_edgecolorZ set_xlabelZ set_ylabelÚCreateRotatedGraticuleÚ enumerateÚplotÚtextZ transAxesZdraw)dÚmZfigrhrgÚunitZxsizeÚwidthr;rwrxÚflipr€ZcbarrjrkZ norm_dictZ graticuleZgraticule_labelsZ rot_graticuleZgraticule_coordZ!override_rot_graticule_propertiesZreturn_only_dataZprojection_typeZcb_orientationrErFZlongitude_grid_spacingZlatitude_grid_spacingZoverride_plot_propertiesrGZrlabelZllabelZxtick_label_colorZytick_label_colorZgraticule_colorrorsZphi_conventionZcustom_xtick_labelsZcustom_ytick_labelsrZshow_tickmarkersr|ZinvRotÚsubZ reuse_axesZmarginsZholdZ remove_dipZ remove_monor:ÚkwargsZgeographic_projectionsZnorm_dict_defaultsrBZfontsize_defaultsZdecsZlpadÚratioZ custom_widthryrzZplot_propertiesZg_colZrot_graticule_propertiesZnrowsZncolsÚidxZextentrcrfrdrerœZysizer#r$r%r&ZPHIZTHETAÚrZnsideÚwÚcmÚnnZgrid_pixZgrid_mapÚretZ LONGITUDEZLATITUDEZxtick_formatterr{ÚcbÚlabelsÚrotated_grid_linesÚ where_zeroÚiZg_liner‡r)rr€rr@s¦R                                                                 " "                        c Ksnddlm}| ¡}t|ddƒ}t||ƒ\}}|dkr>|d}|dkrX|j||f|Ž} n|j|||f|Ž} | S)a¿newprojplot is a wrapper around :func:`matplotlib.Axes.plot` to support colatitude theta and longitude phi and take into account the longitude convention (see the `flip` keyword of :func:`projview`) You can call this function as:: newprojplot(theta, phi) # plot a line going through points at coord (theta, phi) newprojplot(theta, phi, 'bo') # plot 'o' in blue at coord (theta, phi) Parameters ---------- theta, phi : float, array-like Coordinates of point to plot in radians. fmt : str A format string (see :func:`matplotlib.Axes.plot` for details) Notes ----- Other keywords are passed to :func:`matplotlib.Axes.plot`. rNr’r-r)Úmatplotlib.pyplotÚpyplotrÚgetattrr'rŸ) r#r$Úfmtr¦rrœr¤r%r&r­rrrr s  c Csð|d}y |d}Wnd}YnXd}tjd}t dd||¡}t dd||¡}t t |dk¡d|jt |dk¡df¡} t t dd|¡¡} t t dd|¡¡|} g} x`|D]X} t t |¡| ¡|}t d|fd d }||| ƒ\}}||}||}|   ||g¡q¾Wxt|D]l}t t |¡|¡}t |dfd d }|| |ƒ\}}t d|fd d }|||ƒ\}}||}|   ||g¡q WxV| D]N}t t  t  |d¡¡t d ¡k¡}tj  |d¡|d<tj j|d|<q–W| | fS) Nrrrari¦ÿÿÿrOiLÿÿÿrbT)rhrié-)rrZarangeZhstackÚwhereÚsizer—rZzerosr ÚappendÚabsÚdiffÚmarŽZmasked)rhr^r_r$r#Z pointDensityZconventionThetaOffsetZ thetaSpacingZ phiSpacingr±Zgline_phi_fixedZgline_theta_fixedr°Z thetaSpaceZ gline_thetar©Zgline_theta_rotZgline_phi_fixed_rotZphiSpaceZ gline_phiZgline_theta_fixed_rotZ gline_phi_rotZg_linesÚmaskrrrr/sF      $r)3NNNNr+r,NFNNr-r.Tr/NNFFFNNFr0r1NNr2r3NNNNr4r4NNNr5NNNFNTr6FNFFFr)N)r3r3)Ú__all__ÚnumpyrZ pixelfuncrrrrÚrotatorr r Zprojaxesr r³r´rZmatplotlib.projections.geor Zmatplotlib.tickerr r‰ZThetaFormatterrr r!r"r'r*rrrrrrrÚs        $