o piz@sdZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZddlZddlmZddlmZmZmZddlmZmZdd lmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*errddl+Z,Gd ddZ-dS)a ########## Annotation ########## .. plot:: pyplots/annotation.py :class:`pyannote.core.Annotation` instances are ordered sets of non-empty tracks: - ordered, because segments are sorted by start time (and end time in case of tie) - set, because one cannot add twice the same track - non-empty, because one cannot add empty track A track is a (support, name) pair where `support` is a Segment instance, and `name` is an additional identifier so that it is possible to add multiple tracks with the same support. To define the annotation depicted above: .. code-block:: ipython In [1]: from pyannote.core import Annotation, Segment In [6]: annotation = Annotation() ...: annotation[Segment(1, 5)] = 'Carol' ...: annotation[Segment(6, 8)] = 'Bob' ...: annotation[Segment(12, 18)] = 'Carol' ...: annotation[Segment(7, 20)] = 'Alice' ...: which is actually a shortcut for .. code-block:: ipython In [6]: annotation = Annotation() ...: annotation[Segment(1, 5), '_'] = 'Carol' ...: annotation[Segment(6, 8), '_'] = 'Bob' ...: annotation[Segment(12, 18), '_'] = 'Carol' ...: annotation[Segment(7, 20), '_'] = 'Alice' ...: where all tracks share the same (default) name ``'_'``. In case two tracks share the same support, use a different track name: .. code-block:: ipython In [6]: annotation = Annotation(uri='my_video_file', modality='speaker') ...: annotation[Segment(1, 5), 1] = 'Carol' # track name = 1 ...: annotation[Segment(1, 5), 2] = 'Bob' # track name = 2 ...: annotation[Segment(12, 18)] = 'Carol' ...: The track name does not have to be unique over the whole set of tracks. .. note:: The optional *uri* and *modality* keywords argument can be used to remember which document and modality (e.g. speaker or face) it describes. Several convenient methods are available. Here are a few examples: .. code-block:: ipython In [9]: annotation.labels() # sorted list of labels Out[9]: ['Bob', 'Carol'] In [10]: annotation.chart() # label duration chart Out[10]: [('Carol', 10), ('Bob', 4)] In [11]: list(annotation.itertracks()) Out[11]: [(, 1), (, 2), (, u'_')] In [12]: annotation.label_timeline('Carol') Out[12]: , ])> See :class:`pyannote.core.Annotation` for the complete reference. N) defaultdict) HashableOptionalDictUnionIterableListSetTextIOTupleIteratorText TYPE_CHECKING) SortedDict)PYANNOTE_SEGMENTPYANNOTE_TRACKPYANNOTE_LABEL)Segment SlidingWindow)Timeline)SlidingWindowFeature)string_generator int_generator)LabelKeySupportLabelGenerator TrackNameCropModec @seZdZdZe  ddddeedeeddfdd Zddeedeefd d Ze d d Z e j defdd Z ddZ ddZ ddZddZdeeeeffddZddZ ddedeeeeefeeeefffddZdd Zdd"edefd#d$Zdd&d'Zdd(d)Zd*eeeffd+d,Zdee fd-d.Z!de fd/d0Z"d1e#fd2d3Z$dee fd4d5Z%de fd6d7Z&d1e#fd8d9Z'dd;e(dZ* :dd?e(ddYedefdkdlZ?dYede@fdmdnZAddoede9eee@ffdpdqZBdd;ee(deefdrdsZCddueDddfdvdwZE  t !ddxeeFdueDd"eddfdydzZGddueDddfd{d|ZHdd~e@ddfddZId%ddeeeeefeeefffddZJd%ddeKjLfddZM   dd;eedee@eNfdBee9eOdee@fddZPe  ddeeeeefdeedeeddfddZQddZRdS) AnnotationaAnnotation Parameters ---------- uri : string, optional name of annotated resource (e.g. audio or video file) modality : string, optional name of annotated modality Returns ------- annotation : Annotation New annotation Ndfz pd.DataFrameurimodalityreturncCs$|tttg}t|jdd||S)NF)index)rrrr from_records itertuples)clsr!r"r#r)L/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/core/annotation.pyfrom_dfszAnnotation.from_dfcCs0||_||_t|_i|_i|_d|_d|_dS)NT)_urir#r_tracks_labels_labelNeedsUpdate _timeline_timelineNeedsUpdate)selfr"r#r)r)r*__init__s zAnnotation.__init__cCs|jSN)r,r2r)r)r*r"szAnnotation.uricCs>|D] }|j|dd}||_q|jdd}||_||_dS)NFcopy)labelslabel_timeliner" get_timeliner,)r2r"labeltimeliner)r)r*r"s   cCstdd|jD}dd|D}|jddD]\}}}||vr)|||q|D]&}||rDt|||jd|j|<d|j|<q,|j|d|j|dq,dS) Ncss|] \}}|r|VqdSr4r)).0r;updater)r)r* s  z+Annotation._updateLabels..cSsi|]}|gqSr)r)r=r;r)r)r* z,Annotation._updateLabels..T yield_labelsegmentsr"F) setr/items itertracksappendrr"r.pop)r2r> _segmentssegmenttrackr;r)r)r* _updateLabelss zAnnotation._updateLabelscC t|jS)zhNumber of segments >>> len(annotation) # annotation contains three segments 3 lenr-r5r)r)r*__len__s zAnnotation.__len__cCs|Sr4)__bool__r5r)r)r* __nonzero__szAnnotation.__nonzero__cCst|jdkS)zEmptiness >>> if annotation: ... # annotation is not empty ... else: ... # annotation is empty rrQr5r)r)r*rTszAnnotation.__bool__ccs(|jddD] \}}}||fVqdS)zIterate over (segment, label) pairs in chronological order Examples -------- >>> for segment, label in annotation: ... # do something with the segment and its label TrCN)rI)r2rM_r;r)r)r*__iter__s zAnnotation.__iter__cCrP)a Iterate over segments (in chronological order) >>> for segment in annotation.itersegments(): ... # do something with the segment See also -------- :class:`pyannote.core.Segment` describes how segments are sorted. )iterr-r5r)r)r* itersegmentss zAnnotation.itersegmentsFrDccsT|jD]!\}}t|dddD]\}}|r!|||fVq||fVqqdS)aNIterate over tracks (in chronological order) Parameters ---------- yield_label : bool, optional When True, yield (segment, track, label) tuples, such that annotation[segment, track] == label. Defaults to yielding (segment, track) tuple. Examples -------- >>> for segment, track in annotation.itertracks(): ... # do something with the track >>> for segment, track, label in annotation.itertracks(yield_label=True): ... # do something with the track and its label cSst|dt|dfS)Nrr)str)tlr)r)r**sz'Annotation.itertracks..keyN)r-rHsorted)r2rDrMtracksrNlblr)r)r*rIs  zAnnotation.itertrackscCst|j|jd|_d|_dS)NrEF)rr-r"r0r1r5r)r)r*_updateTimeline1s zAnnotation._updateTimelineTr7cCs"|jr||r|jS|jS)aMGet timeline made of all annotated segments Parameters ---------- copy : bool, optional Defaults (True) to returning a copy of the internal timeline. Set to False to return the actual internal timeline (faster). Returns ------- timeline : Timeline Timeline made of all annotated segments. Note ---- In case copy is set to False, be careful **not** to modify the returned timeline, as it may lead to weird subsequent behavior of the annotation instance. )r1rbr0r7)r2r7r)r)r*r:5s  zAnnotation.get_timelineothercC.t|jdd|jdd}tdd|DS)zEquality >>> annotation == other Two annotations are equal if and only if their tracks and associated labels are equal. TrCcss|] \}}||kVqdSr4r)r=t1t2r)r)r*r?[z$Annotation.__eq__..) itertools zip_longestrIallr2rc pairOfTracksr)r)r*__eq__PszAnnotation.__eq__cCrd) InequalityTrCcss|] \}}||kVqdSr4r)rer)r)r*r?crhz$Annotation.__ne__..)rirjrIanyrlr)r)r*__ne__]szAnnotation.__ne__includedcCs||jddvS)aInclusion Check whether every segment of `included` does exist in annotation. Parameters ---------- included : Segment or Timeline Segment or timeline being checked for inclusion Returns ------- contains : bool True if every segment in `included` exists in timeline, False otherwise Fr6)r:)r2rrr)r)r* __contains__eszAnnotation.__contains__c cs|jr|jnd}t|trd|vrd|d}t||jddD],\}}}t|tr:d|vr:d|d}t|d|d |jd d|jd d |d Vq"d S)zGenerate lines for an RTTM file for this annotation Returns ------- iterator: Iterator[str] An iterator over RTTM text lines z zSSpace-separated RTTM file format does not allow file URIs containing spaces (got: "").TrCzPSpace-separated RTTM file format does not allow labels containing spaces (got: "zSPEAKER z 1 .3fz z N)r" isinstancer ValueErrorrIstartduration)r2r"msgrMrVr;r)r)r* _iter_rttmxs(zAnnotation._iter_rttmcCddd|DS)zSerialize annotation as a string using RTTM format Returns ------- serialized: str RTTM string cSg|]}|qSr)r)r=liner)r)r* z&Annotation.to_rttm..)joinr|r5r)r)r*to_rttmzAnnotation.to_rttmfilecC|D]}||qdS)zDump annotation to file using RTTM format Parameters ---------- file : file object Usage ----- >>> with open('file.rttm', 'w') as file: ... annotation.write_rttm(file) N)r|writer2rrr)r)r* write_rttm zAnnotation.write_rttmccsj|jddD]+\}}}t|trd|vrd|d}t||jdd|j|jdd|dVqdS) zGenerate lines for a LAB file for this annotation Returns ------- iterator: Iterator[str] An iterator over LAB text lines TrCrtzOSpace-separated LAB file format does not allow labels containing spaces (got: "rurv N)rIrwr rxryrz)r2rMrVr;r{r)r)r* _iter_labs(zAnnotation._iter_labcCr})zSerialize annotation as a string using LAB format Returns ------- serialized: str LAB string r~cSrr)r)rr)r)r*rrz%Annotation.to_lab..)rrr5r)r)r*to_labrzAnnotation.to_labcCr)zDump annotation to file using LAB format Parameters ---------- file : file object Usage ----- >>> with open('file.lab', 'w') as file: ... annotation.write_lab(file) N)rrrr)r)r* write_labrzAnnotation.write_lab intersectionsupportmodec Cst|trt|g|jd}|j||dSt|tr|}|j|j|jd}|dkrmi}tg}|j dd |D]\}}t |j |}|||<| |q9t||_ dd|D|_d d|D|_d |_d |_|S|d kri}tg}|j dd |D]\}} || vrqt |j |}|||<| |qt||_ d d|D|_dd|D|_d |_d |_|S|dkr|j dd |D]!\}} || @} |j |D]\} } |j| | d} | || | f<qq|Std|d S)aCrop annotation to new support Parameters ---------- support : Segment or Timeline If `support` is a `Timeline`, its support is used. mode : {'strict', 'loose', 'intersection'}, optional Controls how segments that are not fully included in `support` are handled. 'strict' mode only keeps fully included segments. 'loose' mode keeps any intersecting segment. 'intersection' mode keeps any intersecting segment but replace them by their actual intersection. Returns ------- cropped : Annotation Cropped annotation Note ---- In 'intersection' mode, the best is done to keep the track names unchanged. However, in some cases where two original segments are cropped into the same resulting segments, conflicting track names are modified to make sure no track is lost. rErr"r#looseFr6cSi|]}|dqSTr)r@r)r)r*rA rBz#Annotation.crop..cSrr4r)r@r)r)r*rA rBTNstrictcSrrr)r@r)r)r*rA#rBcSrr4r)r@r)r)r*rA$rBr) candidatezunsupported mode: '%s')rwrrr"cropr __class__r#rGr:co_iterdictr-r>valuesrr/r.r1r0rH new_trackNotImplementedError) r2rrcroppedr-r.rMrVr` other_segmentrrNr;r)r)r*rs`         zAnnotation.cropremovedcCs`t|tr t|g}t|g|jd}|j|d}|dkr#d}n|dkr)d}|j||dS)aRemove segments that overlap `removed` support. A simple illustration: .. code-block:: text A |------| |------| # B |----------| # self C |--------------| |------| # # |-------| |-----------| # removed # A # B |---| # mode="intersection" C |--| |------| # A # B # mode="loose" C |------| # A |------| # B |----------| # mode="strict" C |--------------| |------| # Parameters ---------- removed : Segment or Timeline If `support` is a `Timeline`, its support is used. mode : {'strict', 'loose', 'intersection'}, optional Controls how segments that are not fully included in `removed` are handled. 'strict' mode only removes fully included segments. 'loose' mode removes any intersecting segment. 'intersection' mode removes the overlapping part of any intersecting segment. Returns ------- extruded : Annotation Extruded annotation Note ---- In 'intersection' mode, the best is done to keep the track names unchanged. However, in some cases where two original segments are cropped into the same resulting segments, conflicting track names are modified to make sure no track is lost. r")rrrr)rwrrr:extentr"gapsr)r2rr extent_tltruncating_supportr)r)r*extrude;s 4  zAnnotation.extruder8rcCsl|r||}n|}t|jd}||D]\\}}\}}|||f|||fkr*q|||@q|S)aGet overlapping parts of the annotation. A simple illustration: .. code-block:: text annotation A |------| |------| |----| B |--| |-----| |----------| C |--------------| |------| annotation.get_overlap() |------| |-----| |--------| annotation.get_overlap(for_labels=["A", "B"]) |--| |--| |----| Parameters ---------- labels : optional list of labels Labels for which to consider the overlap Returns ------- overlap : `pyannote.core.Timeline` Timeline of the overlaps. r)subsetrr"raddr)r2r8 annotation overlaps_tls1rfs2rgr)r)r* get_overlap{s  zAnnotation.get_overlaprMcCst|j|iS)aQuery tracks by segment Parameters ---------- segment : Segment Query Returns ------- tracks : set Set of tracks Note ---- This will return an empty set if segment does not exist. )rGr-getkeys)r2rMr)r)r* get_tracksszAnnotation.get_tracksrNcCs||j|ivS)aCheck whether a given track exists Parameters ---------- segment : Segment Query segment track : Query track Returns ------- exists : bool True if track exists for segment )r-r)r2rMrNr)r)r* has_trackszAnnotation.has_trackcCs|j|j|jd}gtg}}|jD]\}}||||t |fqt ||_dd|D|_ dd|D|_ d|_ d|_|S)zGet a copy of the annotation Returns ------- annotation : Annotation Copy of the annotation rcSrr4r)r@r)r)r*rArBz#Annotation.copy..cSrrr)r@r)r)r*rArBNT)rr"r#rGr-rHr>rrJrrr.r/r0r1)r2copiedr-r.r^valuer)r)r*r7s  zAnnotation.copyrprefixcCsjt|j|i}|dur||vr|S|durd}d}d||f|vr/|d7}d||f|vs#d||fS)a1Generate a new track name for given segment Ensures that the returned track name does not already exist for the given segment. Parameters ---------- segment : Segment Segment for which a new track name is generated. candidate : any valid track name, optional When provided, try this candidate name first. prefix : str, optional Track name prefix. Defaults to the empty string ''. Returns ------- name : str New track name Nr~rz%s%dr)rGr-r)r2rMrrexisting_trackscountr)r)r*rs zAnnotation.new_trackcCsddd|jddDS)zHuman-friendly representationrcSs g|] \}}}d|||fqS)z%s %s %sr))r=stlr)r)r*rs z&Annotation.__str__..TrC)rrIr5r)r)r*__str__szAnnotation.__str__r^cCst|tr|j|}d|_|D] \}}d|j|<qdSt|trMt|dkrM|j|d}||d}d|j|<|sK|j|dd|_dSdSt d)zDelete one track >>> del annotation[segment, track] Delete all tracks of a segment >>> del annotation[segment] Trrz:Deletion only works with Segment or (Segment, track) keys.N) rwrr-rKr1rHr/tuplerRr)r2r^r`rNr;r)r)r* __delitem__s"    zAnnotation.__delitem__cCs(t|tr |df}|j|d|dS)zGet track label >>> label = annotation[segment, track] Note ---- ``annotation[segment]`` is equivalent to ``annotation[segment, '_']`` rVrr)rwrr-)r2r^r)r)r* __getitem__Rs zAnnotation.__getitem__r;cCs~t|tr |df}|\}}|sdS||jvri|j|<d|_||j|vr1|j||}d|j|<||j||<d|j|<dS)aAdd new or update existing track >>> annotation[segment, track] = label If (segment, track) does not exist, it is added. If (segment, track) already exists, it is updated. Note ---- ``annotation[segment] = label`` is equivalent to ``annotation[segment, '_'] = label`` Note ---- If `segment` is empty, it does nothing. rVNT)rwrr-r1r/)r2r^r;rMrN old_labelr)r)r* __setitem__cs    zAnnotation.__setitem__cCs|j|j|jdS)zReturn an empty copy Returns ------- empty : Annotation Empty annotation using the same 'uri' and 'modality' attributes. r)rr"r#r5r)r)r*emptys zAnnotation.emptycCs.tdd|jDr|t|jtdS)z{Get sorted list of labels Returns ------- labels : list Sorted list of labels cSrr)r))r=lnur)r)r*rrz%Annotation.labels..r])rpr/rrOr_r.rZr5r)r)r*r8szAnnotation.labelsuniquecCs&|j|i}|rt|St|S)aZQuery labels by segment Parameters ---------- segment : Segment Query unique : bool, optional When False, return the list of (possibly repeated) labels. Defaults to returning the set of labels. Returns ------- labels : set or list Set (resp. list) of labels for `segment` if it exists, empty set (resp. list) otherwise if unique (resp. if not unique). Examples -------- >>> annotation = Annotation() >>> segment = Segment(0, 2) >>> annotation[segment, 'speaker1'] = 'Bernard' >>> annotation[segment, 'speaker2'] = 'John' >>> print sorted(annotation.get_labels(segment)) set(['Bernard', 'John']) >>> print annotation.get_labels(Segment(1, 2)) set([]) )r-rrrGlist)r2rMrr8r)r)r* get_labelss zAnnotation.get_labelsinvertc st|rt|nt|@|j|j|jd}itg}}|jD]\}}fdd|D}|rH|||<||q,t ||_dd|D|_ dd|D|_ d|_ d|_ |S)aHFilter annotation by labels Parameters ---------- labels : iterable List of filtered labels invert : bool, optional If invert is True, extract all but requested labels Returns ------- filtered : Annotation Filtered annotation rcsi|] \}}|vr||qSr)r)r=rNr;r8r)r*rAsz%Annotation.subset..cSrrr)r@r)r)r*rArBcSrr4r)r@r)r)r*rArBTN)rGr8rr"r#r-rHr>rrr/r.r1r0) r2r8rsubr-r.rMr` sub_tracksr)rr*rs(  zAnnotation.subsetrcCs8|r|n|}|jddD] \}}}||||f<q|S)aAdd every track of an existing annotation (in place) Parameters ---------- annotation : Annotation Annotation whose tracks are being added copy : bool, optional Return a copy of the annotation. Defaults to updating the annotation in-place. Returns ------- self : Annotation Updated annotation Note ---- Existing tracks are updated with the new label. TrC)r7rI)r2rr7resultrMrNr;r)r)r*r>szAnnotation.updatecCsF||vr t|jdS|j|r||r|j|S|j|S)aQuery segments by label Parameters ---------- label : object Query copy : bool, optional Defaults (True) to returning a copy of the internal timeline. Set to False to return the actual internal timeline (faster). Returns ------- timeline : Timeline Timeline made of all segments for which at least one track is annotated as label Note ---- If label does not exist, this will return an empty timeline. Note ---- In case copy is set to False, be careful **not** to modify the returned timeline, as it may lead to weird subsequent behavior of the annotation instance. r)r8rr"r/rOr.r7)r2r;r7r)r)r*r9s    zAnnotation.label_timelinecC|j|ddS)aLabel support Equivalent to ``Annotation.label_timeline(label).support()`` Parameters ---------- label : object Query Returns ------- support : Timeline Label support See also -------- :func:`~pyannote.core.Annotation.label_timeline` :func:`~pyannote.core.Timeline.support` Fr6)r9rr2r;r)r)r* label_support;szAnnotation.label_supportcCr)aLabel duration Equivalent to ``Annotation.label_timeline(label).duration()`` Parameters ---------- label : object Query Returns ------- duration : float Duration, in seconds. See also -------- :func:`~pyannote.core.Annotation.label_timeline` :func:`~pyannote.core.Timeline.duration` Fr6)r9rzrr)r)r*label_durationRszAnnotation.label_durationpercentcsRtfddDdddd}|r'tdd|Dfd d|D}|S) aGet labels chart (from longest to shortest duration) Parameters ---------- percent : bool, optional Return list of (label, percentage) tuples. Defaults to returning list of (label, duration) tuples. Returns ------- chart : list List of (label, duration), sorted by duration in decreasing order. c3|] }||fVqdSr4r)r=Lr5r)r*r?zz#Annotation.chart..cS|dSNrr)xr)r)r*r\{z"Annotation.chart..T)r^reversecSsg|]\}}|qSr)r))r=rVrzr)r)r*rsz$Annotation.chart..csg|] \}}||fqSr)r))r=r;rz)totalr)r*rs)r_r8npsum)r2rchartr))r2rr*rjszAnnotation.chartcsH||dur j|ddsdStfddDdddd S) aMGet label with longest duration Parameters ---------- support : Segment or Timeline, optional Find label with longest duration within provided support. Defaults to whole extent. Returns ------- label : any existing label or None Label with longest intersection Examples -------- >>> annotation = Annotation(modality='speaker') >>> annotation[Segment(0, 10), 'speaker1'] = 'Alice' >>> annotation[Segment(8, 20), 'speaker1'] = 'Bob' >>> print "%s is such a talker!" % annotation.argmax() Bob is such a talker! >>> segment = Segment(22, 23) >>> if not annotation.argmax(support): ... print "No label intersecting %s" % segment No label intersection [22 --> 23] Nrrc3rr4r)r=rVrr)r*r?rz$Annotation.argmax..cSrrr)rr)r)r*r\rz#Annotation.argmax..r]r)rmaxr8)r2rr)rr*argmaxszAnnotation.argmaxstring generatorcCs\|j|j|jd}|dkrt}n|dkrt}|jddD] \}}}|||t|f<q|S)aRename all tracks Parameters ---------- generator : 'string', 'int', or iterable, optional If 'string' (default) rename tracks to 'A', 'B', 'C', etc. If 'int', rename tracks to 0, 1, 2, etc. If iterable, use it to generate track names. Returns ------- renamed : Annotation Copy of the original annotation where tracks are renamed. Example ------- >>> annotation = Annotation() >>> annotation[Segment(0, 1), 'a'] = 'a' >>> annotation[Segment(0, 1), 'b'] = 'b' >>> annotation[Segment(1, 2), 'a'] = 'a' >>> annotation[Segment(1, 3), 'c'] = 'c' >>> print(annotation) [ 00:00:00.000 --> 00:00:01.000] a a [ 00:00:00.000 --> 00:00:01.000] b b [ 00:00:01.000 --> 00:00:02.000] a a [ 00:00:01.000 --> 00:00:03.000] c c >>> print(annotation.rename_tracks(generator='int')) [ 00:00:00.000 --> 00:00:01.000] 0 a [ 00:00:00.000 --> 00:00:01.000] 1 b [ 00:00:01.000 --> 00:00:02.000] 2 a [ 00:00:01.000 --> 00:00:03.000] 3 c rrintTrC)rr"r#rrrInext)r2rrenamedrrVr;r)r)r* rename_trackss"zAnnotation.rename_tracksmappingc sdurdkr tndkrtfdd|D|r$|n|}D]\}}d|j|<d|j|<q*|jD]\}}fdd|D} | |j|<q>|S)a/Rename labels Parameters ---------- mapping : dict, optional {old_name: new_name} mapping dictionary. generator : 'string', 'int' or iterable, optional If 'string' (default) rename label to 'A', 'B', 'C', ... If 'int', rename to 0, 1, 2, etc. If iterable, use it to generate labels. copy : bool, optional Set to True to return a copy of the annotation. Set to False to update the annotation in-place. Defaults to True. Returns ------- renamed : Annotation Annotation where labels have been renamed Note ---- Unmapped labels are kept unchanged. Note ---- Parameter `generator` has no effect when `mapping` is provided. Nrrcsi|]}|tqSr))rr@)rr)r*rAz,Annotation.rename_labels..Tcsi|] \}}|||qSr))rr)rr)r*rA s)rrr8r7rHr/r-) r2rrr7rr new_labelrMr` new_tracksr))rrr* rename_labelss "    zAnnotation.rename_labelscCsR|dkrt}n|dkrt}|}|jddD] \}}}t||||f<q|S)aRelabel tracks Create a new annotation where each track has a unique label. Parameters ---------- generator : 'string', 'int' or iterable, optional If 'string' (default) relabel tracks to 'A', 'B', 'C', ... If 'int' relabel to 0, 1, 2, ... If iterable, use it to generate labels. Returns ------- renamed : Annotation New annotation with relabeled tracks. rrTrC)rrrrIr)r2r relabeledrrrVr)r)r*relabel_tracksszAnnotation.relabel_trackscollarcCsVt}|}|D]}|j|dd}||}|D] }|||t|f<qq |S)aAnnotation support The support of an annotation is an annotation where contiguous tracks with same label are merged into one unique covering track. A picture is worth a thousand words:: collar |---| annotation |--A--| |--A--| |-B-| |-B-| |--C--| |----B-----| annotation.support(collar) |------A------| |------B------| |-B-| |--C--| Parameters ---------- collar : float, optional Merge tracks with same label and separated by less than `collar` seconds. This is why 'A' tracks are merged in above figure. Defaults to 0. Returns ------- support : Annotation Annotation support Note ---- Track names are lost in the process. Tr6)rrr8r9rr)r2rrrr;r<rMr)r)r*r/s$   zAnnotation.supportc cs~|jdd}|jdd}||D]*\}}t||td}t||td}t||D] \}} ||f|| ffVq.qdS)aIterate over pairs of intersecting tracks Parameters ---------- other : Annotation Second annotation Returns ------- iterable : (Segment, object), (Segment, object) iterable Yields pairs of intersecting tracks, in chronological (then alphabetical) order. See also -------- :func:`~pyannote.core.Timeline.co_iter` Fr6r]N)r:rr_rrZriproduct) r2rcr<other_timelinerSr` other_tracksrTr)r)r*rfs  zAnnotation.co_itercCst|ts td|}|}ddt|D}ddt|D}tt|t|f}||D]'\\}}\} } ||||f} ||| | f} || @j } || | f| 7<q3|S)aCooccurrence (or confusion) matrix >>> matrix = annotation * other Parameters ---------- other : Annotation Second annotation Returns ------- cooccurrence : (n_self, n_other) np.ndarray Cooccurrence matrix where `n_self` (resp. `n_other`) is the number of labels in `self` (resp. `other`). zCcomputing cooccurrence matrix only works with Annotation instances.cSi|]\}}||qSr)r))r=ir;r)r)r*rArz&Annotation.__mul__..cSrr)r))r=jr;r)r)r*rAr) rwr TypeErrorr8 enumeraterzerosrRrrz)r2rci_labelsj_labelsIJmatrixrMrNr other_trackrrrzr)r)r*__mul__s  zAnnotation.__mul__{Gz? resolutionrzcCs.|dur |}|\}}|j|dd}|dur|}t|tr,t||j|jd}nt|||d}||}|durF||} | |} n t t ||j} t j | t |ft jd} t|D](\} } || }|j|dddD]\}}| td |t|| | fd 7<qpq_t j| d | d } t| ||d S) aDiscretize Parameters ---------- support : Segment, optional Part of annotation to discretize. Defaults to annotation full extent. resolution : float or SlidingWindow, optional Defaults to 10ms frames. labels : list of labels, optional Defaults to self.labels() duration : float, optional Overrides support duration and ensures that the number of returned frames is fixed (which might otherwise not be the case because of rounding errors). Returns ------- discretized : SlidingWindowFeature (num_frames, num_labels)-shaped binary features. Nrr)rysteprz)dtypecenterT)r return_rangesrr)outr)r:rrr8rwrr rz closest_framerroundrrrRuint8rr9rminminimumr)r2rrr8rz start_timeend_timer start_frame end_frame num_framesdatakr;rFrystopr)r)r* discretizes8       &zAnnotation.discretizerecordsc Cs||||d}tt}t}|D]\}}} | |||<|| qt||_dd|D|_dd|jD|_d|_d|_ |S)aAnnotation Parameters ---------- records : iterator of tuples (segment, track, label) tuples uri : string, optional name of annotated resource (e.g. audio or video file) modality : string, optional name of annotated modality Returns ------- annotation : Annotation New annotation rcSrr4r)r@r)r)r*rA rBz+Annotation.from_records..cSrrr)r@r)r)r*rA rBNT) rrrGrrr-r.r/r0r1) r(rr"r#rr`r8rMrNr;r)r)r*r&s    zAnnotation.from_recordscCsBddlm}m}|st|j|jjddSddlm}||S)zjIPython notebook support See also -------- :mod:`pyannote.core.notebook` r)MATPLOTLIB_IS_AVAILABLEMATPLOTLIB_WARNING)klassN)repr_annotation) notebookrrwarningswarnformatr__name__r )r2rrr r)r)r* _repr_png_s  zAnnotation._repr_png_)NN)Fr)rcr )rr4)r$r )r)NrT)r)NrNN)Sr% __module__ __qualname____doc__ classmethodrrZr+r3propertyr"setterrOrSrUrTr r rrrWrYboolrrrIrbrr:rnrqrsr r|rr rrrrrrrrrrr rrr7rrrrrrrrr8rrr>r9rfloatrrrrrrrrrrrndarrayrrrrr&r&r)r)r)r*r s              f @)  25 *   ',' (0 87 )   A %r ).r)rir" collectionsrtypingrrrrrrr r r r r rnumpyrsortedcontainersrr~rrrrMrrr<rfeaturerutils.generatorsrr utils.typesrrrrrrpandaspdr r)r)r)r*s O 8