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
|
from __future__ import print_function
class LogEntry(object):
def __init__(self):
self.slots = 0
self.objects = 0
self.element_typenames = {}
def add(self, size, element_typename):
self.slots += size
self.objects += 1
if element_typename:
self.element_typenames[element_typename] = None
def classnames(self):
return self.element_typenames.keys()
class Logger(object):
_attrs_ = ["active", "aggregate", "logs"]
_immutable_fields_ = ["active?", "aggregate?", "logs"]
def __init__(self):
self.active = False
self.aggregate = False
self.logs = {}
def activate(self, aggregate=False):
self.active = True
self.aggregate = self.aggregate or aggregate
def log(self, new_strategy, size, cause="", old_strategy="", typename="", element_typename=""):
if self.aggregate:
key = (cause, old_strategy, new_strategy, typename)
if key not in self.logs:
self.logs[key] = LogEntry()
entry = self.logs[key]
entry.add(size, element_typename)
else:
element_typenames = [ element_typename ] if element_typename else []
self.output(cause, old_strategy, new_strategy, typename, size, 1, element_typenames)
def print_aggregated_log(self):
if not self.aggregate:
return
for key, entry in self.logs.items():
cause, old_strategy, new_strategy, typename = key
slots, objects, element_typenames = entry.slots, entry.objects, entry.classnames()
self.output(cause, old_strategy, new_strategy, typename, slots, objects, element_typenames)
def output(self, cause, old_strategy, new_strategy, typename, slots, objects, element_typenames):
old_strategy_string = "%s -> " % old_strategy if old_strategy else ""
classname_string = " of %s" % typename if typename else ""
element_string = (" elements: " + " ".join(element_typenames)) if element_typenames else ""
format = (cause, old_strategy_string, new_strategy, classname_string, slots, objects, element_string)
self.do_print("%s (%s%s)%s size %d objects %d%s" % format)
def do_print(self, str):
# Hook to increase testability
print(str)
|
