o ॵi@svddlZddlmZddlZddlmZddlmZddl m Z dddZ dd d Z Gd d d Z dejfddZdS)N)Iterableget_terminal_size)Timerfile "{}" does not existcCst|s t||dSN)ospisfileFileNotFoundErrorformat)filenamemsg_tmplrT/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/preprocessors/cv/util.pycheck_file_exist s rcCs*|dkrdSt|}tj||dddS)NT)modeexist_ok)r expanduserosmakedirs)dir_namerrrrmkdir_or_exists rc@sBeZdZdZdddejfddZeddZd d Z dd d Z dS) ProgressBarz,A progress bar which can print the progress.r2TcCs,||_||_d|_||_|r|dSdS)Nr)task_num bar_width completedfilestart)selfrrr!r rrr__init__s zProgressBar.__init__cCst\}}|Srr)r"width_rrrterminal_width$s zProgressBar.terminal_widthcCsP|jdkr|jdd|jd|jdn|jd|jt|_dS)Nr[ z] 0/z, elapsed: 0s, ETA:zcompleted: 0, elapsed: 0s)rr writerflushrtimer)r"rrrr!)s $   zProgressBar.startrc CsH|dksJ|j|7_|j}|dkr|j|}ntd}|jdkr|jt|j}t|d||d}d|jd|jd|dd t|dd |d d }t|jt|jt |d t|jd}t d |}t||}d|d||} |j | | n|j d|jdt|dd|dd|j dS)Nrinfrg?z [{}] /z, z.1fz task/s, elapsed: zs, ETA: 5sg333333?>r(z completed: z , elapsed: zs, z tasks/s)rr+ since_startfloatrintminrr&lenmaxr r)r r*) r" num_taskselapsedfps percentageetamsgr mark_width bar_charsrrrupdate2s<        zProgressBar.updateN)r) __name__ __module__ __qualname____doc__sysstdoutr#propertyr&r!r@rrrrrs  rrc Kst|tr(t|dks Jt|dtsJt|dtsJ|d}|d}nt|tr2t|}ntdt|||d}g}|D]}|||fi||qA|j d|S)atTrack the progress of tasks execution with a progress bar. Tasks are done with a simple for-loop. Args: func (callable): The function to be applied to each task. tasks (list or tuple[Iterable, int]): A list of tasks or (tasks, total num). bar_width (int): Width of progress bar. Returns: list: The task results. r0rrz="tasks" must be an iterable object or a (iterator, int) tuple)r  ) isinstancetupler6rr4 TypeErrorrappendr@r r)) functasksrr kwargsrprog_barresultstaskrrrtrack_progressOs$      rS)r)r)ros.pathpathr rEcollections.abcrshutilrr+rrrrrFrSrrrrs      6