U cP @sfdZddlZddlZddlZddlZddZddZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZd\ddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!dd?d@dAd>d?dBZ#dCdDdEdFdGdHdIdJdKdLdM Z$d@dNdOdPdQdRdSdTdUdVdVdW Z%dXdYdZd[Z&dS)]zG Tools A module for converting from mpl language to plotly language. NcCsg}||d|dkf7}||d|dkf7}|ddkr|d|d}|d|d}|||dkf7}||d|dkf7}nT|ddkr|d |d}|d |d}|||dkf7}||d|dkf7}t|rd Sd Sd S) zCheck if two bars belong in the same collection (bar chart). Positional arguments: old_bar -- a previously sorted bar dictionary. new_bar -- a new bar dictionary that needs to be sorted. Z orientation facecolorvx1x0gư>y0hy1TFN)all)Zold_barZnew_bartestsZ new_widthZ old_widthZ new_heightZ old_heightr @/tmp/pip-unpacked-wheel-fki0jwev/plotly/matplotlylib/mpltools.pycheck_bar_matchs   r cCs|}|}|dd|ddkr4dS|dd|ddkrPdS|dd|ddkrldS|dd|ddkrdSdSdS)NrFT)Zget_window_extentZcorners)Z inner_objZ outer_objZ inner_cornersZ outer_cornersr r r check_corners)s  rcCs|tkrt|S|d}t|dkr*dStt|ddrBdSddd|D}|dkrdd }n|d krrd }n |d kr~d }|SdS)z@Convert mpl line symbol to plotly line symbol and return symbol.,solidrcSsg|] }|dqS)Zpxr ).0xr r r Isz convert_dash..z 7.4px,3.2pxZdashedz12.8px,3.2px,2.0px,3.2pxdashdotz 2.0px,3.3pxZdottedN)DASH_MAPsplitlenmathisclosefloatjoin)Zmpl_dashZ dash_arrayZdashpxr r r convert_dash8s  r cCs,|d}t|d}|tkr$t|SdSdS)Nrr)tuplePATH_MAP)pathZvertscoder r r convert_pathVs  r%cCsDt|tr,t}|D]}|t|g7}q|S|tkr (255, 255, 255) #c3s(|] }t||ddVqdS)rN)intrilvvaluer r zszhex_to_rgb..rr)lstriprr!range)r4r r2r hex_to_rgbls r8cCs:|dkr dSt|}|dkr&d|S||f}d|S)a Merge hex color with an alpha (opacity) to get an rgba tuple. :param (str|unicode) color: A hex color string. :param (float|int) opacity: A value [0, 1] for the 'a' in 'rgba'. :return: (int, int, int, float) The rgba color and alpha tuple. Nzrgb {}zrgba {})r8format)colorZopacityZrgb_tupZrgba_tupr r r merge_color_and_opacity}s   r;cCs|tkrt|SdSdS)aConvert mpl vertical alignment word to equivalent HTML word. Text alignment specifiers from mpl differ very slightly from those used in HTML. See the VA_MAP for more details. Positional arguments: mpl_va -- vertical mpl text alignment spec. N)VA_MAP)Zmpl_var r r convert_vas r=cCsX|d|d|dg}|d|d}|d|d|}|d|d|}||gS)aHMap x dimension of current plot to plotly's domain space. The bbox used to locate an axes object in mpl differs from the method used to locate axes in plotly. The mpl version locates each axes in the figure so that axes in a single-plot figure might have the bounds, [0.125, 0.125, 0.775, 0.775] (x0, y0, width, height), in mpl's figure coordinates. However, the axes all share one space in plotly such that the domain will always be [0, 0, 1, 1] (x0, y0, x1, y1). To convert between the two, the mpl figure bounds need to be mapped to a [0, 1] domain for x and y. The margins set upon opening a new figure will appropriately match the mpl margins. Optionally, setting margins=0 and simply copying the domains from mpl to plotly would place axes appropriately. However, this would throw off axis and title labeling. Positional arguments: mpl_plot_bounds -- the (x0, y0, width, height) params for current ax ** mpl_max_x_bounds -- overall (x0, x1) bounds for all axes ** ** these are all specified in mpl figure coordinates rrrr )mpl_plot_boundsZmpl_max_x_boundsZ mpl_x_domZplotting_widthrrr r r convert_x_domains r?cCsX|d|d|dg}|d|d}|d|d|}|d|d|}||gS)aHMap y dimension of current plot to plotly's domain space. The bbox used to locate an axes object in mpl differs from the method used to locate axes in plotly. The mpl version locates each axes in the figure so that axes in a single-plot figure might have the bounds, [0.125, 0.125, 0.775, 0.775] (x0, y0, width, height), in mpl's figure coordinates. However, the axes all share one space in plotly such that the domain will always be [0, 0, 1, 1] (x0, y0, x1, y1). To convert between the two, the mpl figure bounds need to be mapped to a [0, 1] domain for x and y. The margins set upon opening a new figure will appropriately match the mpl margins. Optionally, setting margins=0 and simply copying the domains from mpl to plotly would place axes appropriately. However, this would throw off axis and title labeling. Positional arguments: mpl_plot_bounds -- the (x0, y0, width, height) params for current ax ** mpl_max_y_bounds -- overall (y0, y1) bounds for all axes ** ** these are all specified in mpl figure coordinates rrrr )r>Zmpl_max_y_boundsZ mpl_y_domZplotting_heightrrr r r convert_y_domains r@cCsp||dd}|d|dd|dd}||dd}|d|dd|dd}||||fS)aConvert mpl display coordinates to plotly paper coordinates. Plotly references object positions with an (x, y) coordinate pair in either 'data' or 'paper' coordinates which reference actual data in a plot or the entire plotly axes space where the bottom-left of the bottom-left plot has the location (x, y) = (0, 0) and the top-right of the top-right plot has the location (x, y) = (1, 1). Display coordinates in mpl reference objects with an (x, y) pair in pixel coordinates, where the bottom-left corner is at the location (x, y) = (0, 0) and the top-right corner is at the location (x, y) = (figwidth*dpi, figheight*dpi). Here, figwidth and figheight are in inches and dpi are the dots per inch resolution. marginlwidthrbheighttr )ryZlayoutZnum_xZden_xZnum_yZden_yr r r display_to_papers   rIcCsggggf\}}}}|D]V}|j}||d||d|d||d||d|dqt|t|t|t|f\}}}}||f||ffS)asReturn the entire axes space for figure. An axes object in mpl is specified by its relation to the figure where (0,0) corresponds to the bottom-left part of the figure and (1,1) corresponds to the top-right. Margins exist in matplotlib because axes objects normally don't go to the edges of the figure. In plotly, the axes area (where all subplots go) is always specified with the domain [0,1] for both x and y. This function finds the smallest box, specified by two points, that all of the mpl axes objects fit into. This box is then used to map mpl axes domains to plotly axes domains. rrrr)Zget_axes get_positionboundsappendminmax)ZfigZx_minZx_maxZy_minZy_maxZaxes_objrKr r r get_axes_boundss  $rOcCs,|r |r dS|r|sdS|s$|r$dSdSdS)NZticksFr )Z main_spineZ mirror_spiner r r get_axis_mirror srP绽|=cstt|t|krpt|dkrp|dd}|dd}ddt||D}|dtfdd|D}|rpSdS)NrcSsg|]\}}||qSr r )rs2s1r r r rszget_bar_gap..rcsg|]}t|kqSr )abs)rZgapZgap0tolr r rs)rzipr )Z bar_startsZbar_endsrWZsides1Zsides2Zgapsuniformr rVr get_bar_gaps  rZcCst}|D]B}|tt|ddt|ddt|dd|ddg7}q t}|D]}|djf|g7}qXt|dkr|dS|SdS)Nrrrr)rDgrEazrgba({r},{g},{b},{a}))r(dictr/r9r)Z color_listZclean_color_listcZ plotly_colorsZrgbar r r convert_rgba_array#s6 r`cCs:t}|D]}|t|g7}q t|dkr2|dS|SdSNrr)r(r%r)Z path_arraysymbolsr#r r r convert_path_array2s  rccCst|dkr|dS|SdSrar)Z width_arrayr r r convert_linewidth_array<s recCs*dd|D}t|dkr"|dS|SdS)NcSsg|]}t|qSr )rsqrt)rr,r r r rDsz&convert_size_array..rrrd)Z size_arraysizer r r convert_size_arrayCs rhc CsPtdt|ddt|dt|ddt|dt|ddd}|S)NZstylesrpathsZ linewidthmplobj edgecolor)alpharmarker edgewidthZ markersizerk)r^r`rcrerhZ get_sizes)propsZ markerstyler r r get_markerstyle_from_collectionKs  rpcCs.t|dd|dd|dd|ddS)z.Find minimum x value from four (x,y) vertices.rrrrrMdatar r r get_rect_xminXsrtcCs.t|dd|dd|dd|ddS)z.Find maximum x value from four (x,y) vertices.rrrrrNrrr r r get_rect_xmax]srvcCs.t|dd|dd|dd|ddS)z.Find minimum y value from four (x,y) vertices.rrrrrqrrr r r get_rect_yminbsrwcCs.t|dd|dd|dd|ddS)z.Find maximum y value from four (x,y) vertices.rrrrrurrr r r get_rect_ymaxgsrxcCs|j|}|}|jpd}t|tr@|dkr4d}n |dkr@d}|\}}|dkr^|dkr^d}nd }|snd S|jd s|d S|s|rd S|r|rdS|s|sdSd Sd S) z8Return some spine parameters for the spine, `spine_key`.)outwardrcenter)axesg?Zzero)rsrryrTFrRN)ZspinesZ get_frame_on _positionr'strZ get_visibleZ _edgecolor)axZ spine_keyZspineZ ax_frame_onpositionZ position_typeamountZspine_frame_liker r r get_spine_visiblels.    rcKs |D]}|d|krdSqdS)zCA test to decide whether a path is a bar from a vertical bar chart.rjTFr )Zbar_containersro containerr r r is_bars rc Ksp|dt|dt|dt|dt|d|dd|dd|dd|dd|dd|dd d S) aMake an intermediate bar dictionary. This creates a bar dictionary which aids in the comparison of new bars to old bars from other bar chart (patch) collections. This is not the dictionary that needs to get passed to plotly as a data dictionary. That happens in PlotlyRenderer in that class's draw_bar method. In other words, this dictionary describes a SINGLE bar, whereas, plotly will require a set of bars to be passed in a data dictionary. rjrsstylerlrkrrn dasharrayzorder) barrrrrrlrkrrnrr)rtrwrvrx)ror r r make_bars           rc st}|dkr|}n|dkr*|}ntS|d|d}|dkrz|d|dd}fdd td td Dtfd d td td Drd d}ntd |t Wn,t t fk r|d|d |d <YnX||d<||d<d|d<n,|dkr@zP|d|dd|d<|d|dd |d|dd|d<d|d<Wn0t t fk rt|d|d d}YnX| j } | dkr"|dkrt |ddt |dd g|d<n2|dkr>t |ddt |dd g|d<ntddd}td|ntS|jj} |dkrd| krd|d<z2zt|d| |d<Wntk rYnXW5|dd|ddt|d| |d<X| dkrd|d<|S) zPrepare axis obj belonging to axes obj. positional arguments: ax - the mpl axes instance index - the index of the axis in `props` ax_type - 'x' or 'y' (for now) props - an mplexporter poperties dictionary rrHr{scaleZlinear tickvaluesrcs&g|]}t||ddqS)r )roundr0)rr r rszprep_ticks..rcs g|]}||dkqS)rr r0)dticksr r rszF'linear' {0}-axis tick spacing not even, ignoring mpl tick formatting.ntickstick0dtickNZtickmodelog)r xlimr7ylim)r7typez3Converted non-base10 {0}-axis log scale to 'linear'Z DateFormatterdaterZLogFormatterMathtexteZexponentformat)r^Z get_xaxisZ get_yaxisr7rr warningswarnr9 TypeError IndexErrorZ get_transformbaserlog10Zget_major_formatter __class____name__popmpl_dates_to_datestringsKeyError) r~indexZax_typeroZ axis_dictZaxisrrrr formatterr )rrr prep_tickss     $               rc Cst|dddt|d|ddddt|d||dddt|ddd d d }|t|dd |t|dd dt|d|dd ddt|d||dd dt|dd d d d }|t|d d|||fS)Nr{rrrZgridZgridOnrKrZfontsize)rg)rr7ZshowgriddomainZsideZtickfontrrrrH)r^r(r?updaterr@)r~roZx_boundsZy_boundsZxaxisZyaxisr r r prep_xy_axiss&    rcCsz|}|dkrFz&tjdd|D}tj|}Wqh|YSXn"ztj|}Wn|YSXdd|D}|S)zConvert matplotlib dates to iso-formatted-like time strings. Plotly's accepted format: "YYYY-MM-DD HH:MM:SS" (e.g., 2001-01-01 00:00:00) Info on mpl dates: http://matplotlib.org/api/dates_api.html ZTimeSeries_DateFormattercSsg|]}|dddqS)<r rrr r r r*sz,mpl_dates_to_datestrings..cSs*g|]"}d|dddqS) +rT)r isoformatrrr r r r8s) matplotlibdatesZ epoch2numZnum2date)rZ mpl_formatterZ_datesZ time_stingsr r r rs  rrZdashr&r)z10,0z6,6z2,2z4,4,2,4nonez7.4,3.2o*8rp1r,r^|) ) MCrrrrrrrZ) rLrrrrrrrrr) rrrrrrrrr)rrrrrrr)rrrrrr)rrrrrr)rrrrr)rrrr)rrrr)rrz triangle-downz triangle-upz triangle-leftztriangle-rightZsquareZcrossrZstarZdiamond) rrr<>r,rrrDdZmiddleZbottomtop)rzZbaseliner)rQ)'__doc__rdatetimerZmatplotlib.datesrr rr r%r)r8r;r=r?r@rIrOrPrZr`rcrerhrprtrvrwrxrrrrrrrr"r*r<r r r r s      Y$