Jeremy Hylton | 65d4ea0 | 2000-02-16 00:49:47 +0000 | [diff] [blame] | 1 | import types |
| 2 | |
| 3 | def flatten(tup): |
| 4 | elts = [] |
| 5 | for elt in tup: |
| 6 | if type(elt) == types.TupleType: |
| 7 | elts = elts + flatten(elt) |
| 8 | else: |
| 9 | elts.append(elt) |
| 10 | return elts |
| 11 | |
Jeremy Hylton | 8b6323d | 2000-02-04 00:28:21 +0000 | [diff] [blame] | 12 | class Set: |
| 13 | def __init__(self): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 14 | self.elts = {} |
Jeremy Hylton | 95b8614 | 2000-02-10 20:54:27 +0000 | [diff] [blame] | 15 | def __len__(self): |
| 16 | return len(self.elts) |
Jeremy Hylton | 314e3fb | 2000-11-06 03:43:11 +0000 | [diff] [blame] | 17 | def __contains__(self, elt): |
| 18 | return self.elts.has_key(elt) |
Jeremy Hylton | 8b6323d | 2000-02-04 00:28:21 +0000 | [diff] [blame] | 19 | def add(self, elt): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 20 | self.elts[elt] = elt |
Jeremy Hylton | e4d6293 | 2000-03-16 20:02:38 +0000 | [diff] [blame] | 21 | def elements(self): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 22 | return self.elts.keys() |
Jeremy Hylton | 8b6323d | 2000-02-04 00:28:21 +0000 | [diff] [blame] | 23 | def has_elt(self, elt): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 24 | return self.elts.has_key(elt) |
Jeremy Hylton | 69926ea | 2000-02-10 00:43:22 +0000 | [diff] [blame] | 25 | def remove(self, elt): |
| 26 | del self.elts[elt] |
Jeremy Hylton | 314e3fb | 2000-11-06 03:43:11 +0000 | [diff] [blame] | 27 | def copy(self): |
| 28 | c = Set() |
| 29 | c.elts.update(self.elts) |
| 30 | return c |
Jeremy Hylton | 8b6323d | 2000-02-04 00:28:21 +0000 | [diff] [blame] | 31 | |
| 32 | class Stack: |
| 33 | def __init__(self): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 34 | self.stack = [] |
| 35 | self.pop = self.stack.pop |
Jeremy Hylton | 95b8614 | 2000-02-10 20:54:27 +0000 | [diff] [blame] | 36 | def __len__(self): |
| 37 | return len(self.stack) |
Jeremy Hylton | 8b6323d | 2000-02-04 00:28:21 +0000 | [diff] [blame] | 38 | def push(self, elt): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 39 | self.stack.append(elt) |
Jeremy Hylton | 8b6323d | 2000-02-04 00:28:21 +0000 | [diff] [blame] | 40 | def top(self): |
Jeremy Hylton | 772dd41 | 2000-02-21 22:46:00 +0000 | [diff] [blame] | 41 | return self.stack[-1] |