Merge pull request #104 from jiangyi15/2d_pull

feat: plot 2d pull directly
jiangyi15 · Aug 26, 2023 · 5a7ef7e · 5a7ef7e
2 parents 11f56b3 + 2b6b523
commit 5a7ef7e
Show file tree

Hide file tree

Showing 4 changed files with 299 additions and 154 deletions.
diff --git a/tf_pwa/adaptive_bins.py b/tf_pwa/adaptive_bins.py
@@ -17,7 +17,8 @@
 class AdaptiveBound(object):
     """adaptive bound cut for data value"""
 
-    def __init__(self, base_data, bins):
+    def __init__(self, base_data, bins, base_bound=None):
+        self._base_bound = base_bound
         if isinstance(bins, int):
             self._base_data = np.array([base_data])
             self.bins = [[bins]]
@@ -33,9 +34,10 @@ def __init__(self, base_data, bins):
     @functools.lru_cache()
     def get_bounds_data(self):
         """get split data bounds, and the data after splitting"""
-        base_bound = AdaptiveBound.base_bound(self._base_data)
+        if self._base_bound is None:
+            self._base_bound = AdaptiveBound.base_bound(self._base_data)
         bounds, datas = AdaptiveBound.loop_split_bound(
-            self._base_data, self.bins
+            self._base_data, self.bins, self._base_bound
         )
         return bounds, datas