|h3^ddlZddlmZmZmZddlZddlmZmZm Z ddl m Z GddeZ y)N)AnyDictList) BaseSolutionSolutionAnnotatorSolutionResults)colorsc XeZdZdZdeddffd Zdededed ed eddf d Zdefd Z xZ S) DistanceCalculationa A 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, List[float]]): Dictionary to store selected bounding boxes and their track IDs. 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 i/var/www/html/test/engine/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)a Handle 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)colorlabelr rr!showzUltralytics Solutions)plot_impixels_distance total_tracks)extract_tracksrr,r$r%r&clssconfs box_labelr intadjust_box_labellenrkeysrextendvaluesmathsqrtplot_distance_and_lineresultdisplay_outputCFGget env_checkr"setMouseCallbackr*r) rim0 annotatorr1r(r)clsconftrk_idr0s rprocesszDistanceCalculation.processGsF. C %cdooF (+DJJ SWS]S](^ < $C3   6#c(D+AI^I^_bdhjrIs  t4&&'1,"11668rTs, "" \\-rs,rsr