o p i6@sddlmZddlZddlmZddlZddlZer&ddlmZddl m Z GdddZ e Z e dSddZe dTdUddZe dTdVddZe dTdUddZe dWd!d"Ze dXd#d$Ze dYd%d&Ze dZd)d*Ze dTdUd+d,Ze dTd[d/d0Ze dTd\d2d3Ze d]dUd6d7Ze dTdUd8d9Ze dTdUd:d;Ze d^dUd=d>Ze dTd_d@dAZe d`dUdCdDZe dTdUdEdFZe dTdUdGdHZ e dTdUdIdJZ!e dTdUdKdLZ"  dTdadQdRZ#dS)b) annotationsN) TYPE_CHECKING)Tensor)_WindowLiteralc@s&eZdZddZdddZddZdS) WindowFunctionRegistercCs i|_dSN_functions_dictselfr e/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/audio/functional/window.py__init__ zWindowFunctionRegister.__init__Ncsfdd}|S)Ncs|j}|j|<|Sr)__name__r )funcnamer r radd_subfunction!s z8WindowFunctionRegister.register..add_subfunctionr )r rrr r rregister s zWindowFunctionRegister.registercCs |j|Srr )r rr r rget(rzWindowFunctionRegister.getr)r __module__ __qualname__rrrr r r rrs  rx list[Tensor] data_typestrreturnrcCsTg}|D] }t|rt|ts|t|g|q|t||qt|Sr)npZisscalar isinstancerappendpaddle to_tensorconcat)rrltr r r_cat/s  r&Tfloat64MintsymbooldtypecCst|r tj|f|dSt||\}}tjd||d}tj||d}tt||ddd||ddd||d}t||S)zm Computes the Bartlett window. This function is consistent with scipy.signal.windows.bartlett(). r,r@) _len_guardsr!ones_extendaranger"whereZ less_equal _truncater(r*r, needs_truncnwr r r _bartlett:s r:betafloatcCst|r tj|f|dSt||\}}tj||d}tjd||d}tj||d}|dd}t|td|||dt|}t||S)zcCompute the Kaiser window. This function is consistent with scipy.signal.windows.kaiser(). r-rr.r/) r0r!r1r2r"r3Zi0sqrtr5)r(r;r*r,r7r8alphar9r r r_kaiserOs  r?cCs"tjgd|d}t||||dS)zYNuttall window. This function is consistent with scipy.signal.windows.nuttall(). )gzD?g;%N?g1|?gC˅?r-)ar*r,)r!r"_general_cosine)r(r*r,r@r r r_nuttallfsrBTensor | floatcCsDt|trt|t|ddSt|tt|dS)Nrr.)rr<mathlogr=r!Zsquare)rr r r_acoshqs rFcCs|s|ddfS|dfS)z:Extend window by 1 sample if needed for DFT-even symmetry.r.TFr )r(r*r r rr2xs r2cCs$t||ks |dkrtd|dkS)z)Handle small or incorrect window lengths.rz.Window length M must be a non-negative integerr.)r) ValueError)r(r r rr0sr0r9neededcCs|r|ddS|S)zcCst||d|g||dS)zvCompute a generalized Hamming window. This function is consistent with scipy.signal.windows.general_hamming() rJr-rA)r(r>r*r,r r r_general_hammingsrYc str tjf|dSt|\}d|d}t|tj}|d|d|dd} tjd||dtj|df|dt } d| ddd<d| ddd<} t t D]u} | | t d| | | |ddd} | d krdt d| | | | dd}n<| t dkrdt d| | | d| }n"dt d| | | d| t d| | | | dd}| || <q]fd d }|tjd |d}|rd |dd}||9}| }t||S) zCompute a Taylor window. The Taylor window taper function approximates the Dolph-Chebyshev window's constant sidelobe level for a parameterized number of near-in sidelobes. r- r?r.NrIrc sDddtdtdtjd|ddS)Nr.rrr/r^)r!matmulZ unsqueezerUrDrR)r8ZFmr(mar rWs,z_taylor..WrJ)r0r!r1r2rFrDrRr3emptyZ empty_likerSrTprodZsqueezer5)r(ZnbarZsllZnormr*r,r7BAs2Zsignsm2minumberdenomrbr9scaler r`r_taylorsD   "($  rmcCt|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?r-rYr(r*r,r r r_hammingrqcCrn)zCompute a Hann window. The Hann window is a taper formed by using a raised cosine or sine-squared with ends that touch zero. r^r-rorpr r r_hannrrrsr^c 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. r-rrJ)r*r.r/Nr^rIg) r0r!r1rsr2r3r)rUrDrRshaper#r5)r(r>r*r,r7r8widthZn1Zn2Zn3Zw1Zw2Zw3r9r r r_tukey s* ,* rvstdcCslt|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). r-rrJr/r)r0r!r1r2r3rKr5)r(rwr*r,r7r8Zsig2r9r r r _gaussian,s  rxrJcCs|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.r-r.rr) rGr0r!r1r2r3rKrLr5)r(centertaur*r,r7r8r9r r r _exponential>s   r{cCst|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 exponentialZkaiserzThe 'z6' window needs one or more parameters -- pass a tuple.z! as window type is not supported._zUnknown window type.r*r,) rtuplerTrrGtypewindow_function_registerrKeyError) rrrr,r*argsZwinstrZwinfunceparamskwargsr r r get_windows.       r)rrrrrr)Tr')r(r)r*r+r,rrr) r(r)r;r<r*r+r,rrr)rrCrr)r(r)r*r+rr+)r(r)rr+)r9rrHr+rr) r(r)r@rPr*r+r,rrr) r(r)r>r<r*r+r,rrr)rZr[TTr')r^Tr') r(r)rwr<r*r+r,rrr)NrJTr') rrrr)rr+r,rrr)$ __future__rrDtypingrnumpyrr!rZfeatures.layersrrrrr&r:r?rBrFr2r0r5rOrArYrmrqrsrvrxr{r}rrrrr r r rs~           8!