o si 2@sdZddlZddlZddlmZddlmZddl m Z ddl m Z ddd Z ej ddd fd d Zd dZdej ddfddZddZddZdS)z} Methods which sonify annotations for "evaluation by ear". All functions return a raw signal at the specified sampling rate. N) as_strided)interp1d)util)chordcCs|dur*tdtjt|ddd|}|tt|d |d9}|dur=t|||jdd}t|}|D]/}t||}||jd}||krZ|S||krm|d||||d<|S||||<qD|S) aReturn a signal with the signal 'click' placed at each specified time Parameters ---------- times : np.ndarray times to place clicks, in seconds fs : int desired sampling rate of the output signal click : np.ndarray click signal, defaults to a 1 kHz blip length : int desired number of samples in the output signal, defaults to ``times.max()*fs + click.shape[0] + 1`` Returns ------- click_signal : np.ndarray Synthesized click signal Ng?ig?{Gz?rr) npsinpiarangeexpintmaxshapezeros)timesfsclicklength click_signaltimestartendrC/home/ubuntu/.local/lib/python3.10/site-packages/mir_eval/sonify.pyclickss$*"  rrc CsBd}|jdkrt|ddtjf|ddtjff}d}|dur+tt||}t||} |rI|jd|jdkrIt|t|| gf}|jd|jdkr_t d|jd|j|jd|jdkrut d |jd|jddg} g} | dkrd| d<| d| g| ||| krd| d<| || gtj |d | fd d }t| }t |dddfdk|dddf| k} |dd| f}t|| d| }|jd} t|d}t|}|jddkr|Stj|dd |k}||ddf}||}| dkr>t|dddf||ddd| fdd|dddf|dddffd}|t|}nt|dddfd|f}t|D]9\}}t|||||}dt||}t||}tjj|||dd }|dd|dt||7<qPt|}|t|jjkr||}|S)aKReverse synthesis of a time-frequency representation of a signal Parameters ---------- gram : np.ndarray ``gram[n, m]`` is the magnitude of ``frequencies[n]`` from ``times[m]`` to ``times[m + 1]`` Non-positive magnitudes are interpreted as silence. frequencies : np.ndarray array of size ``gram.shape[0]`` denoting the frequency (in Hz) of each row of gram times : np.ndarray, shape= ``(gram.shape[1],)`` or ``(gram.shape[1], 2)`` Either the start time (in seconds) of each column in the gram, or the time interval (in seconds) corresponding to each column. fs : int desired sampling rate of the output signal function : function function to use to synthesize notes, should be :math:`2\pi`-periodic length : int desired number of samples in the output signal, defaults to ``times[-1]*fs`` n_dec : int the number of decimals used to approximate each sonfied frequency. Defaults to 1 decimal place. Higher precision will be slower. threshold : float optimizes synthesis to only occur for frequencies that have a linear magnitude of at least one element in gram above the given threshold. Returns ------- output : np.ndarray synthesized version of the piano roll FrNTrz times.shape=z! is incompatible with gram.shape=zfrequencies.shape=)rrconstant)mode)axisprevious)kind bounds_error fill_valuersame)ndimr hstacknewaxisrrfloatrvstack ValueErrorminappendpad logical_andclipmaximumrrr tile enumerate_fast_synthesizeonesscipysignalconvolvelenabsfinfodtypetiny)gram frequenciesrrfunctionrn_dec thresholdtime_convertedlast_time_in_secspaddingstackingidxn_timesoutput freq_keep interpolatorr7n frequencywaveperiodfiltersignal_nnormrrrtime_frequency@sv. *     ,    (rSc Cszt||}td||}|dtjt|||}tt|t|jd}t||t |fd|j fd}|j S)zEfficiently synthesize a signal. Generate one cycle, and simulate arbitrary repetitions using array indexing tricks. g$@@r)rstrides) r roundrr r ceilr)rrr9itemsizeflat) rMrArr@r n_samples short_signal n_repeats long_signalrrrr4s   r4linearc Cst|}|durt||}t|d}tj|dd}t||dtj|||dddd}|t|}|durBt |f} nt||||dddd} | t|} | |t |S)aSonify a pitch contour. Parameters ---------- times : np.ndarray time indices for each frequency measurement, in seconds frequencies : np.ndarray frequency measurements, in Hz. Non-positive measurements or NaNs will be interpreted as un-voiced samples. fs : int desired sampling rate of the output signal amplitudes : np.ndarray amplitude measurements, nonnegative defaults to ``np.ones((length,))`` function : function function to use to synthesize notes, should be :math:`2\pi`-periodic length : int desired number of samples in the output signal, defaults to ``max(times)*fs`` kind : str Interpolation mode for the frequency and amplitude values. See: ``scipy.interpolate.interp1d`` for valid settings. Returns ------- output : np.ndarray synthesized version of the pitch contour NgF)copyr)r"r$r#r_) r)rrr r1 nan_to_numrr r r5cumsum) rr?r amplitudesr@rr"f_interpf_esta_esta_interprrr pitch_contours4  rgc Ksd}d}d}d}td||}t||d d|d} t|j|j} | t} | | dd9} d d|d d } t| | ||fi|S) aReverse synthesis of a chromagram (semitone matrix) Parameters ---------- chromagram : np.ndarray, shape=(12, times.shape[0]) Chromagram matrix, where each row represents a semitone [C->Bb] i.e., ``chromagram[3, j]`` is the magnitude of D# from ``times[j]`` to ``times[j + 1]`` times : np.ndarray, shape=(len(chord_labels),) or (len(chord_labels), 2) Either the start time of each column in the chromagram, or the time interval corresponding to each column. fs : int Sampling rate to synthesize audio data at **kwargs Additional keyword arguments to pass to :func:`mir_eval.sonify.time_frequency` Returns ------- output : np.ndarray Synthesized chromagram H rTrrg{@Eg(@) r r r r2Tastyper)reshaperS) chromagramrrkwargs n_octaves base_notemeanstdnotesshepard_weightr>r?rrrchroma@s  rycKsJt|t|\}}}tddt||Dj}t|||fi|S)aSynthesizes chord labels Parameters ---------- chord_labels : list of str List of chord label strings. intervals : np.ndarray, shape=(len(chord_labels), 2) Start and end times of each chord label fs : int Sampling rate to synthesize at **kwargs Additional keyword arguments to pass to :func:`mir_eval.sonify.time_frequency` Returns ------- output : np.ndarray Synthesized chord labels cSsg|] \}}t||qSr)r roll).0interval_bitmaprootrrr s zchords..) rvalidate_intervalsr encode_manyr arrayziprnry) chord_labels intervalsrrrrootsinterval_bitmaps_rqrrrchordsos r)NN)__doc__numpyr scipy.signalr6numpy.lib.stride_tricksrscipy.interpolaterrrrr rSr4rgryrrrrrs       2  ! I /