ge2 >ddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl Z ddZe jed ejZd e_e jedejZd e_e jed ejZd e_e jedejZd e_ ddZe jed ejZde_e jedejZde_e jed ejZde_e jedejZde_dS)N) pypocketfft) _asfarray_init_nd_shape_and_axes _datacopied _fix_shape _fix_shape_1d_normalization_workersFc Rt|} |pt| |}t||}t|}|s|dkrd}n|dkrd}|t | ||\} } |p| }n>| j|dkr-t d| j||r| nd} tj |rP| 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 r Nrz,invalid number of data points ({}) specified) rrr r r shape ValueErrorformatnp iscomplexobj empty_likerealimag) forward transformxtypenaxisnorm overwrite_xworkers orthogonalizetmpcopiedouts Clib/python3.11/site-packages/scipy/fft/_pocketfft/realtransforms.py_r2rr&sc A,,C4S!!4!4K $ ( (DwG  199DD QYYD}#CD11 V!+V 41  G &41133 3 '334C q$'KbmC   S #(D4'47CCC #(D4'47CCC 9S$sG] K KKTdctidctdstidstc t|} t| ||\} }|pt| |}t|dkr|St | | |\} } |p| }|s|dkrd}n|dkrd}t ||}t |}|r| nd} tj|rN| 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)rrrrsaxesrrr r!r"rr#r$s r%_r2rnr0<sD A,,C)#q$77KE44S!!4!4K 4yyA~~S%..KC'K  199DD QYYD $ ( (DwG '334C q$'KbmC   S #(D$ch@@@ #(D$ch@@@ 9S$dC- H HHr'dctnidctndstnidstn)r Nr NFNN)r NNNFNN)numpyrrpffthelperrrrrr r r functoolsr&partialr(__name__r)r*r+r0r1r2r3r4r'r%r=s!!!!!!JJJJJJJJJJJJJJJJJJ?C8<&L&L&L&LRidDH-- yudh// idDH-- yudh// BF9='I'I'I'ITydh//  %11ydh//  %11r'