blob: 4cee6d67485c77552752ff6d19604c26b94861f2 (
plain) (
tree)
|
|
include prelude
proc main(input, output: string) =
type NodeKind = enum
local, localInvalid, global, globalInvalid
#c_fprintf(file, "%s %p %d rc=%ld color=%c\n",
# msg, c, kind, c.refcount shr rcShift, col)
# cell 0x10a908190 22 rc=2 color=w
var i, o: File
var roots = initTable[string, NodeKind]()
if open(i, input):
if open(o, output, fmWrite):
o.writeLine("digraph $1 {\n" % extractFilename(input))
var currNode = ""
for line in lines(i):
let data = line.split()
if data.len == 0: continue
case data[0]
of "end":
currNode = ""
of "cell":
currNode = data[1]
let rc = data[3].substr("rc=".len)
let col = case data[4].substr("color=".len)
of "b": "black"
of "w": "green"
of "g": "grey"
else: ""
o.write("N" & currNode)
if currNode in roots:
let v = roots[currNode]
case v
of local: o.write(" [label=\"local \\N\" fillcolor=$1]" % col)
of localInvalid: o.write(" [label=\"local invalid \\N\" fillcolor=$1]" % col)
of global: o.write(" [label=\"global \\N\" fillcolor=$1]" % col)
of globalInvalid: o.write(" [label=\"global invalid \\N\" fillcolor=$1]" % col)
else:
o.write(" [fillcolor=$1]" % col)
o.writeLine(";")
of "child":
assert currNode.len > 0
o.writeLine("N$1 -> N$2;" % [currNode, data[1]])
of "global_root":
roots[data[1]] = global
of "global_root_invalid":
roots[data[1]] = globalInvalid
of "onstack":
roots[data[1]] = local
of "onstack_invalid":
roots[data[1]] = localInvalid
else: discard
close(i)
o.writeLine("\n}")
close(o)
else:
quit "error: cannot open " & output
else:
quit "error: cannot open " & input
if paramCount() == 1:
main(paramStr(1), changeFileExt(paramStr(1), "dot"))
elif paramCount() == 2:
main(paramStr(1), paramStr(2))
else:
quit "usage: heapdump2dot inputfile outputfile"
|