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

Rectangle selection functions now take bytes_per_row as required parameter

Browse Source
This commit is contained in:
Rob McMullen 2017-04-02 09:56:18 -07:00
parent b6a81a10b2
commit 510366051c
1 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
atrcopy/segments.py
View File

@ -691,7 +691,7 @@ class DefaultSegment(object):
return ranges[match_index][0] return ranges[match_index][0]
return None return None
def get_rect_indexes(self, anchor_start, anchor_end): def get_rect_indexes(self, anchor_start, anchor_end, bytes_per_row):
# 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)
# is 2 rows and 3 columns. Columns need to be adjusted slightly # is 2 rows and 3 columns. Columns need to be adjusted slightly
@ -699,9 +699,8 @@ class DefaultSegment(object):
# measured as cursor positions, that is: positions between the # measured as cursor positions, that is: positions between the
# bytes where as rect select needs to think of the selections as # bytes where as rect select needs to think of the selections as
# on the byte positions themselves, not in between. # on the byte positions themselves, not in between.
bpr = self.map_width r1, c1 = divmod(anchor_start, bytes_per_row)
r1, c1 = divmod(anchor_start, bpr) r2, c2 = divmod(anchor_end, bytes_per_row)
r2, c2 = divmod(anchor_end, bpr)
if c1 >= c2: if c1 >= c2:
# start column is to the right of the end column so columns # start column is to the right of the end column so columns
# need to be swapped # need to be swapped
@ -713,7 +712,7 @@ class DefaultSegment(object):
# When the cursor is at the end of a line, anchor_end points # When the cursor is at the end of a line, anchor_end points
# to the first character of the next line. Handle this # to the first character of the next line. Handle this
# special case by pointing to end of the previous line. # special case by pointing to end of the previous line.
c2 = bpr c2 = bytes_per_row
r2 -= 1 r2 -= 1
else: else:
c1, c2 = c2 - 1, c1 + 1 c1, c2 = c2 - 1, c1 + 1
@ -724,16 +723,16 @@ class DefaultSegment(object):
# start row is below end row # start row is below end row
r1, r2 = r2, r1 r1, r2 = r2, r1
c2 += 1 c2 += 1
anchor_start = r1 * bpr + c1 anchor_start = r1 * bytes_per_row + c1
anchor_end = r2 * bpr + c2 anchor_end = r2 * bytes_per_row + c2
r2 += 1 r2 += 1
return anchor_start, anchor_end, (r1, c1), (r2, c2) return anchor_start, anchor_end, (r1, c1), (r2, c2)
def set_style_ranges_rect(self, ranges, **kwargs): def set_style_ranges_rect(self, ranges, bytes_per_row, **kwargs):
style_bits = self.get_style_bits(**kwargs) style_bits = self.get_style_bits(**kwargs)
s = self.style s = self.style
for start, end in ranges: for start, end in ranges:
start, end, (r1, c1), (r2, c2) = self.get_rect_indexes(start, end) start, end, (r1, c1), (r2, c2) = self.get_rect_indexes(start, end, bytes_per_row)
# Numpy tricks! # Numpy tricks!
# >>> c1 = 15 # >>> c1 = 15
# >>> r = 4 # r2 - r1 # >>> r = 4 # r2 - r1
@ -753,15 +752,14 @@ class DefaultSegment(object):
# 101, 102, 103, 104, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144]) # 101, 102, 103, 104, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144])
r = r2 - r1 r = r2 - r1
c = c2 - c1 c = c2 - c1
indexes = np.tile(np.arange(c), r) + np.repeat(np.arange(r) * self.map_width, c) + start indexes = np.tile(np.arange(c), r) + np.repeat(np.arange(r) * bytes_per_row, c) + start
s[indexes] |= style_bits s[indexes] |= style_bits
def rects_to_ranges(self, rects): def rects_to_ranges(self, rects, bytes_per_row):
ranges = [] ranges = []
bpr = self.map_width
for (r1, c1), (r2, c2) in rects: for (r1, c1), (r2, c2) in rects:
start = r1 * bpr + c1 start = r1 * bytes_per_row + c1
end = (r2 - 1) * bpr + c2 end = (r2 - 1) * bytes_per_row + c2
ranges.append((start, end)) ranges.append((start, end))
return ranges return ranges