o W h3@sfddlZddlZddlZddlmZmZmZddlm Z ddl m Z GdddZ GdddeZ dS) N) BaseSolutionSolutionAnnotatorSolutionResults)LOGGER) check_imshowc@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)ParkingPtsSelectiona A class for selecting and managing parking zone points on images using a Tkinter-based UI. This class provides functionality to upload an image, select points to define parking zones, and save the selected points to a JSON file. It uses Tkinter for the graphical user interface. Attributes: tk (module): The Tkinter module for GUI operations. filedialog (module): Tkinter's filedialog module for file selection operations. messagebox (module): Tkinter's messagebox module for displaying message boxes. master (tk.Tk): The main Tkinter window. canvas (tk.Canvas): The canvas widget for displaying the image and drawing bounding boxes. image (PIL.Image.Image): The uploaded image. canvas_image (ImageTk.PhotoImage): The image displayed on the canvas. rg_data (List[List[Tuple[int, int]]]): List of bounding boxes, each defined by 4 points. current_box (List[Tuple[int, int]]): Temporary storage for the points of the current bounding box. imgw (int): Original width of the uploaded image. imgh (int): Original height of the uploaded image. canvas_max_width (int): Maximum width of the canvas. canvas_max_height (int): Maximum height of the canvas. Methods: initialize_properties: Initializes the necessary properties. upload_image: Uploads an image, resizes it to fit the canvas, and displays it. on_canvas_click: Handles mouse clicks to add points for bounding boxes. draw_box: Draws a bounding box on the canvas. remove_last_bounding_box: Removes the last bounding box and redraws the canvas. redraw_canvas: Redraws the canvas with the image and all bounding boxes. save_to_json: Saves the bounding boxes to a JSON file. Examples: >>> parking_selector = ParkingPtsSelection() >>> # Use the GUI to upload an image, select parking zones, and save the data c Cszzddl}ddlm}m}Wn"ty0ddl}dddd|d}td |YdSwt d d s8dS||||_ |_|_|j |_ |j d |j d d |j j|j dd|_|jj|j jdd|_d|_d|_d|_d|_d|_d|_d|_|j |j }|j|j jdd|jfd|jfd|jffD]\}}|j j |||dj|j j!dq|"|j #dS)zhInitialize the ParkingPtsSelection class, setting up UI and properties for parking zone point selection.rN) filedialog messageboxzrsudo apt install python3-tk (Debian/Ubuntu) | sudo dnf install python3-tkinter (Fedora) | sudo pacman -S tk (Arch)zgreinstall Python and enable the checkbox `tcl/tk and IDLE` on **Optional Features** during installationzYreinstall Python from https://www.python.org/downloads/macos/ or `brew install python-tk`)LinuxWindowsDarwinz+Unknown OS. Check your Python installation.z8 Tkinter is not configured or supported. Potential fix: T)warnz)Ultralytics Parking Zones Points SelectorFwhite)bg)sidez Upload ImagezRemove Last BBoxSave)textcommand)$tkinterrr ImportErrorplatformgetsystemrwarningrtkTkmastertitle resizableCanvascanvaspackBOTTOMimage canvas_imagecanvas_max_widthcanvas_max_heightrg_data current_boximghimgwFrameTOP upload_imageremove_last_bounding_box save_to_jsonButtonLEFTinitialize_propertiesmainloop) selfrrr r install_cmd button_framercmdr8\/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/solutions/parking_management.py__init__1sN      zParkingPtsSelection.__init__cCs6d|_|_gg|_|_d|_|_d\|_|_dS)zHInitialize properties for image, canvas, bounding boxes, and dimensions.Nr)ii)r#r$r'r(r*r)r%r&r4r8r8r9r2es  z)ParkingPtsSelection.initialize_propertiescCsddlm}m}||jjdgd|_|jsdS|jj\|_|_ |j|j }|dkr2t |j |jnt |j |}|dkrDt |j |j nt ||}|jj||d||j||f|_|jjdd|jj|jd|jd |j|j|jfdS) zZUpload and display an image on the canvas, resizing it to fit within specified dimensions.r)ImageImageTk)z Image Filesz*.png *.jpg *.jpeg) filetypesN)widthheightanchorr#z )PILr<r=openraskopenfilenamer#sizer*r)minr%intr&r config PhotoImageresizer$ create_imagerNWbindon_canvas_clickr'clearr()r4r<r= aspect_ratio canvas_width canvas_heightr8r8r9r-ls " z ParkingPtsSelection.upload_imagecCs|j|j|jf|jj|jd|jd|jd|jdddt|jdkr>|j|j| |j|j dSdS)zCHandle mouse clicks to add points for bounding boxes on the canvas.red)fillN) r(appendxyr create_ovallenr'copydraw_boxrQ)r4eventr8r8r9rPs. z#ParkingPtsSelection.on_canvas_clickcCs6tdD]}|jj||||dddddqdS)zADraw a bounding box on the canvas using the provided coordinates.rXr?blue)rWr@N)ranger create_line)r4boxir8r8r9r_s &zParkingPtsSelection.draw_boxcCs.|js |jdddS|j|dS)zARemove the last bounding box from the list and redraw the canvas.WarningzNo bounding boxes to remove.N)r'r showwarningpop redraw_canvasr;r8r8r9r.s   z,ParkingPtsSelection.remove_last_bounding_boxcCs@|jd|jjdd|jj|jd|jD]}||qdS)z8Redraw the canvas with the image and all bounding boxes.allrrBN)r deleterMrrNr$r'r_)r4rer8r8r9rjs   z!ParkingPtsSelection.redraw_canvascs|j|j|j|jfdd|jD}ddlm}|}tj ||ddt ddd d }| | Wd n1sFwY|j d d d S)zMSave the selected parking zone points to a JSON file with scaled coordinates.cs$g|]}dfdd|DiqS)pointscs(g|]\}}t|t|fqSr8)rI).0rZr[scale_hscale_wr8r9 s(z?ParkingPtsSelection.save_to_json...r8)rnreror8r9rrs$z4ParkingPtsSelection.save_to_json..r)StringIOrX)indentzbounding_boxes.jsonwzutf-8)encodingNSuccessz+Bounding boxes saved to bounding_boxes.json)r*r winfo_widthr) winfo_heightr'iorsjsondumprEwritegetvaluer showinfo)r4datars write_bufferfr8ror9r/s" z ParkingPtsSelection.save_to_jsonN) __name__ __module__ __qualname____doc__r:r2r-rPr_r.rjr/r8r8r8r9r s#4  rcs(eZdZdZfddZddZZS)ParkingManagementa* Manages parking occupancy and availability using YOLO model for real-time monitoring and visualization. This class extends BaseSolution to provide functionality for parking lot management, including detection of occupied spaces, visualization of parking regions, and display of occupancy statistics. Attributes: json_file (str): Path to the JSON file containing parking region details. json (List[Dict]): Loaded JSON data containing parking region information. pr_info (Dict[str, int]): Dictionary storing parking information (Occupancy and Available spaces). arc (Tuple[int, int, int]): RGB color tuple for available region visualization. occ (Tuple[int, int, int]): RGB color tuple for occupied region visualization. dc (Tuple[int, int, int]): RGB color tuple for centroid visualization of detected objects. Methods: process: Processes the input image for parking lot management and visualization. Examples: >>> from ultralytics.solutions import ParkingManagement >>> parking_manager = ParkingManagement(model="yolo11n.pt", json_file="parking_regions.json") >>> print(f"Occupied spaces: {parking_manager.pr_info['Occupancy']}") >>> print(f"Available spaces: {parking_manager.pr_info['Available']}") c stjd i||jd|_|jdurtdtdt|j}t ||_Wdn1s3wYddd|_ d|_ d|_ d |_ dS) zVInitialize the parking management system with a YOLO model and visualization settings. json_fileNu@❌ json_file argument missing. Parking region details required.u#❌ Json file path can not be emptyr) Occupancy Available)rr)rrr)rrr8)superr:CFGrrr ValueErrorrEr{loadpr_infoarcoccdc)r4kwargsr __class__r8r9r:s      zParkingManagement.__init__c Cs~||t|jd}}t||j}|jD]y}tj|dtjdd}d}t |j |j D]?\}} t |d|ddt |d|dd} } t || | fd} | dkrm|||jjt | d d | | d d }nq.|rx|d|dfn||f\}}t j||gd |r|jn|jdd q|||jd<|jd<|||jd d d |} || t| |jd|jdt|jdS)a Process the input image for parking lot management and visualization. This function analyzes the input image, extracts tracks, and determines the occupancy status of parking regions defined in the JSON file. It annotates the image with occupied and available parking spots, and updates the parking information. Args: im0 (np.ndarray): The input inference image. Returns: (SolutionResults): Contains processed image `plot_im`, 'filled_slots' (number of occupied parking slots), 'available_slots' (number of available parking slots), and 'total_tracks' (total number of tracked objects). Examples: >>> parking_manager = ParkingManagement(json_file="parking_regions.json") >>> image = cv2.imread("parking_lot.jpg") >>> results = parking_manager.process(image) rrm)dtype)r?rbFrbr?rU)h)rrr T)isClosedcolor thicknessrr)plot_im filled_slotsavailable_slots total_tracks)extract_tracksr]r{r line_widthnparrayint32reshapezipboxesclssrIcv2pointPolygonTestdisplay_objects_labelsmodelnames polylinesrrrdisplay_analyticsresultdisplay_outputr track_ids)r4im0esfs annotatorregion pts_array rg_occupiedreclsxcycdistrr8r8r9processs8   2 $ zParkingManagement.process)rrrrr:r __classcell__r8r8rr9rs r)r{rnumpyrultralytics.solutions.solutionsrrrultralytics.utilsrultralytics.utils.checksrrrr8r8r8r9s  #