o xi8@sdZddlmZmZmZmZmZddlmZddl m Z m Z ddl m Z ddl mZddl mZddl mZdd lmZd d Zdd ee fddZGdddejZd S)ze A PointTier is a tier containing an array of points -- data that exists at a specific point in time )ListTupleOptionalAnySequence)Literal)Point POINT_TIER) constants)errors)utils)my_math) textgrid_tiercCsdd|D}||S)z Enforces consistency in points - converts all entries to points - removes whitespace in labels - sorts values by time cSs"g|] \}}tt||qS)rfloatstrip.0timelabelrrb/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/praatio/data_classes/point_tier.py s"z&_homogenizeEntries..)sort)entriesprocessedEntriesrrr_homogenizeEntriessrNrcCsldd|D}|dur|t||dur|t|z t|}t|}W||fSty5tw)NcSg|]\}}|qSrrrrrrr#z+_calculateMinAndMaxTime..)appendrminmax ValueErrorr TimelessTextgridTierException)rminTmaxTtimeListcalculatedMinTcalculatedMaxTrrr_calculateMinAndMaxTime"s  r(c seZdZeZeZ  ddejde ddfddZ d?de dedddfd d!Z "d@d#eee d$fd%edeeed$ffd&d'Z ( dAd)e d*e d+ed,d-eddf d.d/Z ( dBded+ed0d1ed2ddfd3d4Z (dCd)e d5e d6ed7ddfd8d9Z d?deddefd:d;ZZS)D PointTierNnamerr#r$cs2t|}t|||\}}tt|||||dS)a{A point tier is for annotating instaneous events The entries is of the form: [(timeVal1, label1), (timeVal2, 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. ('peak point here') or numerical data e.g. (pitch values like '132')) N)rr(superr)__init__)selfr*rr#r$r&r' __class__rrr,6szPointTier.__init__returncCs(dd|jD}tt|}||S)z'All unique timestamps used in this tiercSrrr)rr_rrrrOrz(PointTier.timestamps..)rlistsetr)r- tmpTimestampsuniqueTimestampsrrr timestampsLs zPointTier.timestampslaxT cropStartcropEndmode)strictr7 truncated rebaseToZeroc s|krtdd|dg}|jD]}|j}|kr'||kr'||q|dur<fdd|D}d}|} n}|} t|j||| S)aCreates a new tier containing all entries inside the new interval Args: cropStart: cropEnd: mode: Mode is ignored. This parameter is kept for compatibility with IntervalTier.crop() rebaseToZero: if True, all entries will have their timestamps subtracted by *cropStart*. Returns: the modified version of the current tier zCrop error: start time (z) must occur before end time ()Tcsg|] \}}t||qSr)r)rtimeVrr8rrrxsz"PointTier.crop..g)r ArgumentErrorrrrr)r*) r-r8r9r:r= newEntriesentry timestampr#r$rr@rcropVs&    zPointTier.croprCcCs|j|j|dS)z!Removes an entry from the entriesN)_entriespopindex)r-rCrrr deleteEntryszPointTier.deleteEntryMbP? referenceTier maxDifferencecs`|j}g}|jD]!\}t|fddd}tt||r"|||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 cs t|S)N)abs)xrrrs z$PointTier.dejitter..)key)r)r6rrr lessThanOrEqualrMrnew)r-rKrLreferenceTimestampsrBr timeComparerrOrdejitters zPointTier.dejitterwarningoffset reportingMode)silencerWerrorc Cstd|tjt|}g}|jD]%\}}||}t||j|t||j ||dkr/q| t ||qdd|D}t |} t |} | |jkrO|j} | |j krW|j } t|j|| | S)atModifies all timestamps by a constant amount Args: offset: reportingMode: one of "silence", "warning", or "error". This flag determines the behavior if an entries moves outside of minTimestamp or maxTimestamp after being edited Returns: the modified version of the current tier rYrcSsg|]}t|jqSr)rr)rpointrrrrsz,PointTier.editTimestamps..)r validateOptionr ErrorReportingModegetErrorReporterrcheckIsUndershoot minTimestampcheckIsOvershoot maxTimestamprrrr r)r*) r-rXrY errorReporterrBrDr newTimestampr%newMinnewMaxrrreditTimestampss(   zPointTier.editTimestampsF dataTupleList. fuzzyMatchingc CsHd}g}t|}|jD]\}}tj||||d}|\} }|| q |S)aGet the values that occur at points in the point tier The procedure assumes that all data is ordered in time. dataTupleList should be in the form [(t1, v1a, v1b, ..), (t2, v2a, v2b, ..), ..] It returns the data in the form of [(t1, v1a, v1b, ..), (t2, v2a, v2b), ..] The procedure makes one pass through dataTupleList and one pass through self.entries. If the data is not sequentially ordered, the incorrect response will be returned. Args: dataTupleList: fuzzyMatching: if True, if there is not a feature value at a point, the nearest feature value will be taken. Returns: A list of values that exist at the given timepoints r)rjstartI)sortedrr getValueAtTimer) r-rirj currentIndexretListsortedDataTupleListrDrretTupleretRowrrrgetValuesAtPointss zPointTier.getValuesAtPointsr[startend collisionMode)truncate categoricalr[doShrinkc Cs|}|||tjjd}|j}t|dkr&|dddD]}||q|dur^g} ||} |jD]}|j|kr@| |q3|j|krQ| t |j| |j q3|j | } |j| | d}|S)aMakes a region in a tier blank (removes all contained entries) Args: start: the start of the deletion interval end: the end of the deletion interval collisionMode: Ignored for the moment (added for compatibility with eraseRegion() for Interval Tiers) doShrink: if True, moves leftward by (/end/ - /start/) all points to the right of /end/ Returns: The modified version of the current tier FrNTrrc) rSrEr CropCollision TRUNCATEDrlenrIrrrrrc) r-rtrurvrynewTier croppedTier matchListr\rBdiffrgrrr eraseRegions$       zPointTier.eraseRegion)replacemerger[collisionReportingMode)rZrWc Cs|td|tjtd|tjt|}t|ts$t|d|d}n|}g}d}t|j D]\}}|j |j kr@| |nq/t |dkrN|j |nQ|tjjkrc||j ||j |n<|tjjkr|j |} t|j d| j|jg} ||j ||j | ntd|d|jd d d |Dd |t |dkr|tjd |d|d|jddSdS)aInserts an interval into the tier Args: entry: the entry to insert collisionMode: determines the behavior if intervals exist in the insertion area. - 'replace', existing items will be removed - 'merge', inserting item will be fused with existing items - 'error', will throw TextgridCollisionException collisionReportingMode: Returns: None rvrrN-zAttempted to insert interval z into tier z% of textgrid but overlapping entries cSsg|]}t|qSr)tuple)rintervalrrrrjrz)PointTier.insertEntry..z already existzCollision warning for (z) with items (z ) of tier '')r r]r IntervalCollisionr^r_ isinstancer enumeraterrrr~rFREPLACErIMERGEjoinrr CollisionErrorr*r) r-rCrvrcollisionReporternewPointrir\oldPoint mergedPointrrr insertEntry,sV          zPointTier.insertEntryduration_collisionMode)stretchsplit no_changer[cCs`g}|jD]}|j|kr||q|j|kr#|t|j||jq|j||j|d}|S)a{Inserts a region into the tier Args: start: the start time to insert a space at duration: the duration of the space to insert collisionMode: Ignored for the moment (added for compatibility with insertSpace() for Interval Tiers) Returns: PointTier: the modified version of the current tier r{)rrrrrrSrc)r-rtrrrBr\rrrr insertSpacevs     zPointTier.insertSpacecCstd|tjt|}d}d}|jD]1}|r-|j|jkr-d}|tjd|d|dt |j|j |r8d}t |j|j |rCd}|}q|S)aValidate this tier Returns whether the tier is valid or not. If reportingMode is "warning" or "error" this will also print on error or stop execution, respectively. Args: reportingMode: Determines the behavior if there is a size difference between the maxTimestamp in the tier and the current textgrid. Returns: True if this tier is valid; False otherwise rYTNFz!Points are not sorted in time: [(z), (z)]) r r]r r^r_rrr TextgridStateErrorr`rarbrc)r-rYrdisValid previousPointr\rrrvalidates.  zPointTier.validateNN)r7T)rJ)rW)F)r[T)r[rW)r[)__name__ __module__ __qualname__r tierTyper entryTypestrrrrr,propertyr6rboolrErIr TextgridTierrVrhrrrsrrrr __classcell__rrr.rr)2s  - " 0 / 1 N  r)r)__doc__typingrrrrrtyping_extensionsrpraatio.utilities.constantsrr praatio.utilitiesr r r r praatio.data_classesrrr(rr)rrrrs