-Ph#nddlZddlmZddlmZddlmZddejdz dfdZGd d e Z dS) N) exceptions) graph_objs)utilsg?g333333? c tj||||tj|||t|||||||} nt||||||||} | \} } | \} } t jd | | z| | zdd|}|g}|t jd}n%t jdt|d }t j || S) a Returns data for a quiver plot. :param (list|ndarray) x: x coordinates of the arrow locations :param (list|ndarray) y: y coordinates of the arrow locations :param (list|ndarray) u: x components of the arrow vectors :param (list|ndarray) v: y components of the arrow vectors :param (float in [0,1]) scale: scales size of the arrows(ideally to avoid overlap). Default = .1 :param (float in [0,1]) arrow_scale: value multiplied to length of barb to get length of arrowhead. Default = .3 :param (angle in radians) angle: angle of arrowhead. Default = pi/9 :param (positive float) scaleratio: the ratio between the scale of the y-axis and the scale of the x-axis (scale_y / scale_x). Default = None, the scale ratio is not fixed. :param kwargs: kwargs passed through plotly.graph_objs.Scatter for more information on valid kwargs call help(plotly.graph_objs.Scatter) :rtype (dict): returns a representation of quiver figure. Example 1: Trivial Quiver >>> from plotly.figure_factory import create_quiver >>> import math >>> # 1 Arrow from (0,0) to (1,1) >>> fig = create_quiver(x=[0], y=[0], u=[1], v=[1], scale=1) >>> fig.show() Example 2: Quiver plot using meshgrid >>> from plotly.figure_factory import create_quiver >>> import numpy as np >>> import math >>> # Add data >>> x,y = np.meshgrid(np.arange(0, 2, .2), np.arange(0, 2, .2)) >>> u = np.cos(x)*y >>> v = np.sin(x)*y >>> #Create quiver >>> fig = create_quiver(x, y, u, v) >>> fig.show() Example 3: Styling the quiver plot >>> from plotly.figure_factory import create_quiver >>> import numpy as np >>> import math >>> # Add data >>> x, y = np.meshgrid(np.arange(-np.pi, math.pi, .5), ... np.arange(-math.pi, math.pi, .5)) >>> u = np.cos(x)*y >>> v = np.sin(x)*y >>> # Create quiver >>> fig = create_quiver(x, y, u, v, scale=.2, arrow_scale=.3, angle=math.pi/6, ... name='Wind Velocity', line=dict(width=1)) >>> # Add title to layout >>> fig.update_layout(title='Quiver Plot') # doctest: +SKIP >>> fig.show() Example 4: Forcing a fix scale ratio to maintain the arrow length >>> from plotly.figure_factory import create_quiver >>> import numpy as np >>> # Add data >>> x,y = np.meshgrid(np.arange(0.5, 3.5, .5), np.arange(0.5, 4.5, .5)) >>> u = x >>> v = y >>> angle = np.arctan(v / u) >>> norm = 0.25 >>> u = norm * np.cos(angle) >>> v = norm * np.sin(angle) >>> # Create quiver with a fix scale ratio >>> fig = create_quiver(x, y, u, v, scale = 1, scaleratio = 0.5) >>> fig.show() ) arrow_scalescaleNlines)xymodeclosest) hovermoder ) scaleratio scaleanchor)ryaxis)datalayout) rvalidate_equal_lengthvalidate_positive_scalars_Quiver get_barbsget_quiver_arrowsrScatterLayoutdictFigure)r r uvr ranglerkwargs quiver_objbarb_xbarb_yarrow_xarrow_y quiver_plotrrs ]/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/plotly/figure_factory/_quiver.py create_quiverr*s3t 1a+++ # 5IIIIQ1a UCC Q1a UJOO ))++NFF!3355GW$ 7 fw.W@FK =D"Y777"tzs'S'S'S     $v 6 6 66c,eZdZdZddZdZdZdZdS) rz> Refer to FigureFactory.create_quiver() for docstring c  n tj|}n#tj$rYnwxYw tj|}n#tj$rYnwxYw tj|}n#tj$rYnwxYw tj|}n#tj$rYnwxYw||_||_||_||_||_||_ ||_ ||_ g|_ g|_ ||\} } |\} } dS)N)rflattenr PlotlyErrorr r rr r rrr!end_xend_yscale_uvrr)selfr r rr r rr!rr"r$r%r&r's r)__init__z_Quiver.__init__si  a  AA%    D   a  AA%    D   a  AA%    D   a  AA%    D  $&    ))1133sA))AAAA--A?>A?BB*)B*chfdjD_fdjD_dS)z Scales u and v to avoid overlap of the arrows. u and v are added to x and y to get the endpoints of the arrows so a smaller scale value will result in less overlap of arrows. c4g|]}|jzjzSr)r r.0ir4s r) z$_Quiver.scale_uv..s&CCCq!dj.4?2CCCr+c$g|] }|jz Sr)r r8s r)r;z$_Quiver.scale_uv..s111Q!dj.111r+N)rr )r4s`r)r3z_Quiver.scale_uvsDDCCCDFCCC1111$&111r+cdt|j|jD|_dt|j|jD|_dgt|jz}tj t|j|j|}tj t|j|j|}||fS)a Creates x and y startpoint and endpoint pairs After finding the endpoint of each barb this zips startpoint and endpoint pairs to create 2 lists: x_values for barbs and y values for barbs :rtype: (list, list) barb_x, barb_y: list of startpoint and endpoint x_value pairs separated by a None to create the barb of the arrow, and list of startpoint and endpoint y_value pairs separated by a None to create the barb of the arrow. cg|] \}}||z Srrr9r:js r)r;z%_Quiver.get_barbs.. <<<1a!e<<.rAr+N) zipr rr1r r r2lenrr/)r4emptyr$r%s r)rz_Quiver.get_barbss=<DFDF(;(;<<< <<DFDF(;(;<<< TV$s464:u==>>s464:u==>>v~r+cdtjjD}dtjjD}dgt jz}t t |D].}tj||j z ||||</dgt jz}fd|D}dgt jz}t t |D].}tj ||||j z ||</fd|D}fd|D}dgt |z} t t |D]}tj ||| |< dt|| D} dgt |z} t t |D]}tj ||| |< dt|| D} dgt |z} t t |D]}tj ||| |< d t|| D}dgt |z}t t |D]}tj ||||< d t||D}t t jD]}fd tj| D}d tj| D}fd tj|D}dtj|D}dgt jz}tjt|j||}tjt|j||}||fS)a0 Creates lists of x and y values to plot the arrows Gets length of each barb then calculates the length of each side of the arrow. Gets angle of barb and applies angle to each side of the arrowhead. Next uses arrow_scale to scale the length of arrowhead and creates x and y values for arrowhead point1 and point2. Finally x and y values for point1, endpoint and point2s for each arrowhead are separated by a None and zipped to create lists of x and y values for the arrows. :rtype: (list, list) arrow_x, arrow_y: list of point1, endpoint, point2 x_values separated by a None to create the arrowhead and list of point1, endpoint, point2 y_values separated by a None to create the barb of the arrow. cg|] \}}||z  Srrr?s r)r;z-_Quiver.get_quiver_arrows.. ;;;41aQ;;;r+cg|] \}}||z  Srrr?s r)r;z-_Quiver.get_quiver_arrows..rHr+Nc$g|] }|jz Sr)rr8s r)r;z-_Quiver.get_quiver_arrows..s!<<.1111DJ111r+c$g|] }|jz  SrrLr8s r)r;z-_Quiver.get_quiver_arrows..rMr+cg|] \}}||z Srrr?s r)r;z-_Quiver.get_quiver_arrows.. ===DAq!a%===r+cg|] \}}||z Srrr?s r)r;z-_Quiver.get_quiver_arrows..rPr+cg|] \}}||z Srrr?s r)r;z-_Quiver.get_quiver_arrows..rPr+cg|] \}}||z Srrr?s r)r;z-_Quiver.get_quiver_arrows..rPr+c0g|]\}}||jzz Srrr9r:r@r4s r)r;z-_Quiver.get_quiver_arrows..)TTTDAqA//TTTr+cg|] \}}||z  Srrr?s r)r;z-_Quiver.get_quiver_arrows.. BBB$!QABBBr+c0g|]\}}||jzz SrrUrVs r)r;z-_Quiver.get_quiver_arrows..rWr+cg|] \}}||z  Srrr?s r)r;z-_Quiver.get_quiver_arrows..rYr+)rCr1r r2r rDrangemathhypotratan2cossinrr/)r4dif_xdif_ybarb_lenindex arrow_lenbarb_angang1ang2cos_ang1seg1_xsin_ang1seg1_ycos_ang2seg2_xsin_ang2seg2_ypoint1_xpoint1_ypoint2_xpoint2_yrEr&r's` r)rz_Quiver.get_quiver_arrowss"<;3tz46#:#:;;;;;3tz46#:#:;;;6CKK'3x==)) W WE"ju)GuVVHUOOFS[[( <<<<8<<< 6CKK'3x==)) W WE"juuU|do7UVVHUOO211111111111116CII%3t99%% 4 4E"htE{33HUOO==C 8$<$<===6CII%3t99%% 4 4E"htE{33HUOO==C 8$<$<===6CII%3t99%% 4 4E"htE{33HUOO==C 8$<$<===6CII%3t99%% 4 4E"htE{33HUOO==C 8$<$<===3tz??++ C CETTTTC Frs ((((((''''''s$'A+$r7r7r7r7jL L L L L fL L L L L r+