ag/pixelocr
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Move _iter_lines() method from SubImage to Image.

Browse source
This commit is contained in:
Andrey Golovizin 2014-08-10 21:40:10 +02:00
parent 0ae6f19e60
commit 2450f7f002
1 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

34
pixelocr/image.py
View file

@ -151,6 +151,23 @@ class Image(object):
bottom_margin = _get_margin_height(reversed(self.bitmap)) bottom_margin = _get_margin_height(reversed(self.bitmap))
return self[top_margin:self.height - bottom_margin, :] return self[top_margin:self.height - bottom_margin, :]
def _iter_lines(self, min_space, T=False):
line_start = None
prev_line_end = 0
for i, row in enumerate(self.bitmap):
if _is_nonblank(row):
if line_start is None:
line_start = i
height = line_start - prev_line_end
if height >= min_space:
yield self[prev_line_end:line_start]
else:
if line_start is not None:
yield self[line_start:i,:]
line_start = None
prev_line_end = i
class SubImage(Image): class SubImage(Image):
def __init__(self, parent, y1, y2, x1, x2): def __init__(self, parent, y1, y2, x1, x2):
@ -171,20 +188,3 @@ class SubImage(Image):
@property @property
def T(self): def T(self):
return type(self)(self.parent.T, self.x1, self.x2, self.y1, self.y2) return type(self)(self.parent.T, self.x1, self.x2, self.y1, self.y2)
def _iter_lines(self, min_space, T=False):
line_start = None
prev_line_end = 0
for i, row in enumerate(self.bitmap):
if _is_nonblank(row):
if line_start is None:
line_start = i
height = line_start - prev_line_end
if height >= min_space:
yield self[prev_line_end:line_start]
else:
if line_start is not None:
yield self[line_start:i,:]
line_start = None
prev_line_end = i