Apple-2-Tools/atrcopy
1
0
Fork 0
mirror of https://github.com/robmcmullen/atrcopy.git synced 2024-06-17 21:30:07 +00:00
Code Issues Projects Releases Wiki Activity

Added text comments

Browse Source
This commit is contained in:
Rob McMullen 2016-04-05 22:55:08 -07:00
parent d8236e4edd
commit 9835c0521f
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
atrcopy/segments.py
View File

@ -116,6 +116,16 @@ class DefaultSegment(object):
data_start, data_end = np.byte_bounds(self.data)
base_start, base_end = np.byte_bounds(self.data.base)
return int(data_start - base_start), int(data_end - base_start)
def get_raw_index(self, i):
"""Get index into base array's raw data, given the index into this
segment
"""
if self.data.base is None:
return i
data_start, data_end = np.byte_bounds(self.data)
base_start, base_end = np.byte_bounds(self.data.base)
return int(data_start - base_start + i)
def tostring(self):
return self.data.tostring()
@ -253,6 +263,16 @@ class DefaultSegment(object):
style_mask = self.get_style_mask(**kwargs)
self.style &= style_mask
def set_comment(self, ranges, text):
self.set_style_ranges(ranges, comment=True)
for start, end in ranges:
rawindex = self.get_raw_index(start)
self.rawdata.comments[rawindex] = text
def get_comment(self, index):
rawindex = self.get_raw_index(index)
return self.rawdata.comments.get(rawindex, None)
def label(self, index, lower_case=True):
if lower_case:
return "%04x" % (index + self.start_addr)
@ -426,5 +446,12 @@ class IndexedByteSegment(DefaultSegment):
b = DefaultSegment.byte_bounds_offset(self)
return (b[0] + self.order[0], b[0] + self.order[-1])
def get_raw_index(self, i):
if self.data.base is None:
return self.order[i]
data_start, data_end = np.byte_bounds(self.data)
base_start, base_end = np.byte_bounds(self.data.base)
return int(data_start - base_start + self.order[i])
def tostring(self):
return self.data[self.order[:]].tostring()