| 6 | 1 # | 
|  | 2 # Copyright INRA-URGI 2009-2010 | 
|  | 3 # | 
|  | 4 # This software is governed by the CeCILL license under French law and | 
|  | 5 # abiding by the rules of distribution of free software. You can use, | 
|  | 6 # modify and/ or redistribute the software under the terms of the CeCILL | 
|  | 7 # license as circulated by CEA, CNRS and INRIA at the following URL | 
|  | 8 # "http://www.cecill.info". | 
|  | 9 # | 
|  | 10 # As a counterpart to the access to the source code and rights to copy, | 
|  | 11 # modify and redistribute granted by the license, users are provided only | 
|  | 12 # with a limited warranty and the software's author, the holder of the | 
|  | 13 # economic rights, and the successive licensors have only limited | 
|  | 14 # liability. | 
|  | 15 # | 
|  | 16 # In this respect, the user's attention is drawn to the risks associated | 
|  | 17 # with loading, using, modifying and/or developing or reproducing the | 
|  | 18 # software by the user in light of its specific status of free software, | 
|  | 19 # that may mean that it is complicated to manipulate, and that also | 
|  | 20 # therefore means that it is reserved for developers and experienced | 
|  | 21 # professionals having in-depth computer knowledge. Users are therefore | 
|  | 22 # encouraged to load and test the software's suitability as regards their | 
|  | 23 # requirements in conditions enabling the security of their systems and/or | 
|  | 24 # data to be ensured and, more generally, to use and operate it in the | 
|  | 25 # same conditions as regards security. | 
|  | 26 # | 
|  | 27 # The fact that you are presently reading this means that you have had | 
|  | 28 # knowledge of the CeCILL license and that you accept its terms. | 
|  | 29 # | 
|  | 30 import sys | 
|  | 31 import time | 
|  | 32 | 
|  | 33 class UnlimitedProgress(object): | 
|  | 34     """Show the progress of a process when no upper bound is known""" | 
|  | 35 | 
|  | 36     def __init__(self, step = 1000, message = "Progress", verbosity = 0): | 
|  | 37         self.step           = step | 
|  | 38         self.progress       = 0 | 
|  | 39         self.message        = message | 
|  | 40         self.verbosity      = verbosity | 
|  | 41         self.maxMessageSize = 50 | 
|  | 42         self.startTime      = time.time() | 
|  | 43         self.elapsed        = 0 | 
|  | 44         if len(self.message) > self.maxMessageSize: | 
|  | 45             self.message = self.message[0:self.maxMessageSize-3] + "..." | 
|  | 46         self.show() | 
|  | 47 | 
|  | 48 | 
|  | 49     def inc(self): | 
|  | 50         self.progress += 1 | 
|  | 51         self.show() | 
|  | 52 | 
|  | 53 | 
|  | 54     def getPrintableElapsedTime(self, time): | 
|  | 55         timeHou = int(time) / 3600 | 
|  | 56         timeMin = int(time) / 60 - 60 * timeHou | 
|  | 57         timeSec = int(time) % 60 | 
|  | 58         if timeHou > 0: | 
|  | 59             return "%3dh %2dm" % (timeHou, timeMin) | 
|  | 60         if timeMin > 0: | 
|  | 61             return "%2dm %2ds" % (timeMin, timeSec) | 
|  | 62         return "%2ds" % (timeSec) | 
|  | 63 | 
|  | 64 | 
|  | 65     def show(self): | 
|  | 66         if self.verbosity <= 0: | 
|  | 67             return | 
|  | 68         elapsed = int(time.time() - self.startTime) | 
|  | 69         if (self.progress % self.step == 0) or (elapsed > self.elapsed + 10): | 
|  | 70             self.elapsed = elapsed | 
|  | 71             string = "%s %d -- time spent: %s\r" % (self.message, self.progress, self.getPrintableElapsedTime(elapsed)) | 
|  | 72             sys.stdout.write(string) | 
|  | 73             sys.stdout.flush() | 
|  | 74 | 
|  | 75 | 
|  | 76     def done(self): | 
|  | 77         if self.verbosity > 0: | 
|  | 78             elapsed = time.time() - self.startTime | 
|  | 79             string = "%s %d -- time spent: %s\r" % (self.message, self.progress, self.getPrintableElapsedTime(elapsed)) | 
|  | 80             print string | 
|  | 81 |