o j-0 @s ddlZddlmZmZmZddlZddlZddlmZGdddZ e Z e deede defd d Z e deeefdefd d Ze d ededefddZe d edefddZe dededefddZe d@d edede defddZe d@d ededede def ddZe d@d ededede def d d!Ze dAd edede defd$d%Ze d@d edede defd&d'Ze d@d edede defd(d)Ze dBd edede defd+d,Ze d@d ed-edede def d.d/Ze dCd edede defd1d2Ze d@d edede defd3d4Ze d@d edede defd5d6Ze d@d edede defd7d8Z e d@d edede defd9d:Z!  d@d;ee ee effdd?Z"dS)DN)ListTupleUnion)Tensorc@s&eZdZddZdddZddZdS) WindowFunctionRegistercCs i|_dSN_functions_dictselfr _/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/audio/functional/window.py__init__ zWindowFunctionRegister.__init__Ncsfdd}|S)Ncs|j}|j|<|Sr)__name__r )funcnamer r r add_subfunctions z8WindowFunctionRegister.register..add_subfunctionr )r rrr r r registers zWindowFunctionRegister.registercCs |j|Srr)r rr r r get"rzWindowFunctionRegister.getr)r __module__ __qualname__rrrr r r r rs  rx data_typereturncCsTg}|D] }t|rt|ts|t|g|q|t||qt|Sr)npZisscalar isinstancestrappendpaddleZ to_tensorconcat)rrltr r r _cat)s  r#cCsDt|trt|t|ddSt|tt|dS)N)rfloatmathlogsqrtrZsquare)rr r r _acosh4s r*MsymcCs|s|ddfS|dfS)z:Extend window by 1 sample if needed for DFT-even symmetry.r%TFr )r+r,r r r _extend;s r-cCs$t||ks |dkrtd|dkS)z)Handle small or incorrect window lengths.rz.Window length M must be a non-negative integerr%)int ValueError)r+r r r _len_guardsDsr0wneededcCs|r|ddS|S)z.Wr8)r0rr:r-r*r'rEr;emptyZ empty_likerFrGprodZsqueezer4)r+ZnbarZsllZnormr,r6r@BAs2Zsignsm2miZnumerdenomrWr1scaler rUr _taylorsD   "($  racCt|d||dS)zCompute a Hamming window. The Hamming window is a taper formed by using a raised cosine with non-zero endpoints, optimized to minimize the nearest side lobe. gHzG?r7rNr+r,r6r r r _hammingrecCrb)zCompute a Hann window. The Hann window is a taper formed by using a raised cosine or sine-squared with ends that touch zero. rSr7rcrdr r r _hannrfrgrSc Cs<t|r tj|f|dS|dkrtj|f|dS|dkr"t||dSt||\}}tjd||d}t||dd}|d|d}||d||d}|||dd} ddttj d d|||d} tj|j |d} ddttj d |dd| ||d} t | | | g} t | |S) z[Compute a Tukey window. The Tukey window is also known as a tapered cosine window. r7rr8)r,r%r9NrSr3g) r0rr:rgr-r;r.rHr'rEshaper r4)r+rLr,r6r@rAwidthZn1Zn2Zn3Zw1Zw2Zw3r1r r r _tukeys* ,* rjstdcCslt|r tj|f|dSt||\}}tjd||d|dd}d||}t|d |}t||S)ztCompute a Gaussian window. The Gaussian widows has a Gaussian shape defined by the standard deviation(std). r7rr8r9r$)r0rr:r-r;r<r4)r+rkr,r6r@rAZsig2r1r r r _gaussians  rlr8cCs|r |dur tdt|rtj|f|dSt||\}}|dur'|dd}tjd||d}tt|| |}t||S)z+Compute an exponential (or Poisson) window.Nz"If sym==True, center must be None.r7r%r$r) r/r0rr:r-r;r<r=r4)r+centertaur,r6r@rAr1r r r _exponentials   rocCst|r tj|f|dSt||\}}tjd|ddd|d}|ddkr>> import paddle >>> n_fft = 512 >>> cosine_window = paddle.audio.functional.get_window('cosine', n_fft) >>> std = 7 >>> gaussian_window = paddle.audio.functional.get_window(('gaussian',std), n_fft) r rr%N)ZgaussianZ exponentialzThe 'z6' window needs one or more parameters -- pass a tuple.z#%s as window type is not supported._zUnknown window type.r,r6) rtuplerGrr/typewindow_function_registerrKeyError) rvrwrxr6r,argsZwinstrZwinfunceparamskwargsr r r get_windowOs2       r)Tr5)rOrPTTr5)rSTr5)Nr8Tr5)#r'typingrrrnumpyrrrrr|rrr#r&r*r.boolr-r0r4rBrKrNrarergrjrlrorqrsrtrurr r r r s        8 !