idZdZdZdZddlmZmZddlZejZe Z e Z da da dZddZd Zdd ZGd d ZGd dZddZy)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@c@|r|an tattay)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_workerss a/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/pygame/threads/__init__.pyinitr 's ( (*  l #Cc4tjdaday)zcleans up everything.NF)r stoprrr quitr8sHHJ CLrc6ddlddlddl}|sfd}n|}|s*tdDcgc]}j ddd}}n|}|jdz}d}tdt D]}t |} |j} tdD]0} td tjt||| 2|j} | j| | z } td |d | d | |kr|}| }||z dkDs|S|Scc}w)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. rNc<jj|dS)N)i i@) transformscale)xpygames r doitzbenchmark_workers..doitVs##))!Z8 8r )r iz active count:) worker_queueztotal time num_workers:z: time::) rpygame.transformtimerangeSurfaceMAX_WORKERS_TO_TESTr print threading active_counttmapr) a_bench_functhe_datar"rrthedatabest best_number num_workerswqt1_t2 total_timers @r rr@s3   9 >CBiH6>>*a4HH 99; "DKQ 34  % YY[r 1A M)"8"8":!;< = wR 0 1YY[  "W  ' }GJ>#Dt| t$ $$& 'Qa,DG, $$& $$&s B B'c8|jjy)z#waits until all tasks are complete.N)r8rM)r;s r waitzWorkerQueue.waits rN)r) __name__ __module__ __qualname__r<r:rJrrArTrrr r r s ) * 'rr ceZdZdZddZdZy) FuncResultzsUsed for wrapping up a function call so that the results are stored inside the instances result attribute. NcJ||_d|_d|_||_||_y)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)rG exceptionresultcallbackerrback)r;rGr]r^s r r<zFuncResult.__init__s'     rc |j|i||_|jr|j|jyy#t$r=}||_|j r |j |jYd}~yYd}~yd}~wwxYwr7)rGr\r] Exceptionr[r^)r;rHkwargses r __call__zFuncResult.__call__sp - $&&$1&1DK}} dkk* -DN|| T^^,, -s>A B .BBNN)rUrVrW__doc__r<rcrrr rYrYs  -rrYc|r|}n)trt}n|dk(r t||St|}t|jdk(r t||Sg}|D]1}|j t ||j|d|3|r|j|jjr td|s]tsW|j|jjr-|jj} | tur td|r'tt!d|} | r| dj"d|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 threadmapc|jSr7)r[)rs r ztmap.. s q{{rc34K|]}|jywr7)r\).0rs r ztmap..s*Q*s)r mapr lenr9r?rYrJrTr8qsize RuntimeErrorrrQrLlistfilterr[) rGseq_argsr/rrT stop_on_errorr0resultssaum error_oness r r)r)s3  Ba1h''[)B 277|q1hGz!}% gbk2     88>> 01 1C GGIxx~~XX\\^T>&'899 f%:GDEJ m---*'** =r)rrd)rNTT)re __author__ __version__ __license__r8rrr'r@objectrLFINISHr rr%r rrr rYr)rrr r~s}       x    $"<~..b--:Br