o Me1 @sddlZddlmZddlmZddlmZddlZddlmZGdddeZeZ e deed e d efd d Z e deee fd efd dZe deded efddZe ded efddZe deded efddZe dBdedede d efddZe dBdede dede d ef dd Ze dBded!e dede d ef d"d#Ze dCdedede d efd&d'Ze dBdedede d efd(d)Ze dBdedede d efd*d+Ze dDdedede d efd-d.Ze dBded/e dede d ef d0d1Ze dEdedede d efd3d4Ze dBdedede d efd5d6Ze dBdedede d efd7d8Ze dBdedede d efd9d:Ze dBdedede d efd;d<Z  dBd=ee ee e ffd>ed?ede d ef d@dAZ!dS)FN)List)Tuple)Union)Tensorc@s&eZdZddZdddZddZdS) WindowFunctionRegistercCs t|_dSN)dict_functions_dictselfr ND:\Projects\ConvertPro\env\Lib\site-packages\paddle/audio/functional/window.py__init__s 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|Sr)r )r rr r r get"s zWindowFunctionRegister.getr)r __module__ __qualname__rrrr r r r rs  rx data_typereturncsfdd|D}t|S)Ncsg|]}t|qSr )paddleZ to_tensor).0_rr r +sz_cat..)rconcat)rrlr rr _cat)s r!cCsDt|trt|t|ddSt|tt|dS)N) isinstancefloatmathlogsqrtrZsquare)rr r r _acosh/s r)MsymcCs|s|ddfS|dfS)z:Extend window by 1 sample if needed for DFT-even symmetry.r#TFr )r*r+r r r _extend6s 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_guards?sr/wneededcCs|r|ddS|S)z.Wr7)r/rr9r,r)r&rDr:emptyZ empty_likerErFprodZsqueezer3)r*ZnbarZsllZnormr+r5r?BAs2Zsignsm2miZnumerdenomrVr0scaler rTr _taylor}sD   "($  r`cCt|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?r6rMr*r+r5r r r _hammingrdcCra)zCompute a Hann window. The Hann window is a taper formed by using a raised cosine or sine-squared with ends that touch zero. rRr6rbrcr r r _hannrerfrRc 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. r6rr7)r+r#r8NrRr2g) r/rr9Zhannr,r:r-rGr&rDshaperr3)r*rKr+r5r?r@widthZn1Zn2Zn3Zw1Zw2Zw3r0r r r _tukeys* ,* ristdcCslt|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). r6rr7r8r")r/rr9r,r:r;r3)r*rjr+r5r?r@Zsig2r0r r r _gaussians  rkr7cCs|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.r6r#r"r) r.r/rr9r,r:r;r<r3)r*centertaur+r5r?r@r0r r r _exponentials   rncCst|r tj|f|dSt||\}}tjd|ddd|d}|ddkrs         8 !