a Df@svddlZddlZddlZddlZddlmZddlmZGdddeZGdddeZ Gd d d eZ dd d Z dS)N) clear_output)ProgressIndicatorcseZdZdZejdgdddZejdddZej d d dZ ej d d dZ ej d ddZ iZgZfddZddZdddZdddZZS) ProgressBarz A simple text progress bar suitable for both the IPython notebook and the IPython interactive prompt. ProgressBars are automatically nested if a previous instantiated progress bars has not achieved 100% completion. stdout)rdisabled broadcasta Parameter to control display of the progress bar. By default, progress is shown on stdout but this may be disabled e.g. for jobs that log standard output to file. If the output mode is set to 'broadcast', a socket is opened on a stated port to broadcast the completion percentage. The RemoteProgress class may then be used to view the progress from a different process.)defaultZobjectsdocFzB The width of the progress bar as the number of charactersr r #z5 The character used to fill the progress bar. zA The character for the blank portion of the progress bar.Tz If enabled, the progress bar will disappear and display the total elapsed time once 100% completion is reached.c s4d|_|ddtj|tjfi|dS)NrF) start_time_stdout_displayrcurrent_progressappendsuper__init__selfparams __class__f/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/holoviews/ipython/widgets.pyr0s  zProgressBar.__init__cCs8|jdurt|_|jd|jd}|jd|d|}|jdkrLdS|jdkr|dkr|jrt|j}tr|tdd |j|d |d |d f|_d d d|j D}t j |n |||dkrtj rtj dSd|jvr||jd<|jddur4|jd|d|jdS)z; Update the progress bar within the specified percent_rangeNrY@rrd z100%% %s %02d:%02d:%02di<cSsg|] }|jqSrout.0Zpgrrr Fz(ProgressBar.__call__..socket|)rtimeZ percent_rangedisplay elapsed_timerlabellowerr#joinrsysrwriterrpopcache _get_socketsend)r percentagespanelapsedoutputrrr__call__6s0     zProgressBar.__call__cCstr td|j}t|dkr,t||n|j}|j|}t|jdkrPdnd}|d|jrj|jdnd|j |dt|j |||_ |rt j ddd |jDt j td dS) Nrr r!z{}[{}{}] {:0.1f}%z: rcSsg|] }|jqSrr"r$rrrr&br'z/ProgressBar._stdout_display..g-C6?)rwidthintmathfloorlenrformatr- fill_charr#r0rr1r/flushr*sleep)rr6r+Zpercent_per_charZ char_countZ blank_countprefixrrrrUs&   zProgressBar._stdout_displaycCsjddl}|}||j}z(|jd|||d}|jd||WStyd|jdYdS0dS)Nrztcp://*)min_portmax_port max_triesz#Progress broadcast bound to port %dz.No suitable port found for progress broadcast.)zmqContextr(ZPUBZbind_to_random_portparammessage Exception)rrIrJrKrLcontextsockportrrrr4gs   zProgressBar._get_socket)T)rFrGrH)__name__ __module__ __qualname____doc__rNZObjectSelectorr+Integerr<StringrBZ blank_charBooleanr,r3rrr:rr4 __classcell__rrrrr s   rcsHeZdZdZejdddZejdddZfddZ fd d Z Z S) RemoteProgressz Connect to a progress bar in a separate process with output_mode set to 'broadcast' in order to display the results (to stdout). localhostz4 Hostname where progress is being broadcast.r rFzTarget port on hostname.c stjfd|i|dS)NrSrr)rrSrrrrrszRemoteProgress.__init__csddl}|}||j}||jd|d|jdt|j d}z4| }| d\}}t |}||_ t|WqLty|dur|jd|ddYqYqLty|jd |YqL0qLdS) Nrr!ztcp://:r)z Exited at z.3fz % completionz"Could not process socket message: )rLrMr(ZSUB setsockoptZ SUBSCRIBEconnecthostnamestrrSrecvsplitfloatr-rr:KeyboardInterruptrNrOrP)rrLrQrRpercentrOZ percent_strr-rrrr:s$   zRemoteProgress.__call__) rTrUrVrWrNrYrbrXrSrr:r[rrrrr\ws  r\csNeZdZdZejdddZejejj j ddZ fddZ fdd Z ZS) RunProgressa RunProgress breaks up the execution of a slow running command so that the level of completion can be displayed during execution. This class is designed to run commands that take a single numeric argument that acts additively. Namely, it is expected that a slow running command 'run_hook(X+Y)' can be arbitrarily broken up into multiple, faster executing calls 'run_hook(X)' and 'run_hook(Y)' without affecting the overall result. For instance, this is suitable for simulations where the numeric argument is the simulated time - typically, advancing 10 simulated seconds takes about twice as long as advancing by 5 seconds. rzG The run interval used to break up updates to the progress bar.r z By default updates time in param which is very fast and does not need a progress bar. Should be set to some slower running callable where display of progress level is desired.c stjfi|dS)Nr^rrrrrszRunProgress.__init__csld}|||jkrB||j||j7}td|t|q||}|dkrh||tddS)z Execute the run_hook to a total of value, breaking up progress updates by the value specified by interval. rrN)intervalrun_hookrr:rf)rvalueZ completed remainingrrrr:s   zRunProgress.__call__)rTrUrVrWrNNumberrjCallableZDynamicZtime_fnZadvancerkrr:r[rrrrris  riFccs^t}|durt|n|}t|}t|\}}||d|d|rR||fVq"|Vq"dS)z A helper utility to display a progress bar when iterating over a collection of a fixed length or a generator (with a declared length). If enum=True, then equivalent to enumerate with a progress bar. Ng?r)rr@ enumeratenext)iteratorenumlengthprogressgenivalrrrrus  ru)FN) r>r0r*rNZIPython.core.displayrZ core.utilrrr\rirurrrrs  l%,