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
|
#!/usr/bin/python3
'''
Reads memento squeeze output from stdin, and writes brief info to stdout or
file.
E.g.:
MEMENTO_SQUEEZEAT=1 ./membin/gpdl -sDEVICE=bit -o /dev/null examples/tiger.eps 2>&1 | toolbin/squeeze2text.py -o squeeze.txt
Args:
-a <filename>
Append output to <filename>. Write log to stdout.
-o <filename>
Write output to <filename>. Write log to stdout.
-p <N>
Write progress information to log, every <N> blocks.
If -o is not specified, we write output to stdout, and log to stderr.
If -p is not specified, default is -p 1.
'''
import io
import re
import sys
if __name__ == '__main__':
verbose = False
progress_n = 1
num_segv = 0
num_leak = 0
out = sys.stdout
log = sys.stderr
args = iter( sys.argv[1:])
while 1:
try:
arg = next(args)
except StopIteration:
break
if arg == '-p':
progress_n = int( next(args))
elif arg == '-a':
filename = next(args)
out = open( filename, 'a')
log = sys.stdout
elif arg == '-o':
filename = next(args)
out = open( filename, 'w')
log = sys.stdout
else:
raise Exception( 'unrecognised arg: %r' % arg)
# Use latin_1 encoding to ensure we can read all 8-bit bytes from stdin as
# characters.
f = io.TextIOWrapper( sys.stdin.buffer, encoding='latin_1')
progress_n_next = 0
memento_n = None
def print_progress():
print( 'memento_n=%s. num_segv=%s num_leak=%s' % (
memento_n,
num_segv,
num_leak
),
file=log,
)
for line in f:
m = re.match( '^Memory squeezing @ ([0-9]+)$', line)
if m:
memento_n = int( m.group(1))
if memento_n >= progress_n_next:
print_progress()
progress_n_next = int(memento_n / progress_n + 1) * progress_n
log.flush() # Otherwise buffered and we see no output for ages.
elif line.startswith( 'SEGV at:'):
num_segv += 1
print( 'memento_n=%s: segv' % memento_n, file=out)
elif line.startswith( 'Allocated blocks'):
num_leak += 1
print( 'memento_n=%s: leak' % memento_n, file=out)
print_progress()
|
