summaryrefslogtreecommitdiff
path: root/source4/script/depfilter.py
blob: 74d1d179c722700b02cf6ff43b5fdcd4230e20a9 (plain)
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
#!/usr/bin/python
#
# Filter out arcs in a dotty graph that are at or below a certain
# node.  This is useful for visualising parts of the dependency graph.
#

# Command line stuff

import sys, sre

if len(sys.argv) != 2:
    print 'Usage: depfilter.py NODE'
    sys.exit(1)

top = sys.argv[1]

# Read in dot file

lines = sys.stdin.readlines()

graph = {}

for arc in lines[1:-1]:
    match = sre.search('"(.*)" -> "(.*)"', arc)
    n1, n2 = match.group(1), match.group(2)
    if not graph.has_key(n1):
        graph[n1] = []
    graph[n1].append(n2)

# Create subset of 'graph' rooted at 'top'

subgraph = {}

def add_deps(node):
    if graph.has_key(node) and not subgraph.has_key(node):
        subgraph[node] = graph[node]
        for n in graph[node]:
            add_deps(n)

add_deps(top)

# Generate output

print lines[0],

for key, value in subgraph.items():
    for n in value:
        print '\t"%s" -> "%s"' % (key, n)

print lines[-1],