a _$a(@sddlmZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZm Z ddl!m"Z#ddl$Z$Gdd d Z%e%Z&gd Z'd Z(e)Z*d d Z+ddZ,ddZ-d5ddZ.ddZ/ddZ0GdddZ1e,ejd6ddZe,ej2d7ddZ2e,ej3d8d!d"Z3e,ej4d9d#d$Z4e,ej5d:d%d&Z5e,ej6d;d'd(Z6e,ej7dd-d.Z9e,ej:d?d/d0Z:e,ej;d@d1d2Z;e,ej<dAd3d4Z|}|jdur8t|_|j|kr8td|j||jS)Nzos.cpu_count() returned value of {} greater than mkl.get_max_threads()'s value of {}. Using negative values of worker option may amount to requesting more threads than Intel(R) MKL can acommodate.)get_max_threads_countr os_cpu_countwarningswarnformat)r#Z max_threadsr$r$r% get_cpu_count2s  z$_cpu_max_threads_count.get_cpu_countcCs|jdurt|_|jSr )r!mklZget_max_threadsr"r$r$r%r'>s  z,_cpu_max_threads_count.get_max_threads_countN)__name__ __module__ __qualname__r&r,r'r$r$r$r%r-s r)!fftifftfft2ifft2fftnifftnrfftirfftrfft2irfft2rfftnirfftnZhfftZihfftr rrrrrrrr r r r rrrrrrZ next_fast_lenznumpy.scipy.fftcCs&t|d}|durtS||i|S)zFetch registered UA function.N) __implementedgetNotImplemented)methodargskwargsfnr$r$r%__ua_function__Rs rDcsfdd}|S)zEDecorator adds function to the dictionary of implemented UA functionscs |t<|Sr )r=)func scipy_funcr$r%inner\sz_implements..innerr$)rGrHr$rFr% _implementsZs rIcCs|dvrtd||duS)N)NZorthoz1Invalid norm value %s, should be None or "ortho".) ValueError)normr$r$r%_unitarycs rLcCs|dur.d}|dur t|j}q2t|j|}nd}t|}|durVttt| d}t|t|krntd|r|r|j|ddd|d<||fS)Nrrz&Shape and axes have different lengths.)listrrrangelenrJ)asaxesZinvrealZ shapelessr$r$r% _cook_nd_argsjs rUcs*|j|dur|jStfdd|DS)Ncsg|] }|qSr$r$).0ZairSr$r% z_tot_size..)rsizer)xrTr$rWr% _tot_size}sr\cCs`|durtSt|}|dkr&td|dkr\|td7}|dkr\td|t |S)ztHandle conversion of workers to a positive number of threads in the same way as scipy.fft.helpers._workers. Nrz!Number of workers must be nonzerorzsV   P$