a
S/bL @ s d dl Z d dlZd dlZdddZdZdddZd d d
Zd d d Zd ddZ d ddZ
d ddZ dd Z d!ddZ
dd ZdS )" N c K s t tt| }t| r$tdt|s4tdt| dkrHtdt| dkrht | fd|i|S t| dkrt
| fd|i|S t| dkrt | fd|i|S t | fd|i|S d S )
zConvert chunks from Dask Array into an SVG Image
Parameters
----------
chunks: tuple
size: int
Rough size of the image
Returns
-------
text: An svg string depicting the array as a grid of chunks
zbCan't generate SVG with unknown chunk sizes.
A possible solution is with x.compute_chunk_sizes()z+Can't generate SVG with 0-length dimensionsr z$Can't generate SVG with 0 dimensions size N)
tuplemapsumnpZisnananyNotImplementedErroralllensvg_1dsvg_2dsvg_3dsvg_nd)chunksr kwargsshape r -lib/python3.9/site-packages/dask/array/svg.pysvg s
r z9font-size="1.0rem" font-weight="100" text-anchor="middle"r r c C s t tt| }|p t||d}t| |\}}t|||||d\}\} }
} } d|
d | d f }
d}|d dkrtd}nd}d d
d |
d | d
t|d f d|
d
| d t||
d
| d |d f g}|
d|| | S )Nr offsetskewr Gr d # %dr r ? %d
r r r
draw_sizes grid_pointssvg_grid
text_stylejoin)r r r r sizesr yxlinesmin_xmax_xmin_ymax_yheaderfooterrotatetextr r r r + s0
$ r c C s t tt| }|p t||d}t| |\}}}|\}} t|d ||d | d fd|d\}
\} } }
}t||d |d | d fd|d\}\}}}}t|||| d | | fd|d\}\}}}}d |d
|d
f }d }|d d
krd}nd}ddd|| d |d t|d f d|d || d t||d || d |d f d| | d d || | d d t| | d d || | d d |d f g}|d|
| | | | S )Nr g333333?
r )r r r )r r r r r r r r! r" r# r$ r% r r& r' z? %dr( r) )r r r/ r r r1 r0 zZoxZoyZxyZmnxZmxxZmnyZmxyZzx_r4 ZzyZmin_zZmax_zr5 r6 r7 r8 r9 r: r r r r K s^
r c C s. t | d dkrd| } ttt| }t||d}| }|}g }d}d}|r t |d pZd} t|d | |d | |dfd}
|| d }|| d }|
d} | d } tt d|
d }
t ||
}tt d |
d }||d
7 }d | dd }
|
|
qFd ||f } d
}| d | | S )Nr r )r r r )r/ r r( zheight="(\d*\.?\d*)"zwidth="(\d*\.?\d*)"r; r r z
)r r r r r* r splitfloatresearchgroupsmaxr. append)r r r r/ Zchunks2Zsizes2outleftZ total_heightnor2 r7 Zheightwidthr8 r r r r s8 "
r r& c st t }||kr(tjd|d |dd}nt|} fdd|D }|d dd|d< |d dd|d < |S )
a5 Convert points into lines of text for an SVG plot
Examples
--------
>>> svg_lines([0, 1], [0, 0], [10, 11], [1, 1]) # doctest: +NORMALIZE_WHITESPACE
[' ',
' ']
r r int)Zdtypec s, g | ]$}d | | | | f qS )z* r ).0ix1x2y1y2r r
s z svg_lines..z /z style="stroke-width:2" /r? )r r
Zlinspacerangereplace)rP rR rQ rS max_nrI indicesr2 r rO r svg_lines s rY c C s6 t ||d }||d }t || d |d }||d }|d r^|| |d 7 }|d r||d | 7 }||d | 7 }t| | } t| | }
t| | } t| | } |d }
ddgt|||||
}| |d }t | |d }| |d }t | |d |d }|d rZ||d | 7 }||d | 7 }|d rx||d | 7 }ddgt|||||
}t| |
k rt||
k rdnd }|d d
|d d |d d
|d d |d d
|d d |d d
|d }dd d
| d| dg}|| | | | |
| ffS )zCreate lines of SVG text that show a grid
Parameters
----------
x: numpy.ndarray
y: numpy.ndarray
offset: tuple
translational displacement of the grid in SVG coordinates
skew: tuple
r r r? r# z z ZECB172Z8B4903, z z )r
Z
zeros_likeZ full_likerE minrY r )r1 r0 r r r rP rR rQ rS r3 r5 r4 r6 rW Zh_linesZv_linesZcolorZcornersZrectr r r r, s@
$Rr, c K s t d| fi |S )Nr> )r )r r/ r r r r r s r c C s& dd | D }dd t ||D }|S )Nc S s g | ]}t d | qS ))r )r
Zcumsum)rM cr r r rT z grid_points..c S s g | ]\}}|| |d qS )r? r )rM r1 r r r r rT r_ )zip)r r/ Z cumchunksZpointsr r r r+ s r+ c s> t | fdd| D }dd |D }tfdd|D S )z%Get size in pixels for all dimensionsc s g | ]} t d | qS )g?)rE )rM d)mxr r rT r_ z draw_sizes..c S s g | ] }t |qS r )ratio_responserM rr r r rT r_ c 3 s | ]} | V qd S )Nr rd r r r r_ z draw_sizes..)rE r )r r Zratiosr )rb r r r* s r* c C s2 | t jk r| S | dkr$t | d S t dS dS )a How we display actual size ratios
Common ratios in sizes span several orders of magnitude,
which is hard for us to perceive.
We keep ratios in the 1-3 range accurate, and then apply a logarithm to
values up until about 100 or so, at which point we stop scaling.
r! g(@g\@N)mathelog)r1 r r r rc s
rc )r )r r r N)r Nr )r )r& )r r r )N)r )rg rB Znumpyr
r r- r r r rY r, r r+ r* rc r r r r s
!
9
#
8