Ej'FdZddlZ ddZ ddZddZy) zDeals with making images (np arrays). It provides drawing methods that are difficult to do with the existing Python libraries. Nc |\} } tj|jt}tj|jt}|dk(r|8| t dtj|tj|z }||D cgc]} | c} fD cgc]} t ||| ddd|c} \} } tj | | }|jdkDrtjd |gz}||zd|z |zzStj|ddd jt}tjtjt| t| ddd jt}|dk(r|,|tj|ddd |z }n&t dtj|ddd }tjj|}||d zz }|||zz}||z j|d|z z }tjdtj d |}|jdkDrtjd |gz}||zd|z |zzS|d k(r|xsd d k(rtj| | f}nftj ||z d zj#d ||zz }|d|z |zz }tjdtj d |}|jdkDrtjd |gz}d|z |z||zzSt dcc} wcc} w)aq Draw a linear, bilinear, or radial gradient. The result is a picture of size ``size``, whose color varies gradually from color `color_1` in position ``p1`` to color ``color_2`` in position ``p2``. If it is a RGB picture the result must be transformed into a 'uint8' array to be displayed normally: Parameters ---------- size : tuple or list Size (width, height) in pixels of the final image array. p1 : tuple or list Position for the first coordinate of the gradient in pixels (x, y). The color 'before' ``p1`` is ``color_1`` and it gradually changes in the direction of ``p2`` until it is ``color_2`` when it reaches ``p2``. p2 : tuple or list, optional Position for the second coordinate of the gradient in pixels (x, y). Coordinates (x, y) of the limit point for ``color_1`` and ``color_2``. vector : tuple or list, optional A vector (x, y) in pixels that can be provided instead of ``p2``. ``p2`` is then defined as (p1 + vector). color_1 : tuple or list, optional Starting color for the gradient. As default, black. Either floats between 0 and 1 (for gradients used in masks) or [R, G, B] arrays (for colored gradients). color_2 : tuple or list, optional Color for the second point in the gradient. As default, white. Either floats between 0 and 1 (for gradients used in masks) or [R, G, B] arrays (for colored gradients). shape : str, optional Shape of the gradient. Can be either ``"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 symmetrically from ``p1`` in the other direction. In a circular gradient it goes from ``color_1`` to ``color_2`` in all directions. radius : float, optional If ``shape="radial"``, the radius of the gradient is defined with the parameter ``radius``, in pixels. offset : float, optional 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 ``color_1``) 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 (width, height, n_colors) of type float representing the image of the gradient. Examples -------- .. code:: python color_gradient((10, 1), (0, 0), p2=(10, 0)) # from white to black #[[1. 0.9 0.8 0.7 0.6 0.5 0.4 0.3 0.2 0.1]] # from red to green color_gradient( (10, 1), (0, 0), p2=(10, 0), color_1=(255, 0, 0), color_2=(0, 255, 0) ) # [[[ 0. 255. 0. ] # [ 25.5 229.5 0. ] # [ 51. 204. 0. ] # [ 76.5 178.5 0. ] # [102. 153. 0. ] # [127.5 127.5 0. ] # [153. 102. 0. ] # [178.5 76.5 0. ] # [204. 51. 0. ] # [229.5 25.5 0. ]]] bilinearNz(You must provide either 'p2' or 'vector'?linear)vectorcolor_1color_2shapeoffsetrradial)axisz@Invalid shape, should be either 'radial', 'linear' or 'bilinear')nparrayastypefloat ValueErrorcolor_gradientmaximumsizedstackmeshgridrangelinalgnormdotminimumonessqrtsum)rp1p2rradiusr r r r whvm1m2arrMrn_vecs Y/root/workspace/ytshorts/venv/lib/python3.12/site-packages/moviepy/video/tools/drawing.pyrrsN DAqhhw&&u-Ghhw&&u-G  >z !KLLXXb\BHHRL0F62ar23      BjjR  <~"TrT(+b0 !KLLXXfTrTl+Fyy~~f%q  &6/ !2vll5!QZ0jjBJJq#./ <XXb\BHHRL0F ]XXq$i(F1a&!B ]XXsCj)F1a&!B11qb'"))'2yy~~f%S.)F2T9 "VWgX  1++g.AQ3w<4Hhhuo  C2A2J C12J CGCG c D|r d||z z|z nd}t|||||d|S)aDraw an image with a circle. Draws a circle of color ``color``, on a background of color ``bg_color``, 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. Parameters ---------- screensize : tuple or list Size of the canvas. center : tuple or list Center of the circle. radius : float Radius of the circle, in pixels. bg_color : tuple or float, optional Color for the background of the canvas. As default, black. blur : float, optional Blur for the border of the circle. Examples -------- .. code:: python from moviepy.video.tools.drawing import circle circle( (5, 5), # size (2, 2), # center 2, # radius ) # array([[0. , 0. , 0. , 0. , 0. ], # [0. , 0.58578644, 1. , 0.58578644, 0. ], # [0. , 1. , 1. , 1. , 0. ], # [0. , 0.58578644, 1. , 0.58578644, 0. ], # [0. , 0. , 0. , 0. , 0. ]]) rrr)r%r'r r r r )r) screensizecenterr'colorbg_colorblurr s r0circlerA s>X06SFTM "V +1F   r:)NNNrrrr)NNNNNrrr)rrr )__doc__numpyrrr9rAr:r0rEsT       fYV     Vr5r: