From 230e37096d4c386eeed36a9df329e8a52b1b808c Mon Sep 17 00:00:00 2001
From: Tor Nordam <tor.nordam@sintef.no>
Date: Wed, 21 Aug 2024 14:19:17 +0200
Subject: [PATCH] Quick implementation of progress bar option for the run
 method

---
 opendrift/models/basemodel/__init__.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/opendrift/models/basemodel/__init__.py b/opendrift/models/basemodel/__init__.py
index 5d14e86ad..83a14dd0e 100644
--- a/opendrift/models/basemodel/__init__.py
+++ b/opendrift/models/basemodel/__init__.py
@@ -1738,7 +1738,8 @@ def run(self,
             outfile=None,
             export_variables=None,
             export_buffer_length=100,
-            stop_on_error=False):
+            stop_on_error=False,
+            progress_bar=False):
         """Start a trajectory simulation, after initial configuration.
 
         Performs the main loop:
@@ -2009,7 +2010,13 @@ def run(self,
         self.timer_end('preparing main loop')
         self.timer_start('main loop')
         self.memory_usage = np.array([])
-        for i in range(self.expected_steps_calculation):
+
+        if progress_bar:
+            from tqdm.auto import trange
+            iterator = trange
+        else:
+            iterator = range
+        for i in iterator(self.expected_steps_calculation):
             self.memory_usage = np.append(self.memory_usage, psutil.virtual_memory().used / (1024.0**3))
             try:
                 # Release elements