o tVg@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)number_of_workersr K/home/ubuntu/.local/lib/python3.10/site-packages/pygame/threads/__init__.pyinit's r cCstdadadS)zcleans up everything.NF)rstoprr r r r quit8src 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 )Surface.0rrr r ]sz%benchmark_workers.. iz active count:) worker_queueztotal time num_workers:z: time::) rpygame.transformtimerangeMAX_WORKERS_TO_TESTrprint threading active_counttmapr ) a_bench_functhe_datar"rthedatabest 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)r6r#appendThread threadloop setDaemonstart)r8r.r1a_threadr r r r7s    zWorkerQueue._setup_workerscOs|j|||fdS)z-puts a function on a queue for running later.N)r5put)r8fargskwArgsr r r doszWorkerQueue.docCs$|jt|jD]}|q dS)zAStops the WorkerQueue, waits for all of the threads to finish up.N)r5rASTOPr6join)r8threadr r r r s   zWorkerQueue.stopcCsf |j}|tur|jt|jdSz|d|di|dW|jn|jwq)z*Loops until all of the tasks are finished.Trr N)r5getrFrA task_done)r8rCr r r r=s   zWorkerQueue.threadloopcCs|jdS)z#waits until all tasks are complete.N)r5rG)r8r r r waitszWorkerQueue.waitN)r) __name__ __module__ __qualname__r9r7rEr r=rLr 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)rB exceptionresultcallbackerrback)r8rBrSrTr r r r9s  zFuncResult.__init__c Os~z|j|i||_|jr||jWdSWdSty>}z||_|jr3||jWYd}~dSWYd}~dSd}~wwr4)rBrRrS ExceptionrQrT)r8rCkwargser r r __call__szFuncResult.__call__NN)rMrNrO__doc__r9rXr r r r rPs  rPrTc 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|jSr4)rQrr r r  sztmap..css|]}|jVqdSr4)rRrr r r sztmap..)rmaprlenr6r;rPrErLr5qsize RuntimeErrorr rJrFlistfilterrQ) rBseq_argsr.rrL stop_on_errorr/resultssaum error_onesr r r r(s:       r()rrY)rNTT)rZ __author__ __version__ __license__r5rrr&r<objectrFFINISHrrr$r rrrrPr(r r r r s$   ?1