U ?h3 @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|} |pt| |}t||}t|}|sF|dkr:d}n |dkrFd}|dk rht| ||\} } |pd| }n"| j|dkrtd| j||r| nd} t |r| dkrt | 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 Nrz-invalid number of data points ({0}) specified) rrr r rshape ValueErrorformatnp iscomplexobj empty_likerealimag) forward transformxtypenZaxisnorm overwrite_xworkers orthogonalizetmpcopiedoutr"U/var/www/html/venv/lib/python3.8/site-packages/scipy/fft/_pocketfft/realtransforms.py_r2rs.     r$Tdctidctdstidstc Cst|} t| ||\} }|p$t| |}t|dkr6|St| | |\} } |pL| }|sl|dkr`d}n |dkrld}t||}t|}|r| nd} t|r| dkrt | 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)rrrrsZaxesrrrrrrr r!r"r"r#_r2rn<s*     r+dctnidctndstnidstn)r Nr NFNN)r NNNFNN)numpyrrZpffthelperrrrrrr r functoolsr$partialr%__name__r&r'r(r+r,r-r.r/r"r"r"r#s: $ ) *