sweep highlight updates

Author: danielparthier

ephys/classes/plot/CurveArray.py

@@ -5,7 +5,7 @@
 """
 
 import numpy as np
-from pyqtgraph import PlotCurveItem
+from pyqtgraph import PlotCurveItem, mkPen
 
 
 class CurveArray:
@@ -64,15 +64,20 @@ def __init__(
         self.update_data(sweep_index=sweep_index)
 
     def update_data(
-        self, sweep_index: int | None, sweep_trace: "TraceCurve |  None" = None
+        self,
+        sweep_index: int | None,
+        sweep_trace: "TraceCurve |  None" = None,
+        pen=mkPen(color="red", alpha=1, width=1),
     ) -> None:
+
         if sweep_trace is not None:
             self.trace_curve = sweep_trace
         if isinstance(self.trace_curve, TraceCurve):
             data_xy = self.trace_curve.return_sweep_data(sweep_index)
             if data_xy is not None:
                 self.setData(x=data_xy[0], y=data_xy[1])
                 self.sweep_index = sweep_index
+                self.setPen(pen)
             else:
                 self.setData(x=np.array([]), y=np.array([]))
                 self.sweep_index = None