j^ddlZddlmZddlZddlmZmZmZddlm Z GddeZ dS)N)Any) BaseSolutionSolutionAnnotatorSolutionResults)colorsc XeZdZdZdeddffd Zdededed ed eddf d Zdefd Z xZ S) DistanceCalculationaA class to calculate distance between two objects in a real-time video stream based on their tracks. This class extends BaseSolution to provide functionality for selecting objects and calculating the distance between them in a video stream using YOLO object detection and tracking. Attributes: left_mouse_count (int): Counter for left mouse button clicks. selected_boxes (dict[int, Any]): Dictionary to store selected bounding boxes keyed by track ID. centroids (list[list[int]]): List to store centroids of selected bounding boxes. Methods: mouse_event_for_distance: Handle mouse events for selecting objects in the video stream. process: Process video frames and calculate the distance between selected objects. Examples: >>> distance_calc = DistanceCalculation() >>> frame = cv2.imread("frame.jpg") >>> results = distance_calc.process(frame) >>> cv2.imshow("Distance Calculation", results.plot_im) >>> cv2.waitKey(0) kwargsreturnNc dtjdi|d|_i|_g|_dS)zYInitialize the DistanceCalculation class for measuring object distances in video streams.rN)super__init__left_mouse_countselected_boxes centroids)selfr __class__s o/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/solutions/distance_calculation.pyrzDistanceCalculation.__init__#s=""6"""!"68*,eventxyflagsparamcz|tjkr|xjdz c_|jdkrkt|j|jD]R\}}|d|cxkr |dkr2n!|d|cxkr |dkrn=||jvr ||j|<QdSdS|tjkri|_d|_dSdS)aHandle mouse events to select regions in a real-time video stream for distance calculation. Args: event (int): Type of mouse event (e.g., cv2.EVENT_MOUSEMOVE, cv2.EVENT_LBUTTONDOWN). x (int): X-coordinate of the mouse pointer. y (int): Y-coordinate of the mouse pointer. flags (int): Flags associated with the event (e.g., cv2.EVENT_FLAG_CTRLKEY, cv2.EVENT_FLAG_SHIFTKEY). param (Any): Additional parameters passed to the function. Examples: >>> # Assuming 'dc' is an instance of DistanceCalculation >>> cv2.setMouseCallback("window_name", dc.mouse_event_for_distance) rN)cv2EVENT_LBUTTONDOWNrzipboxes track_idsrEVENT_RBUTTONDOWN)rrrrrrboxtrack_ids rmouse_event_for_distancez,DistanceCalculation.mouse_event_for_distance,s C) ) )  ! !Q & ! !$))%(T^%D%D<>> import numpy as np >>> from ultralytics.solutions import DistanceCalculation >>> dc = DistanceCalculation() >>> frame = np.random.randint(0, 255, (480, 640, 3), dtype=np.uint8) >>> results = dc.process(frame) >>> print(f"Distance: {results.pixels_distance:.2f} pixels") ) line_widthrT)colorlabelrcg|]D}t|d|dzdzt|d|dzdzgES)rrrr)int).0r&s r z/DistanceCalculation.process..lsPtttPS#s1vA1,--sCFSVO3I/J/JKtttrrshowzUltralytics Solutions)plot_impixels_distance total_tracks)extract_tracksrr*r"r#r$clssconfs box_labelrr.adjust_box_labellenrkeysrextendvaluesmathsqrtplot_distance_and_lineresultdisplay_outputCFGget env_checkr setMouseCallbackr(r) rim0 annotatorr3r&r'clsconftrk_idr2s rprocesszDistanceCalculation.processEsJ, C   %cdoFFF (+DJ SWS](^(^ < < $C3   6#c((D+A+AI^I^_bdhjrIsIs  t t t4&''1,,"16688<typingrr ultralytics.solutions.solutionsrrrultralytics.utils.plottingrr r rrrUs  \\\\\\\\\\------ososososos,osososososr