o si&x@sdZddlmZddlmZddlZddlmZddl Z ddl m Z ddl mZmZddl mZdd lmZmZmZdd lmZmZd d lmZd d lmZmZeZd0ddZ      d1ddZ       d2ddZ GdddeZ!d0ddZ"      d3ddZ# d4ddZ$ d4ddZ%d5d d!Z& "  #   $ % d6d&d'Z'd(d)Z(d*d+Z)d7d,d-Z*d7d.d/Z+ee(Z,ee)Z-dS)8zDisplay functions) defaultdict)WeakKeyDictionaryN) spectrogram) Rectangle) FuncFormatterMultipleLocator) Formatter)LinearSegmentedColormapLogNormColorConverter)BboxTransformedBbox)freq_to_voicing) midi_to_hz hz_to_midicCsVd}|dur|durddlm}|}|sd}|}|tvr'tt|<||fS)a{Get or construct the target axes object for a new plot. Parameters ---------- ax : matplotlib.pyplot.axes, optional If provided, return this axes object directly. fig : matplotlib.figure.Figure, optional The figure to query for axes. By default, uses the current figure `plt.gcf()`. Returns ------- ax : matplotlib.pyplot.axes An axis handle on which to draw the segmentation. If none is provided, a new set of axes is created. new_axes : bool If `True`, the axis object was newly constructed. If `False`, the axis object already existed. FNrT)matplotlib.pyplotpyplotgcfget_axesgca__AXMAPdict)axfignew_axespltrD/home/ubuntu/.local/lib/python3.10/site-packages/mir_eval/display.py __get_axess  rFc Ks^|durt}|dd|dd|dtddd t|}td d } t|d \}} |durKt|d tjdt|dttjdndt|vr_|t|d <t|t|d<t|d }t|d} | rr| g|dur|durd\}}| } n|dur|durd} nt dt} |D]V}|| vrqzt | }Wnt yt|} | t|d<t | }Ynwdd|D}|d|d|dd| | |<| ||| |||| |d<qt||D]9\}}|j|d|d f||d| |}| |dd|r,|j|f|d|f| ddd|}||q|S)aUPlot a segmentation as a set of disjoint rectangles. Parameters ---------- intervals : np.ndarray, shape=(n, 2) segment intervals, in the format returned by :func:`mir_eval.io.load_intervals` or :func:`mir_eval.io.load_labeled_intervals`. labels : list, shape=(n,) reference segment labels, in the format returned by :func:`mir_eval.io.load_labeled_intervals`. base : number The vertical position of the base of the rectangles. By default, this will be the bottom of the plot. height : number The height of the rectangles. By default, this will be the top of the plot (minus ``base``). .. note:: If either `base` or `height` are provided, both must be provided. text : bool If true, each segment's label is displayed in its upper-left corner text_kw : dict If ``text == True``, the properties of the text object can be specified here. See ``matplotlib.pyplot.Text`` for valid parameters ax : matplotlib.pyplot.axes An axis handle on which to draw the segmentation. If none is provided, a new set of axes is created. prop_cycle : cycle.Cycler An optional property cycle object to specify style properties. If not provided, the default property cycler will be retrieved from matplotlib. **kwargs Additional keyword arguments to pass to ``matplotlib.patches.Rectangle``. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes Nvatopclip_onTbboxroundwhiteboxstyle facecolorr linewidthr prop_cycleaxes.prop_cycle prop_iterrr6When specifying base or height, both must be provided.cSi|] \}}|dvr||qS)colorr( edgecolorr*r.0kvrrr zsegments..r(r3labelr)yminymaxi offset pointsxyxycoordsxytext textcoords)r setdefaultnp atleast_2drrmplrcParamsiter set_yticksget_xaxis_transform ValueErrornext StopIterationitemspopcopyupdatezipaxvspanannotate set_clip_path) intervalslabelsbaseheighttexttext_kwrr,kwargs seg_def_stylerr. transformseg_maplab propertiesstyleivalrectannrrrsegmentsBsz3                &  riTc Kst|d\}} |dur$t|dtjdt|dttjdndt|vr8|t|d<t|t|d<t|d}t|d} t|}|durVt|dg}nt |}d} |rw|t t |t |} | |krvt |dkrvd } n |r||} nt t |} | t|d<t d d }zt| }Wntyt|} | t|d<t| }Ynwd d |D}|d|d|dd|| |durtt | }|durd }t|r|t|}t }t||| D] \}}}||f||<qtt }t||D]\}}||vr q|||d|d |dfq|D]}|j||||fi||ddq"| rG|jt |ddd|rk|jd dd|g|||j| dd|jt || |S)abPlot labeled intervals with each label on its own row. Parameters ---------- intervals : np.ndarray, shape=(n, 2) segment intervals, in the format returned by :func:`mir_eval.io.load_intervals` or :func:`mir_eval.io.load_labeled_intervals`. labels : list, shape=(n,) reference segment labels, in the format returned by :func:`mir_eval.io.load_labeled_intervals`. label_set : list An (ordered) list of labels to determine the plotting order. If not provided, the labels will be inferred from ``ax.get_yticklabels()``. If no ``yticklabels`` exist, then the sorted set of unique values in ``labels`` is taken as the label set. base : np.ndarray, shape=(n,), optional Vertical positions of each label. By default, labels are positioned at integers ``np.arange(len(labels))``. height : scalar or np.ndarray, shape=(n,), optional Height for each label. If scalar, the same value is applied to all labels. By default, each label has ``height=1``. extend_labels : bool If ``False``, only values of ``labels`` that also exist in ``label_set`` will be shown. If ``True``, all labels are shown, with those in `labels` but not in `label_set` appended to the top of the plot. A horizontal line is drawn to indicate the separation between values in or out of ``label_set``. ax : matplotlib.pyplot.axes An axis handle on which to draw the intervals. If none is provided, a new set of axes is created. tick : bool If ``True``, sets tick positions and labels on the y-axis. prop_cycle : cycle.Cycler An optional property cycle object to specify style properties. If not provided, the default property cycler will be retrieved from matplotlib. **kwargs Additional keyword arguments to pass to `matplotlib.collection.PolyCollection`. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes r+Nr,r-r.rZFrTrr)cSr1r2rr5rrrr9;r:z%labeled_intervals..r(r3r;r7?)r3alphay)axisbottom)r )!rrrFrIrJrKrGrHgetlistsortedsetlenrrOrPrQrRrTarangeisscalar ones_likerUrappend broken_barhaxhlinegridrLset_yticklabelsyaxisset_major_formatterIntervalFormatter)rYrZ label_setr[r\ extend_labelsrtickr,r_rr.extendedticksrerdseg_yybaseyheightrcxvalsrfrrrlabeled_intervalss~H                &  rc@s"eZdZdZddZdddZdS)r~zTicker formatter for labeled interval plots. Parameters ---------- base : array-like of int The base positions of each label ticks : array-like of string The labels for the ticks cCsddt||D|_dS)NcSsi|] \}}t||qSr)intr5rrrr9ysz.IntervalFormatter.__init__..)rU_map)selfr[rrrr__init__xszIntervalFormatter.__init__NcCs|jt|dS)z:Map the input position to its corresponding interval label)rror)rxposrrr__call__{szIntervalFormatter.__call__N)__name__ __module__ __qualname____doc__rrrrrrr~ls r~c Ks|dur ttt|}t|d\}}t|j}t|ddd|ddd|dddD]\}}} t||f| |d|q-|S)apPlot a hierarchical segmentation Parameters ---------- intervals_hier : list of np.ndarray A list of segmentation intervals. Each element should be an n-by-2 array of segment intervals, in the format returned by :func:`mir_eval.io.load_intervals` or :func:`mir_eval.io.load_labeled_intervals`. Segmentations should be ordered by increasing specificity. labels_hier : list of list-like A list of segmentation labels. Each element should be a list of labels for the corresponding element in `intervals_hier`. levels : list of string Each element ``levels[i]`` is a label for the ```i`` th segmentation. This is used in the legend to denote the levels in a segment hierarchy. ax : matplotlib.pyplot.axes An axis handle on which to draw the intervals. If none is provided, a new set of axes is created. **kwargs Additional keyword arguments to `labeled_intervals`. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes Nr+)r;r)rprangersrpatchesrUr) intervals_hier labels_hierlevelsrr__ n_patchesintslabskeyrrr hierarchys 4rcKs|durt}|dd|dd|dtddd t|}t|d \}}|durFt|d tjd t|d ttjd nd t|vrZ|t|d <t|t|d <t|d }t|d } |durw|durwd\}}| } n|dur|durd} nt dzt | } Wnt yt|} | t|d <t | } Ynwdd| D} | |d| vr| dd|j||||fd| i| } |rt| |D]\}}|j|f|jdd|f| ddd|q|r|g|S)aPlot event times as a set of vertical lines Parameters ---------- times : np.ndarray, shape=(n,) event times, in the format returned by :func:`mir_eval.io.load_events` or :func:`mir_eval.io.load_labeled_events`. labels : list, shape=(n,), optional event labels, in the format returned by :func:`mir_eval.io.load_labeled_events`. base : number The vertical position of the base of the line. By default, this will be the bottom of the plot. height : number The height of the lines. By default, this will be the top of the plot (minus `base`). .. note:: If either `base` or `height` are provided, both must be provided. ax : matplotlib.pyplot.axes An axis handle on which to draw the segmentation. If none is provided, a new set of axes is created. text_kw : dict If `labels` is provided, the properties of the text objects can be specified here. See `matplotlib.pyplot.Text` for valid parameters prop_cycle : cycle.Cycler An optional property cycle object to specify style properties. If not provided, the default property cycler will be retrieved from matplotlib. **kwargs Additional keyword arguments to pass to `matplotlib.pyplot.vlines`. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes Nr r!r"Tr#r$r%r&r+r,r-r.r/r0cSr1))r3 linestyler*rr5rrrr9szevents..colorsr3rarr>r@rA)rrFrGasarrayrrrIrJrKrMrNrOrPrQrTrRvlinesrU get_pathsrWverticesrL)timesrZr[r\rr^r,r_rr.rardrelinespathrcrrreventssd/               rcKs8t|d\}}|dur$t|dtjdt|dttjdndt|vr8|t|d<t|t|d<t|d}t|d}t|}ttj|tj d\}} | t } dt | dd| ddk} t td g| t| gg} gg} } t| | ddD]\} }t| |}| | r| |q||r| |qzt|}Wntyt|}|t|d<t|}Ynw|||r|d k}t||||<|jtd| D]}|j||||fi||d dq|r|d d d |d <| D]}|j||||fi|q|S)aVisualize pitch contours Parameters ---------- times : np.ndarray, shape=(n,) Sample times of frequencies frequencies : np.ndarray, shape=(n,) frequencies (in Hz) of the pitch contours. Voicing is indicated by sign (positive for voiced, non-positive for non-voiced). midi : bool If `True`, plot on a MIDI-numbered vertical axis. Otherwise, plot on a linear frequency axis. unvoiced : bool If `True`, unvoiced pitch contours are plotted and indicated by transparency. Otherwise, unvoiced pitch contours are omitted from the display. ax : matplotlib.pyplot.axes An axis handle on which to draw the pitch contours. If none is provided, a new set of axes is created. prop_cycle : cycle.Cycler An optional property cycle object to specify style properties. If not provided, the default property cycler will be retrieved from matplotlib. **kwargs Additional keyword arguments to `matplotlib.pyplot.plot`. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes r+Nr,r-r.dtyperrrr;rk?rj)rrrFrIrJrKrGrrfloat64astypebool flatnonzerounique concatenatersrUslicerwallrOrPrTrr|set_minor_locatorrplotrRro)r frequenciesmidiunvoicedrr,r_rr.voicings v_changesv_slicesu_slicesstartendidxrerrrpitch!sT)      "           rcKst|d\}}|dur$t|dtjdt|dttjdndt|vr8|t|d<t|t|d<t|d}t|d}zt|} Wntyat|}|t|d<t|} Ynw| || } | dd| ddd | d<g} g} g} g}t ||D]S\}}t |sqttj|tjd \}}|d k}||}||t}|rt|}t|}| |gt|| ||| |gt |||||q|j| | fi| |r|j| |fi| |r|jtd |S) aVisualize multiple f0 measurements Parameters ---------- times : np.ndarray, shape=(n,) Sample times of frequencies frequencies : list of np.ndarray frequencies (in Hz) of the pitch measurements. Voicing is indicated by sign (positive for voiced, non-positive for non-voiced). `times` and `frequencies` should be in the format produced by :func:`mir_eval.io.load_ragged_time_series` midi : bool If `True`, plot on a MIDI-numbered vertical axis. Otherwise, plot on a linear frequency axis. unvoiced : bool If `True`, unvoiced pitches are plotted and indicated by transparency. Otherwise, unvoiced pitches are omitted from the display. ax : matplotlib.pyplot.axes An axis handle on which to draw the pitch contours. If none is provided, a new set of axes is created. prop_cycle : cycle.Cycler An optional property cycle object to specify style properties. If not provided, the default property cycler will be retrieved from matplotlib. **kwargs Additional keyword arguments to `plt.scatter`. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes r+Nr,r-r.r;rkrrjrrr)rrrFrIrJrKrOrPrTrSrRrorUrsrrGrrrrrsumextendrscatterr|rr)rrrrrr,r_rr. style_voicedstyle_unvoiced voiced_times voiced_freqsunvoiced_timesunvoiced_freqstfreqsrrn_voicedrrr multipitchsX-          rcKsd|dur|dur tdt|}td}t|t|tf|d|d|}|j t d|S)aPlot a quantized piano roll as intervals Parameters ---------- intervals : np.ndarray, shape=(n, 2) timing intervals for notes pitches : np.ndarray, shape=(n,), optional pitches of notes (in Hz). midi : np.ndarray, shape=(n,), optional pitches of notes (in MIDI numbers). At least one of ``pitches`` or ``midi`` must be provided. ax : matplotlib.pyplot.axes An axis handle on which to draw the intervals. If none is provided, a new set of axes is created. **kwargs Additional keyword arguments to :func:`labeled_intervals`. Returns ------- ax : matplotlib.pyplot.axes._subplots.AxesSubplot A handle to the (possibly constructed) plot axes Nz5At least one of `midi` or `pitches` must be provided.F)rrrr) rNrrGrtrr$rrr|rr)rYpitchesrrr_scalerrr piano_rolls"  r"V?Nonegouraudc  Kst|d\}} t|}|durddtt|D}| ddd} g} t|D]#\} }t|fd|i| \}}}| || durH| } q)| |7} q)| }|d}t }|durut |d t jd t |d tt jd nd t |vr|t |d <t|t |d <t |d }t |d }t| D]Z\} }zt|}Wntyt|}|t |d <t|}Ynw|j|d |d }t|| d|g}|j||||t||d|||d|t||fdd||| dq|S)aSource-separation visualization Parameters ---------- sources : np.ndarray, shape=(nsrc, nsampl) A list of waveform buffers corresponding to each source fs : number > 0 The sampling rate labels : list of strings An optional list of descriptors corresponding to each source alpha : float in [0, 1] Maximum alpha (opacity) of spectrogram values. ax : matplotlib.pyplot.axes An axis handle on which to draw the spectrograms. If none is provided, a new set of axes is created. rasterized : bool If `True`, the spectrogram is rasterized. edgecolors : str or None The color of the edges of the spectrogram patches. Set to "None" (default) to disable edge coloring. shading : str The shading method to use for the spectrogram. See `matplotlib.pyplot.pcolormesh` for valid options. prop_cycle : cycle.Cycler An optional property cycle object to specify colors for each signal. If not provided, the default property cycler will be retrieved from matplotlib. **kwargs Additional keyword arguments to ``scipy.signal.spectrogram`` Returns ------- ax The axis handle for this plot r+NcSsg|]}d|dqS)zSource dr)r6rrrr ^szseparation..scalingspectrumfsgư>r,r-r.r3)rk)rrrg)vminvmax)cmapnorm rasterized edgecolorsshadingr)r3r;)rrGrHrrsrF enumeraterrwrSmaxr rrIrJrKrOrPto_rgbar from_list pcolormeshr add_patchrmin)sourcesrrZrkrrrrr,r_rcumspecspecsisrcrrspecref_maxref_min color_convr.rdr3rrrr separation)sd/                rcCsgd}tt|d}|dkr|d}|d}t|d}t|dd}|dkr4||d|dS||d|dt|d d S) zFormat midi notes for ticker decoration. Inputs x are interpreted as midi numbers, and converted to [NOTE][OCTAVE]+[cents]. ) CzC#DzD#EFzF#GzG#AzA#Brrj rrs2ddz+02d)floatrGmodr)rrNOTEScentsroctaverrr__ticker_midi_notes "rcCs t|dS)zwFormat midi pitches for ticker decoration. Inputs x are interpreted as midi numbers, and converted to Hz. g)r)rrrrr__ticker_midi_hzs rcCs8t|d\}}|jt|jD]}|dqdS)zSet the y-axis of the given axes to MIDI notes Parameters ---------- ax : matplotlib.pyplot.axes The axes handle to apply the ticker. By default, uses the current axes handle. r+baselineN)rr|r} FMT_MIDI_NOTEget_ticklabelsset_verticalalignment)rrrrrr ticker_notess   r cCst|d\}}|jtdS)zSet the y-axis of the given axes to MIDI frequencies Parameters ---------- ax : matplotlib.pyplot.axes The axes handle to apply the ticker. By default, uses the current axes handle. r+N)rr|r} FMT_MIDI_HZ)rrrrr ticker_pitchs r)NN)NNFNNN)NNNTNTN)NNNNNN)FFNN)NNN)rNrNTrrNr).r collectionsrweakrefrnumpyrG scipy.signalr matplotlibrImatplotlib.patchesrmatplotlib.tickerrrrmatplotlib.colorsr r r matplotlib.transformsr r melodyrutilrrrrrirr~rrrrrrrrr rr rrrrrsz       ,  ' / u h o4 {