L iddlmZmZmZmZmZddlmZddlZ dZ hdZ dZ dZ ddd d Zddd d Z ddd d Z ddd d Z ddd dZ ddd dZ ddd dZ ddd dZ ddd dZ ddd dZ ddd dZ ddd dZ ddd dZ ddd dZdZ ddd dZ ddd dZ ddd dZ ddd dZ y) )array_namespaceis_numpyxp_unsupported_param_msg is_complexxp_float_to_complex) _pocketfftNcf|ttd|ttd|d}|S)Nworkersplanbackward) ValueErrorr)r r norms ^/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/fft/_basic_backend.py_validate_fft_argsrs?1)<== 1&9:: | K>fftfftnhfftifftifftnirfftirfftnc t|} t| r$tj|}||||||||St |||}t | dr9t | j|} |tvr | ||||} | S| ||||Stj|}|||||} | j| S#t|| }| ||||} Y| SxYw)Nnaxisr overwrite_xr r r)rrr rrnpasarrayrhasattrgetattrr complex_funcsr) func_strpocketfft_funcxrrrrr r xpxp_funcresys r _execute_1Dr,  B| JJqMa14d*5wTS S gtT 2Dr5"&&(+ } $ <a14d;JqADt44 1 AqADt4A ::a= <'2.a14d;J 3 B??Cc t|} t| r$tj|}||||||||St |||}t | dr9t | j|} |tvr | ||||} | S| ||||Stj|}|||||} | j| S#t|| }| ||||} Y| SxYw)Nsaxesrrr r r)r1r2rr) r%r&r'r1r2rrr r r(r)r*r+s r _execute_nDr34r-r.r c Ftdtj||||||| S)Nrr)r,r rr'rrrrr r s rrrMs( ujnna14d#.d LLrc Ftdtj||||||| S)Nrr)r,r rr6s rrrS( vzQT#.d LLrc Ftdtj||||||| S)Nrfftr)r,r r:r6s rr:r:Yr8rc Ftdtj||||||| S)Nrr)r,r rr6s rrr_* w 0 0!qt$#.d LLrc Ftdtj||||||| S)Nrr)r,r rr6s rrrer8rc Ftdtj||||||| S)Nihfftr)r,r r?r6s rr?r?kr<rc Ftdtj||||||| S)Nrr0)r3r rr'r1r2rrr r s rrrqr8rc Ftdtj||||||| S)Nrr0)r3r rrAs rrrxr<rc &t|||||||SNr4)rrAs rfft2rE~s 1dD+wT BBrc &t|||||||SrD)rrAs rifft2rG AtT;d CCrc Ftdtj||||||| S)Nrfftnr0)r3r rJrAs rrJrJr<rc &t|||||||SrD)rJrAs rrfft2rLrHrc Ftdtj||||||| S)Nrr0)r3r rrAs rrrs* x!2!2AD#.d LLrc &t|||||||SrD)rrAs rirfft2rO !QdKt DDrcP|dvrd}|S|dk(rd}|S|dk7rtd|d|S)N)Nr forwardr orthozInvalid norm value z.; should be "backward", "ortho", or "forward".)r)rs r_swap_directionrTsY !! K   K .tf5223 3 Krc t|}t|r1tj|}t j |||||||St ||r|j|}t|||t||||SrD) rrr r!r hfftnrconjrrTr'r1r2rrr r r(s rrVrVsu  B| JJqM1dD+wTRR!R GGAJ !Qod3wT 33rc &t|||||||SrD)rVrAs rhfft2rZrHrc t|}t|r1tj|}t j |||||||S|j t|||t||||SrD) rrr r!r ihfftnrWrJrTrXs rr\r\si  B| JJqM  AtT;dSS 775At_T%:$gD: ;;rc &t|||||||SrD)r\rAs rihfft2r^rPr)NNFN)NNNFN)N)r_NFN)!scipy._lib._array_apirrrrrr numpyr rr$r,r3rrr:rrr?rrrErGrJrLrrOrTrVrZr\r^rrresL 22"&#'L15L LL #'$(L26L $(%)L37L #'$(L26L $(%)L37L %)$(L26L&*%)L37L )-$(C26C *.%)D37D &*%)L37L *.$(D26D '+&*L48L +/&*E48E &*%) 337 3*.%)D37D '+&*;48;+/&*E48Er