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 convenienc functions to get style, comments at list of indexes

Browse Source
This commit is contained in:
Rob McMullen 2017-02-09 14:20:03 -08:00
parent 10e89f6730
commit 9cf61e5a82
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
atrcopy/segments.py
View File

@ -608,6 +608,20 @@ class DefaultSegment(object):
ranges.append([[start, end], current]) ranges.append([[start, end], current])
return ranges return ranges
def get_style_at_indexes(self, indexes):
return self.style[indexes]
def get_comments_at_indexes(self, indexes):
s = self.style[indexes]
has_comments = np.where(s & comment_bit_mask > 0)[0]
print "has_comments", has_comments
comments = []
for where_index in has_comments:
raw = self.get_raw_index(indexes[where_index])
comment = self.get_comment(indexes[where_index])
comments.append((where_index, comment))
return comments
def set_comment(self, ranges, text): def set_comment(self, ranges, text):
self.set_style_ranges(ranges, comment=True) self.set_style_ranges(ranges, comment=True)
for start, end in ranges: for start, end in ranges: