o W h*@s<ddlZddlmZmZmZddlmZGdddeZdS)N) BaseSolutionSolutionAnnotatorSolutionResults)colorscs0eZdZdZfddZddZddZZS) RegionCountera A class for real-time counting of objects within user-defined regions in a video stream. This class inherits from `BaseSolution` and provides functionality to define polygonal regions in a video frame, track objects, and count those objects that pass through each defined region. Useful for applications requiring counting in specified areas, such as monitoring zones or segmented sections. Attributes: region_template (dict): Template for creating new counting regions with default attributes including name, polygon coordinates, and display colors. counting_regions (list): List storing all defined regions, where each entry is based on `region_template` and includes specific region settings like name, coordinates, and color. region_counts (dict): Dictionary storing the count of objects for each named region. Methods: add_region: Adds a new counting region with specified attributes. process: Processes video frames to count objects in each region. c s6tjdi|ddddddd|_i|_g|_dS) zaInitializes the RegionCounter class for real-time counting in different regions of video streams.zDefault RegionNrF)rr)rrr)namepolygoncountsdragging region_color text_color)super__init__region_template region_countscounting_regions)selfkwargs __class__rX/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/solutions/region_counter.pyrs zRegionCounter.__init__cCs4|j}||||||d|j|dS)a Add a new region to the counting list based on the provided template with specific attributes. Args: name (str): Name assigned to the new region. polygon_points (List[Tuple]): List of (x, y) coordinates defining the region's polygon. region_color (tuple): BGR color for region visualization. text_color (tuple): BGR color for the text within the region. )rr r r N)rcopyupdatePolygonrappend)rrpolygon_pointsr r regionrrr add_region+s zRegionCounter.add_regioncs|t|jd}tjtsdjpi_tjddD]\}\}}t |d}| ||jd |||| q%j D]}d|vrW|d|d<qHtjd d jDtjd }fd d |D} | rt| jjjjD]5\} } } } }|j| | || t | dd j D]}|d| r|dd7<|dj|d<qq}j D]}|j|djt|d|d|djddd|d<q|}|t |t!jjdS)a Process the input frame to detect and count objects within each defined region. Args: im0 (np.ndarray): Input image frame where objects and regions are annotated. Returns: (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (int, total number of tracked objects), and 'region_counts' (dict, counts of objects per region). ) line_widthz Region#01)startTprepared_polygonr cSs4g|]}|d|dd|d|ddfqS)rr#r!r).0boxrrr ^s4z)RegionCounter.process..)dtypecsg|]}|qSr)Point)r&ptrrrr(_s)labelcolorr rr r )r-r. txt_colormarginr)plot_im total_tracksr)"extract_tracksrr isinstancerdictinitialize_region enumerateitemsr draw_regionr get_txt_colorrprepnparrayboxesfloat32zipclss track_idsconfs box_labeladjust_box_labelcontainsr text_labelboundsstrresultdisplay_outputrlen)rim0 annotatoridx region_namereg_ptsr.rboxes_nppointspointclstrack_idr'confr2rr,rprocess@sF    &      zRegionCounter.process)__name__ __module__ __qualname____doc__rrrY __classcell__rrrrr s  r) numpyr=ultralytics.solutions.solutionsrrrultralytics.utils.plottingrrrrrrs