tKg ddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl Z ddZe jedej Zde_e jedej Zd e_e jedej&Zd e_e jedej&Zd e_ dd Ze jedej Zd e_e jedej Zde_e jedej&Zde_e jedej&Zde_y)N) pypocketfft) _asfarray_init_nd_shape_and_axes _datacopied _fix_shape _fix_shape_1d_normalization_workersFc (t|} |xs t| |}t||}t|}|s|dk(rd}n|dk(rd}|t | ||\} } |xs| }n.| j |dkrt d| j |d|r| nd} tj|r_| tj| 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 Nrzinvalid number of data points (z ) specified) rrr r r shape ValueErrornp iscomplexobj empty_likerealimag) forward transformxtypenaxisnorm overwrite_xworkers orthogonalizetmpcopiedouts g/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/fft/_pocketfft/realtransforms.py_r2rr$s A,C4S!!4K $ (DwG  19D QYD}#CD1 V!+V 41 :399T?:K;WXX34C q$'KbmmC S#((D4'47C#((D4'47C S$sG] KKTdctidctdstidstc t|} t| ||\} }|xs t| |}t|dk(r|St | | |\} } |xs| }|s|dk(rd}n|dk(rd}t ||}t |}|r| nd} tj|r]| tj| 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 rN) rrrlenrr r rrrrr)rrrrsaxesrrrrr rr!r"s r#_r2rnr.;s A,C)#q$7KE44S!!4K 4yA~S%.KC'K  19D QYD $ (DwG34C q$'KbmmC S#((D$chh@#((D$chh@ S$dC- HHr%dctnidctndstnidstn)r NNFNN)r NNNFNN)numpyrrpffthelperrrrrr r r functoolsr$partialr&__name__r'r(r)r.r/r0r1r2r%r#r<sM!JJJ?C8<%LPidDHH- yudhh/ idDHH- yudhh/ BF9='ITydhh/  %1ydhh/  %1r%