j#!PddlZddlmZmZmZddlmZmZmZGddZ dS)N)OptionalTupleUnion)TimelineSegment AnnotationceZdZdZ ddededededef d Zded edefd Z d ed edefdZ dded ede ededededede e eefe eeefe eeeefffdZdS)UEMSupportMixinu:Provides 'uemify' method with optional (à la NIST) collarFuem referencecollar skip_overlapreturnc 6|dkr|s|Sgg}}|dkr|D]n}|j}|t|d|zz |d|zz|j}|t|d|zz |d|zzo|rF||D]0\\} } \} } | | kr| | kr|| | z1||z} t | |S)aExtrude reference boundary collars from uem reference |----| |--------------| |-------------| uem |---------------------| |-------------------------------| extruded |--| |--| |---| |-----| |-| |-----| |-----------| |-----| Parameters ---------- uem : Timeline Evaluation map. reference : Annotation Reference annotation. collar : float, optional When provided, set the duration of collars centered around reference segment boundaries that are extruded from both reference and hypothesis. Defaults to 0. (i.e. no collar). skip_overlap : bool, optional Set to True to not evaluate overlap regions. Defaults to False (i.e. keep overlap regions). Returns ------- extruded_uem : Timeline r g?)segments)support) itersegmentsstartappendrendco_iterrrgaps)selfr r rrcollarsoverlap_regionssegmenttsegment1track1segment2track2rs y/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/pyannote/metrics/utils.pyextrudezUEMSupportMixin.extrude&sT< R<< <J#%r B;;$1133 J JMwq2;BKHHIIIKwq2;BKHHIIII  <:C:K:KI:V:V < <6"6$6Xvx''Ff,<,<&&x(':;;;;_,***224499#9FFF hypothesisc|d}||d|S)aReturn timeline common to both reference and hypothesis reference |--------| |------------| |---------| |----| hypothesis |--------------| |------| |----------------| timeline |--|-----|----|---|-|------| |-|---------|----| |----| Parameters ---------- reference : Annotation hypothesis : Annotation Returns ------- timeline : Timeline TcopyF) get_timelineupdate segmentation)rr r&timelines r#common_timelinezUEMSupportMixin.common_timelinebsO"))t)44 //U/;;<<<$$&&&r% annotationr-c |}|d}||D]C\}}||D](}|||}|||f|||f<)D|S)aProject annotation onto timeline segments reference |__A__| |__B__| |____C____| timeline |---|---|---| |---| projection |_A_|_A_|_C_| |_B_| |_C_| Parameters ---------- annotation : Annotation timeline : Timeline Returns ------- projection : Annotation Fr() candidate)emptyr*r get_tracks new_track) rr/r- projection timeline_segment_rtrack_tracks r#projectzUEMSupportMixin.projectws( %%'' +++77 !*!2!28!** Jr%N returns_uemreturns_timelinecH||}|} || z} t| r| gng|j}t jd|||||}||d}||d}|rB|||} | || }| || }||f} |r| |fz } |r| | fz } | S)a Crop 'reference' and 'hypothesis' to 'uem' support Parameters ---------- reference, hypothesis : Annotation Reference and hypothesis annotations. uem : Timeline, optional Evaluation map. collar : float, optional When provided, set the duration of collars centered around reference segment boundaries that are extruded from both reference and hypothesis. Defaults to 0. (i.e. no collar). skip_overlap : bool, optional Set to True to not evaluate overlap regions. Defaults to False (i.e. keep overlap regions). returns_uem : bool, optional Set to True to return extruded uem as well. Defaults to False (i.e. only return reference and hypothesis) returns_timeline : bool, optional Set to True to oversegment reference and hypothesis so that they share the same internal timeline. Returns ------- reference, hypothesis : Annotation Extruded reference and hypothesis annotations uem : Timeline Extruded uem (returned only when 'returns_uem' is True) timeline : Timeline: Common timeline (returned only when 'returns_timeline' is True) N)rurizL'uem' was approximated by the union of 'reference' and 'hypothesis' extents.)rr intersection)mode) r*extentrr>warningswarnr$cropr.r:) rr r&r rrr;r<r_extenth_extentrAr-results r#uemifyzUEMSupportMixin.uemifysX^ ; --//6688H!..007799H(F$>VHHB(}...C M, - - - ll3 &(466NN3^N<< __S~_>>   <++IzBBH Y99Ij(;;JZ(   sf F  " xk !F r%)r F)Nr FFF)__name__ __module__ __qualname____doc__rrfloatboolr$r.r:rrrrHr%r#r r #sDD !$%* :G:G:G%:G:G# :G08 :G:G:G:Gx'''''''**Z>*.!$)#((-NN$N%NX&N N " N ! N"&Nj*,-j*h67j*h@ACNNNNNNr%r ) rBtypingrrr pyannote.corerrrr rOr%r#rRs:))))))))))7777777777~~~~~~~~~~r%