:j#!HddlZddlmZmZmZddlmZmZmZGddZ y)N)OptionalTupleUnion)TimelineSegment AnnotationceZdZdZ ddededededef dZdededefd Z d ed edefd Z ddedede ededededede e eefe eeefe eeeefffdZy )UEMSupportMixinu:Provides 'uemify' method with optional (à la NIST) collaruem referencecollar skip_overlapreturnc |dk(r|s|Sgg}}|dkDr{|jD]h}|j}|jt|d|zz |d|zz|j}|jt|d|zz |d|zzj|r>|j |D]*\\} } \} } | | k(r| | k(r|j| | z,||z} t | jj|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 g?)segments)support) itersegmentsstartappendrendco_iterrrgaps)selfr r r rcollarsoverlap_regionssegmenttsegment1track1segment2track2rs k/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/pyannote/metrics/utils.pyextrudezUEMSupportMixin.extrude&s< R< J#%r B;$113 JMMwq2;BKHIKKwq2;BKHI J :C:K:KI:V <6"6$6Xvx'Ff,<&&x(':;  < _,*22499#9FF hypothesisc|jd}|j|jd|jS)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_timelinebs@"))t)4 //U/;<$$&&r% annotationr-c|j}|jd}|j|D]:\}}|j|D]!}|j ||}|||f|||f<#<|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( %%' +++7 !*!2!28!< J Hg$//9 J",,W,G-7&8H-I 7E>* J Jr%N returns_uemreturns_timelinec|r|jj}|jj} || z} t| r| gng|j}t j d|j ||||}|j|d}|j|d}|r6|j||} |j|| }|j|| }||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) )rurizL'uem' was approximated by the union of 'reference' and 'hypothesis' extents.)r r intersection)mode) r*extentrr>warningswarnr$cropr.r:) rr r&r r rr;r<r_extenth_extentrAr-results r#uemifyzUEMSupportMixin.uemifys^ ; --/668H!..0779H(FVHB(}}.C MM, - ll3 &(46NN3^N< __S~_>  ++IzBH Y9Ij(;JZ(  sf F  xk !F r%)rF)NrFFF)__name__ __module__ __qualname____doc__rrfloatboolr$r.r:rrrrHr%r#r r #s$D !$%* :G:G%:G:G# :G08 :Gx''''**Z>*.!$)#((-N$N%NX&N N " N ! N"&Nj*,-j*h67j*h@ACNr%r ) rBtypingrrr pyannote.corerrrr rOr%r#rRs":))77~~r%