U a+d8"@s8dZddlZdddZddd Zdd d Zdd dZdS)z This module deals with making images (np arrays). It provides drawing methods that are difficult to do with the existing Python libraries. NFcCsp|dkrddg}|\}}td| }td| }|jdd\} } |jdd\} } t| || } t| || }t| | |}t| | |}td|}td|}|| ks||kr|S|||||f}| }|dkr|||||| f<np|||||f}t|jdkr td|g}||||| f}d||d||||||| f<|sl|dS|S)z Blit an image over another. Blits ``im1`` on ``im2`` as position ``pos=(x,y)``, using the ``mask`` if provided. If ``im1`` and ``im2`` are mask pictures (2D float arrays) then ``ismask`` must be ``True``. Nr?Zuint8)maxshapeminlennpdstackastype)Zim1Zim2posmaskZismaskZxpZypx1y1h1Zw1h2Zw2Zxp2Zyp2Zx2y2Zxp1Zyp1ZblittedZnew_im2Z blit_regionr?/tmp/pip-unpacked-wheel-0yp4gafk/moviepy/video/tools/drawing.pyblit s2    (rrlinearc sj\} } t|t}t|t}|dkr|dkrLt|t}fdd|| fD\} } t| | } |jdkrtd| g} | |d| |Stdddt|dkr|rt|ddd}|}nt|ddd}|}|rtj|}tt t | t | dddt}|dkr||d }|| |d} t dtd | } |jdkrtd| g} | |d| |S|d krf|dkr|}|d krt | | f} nFt|d jd d |} | d|} t d td | } |jdkrRtd| g} d| || |SdS)aaDraw a linear, bilinear, or radial gradient. The result is a picture of size ``size``, whose color varies gradually from color `col1` in position ``p1`` to color ``col2`` in position ``p2``. If it is a RGB picture the result must be transformed into a 'uint8' array to be displayed normally: Parameters ------------ size Size (width, height) in pixels of the final picture/array. p1, p2 Coordinates (x,y) in pixels of the limit point for ``col1`` and ``col2``. The color 'before' ``p1`` is ``col1`` and it gradually changes in the direction of ``p2`` until it is ``col2`` when it reaches ``p2``. vector A vector [x,y] in pixels that can be provided instead of ``p2``. ``p2`` is then defined as (p1 + vector). col1, col2 Either floats between 0 and 1 (for gradients used in masks) or [R,G,B] arrays (for colored gradients). shape 'linear', 'bilinear', or 'circular'. In a linear gradient the color varies in one direction, from point ``p1`` to point ``p2``. In a bilinear gradient it also varies symetrically form ``p1`` in the other direction. In a circular gradient it goes from ``col1`` to ``col2`` in all directions. offset Real number between 0 and 1 indicating the fraction of the vector at which the gradient actually starts. For instance if ``offset`` is 0.9 in a gradient going from p1 to p2, then the gradient will only occur near p2 (before that everything is of color ``col1``) If the offset is 0.9 in a radial gradient, the gradient will occur in the region located between 90% and 100% of the radius, this creates a blurry disc of radius d(p1,p2). Returns -------- image An Numpy array of dimensions (W,H,ncolors) of type float representing the image of the gradient. Examples --------- >>> grad = color_gradient(blabla).astype('uint8') ZbilinearNc s"g|]}t|ddddqS)rrr)vectorcol1col2roffsetcolor_gradient).0vrp1sizerr s  z"color_gradient..rrrrradial)Zaxisr)r arrayr floatmaximumr!r linalgnormZmeshgridrangedotZminimumZonessqrtsum)r!r p2rrrrrrwhm1m2arrr*MZn_vecrrrr5sRB      *       " rc Cs^|s|dkr|dkr|dk r2t|t|}nJ|dk rXtddg}t|dg}n$|dk r|tddg}td|g}|\}}t|| gd}tj|} td||| }t|||||dd S|\} } t|r| | fn | | t|f} t | } |r.|| ddd|f<|| dd|df<n|rL|| d|<|| |d<| St d dS) aEMake an image splitted in 2 colored regions. Returns an array of size ``size`` divided in two regions called 1 and 2 in wht follows, and which will have colors col& and col2 respectively. Parameters ----------- x: (int) If provided, the image is splitted horizontally in x, the left region being region 1. y: (int) If provided, the image is splitted vertically in y, the top region being region 1. p1,p2: Positions (x1,y1),(x2,y2) in pixels, where the numbers can be floats. Region 1 is defined as the whole region on the left when going from ``p1`` to ``p2``. p1, vector: ``p1`` is (x1,y1) and vector (v1,v2), where the numbers can be floats. Region 1 is then the region on the left when starting in position ``p1`` and going in the direction given by ``vector``. gradient_width If not zero, the split is not sharp, but gradual over a region of width ``gradient_width`` (in pixels). This is preferable in many situations (for instance for antialiasing). Examples --------- >>> size = [200,200] >>> # an image with all pixels with x<50 =0, the others =1 >>> color_split(size, x=50, col1=0, col2=1) >>> # an image with all pixels with y<50 red, the others green >>> color_split(size, x=50, col1=[255,0,0], col2=[0,255,0]) >>> # An image splitted along an arbitrary line (see below) >>> color_split(size, p1=[20,50], p2=[25,70] col1=0, col2=1) Nrgrgr'g?r)rrrrz)Arguments in color_split not understood !) r r&r r)r*rrZisscalarrzerosprint)r!xyr r/rrrZ grad_widthr*r1r2rr5rrr color_splits<0     r;r#c Cs.|rd|||nd}t|||||d|dS)a Draw an image with a circle. Draws a circle of color ``col1``, on a background of color ``col2``, on a screen of size ``screensize`` at the position ``center=(x,y)``, with a radius ``radius`` but slightly blurred on the border by ``blur`` pixels rrr%)r r0rrrrr)Z screensizecenterZradiusrrZblurrrrrcircles  r=)NNF)NNNrrrr)NNNNNrrr)rrr#)__doc__Znumpyr rrr;r=rrrrs , ~ P