Use state component visualizer label in Y axis

bonsai-rx · Jun 28, 2024 · d01b703 · d01b703
1 parent fb5c41f
commit d01b703
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 13 deletions.
diff --git a/src/Bonsai.ML.Visualizers/StateComponentVisualizer.cs b/src/Bonsai.ML.Visualizers/StateComponentVisualizer.cs
@@ -71,6 +71,7 @@ public override void Load(IServiceProvider provider)
             var areaSeriesName = string.IsNullOrEmpty(Label) ? "Variance" : $"{Label} Variance";
             AreaSeries = Plot.AddNewAreaSeries(areaSeriesName, color: AreaSeriesColor);
 
+            if (Label != null) Plot.ValueLabel = Label;
             Plot.ResetLineSeries(LineSeries);
             Plot.ResetAreaSeries(AreaSeries);
             Plot.ResetAxes();

diff --git a/src/Bonsai.ML.Visualizers/TimeSeriesOxyPlotBase.cs b/src/Bonsai.ML.Visualizers/TimeSeriesOxyPlotBase.cs
@@ -11,15 +11,15 @@ namespace Bonsai.ML.Visualizers
 {
     internal class TimeSeriesOxyPlotBase : UserControl
     {
-        private PlotView view;
-        private PlotModel model;
-        private OxyColor defaultLineSeriesColor = OxyColors.Blue;
-        private OxyColor defaultAreaSeriesColor = OxyColors.LightBlue;
+        private readonly PlotView view;
+        private readonly PlotModel model;
+        private static readonly OxyColor DefaultLineSeriesColor = OxyColors.Blue;
+        private static readonly OxyColor DefaultAreaSeriesColor = OxyColors.LightBlue;
 
-        private Axis xAxis;
-        private Axis yAxis;
+        private readonly Axis xAxis;
+        private readonly Axis yAxis;
 
-        private StatusStrip statusStrip;
+        private readonly StatusStrip statusStrip;
 
         /// <summary>
         /// Gets or sets the datetime value that determines the starting time of the data values.
@@ -42,14 +42,18 @@ internal class TimeSeriesOxyPlotBase : UserControl
         public bool BufferData { get; set; }
 
         /// <summary>
-        /// Initializes a new instance of the <see cref="TimeSeriesOxyPlotBase"/> class
+        /// Gets or sets the label of the value axis in the time series plot.
         /// </summary>
-        public TimeSeriesOxyPlotBase()
+        public string ValueLabel
         {
-            Initialize();
+            get => yAxis.Title;
+            set => yAxis.Title = value;
         }
 
-        private void Initialize()
+        /// <summary>
+        /// Initializes a new instance of the <see cref="TimeSeriesOxyPlotBase"/> class
+        /// </summary>
+        public TimeSeriesOxyPlotBase()
         {
             view = new PlotView
             {
@@ -133,7 +137,7 @@ public void AddComboBoxWithLabel(string label, IEnumerable dataSource, int selec
         /// </summary>
         public LineSeries AddNewLineSeries(string lineSeriesName, OxyColor? color = null)
         {
-            OxyColor _color = color.HasValue ? color.Value : defaultLineSeriesColor;
+            OxyColor _color = color.HasValue ? color.Value : DefaultLineSeriesColor;
             LineSeries lineSeries = new LineSeries {
                 Title = lineSeriesName,
                 Color = _color
@@ -150,7 +154,7 @@ public LineSeries AddNewLineSeries(string lineSeriesName, OxyColor? color = null
         /// </summary>
         public AreaSeries AddNewAreaSeries(string areaSeriesName, OxyColor? color = null, OxyColor? fill = null, byte opacity = 100)
         {
-            OxyColor _color = color.HasValue ? color.Value : defaultAreaSeriesColor;
+            OxyColor _color = color.HasValue ? color.Value : DefaultAreaSeriesColor;
             OxyColor _fill = fill.HasValue? fill.Value : OxyColor.FromArgb(opacity, _color.R, _color.G, _color.B);
             AreaSeries areaSeries = new AreaSeries {
                 Title = areaSeriesName,