Briq
/
camelot-py
6
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add negative tolerance

pull/2/head
Vinayak Mehta 2016-08-31 22:25:33 +05:30 committed by GitHub
parent 2a55621d05
commit 8d56f15130
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
camelot/stream.py
View File

@ -56,7 +56,7 @@ def _group_rows(text, ytol=2):
return rows return rows
def _merge_columns(l, mtol=2): def _merge_columns(l, mtol=0):
"""Merges overlapping columns and returns list with updated """Merges overlapping columns and returns list with updated
columns boundaries. columns boundaries.
@ -76,6 +76,7 @@ def _merge_columns(l, mtol=2):
merged.append(higher) merged.append(higher)
else: else:
lower = merged[-1] lower = merged[-1]
if mtol > 0:
if (higher[0] <= lower[1] or if (higher[0] <= lower[1] or
np.isclose(higher[0], lower[1], atol=mtol)): np.isclose(higher[0], lower[1], atol=mtol)):
upper_bound = max(lower[1], higher[1]) upper_bound = max(lower[1], higher[1])
@ -83,6 +84,16 @@ def _merge_columns(l, mtol=2):
merged[-1] = (lower_bound, upper_bound) merged[-1] = (lower_bound, upper_bound)
else: else:
merged.append(higher) merged.append(higher)
elif mtol < 0:
if higher[0] <= lower[1]:
if np.isclose(higher[0], lower[1], atol=abs(mtol)):
merged.append(higher)
else:
upper_bound = max(lower[1], higher[1])
lower_bound = min(lower[0], higher[0])
merged[-1] = (lower_bound, upper_bound)
else:
merged.append(higher)
return merged return merged