o Xi%)@sdZddlmZddlmZddlmZmZmZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZdZgdZddZddZdZddddd ZdS)!zModule to generate ascii charts. This module provides a single function `plot` that can be used to generate an ascii chart from a series of numbers. The chart can be configured via several options to tune the output. ) annotations)Mapping)ceilfloorisnanzzzzzzzzzzzzzzzzzz)plotblackredgreenyellowbluemagentacyan lightgraydefaultdarkgraylightred lightgreen lightyellow lightblue lightmagenta lightcyanwhiteresetcCs t| SNrnrT/home/ubuntu/.local/lib/python3.10/site-packages/onnx_ir/_thirdparty/asciichartpy.py_isnumRs r cCs|s|S||tSr)r)charcolorrrrcoloredVs r#) u┼u┤u╶u╴u─u╰u╭u╮u╯u│N) bin_edgescfgc*s.t|dkrdSt|dtstdd|DrdS|g}|dur*t|ts*td|p-i}|ddg}|dttt d d |D|d t tt d d |D|d t }krct d}|dd}|d|}|dkr{||ndt t}fddfdd} |} d|D] } t t| q|7|dd} |dd} fdd t| dD}t|dD]6}| ||| r| nd}|||t |t|d<|dkr|dn|d|||d<q|dd}t |r|d|| | ||d<t|D]\}} ||t|}tt| dD]}| |d}| |d}t|rPt|rPq6t|rmt |rmt|d||| | |||<q6t |rt|rt|d||| | |||<q6| |}| |}||krt|d||| |||<q6||krt|d|nt|d||| |||<||krt|d |nt|d!||| |||<t||d}t ||}t||D]}t|d"||| |||<qq6q"d#d$d |D}|dust|dkr|Sd}|t|}| |d}t|}t d||}g} |d}!d}"||"}#t|}$t|$|#}%|%d}&t |"|!|%||&}'||!krt||!|$}(| ||(}| ||t||'7}||!ksed%|d%|'| })|d#|)S)&u Generate an ascii chart for a series of numbers. `series` should be a list of ints or floats. Missing data values in the series can be specified as a NaN. In Python versions less than 3.5, use float("nan") to specify an NaN. With 3.5 onwards, use math.nan to specify a NaN. >>> series = [1,2,3,4,float("nan"),4,3,2,1] >>> print(plot(series)) 4.00 ┤ ╭╴╶╮ 3.00 ┤ ╭╯ ╰╮ 2.00 ┤╭╯ ╰╮ 1.00 ┼╯ ╰ `series` can also be a list of lists to support multiple data series. >>> series = [[10,20,30,40,30,20,10], [40,30,20,10,20,30,40]] >>> print(plot(series, cfg={'height': 3})) 40.00 ┤╮ ╭╮ ╭ 30.00 ┤╰╮╯╰╭╯ 20.00 ┤╭╰╮╭╯╮ 10.00 ┼╯ ╰╯ ╰ `bin_edges` is an optional list of bin edges to display on the x-axis. If provided, the x-axis will be labeled with the bin edges. If there are too many bin edges to fit on the x-axis, some labels will be dropped and they will be spaced out evenly to fit the width of the chart. The labels will be formatted using the `x_format` option in `cfg`. `cfg` is an optional dictionary of various parameters to tune the appearance of the chart. `min` and `max` will clamp the y-axis and all values: >>> series = [1,2,3,4,float("nan"),4,3,2,1] >>> print(plot(series, cfg={'min': 0})) 4.00 ┼ ╭╴╶╮ 3.00 ┤ ╭╯ ╰╮ 2.00 ┤╭╯ ╰╮ 1.00 ┼╯ ╰ 0.00 ┤ >>> print(plot(series, cfg={'min': 2})) 4.00 ┤ ╭╴╶╮ 3.00 ┤ ╭╯ ╰╮ 2.00 ┼─╯ ╰─ >>> print(plot(series, cfg={'min': 2, 'max': 3})) 3.00 ┤ ╭─╴╶─╮ 2.00 ┼─╯ ╰─ `height` specifies the number of rows the graph should occupy. It can be used to scale down a graph with large data values: >>> series = [10,20,30,40,50,40,30,20,10] >>> print(plot(series, cfg={'height': 4})) 50.00 ┤ ╭╮ 40.00 ┤ ╭╯╰╮ 30.00 ┤ ╭╯ ╰╮ 20.00 ┤╭╯ ╰╮ 10.00 ┼╯ ╰ `format` specifies a Python format string used to format the labels on the y-axis. The default value is "{:8.2f} ". This can be used to remove the decimal point: >>> series = [10,20,30,40,50,40,30,20,10] >>> print(plot(series, cfg={'height': 4, 'format':'{:8.0f}'})) 50 ┤ ╭╮ 40 ┤ ╭╯╰╮ 30 ┤ ╭╯ ╰╮ 20 ┤╭╯ ╰╮ 10 ┼╯ ╰ rcss|]}t|VqdSrr).0rrrr szplot..Nz cfg must be a dictionary or NonecolorsmincSg|] }|D]}|qqSrrr'ijrrr zplot..maxcSr+rrr,rrrr/r0symbolsz*The min value cannot exceed the max value.offsetheightcstt|Sr)r*r1r)maximumminimumrrclampszplot..clampcstt|Sr)intround)y)r9min2ratiorrscaledszplot..scaledformatz{:8.2f} x_formatz{:4.4f}csg|]}dgqS) r)r'r-)widthrrr/s  cSsg|] }d|qS)r&)joinrstrip)r'rowrrrr/ r0rB)len isinstancelistallr TypeErrorgetr*filterr r1_DEFAULT_SYMBOLS ValueErrorrrranger@ enumeraterr#rLr:append)*seriesr$r%r)r2intervalr3r5max2r?rowsseries_i placeholder x_placeholderresultr<labeld0r-r"xd1y0y1startendthe_plotcurrent_location leading_spacex_label x_label_sizex_leading_spacex_labelsworkable_width min_spacingnum_labels_can_fit labels_countnum_labels_to_display num_spacesspacing bin_index x_labels_textr)r9r7r=r8r>rCrr`s I         "*    "" $$ "!        r) __doc__ __future__rcollections.abcrmathrrrrr r r r r rrrrrrrrrrrr__all__r r#rVrrrrrs6