Added function to return ranges of selected styles

This commit is contained in:
Rob McMullen 2016-02-12 22:53:58 -08:00
parent c30aaa451a
commit 42d2a977cc

View File

@ -77,6 +77,17 @@ class DefaultSegment(object):
start, end = end, start start, end = end, start
s[start:end] |= style_bits s[start:end] |= style_bits
def get_style_ranges(self, **kwargs):
style_bits = self.get_style_bits(**kwargs)
w = np.where(self.style & style_bits)[0]
# split into groups with consecutive numbers
groups = np.split(w, np.where(np.diff(w) != 1)[0] + 1)
ranges = []
for group in groups:
if np.alen(group) > 0:
ranges.append((int(group[0]), int(group[-1]) + 1))
return ranges
def get_rect_indexes(self, anchor_start, anchor_end): def get_rect_indexes(self, anchor_start, anchor_end):
# determine row,col of upper left and lower right of selected # determine row,col of upper left and lower right of selected
# rectangle. The values are inclusive, so ul=(0,0) and lr=(1,2) # rectangle. The values are inclusive, so ul=(0,0) and lr=(1,2)