support manual outlier filtering

1 files changed, 4 insertions, 1 deletions

diff --git a/binary_data.py b/binary_data.py
index 152afb7..39d6e05 100644
--- a/binary_data.py
+++ b/binary_data.py
@@ -5,7 +5,7 @@ from stats import iqr_remove_outliers, iqr_cutoff
 
 
 def get_data(fname, scale, extent, cutoff=None, maxval=1000.0,
-             stdev=False):
+             stdev=False, manual=None):
     data = load_binary_file(fname)
 
     if cutoff and len(data) > cutoff:
@@ -31,6 +31,9 @@ def get_data(fname, scale, extent, cutoff=None, maxval=1000.0,
 
     min_idx, max_idx = numpy.searchsorted(data, [lower, upper])
 
+    if manual:
+        max_idx -= manual
+
     return [data, max_idx, min_idx, upper, lower]