a K)XT"@sdZdZdZdZddlZddlZddlmZejddkrTddl m Z dd l m Z nNejdd krejd d krddl m Z dd l m Z nddl m Z dd l m Z ddl Z e jZeZeZdadad ZdddZddZdddZGdddeZGdddZdddZdS)z * Experimental * Like the map function, but can use a pool of threads. Really easy to use threads. eg. tmap(f, alist) If you know how to use the map function, you can use threads. z Rene Dudfieldz0.3.0zPython licenseN)geterror)Queue)Empty@cCs|r |antattadS)z Does a little test to see if threading is worth it. Sets up a global worker queue if it's worth it. Calling init() is not required, but is generally better to do. N) _use_workersbenchmark_workers WorkerQueue_wq)Znumber_of_workersrm/home/tom/ab/renpy-build/tmp/install.linux-x86_64/lib/python3.9/site-packages/pygame_sdl2/threads/__init__.pyinit4srcCstdadadS)z cleans up everything. NF)r stopr rrrrquitGsrcsddlddlddl}|s*fdd}n|}|sZg}tdD]}|dddq>n|}|d}d}d }tdtD]} t| } |} td D]"} td t t ||| d } q|}| || }td | |f||kr|}| }|}| |dkr|q q||S)aX does a little test to see if workers are at all faster. Returns the number of workers which works best. Takes a little bit of time to run, so you should only really call it once. You can pass in benchmark data, and functions if you want. a_bench_func - f(data) the_data - data to work on. rNcsj|dS)N)i i@)Z transformZscalexpygamerrdoitiszbenchmark_workers..doit )r izactive count:%s) worker_queuez#total time num_workers:%s: time:%s:r) rZpygame.transformtimerangeappendZSurfaceMAX_WORKERS_TO_TESTr print threading activeCounttmapr)Z a_bench_funcZthe_datarrZthedatarbestZ best_numberZ last_best num_workerswqt1Zxxresultst2Z total_timerrrr Ps>    r c@s>eZdZdddZddZddZdd Zd d Zd d ZdS)r rcCst|_g|_||dSN)rqueuepool_setup_workers)selfr'rrr__init__szWorkerQueue.__init__cCsJg|_t|D]}|jt|jdq|jD]}|d|q.dS)zd Sets up the worker threads NOTE: undefined behaviour if you call this again. )targetTN)r.rr Thread threadloop setDaemonstart)r0r'_Za_threadrrrr/s    zWorkerQueue._setup_workerscOs|j|||fdS)z7 puts a function on a queue for running later. N)r-put)r0fargsZkwArgsrrrdoszWorkerQueue.docCs$|jt|jD] }|qdS)zK Stops the WorkerQueue, waits for all of the threads to finish up. N)r-r8STOPr.join)r0threadrrrrs  zWorkerQueue.stopcCsh|j}|tur,|jt|jqdqz(|d|di|dW|jq|j0qdS)z4 Loops until all of the tasks are finished. rrrN)r-getr<r8 task_done)r0r:rrrr4s   zWorkerQueue.threadloopcCs|jdS)z- waits until all tasks are complete. N)r-r=)r0rrrwaitszWorkerQueue.waitN)r) __name__ __module__ __qualname__r1r/r;rr4rArrrrr s  r c@s"eZdZdZdddZddZdS) FuncResultzy Used for wrapping up a function call so that the results are stored inside the instances result attribute. NcCs||_d|_||_||_dS)z f - is the function we that we call callback(result) - this is called when the function(f) returns errback(exception) - this is called when the function(f) raises an exception. N)r9 exceptioncallbackerrback)r0r9rGrHrrrr1szFuncResult.__init__cOsZz(|j|i||_|jr&||jWn,tyTt|_|jrP||jYn0dSr,)r9resultrG ExceptionrrFrH)r0r:kwargsrrr__call__s zFuncResult.__call__)NN)rBrCrD__doc__r1rLrrrrrEs rErTc Cs|r |}n$trt}n|dkr&t||St|}t|jdkrFt||Sg}|D]"}|t|||d|qN|r||j rt d|sts| |j r|j } | turt d|rttdd|} | r| djtdd|S||gSdS)a' like map, but uses a thread pool to execute. num_workers - the number of worker threads that will be used. If pool is passed in, then the num_workers arg is ignored. worker_queue - you can optionally pass in an existing WorkerQueue. wait - True means that the results are returned when everything is finished. False means that we return the [worker_queue, results] right away instead. results, is returned as a list of FuncResult instances. stop_on_error - rrzbuggy threadmapcSs|jSr,)rFrrrr0ztmap..cSs|jSr,)rIrrrrrN4rON)r mapr lenr.r rEr;rAr-qsizerJrr?r<listfilterrF) r9Zseq_argsr'rrAZ stop_on_errorr(r*saZumZ error_onesrrrr%s:       r%)r)NN)rNTT)rM __author__ __version__ __license__ tracebacksysZ pygame.compatr version_infor-rrZ Py25Queuer#r3objectr<ZFINISHr r r!rrr r rEr%rrrrs4        H: