j2rddlmZddlmZddlmZddlZddlZddl m Z m Z ddl m Z Gdde ZdS) ) annotations)cycle)AnyN) BaseSolutionSolutionResults) plt_settingscXeZdZdZedfd Zdd Z dddZxZS) AnalyticsaA class for creating and updating various types of charts for visual analytics. This class extends BaseSolution to provide functionality for generating line, bar, pie, and area charts based on object detection and tracking data. Attributes: type (str): The type of analytics chart to generate ('line', 'bar', 'pie', or 'area'). x_label (str): Label for the x-axis. y_label (str): Label for the y-axis. bg_color (str): Background color of the chart frame. fg_color (str): Foreground color of the chart frame. title (str): Title of the chart window. max_points (int): Maximum number of data points to display on the chart. fontsize (int): Font size for text display. color_cycle (cycle): Cyclic iterator for chart colors. total_counts (int): Total count of detected objects (used for line charts). clswise_count (dict[str, int]): Dictionary for class-wise object counts. fig (Figure): Matplotlib figure object for the chart. ax (Axes): Matplotlib axes object for the chart. canvas (FigureCanvasAgg): Canvas for rendering the chart. lines (dict): Dictionary to store line objects for area charts. color_mapping (dict[str, str]): Dictionary mapping class labels to colors for consistent visualization. Methods: process: Process image data and update the chart. update_graph: Update the chart with new data points. Examples: >>> analytics = Analytics(analytics_type="line") >>> frame = cv2.imread("image.jpg") >>> results = analytics.process(frame, frame_number=1) >>> cv2.imshow("Analytics", results.plot_im) kwargsrreturnNonec tjdi|ddlm}ddlm}ddlm}|jd|_ |j dvrdnd|_ d |_ d |_ d |_ d |_d |_d|_|jd}t#gd|_d|_i|_|dd|_d|_|j dvri|_||j ||_||j|_|jd|j |_|j dkr,|jggd|j\|_dSdS|j dvr| ||j \|_|_||j|_|j!|j i|_"|j dkr|j#ddSdSdS)zSInitialize Analytics class with various chart types for visual data representation.rN)FigureCanvasAgg)Figureanalytics_type>barpieClasseszFrame#z Total Countsz#F3F3F3z#111E68zUltralytics Solutions-figsizez#DD00BAz#042AFFz#FF4447z#7D24FFz#BD00FF update_every>arealine) facecolorro)rrcyan)color linewidth)rrrequal)$super__init__matplotlib.pyplotpyplotmatplotlib.backends.backend_aggrmatplotlib.figurerCFGtypex_labely_labelbg_colorfg_colortitle max_pointsfontsizer color_cycle total_counts clswise_countgetr last_plot_imlinesfigcanvas add_subplotaxplot line_widthrsubplots set_facecolor color_mappingaxis)selfr pltrrr __class__s d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/solutions/analytics.pyr%zAnalytics.__init__2s, ""6"""''''''CCCCCC,,,,,,H-. $(I$?$?yyX % " ! ,  (9% !X!X!XYY"JJ~r::  9( ( (DJv wGGGDH)/$(33DKh**3$-*HHDGyF""#w||B&DO|\\ #" Y. ( ( # W V V DHdg)/$(33DK G ! !$- 0 0 0!#D yE!! W%%%%%) ("!im0 np.ndarray frame_numberintrcp|jdkr]jD]}xjdz c_|jzdkpjdu}|r|_j}d_njdvroddlm}|fdj D_ |jzdkpjdu}|r'|j j _j}ntd jd t|tjj S) aProcess image data and run object tracking to update analytics charts. Args: im0 (np.ndarray): Input image for processing. frame_number (int): Video frame number for plotting the data. Returns: (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (int, total number of tracked objects) and 'classwise_count' (dict, per-class object count). Raises: ValueError: If an unsupported chart type is specified. Examples: >>> analytics = Analytics(analytics_type="line") >>> frame = np.zeros((480, 640, 3), dtype=np.uint8) >>> results = analytics.process(frame, frame_number=1) rrN)rJ>rrr)Counterc3LK|]}jt|VdS)N)namesrK).0clsrCs rF z$Analytics.process..~s0(S(S#CHH)=(S(S(S(S(S(SrG)rJ count_dictr=zUnsupported analytics_type='z)'. Supported types: line, bar, pie, area.)plot_im total_tracksclasswise_count)extract_tracksr+boxesr4rr7 update_graph collectionsrNclssr5 ValueErrorrlen track_ids)rCrHrJ_update_requiredrUrNs` rFprocesszAnalytics.process_s& C   9  Z ' '!!Q&!!!*T->>!C`tGX\`G`O Q$($5$5<$5$P$P!'G !D   Y0 0 0 + + + + + +!((S(S(S(S(S(S(S!S!SD *T->>!C`tGX\`G`O $($5$5!-$:LSWS\%6%%!'GGpDIpppqq qwS=P=PbfbtuuuurGNrrTdict[str, int] | Noner=strc |}?||S).aUpdate the graph with new data for single or multiple classes. Args: frame_number (int): The current frame number. count_dict (dict[str, int], optional): Dictionary with class names as keys and counts as values for multiple classes. If None, updates a single line graph. plot (str): Type of the plot. Options are 'line', 'bar', 'pie', or 'area'. Returns: (np.ndarray): Updated image containing the graph. Examples: >>> analytics = Analytics(analytics_type="bar") >>> frame_num = 10 >>> results_dict = {"person": 5, "car": 3} >>> updated_image = analytics.update_graph(frame_num, results_dict, plot="bar") NCountsz#7b0068*rrrc8i|]}|tjgSr#)nparray)rQkeys rF z*Analytics.update_graph..s"NNNSsBHRLLNNNrGrMg?)r alphaoz Data Points)r r!marker markersizelabelrc*g|]}j|Sr#)rA)rQrrrCs rF z*Analytics.update_graph..s!HHH$,U3HHHrG)r centerbottom)havar z upper left )locr2r edgecolorrc g|] }|z dz S)dr#)rQsizetotals rFrtz*Analytics.update_graph..s"EEEdte|c1EEErGZr )labels startangle textpropsautopctc&g|]\}}|d|ddS)z (z.1fz%)r#)rQrr percentages rFrtz*Analytics.update_graph..s1 r r rDUE:E!?!?Z!?!?!?!? r r rrGrz center left)rMr?rM)r0r{bbox_to_anchorg?g?)leftrightz#f0f0f0Tz--r) linestyler!rn)r r2)@rjappendr get_xdatafloat get_ydatar4r^r1set_data set_label set_color set_markerset_markersizer>listkeysvaluesrr<r8rkzippadclearitemsnext fill_betweenr=rAr3rtextget_x get_width get_heightrdr/legendsumrr9subplots_adjustr@grid set_titler0r2 set_xlabelr, set_ylabelr-r. get_textsrelimautoscale_viewr:drawrenderer buffer_rgbacv2cvtColorCOLOR_RGBA2BGRdisplay_output)rCrJrTr=x_datay_datarcountsr3 y_data_dictrrl max_lengthr rrcolorsbarsrcount percentages start_anglewedgesr` legend_labelsrrrHrs` @rFrZzAnalytics.update_graphsv(  Yty2244eL6I6IJJFYty2244eD>+;+; C((65+>+>?? [[ %??,,mmC')yS1A5TWCYCY'Z'ZK$;s+,,z99+-6+c2BQ UXYdehYiUjUjHjDk+l+l C(v;;00#ABBZF)00@@+6s+;ABB+? C(( #.#4#4#6#6  KC --EG((uD(QQQGLL#"&/"#'?Q#6!$222!  #KKED$666489I4J4J*51HHHHHHHw{{66{@@"%dF"3"3JCGLL cmmoo&99((E ##"m !#&dF"3"3))JCMM%((((<" aeanooooF EEEEfEEE    GKK6kgW[WdMeos( !s rY\]cepYqYq r r r v}I=iwxxx((c(>>> i(((  TTS DDD $*DMDMRRR 4rs#""""" IIIIIIII******yyyyy yyyyyrG