o pi+@sdZddlZddlZddlmZmZmZmZmZm Z ddl Z ddl m Z ddlmZddlmZddlmZGd d d e jjjZed krOddlZedSdS) zi ######## Features ######## See :class:`pyannote.core.SlidingWindowFeature` for the complete reference. N)TupleOptionalUnionIteratorListText) Alignment)Segment) SlidingWindow)Timelinec @sHeZdZdZ d3dejdedeefddZ dd Z e d d Z e d d Z ddZddZddZdedejfddZddZdeeejffddZddZ d4deedeeeejefejffd d!Z "  #d5d$eeefd%ed&eed'edeejdff d(d)Z d*d+Z!eje"j#fZ$dejfd,d-Z%d.d/Z&d6d1d2Z'dS)7SlidingWindowFeaturezPeriodic feature vectors Parameters ---------- data : (n_frames, n_features) numpy array sliding_window : SlidingWindow labels : list, optional Textual description of each dimension. Ndatasliding_windowlabelscCs||_||_||_d|_dSN)rrr_SlidingWindowFeature__i)selfrrrrI/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/core/feature.py__init__<s zSlidingWindowFeature.__init__cC |jjdS)zNumber of feature vectorsrrshaperrrr__len__Ds zSlidingWindowFeature.__len__cCs|jdt|S)Nr)rrange_to_segmentlenrrrrextentHszSlidingWindowFeature.extentcCr)zDimension of feature vectorsr rrrrr dimensionLs zSlidingWindowFeature.dimensioncCstdt|jjdS)Nz(This is deprecated in favor of `__len__`r)warningswarnDeprecationWarningrrrrrr getNumberQs  zSlidingWindowFeature.getNumbercCtdt|jS)Nz3This is deprecated in favor of `dimension` property)r!r"r#r rrrr getDimensionUz!SlidingWindowFeature.getDimensioncCr%)Nz0This is deprecated in favor of `extent` property)r!r"r#rrrrr getExtent[r'zSlidingWindowFeature.getExtentireturncCs |j|S)zGet ith feature vectorr)rr)rrr __getitem__as z SlidingWindowFeature.__getitem__cCs d|_|Sr)rrrrr__iter__eszSlidingWindowFeature.__iter__c CsH|jd7_z |j|j|j|jfWSty#}ztd}~ww)Nr )rrr IndexError StopIteration)rerrr__next__iszSlidingWindowFeature.__next__cCs|SN)r1rrrrnextpszSlidingWindowFeature.nextFwindowccsH|jjd}t|D]}|r|j||j|fVq |j|Vq dS)zFeature vector iterator Parameters ---------- window : bool, optional When True, yield both feature vector and corresponding window. Default is to only yield feature vector rN)rrranger)rr4 n_samplesr)rrr iterfeaturesss z!SlidingWindowFeature.iterfeatureslooseTfocusmodefixed return_datac s|s t|ts d}t||s|durd}t|jj|||dd}jjd}tjjd}gdd} } } |D]2\} } | t| dt| d7} | t | |t | |7} | dks_| |kr`q;| t | dt| |gg7} q;| r}t fdd | D}nd jjdd}t |}|durt t jd| fd ||t j|d| fd |g}|r|Stj| ddjjjjjd }t||jd S)a@Extract frames Parameters ---------- focus : Segment or Timeline mode : {'loose', 'strict', 'center'}, optional In 'strict' mode, only frames fully included in 'focus' support are returned. In 'loose' mode, any intersecting frames are returned. In 'center' mode, first and last frames are chosen to be the ones whose centers are the closest to 'focus' start and end times. Defaults to 'loose'. fixed : float, optional Overrides `Segment` 'focus' duration and ensures that the number of returned frames is fixed (which might otherwise not be the case because of rounding errors). return_data : bool, optional Return a numpy array (default). For `Segment` 'focus', setting it to False will return a `SlidingWindowFeature` instance. Returns ------- data : `numpy.ndarray` or `SlidingWindowFeature` Frame features. See also -------- SlidingWindow.crop zG"focus" must be a "Segment" instance when "return_data"is set to False.Nz9"fixed" cannot be set when "return_data" is set to False.T)r:r; return_rangesrr cs&g|]\}}j||ddfqSr2r+).0startendrrr s&z-SlidingWindowFeature.crop..)r)r )r?durationstepr) isinstancer ValueErrorrcroprrrminmaxnpvstackemptytiler r?rBrCr r)rr9r:r;r<msgrangesr6 n_dimensionsclipped_ranges repeat_first repeat_lastr?r@rrrrrrrGsP%     zSlidingWindowFeature.cropcCsBddlm}m}|st|j|jjddSddlm}||S)Nr )MATPLOTLIB_IS_AVAILABLEMATPLOTLIB_WARNING)klass) repr_feature) notebookrTrUr!r"format __class____name__rW)rrTrUrWrrr _repr_png_s  zSlidingWindowFeature._repr_png_cCs|jSr2r+rrrr __array__szSlidingWindowFeature.__array__cs|dd}||D]}t|jtfstSq tdd|D}|r0tdd|D|d<t|||i|}t|turKtfdd|DS|dkrQdSt|jj dS) Noutrcs$|] }t|tr |jn|VqdSr2rEr rr>xrrr  z7SlidingWindowFeature.__array_ufunc__..csr_r2r`rarrrrcrdc3s&|]}t|jjdVqdS)rDN)typerrrarrrrcs atrD) getrE_HANDLED_TYPESr NotImplementedtuplegetattrrerr)rufuncmethodinputskwargsr^rbrrrr__array_ufunc__s(     z$SlidingWindowFeature.__array_ufunc__toc s|jj}|jj}|jj}t|}|d|t|||jj}|jj}|jj}t|} |d|t| |tfdd|jj D} t | |j|j dS)a Align features by linear temporal interpolation Parameters ---------- to : SlidingWindowFeature Features to align with. Returns ------- aligned : SlidingWindowFeature Aligned features g?cs(g|]}t|ddtjfqSr2)rJinterpnewaxis)r>old_datanew_told_trrrA>sz.SlidingWindowFeature.align..rD) rr?rCrBrrJarangehstackrTr r) rrq old_startold_step old_duration old_samples new_startnew_step new_duration new_samplesnew_datarruralign#s  zSlidingWindowFeature.alignr2)F)r8NT)rqr r*r )(r[ __module__ __qualname____doc__rJndarrayr rrrrpropertyrr r$r&r(intr,r-rr r1r3rboolrrr7r rfloatrGr\numbersNumberrhr]rprrrrrr 1s^        m  r __main__)rrr!typingrrrrrrnumpyrJpyannote.core.utils.typesrsegmentr r timeliner libmixinsNDArrayOperatorsMixinr r[doctesttestmodrrrrs