Added tqdm progress bar

MintonGroup · Nov 22, 2022 · 9f429d1 · 9f429d1
1 parent 35d1cf0
commit 9f429d1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/swiftest/swiftest/simulation_class.py b/python/swiftest/swiftest/simulation_class.py
@@ -26,6 +26,7 @@
 import subprocess
 import shlex
 import warnings
+from tqdm import tqdm
 from typing import (
     Literal,
     Dict,
@@ -376,7 +377,8 @@ def _run_swiftest_driver(self):
                                   env=env,
                                   universal_newlines=True) as p:
                 process_output = False
-                for line in p.stdout:
+                noutput = int((self.param['TSTOP'] - self.param['T0']) / (self.param['DT'] * self.param['ISTEP_OUT']))
+                for line in tqdm(p.stdout,total=noutput):
                     if "SWIFTEST STOP" in line:
                         process_output = False