FiVddlZddlmZddlZddlmZmZmZddlm Z GddeZ y)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 Nt|di|d|_i|_g|_y)zYInitialize the DistanceCalculation class for measuring object distances in video streams.rN)super__init__left_mouse_countselected_boxes centroids)selfr __class__s l/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/solutions/distance_calculation.pyrzDistanceCalculation.__init__#s, "6"!"68*,eventxyflagsparamc|tjk(r|xjdz c_|jdkrqt|j|j D]M\}}|d|cxkr |dksn|d|cxkr |dks-n0||j vs?||j |<Oyy|tjk(ri|_d|_yy)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>> 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)colorlabelrrrshowzUltralytics Solutions)plot_impixels_distance total_tracks)extract_tracksrr*r"r#r$clssconfs box_labelrintadjust_box_labellenrkeysrextendvaluesmathsqrtplot_distance_and_lineresultdisplay_outputCFGget env_checkr setMouseCallbackr(r) rim0 annotatorr/r&r'clsconftrk_idr.s rprocesszDistanceCalculation.processEsF, C %cdooF (+DJJ SWS]S](^ < $C3   6#c(D+AI^I^_bdhjrIs  t4&&'1,"11668rRs)  \\-os,osr