Benjamin Kramer | f61f22a | 2012-01-17 14:52:12 +0000 | [diff] [blame] | 1 | """ |
| 2 | Load into LLDB with: |
| 3 | script import lldbDataFormatters |
| 4 | type synthetic add -x "^llvm::SmallVectorImpl<.+>$" -l lldbDataFormatters.SmallVectorSynthProvider |
| 5 | """ |
| 6 | |
| 7 | # Pretty printer for llvm::SmallVector/llvm::SmallVectorImpl |
| 8 | class SmallVectorSynthProvider: |
| 9 | def __init__(self, valobj, dict): |
| 10 | self.valobj = valobj; |
| 11 | self.update() # initialize this provider |
| 12 | |
| 13 | def num_children(self): |
| 14 | begin = self.begin.GetValueAsUnsigned(0) |
| 15 | end = self.end.GetValueAsUnsigned(0) |
| 16 | return (end - begin)/self.type_size |
| 17 | |
| 18 | def get_child_index(self, name): |
| 19 | try: |
| 20 | return int(name.lstrip('[').rstrip(']')) |
| 21 | except: |
| 22 | return -1; |
| 23 | |
| 24 | def get_child_at_index(self, index): |
| 25 | # Do bounds checking. |
| 26 | if index < 0: |
| 27 | return None |
| 28 | if index >= self.num_children(): |
| 29 | return None; |
| 30 | |
| 31 | offset = index * self.type_size |
| 32 | return self.begin.CreateChildAtOffset('['+str(index)+']', |
| 33 | offset, self.data_type) |
| 34 | |
| 35 | def get_type_from_name(self): |
| 36 | import re |
| 37 | name = self.valobj.GetType().GetName() |
| 38 | # This class works with both SmallVectors and SmallVectorImpls. |
| 39 | res = re.match("^(llvm::)?SmallVectorImpl<(.+)>$", name) |
| 40 | if res: |
| 41 | return res.group(2) |
| 42 | res = re.match("^(llvm::)?SmallVector<(.+), \d+>$", name) |
| 43 | if res: |
| 44 | return res.group(2) |
| 45 | return None |
| 46 | |
| 47 | def update(self): |
| 48 | self.begin = self.valobj.GetChildMemberWithName('BeginX') |
| 49 | self.end = self.valobj.GetChildMemberWithName('EndX') |
| 50 | data_type = self.get_type_from_name() |
| 51 | # FIXME: this sometimes returns an invalid type. |
| 52 | self.data_type = self.valobj.GetTarget().FindFirstType(data_type) |
| 53 | self.type_size = self.data_type.GetByteSize() |