xf2 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| }n9| j |dkr't dj| j ||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 Nrz,invalid number of data points ({}) specified) rrr r r shape ValueErrorformatnp iscomplexobj empty_likerealimag) forward transformxtypenaxisnorm overwrite_xworkers orthogonalizetmpcopiedouts E/usr/lib/python3/dist-packages/scipy/fft/_pocketfft/realtransforms.py_r2rr%s% A,C4S!!4K $ (DwG  19D QYD}#CD1 V!+V 41 G &413 334C 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)rrrrsaxesrrrr r!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/r0r1r2r3r&r$r=sM!JJJ?C8<&LRidDHH- yudhh/ idDHH- yudhh/ BF9='ITydhh/  %1ydhh/  %1r&