replica
/
hanchenye-llvm-project
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add the Python rich comparison methods for SBBreakpoint, where GetID() returns 

the breakpoint ID and provides the semantics needed for '==' and '!='.  And
modify LLDBIteratorTestCase.lldb_iter_2() to use '==' between two SBBreakpoint's.

llvm-svn: 130531
This commit is contained in:
Johnny Chen 2011-04-29 19:03:02 +00:00
parent 080d98d3b2
commit 0f023fa4af
2 changed files with 27 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
lldb/scripts/Python/modify-python-lldb.py
View File

@ -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

2
lldb/test/python_api/lldbutil/TestLLDBIterator.py
View File

@ -92,7 +92,7 @@ class LLDBIteratorTestCase(TestBase):
if self.TraceOn():
print "yours[%d]='%s'" % (i, get_description(yours[i]))
print "mine[%d]='%s'" % (i, get_description(mine[i]))
self.assertTrue(yours[i].GetID() == mine[i].GetID(),
self.assertTrue(yours[i] == mine[i],
"ID of yours[{0}] and mine[{0}] matches".format(i))
def lldb_iter_3(self):