1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
#!/usr/bin/python
# Copyright 2011 Google Inc. All Rights Reserved.
"""The class to show the banner."""
import datetime
import time
class ExperimentStatus(object):
"""The status class."""
def __init__(self, experiment):
self.experiment = experiment
self.num_total = len(self.experiment.benchmark_runs)
self.completed = 0
self.new_job_start_time = time.time()
def _GetProgressBar(self, num_complete, num_total):
ret = "Done: %s%%" % int(100.0 * num_complete / num_total)
bar_length = 50
done_char = ">"
undone_char = " "
num_complete_chars = bar_length * num_complete / num_total
num_undone_chars = bar_length - num_complete_chars
ret += " [%s%s]" % (num_complete_chars * done_char, num_undone_chars *
undone_char)
return ret
def GetProgressString(self):
"""Get the elapsed_time, ETA."""
current_time = time.time()
if self.experiment.start_time:
elapsed_time = current_time - self.experiment.start_time
else:
elapsed_time = 0
try:
if self.completed != self.experiment.num_complete:
self.completed = self.experiment.num_complete
self.new_job_start_time = current_time
time_completed_jobs = (elapsed_time -
(current_time - self.new_job_start_time))
eta_seconds = (float(self.num_total - self.experiment.num_complete -1) *
time_completed_jobs / self.experiment.num_run_complete
+ (time_completed_jobs / self.experiment.num_run_complete
- (current_time - self.new_job_start_time)))
eta_seconds = int(eta_seconds)
eta = datetime.timedelta(seconds=eta_seconds)
except ZeroDivisionError:
eta = "Unknown"
strings = []
strings.append("Current time: %s Elapsed: %s ETA: %s" %
(datetime.datetime.now(),
datetime.timedelta(seconds=int(elapsed_time)),
eta))
strings.append(self._GetProgressBar(self.experiment.num_complete,
self.num_total))
return "\n".join(strings)
def GetStatusString(self):
"""Get the status string of all the benchmark_runs."""
status_bins = {}
for benchmark_run in self.experiment.benchmark_runs:
if benchmark_run.timeline.GetLastEvent() not in status_bins:
status_bins[benchmark_run.timeline.GetLastEvent()] = []
status_bins[benchmark_run.timeline.GetLastEvent()].append(benchmark_run)
status_strings = []
for key, val in status_bins.items():
status_strings.append("%s: %s" %
(key, self._GetNamesAndIterations(val)))
result = "Thread Status:\n%s" % "\n".join(status_strings)
# Add the machine manager status.
result += "\n" + self.experiment.machine_manager.AsString() + "\n"
return result
def _GetNamesAndIterations(self, benchmark_runs):
strings = []
t = time.time()
for benchmark_run in benchmark_runs:
t_last = benchmark_run.timeline.GetLastEventTime()
elapsed = str(datetime.timedelta(seconds=int(t-t_last)))
strings.append("'{0}' {1}".format(benchmark_run.name, elapsed))
return " %s (%s)" % (len(strings), ", ".join(strings))
|
