From 94a1fdee1ae844720bf49dd4752b12394c1d3ce3 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Mon, 23 Jan 2023 20:25:16 -0500
Subject: [PATCH] Use new cached median method in overlay scaling

Massively speeds up scaling transform cycles (duh).

Also includes a draft for an "overlay transform" type/api; obviously
still a WIP :surfer:..
---
 piker/ui/_interaction.py | 59 +++++++++++++++++++++++++++++++++++-----
 1 file changed, 52 insertions(+), 7 deletions(-)

diff --git a/piker/ui/_interaction.py b/piker/ui/_interaction.py
index 2e437af1..bf119be2 100644
--- a/piker/ui/_interaction.py
+++ b/piker/ui/_interaction.py
@@ -42,6 +42,7 @@ import trio
 from ..log import get_logger
 from .._profile import Profiler
 from .._profile import pg_profile_enabled, ms_slower_then
+from ..data.types import Struct
 # from ._style import _min_points_to_show
 from ._editors import SelectRect
 from . import _event
@@ -343,6 +344,49 @@ async def handle_viewmode_mouse(
             view.order_mode.submit_order()
 
 
+class OverlayT(Struct):
+    '''
+    An overlay co-domain range transformer.
+
+    Used to translate and apply a range from one y-range
+    to another based on a returns logarithm:
+
+    R(ymn, ymx, yref) = (ymx - yref)/yref
+
+    which gives the log-scale multiplier, and
+
+    ymx_t = yref * (1 + R)
+
+    which gives the inverse to translate to the same value
+    in the target co-domain.
+
+    '''
+    viz: Viz  # viz with largest measured dispersion
+
+    mx: float = 0
+    mn: float = float('inf')
+
+    up_swing: float = 0
+    down_swing: float = 0
+    disp: float = 0
+
+    def loglin_from_range(
+        self,
+
+        y_ref: float,  # reference value for dispersion metric
+        mn: float,  # min y in target log-lin range
+        mx: float,  # max y in target log-lin range
+        offset: float,  # y-offset to start log-scaling from
+
+    ) -> tuple[float, float]:
+        r_up = (mx - y_ref) / y_ref
+        r_down = (mn - y_ref) / y_ref
+        ymn = offset * (1 + r_down)
+        ymx = offset * (1 + r_up)
+
+        return ymn, ymx
+
+
 class ChartView(ViewBox):
     '''
     Price chart view box with interaction behaviors you'd expect from
@@ -1034,19 +1078,20 @@ class ChartView(ViewBox):
                     row_start = arr[read_slc.start - 1]
 
                     if viz.is_ohlc:
-                        y_med = np.median(in_view['close'])
+                        y_med = viz.median_from_range(
+                            read_slc.start,
+                            read_slc.stop,
+                        )
                         y_start = row_start['open']
                     else:
-                        y_med = np.median(in_view[viz.name])
+                        y_med = viz.median_from_range(
+                            read_slc.start,
+                            read_slc.stop,
+                        )
                         y_start = row_start[viz.name]
 
                     profiler(f'{viz.name}@{chart_name} MINOR curve median')
 
-                    # x_start = ixrng[0]
-                    # print(
-                    #     f'{viz.name} ->\n'
-                    #     f'(x_start: {x_start}, y_start: {y_start}\n'
-                    # )
                     start_datums[viz.plot.vb] = (
                         viz,
                         y_start,