xref: /llvm-project/clang/utils/analyze_safe_buffer_debug_notes.py (revision a4323586fcbb20f39f00d5d1bc4a94a1aeea15c4) 
1import sys
2from collections import OrderedDict
3
4class Trie:
5    def __init__(self, name):
6        self.name = name
7        self.children = OrderedDict()
8        self.count = 1
9
10    def add(self, name):
11        if name in self.children:
12            self.children[name].count += 1
13        else:
14            self.children[name] = Trie(name)
15        return self.children[name]
16
17    def print(self, depth):
18        if depth > 0:
19            print('|', end="")
20        for i in range(depth):
21            print('-', end="")
22        if depth > 0:
23            print(end=" ")
24        print(self.name, '#', self.count)
25        for key, child in self.children.items():
26            child.print(depth + 1)
27
28
29Root = Trie("Root")
30
31if __name__ == "__main__":
32    for line in sys.stdin:
33        words = line.split('==>')
34        words = [word.strip() for word in words]
35        MyTrie = Root;
36        for word in words:
37            MyTrie = MyTrie.add(word)
38
39    Root.print(0)
40