← Back to branch summary

~webapps/unity-js-scopes/node.js

« back to all changes in this revision

Viewing changes to tools/gyp/tools/graphviz.py

  • Committer: Marcus Tomlinson
  • Date: 2015-11-13 07:59:04 UTC
  • Revision ID: marcus.tomlinson@canonical.com-20151113075904-h0swczmoq1rvstfc
Node v4 (stable)

Show diffs side-by-side

added added

removed removed

Lines of Context:
tools/gyp/tools/graphviz.py
 
1
#!/usr/bin/env python
 
2
 
 
3
# Copyright (c) 2011 Google Inc. All rights reserved.
 
4
# Use of this source code is governed by a BSD-style license that can be
 
5
# found in the LICENSE file.
 
6
 
 
7
"""Using the JSON dumped by the dump-dependency-json generator,
 
8
generate input suitable for graphviz to render a dependency graph of
 
9
targets."""
 
10
 
 
11
import collections
 
12
import json
 
13
import sys
 
14
 
 
15
 
 
16
def ParseTarget(target):
 
17
  target, _, suffix = target.partition('#')
 
18
  filename, _, target = target.partition(':')
 
19
  return filename, target, suffix
 
20
 
 
21
 
 
22
def LoadEdges(filename, targets):
 
23
  """Load the edges map from the dump file, and filter it to only
 
24
  show targets in |targets| and their depedendents."""
 
25
 
 
26
  file = open('dump.json')
 
27
  edges = json.load(file)
 
28
  file.close()
 
29
 
 
30
  # Copy out only the edges we're interested in from the full edge list.
 
31
  target_edges = {}
 
32
  to_visit = targets[:]
 
33
  while to_visit:
 
34
    src = to_visit.pop()
 
35
    if src in target_edges:
 
36
      continue
 
37
    target_edges[src] = edges[src]
 
38
    to_visit.extend(edges[src])
 
39
 
 
40
  return target_edges
 
41
 
 
42
 
 
43
def WriteGraph(edges):
 
44
  """Print a graphviz graph to stdout.
 
45
  |edges| is a map of target to a list of other targets it depends on."""
 
46
 
 
47
  # Bucket targets by file.
 
48
  files = collections.defaultdict(list)
 
49
  for src, dst in edges.items():
 
50
    build_file, target_name, toolset = ParseTarget(src)
 
51
    files[build_file].append(src)
 
52
 
 
53
  print 'digraph D {'
 
54
  print '  fontsize=8'  # Used by subgraphs.
 
55
  print '  node [fontsize=8]'
 
56
 
 
57
  # Output nodes by file.  We must first write out each node within
 
58
  # its file grouping before writing out any edges that may refer
 
59
  # to those nodes.
 
60
  for filename, targets in files.items():
 
61
    if len(targets) == 1:
 
62
      # If there's only one node for this file, simplify
 
63
      # the display by making it a box without an internal node.
 
64
      target = targets[0]
 
65
      build_file, target_name, toolset = ParseTarget(target)
 
66
      print '  "%s" [shape=box, label="%s\\n%s"]' % (target, filename,
 
67
                                                     target_name)
 
68
    else:
 
69
      # Group multiple nodes together in a subgraph.
 
70
      print '  subgraph "cluster_%s" {' % filename
 
71
      print '    label = "%s"' % filename
 
72
      for target in targets:
 
73
        build_file, target_name, toolset = ParseTarget(target)
 
74
        print '    "%s" [label="%s"]' % (target, target_name)
 
75
      print '  }'
 
76
 
 
77
  # Now that we've placed all the nodes within subgraphs, output all
 
78
  # the edges between nodes.
 
79
  for src, dsts in edges.items():
 
80
    for dst in dsts:
 
81
      print '  "%s" -> "%s"' % (src, dst)
 
82
 
 
83
  print '}'
 
84
 
 
85
 
 
86
def main():
 
87
  if len(sys.argv) < 2:
 
88
    print >>sys.stderr, __doc__
 
89
    print >>sys.stderr
 
90
    print >>sys.stderr, 'usage: %s target1 target2...' % (sys.argv[0])
 
91
    return 1
 
92
 
 
93
  edges = LoadEdges('dump.json', sys.argv[1:])
 
94
 
 
95
  WriteGraph(edges)
 
96
  return 0
 
97
 
 
98
 
 
99
if __name__ == '__main__':
 
100
  sys.exit(main())