Apple-2-Tools/atrcopy
1
0
Fork 0
mirror of https://github.com/robmcmullen/atrcopy.git synced 2024-11-26 08:49:50 +00:00
Code Issues Projects Releases Wiki Activity

Added iterator over comments in a segment

Browse Source
This commit is contained in:
Rob McMullen 2017-02-07 19:14:02 -08:00
parent 1f302f8f5b
commit a609bf10db
1 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
atrcopy/segments.py
View File

@ -209,12 +209,19 @@ class SegmentData(object):
Raises IndexError if the base index doesn't map to anything in this
segment's data
"""
if not self.reverse_index_mapping:
self.reverse_index_mapping = dict([(k,i) for i,k in enumerate(self.order)])
try:
return self.reverse_index_mapping[base_index]
except KeyError:
raise IndexError("index %d not mapped in this segment" % base_index)
if self.is_indexed:
if not self.reverse_index_mapping:
self.reverse_index_mapping = dict([(k,i) for i,k in enumerate(self.order)])
try:
return self.reverse_index_mapping[base_index]
except KeyError:
raise IndexError("index %d not mapped in this segment" % base_index)
else:
if self.data.base is None:
return int(base_index)
data_start, data_end = np.byte_bounds(self.data)
base_start, base_end = np.byte_bounds(self.data.base)
return int(base_start + base_index - data_start)
class DefaultSegment(object):
@ -625,6 +632,14 @@ class DefaultSegment(object):
def get_sorted_comments(self):
return sorted([[k, v] for k, v in self.rawdata.extra.comments.iteritems()])
def iter_comments_in_segment(self):
start = self.start_addr
start_index = self.get_raw_index(0)
end_index = self.get_raw_index(len(self.rawdata))
for k, v in self.rawdata.extra.comments.iteritems():
if k >= start_index and k < end_index:
yield self.rawdata.get_reverse_index(k), v
def label(self, index, lower_case=True):
if lower_case: