^jj/ddlmZddlZddlmZddlZddlZddlmZ ddl m Z ddl m Z mZe jdf ddZe jdf ddZe jd f dd Z dd Ze jdf dd Ze jd f dd Ze j,dd dej.df ddZ ddZddZddZy)) annotationsN)cast)Color)PointRectctj||j|j|j||S)ai Draws a line on a given scene. Args: scene: The scene on which the line will be drawn start: The starting point of the line end: The end point of the line color: The color of the line, defaults to Color.ROBOFLOW thickness: The thickness of the line Returns: The scene with the line drawn on it  thickness)cv2lineas_xy_int_tupleas_bgr)scenestartendcolorr s a/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/draw/utils.py draw_liners>(HH      Lctj||jj|jj|j ||S)a; Draws a rectangle on an image. Args: scene: The scene on which the rectangle will be drawn rect: The rectangle to be drawn color: The color of the rectangle thickness: The thickness of the rectangle border Returns: The scene with the rectangle drawn on it r )r rectangletop_leftr bottom_rightr)rrectrr s rdraw_rectangler,sH$MM  %%' ))+   Lrc|dk(rYtj||jj|jj|j d|S|j }tj||jj|jj|j dtj|||d|z d||S)aL Draws a filled rectangle on an image. Args: scene: The scene on which the rectangle will be drawn rect: The rectangle to be drawn color: The color of the rectangle opacity: The opacity of rectangle when drawn on the scene. Returns: The scene with the rectangle drawn on it rrgammadst)r rrrrrcopy addWeighted)rrropacityscene_with_annotationss rdraw_filled_rectangler'Hs$!|  MM ) ) +    - - / LLN   ( L"' " MM ) ) +    - - / LLN     "GUAKqe  Lrc|j\}}}}||z ||z } }t|t|| dz}||z|f||z |ff|||zf|||z ffg} ||z||zf||z ||zf||z||z f||z ||z fg} | D]0} tj|| d| d|j d2| D]*} tj || ||j d,|S)a` Draws a rounded rectangle on an image. Args: scene: The image on which the rounded rectangle will be drawn. rect: The rectangle to be drawn. color: The color of the rounded rectangle. border_radius: The radius of the corner rounding. Returns: The image with the rounded rectangle drawn on it. rrrr)imgpt1pt2rr )r)centerradiusrr )as_xyxy_int_tupleminr rrcircle)rrr border_radiusx1y1x2y2widthheightrectangle_coordinatescircle_centers coordinatesr,s rdraw_rounded_rectangler;rsI$++-NBBGR"W6E s5&'9Q'>?M } b !B$6#;< b= !B](:#;< m R-/0 m R-/0 m R-/0 m R-/0 N-  AA,,.   !   ,,.    LrcXtj||gd|j||S)ajDraw a polygon on a scene. Args: scene: The scene to draw the polygon on. polygon: The polygon to be drawn, given as a list of vertices. color: The color of the polygon. Defaults to Color.ROBOFLOW. thickness: The thickness of the polygon lines, by default 2. Returns: The scene with the polygon drawn on it. T)isClosedrr )r polylinesr)rpolygonrr s r draw_polygonr@s*"MM y4u||~ Lrc |dk(r)tj||g|j|S|j}tj||g|jtj|||d|z d||S)amDraw a filled polygon on a scene. Args: scene: The scene to draw the polygon on. polygon: The polygon to be drawn, given as a list of vertices. color: The color of the polygon. Defaults to Color.ROBOFLOW. opacity: The opacity of polygon when drawn on the scene. Returns: The scene with the polygon drawn on it. r)rrr )r fillPolyrr#r$)rr?rr%r&s rdraw_filled_polygonrCsu"!| UWIU\\^< L "' +gYellnM  "GUAKqe  Lrg? c ltj||||d\} } |j\} } t| | dzz | | dzz | | j |} |t || |}tj ||| | dzz | | dzzf|||j|tj|S)a Draw text with background on a scene. Args: scene: A numpy ndarray representing the image, typically of shape (H, W, 3) for a color BGR image or (H, W) for grayscale, with dtype uint8. text: The text to be drawn. text_anchor: The anchor point for the text, represented as a Point object with x and y attributes. text_color: The color of the text. Defaults to black. text_scale: The scale of the text. Defaults to 0.5. text_thickness: The thickness of the text. Defaults to 1. text_padding: The amount of padding to add around the text when drawing a rectangle in the background. Defaults to 10. text_font: The font to use for the text. Defaults to cv2.FONT_HERSHEY_SIMPLEX. background_color: The color of the background rectangle, if one is to be drawn. Defaults to None. Returns: The input scene with the text drawn on it. Examples: ```pycon >>> import numpy as np >>> from supervision.geometry.core import Point >>> from supervision.draw.utils import draw_text >>> 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 ... ) >>> scene.shape (100, 100, 3) ``` )textfontFace fontScaler rr)xyr6r7)rrr)r)rForgrGrHrr lineType) r getTextSizerrpadr'putTextrLINE_AA)rrF text_anchor text_color text_scaletext_thickness text_padding text_fontbackground_color text_width text_height text_anchor_x text_anchor_y text_rects r draw_textr]sb"oo     J $/#>#>#@ M= */ ) +* *   c, #%i/? KK   Z1_ ,mkQ>N.N O!   Lrc t|trRtjj |st d|dt j|t j}d|cxkrdkstdtdt|j}t|j}t|j}t|j}|dks/|dks*||z|jdkDs||z|jdkDr tdt j |||f}t#t$j&t(j*|}|jd d k(r |d d d d d fn%t)j,||f|j. dz}t j0||z} ||||z|||zf} | j3t(j4dz } t j0d| dt(j6fz | z| dt(j6f|d d d d d d fzz} | ||||z|||zf<|S)a Draws an image onto a given scene with specified opacity and dimensions. Args: scene: Background image where the new image will be drawn. image: Image to draw, either a file path or an already-loaded image array. opacity: Opacity of the image to be drawn. rect: Rectangle specifying where to draw the image. Returns: 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.rrzInvalid rectangle dimensions.rN)dtypego@.) isinstancestrospathexistsFileNotFoundErrorr imreadIMREAD_UNCHANGED ValueErrorintrIrJr6r7shaperesizernptNDArraynpuint8onesraconvertScaleAbsastypefloat32newaxis) rimager%rrect_xrect_y rect_width rect_height alpha_channel alpha_scaled scene_roi alpha_float blended_rois r draw_imager-s50%ww~~e$#mE7:L$MN N 5#"6"67 ' S ?@@ !?@@ [F [FTZZJdkk"K  A: J Q / K %++a. 0899 JJuz;7 8E RXX& .E ;;q>Q  aAg WWk:.ekk BS H &&}w'>?Lfv 33Vfz>Q5QQRI%%bjj1E9K%% [bjj ) )Y6 c2::o &q!RaRx 8 9K JUE&6K' '&:2E)E EF Lrct|dzS)a) Calculate optimal font scale based on image resolution. Adjusts font scale proportionally to the smallest dimension of the given image resolution for consistent readability. Args: resolution_wh: A tuple of `(width, height)` of the image in pixels. Returns: Recommended font scale factor. Examples: ```pycon >>> import supervision as sv >>> sv.calculate_optimal_text_scale((1920, 1080)) 1.08 >>> sv.calculate_optimal_text_scale((640, 480)) 0.48 ``` gMbP?r/ resolution_whs rcalculate_optimal_text_scalerss, }  $$rc"t|dkryy)a1 Calculate optimal line thickness based on image resolution. Adjusts the line thickness for readability depending on the smallest dimension of the provided image resolution. Args: resolution_wh: A tuple of `(width, height)` of the image in pixels. Returns: Recommended line thickness in pixels. Examples: ```pycon >>> import supervision as sv >>> sv.calculate_optimal_line_thickness((1920, 1080)) 4 >>> sv.calculate_optimal_line_thickness((640, 480)) 2 ``` i8rr_rrs r calculate_optimal_line_thicknessrs, =D  r) rnpt.NDArray[np.uint8]rrrrrrr rlreturnr) rrrrrrr rlrr) rrrrrrr%floatrr) rrrrrrr1rlrr) rrr?npt.NDArray[np.int_]rrr rlrr) rrr?rrrr%rrr)rrrFrdrQrrRrrSrrTrlrUrlrVrlrWz Color | Nonerr) rrrxzstr | npt.NDArray[np.uint8]r%rrrrr)rtuple[int, int]rr)rrrrl) __future__rretypingrr numpyrq numpy.typingrosupervision.draw.colorrsupervision.geometry.corerrROBOFLOWrrr'r;r@rCBLACKFONT_HERSHEY_SIMPLEXr]rrrrrrsl"  (1>>          B>>       >>> ' ' ' ' '  'T1 1 1 1 1  1n>>   !    4>>   !    B --%)P P PP P  P  PPP#PPfC C &CC  C  CL%2 r