AVhi%LdZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZdZedej&dej(dfdZedej&dej(dfdZedej&ej(dfdZedej&dej(dfdZedej&dej(dfdZdZy)z*Ops for computing common window functions.N) constant_op)dtypes)ops) tensor_util) array_ops)cond)math_ops)nn_ops)special_math_ops)dispatch) tf_exportc|jstd|ztj|tj }|j jd|S)aRCheck window_length and dtype params. Args: window_length: A scalar value or `Tensor`. dtype: The data type to produce. Must be a floating point type. Returns: window_length converted to a tensor of type int32. Raises: ValueError: If `dtype` is not a floating point type or window_length is not a scalar. -dtype must be a floating point type. Found %sdtyper) is_floating ValueErrorrconvert_to_tensorrint32shapeassert_has_rank) window_lengthrs W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/signal/window_ops.py _check_paramsr sN    DuL MM'' V\\J-%%a( zsignal.kaiser_windowg(@c ^tj|d5t||}tj|}|dk(r!t j dg|cdddStj|tjdz dz }tj| |dztj}tj||}tj||}tjd|}tj||}|tjtj|tj||z z z}tj ||z t#j$|t#j$|z z} ddd| S#1swY SxYw)aGenerate a [Kaiser window][kaiser]. Args: window_length: A scalar `Tensor` indicating the window length to generate. beta: Beta parameter for Kaiser window, see reference below. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. [kaiser]: https://docs.scipy.org/doc/numpy/reference/generated/numpy.kaiser.html kaiser_windowrNg?@g?)r name_scoperrconstant_valueronesr castrfloat32rangesqrtr relusquareexpr bessel_i0e) rbetarnamewindow_length_const halflen_floatargonenumwindows rrr5sf" ~~dO,N!-7M%44]Ca ^^QCu - NN  m6>>:S@CHM ..-)<%~~ /C --5 )C ==U +D --5 )CMM-u=M v{{ hoocM122 45 5C \\#* %##C(+;+F+Ft+LLNF#N& -'N& -s>F"D9F""F,z#signal.kaiser_bessel_derived_windowcFtj|d5t||}|dz}t|dz||}t j |}t j |dd|dz }tj||dddfd}ddd|S#1swYSxYw) aGenerate a [Kaiser Bessel derived window][kbd]. Args: window_length: A scalar `Tensor` indicating the window length to generate. beta: Beta parameter for Kaiser window. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. [kbd]: https://en.wikipedia.org/wiki/Kaiser_window#Kaiser%E2%80%93Bessel-derived_(KBD)_window kaiser_bessel_derived_windowrrNr)axis) rr rrr cumsumr&rconcat) rr+rr,halflenkaiserw kaiserw_csumhalfwr2s rr4r4\s$ ~~d:;<!-7Mq GGaKU;G??7+L MM,s+l2.>> ?E   ueDbDk2 ;F < -< -s A5BB zsignal.vorbis_windowc tj|d5t||}tjtj ||}tj tjdz tjtj tjtj||z |dzzdz}ddd|S#1swYSxYw)aGenerate a [Vorbis power complementary window][vorbis]. Args: window_length: A scalar `Tensor` indicating the window length to generate. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. [vorbis]: https://en.wikipedia.org/wiki/Modified_discrete_cosine_transform#Window_functions vorbis_windowrr?N) rr rr r#r%sinnppipow)rrr,r/r2s rr?r?xs  ~~dO,!-7M --}5U CC \\"%%#+ X\\  m599 s 6)F -  -s B3CCzsignal.hann_windowTc $t|d|||ddS)aGenerate a [Hann window][hann]. Args: window_length: A scalar `Tensor` indicating the window length to generate. periodic: A bool `Tensor` indicating whether to generate a periodic or symmetric window. Periodic windows are typically used for spectral analysis while symmetric windows are typically used for digital filter design. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. Raises: ValueError: If `dtype` is not a floating point type. [hann]: https://en.wikipedia.org/wiki/Window_function#Hann_and_Hamming_windows hann_windowr@_raised_cosine_windowrperiodicrr,s rrFrFs , t]M8$c3 00rzsignal.hamming_windowc $t|d|||ddS)aGenerate a [Hamming][hamming] window. Args: window_length: A scalar `Tensor` indicating the window length to generate. periodic: A bool `Tensor` indicating whether to generate a periodic or symmetric window. Periodic windows are typically used for spectral analysis while symmetric windows are typically used for digital filter design. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. Raises: ValueError: If `dtype` is not a floating point type. [hamming]: https://en.wikipedia.org/wiki/Window_function#Hann_and_Hamming_windows hamming_windowgHzG?gq= ףp?rGrIs rrLrLs!0 t%5}h$dD 22rc \ jstdztj|||g5tjt j djjdtj}|dk(r!tjdgcdddStjtj|t jdt j }|jjddtj d z }tj||zzdz } tjtj"} t%j&d t(j*z| z| z |9tjtj, zz cdddSt/j.tj0dfd  fd cdddS#1swYyxYw) aHelper function for computing a raised cosine window. Args: name: Name to use for the scope. default_name: Default name to use for the scope. window_length: A scalar `Tensor` or integer indicating the window length. periodic: A bool `Tensor` indicating whether to generate a periodic or symmetric window. dtype: A floating point `DType`. a: The alpha parameter to the raised cosine window. b: The beta parameter to the raised cosine window. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. Raises: ValueError: If `dtype` is not a floating point type or `window_length` is not scalar or `periodic` is not scalar. rr)rr,rrrNrJr5c4tjgSNr)rr")rrsrz'_raised_cosine_window..s  e<rcdtjtjzz SrO)r r#cos)abcos_argrsrrPz'_raised_cosine_window..s% a!hll7&;";;5Ir)rrrr rrrrrrr!rr"r r#boolmodr%rconstantrBrCrRrequal) r, default_namerrJrrSrTr-evenncountrUs ` ``` @rrHrHs*    DuL MM ~~dL=(*CDK))-v||/>@M''*%44]Ca ^^QCu - KK}} hfkk K H NN""1% x||M1- -D mho59GA MM(..7 ?E""1ruu9E:UBQFG& ]]1q8<<#888 F%KK& 99}a(<I K'KKKsA3H"6D$H"$4H""H+)__doc__numpyrBtensorflow.python.frameworkrrrrtensorflow.python.opsrrr r r tensorflow.python.utilr tensorflow.python.util.tf_exportr radd_dispatch_supportr$rr4r?rFrLrHrrrfs913.+3+&*(2+6* !" &)d"#"J 01 58'-~~D24 !" '-~~D#.   (,FNN0!00 "# +/v~~2$24.Kr