g6X ddlmZddlmZddlmZddlmZejdZ ejdZ ddl m Z dd d gd fd Zdd ZdZdZdZdZddZdZd dZdZdZdZ d!dZgd ddddddddddf dZy)"N) graph_objs) exceptions)optional_importsnumpyzscipy.interpolate)measurezTernary contour ploti&i )abcct|||tdtt|ddddtt|ddddtt|ddddd 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?outside)titlemin linewidthticks)sumaaxisbaxiscaxisF)rwidthheightternary showlegend)dict)rrr pole_labelss c/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/plotly/figure_factory/_ternary_contour.py_ternary_layoutr s|$  A/TQi A/TQi A/TQi  ! c|dk(}tj|d}d||zz }tj|dkr tdtj||||z}|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. raxisr zMThe provided value of delta led to negativeternary coords.Set a smaller delta)npany ValueErrorwhere) ternary_datadelta zero_maskis_any_coord_zerounity_complements r_replace_zero_coordsr.7sv*!Iyq15#444 vv!# 1  88Iu.>.MNL r!cFtj|}tj|d|dz tjdz }dtjdz tj|d|dz|ddzz z}tj ||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?)r%asarraylogsqrtstack) barycentricx_0x_1 ilr_tdatas r_ilr_transformr:Ys **[)K &&Q+a.0 1BGGAJ >C bggaj266+a.;q>"AKPQNVWDW"WXX#s$I r!c Btj|}tjgdgdgdg}tjddz }tjd}tjdtj||g| |gg|}tj dtj |j d z }tjd|tjd|z|dz||dz|f}tj |}|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 r0)gr r0)r=r0rg? ik, kj -> ijr rr#)r%arrayr4einsumr3exprr5)xmatrixstSkZlog_barycentric iilr_tdatas r _ilr_inverserJrs"  A XX}noF GF  QA A >288aVqb!W,=#> BB BFF2JNNN** ++Aii!a%!A$,AaD!)D EO(J r!ctjdtjddz gddgddgg}tj|dddf|dddftjdg}|tjj |fS)zi Returns the transformation matrix from barycentric to Cartesian coordinates and conversely. r<rrr N)r%r?r4oneslinalginv) tri_vertsMs r _transform_barycentric_cartesianrRsw 3 Q/!Q!Q@AI )AqD/9QT?BGGAJ?@A biimmA r!ct|ttjfs t dtj |}|j ddvr t dt|dk(r]tj|jddd s5tj|jdd d s d }t |t|d k(r|\}}d||zz }n||jdz \}}}tjtj|||fdkr t d tj|||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)rrLzCA point should have 2 (a, b) or 3 (a, b, c)barycentric coordinatesrLr#r r)rtoldz=The sum of coordinates should be 1 or 100 for all data pointsrz+Barycentric coordinates should be positive.) isinstancelistr%ndarrayr'r2shapelenallcloserr&r5)b_coordsmsgABCs r_prepare_barycentric_coordras/ hrzz 2 3 /  zz(#H~~a& T   X!  HLLaL0!$ ? HLLaL0#D AMo 8}1 QKX\\q\111a vvbhh1ay!A%&FGG 88Q1I r!ilrc |dk(r%t\}}tjd||}n'|dk(rt|}t |}n t d|dd\}}|j |j} }|j |j} } tdttjt|} tj|| | } tj| | | }tj| |\}}tj|ddj|||fd }tj|ddj|||fd }|| |fS) a Transform data points with Cartesian or ILR mapping, then Compute interpolation on a regular grid. Parameters ========== coordinates : array-like Barycentric coordinates of data points. values : 1-d array-like Data points, field to be represented as contours. interp_mode : 'ilr' (default) or 'cartesian' Defines how data are interpolated to compute contours. cartesianr>rbz&interp_mode should be cartesian or ilrNrcubic)methodnearest)rRr%r@r.r:r'rmaxintr4rZlinspacemeshgrid scipy_interpgriddataT) coordinatesvalues interp_moderQinvM coord_pointsxxyyx_minx_maxy_miny_maxn_interpgr_xgr_ygrid_xgrid_ygrid_z grid_z_others r _compute_gridrsUk!244yyK@  *;7 %k2 ABB "1 FB668RVVX5E668RVVX5E3BGGCK012H ;;ueX .D ;;ueX .D[[t,NFF " "RaFVV$4W#F ((RaFVV$4Y)L 4 r!c dtjtj|tj|dtj|tj|dz zS)Nr<r )r%absdotroll)rBys r _polygon_arearsE q"''!Q-0266!RWWQ]3KKL LLr!cL|tjjvrtj|}n@tjdj tjjt jdd|}t j|Dcgc]}|d c}}t j|Dcgc]}|d c}}t j||}d|dvr5|Dcgc]*}tjtj|,}}|dg} t|dd|ddD]P\} } || dz || } } | | z | | z z }tj|| dz || |d}| j|R| Scc}wcc}wcc}w)zQ Return a list of ``ncontours`` colors from the ``colormap`` colorscale. zRColorscale must be a valid Plotly Colorscale.The available colorscale names are {}rr #Nrgb) colortype)clrs PLOTLY_SCALESkeysr PlotlyErrorformatr%rkr? searchsorted label_rgb hex_to_rgbzipfind_intermediate_colorappend) ncontourscolormapcmaprqpair vals_cmapcolsindscolcolorsindvalval1val2interms r_colorsrs4%%**,,!!(+$$ 44:F4;M;M;R;R;T4U  [[Ay )Fd3d$q'34I 88.T!W. /D ??9f -D d1g~@DEts34EE1gYFQR&*-SsQw'3d*-** qM49f   c  M4.Fs F F9/F!ctjtj||dz dkxr1tjtj||dz dk}|S)zy Utility function for _contour_trace Contours with an area of the order as 1 pixel are considered spurious. rr)r%allr)rBr too_smalls r_is_invalid_contourr sR rvva!A$h'!+,Mq1Q4x8H18L1MI r!c tj|}|tj|j|tj|j }}tj |}d|z||<tj |}d|z||<ggggf\}} } } ggggf\} } }}t |D]X\}}tj||}tj||}|j|| j|| j|gt|z| j|gt|z| j|Dcgc](}t|jd|jd*c}|j|Dcgc](}t|jd|jd*c}| j||gt|z|j||gt|z[t|t| kr|| | | fS| | ||fScc}wcc}w)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... rr r) r%isnan logical_notrricopy enumerater find_contoursextendrZrro)imrqrmask_nanim_minim_maxzz_minzz_max all_contours1 all_values1 all_areas1 all_colors1 all_contours2 all_values2 all_areas2 all_colors2ircontour_level1contour_level2contours r_extract_contoursrs $xx|H 2>>( #$((* 2>>( #$((* FWWR[F6zF8 WWR[F6zF8:3 ..vs; ..vs;^,^,C53~#667C53~#667FT U7]799Q<1 6 U  FT U7]799Q<1 6 U  F1I;^)<<=F1I;^)<<=> =S//k:{BBk:{BB V Vs -H< -I c dtjgdgdgjz} | g|z}tj|d} tj |d| z g||d| zgf}tj | g|| gf}tj ||k(dd} | t |dz kr| dz} n| dz } || g|z}|| g|z}dg|z}|Dcgc] }||vs| }}t |}tjdd|dz}g}t|D]:\}}|j||||g|j||dz||g<|j||||dz g|||||fScc}w)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). rrr )rr r<rrr ) r%r?rodiff concatenatenonzerorZrkrr) all_contours all_values all_areas all_colorsrq val_outerv_minv_maxr color_min color_max outer_contour delta_valuesindexcolor used_colors color_numberscale discrete_cmrs r_add_outer_contourrFs:9k":;===M!?\1L776?1%L ^^ )l " #VfRj<.G-HIF^^i[&9+> ?F JJv* +A .q 1E s6{Q    -:-J-:-JiI&,DU0C5DKD{#L KK1lQ. /EKk*;5E!Hk!n56E!a%L+a.9:;l+[9I-JKL Y K GGEs  E?'E?c "t|dz|} tj|| |dz} | d| d} } | dd} | dd} |d}n|g|z} t|| | \}}}}tj |ddd}t ||||| ||d|| | | | \}}}}}tj dg|dzf}g}t\}}|j|jz |jz }|j|jz |jz }|D]j}||j\}}||}|dk(rPtj|tj||z||ztj|jf}nM|dk(rHt!tj||z|jz||z|jzf}|dk(rFtj#gd }tj#gd }tj#gd } n\}}} t%||r|d k(r||n|}!t'd ||| d t'|!ddd||ddd|z }"|d k(rd|"d<|j)|"m||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 ===== rrrr Nzrgb(150, 150, 150)rdrb)r rr)rr rrlinesscatterternaryspline)rrYrtoselfTskipz%.3f) typerr r modelinefill fillcolorr hoverinfonamer)rr%rkrargsortrrrRrirsizerorr5rMrYrJr?rrr)#rBrzr colorscale linecolorrrcoloringrrrrqrrrrrrorderrtracesrQrsdxdyr y_contour x_contourr bar_coordsrr r _coltraces# r_contour_tracersXY]J /F[[y1} 5F!!9fRjyI Ab\F Ab\F( y(7H 6673L*i JJy !$B$ 'EDV58   D@L*i[ NNQC+ ,EF.0GAt %%'AEEG qvv %B %%'AEEG qvv %B%+E244 9 + %"y."y."'')//:RSTJE !%"y.15572BNQUUW4LMNJ A:#A#A#A GAq! y) 4 $,$7z% Y!D: '#   w  E&M eK%N ; r!iFBlueredc t tdtjd} | td|d}|d}t |}|j |j }}t||| \}}}t|||| }t|||||| | ||| \}}tj|| }| rd nd }|\}}}|d d z|d zdz|dzdz}|j|||d||dt|dk7dd|||rK| s|}tddgdgdg|j |j |dddd}|j||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 packageskimagezS The create_ternary_contour figure factory requires the scikit-image packageF)rr)rrrr)rrrrrrrr)datalayoutr rz: %{a:.3f}
z: %{b:.3f}
rz3: %{c:.3f}
z: %{marker.color:.3f}markerszrgb(120, 120, 120)r)rr)rrr)rr r rmarkeropacity hovertemplaterT)cmincmaxr showscale)rrr r rr)rm ImportErrorr get_modulerarrirr rgoFigureadd_scatterternaryrjr add_trace)rprqrrrrrrrrrrr showmarkers sk_measurerrrr|r}r contour_tracerfigrrr r rcolorbars rcreate_ternary_contourrsB L  ",,Y7J     ,[9K::<5E&{F TFD$ uV5F"0   "M; ))v 6CaAGGAq!A   a.    a.    0  0 $2SWAT=UV # $J(VVV"JJL"JJL",!%  "   h Jr!)gMb@?)rb)N)NElectriczrgb(150,150,150)llrNrr ) plotly.colorsrrplotly.graph_objsrr plotlyrrr r%rmrrr r.r:rJrRrarrrrrrrrr!rrs.#   )***+>?  !C_#RD2<@(\M:/Cd6Hz   P   ur!