xref: /openbsd-src/gnu/llvm/llvm/utils/chunk-print-before-all.py (revision 46035553bfdd96e63c94e32da0210227ec2e3cf1)
1#!/usr/bin/env python
2# Given a -print-before-all -print-module-scope log from an opt invocation,
3# chunk it into a series of individual IR files, one for each pass invocation.
4# If the log ends with an obvious stack trace, try to split off a separate
5# "crashinfo.txt" file leaving only the valid input IR in the last chunk.
6# Files are written to current working directory.
7
8import sys
9
10basename = "chunk-"
11chunk_id = 0
12
13def print_chunk(lines):
14    global chunk_id
15    global basename
16    fname = basename + str(chunk_id) + ".ll"
17    chunk_id = chunk_id + 1
18    print "writing chunk " + fname + " (" + str(len(lines)) + " lines)"
19    with open(fname, "w") as f:
20        f.writelines(lines)
21
22is_dump = False
23cur = []
24for line in sys.stdin:
25    if line.startswith("*** IR Dump Before ") and len(cur) != 0:
26        print_chunk(cur);
27        cur = []
28        cur.append("; " + line)
29    elif line.startswith("Stack dump:"):
30        print_chunk(cur);
31        cur = []
32        cur.append(line)
33        is_dump = True
34    else:
35        cur.append(line)
36
37if is_dump:
38    print "writing crashinfo.txt (" + str(len(cur)) + " lines)"
39    with open("crashinfo.txt", "w") as f:
40        f.writelines(cur)
41else:
42    print_chunk(cur);
43