o
Nrfp @ s d dl mZ d dlZd dlZd dlZdddZdZd dd Zd d
d Z dd d
Z
d ddZ d ddZ d ddZ
dd ZdddZdd ZdS )! ) annotationsN c K s t tt| }t| rtdt|stdt| dkr$tdt| dkr4t | fd|i|S t| dkrDt
| fd|i|S t| dkrTt | fd|i|S t | fd|i|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 )
tuplemapsumnpisnananyNotImplementedErroralllensvg_1dsvg_2dsvg_3dsvg_nd)chunksr kwargsshape r W/var/www/html/software/conda/envs/catlas/lib/python3.10/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| }|pt||d}t| |\}}t|||||d\}\} }
} } d|
d | d f }
d}|d dkr:d}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| }|pt||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
krud}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 r1 r r r3 r2 zoxZoyZxyZmnxZmxxZmnyZmxyZzx_r6 ZzyZmin_zZmax_zr7 r8 r9 r: r; r< r r r r M s^
r c C s. t | d dkr d| } ttt| }t||d}| }|}g }d}d}|rt |d p,d} t|d | |d | |dfd}
|| d }|| d }|
d} | d } tt d|
d }
t ||
}tt d |
d }||d
7 }d | dd }
|
|
|s%d ||f } d
}| d | | S )Nr r )r r r )r1 r r* zheight="(\d*\.?\d*)"zwidth="(\d*\.?\d*)"r= r r" z
)r r r r
r, r splitfloatresearchgroupsmaxr0 append)r r r r1 Zchunks2Zsizes2outleftZ total_heightnor4 r9 heightwidthr: r r r r s: "
r r( c st t }||krtjd|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" /rB )r r Zlinspacerangereplace)rT rV rU rW max_nrL indicesr4 r rS r svg_lines s r] c C s. t ||d }||d }t || d |d }||d }|d r/|| |d 7 }|d rC||d | 7 }||d | 7 }t| | } t| | }
t| | } t| | } |d }
ddgt|||||
}| |d }t | |d }| |d }t | |d |d }|d r||d | 7 }||d | 7 }|d r||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 rB r% z z ZECB172Z8B4903, z z )r Z
zeros_likeZ full_likerH minr] r )r3 r2 r r r rT rV rU rW r5 r7 r6 r8 r[ Zh_linesZv_linescolorZcornersrectr r r r. s@ Rr. c K s t d| fi |S )NrA )r )r r1 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)rQ cr r r rX z grid_points..c S s g | ] \}}|| |d qS )rB r )rQ r3 r r r r rX s )zip)r r1 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?)rH )rQ d)mxr r rX re z draw_sizes..c S s g | ]}t |qS r )ratio_responserQ rr r r rX s c 3 s | ]} | V qd S Nr rj r r r
s z draw_sizes..)rH r )r r ratiosr )rh r r r, s r, c C s. | t jk r| S | dkrt | d S t 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\@)mathelog)r3 r r r ri
s
ri )r )r r r N)r Nr )r( )r r r rl )
__future__r ro rE numpyr r r/ r r r r] r. r r- r, ri r r r r s
!
9
#
8