��c@s�dd
lm
Z

ddlZddlZddlZddlZddlZddlZd�Zd�Z	ej
dkr�dZndZd�Zd	�Z
d
�Zd�Zej
dkr�eZeZnd
�ZdS(i�(
t
_Nc
Cs�y)tt
jd
��}|dkr(|SWnttfk
rB


n
Xy'tt
jd�}|dkri|SWnttfk
r�


n
XdS(s-try to count the number of CPUs on the systemtSC_NPROCESSORS_ONLNitNUMBER_OF_PROCESSORSi
(tinttostsysconftAttributeErrort
ValueErrortenvirontKeyError(
t
n((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyt	countcpuss









c
Cs}|jd
d�}
|
rdy t
|
�}|dkr7|SWqdtk
r`


tjtd���qdXnttt�d�d�S(Ntworkertnumcpusi
s!number of cpus must be an integerii (	tconfigRRtutiltAbortRtmintmaxR(tuit
sR
((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyt_numworkers s









tposixgz��?g錠9Y>)FcCs6|
|}t|�}|t
|||}|d
kS(setry to determine whether the benefit of multiple processes can
    outweigh the cost of starting themg333333�(Rt_startupcost(Rt	costperoptnopstlineartworkerstbenefit((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyt
worthwhile0s



cCs<t||
t
|��r+t||||�S|||f
�S(
sF
run a function, possibly in parallel in multiple worker
    processes.

    returns a progress iterator

    costperarg - cost of a single task

    func - function to run

    staticargs - arguments to pass to every invocation of the function

    args - arguments to split into chunks, to pass to individual
    workers
    (Rtlent_platformworker(Rt
costperargtfunct
staticargstargs((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyR8s

c#s8tj
�\}}t|�}tjtj��tjtjtj�
gd
g
��x�||�D]�tj�}|d
kr<
tjtj��
y\tj	|�
x;|
||f
�D]&\}	}
tj
|d|	|
f�
q�Wtjd
�
Wq<
tk
r



tjd�
q<



z|j
�
Wdtjd�
Xq<
Xn�j|�
qfW�j�
tj	|�
tj|dd
�}�f
d�����fd�}tjd|�
��j�
���fd�}
yDx=|D]5}|jd	d
�}t|d
�|d
d fV
q�WWn


��
|
�
�n
X|
�
dS(Nis%d %s
i�trbcsZxS�D]K}ytj
|tj�
Wqtk
rQ
}

|
jtjkrR�qRqXqWdS(
N(RtkilltsignaltSIGTERMtOSErrorterrnotESRCH(t
pterr(
tpids(s4/sys/lib/python2.7/site-packages/mercurial/worker.pytkillworkerses





csLxE�D]=}tt
j�d
�}
|
r�r|
�d<��
qqWdS(Ni
i(t_exitstatusRtwait(Rtst(R.R-tproblem(s4/sys/lib/python2.7/site-packages/mercurial/worker.pytwaitforworkersms








ttargetc
sgtjtj
��
�j�
�d
}|rc|d
krStjtj�|�
ntj|�
ndS(Ni(R&tSIGINTtjoinRR%tgetpidtsystexit(
tstatus(t
oldhandlerR2t
t(s4/sys/lib/python2.7/site-packages/mercurial/worker.pytcleanupus








t
 i
i�(RtpipeRR&t	getsignalR5tSIG_IGNt	partitiontforktclosetwritet_exittKeyboardInterruptt	tracebacktappendtreversetfdopent	threadingtThreadtstarttsplitR(RR!R"R#trfdtwfdRtpargstpidt
ititemtfpR3R=tlinet
l((R.R;R-R2R<s4/sys/lib/python2.7/site-packages/mercurial/worker.pyt_posixworkerKsL












 





















%




c

Cs=tj
|�rtj|�Stj|�r9tj|�Sd
S(s�convert a posix exit status into the same form returned by
    os.spawnv

    returns None if the process was stopped instead of exitingN(Rt	WIFEXITEDtWEXITSTATUStWIFSIGNALEDtWTERMSIG(
tcode((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyt_posixexitstatus�s



tntccs{t|�}||
||
}}d
}xMt
|
�D]?}|}||}|rg|d7}|d8}n|||!V
q4WdS(s,partition a list into N slices of equal sizeii
N(Rtxrange(tlsttnslicesR
tchunktsloptendRTRN((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyRB�s











(ti18nRR)RR&R8RLRRRtnameRRRRYR_RR/RB(((s4/sys/lib/python2.7/site-packages/mercurial/worker.pyt<module>s
H		
				<