o xim@sdZddlmZmZmZmZmZddlmZddl m Z m Z m Z ddl mZddl mZddl mZddl mZdd lmZd d Zdd ee fddZGdddejZd S)z` An IntervalTier is a tier containing an array of intervals -- data that spans a period of time )CallableListOptionalTupleSequence)Literal)Interval INTERVAL_TIER CropCollision)errors)utils)my_math) constants) textgrid_tiercCsdd|D}||S)z Enforces consistency in intervals - converts all entries to intervals - removes whitespace in labels - sorts values by time cSs*g|]\}}}tt|t||qS)rfloatstrip.0startendlabelrre/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/praatio/data_classes/interval_tier.py sz&_homogenizeEntries..)sort)entriesprocessedEntriesrrr_homogenizeEntriess rNrcCszdd|D}dd|D}|dur|t||dur$|t|z t|}t|}W||fSty<tw)NcSg|]}|jqSrrrintervalrrrr)z+_calculateMinAndMaxTime..cSrrrr rrrr*r")appendrminmax ValueErrorr TimelessTextgridTierException)rminTmaxT minTimeList maxTimeList resolvedMinT resolvedMaxTrrr_calculateMinAndMaxTime(s  r/c seZdZeZeZ  dIdedeede e de e ffdd Z dd Z e d ee fd d Zd e de dedded df ddZ dJdejde d ejfddZded dfddZdKddZ dLd e d!ed"d dfd#d$Z % &dMd'e d(e d)ed*d+ed df d,d-Zd.ed eeeeffd/d0Zd eefd1d2Z % dNded)ed3d4ed5d dfd6d7Zd'e d8e d)ed9d dfd:d;ZdOdKd=d>Z ?dPddd@ed dfdAdBZ dQdCddDe e egefd dfdEdFZ! dLd!ed"d efdGdHZ"Z#S)R IntervalTierNnamerr)r*cs:t|}t|||\}}tt||||||dS)aAn interval tier is for annotating events that have duration The entries is of the form: [(startTime1, endTime1, label1), (startTime2, endTime2, label2), ] The data stored in the labels can be anything but will be interpreted as text by praatio (the label could be descriptive text e.g. ('erase this region') or numerical data e.g. (average pitch values like '132')) N)rr/superr0__init__ _validate)selfr1rr)r*calculatedMinTcalculatedMaxT __class__rrr3>s   zIntervalTier.__init__cCs|jD]}|j|jkrtd|jd|jdqt|jdd|jddD](\}}|j|jkrQtd|jd|jd|jd |jd|jd|jd q)dS) zVAn interval tier is invalid if the entries are out of order or overlap with each otherzThe start time of an interval (z#) cannot occur after its end time ()rNz1Two intervals in the same tier overlap in time: (, z) and ()rrrr TextgridStateErrorzipr)r5entry nextEntryrrrr4Ws6   & zIntervalTier._validatereturncCs(dd|jD}tt|}||S)z'All unique timestamps used in this tiercSs$g|]\}}}||fD]}|q qSrr)rrstop_timerrrrksz+IntervalTier.timestamps..)rlistsetr)r5 tmpTimestampsuniqueTimestampsrrr timestampshs  zIntervalTier.timestamps cropStartcropEndmode)strictlax truncated rebaseToZeroc std|t||krtd|d|dt|||j|}|durC|dd}||kr1|n|fdd|D}d }||}n|}|}t|j|||} | S) aCreates a new tier with all entries that fit inside the new interval Args: cropStart: cropEnd: mode: determines cropping behavior - 'strict', only intervals wholly contained by the crop interval will be kept - 'lax', partially contained intervals will be kept - 'truncated', partially contained intervals will be truncated to fit within the crop region. rebaseToZero: if True, the cropped textgrid values will be subtracted by the cropStart Returns: the modified version of the current tier rLzCrop error: start time (z) must occur before end time (r:Trcs&g|]\}}}t|||qSr)rrtimeDiffrrrsz%IntervalTier.crop..g) r validateOptionr r ArgumentErrorgetIntervalsInIntervalrr0r1) r5rJrKrLrP newEntryListnewSmallestValuer)r* croppedTierrrQrcropys,    zIntervalTier.cropMbP? referenceTier maxDifferencecs|j}g}|jD]9\}t|fddd}t|fddd}tt||r-|tt||r9|||fq|j|dS)aC Set timestamps in this tier to be the same as values in the reference tier Timestamps will only be moved if they are less than maxDifference away from the reference time. This can be used to correct minor alignment errors between tiers, as made when annotating files manually, etc. Args: referenceTier: the IntervalTier or PointTier to use as a reference maxDifference: the maximum amount to allow timestamps to be moved by Returns: the modified version of the current tier c t|SNabsxrrr z'IntervalTier.dejitter..)keycr]r^r_ra)rBrrrcrdr)rIrr%r lessThanOrEqualr`r$new)r5r[r\referenceTimestamps newEntriesr startCompare stopComparer)rrBrdejitters zIntervalTier.dejitterr?cCs|j|j|dS)z!Removes an entry from the entriesN)_entriespopindex)r5r?rrr deleteEntryszIntervalTier.deleteEntrytiercCs2|}|jD]}|j|j|jtjjdd}q|S)aFTakes the set difference of this tier and the given one Any overlapping portions of entries with entries in this textgrid will be removed from the returned tier. Args: tier: the tier to subtract from this one Returns: the modified version of the current tier F) collisionModedoShrink)rhr eraseRegionrrrEraseCollisionTRUNCATE)r5rrretTierr?rrr differences zIntervalTier.differencewarningoffset reportingMode)silencerzerrorc Cstd|tjt|}g}|jD]1}||j}||j}t||j |t ||j ||dkr3q|dkr9d}| t |||jqtdd|D}tdd|D} ||j kr^|j }| |j krf|j } t|j||| S)aSModifies all timestamps by a constant amount Args: offset: the amount to shift all intervals reportingMode: Determines the behavior if an entries moves outside of minTimestamp or maxTimestamp after being edited Returns: the modified version of the current tier r|rcSrrrr rrrrr"z/IntervalTier.editTimestamps..cSrrr#r rrrrr")r rSrErrorReportingModegetErrorReporterrrrcheckIsUndershoot minTimestampcheckIsOvershoot maxTimestampr$rrr%r&r0r1) r5r{r| errorReporterrVr!newStartnewEndnewMinnewMaxrrreditTimestampss,      zIntervalTier.editTimestampsr~Trrrs)truncate categoricalr~rtcCs td|tj|||tjdj}|}t |dkrnW|tjj kr/t d|d|d|dddD]}| |q6|tjjkrt|dj|kr\t|dj||dj}|||dj|krtt||dj|dj}|||d ur||} g} |jD]"}|j|kr| |q|j|kr| t|j| |j| |jqtdt | d D]I} | | j|k} | | d j|k} | | j| | d jk}| r| r|rt| | j| | d j| | j}| | d | | | | |nq|j| }|j| |d }|S) aIMakes a region in a tier blank (removes all contained entries) Args: start: end: collisionMode: Determines the behavior when the region to erase overlaps with existing intervals. - 'truncate' partially contained entries will have the portion removed that overlaps with the target entry - 'categorical' all entries that overlap, even partially, with the target entry will be completely removed - None or any other value throws IntervalCollision doShrink: If True, moves leftward by (/end/ - /start/) amount, each item that occurs after /end/ Returns: The modified version of the current tier Raises: CollisionError rsFrzErase region (r<zV)overlapped with an interval. If this was expected, consider setting the collisionModeNTr;rr)r rSrrvrYr LAXrrhlenERRORr CollisionErrorrqrwrrr insertEntryrr$rangeroinsertr)r5rrrsrt matchListnewTierr!newEntrydiffrVi rightEdgeleftEdge sameLabel newIntervalrrrrru sb                zIntervalTier.eraseRegion dataTupleListcCs4g}|jD]}t||j|j}|||fq|S)aReturns data from dataTupleList contained in labeled intervals Each labeled interval will get its own list of data values. dataTupleList should be of the form: [(time1, value1a, value1b,...), (time2, value2a, value2b...), ...] )rr getValuesInIntervalrrr$)r5r returnListr!intervalDataListrrrgetValuesInIntervalss  z!IntervalTier.getValuesInIntervalscs|jfddttdD}dd|D}djdkr-|dtddjddj|jkrB|tdj|jddd|D}|S) zReturns the regions of the textgrid without labels This can include unlabeled segments and regions marked as silent. cs(g|]}t|j|djdqS)r;)rrr)rrrfrrrsz.IntervalTier.getNonEntries..r;cSsg|] }|j|jkr|qSr)rrr rrrrsrrrcSs"g|] }t|tr |nt|qSr) isinstancerr rrrrs) rrrrrrrrr$)r5invertedEntryListrrfr getNonEntriess zIntervalTier.getNonEntries)replacemerger~collisionReportingMode)r}rzc Cstd|tjtd|tjt|}t|tst|}n|}||j |j t j dj }t|dkr:|j |nq|tjjkrQ|D]}||qB|j |nZ|tjjkr|D]}||qY|||ttdd|Dtdd|Ddd d|D}|j |ntd |j d |j d |jd |jddd|Dd ||j dd|jkr|j dd|_|j dd|jkr|j dd|_t|dkr|tjd|d|d|jddSdS)aiInserts an interval into the tier Args: entry: the Interval to insert collisionMode: determines the behavior in the event that intervals exist in the insertion area. - 'replace' will remove existing items - 'merge' will fuse the inserting item with existing items - None or any other value will throw a CollisionError collisionReportingMode: Determines the behavior if the new entry overlaps with an existing one Returns: the modified version of the current tier rsrFrcSrrrr tmpIntervalrrrrr"z,IntervalTier.insertEntry..cSrrr#rrrrrr"-cSrrrrrrrrr"zAttempted to insert interval (r<z, 'z ') into tier z% of textgrid but overlapping entries cSsg|]}t|qSr)tupler rrrrsz already existrr;zCollision warning for (z) with items (z ) of tier ''N)r rSrIntervalCollisionrrrrrYrrr rrnrr$REPLACErqMERGErr%r&joinr rrr1rr) r5r?rsrcollisionReporterr!r matchEntryrrrrrs|           zIntervalTier.insertEntryduration)stretchsplit no_changer~c Cs:td|tjg}|jD]}|j|kr||q |j|kr0|t|j||j||j q |j|kr|j|kr|tjj krO|t|j|j||j q |tjj krs|t|j||j ||||||j||j fq |tjj kr||q t d|d|d||dq |j||j|d}|S)aInserts a blank region into the tier Args: start: duration: collisionMode: Determines the behavior that occurs if an interval stradles the starting point - 'stretch' stretches the interval by /duration/ amount - 'split' splits the interval into two--everything to the right of 'start' will be advanced by 'duration' seconds - 'no change' leaves the interval as is with no change - 'error' will stop execution and raise an error Returns: the modified version of the current tier rsz5Collision occured during insertSpace() for interval 'z,' and given white space insertion interval (r<r:r)r rSrWhitespaceCollisionrrr$rrrSTRETCHSPLIT NO_CHANGEr rTrhr)r5rrrsrVr!rrrr insertSpace sX         zIntervalTier.insertSpacercshg}|jD]|jjtjd}fdd|jD}||q|jd|j}|||}|S)aTakes the set intersection of this tier and the given one - The output will contain one interval for each overlapping pair e.g. [(1, 2, 'foo')] and [(1, 1.3, 'bang'), (1.7, 2, 'wizz')] -> [(1, 1.3, 'foo-bang'), (1.7, 2, 'foo-wizz')] - Only intervals that exist in both tiers will remain in the returned tier. e.g. [(1, 2, 'foo'), (3, 4, 'bar')] and [(1, 2, 'bang'), (2, 3, 'wizz')] -> [(1, 2, 'foo-bang')] - If intervals partially overlap, only the overlapping portion will be returned. e.g. [(1, 2, 'foo')] and [(0.5, 1.5, 'bang')] -> [(1, 1.5, 'foo-bang')] Compare with IntervalTier.mergeLabels Args: tier: the tier to intersect with demarcator: the character to separate the labels of the overlapping intervals Returns: IntervalTier: the modified version of the current tier Fcs*g|]}|j|j|jjfqSr)rrr)r subInterval demarcatorr!rrrqsz-IntervalTier.intersection..r) rrYrrr TRUNCATEDextendr1rh)r5rrr retEntryListsubTier subEntryListnewNamerxrrr intersectionTs    zIntervalTier.intersection,rc Csg}|jD]G}||j|jtjd}t|jdkrq|dd|jD}|j d|d}t |j|jdj}t |j|jdj} || |f} | | q|j d|j } || |} | S) a@Merges labels of overlapping tiers into this tier - All intervals in this tier will appear in the output; for the given tier, only intervals that overlap with content in this tier will appear in the output e.g. [(1, 2, 'foo'), (3, 4, 'bar')] and [(1, 2, 'bang'), (2, 3, 'wizz')] -> [(1, 2, 'foo(bang)'), (3, 4, 'bar()')] - If multiple entries exist in a subinterval, their labels will be concatenated e.g. [(1, 2, 'hi')] and [(1, 1.5, 'h'), (1.5, 2, 'ai')] -> [(1, 2, 'hi(h,ai)')] compare with IntervalTier.intersection Args: tier: the tier to intersect with demarcator: the string to separate items that fall in the same subinterval Returns: IntervalTier: the modified version of the current tier FrcSrrr)rr?rrrrr"z,IntervalTier.mergeLabels..(r:rr)rrYrrr rrrnrrr%r&r$r1rh) r5rrrrr!rsubLabelrrrintersectedIntervalrrxrrr mergeLabelss&   zIntervalTier.mergeLabels targetTier filterFunccCsd}g}|j|jg}t|dD]7\}}|j|}|j|j} |dus(||jr9|j|j} || | 7}|| } n|| } |t|| |jq|j} |dj|jdj} |j | }t |j || |S)aMorphs the duration of segments in this tier to those in another This preserves the labels and the duration of silence in this tier while changing the duration of labeled segments. Args: targetTier: filterFunc: if specified, filters entries. The functor takes one argument, an Interval. It returns true if the Interval should be modified and false if not. Returns: The modified version of the current tier rTNr) rr safeZiprrrr$rrrr0r1)r5rrcumulativeAdjustAmountrV allIntervalssourceIntervaltargetIntervalrcurrIntervalDurationnewIntervalDurationrrcumulativeDifferencerrrrmorphs        zIntervalTier.morphcCstd|tjt|}d}d}|jD]C}|j|jkr(d}|tj d|d|r?|j|jkr?d}|tj d|d|d t |j|j |rJd}t |j|j |rUd}|}q|S) zValidate this tier Args: reportingMode (str): Determines the behavior if validation fails. Returns: True if the tier is valid; False if not r|TNFz>Invalid interval. End time occurs before or on the start time(z).z$Intervals are not sorted in time: [(z), (z)])r rSrrrrrrr r=rrrr)r5r|risValidpreviousIntervalr!rrrvalidates>      zIntervalTier.validateNN)rZ)rrr0rAr0)rz)r~T)r~rz)r)rr^)$__name__ __module__ __qualname__r tierTyper entryTypestrrrrr3r4propertyrIrboolrYr TextgridTierrmrqryrrurrrrrrrrrr __classcell__rrr8rr0:s ; $  3 e! U I/ 4 ,r0r)__doc__typingrrrrrtyping_extensionsrpraatio.utilities.constantsrr r praatio.utilitiesr r r rpraatio.data_classesrrr/rr0rrrrs