ó ì[Jc@s~dZdZdZdZddlZddlZddlmZejddkrvdd l m Z dd l m Z niejdd kr¿ejd d kr¿ddlm Z dd lm Z n ddl m Z dd l m Z ddlZejZeƒZeƒZdadadZdd„Zd„Zddd„Zdefd„ƒYZddd„ƒYZddeed„ZdS(sº * 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. s Rene Dudfields0.3.0sPython licenseiÿÿÿÿN(tgeterrorii(t JoinableQueue(tEmptyiii(tQueuei@cCs(|r|an tƒattƒadS(s» 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(t _use_workerstbenchmark_workerst WorkerQueuet_wq(tnumber_of_workers((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pytinit4s  cCstjƒdatadS(s cleans up everything. N(RtstoptNonetFalseR(((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pytquitGs cs{ddl‰ddl‰ddl}|s<‡fd†}n|}|s„g}x9tdƒD]"}|jˆjdddƒƒq[Wn|}|jƒd}d}d}xÎtdtƒD]½} t| ƒ} |jƒ} x8td ƒD]*} d tj ƒGHt ||d | ƒ} qáW|jƒ}| j ƒ|| }d | |fGH||kr_|}| }|}n| |d kr¶Pq¶q¶W|S(sX 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. iÿÿÿÿNcsˆjj|dƒS(Ni i@(i i@(t transformtscale(tx(tpygame(sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pytdoitisi i›ii iáõisactive count:%st worker_queues#total time num_workers:%s: time:%s:i(i›i›( Rtpygame.transformttimetrangetappendtSurfacetMAX_WORKERS_TO_TESTRt threadingt activeCountttmapR (t a_bench_functthe_dataRRtthedataRtbestt best_numbert last_bestt num_workerstwqtt1txxtresultstt2t total_time((RsDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyRPs>   #       RcBsAeZdd„Zd„Zd„Zd„Zd„Zd„ZRS(icCs&tƒ|_g|_|j|ƒdS(N(Rtqueuetpoolt_setup_workers(tselfR#((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyt__init__šs  cCskg|_x0t|ƒD]"}|jjtd|jƒƒqWx(|jD]}|jtƒ|jƒqFWdS(sd Sets up the worker threads NOTE: undefined behaviour if you call this again. ttargetN(R+RRtThreadt threadloopt setDaemontTruetstart(R-R#t_ta_thread((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyR,Ÿs    cOs|jj|||fƒdS(s7 puts a function on a queue for running later. N(R*tput(R-tftargstkwArgs((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pytdo­scCs2|jjtƒx|jD]}|jƒqWdS(sK Stops the WorkerQueue, waits for all of the threads to finish up. N(R*R7tSTOPR+tjoin(R-tthread((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyR ³scCs{xttrv|jjƒ}|tkrE|jjtƒ|jjƒPqz|d|d|dŽWd|jjƒXqWdS(s4 Loops until all of the tasks are finished. iiiN(R3R*tgetR<R7t task_done(R-R9((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyR1»s   cCs|jjƒdS(s- waits until all tasks are complete. N(R*R=(R-((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pytwaitÍs(t__name__t __module__R.R,R;R R1RA(((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyR˜s      t FuncResultcBs&eZdZddd„Zd„ZRS(sy Used for wrapping up a function call so that the results are stored inside the instances result attribute. cCs(||_d|_||_||_dS(sö 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(R8R t exceptiontcallbackterrback(R-R8RFRG((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyR.Ös   cOsuy5|j||Ž|_|jr4|j|jƒnWn9tk rptƒ|_|jrq|j|jƒqqnXdS(N(R8tresultRFt ExceptionRRERG(R-R9tkwargs((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyt__call__ás    N(RBRCt__doc__R R.RK(((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyRDÒs ic Csw|r|}n4trt}n%|dkr7t||ƒSt|ƒ}t|jƒdkret||ƒSg}x5|D]-}|jt|ƒƒ|j|d|ƒqrW|ri|jƒ|j j ƒrÑt dƒ‚n| r(t r(|j ƒ|j j ƒr(|j j ƒ} | tk r%t dƒ‚q%q(n|rYtd„|ƒ} | rY| dj‚qYntd„|ƒS||gSdS(s' 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 - iiÿÿÿÿsbuggy threadmapcSs|jS(N(RE(R((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyt0scSs|jS(N(RH(R((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyRM4sN(RtmapRtlenR+RRDR;RAR*tqsizeRIR R?R<tfilterRE( R8tseq_argsR#RRAt stop_on_errorR$R'tsatumt error_ones((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyRîs:           ((RLt __author__t __version__t __license__t tracebacktsyst pygame.compatRt version_infotmultiprocessingRRR*Rt Py25QueueRR0tobjectR<tFINISHR RRRR R RRRDR3R(((sDc:\mingw\msys\1.0\newbuild\install\python\pygame\threads\__init__.pyt s4&      H: