U SZf6X @sddlmZddlmZddlmZddlmZe dZ e dZ ddl m Z dd d d d d gfddZd3ddZddZddZddZddZd4ddZddZd5d d!Zd"d#Zd$d%Zd&d'Zd6d,d-Zd d d gd.d.dd/dd0dddd/f d1d2ZdS)7N) graph_objs) exceptions)optional_importsZnumpyzscipy.interpolate)measurezTernary contour ploti&i abccCs`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)text{Gz?Zoutside)titleminZ linewidthZticks)sumZaaxisZbaxisZcaxisF)r widthheightZternary showlegend)dict)r rr pole_labelsrJ/tmp/pip-unpacked-wheel-5ksk5baj/plotly/figure_factory/_ternary_contour.py_ternary_layouts6    rMb@?cCsN|dk}tj|dd}d||}t|dkr8tdt||||}|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. rZaxisr zMThe provided value of delta led to negativeternary coords.Set a smaller delta)npany ValueErrorwhere)Z ternary_datadeltaZ zero_maskZis_any_coord_zeroZunity_complementrrr_replace_zero_coords7s rcCslt|}t|d|dtd}dtdt|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 ?)rasarraylogsqrtstack)Z barycentricZx_0Zx_1Z ilr_tdatarrr_ilr_transformYs   .r&c Cst|}tdddgdddgdddgg}tdd}td}tdt||g| |gg|}tdt|jd d  }td|td||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. ?r r ggr g? ik, kj -> ijrr)rarrayr$einsumr#exprr%)xZmatrixstZSkZZlog_barycentricZ iilr_tdatarrr _ilr_inversers " "" r0cCsdtdtddgddgddgg}t|dddf|dddftdg}|tj|fS)zi Returns the transformation matrix from barycentric to Cartesian coordinates and conversely. r'r rr N)rr)r$onesZlinalginv)Z tri_vertsMrrr _transform_barycentric_cartesians&.r5cCst|ttjfstdt|}|jddkr8tdt|dkrtj|j ddddd stj|j ddd dd sd }t|t|d kr|\}}d||}n||j dd\}}}t t |||fdkrtd t |||fS)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 r1zCA point should have 2 (a, b) or 3 (a, b, c)barycentric coordinatesr1rr r )Zrtoldz=The sum of coordinates should be 1 or 100 for all data pointsr z+Barycentric coordinates should be positive.) isinstancelistrZndarrayrr"shapelenZallcloserrr%)Zb_coordsmsgABCrrr_prepare_barycentric_coords0   r?ilrcCs|dkr"t\}}td||}n"|dkrsz_colors..cSsg|] }|dqS)r rrVrrrrYs#cSsg|]}tt|qSr)clrsZ label_rgbZ hex_to_rgb)rWcolrrrrYsNZrgb)Z colortype) r[Z PLOTLY_SCALESkeysrZ PlotlyErrorformatrrFr)Z searchsortedzipZfind_intermediate_colorappend) ncontoursZcolormapZcmaprJZ vals_cmapcolsZindscolorsindvalZval1Zval2Zintermr\rrr_colorss2    "  rfcCs<tt||ddko6tt||ddk}|S)zy Utility function for _contour_trace Contours with an area of the order as 1 pixel are considered spurious. rr )rallrR)r,rTZ too_smallrrr_is_invalid_contour s8rhcCspt|}|t||t|}}t|}d|||<t|}d|||<ggggf\}} } } ggggf\} } }}t|D]\}}t||}t||}| || || |gt || |gt || dd|D| dd|D| ||gt || ||gt |qt |t | kr`|| | | 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|jdqSr rrUrHrWZcontourrrrrY9sz%_extract_contours..cSs"g|]}t|jd|jdqSrirjrkrrrrY<sN) risnanZ logical_notrrDcopy enumeraterZ find_contoursextendr:)ZimrJrcZmask_nanZim_minZim_maxZzz_minZzz_maxZ all_contours1Z all_values1Z all_areas1Z all_colors1Z all_contours2Z all_values2Z all_areas2Z all_colors2ireZcontour_level1Zcontour_level2rrr_extract_contourss8            rqc sjdtdddgdddggj} | g|}t|d} t|d| g||d| gf}t| g|| gf}t||kdd} | t|dkr| d8} n| d7} || g|| g|}dg|}fdd|D}t|}tdd|d}g}t|D]:\}}| ||||g| ||d||gq| ||||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 r'r csg|]}|kr|qSrr)rWcolor all_colorsrrrYrsz&_add_outer_contour..) rr)rHZdiff concatenateZnonzeror:rFrnr`) all_contours all_values all_areasrvrJZ val_outerv_minv_maxrc color_min color_maxZ outer_contourZ delta_valuesindexZ used_colorsZ color_numberZscale discrete_cmrprtrrur_add_outer_contourFs.    rElectricrgb(150,150,150)llrr c #CsRt|d|} t|| |d} | d| d} } | dd} | dd} |dkrXd}n |g|} t|| | \}}}}t|ddd}t||||| ||d|| | | | \}}}}}tdg|df}g}t\}}|| |j }|| |j }|D]<}||j \}}||}|dkrZt |t ||||t|jf}n4|dkrtt ||| ||| f}|dkrtdddg}tdddg}tdddg} n |\}}} t||rq |d kr||n|}!td ||| d t|!d dd d ||ddd|d }"|d kr>> 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 packageskimagezS The create_ternary_contour figure factory requires the scikit-image packageF)rK)rrrr )rarrrKrr{r|)datalayoutr rz: %{a:.3f}
z: %{b:.3f}
r z3: %{c:.3f}
z: %{marker.color:.3f}markerszrgb(120, 120, 120)r)rtr)rtrr)rrrrmarkeropacity hovertemplaterT)ZcminZcmaxr showscale)rrrrrr)rG ImportErrorr get_moduler?rrDrQrrgoZFigureZadd_scatterternaryrErZ add_trace)rIrJrrrrarrrrr rKZ showmarkersZ sk_measurer{r|rPrNrOrZ contour_tracerZfigrrrrrZcolorbarrrrcreate_ternary_contoursa       r)r)r@)N)NrrrNrr )Z plotly.colorsrcr[Zplotly.graph_objsrrZplotlyrrrrrGrrrrr&r0r5r?rQrUrfrhrqrrrrrrrsV        ) "  .  2=