sbmy_control/progress_bar.py

15 lines
No EOL
858 B
Python

if __name__ == "__main__":
from argparse import ArgumentParser
from tqdm import tqdm
from low_level import progress_bar_from_logfile
parser = ArgumentParser(description="Progress bar from log files.")
parser.add_argument("-l","--logdir", type=str, help="Log directory.")
parser.add_argument("-s","--simname", type=str, help="Simulation name.")
parser.add_argument("-n","--N_tiles", type=int, help="Number of tiles.")
parser.add_argument("-v","--verbose", type=int, default=1, help="Verbose level.")
parsed_args = parser.parse_args()
for b in tqdm(range(1,parsed_args.N_tiles**3+1), desc="sCOLA", unit="box", disable=(parsed_args.verbose==0)):
progress_bar_from_logfile(parsed_args.logdir+parsed_args.simname+".log_"+str(b), desc=f"Box {b}/{parsed_args.N_tiles**3}", verbose=parsed_args.verbose, leave=False)