Merge pull request #44 from drvinceknight/progress_bar

Adding tqdm

Author: geraintpalmer

ciw/simulation.py

@@ -1,26 +1,21 @@
 from __future__ import division
 import os
-from random import (random, expovariate, uniform, triangular,
-    gammavariate, gauss, lognormvariate, weibullvariate)
+import tqdm
+from random import (expovariate, uniform, triangular, gammavariate,
+                    lognormvariate, weibullvariate)
 from csv import writer, reader
-import copy
-from decimal import Decimal, getcontext
+from decimal import getcontext
 from collections import namedtuple
 
-import yaml
 import numpy.random as nprandom
 
 from .node import Node
 from .exactnode import ExactNode, ExactArrivalNode
 from .arrival_node import ArrivalNode
 from .exit_node import ExitNode
-from .server import Server
-from .individual import Individual
-from .data_record import DataRecord
 from .state_tracker import *
 from .deadlock_detector import *
 
-
 Record = namedtuple('Record', 'id_number customer_class node arrival_date waiting_time service_start_date service_time service_end_date time_blocked exit_date destination queue_size_at_arrival queue_size_at_departure')
 
 class Simulation(object):
@@ -236,20 +231,35 @@ def simulate_until_deadlock(self):
             time_of_deadlock - self.times_dictionary[state]
             for state in self.times_dictionary.keys()}
 
-    def simulate_until_max_time(self, max_simulation_time):
+    def simulate_until_max_time(self, max_simulation_time, progress_bar=False):
         """
         Runs the simulation until max_simulation_time is reached.
         """
         self.nodes[0].update_next_event_date()
         next_active_node = self.find_next_active_node()
         current_time = next_active_node.next_event_date
+
+        if progress_bar is not False:
+            self.progress_bar = tqdm.tqdm(total=max_simulation_time)
+
         while current_time < max_simulation_time:
             next_active_node.have_event()
             for node in self.transitive_nodes:
                 node.update_next_event_date(current_time)
             next_active_node = self.find_next_active_node()
+
+            if progress_bar:
+                remaining_time = max_simulation_time - self.progress_bar.n
+                time_increment = next_active_node.next_event_date - current_time
+                self.progress_bar.update(min(time_increment, remaining_time))
+
             current_time = next_active_node.next_event_date
 
+        if progress_bar:
+            remaining_time = max(max_simulation_time - self.progress_bar.n, 0)
+            self.progress_bar.update(remaining_time)
+            self.progress_bar.close()
+
     def source(self, c, n, kind):
         """
         Returns the location of class c node n's arrival or

ciw/tests/test_simulation.py

@@ -104,6 +104,16 @@ def test_simulate_until_max_time_method(self):
             drl.append((dr.customer_class, dr.service_time))
         self.assertEqual(drl, [(1, 10.0), (0, 5.0), (0, 5.0)])
 
+    def test_simulate_until_max_time_with_pbar_method(self):
+        Q = ciw.Simulation(ciw.create_network(
+            'ciw/tests/testing_parameters/params.yml'))
+        Q.simulate_until_max_time(150, progress_bar=True)
+        self.assertEqual(Q.progress_bar.total, 150)
+        self.assertEqual(Q.progress_bar.n, 150)
+
+
+
+
     def test_simulate_until_deadlock_method(self):
         ciw.seed(3)
         Q = ciw.Simulation(ciw.create_network(

requirements.txt

@@ -1,4 +1,5 @@
 PyYAML
 networkx
 hypothesis
-numpy
+numpy
+tqdm

setup.py

@@ -18,5 +18,5 @@
     packages=['ciw'],
     description='A discrete event simulation library for open queueing networks',
     long_description=readme + '\n\n' + changes + '\n\n' + authors,
-    install_requires=["PyYAML", "networkx", "hypothesis", "numpy"],
+    install_requires=["PyYAML", "networkx", "hypothesis", "numpy", "tqdm"]
 )