o xiT@sdZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z m Z mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZGdddZdedefddZdS)z^ Functions for reading/writing/manipulating Textgrid classes This is the 'heart' of praatio. N)OptionalTupleSequence)Literal) OrderedDict)TextgridFormatsMIN_INTERVAL_LENGTH CropCollision) textgrid_tier) point_tier) interval_tier) constants)errors)my_math) textgrid_io)utilsc@seZdZdZdPdedefddZddZd d Zd d Ze d e e dffddZ e d e e jdffddZ  dQde jdeededd dfddZddded dfddZdeded ed!d"ed df d#d$Zd%ed&ed'ed dfd(d)Z dRd*ededd dfd+d,Zd-e d e jfd.d/Z 0dSd%ed1ed2ed3d dfd4d5Z 6dTd7eee d8ed dfd9d:ZdUd;d<Zddedfd=e d>ed?d@edeedeedAededd dfdBdCZdDe dEe d dfdFdGZ dHe d e jfdIdJZ! dRdHe dKe jdedd dfdLdMZ" dRdedd efdNdOZ#dS)VTextgridaA container that stores and operates over interval and point tiers Textgrids are used by the Praat software to group tiers. Each tier contains different annotation information for an audio recording. Attributes: tierNames(Tuple[str]): the list of tier names in the textgrid tiers(Tuple[TextgridTier]): the list of ordered tiers minTimestamp(float): the minimum allowable timestamp in the textgrid maxTimestamp(float): the maximum allowable timestamp in the textgrid N minTimestamp maxTimestampcCst|_||_||_dS)zConstructor for Textgrids Args: minTimestamp: the minimum allowable timestamp in the textgrid maxTimestamp: the maximum allowable timestamp in the textgrid N)r _tierDictrr)selfrrr`/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/praatio/data_classes/textgrid.py__init__+s zTextgrid.__init__cCs t|jSN)lenrrrrr__len__9s zTextgrid.__len__ccs|jD]}|VqdSr)tiers)rentryrrr__iter__<s zTextgrid.__iter__cCsvt|tsdSd}|t|j|jM}|t|j|jM}||j|jkM}|r9|jD]}|||||kM}q*|S)NFT) isinstancerriscloserr tierNamesgetTier)rotherisEqualtierNamerrr__eq__@s  zTextgrid.__eq__return.cCt|jSr)tuplerkeysrrrrr#OzTextgrid.tierNamescCr*r)r+rvaluesrrrrrSr-zTextgrid.tierswarningtier tierIndex reportingMode)silencer/errorc Cs6td|tjt|}|j|jvrtd|dur#||j |j<n$t |j}| ||jt }||j |j<|D] }| |||<q:||_ |j}|jdurb||jkrb|tjd|jd|d|jdusl||jkro||_|j} |jdur| |jkr|tjd|jd| d|jdus| |jkr| |_dSdS)a-Add a tier to this textgrid. Args: tier: The tier to add to the textgrid tierIndex: Insert the tier into the specified position; if None, the tier will appear after all existing tiers reportingMode: This flag determines the behavior if there is a size difference between the maxTimestamp in the tier and the current textgrid. Returns: None Raises: TierNameExistsError: The textgrid already contains a tier with the same name as the tier being added TextgridStateAutoModified: The minimum or maximum timestamp was changed when not permitted IndexError: TierIndex is too large for the size of the existing tier list r2zTier name already in tierNz,Minimum timestamp in Textgrid changed from (z) to ()z,Maximum timestamp in Textgrid changed from ()rvalidateOptionr ErrorReportingModegetErrorReporternamer#rTierNameExistsErrorrlistinsertrr$rTextgridStateAutoModifiedr) rr0r1r2 errorReporternewOrderedTierNameList newTierDicttmpNameminVmaxVrrraddTierWs@      zTextgrid.addTiertgonlyMatchingNamesc CsF|j}|j|j}t||}t|j}|jD] }||vr!||qg}|dur+|}n|D]}||jvr>||jvr>||q-|D]}||jvrR||} || qA|D]K}||jvr||} | j||d} | |j} ||jvr||} ||j } | | j 7} | j| ||d} | || qU| } | j||d} || qU|S)a1Append one textgrid to the end of this one Args: tg: the textgrid to add to this one onlyMatchingNames: if False, tiers that don't appear in both textgrids will also appear Returns: the modified version of the current textgrid F)rr)entriesrr) rrrr;r#appendr$rDneweditTimestampsrG replaceTier) rrErFminTimemaxTimeretTGcombinedTierNamesr'finalTierNamesr0 appendTier newEntriesrrrappendTextgridsR                   zTextgrid.appendTextgrid cropStartcropEndmode)strictlax truncated rebaseToZeroc Cstd|t||krtd|d|d|dur"d}||}n|}|}t||}|jD]$}||} | ||||} t j j } |t jj krKt j j } |j| | dq.|S)a/Creates a textgrid where all intervals fit within the crop region Args: cropStart: The start time of the crop interval cropEnd: The stop time of the crop interval mode: Determines the crop 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 timestamps will be subtracted by the cropStart; if False, timestamps will not be changed Returns: the modified version of the current textgrid rVzCrop error: start time () must occur before end time (r5Tgr2)rr6r r ArgumentErrorrr#r$cropr r7WARNINGLAXSILENCErD) rrTrUrVrZminTmaxTnewTGr'r0newTierr2rrrr^s(     z Textgrid.cropstartenddoShrinkc Cs||krtd|d|d||}|j}|dur||8}t|j|j}|jD]}|||tjj |}| |q)||_|S)aMakes a region in a tier blank (removes all contained entries) Intervals that span the region to erase will be truncated. Args: start: end: doShrink: if True, all entries appearing after the erased interval will be shifted to fill the void (ie the duration of the textgrid will be reduced by *start* - *end*) Returns: the modified version of the current textgrid Raises: ArgumentError zEraseRegion error: start time (r[r5T) rr]rrrr eraseRegionr EraseCollisionTRUNCATErD) rrfrgrhdiffrrdr0 shrunkTierrrrris    zTextgrid.eraseRegionoffsetcCsVtd|tjt|j|j}|jD]}t|j dkr!| ||}|j ||dq|S)aModifies all timestamps by a constant amount Args: offset: the amount to offset in seconds reportingMode: one of "silence", "warning", or "error". This flag determines the behavior if there is a size difference between the maxTimestamp in the tier and the current textgrid. Returns: Textgrid: the modified version of the current textgrid r2rr\) rr6r r7rrrrrrGrJrD)rrnr2rEr0rrrrJ9s  zTextgrid.editTimestampsr'cCs |j|S)z$Get the tier with the specified name)r)rr'rrrr$V zTextgrid.getTierr4duration collisionMode)stretchsplit no_changer4cCsZtd|tjt|j|j}|j|_|j||_|jD]}||||}| |q|S)a=Inserts a blank region into a textgrid Every item that occurs after *start* will be pushed back by *duration* seconds Args: start: duration: collisionMode: Determines behavior in the event that an interval stradles the starting point. One of ['stretch', 'split', 'no change', None] - '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 - None or any other value throws an AssertionError Returns: Textgrid: the modified version of the current textgrid rq) rr6r WhitespaceCollisionrrrr insertSpacerD)rrfrprqrdr0rerrrrvZs   zTextgrid.insertSpaceTr#preserveOtherTiersc Cs&|dur|j}g}g}|D]}||}t|tjr ||q t|tjr+||q d}t|dkrI|| d}|D] }| ||}q>d}t|dkrf|| d}|D] }| ||}q[t |j |j } |r|jD] }|j|vr~| |qr|dur| ||dur| || S)aHCombine tiers Args: tierList: A list of tier names to combine. If none, combine all tiers. preserveOtherTiers: If false, uncombined tiers are not included in the output. Returns: Textgrid: the modified version of the current textgrid Nr)r#r$r!r IntervalTierrHr PointTierrpopunionrrrrr9rD) rr#rwintervalTierNamespointTierNamesr'r0 intervalTier pointTierrErrr mergeTierss@            zTextgrid.mergeTierscCs t|S)zReturns a copy of this Textgrid)copydeepcopyrrrrrIroz Textgrid.newfnformat)short_textgrid long_textgridjson textgrid_jsonincludeBlankSpacesminimumIntervalLengthc Cstd|ttd|tj||t|}t||||||} t j |ddd} | | WddS1s;wYdS)a;Save the current textgrid to a file Args: fn: the fullpath filename of the output format: one of ['short_textgrid', 'long_textgrid', 'json', 'textgrid_json'] 'short_textgrid' and 'long_textgrid' are both used by praat 'json' and 'textgrid_json' are two json variants. 'json' cannot represent tiers with different min and max timestamps than the textgrid. includeBlankSpaces: if True, blank sections in interval tiers will be filled in with an empty interval (with a label of ""). If you are unsure, True is recommended as Praat needs blanks to render textgrids properly. minTimestamp: the minTimestamp of the saved Textgrid; if None, use whatever is defined in the Textgrid object. If minTimestamp is larger than timestamps in your textgrid, an exception will be thrown. maxTimestamp: the maxTimestamp of the saved Textgrid; if None, use whatever is defined in the Textgrid object. If maxTimestamp is smaller than timestamps in your textgrid, an exception will be thrown. minimumIntervalLength: any labeled intervals smaller than this will be removed, useful for removing ultrashort or fragmented intervals; if None, don't remove any. Removed intervals are merged (without their label) into adjacent entries. reportingMode: one of "silence", "warning", or "error". This flag determines the behavior if there is a size difference between the maxTimestamp in the tier and the current textgrid. Returns: a string representation of the textgrid rr2wzutf-8)encodingN) rr6rr r7validate_tgToDictionaryrgetTextgridAsStrioopenwrite) rrrrrrrr2tgAsDict textgridStrfdrrrsaves"+  "z Textgrid.saveoldNamenewNamecCs:||}|j|}||||||j|dSr)r$r#index removeTierrDrIrG)rrroldTierr1rrr renameTiers   zTextgrid.renameTierr9cCs |j|Sr)rrz)rr9rrrrs zTextgrid.removeTierrecCs(|j|}||||||dSr)r#rrrD)rr9rer2r1rrrrKs  zTextgrid.replaceTiercCstd|tjt|}d}t|jtt|jkr'd}|tj d|j|j D]7}|j |j krCd}|tj d|j d|j d|j |j krZd}|tj d|j d|j d|o`||}q*|S) aValidates this textgrid Returns whether the textgrid is valid or not. If reportingMode is "warning" or "error" this will also print on error or stop execution, respectively. Args: reportingMode: one of "silence", "warning", or "error". This flag determines the behavior if there is a size difference between the maxTimestamp in a tier and the current textgrid. Returns: True if this Textgrid is valid; False if not Raises: TierNameExistsError: Two tiers have the same name TextgridStateError: A timestamp fall outside of the allowable range r2TFzTier names not unique: z!Textgrid has a min timestamp of (z) but tier has (r5z!Textgrid has a max timestamp of ()rr6r r7r8rr#setrr:rrTextgridStateErrorrr)rr2r>isValidr0rrrrs<       zTextgrid.validate)NN)Nr/)r/)r4)NT)r)r)$__name__ __module__ __qualname____doc__floatrrr r(propertyrstrr#r TextgridTierrrintrrDboolrSr^rirJr$rvrrrIrrrrrKrrrrrrs  @A 8,  )  9  @  rrEr)cCsDg}|jD]}|j|j|j|j|jd}||q|j|j|dS)N)classr9xminxmaxrG)rrr)rtierTyper9rrrGrH)rErr0 tierAsDictrrrrIs  r)rrrtypingrrrtyping_extensionsr collectionsrpraatio.utilities.constantsrrr praatio.data_classesr r r praatio.utilitiesr rrrrrdictrrrrrs*          /