o pi@sddlmZddlmZmZmZmZddlmZm Z ddl m Z m Z ddl mZGdddZGdd d ZGd d d ZGd d d ZdS))reduce)DictListOptionalSet)Audioget_audio_metadata) AnnotationSegment) ProtocolFilecs>eZdZdZdZd deffdd Zdedefd d Z Z S) LowerTemporalResolutionzArtificially degrade temporal resolution of reference annotation Parameters ---------- resolution : float, optional Target temporal resolution, in seconds. Defaults to 0.1 (100ms). annotation皙? resolutioncst||_dSN)super__init__r)selfr __class__V/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/audio/utils/preprocessors.pyr1s  z LowerTemporalResolution.__init__ current_filereturnc Cs|d}|}t|jddD]-\}\}}}|jt|j|jd}|jt|j|jd} t|| d} ||| |f<q|d} | | S)Nr T) yield_labelg?)startend annotated) empty enumerate itertracksrintrrr extentsupportcrop) rrr new_annotation new_tracksegment_label new_startnew_end new_segmentr#rrr__call__5s   z LowerTemporalResolution.__call__)r) __name__ __module__ __qualname____doc__preprocessed_keyfloatrr r r- __classcell__rrrrr &s r c @sxeZdZdZ  ddeedeeeeefdeeeeeffddZe deefd d Z d e de fd d Z dS)DeriveMetaLabelsaFilters out classes not defined in the `classes` list and synthesizes additional classes based on unions or intersection of classes. Parameters ---------- classes : List[str] All the "base" classes that should be used in the protocol's annotation's unions: Dict[str, List[str]], optional Unions of classes. The keys are the name of the new union classes, and the values are the list of classes that should used for these unions. intersections: Dict[str, List[str]], optional Intersections of classes. The keys are the name of the new intersections classes, and the values are the list of classes that should used for these intersections. Nclassesunions intersectionscCs<t||_|dur |nt|_|dur||_dSt|_dSr)setr6dictr7r8)rr6r7r8rrrrUs zDeriveMetaLabels.__init__rcCs*tt|jt|jt|jS)zA list of all the classes (base, union-based and intersection-based) that can be found in output annotations from this preprocessor)sortedlistr6r7keysr8rrrr all_classes_s  zDeriveMetaLabels.all_classesrc s|d|j}|jD]\}fdd|D}|j|d}||q|jD]\}}fdd|D}tdd|}|D]} ||| <qHq2|S) Nr csi|]}|qSrr).0k) union_labelrr osz-DeriveMetaLabels.__call__..)mappingcsg|]}|qSr)label_timeline)r@subclass)r rr ys z-DeriveMetaLabels.__call__..cSs ||Sr)r$)xyrrr~s z+DeriveMetaLabels.__call__..) subsetr6r7items rename_labelsupdater#r8r) rrderived subclassesrDmetalabel_annotintersect_label subclasses_tl overlap_tlsegr)r rBrr-is"    zDeriveMetaLabels.__call__)NN)r.r/r0r1rstrrrrpropertyr?r r r-rrrrr5Es   r5c@s"eZdZddZdefddZdS)WaveformcCs t|_dSr)r_audior>rrrrs zWaveform.__init__filecCs||\}}|Sr)rY)rrZwaveformr(rrrr-szWaveform.__call__N)r.r/r0rr r-rrrrrXsrXc@seZdZdefddZdS) SampleRaterZcCs t|jSr)r sample_rate)rrZrrrr-s zSampleRate.__call__N)r.r/r0r r-rrrrr\sr\N) functoolsrtypingrrrrpyannote.audio.core.iorr pyannote.corer r pyannote.databaser r r5rXr\rrrrs  @