diff options
Diffstat (limited to 'scripts/oe-depends-dot')
-rwxr-xr-x | scripts/oe-depends-dot | 121 |
1 files changed, 121 insertions, 0 deletions
diff --git a/scripts/oe-depends-dot b/scripts/oe-depends-dot new file mode 100755 index 0000000000..5cec23bf0a --- /dev/null +++ b/scripts/oe-depends-dot | |||
@@ -0,0 +1,121 @@ | |||
1 | #!/usr/bin/env python3 | ||
2 | # | ||
3 | # Copyright (C) 2018 Wind River Systems, Inc. | ||
4 | # | ||
5 | # This program is free software; you can redistribute it and/or modify | ||
6 | # it under the terms of the GNU General Public License version 2 as | ||
7 | # published by the Free Software Foundation. | ||
8 | # | ||
9 | # This program is distributed in the hope that it will be useful, | ||
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. | ||
12 | # See the GNU General Public License for more details. | ||
13 | # | ||
14 | # You should have received a copy of the GNU General Public License | ||
15 | # along with this program; if not, write to the Free Software | ||
16 | # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | ||
17 | |||
18 | import os | ||
19 | import sys | ||
20 | import argparse | ||
21 | import logging | ||
22 | import re | ||
23 | |||
24 | class Dot(object): | ||
25 | def __init__(self): | ||
26 | parser = argparse.ArgumentParser( | ||
27 | description="Analyse recipe-depends.dot generated by bitbake -g", | ||
28 | epilog="Use %(prog)s --help to get help") | ||
29 | parser.add_argument("dotfile", | ||
30 | help = "Specify the dotfile", nargs = 1, action='store', default='') | ||
31 | parser.add_argument("-k", "--key", | ||
32 | help = "Specify the key, e.g., recipe name", | ||
33 | action="store", default='') | ||
34 | parser.add_argument("-d", "--depends", | ||
35 | help = "Print the key's dependencies", | ||
36 | action="store_true", default=False) | ||
37 | parser.add_argument("-w", "--why", | ||
38 | help = "Print why the key is built", | ||
39 | action="store_true", default=False) | ||
40 | parser.add_argument("-r", "--remove", | ||
41 | help = "Remove duplicated dependencies to reduce the size of the dot files." | ||
42 | " For example, A->B, B->C, A->C, then A->C can be removed.", | ||
43 | action="store_true", default=False) | ||
44 | |||
45 | self.args = parser.parse_args() | ||
46 | |||
47 | if len(sys.argv) != 3 and len(sys.argv) < 5: | ||
48 | print('ERROR: Not enough args, see --help for usage') | ||
49 | |||
50 | def main(self): | ||
51 | #print(self.args.dotfile[0]) | ||
52 | # The format is {key: depends} | ||
53 | depends = {} | ||
54 | with open(self.args.dotfile[0], 'r') as f: | ||
55 | for line in f.readlines(): | ||
56 | if ' -> ' not in line: | ||
57 | continue | ||
58 | line_no_quotes = line.replace('"', '') | ||
59 | m = re.match("(.*) -> (.*)", line_no_quotes) | ||
60 | if not m: | ||
61 | print('WARNING: Found unexpected line: %s' % line) | ||
62 | continue | ||
63 | key = m.group(1) | ||
64 | if key == "meta-world-pkgdata": | ||
65 | continue | ||
66 | dep = m.group(2) | ||
67 | if key in depends: | ||
68 | if not key in depends[key]: | ||
69 | depends[key].add(dep) | ||
70 | else: | ||
71 | print('WARNING: Fonud duplicated line: %s' % line) | ||
72 | else: | ||
73 | depends[key] = set() | ||
74 | depends[key].add(dep) | ||
75 | |||
76 | if self.args.remove: | ||
77 | reduced_depends = {} | ||
78 | for k, deps in depends.items(): | ||
79 | child_deps = set() | ||
80 | added = set() | ||
81 | # Both direct and indirect depends are already in the dict, so | ||
82 | # we don't have to do this recursively. | ||
83 | for dep in deps: | ||
84 | if dep in depends: | ||
85 | child_deps |= depends[dep] | ||
86 | |||
87 | reduced_depends[k] = deps - child_deps | ||
88 | outfile= '%s-reduced%s' % (self.args.dotfile[0][:-4], self.args.dotfile[0][-4:]) | ||
89 | with open(outfile, 'w') as f: | ||
90 | print('Saving reduced dot file to %s' % outfile) | ||
91 | f.write('digraph depends {\n') | ||
92 | for k, v in reduced_depends.items(): | ||
93 | for dep in v: | ||
94 | f.write('"%s" -> "%s"\n' % (k, dep)) | ||
95 | f.write('}\n') | ||
96 | sys.exit(0) | ||
97 | |||
98 | if self.args.key not in depends: | ||
99 | print("ERROR: Can't find key %s in %s" % (self.args.key, self.args.dotfile[0])) | ||
100 | sys.exit(1) | ||
101 | |||
102 | if self.args.depends: | ||
103 | if self.args.key in depends: | ||
104 | print('Depends: %s' % ' '.join(depends[self.args.key])) | ||
105 | |||
106 | reverse_deps = [] | ||
107 | if self.args.why: | ||
108 | for k, v in depends.items(): | ||
109 | if self.args.key in v and not k in reverse_deps: | ||
110 | reverse_deps.append(k) | ||
111 | print('Because: %s' % ' '.join(reverse_deps)) | ||
112 | |||
113 | if __name__ == "__main__": | ||
114 | try: | ||
115 | dot = Dot() | ||
116 | ret = dot.main() | ||
117 | except Exception as esc: | ||
118 | ret = 1 | ||
119 | import traceback | ||
120 | traceback.print_exc() | ||
121 | sys.exit(ret) | ||