o  h @sddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl Z  ddd Ze ed ejZd e_e edejZd e_e ed ejZd e_e edejZde_  dddZe ed ejZde_e edejZde_e ed ejZde_e edejZde_dS)N) pypocketfft) _asfarray_init_nd_shape_and_axes _datacopied _fix_shape _fix_shape_1d_normalization_workersFc Cst|} |p t| |}t||}t|}|s#|dkrd}n|dkr#d}|dur4t| ||\} } |p2| }n| j|dkrFtd| j|d|rJ| nd} t|rv| durZt | n| } || j ||f|| j ||| j ||f|| j || S|| ||f|| || S)zForward or backward 1-D DCT/DST Parameters ---------- forward : bool Transform direction (determines type and normalisation) transform : {pypocketfft.dct, pypocketfft.dst} The transform to perform r Nrzinvalid number of data points (z ) specified) rrr r rshape ValueErrornp iscomplexobj empty_likerealimag) forward transformxtypenaxisnorm overwrite_xworkers orthogonalizetmpcopiedoutr"W/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/fft/_pocketfft/realtransforms.py_r2rs*     r$Tdctidctdstidstc Cst|} t| ||\} }|pt| |}t|dkr|St| | |\} } |p&| }|s6|dkr0d}n|dkr6d}t||}t|}|rC| nd} t|rm| durSt | n| } || j |||| j ||| j |||| j || S|| |||| || S)zForward or backward nd DCT/DST Parameters ---------- forward : bool Transform direction (determines type and normalisation) transform : {pypocketfft.dct, pypocketfft.dst} The transform to perform rr r N) rrrlenrr r rrrrr)rrrrsaxesrrrrrrr r!r"r"r#_r2rn;s*     r,dctnidctndstnidstn)r Nr NFNN)r NNNFNN)numpyrrpffthelperrrrrrr r functoolsr$partialr%__name__r&r'r(r,r-r.r/r0r"r"r"r#s4 $ ( *