B &[¨_žXã @sddlmZddlmZddlmZddlm Z m Z ddlm Z ddlmZe   d¡Z e   d¡Z ddlmZd d d d d dgfdd„Zd4dd„Zdd„Zdd„Zdd„Zdd„Zd5dd„Zdd „Zd6d!d"„Zd#d$„Zd%d&„Zd'd(„Zd7d-d.„Zd d dgd/d/dd0dd1dddd0f d2d3„ZdS)8é)Úabsolute_importN)Ú graph_objs)Ú exceptionsÚoptional_imports)rZnumpyzscipy.interpolate)ÚmeasurezTernary contour ploti&i ÚaÚbÚccCs`t|||tdtt|ddddddtt|ddddddtt|dddddddd d S) a] Layout of ternary contour plot, to be passed to ``go.FigureWidget`` object. Parameters ========== title : str or None Title of ternary plot width : int Figure width. height : int Figure height. pole_labels : str, default ['a', 'b', 'c'] Names of the three poles of the triangle. ér)Útextg{®Gáz„?éZoutside)ÚtitleÚminZ linewidthZticks)ÚsumZaaxisZbaxisZcaxisF)r ÚwidthÚheightZternaryÚ showlegend)Údict)r rrÚ pole_labels©rúJ/tmp/pip-install-l29rncou/plotly/plotly/figure_factory/_ternary_contour.pyÚ_ternary_layoutsrçü©ñÒMb@?cCsN|dk}tj|dd}d||}t |¡dkr8tdƒ‚t ||||¡}|S)aû Replaces zero ternary coordinates with delta and normalize the new triplets (a, b, c). Parameters ---------- ternary_data : ndarray of shape (N, 3) delta : float Small float to regularize logarithm. Notes ----- Implements a method by J. A. Martin-Fernandez, C. Barcelo-Vidal, V. Pawlowsky-Glahn, Dealing with zeros and missing values in compositional data sets using nonparametric imputation, Mathematical Geology 35 (2003), pp 253-278. r)Úaxisr zMThe provided value of delta led to negativeternary coords.Set a smaller delta)ÚnpÚanyÚ ValueErrorÚwhere)Z ternary_dataÚdeltaZ zero_maskZis_any_coord_zeroZunity_complementrrrÚ_replace_zero_coords9s rcCslt |¡}t |d|d¡t d¡}dt d¡t |d|d|dd¡}t ||f¡}|S)aÝ Perform Isometric Log-Ratio on barycentric (compositional) data. Parameters ---------- barycentric: ndarray of shape (3, N) Barycentric coordinates. References ---------- "An algebraic method to compute isometric logratio transformation and back transformation of compositional data", Jarauta-Bragulat, E., Buenestado, P.; Hervada-Sala, C., in Proc. of the Annual Conf. of the Intl Assoc for Math Geology, 2003, pp 31-30. rr r gð?é)rÚasarrayÚlogÚsqrtÚstack)Z barycentricZx_0Zx_1Z ilr_tdatarrrÚ_ilr_transform[s   0r%c CsÀt |¡}t dddgdddgdddgg¡}t d¡d}t d¡}t dt ||g| |gg¡|¡}t dt |¡jd d ¡ }t d|t d||d ||d|f¡¡}t |¡}|S) aø Perform inverse Isometric Log-Ratio (ILR) transform to retrieve barycentric (compositional) data. Parameters ---------- x : array of shape (2, N) Coordinates in ILR space. References ---------- "An algebraic method to compute isometric logratio transformation and back transformation of compositional data", Jarauta-Bragulat, E., Buenestado, P.; Hervada-Sala, C., in Proc. of the Annual Conf. of the Intl Assoc for Math Geology, 2003, pp 31-30. gà?r gð?gà¿gr gø?z ik, kj -> ijr)r)rÚarrayr#Úeinsumr"Úexprr$)ÚxZmatrixÚsÚtZSkÚZZlog_barycentricZ iilr_tdatarrrÚ _ilr_inversets " "* r-cCsdt dt d¡dgddgddgg¡}t |dd…df|dd…dft d¡g¡}|tj |¡fS)zi Returns the transformation matrix from barycentric to Cartesian coordinates and conversely. gà?ér rr N)rr&r#ÚonesZlinalgÚinv)Z tri_vertsÚMrrrÚ _transform_barycentric_cartesian’s&.r2cCsêt|ttjfƒstdƒ‚t |¡}|jddkr8tdƒ‚t|ƒdkr€tj|j ddddd s€tj|j ddd dd s€d }t|ƒ‚t|ƒd kr¢|\}}d||}n||j dd\}}}t  t  |||f¡dk¡rÚtd ƒ‚t  |||f¡S)zQ Check ternary coordinates and return the right barycentric coordinates. zOData should be either an array of shape (n,m),or a list of n m-lists, m=2 or 3r)r r.zCA point should have 2 (a, b) or 3 (a, b, c)barycentric coordinatesr.)rr g{®Gáz„?)Zrtolédz=The sum of coordinates should be 1 or 100 for all data pointsr z+Barycentric coordinates should be positive.) Ú isinstanceÚlistrZndarrayrr!ÚshapeÚlenZallcloserrr$)Zb_coordsÚmsgÚAÚBÚCrrrÚ_prepare_barycentric_coords&   r<ÚilrcCs|dkr"tƒ\}}t d||¡}n"|dkr ijr=z&interp_mode should be cartesian or ilrNr éÈZcubic)ÚmethodZnearest)r2rr'rr%rrÚmaxÚintr#r7ÚlinspaceZmeshgridÚ scipy_interpZgriddataÚT)Ú coordinatesÚvaluesÚ interp_moder1ÚinvMZ coord_pointsZxxÚyyZx_minZx_maxZy_minZy_maxZn_interpÚgr_xÚgr_yZgrid_xZgrid_yÚgrid_zZ grid_z_otherrrrÚ _compute_grid½s&  rNc Cs2dt t |t |d¡¡t |t |d¡¡¡S)Ngà?r )rÚabsÚdotZroll)r)ÚyrrrÚ _polygon_areaësrRcCs|tj ¡krtj|}nt d tj ¡¡¡‚t dd|¡}t dd„|Dƒ¡}t dd„|Dƒ¡}t  ||¡}d|dkrŒdd„|Dƒ}|dg}xtt |dd …|dd …ƒD]V\}} ||d||} } | | | | } tj ||d||| d d } |  | ¡q²W|S) zQ Return a list of ``ncontours`` colors from the ``colormap`` colorscale. zRColorscale must be a valid Plotly Colorscale.The available colorscale names are {}rr cSsg|] }|d‘qS)rr)Ú.0Úpairrrrú ûsz_colors..cSsg|] }|d‘qS)r r)rSrTrrrrUüsú#cSsg|]}t t |¡¡‘qSr)ÚclrsZ label_rgbZ hex_to_rgb)rSÚcolrrrrUÿsNZrgb)Z colortype) rWZ PLOTLY_SCALESÚkeysrZ PlotlyErrorÚformatrrCr&Z searchsortedÚzipZfind_intermediate_colorÚappend)Ú ncontoursZcolormapZcmaprGZ vals_cmapÚcolsZindsÚcolorsÚindÚvalZval1Zval2ZintermrXrrrÚ_colorsïs&    $rbcCs<t t ||d¡dk¡o6t t ||d¡dk¡}|S)zy Utility function for _contour_trace Contours with an area of the order as 1 pixel are considered spurious. rr )rÚallrO)r)rQZ too_smallrrrÚ_is_invalid_contour s8rdcCstt |¡}|t |¡ ¡|t |¡ ¡}}t |¡}d|||<t |¡}d|||<ggggf\}} } } ggggf\} } }}xÀt|ƒD]´\}}t ||¡}t ||¡}|  |¡|   |¡|   |gt |ƒ¡|   |gt |ƒ¡|   dd„|Dƒ¡|  dd„|Dƒ¡|   ||gt |ƒ¡|  ||gt |ƒ¡qŽWt |ƒt | ƒkrd|| | | fS| | ||fSdS)a{ Utility function for _contour_trace. In ``im`` only one part of the domain has valid values (corresponding to a subdomain where barycentric coordinates are well defined). When computing contours, we need to assign values outside of this domain. We can choose a value either smaller than all the values inside the valid domain, or larger. This value must be chose with caution so that no spurious contours are added. For example, if the boundary of the valid domain has large values and the outer value is set to a small one, all intermediate contours will be added at the boundary. Therefore, we compute the two sets of contours (with an outer value smaller of larger than all values in the valid domain), and choose the value resulting in a smaller total number of contours. There might be a faster way to do this, but it works... r cSs"g|]}t|jd|jdƒ‘qS)r r)rRrE)rSÚcontourrrrrU;sz%_extract_contours..cSs"g|]}t|jd|jdƒ‘qS)r r)rRrE)rSrerrrrU>sN) rÚisnanZ logical_notrrAÚcopyÚ enumeraterZ find_contoursÚextendr7)ZimrGr_Zmask_nanZim_minZim_maxZzz_minZzz_maxZ all_contours1Z all_values1Z all_areas1Z all_colors1Z all_contours2Z all_values2Z all_areas2Z all_colors2ÚiraZcontour_level1Zcontour_level2rrrÚ_extract_contourss2          rkc sndt dddgdddgg¡j} | g|}t |¡d} t |d| g||d| gf¡}t | g|| gf¡}t ||k¡dd} | t|ƒdkr | d8} n| d7} || gˆ‰|| g|}dg|}‡fdd„|Dƒ}t|ƒ}t dd|d¡}g}xFt|ƒD]:\}}|  ||||g¡|  ||d||g¡qW|  ||||dg¡|||ˆ|fS) a Utility function for _contour_trace Adds the background color to fill gaps outside of computed contours. To compute the background color, the color of the contour with largest area (``val_outer``) is used. As background color, we choose the next color value in the direction of the extrema of the colormap. Then we add information for the outer contour for the different lists provided as arguments. A discrete colormap with all used colors is also returned (to be used by colorscale trace). érr gà?éÿÿÿÿr csg|]}|ˆkr|‘qSrr)rSÚcolor)Ú all_colorsrrrUtsz&_add_outer_contour..) rr&rEZdiffÚ concatenateZnonzeror7rCrhr\)Ú all_contoursÚ all_valuesÚ all_areasrorGZ val_outerÚv_minÚv_maxr_Ú color_minÚ color_maxZ outer_contourZ delta_valuesÚindexZ used_colorsZ color_numberZscaleÚ discrete_cmrjrnr)rorÚ_add_outer_contourHs,      rzÚElectricúrgb(150,150,150)Úllrr c #CsXt|d|ƒ} t || |d¡} | d| d} } | dd…} | dd…} |dkrXd}n |g|} t|| | ƒ\}}}}t |¡ddd…}t||||| ||d|| | | | ƒ \}}}}}t dg|df¡}g}tƒ\}}| ¡|  ¡|j }| ¡|  ¡|j }xF|D]<}||j \}}||}|dkr^t  |t  ||||t |j¡f¡¡}n4|dkr’tt  |||  ¡|||  ¡f¡ƒ}|dkrÎt dddg¡}t dddg¡}t dddg¡} n |\}}} t||ƒrèq|d krú||n|}!td ||| d t|!d dd d ||ddd|d }"|d kr@d|"d<| |"¡qW||fS)aî Contour trace in Cartesian coordinates. Parameters ========== x, y : array-like Cartesian coordinates z : array-like Field to be represented as contours. ncontours : int or None Number of contours to display (determined automatically if None). colorscale : None or str (Plotly colormap) colorscale of the contours. linecolor : rgb color Color used for lines. If ``colorscale`` is not None, line colors are determined from ``colorscale`` instead. interp_mode : 'ilr' (default) or 'cartesian' Defines how data are interpolated to compute contours. If 'irl', ILR (Isometric Log-Ratio) of compositional data is performed. If 'cartesian', contours are determined in Cartesian space. coloring : None or 'lines' How to display contour. Filled contours if None, lines if ``lines``. vmin, vmax : float Bounds of interval of values used for the colorspace Notes ===== r rrmr Nzrgb(150, 150, 150)r>r=ÚlinesÚscatterternaryZspline)rnr6rZtoselfTÚskipz%.3f) Útyperrr ÚmodeÚlineÚfillZ fillcolorrZ hoverinfoÚnamer„)rbrrCrkZargsortrzrpr2rArÚsizerErPr$r/r6r-r&rdrr\)#r)rQÚzr]Ú colorscaleÚ linecolorrHÚcoloringrtrur_rGrvrwrqrrrsroÚorderryZtracesr1rIZdxZdyrxZ y_contourZ x_contourraZ bar_coordsrrr Ú_colÚtracerrrÚ_contour_traces|,      $ (      rŽiôFZBlueredc  Csbtdkrtdƒ‚t d¡} | dkr*tdƒ‚|dkr6d}|dkrBd}t|ƒ}| ¡| ¡}}t||| d\}}}t|||| d}t |||||| | |||d \}}t j ||d }| r¶d nd }|\}}}|d d |d d|dd}|j |||d||dt |dkƒdœdœ||d|r^| s&|}tddgdgdg| ¡| ¡|ddœddœƒ}| |¡|S)a´ Ternary contour plot. Parameters ---------- coordinates : list or ndarray Barycentric coordinates of shape (2, N) or (3, N) where N is the number of data points. The sum of the 3 coordinates is expected to be 1 for all data points. values : array-like Data points of field to be represented as contours. pole_labels : str, default ['a', 'b', 'c'] Names of the three poles of the triangle. width : int Figure width. height : int Figure height. ncontours : int or None Number of contours to display (determined automatically if None). showscale : bool, default False If True, a colorbar showing the color scale is displayed. coloring : None or 'lines' How to display contour. Filled contours if None, lines if ``lines``. colorscale : None or str (Plotly colormap) colorscale of the contours. linecolor : None or rgb color Color used for lines. ``colorscale`` has to be set to None, otherwise line colors are determined from ``colorscale``. title : str or None Title of ternary plot interp_mode : 'ilr' (default) or 'cartesian' Defines how data are interpolated to compute contours. If 'irl', ILR (Isometric Log-Ratio) of compositional data is performed. If 'cartesian', contours are determined in Cartesian space. showmarkers : bool, default False If True, markers corresponding to input compositional points are superimposed on contours, using the same colorscale. Examples ======== Example 1: ternary contour plot with filled contours >>> import plotly.figure_factory as ff >>> import numpy as np >>> # Define coordinates >>> a, b = np.mgrid[0:1:20j, 0:1:20j] >>> mask = a + b <= 1 >>> a = a[mask].ravel() >>> b = b[mask].ravel() >>> c = 1 - a - b >>> # Values to be displayed as contours >>> z = a * b * c >>> fig = ff.create_ternary_contour(np.stack((a, b, c)), z) >>> fig.show() It is also possible to give only two barycentric coordinates for each point, since the sum of the three coordinates is one: >>> fig = ff.create_ternary_contour(np.stack((a, b)), z) Example 2: ternary contour plot with line contours >>> fig = ff.create_ternary_contour(np.stack((a, b, c)), z, coloring='lines') Example 3: customize number of contours >>> fig = ff.create_ternary_contour(np.stack((a, b, c)), z, ncontours=8) Example 4: superimpose contour plot and original data as markers >>> fig = ff.create_ternary_contour(np.stack((a, b, c)), z, coloring='lines', ... showmarkers=True) Example 5: customize title and pole labels >>> fig = ff.create_ternary_contour(np.stack((a, b, c)), z, ... title='Ternary plot', ... pole_labels=['clay', 'quartz', 'fledspar']) NzH The create_ternary_contour figure factory requires the scipy packageÚskimagezS The create_ternary_contour figure factory requires the scikit-image packageFé)rH)rrrr )r]rˆr‰rHrŠrtru)ÚdataÚlayoutr rz: %{a:.3f}
z: %{b:.3f}
r z3: %{c:.3f}
z: %{marker.color:.3f}Úmarkerszrgb(120, 120, 120)r~)rnr)rnrˆrƒ)rrr r‚ÚmarkerÚopacityÚ hovertemplaterT)ZcminZcmaxrˆÚ showscale)rrrr r”r‚)rDÚ ImportErrorrÚ get_moduler<rrArNrrŽÚgoZFigureZadd_scatterternaryrBrZ add_trace)rFrGrrrr]r—rŠrˆr‰r rHZ showmarkersZ sk_measurertrurMrKrLr’Z contour_traceryZfigr•rrr r–ZcolorbarrrrÚcreate_ternary_contourspa    $  r›)r)r=)N)Nr{r|r}Nrr )Ú __future__rZ plotly.colorsr_rWZplotly.graph_objsrršZplotlyrrr™rrDrrrrr%r-r2r<rNrRrbrdrkrzrŽr›rrrrÚsL        ( "  .  2= ~