o eU @sddlZddlmZmZmZddlZddlZddlm Z ddl m Z m Z  d,dej de de d e d ed ej f d d Z d,dej de d e d ed ej f ddZdej de d e d ej fddZ d,dej dej d e d ed ej f ddZe dddejdfdej dede de dedededed ee d ej fd!d"Zdej d#eeej fd$ede d ej f d%d&Zd'eeefd efd(d)Zd'eeefd efd*d+ZdS)-N)OptionalTupleUnion)Color)PointRectscenestartendcolor thicknessreturncCs$tj|||||d|S)a Draws a line on a given scene. Parameters: scene (np.ndarray): The scene on which the line will be drawn start (Point): The starting point of the line end (Point): The end point of the line color (Color): The color of the line thickness (int): The thickness of the line Returns: np.ndarray: The scene with the line drawn on it r )cv2lineas_xy_int_tupleas_bgr)r r r r r rE/var/www/myenv/lib/python3.10/site-packages/supervision/draw/utils.py draw_line srrectcCs(tj||j|j||d|S)ao Draws a rectangle on an image. Parameters: scene (np.ndarray): The scene on which the rectangle will be drawn rect (Rect): The rectangle to be drawn color (Color): The color of the rectangle thickness (int): The thickness of the rectangle border Returns: np.ndarray: The scene with the rectangle drawn on it rr rectangletop_leftr bottom_rightr)r rr r rrrdraw_rectangle%srcCs&t||j|j|d|S)a7 Draws a filled rectangle on an image. Parameters: scene (np.ndarray): The scene on which the rectangle will be drawn rect (Rect): The rectangle to be drawn color (Color): The color of the rectangle Returns: np.ndarray: The scene with the rectangle drawn on it rr rr rrrdraw_filled_rectangle>s rpolygoncCstj||gd||d|S)aDraw a polygon on a scene. Parameters: scene (np.ndarray): The scene to draw the polygon on. polygon (np.ndarray): The polygon to be drawn, given as a list of vertices. color (Color): The color of the polygon. thickness (int, optional): The thickness of the polygon lines, by default 2. Returns: np.ndarray: The scene with the polygon drawn on it. T)isClosedr r )r polylinesr)r r r r rrr draw_polygonTsr#g? text text_anchor text_color text_scaletext_thickness text_padding text_fontbackground_colorc Cstj||||dd\} } t|j| d|j| d| | d|} |dur-t|| |d}tj|||j| d|j| df||||tj d|S)a Draw text with background on a scene. Parameters: scene (np.ndarray): A 2-dimensional numpy ndarray representing an image or scene text (str): The text to be drawn. text_anchor (Point): The anchor point for the text, represented as a Point object with x and y attributes. text_color (Color, optional): The color of the text. Defaults to black. text_scale (float, optional): The scale of the text. Defaults to 0.5. text_thickness (int, optional): The thickness of the text. Defaults to 1. text_padding (int, optional): The amount of padding to add around the text when drawing a rectangle in the background. Defaults to 10. text_font (int, optional): The font to use for the text. Defaults to cv2.FONT_HERSHEY_SIMPLEX. background_color (Color, optional): The color of the background rectangle, if one is to be drawn. Defaults to None. Returns: np.ndarray: The input scene with the text drawn on it. Examples: ```python >>> scene = np.zeros((100, 100, 3), dtype=np.uint8) >>> text_anchor = Point(x=50, y=50) >>> scene = draw_text(scene=scene, text="Hello, world!",text_anchor=text_anchor) ``` )r&fontFace fontScaler rr)xywidthheightNr)imgr&orgr.r/r r lineType) r getTextSizerr0r1padrputTextrLINE_AA) r r&r'r(r)r*r+r,r- text_width text_height text_rectrrr draw_texths>'   r>imageopacityc Cst|trtj|std|dt|tj}d|kr'dks,t dt d|j dksL|j dksL|j |j |j dksL|j |j|j dkrPt dt||j |jf}|j d d krl|d d d d d fn tj|j|j f|jd d}t||}||j |j |j|j |j |j f}|tjd}td|dtjf||dtjf|d d d d d d f}|||j |j |j|j |j |j f<|S)a" Draws an image onto a given scene with specified opacity and dimensions. Args: scene (np.ndarray): Background image where the new image will be drawn. image (Union[str, np.ndarray]): Image to draw. opacity (float): Opacity of the image to be drawn. rect (Rect): Rectangle specifying where to draw the image. Returns: np.ndarray: The updated scene. Raises: FileNotFoundError: If the image path does not exist. ValueError: For invalid opacity or rectangle dimensions. z Image path ('z') does not exist.gg?z$Opacity must be between 0.0 and 1.0.rr$zInvalid rectangle dimensions.rN)dtypego@.) isinstancestrospathexistsFileNotFoundErrorrimreadIMREAD_UNCHANGED ValueErrorr0r1r2shaper3resizenponesrCconvertScaleAbsastypefloat32newaxis) r r?r@r alpha_channel alpha_scaled scene_roi alpha_float blended_roirrr draw_images8    (&(r[ resolution_whcCs t|dS)a Calculate a dynamic font scale based on the resolution of an image. Parameters: resolution_wh (Tuple[int, int]): A tuple representing the width and height of the image. Returns: float: The calculated font scale factor. gMbP?min)r\rrrcalculate_dynamic_text_scales r_cCs<t|}|dkr dS|dkrdS|dkrdS|dkrdSdS)a Calculate a dynamic line thickness based on the resolution of an image. Parameters: resolution_wh (Tuple[int, int]): A tuple representing the width and height of the image. Returns: int: The calculated line thickness in pixels. irii8iprAr])r\ min_dimensionrrr calculate_dynamic_line_thicknesss ra)r)rGtypingrrrrnumpyrPsupervision.draw.colorrsupervision.geometry.corerrndarrayintrrrr#blackFONT_HERSHEY_SIMPLEXrFfloatr>r[r_rarrrrs      F  >