10 class LLVMProjectInfo(object):
12 def load_infos_from_path(llvmbuild_source_root):
13 # FIXME: Implement a simple subpath file list cache, so we don't restat
14 # directories we have already traversed.
16 # First, discover all the LLVMBuild.txt files.
17 for dirpath,dirnames,filenames in os.walk(llvmbuild_source_root,
19 # If there is no LLVMBuild.txt file in a directory, we don't recurse
20 # past it. This is a simple way to prune our search, although it
21 # makes it easy for users to add LLVMBuild.txt files in places they
23 if 'LLVMBuild.txt' not in filenames:
27 # Otherwise, load the LLVMBuild file in this directory.
28 assert dirpath.startswith(llvmbuild_source_root)
29 subpath = '/' + dirpath[len(llvmbuild_source_root)+1:]
30 llvmbuild_path = os.path.join(dirpath, 'LLVMBuild.txt')
31 for info in componentinfo.load_from_path(llvmbuild_path, subpath):
35 def load_from_path(source_root, llvmbuild_source_root):
37 LLVMProjectInfo.load_infos_from_path(llvmbuild_source_root))
39 return LLVMProjectInfo(source_root, infos)
41 def __init__(self, source_root, component_infos):
42 # Store our simple ivars.
43 self.source_root = source_root
44 self.component_infos = component_infos
46 # Create the component info map and validate that component names are
48 self.component_info_map = {}
49 for ci in component_infos:
50 existing = self.component_info_map.get(ci.name)
51 if existing is not None:
52 # We found a duplicate component name, report it and error out.
53 fatal("found duplicate component %r (at %r and %r)" % (
54 ci.name, ci.subpath, existing.subpath))
55 self.component_info_map[ci.name] = ci
57 # Add the root component.
58 if '$ROOT' in self.component_info_map:
59 fatal("project is not allowed to define $ROOT component")
60 self.component_info_map['$ROOT'] = componentinfo.GroupComponentInfo(
62 self.component_infos.append(self.component_info_map['$ROOT'])
64 # Topologically order the component information according to their
65 # component references.
66 def visit_component_info(ci, current_stack, current_set):
69 # We found a cycle, report it and error out.
70 cycle_description = ' -> '.join(
71 '%r (%s)' % (ci.name, relation)
72 for relation,ci in current_stack)
73 fatal("found cycle to %r after following: %s -> %s" % (
74 ci.name, cycle_description, ci.name))
76 # If we have already visited this item, we are done.
77 if ci not in components_to_visit:
80 # Otherwise, mark the component info as visited and traverse.
81 components_to_visit.remove(ci)
83 # Validate the parent reference, which we treat specially.
84 if ci.parent is not None:
85 parent = self.component_info_map.get(ci.parent)
87 fatal("component %r has invalid reference %r (via %r)" % (
88 ci.name, ci.parent, 'parent'))
89 ci.set_parent_instance(parent)
91 for relation,referent_name in ci.get_component_references():
92 # Validate that the reference is ok.
93 referent = self.component_info_map.get(referent_name)
95 fatal("component %r has invalid reference %r (via %r)" % (
96 ci.name, referent_name, relation))
98 # Visit the reference.
99 current_stack.append((relation,ci))
101 visit_component_info(referent, current_stack, current_set)
102 current_set.remove(ci)
105 # Finally, add the component info to the ordered list.
106 self.ordered_component_infos.append(ci)
108 # FIXME: We aren't actually correctly checking for cycles along the
109 # parent edges. Haven't decided how I want to handle this -- I thought
110 # about only checking cycles by relation type. If we do that, it falls
111 # out easily. If we don't, we should special case the check.
113 self.ordered_component_infos = []
114 components_to_visit = set(component_infos)
115 while components_to_visit:
116 visit_component_info(iter(components_to_visit).next(), [], set())
118 # Canonicalize children lists.
119 for c in self.ordered_component_infos:
120 c.children.sort(key = lambda c: c.name)
122 def print_tree(self):
123 def visit(node, depth = 0):
124 print '%s%-40s (%s)' % (' '*depth, node.name, node.type_name)
125 for c in node.children:
127 visit(self.component_info_map['$ROOT'])
130 from optparse import OptionParser, OptionGroup
131 parser = OptionParser("usage: %prog [options]")
132 parser.add_option("", "--source-root", dest="source_root", metavar="PATH",
133 help="Path to the LLVM source (inferred if not given)",
134 action="store", default=None)
135 parser.add_option("", "--print-tree", dest="print_tree",
136 help="Print out the project component tree [%default]",
137 action="store_true", default=False)
139 "", "--llvmbuild-source-root", dest="llvmbuild_source_root",
140 help="If given, an alternate path to search for LLVMBuild.txt files",
141 action="store", default=None, metavar="PATH")
142 (opts, args) = parser.parse_args()
144 # Determine the LLVM source path, if not given.
145 source_root = opts.source_root
147 if not os.path.exists(os.path.join(source_root, 'lib', 'VMCore',
149 parser.error('invalid LLVM source root: %r' % source_root)
151 llvmbuild_path = os.path.dirname(__file__)
152 llvm_build_path = os.path.dirname(llvmbuild_path)
153 utils_path = os.path.dirname(llvm_build_path)
154 source_root = os.path.dirname(utils_path)
155 if not os.path.exists(os.path.join(source_root, 'lib', 'VMCore',
157 parser.error('unable to infer LLVM source root, please specify')
159 # Construct the LLVM project information.
160 llvmbuild_source_root = opts.llvmbuild_source_root or source_root
161 project_info = LLVMProjectInfo.load_from_path(
162 source_root, llvmbuild_source_root)
164 # Print the component tree, if requested.
166 project_info.print_tree()
168 if __name__=='__main__':