diff options
Diffstat (limited to 'lldb/scripts/Python/modify-python-lldb.py')
| -rw-r--r-- | lldb/scripts/Python/modify-python-lldb.py | 36 |
1 files changed, 26 insertions, 10 deletions
diff --git a/lldb/scripts/Python/modify-python-lldb.py b/lldb/scripts/Python/modify-python-lldb.py index 4026d8930a6..afa5494cb60 100644 --- a/lldb/scripts/Python/modify-python-lldb.py +++ b/lldb/scripts/Python/modify-python-lldb.py @@ -40,6 +40,10 @@ def lldb_iter(obj, getsize, getelem): iter_def = " def __iter__(self): return lldb_iter(self, '%s', '%s')" module_iter = " def module_iter(self): return lldb_iter(self, '%s', '%s')" breakpoint_iter = " def breakpoint_iter(self): return lldb_iter(self, '%s', '%s')" +# +# This supports the rich comparison methods of __eq__ and __ne__. +eq_def = " def __eq__(self, other): return isinstance(other, %s) and self.%s() == other.%s()" +ne_def = " def __ne__(self, other): return not self.__eq__(other)" # # The dictionary defines a mapping from classname to (getsize, getelem) tuple. @@ -64,6 +68,11 @@ d = { 'SBBreakpoint': ('GetNumLocations', 'GetLocationAtIndex'), } } +# +# This dictionary defines a mapping from classname to equality method name. +# +e = { 'SBBreakpoint': 'GetID' } + # The new content will have the iteration protocol defined for our lldb objects. new_content = StringIO.StringIO() @@ -79,7 +88,7 @@ init_pattern = re.compile("^ def __init__\(self, \*args\):") # These define the states of our state machine. NORMAL = 0 DEFINING_ITERATOR = 1 -DEFINING_TARGET_ITERATOR = 2 +DEFINING_EQUALITY = 2 # The lldb_iter_def only needs to be inserted once. lldb_iter_defined = False; @@ -91,25 +100,32 @@ for line in content.splitlines(): if not lldb_iter_defined and match: print >> new_content, lldb_iter_def lldb_iter_defined = True - if match and match.group(1) in d: - # Adding support for iteration for the matched SB class. + if match: cls = match.group(1) - # Next state will be DEFINING_ITERATOR. - state = DEFINING_ITERATOR - elif state == DEFINING_ITERATOR: + if cls in d: + # Adding support for iteration for the matched SB class. + state = (state | DEFINING_ITERATOR) + if cls in e: + # Adding support for eq and ne for the matched SB class. + state = (state | DEFINING_EQUALITY) + elif state > NORMAL: match = init_pattern.search(line) if match: # We found the beginning of the __init__ method definition. - # This is a good spot to insert the iteration support. + # This is a good spot to insert the iter and/or eq-ne support. # # But note that SBTarget has two types of iterations. if cls == "SBTarget": print >> new_content, module_iter % (d[cls]['module']) print >> new_content, breakpoint_iter % (d[cls]['breakpoint']) else: - print >> new_content, iter_def % d[cls] - # Next state will be NORMAL. - state = NORMAL + if (state & DEFINING_ITERATOR): + print >> new_content, iter_def % d[cls] + if (state & DEFINING_EQUALITY): + print >> new_content, eq_def % (cls, e[cls], e[cls]) + print >> new_content, ne_def + # Next state will be NORMAL. + state = NORMAL # Pass the original line of content to the ew_content. print >> new_content, line |
