o OZh@sdZdZdZdZddlmZmZddlZejZe Z e Z da da dZddd Zd d Zdd d ZGdddZGdddZdddZdS)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 license)QueueEmptyN@cCs|r|antattadS)zDoes 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_workersr F/var/www/auris/lib/python3.10/site-packages/pygame/threads/__init__.pyinit's r cCstdadadS)zcleans up everything.NF)rstoprr r r r quit8sr c sddlddlddl}|sfdd}n|}|s%fddtdD}n|}|d}d}tdtD]I}t|}|} td D]} td tt |||d qB|} | | | } td |d | d| |krt|}| }||dkr}|Sq4|S)a?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 transformscalexpygamer r doitVszbenchmark_workers..doitcsg|] }dddqS))rr )ZSurface.0rrr r ]sz%benchmark_workers.. iz active count:) worker_queueztotal time num_workers:z: time::) rZpygame.transformtimerangeMAX_WORKERS_TO_TESTrprint threading active_counttmapr ) Z a_bench_funcZthe_datarrZthedatabestZ best_number num_workerswqt1_t2 total_timer rr r@s:   rc@s>eZdZdddZddZddZdd Zd d Zd d ZdS)rrcCst|_g|_||dSN)rqueuepool_setup_workers)selfr&r r r __init__szWorkerQueue.__init__cCsJg|_t|D] }|jt|jdq|jD] }|d|qdS)z]Sets up the worker threads NOTE: undefined behaviour if you call this again. )targetTN)r.rappendThread threadloop setDaemonstart)r0r&r)Za_threadr r r r/s    zWorkerQueue._setup_workerscOs|j|||fdS)z-puts a function on a queue for running later.N)r-put)r0fargsZkwArgsr r r doszWorkerQueue.docCs$|jt|jD]}|q dS)zAStops the WorkerQueue, waits for all of the threads to finish up.N)r-r8STOPr.join)r0threadr r r r s   zWorkerQueue.stopcCsf |j}|tur|jt|jdSz|d|di|dW|jn|jwq)z*Loops until all of the tasks are finished.TrrN)r-getr<r8 task_done)r0r:r r r r5s   zWorkerQueue.threadloopcCs|jdS)z#waits until all tasks are complete.N)r-r=)r0r r r waitszWorkerQueue.waitN)r) __name__ __module__ __qualname__r1r/r;r r5rBr r r r rs   rc@s"eZdZdZdddZddZdS) FuncResultzsUsed for wrapping up a function call so that the results are stored inside the instances result attribute. NcCs"||_d|_d|_||_||_dS)zf - 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 exceptionresultcallbackerrback)r0r9rIrJr r r r1s  zFuncResult.__init__c Os~z|j|i||_|jr||jWdSWdSty>}z||_|jr3||jWYd}~dSWYd}~dSd}~wwr,)r9rHrI ExceptionrGrJ)r0r:kwargser r r __call__szFuncResult.__call__NN)rCrDrE__doc__r1rNr r r r rFs  rFrTc Cs|r|}ntr t}n |dkrt||St|}t|jdkr#t||Sg}|D]}|t|||d|q'|r{||j rHt d|sbtsb| |j rb|j } | turbt d|rtttdd|} | rt| djdd|DS||gS)alike 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 - rzbuggy threadmapcSs|jSr,)rGrr r r  sztmap..css|]}|jVqdSr,)rHrr r r sztmap..)rmaprlenr.r3rFr;rBr-qsize RuntimeErrorr r@r<listfilterrG) r9Zseq_argsr&rrBZ stop_on_errorr'resultssaZumZ error_onesr r r r$s:       r$)rrO)rNTT)rP __author__ __version__ __license__r-rrr"r4objectr<ZFINISHrrr r r rrrFr$r r r r s$   ?1