Mercurial > repos > yufei-luo > s_mart
comparison SMART/Java/Python/misc/Progress.py @ 6:769e306b7933
Change the repository level.
| author | yufei-luo |
|---|---|
| date | Fri, 18 Jan 2013 04:54:14 -0500 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 5:ea3082881bf8 | 6:769e306b7933 |
|---|---|
| 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 Progress(object): | |
| 34 """Show the progress of a process""" | |
| 35 | |
| 36 def __init__(self, aim, message = "Progress", verbosity = 0): | |
| 37 self.aim = aim | |
| 38 self.progress = 0 | |
| 39 self.message = message | |
| 40 self.length = -1 | |
| 41 self.verbosity = verbosity | |
| 42 self.maxMessageSize = 50 | |
| 43 self.barSize = 80 | |
| 44 self.startTime = time.time() | |
| 45 self.elapsed = 0 | |
| 46 if len(self.message) > self.maxMessageSize: | |
| 47 self.message = self.message[0:self.maxMessageSize-3] + "..." | |
| 48 self.show() | |
| 49 | |
| 50 | |
| 51 def inc(self): | |
| 52 self.progress += 1 | |
| 53 self.show() | |
| 54 | |
| 55 | |
| 56 def getPrintableElapsedTime(self, time): | |
| 57 timeHou = int(time) / 3600 | |
| 58 timeMin = int(time) / 60 - 60 * timeHou | |
| 59 timeSec = int(time) % 60 | |
| 60 if timeHou > 0: | |
| 61 return "%3dh %2dm" % (timeHou, timeMin) | |
| 62 if timeMin > 0: | |
| 63 return "%2dm %2ds" % (timeMin, timeSec) | |
| 64 return "%2ds " % (timeSec) | |
| 65 | |
| 66 | |
| 67 def show(self): | |
| 68 if self.verbosity <= 0: | |
| 69 return | |
| 70 if self.aim == 0: | |
| 71 return | |
| 72 messageSize = len(self.message) | |
| 73 length = int(self.progress / float(self.aim) * self.barSize) | |
| 74 elapsed = int(time.time() - self.startTime) | |
| 75 if (length > self.length) or (elapsed > self.elapsed + 10): | |
| 76 self.length = length | |
| 77 self.elapsed = elapsed | |
| 78 string = "%s%s[%s%s] %d/%d" % (self.message, " " * max(0, self.maxMessageSize - messageSize), "=" * self.length, " " * (self.barSize - self.length), self.progress, self.aim) | |
| 79 if elapsed > 5: | |
| 80 done = float(self.progress) / self.aim | |
| 81 total = elapsed / done | |
| 82 remaining = total - elapsed | |
| 83 string += " ETA: %s " % (self.getPrintableElapsedTime(remaining)) | |
| 84 string += "\r" | |
| 85 sys.stdout.write(string) | |
| 86 sys.stdout.flush() | |
| 87 | |
| 88 | |
| 89 def done(self): | |
| 90 if self.verbosity > 0: | |
| 91 messageSize = len(self.message) | |
| 92 elapsed = time.time() - self.startTime | |
| 93 print "%s%s[%s] %d completed in %s " % (self.message, " " * max(0, self.maxMessageSize - messageSize), "=" * self.barSize, self.aim, self.getPrintableElapsedTime(elapsed)) |
