o xi*=@sVdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZmZddlmZddlmZejZeedrLed d Zned d  ZeZWdn1s_wYd eefd d Zdeeded dfddZdeeded efddZdeeded dfddZdedfddZ de!de!d e"fddZ#de!de!d e"fd d!Z$d"d#Z%   $did%ed&ed'e!d(e!d)e"d e"f d*d+Z&d,e!d-e!d.e ed/ed0d e ef d1d2Z'ded efd3d4Z(d5ed e!fd6d7Z) $ djd8e!d9e e e d:fd;e"dZ*d?e d,e!d-e!d e fd@dAZ+dBe!d efdCdDZ, dkdEe e e!e!fdFe!dGe!d e e e!e!ffdHdIZ-dJed dfdKdLZ.dMedNed e efdOdPZ/ dldQedRedSe e dTed df dUdVZ0dWe e1dXe"d e e fdYdZZ2d[ed e!fd\d]Z3d^e!d_ee!d`ee!d e!fdadbZ4dce!dde!dee!dfe"d e e!e!ff dgdhZ5dS)mz# Various generic utility functions N) resources)Literal)AnyIteratorListTupleNoReturnTypeOptional)errors) constantsfilespraatio praatScriptsreturncCsB||vrdS|rt||ddd|d}|S||}|S)z1Returns the first/last index of an item in a listN)lenindex)listvaluereverserrZ/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/praatio/utilities/utils.pyfinds  r _exception_textcCsdSNr)rrrrr reportNoop(sr exceptiontextcCs||rr)rr rrrreportException,sr!cCs t|dSr)print)rr rrr reportWarning0 r# reportingMode)silencewarningerrorcCs$tjjttjjttjjti}||Sr)r ErrorReportingModeSILENCErWARNINGr#ERRORr!)r% modeToFuncrrrgetErrorReporter4s r.time referenceTimecCs*||kr|tjd|d|ddSdS)N'z&' occurs before minimum allowed time 'TFr OutOfBoundsr/r0 errorReporterrrrcheckIsUndershoot>r6cCs*||kr|tjd|d|ddSdS)Nr1z%' occurs after maximum allowed time 'TFr2r4rrrcheckIsOvershootIr7r8cCs||jvr t|||jdSr) validOptionsr WrongOption) variableNamer optionClassrrrvalidateOptionTs r=Finterval cmprIntervalpercentThreshold timeThresholdboundaryInclusivecCs|dd\}}|dd\}}tdt||t||} | dk} d} |r-||kp,||k} d} |dkrK| rKt||t||} | t| }||k} | } d}|dkrY| rY| |k}|} | p`| p`| p`|} | S)aChecks whether two intervals overlap Args: interval: cmprInterval: percentThreshold: if percentThreshold is greater than 0, then if the intervals overlap, they must overlap by at least this threshold (0.2 would mean 20% overlap considering both intervals) (eg [0, 6] and [3,8] have an overlap of 50%. If percentThreshold is set to higher than 50%, the intervals will be considered to not overlap.) timeThreshold: if greater than 0, then if the intervals overlap, they must overlap by at least this threshold boundaryInclusive: if true, then two intervals are considered to overlap if they share a boundary Returns: bool: NrF)maxminfloat)r>r?r@rArB startTimeendTime cmprStartTime cmprEndTime overlapTime overlapFlagboundaryOverlapFlagpercentOverlapFlag totalTimepercentOverlaptimeOverlapFlagrrrintervalOverlapCheckYs(   rRstartend intervalsmode)strictlax truncatedcCs0td|tjg}|D]}d}|j|ks|j|krq |j|kr'|j|kr'|}ne|tjjkr:|j|ks7|j|kr:|}nR|j|krS|j|krS|tjjkrRt|j||j}n9|j|krl|j|krl|tjjkrkt||j|j}n |j|kr|j|kr|tjjkr|}n |tjjkrt|||j}|dur| |q |S)avGets all intervals that exist between /start/ and /end/ Args: start: the target interval start time end: the target interval stop time intervals: the list of intervals to check mode: Determines judgement criteria - 'strict', only intervals wholly contained by the target interval will be kept - 'lax', partially contained intervals will be kept - 'truncated', partially contained intervals will be truncated to fit within the crop region. Returns: The list of intervals that overlap with the target interval rVN) r=r CropCollisionrTrSLAX TRUNCATEDIntervallabelappend)rSrTrUrVcontainedIntervalsr> matchedEntryrrrgetIntervalsInIntervals8      rbcCs |ddS)N"z"")replace)r rrr escapeQuotesr$reinputStrcCsd|vrt|St|S)N.)rFint)rfrrrstrToIntOrFloatsri timestampsortedDataTupleList. fuzzyMatchingstartIc Cs|}d}|dur. z||}Wn tyYnw|d}||kr(||kr'|}n|d7}q nM||d}||} z||} Wn tyL|d8}Yn/w| d}| }t||} t||} | | krm|}|}| dkrlnn | | krv|d8}n|d7}q9|} | |fS)atGet the value in the data list (sorted by time) that occurs at this point If fuzzyMatching is True, if there is not a value at the requested timestamp, the nearest feature value will be taken. The procedure assumes that all data is ordered in time. dataTupleList should be in the form [(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. For efficiency purposes, it takes a starting index and returns the ending index. rFTrr) IndexErrorabs) rjrkrlrmibestRowcurrRowcurrTimebestTime dataTuplecurrDiffbestDiffretRowrrrgetValueAtTimesV       ry dataTupleListcCs4g}|D]}|d}||kr||kr||q|S)zGets the values that exist within an interval The function assumes that the data is formated as [(t1, v1a, v1b, ...), (t2, v2a, v2b, ...)] r)r_)rzrSrTintervalDataListrur/rrrgetValuesInInterval*s r|xcCs$d}|dkr d}|S|dkrd}|S)z9Returns 1 if x is positive, 0 if x is 0, and -1 otherwiserrrr)r}retValrrrsign:sr inputListminValuemaxValuecstddDrtdttdkr'|dur'|dur'||fg}|S|dur;dd|kr;dd|f|durPdd|krP||dffddtdtdD}d d|D}|S) zInverts the segments of a list of intervals e.g. [(0,1), (4,5), (7,10)] -> [(1,4), (5,7)] [(0.5, 1.2), (3.4, 5.0)] -> [(0.0, 0.5), (1.2, 3.4)] cSsg|] }|d|dkqSrrr.0r>rrr Msz&invertIntervalList..z*Interval start occured before interval endrNrrcs(g|]}|d|ddfqS)rrr)rrprrrrascSs g|] }|d|dkr|qSrrrrrrrgs )anyr ArgumentErrorsortedrinsertr_range)rrrinvListrrrinvertIntervalListDs   rpathcCstj|s t|dSdS)zpCreate a new directory Unlike os.mkdir, it does not throw an exception if the directory already exists N)osrexistsmkdir)rrrrmakeDirls rtxtsubStrcCsHg}d} z|||}Wn tyY|Sw|t||d7}q)z.z--run)r) rrrr FileNotFound subprocessPopenwaitPraatExecutionFailed)rrrrcmdList myProcessrrrrunPraatScripts     r listOfLists enforceLengthcs>|durt|dtfdd|Dstdtj|S)zA safe version of python's zip() If two sublists are of different sizes, python's zip will truncate the output to be the smaller of the two. safeZip throws an exception if the size of the any sublist is different from the rest. Trcsg|]}t|kqSr)r)rsubListlengthrrrszsafeZip..z"Lists to zip have different sizes.)rallr SafeZipException itertools zip_longest)rrrrrsafeZips    rwavFNcCs4t|d}|}|d}|d}t||}|S)z@For internal use. See praatio.audio.QueryWav() for general use.rrC)waveopen getparamsrF)r audiofileparams frameratenframesdurationrrrgetWavDurations  r targetTime candidateA candidateBcCs|dur |dur td|dur|dur|}|S|dur%|dur%|}|S|durC|durCt||}t||}||krA|}|S|}|S)aqChooses the closest time between two options that straddle the target time Args: targetTime: the time to compare against candidateA: the first candidate candidateB: the second candidate Returns: the closer of the two options to the target time Raises: ArgumentError: When no left or right candidate is provided NzMust provide at)r rro)rrr closestTimeaDiffbDiffrrrchooseClosestTimes      rrGrrDrcCsF|dur |}||8}n||}|dkrd}||fS||kr|}||fS)a}returns an interval before or after some start time The returned timestamps will be between 0 and max Args: startTime: the time to get the interval from duration: duration of the interval max: the maximum allowed time reverse: is the interval before or after the targetTime? Returns: the start and end time of an interval Trr)rGrrDrrHrrr getIntervals r)rrF)Fr)NNr)6__doc__rrrr importlibrtyping_extensionsrtypingrrrrrr r praatio.utilitiesr r r]hasattrr scriptsPathrrhr BaseExceptionstrrr!r#r.rFboolr6r8r=rRrbreriryr|rrrrrrrrrrrrrrs  $        < E H  (    #