\;j1 ddlZddlmZmZmZddlZddlZddlmZGddZ e Z e jdeede defd Z e jdeeefdefd Ze jd ed edefd Ze jd edefdZe jdededefdZe j d'd ed ede defdZe j d'd eded ede def dZe j d'd eded ede def dZe j d(d ed ede defdZe jd'd ed ede defdZe jd'd ed ede defdZe j d)d ed ede defdZe j d'd eded ede def dZe j d*d ed ede defdZe jd'd ed ede defdZe jd'd ed ede defd Ze jd'd ed ede defd!Z e jd'd ed ede defd"Z! d'd#ee ee effd$ed%ede def d&Z"y)+N)ListTupleUnion)Tensorc eZdZdZddZdZy)WindowFunctionRegisterci|_yN_functions_dict)selfs gG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/audio/functional/window.py__init__zWindowFunctionRegister.__init__s !Ncfd}|S)Nc>|j}|j|<|Sr )__name__r )funcnamer s radd_subfunctionz8WindowFunctionRegister.register..add_subfunctions!==D)-D  &Kr)r rrs` rregisterzWindowFunctionRegister.registers  rc |j|Sr r )r rs rgetzWindowFunctionRegister.get"s##D))rr )r __module__ __qualname__rrrrrrrrs"*rrx data_typereturnc g}|D]s}tj|r7t|ts'|j t j |g|O|j t j ||ut j|Sr )npisscalar isinstancestrappendpaddle to_tensorconcat)rrlts r_catr+)sj A  ;;q>*Q"4 HHV%%qc95 6 HHV%%a3 4  == rct|tr1tj|tj|dzdz zSt j|t jt j |dz zS)N)r#floatmathlogsqrtr&square)rs r_acoshr44s\!UxxDIIadQh//00 ::a&++fmmA&6&:;; <s r _truncaterAMs"v rdtypec(t|rtj|f|St||\}}tjd|||dz dz z }tj dtj ||z d|zzz}t||S)zCompute a window with a generalized Gaussian shape. This function is consistent with scipy.signal.windows.general_gaussian(). rBr?@gr-)r<r&onesr8arangeexpabsrA)r5psigr6rB needs_truncnr=s r_general_gaussianrOVs1~{{A4u--Q_NA{ a%(AGs?:A 4&**QW-!a%889A Q $$ract|rtj|f|St||\}}tjt j t j ||}tj|f|}tt|D]#}|||tj||zzz }%t||S)zCompute a generic weighted sum of cosine terms window. This function is consistent with scipy.signal.windows.general_cosine(). rD) r<r&rGr8linspacer0pizerosrangelencosrA)r5rPr6rBrMfacr=ks r_general_cosinerZgs1~{{A4u--Q_NA{ //477(DGGQe .Wse1v}} LLO JJq477{R\\!_4AG c8IJQN O    rrE)r<r&rGr8r4r0rSrHempty empty_likerUrVprodsqueezerA)r5nbarsllnormr6rBrMBAs2signsm2minumerdenomrhr=scalerfrgs` @@r_taylorrysJ1~{{A4u--Q_NA{ sRxAq DGGA qAqDD3J1,, -B q$e ,B taxk /B   b !EE#A#JE!$Q$K bBCGnb FKK 2 q!trCxAo56 6   7 A2BFH(=$=>>E 3r7Q;  A2CR(8$899E++a"R&2cr7"2234++a"R&2b1fh<"7789  2" &--1E *+A aQ! n$ U  A Q $$rc t|d||S)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?rDr^r5r6rBs r_hammingr}s AtS 66rc t|d||S)zCompute a Hann window. The Hann window is a taper formed by using a raised cosine or sine-squared with ends that touch zero. rbrDr{r|s r_hannrs AsCu 55rc t|rtj|f|S|dkrtj|f|S|dk\r t||St ||\}}tj d||}t ||dz zdz }|d|dz}||dz||z dz }|||z dz d} ddtjtjd d|z|z |dz z zzzz} tj|j|} ddtjtjd |z dzd| z|z |dz z zzzz} tj| | | g} t| |S) z[Compute a Tukey window. The Tukey window is also known as a tapered cosine window. rDrrE)r6r.rFNrbr@g) r<r&rGrr8rHr:rWr0rSshaper(rA)r5r[r6rBrMrNwidthn1n2n3w1w2w3r=s r_tukeyrsw1~{{A4u-- z{{A4u-- #QC  Q_NA{ a%(A Q#% &E 1uqy B 519q5y1} %B 1u9q=? B FJJtww"sRx%/?1q5/I*IJKK LB RXXU +B  **TWWu q 038e3Cq1u3M MN O P B  r2rl#A Q $$rstdct|rtj|f|St||\}}tjd|||dz dz z }d|z|z}tj |dz |z }t ||S)ztCompute a Gaussian window. The Gaussian widows has a Gaussian shape defined by the standard deviation(std). rDrrErFr-)r<r&rGr8rHrIrA)r5rr6rBrMrNsig2r=s r _gaussianrs~1~{{A4u--Q_NA{ a%(AGs?:A s7S=D QT7T>"A Q $$rc>|r | tdt|rtj|f|St ||\}}||dz dz }tj d||}tj tj||z  |z }t||S)z+Compute an exponential (or Poisson) window.z"If sym==True, center must be None.rDr.r-r) r;r<r&rGr8rHrIrJrA)r5centertaur6rBrMrNr=s r _exponentialrs  v!=>>1~{{A4u--Q_NA{ ~a%1 a%(A FJJq6z**S01A Q $$rcrt|rtj|f|St||\}}tjd|dzdzdz|}|dzdk(r)d|zdz |z }tj ||dddg}n(d|z|dzz }tj ||dddg}t ||S) zCompute a triangular window.rDr.r-rrENr@)r<r&rGr8rHr(rA)r5r6rBrMrNr=s r_triangrs1~{{A4u--Q_NA{ a!a%A)7A1uz US[A  MM1a"g, ' EQW  MM1aBi. ) Q $$rct|rtj|f|St||\}}tjtj dd||dd}d|z tj tj|zzdtjz tjtj|zzz}td|dg|}t||S)z^Compute a Bohman window. The Bohman window is the autocorrelation of a cosine window. rDr@r.rEr) r<r&rGr8rJrRrWr0rSsinr+rA)r5r6rBrMrXr=s r_bohmanr's 1~{{A4u--Q_NA{ **V__RAU;AbA BC SFJJtww}--dgg   # A1 A aAYA Q $$rc$t|gd||S)aCompute a Blackman window. The Blackman window is a taper formed by using the first three terms of a summation of cosines. It was designed to have close to the minimal leakage possible. It is close to optimal, only slightly worse than a Kaiser window. )gzG?rbg{Gz?rDr]r|s r _blackmanr9s 10#U CCrct|rtj|f|St||\}}tjt j |z tjd||dzz}t||S)z,Compute a window with a simple cosine shape.rDrrb) r<r&rGr8rr0rSrHrA)r5r6rBrMr=s r_cosinerDsg1~{{A4u--Q_NA{ 477Q;&--1E"BS"HIJA Q $$rwindow win_lengthfftbinsc| }d}t|tr|d}t|dkDrN|dd}nHt|tr|dvrt d|zdz|}n t dtt |z t jd |z}|f|z} d |i} || d |i| S#t$r}t d |d}~wwxYw) aReturn a window of a given length and type. Args: window (Union[str, Tuple[str, float]]): The window function applied to the signal before the Fourier transform. Supported window functions: 'hamming', 'hann', 'gaussian', 'general_gaussian', 'exponential', 'triang', 'bohman', 'blackman', 'cosine', 'tukey', 'taylor'. win_length (int): Number of samples. fftbins (bool, optional): If True, create a "periodic" window. Otherwise, create a "symmetric" window, for use in filter design. Defaults to True. dtype (str, optional): The data type of the return window. Defaults to 'float64'. Returns: Tensor: The window represented as a tensor. Examples: .. code-block:: python >>> 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) rrr.N)gaussian exponentialzThe 'z6' window needs one or more parameters -- pass a tuple.z#%s as window type is not supported._zUnknown window type.r6rB) r#tuplerVr$r;typewindow_function_registerrKeyError) rrrrBr6argswinstrwinfunceparamskwargss r get_windowrOs8+C D&%  v;?!":D FC 0 0& $33  F 1CV 4E E  8*..sV|<]T !FS\F F 2% 26 22 8/0a78s8B$$ B>- B99B>)Tfloat64)TTr)rbTr)NrETr)#r0typingrrrnumpyr!r&rrrrr$r+r/r4r:boolr8r<rArOrZr^ryr}rrrrrrrrrrrrrs %%  * * 23""$DLSV%""$=eFEM"=v=%= ""$s$%""$34%""$&%""$3< %  % %-0 %  %% % ""$5> %  % % %/2 %  %% % ""$9BF FF#F36F F%F""$FO5% 5%,05%@C5% 5%%5%p""$77$7c7&7%7""$6S6t636v6%6""$6?% % %03% %%%B""$7@% %%!%14% %%%"""$AJ% %'+%;>% %%%&""$%s%%S%%%%"""$%s%%S%%%%"""$DD4DsD6D%D""$%s%%S%%%% 73 #uS%Z(( )737373  73  73r