o i@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 Q/home/kim/smarthome/.venv/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@) 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/r appendThread threadloop setDaemonstart)r1r'r*Za_threadr r r r0s    zWorkerQueue._setup_workerscOs|j|||fdS)z-puts a function on a queue for running later.N)r.put)r1fargsZkwArgsr r r doszWorkerQueue.docCs$|jt|jD]}|q dS)zAStops the WorkerQueue, waits for all of the threads to finish up.N)r.r9STOPr/join)r1threadr 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=r9 task_done)r1r;r r r r6s   zWorkerQueue.threadloopcCs|jdS)z#waits until all tasks are complete.N)r.r>)r1r r r waitszWorkerQueue.waitN)r) __name__ __module__ __qualname__r2r0r<r r6rCr 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)r: exceptionresultcallbackerrback)r1r:rJrKr r r r2s  zFuncResult.__init__c Os~z|j|i||_|jr||jWdSWdSty>}z||_|jr3||jWYd}~dSWYd}~dSd}~wwr-)r:rIrJ ExceptionrHrK)r1r;kwargser r r __call__szFuncResult.__call__NN)rDrErF__doc__r2rOr r r r rGs  rGrTc 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-)rHrr r r  sztmap..css|]}|jVqdSr-)rIrr r r sztmap..)rmaprlenr/r4rGr<rCr.qsize RuntimeErrorr rAr=listfilterrH) r:Zseq_argsr'rrCZ stop_on_errorr(resultssaZumZ error_onesr r r r%s:       r%)rrP)rNTT)rQ __author__ __version__ __license__r.rrr#r5objectr=ZFINISHrrr!r r rrrGr%r r r r s$   ?1