paul@0 | 1 | |
paul@0 | 2 | def flatten(tup): |
paul@0 | 3 | elts = [] |
paul@0 | 4 | for elt in tup: |
paul@0 | 5 | if isinstance(elt, tuple): |
paul@0 | 6 | elts = elts + flatten(elt) |
paul@0 | 7 | else: |
paul@0 | 8 | elts.append(elt) |
paul@0 | 9 | return elts |
paul@0 | 10 | |
paul@0 | 11 | class Set: |
paul@0 | 12 | def __init__(self): |
paul@0 | 13 | self.elts = {} |
paul@0 | 14 | def __len__(self): |
paul@0 | 15 | return len(self.elts) |
paul@0 | 16 | def __contains__(self, elt): |
paul@0 | 17 | return elt in self.elts |
paul@0 | 18 | def add(self, elt): |
paul@0 | 19 | self.elts[elt] = elt |
paul@0 | 20 | def elements(self): |
paul@0 | 21 | return self.elts.keys() |
paul@0 | 22 | def has_elt(self, elt): |
paul@0 | 23 | return elt in self.elts |
paul@0 | 24 | def remove(self, elt): |
paul@0 | 25 | del self.elts[elt] |
paul@0 | 26 | def copy(self): |
paul@0 | 27 | c = Set() |
paul@0 | 28 | c.elts.update(self.elts) |
paul@0 | 29 | return c |
paul@0 | 30 | |
paul@0 | 31 | class Stack: |
paul@0 | 32 | def __init__(self): |
paul@0 | 33 | self.stack = [] |
paul@0 | 34 | self.pop = self.stack.pop |
paul@0 | 35 | def __len__(self): |
paul@0 | 36 | return len(self.stack) |
paul@0 | 37 | def push(self, elt): |
paul@0 | 38 | self.stack.append(elt) |
paul@0 | 39 | def top(self): |
paul@0 | 40 | return self.stack[-1] |
paul@0 | 41 | def __getitem__(self, index): # needed by visitContinue() |
paul@0 | 42 | return self.stack[index] |
paul@0 | 43 | |
paul@0 | 44 | MANGLE_LEN = 256 # magic constant from compile.c |
paul@0 | 45 | |
paul@0 | 46 | def mangle(name, klass): |
paul@0 | 47 | if not name.startswith('__'): |
paul@0 | 48 | return name |
paul@0 | 49 | if len(name) + 2 >= MANGLE_LEN: |
paul@0 | 50 | return name |
paul@0 | 51 | if name.endswith('__'): |
paul@0 | 52 | return name |
paul@0 | 53 | try: |
paul@0 | 54 | i = 0 |
paul@0 | 55 | while klass[i] == '_': |
paul@0 | 56 | i = i + 1 |
paul@0 | 57 | except IndexError: |
paul@0 | 58 | return name |
paul@0 | 59 | klass = klass[i:] |
paul@0 | 60 | |
paul@0 | 61 | tlen = len(klass) + len(name) |
paul@0 | 62 | if tlen > MANGLE_LEN: |
paul@0 | 63 | klass = klass[:MANGLE_LEN-tlen] |
paul@0 | 64 | |
paul@0 | 65 | return "_%s%s" % (klass, name) |
paul@0 | 66 | |
paul@0 | 67 | def set_filename(filename, tree): |
paul@0 | 68 | """Set the filename attribute to filename on every node in tree""" |
paul@0 | 69 | worklist = [tree] |
paul@0 | 70 | while worklist: |
paul@0 | 71 | node = worklist.pop(0) |
paul@0 | 72 | node.filename = filename |
paul@0 | 73 | worklist.extend(node.getChildNodes()) |