Make all graph visualizers buffered

bonsai-rx · Jun 22, 2023 · 7e62da2 · 7e62da2
1 parent 3dcb2b9
commit 7e62da2
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 10 deletions.
diff --git a/Bonsai.Design.Visualizers/BarGraphVisualizer.cs b/Bonsai.Design.Visualizers/BarGraphVisualizer.cs
@@ -8,7 +8,7 @@ namespace Bonsai.Design.Visualizers
     /// <summary>
     /// Provides a type visualizer to display an object as a bar graph.
     /// </summary>
-    public class BarGraphVisualizer : DialogTypeVisualizer
+    public class BarGraphVisualizer : BufferedVisualizer
     {
         BarGraphBuilder.VisualizerController controller;
         BarGraphView view;

diff --git a/Bonsai.Design.Visualizers/GraphHelper.cs b/Bonsai.Design.Visualizers/GraphHelper.cs
@@ -79,6 +79,11 @@ static bool IsIntegralType(Type type)
         }
 
         internal static Expression SelectIndexMember(Expression expression, string indexSelector, out string indexLabel)
+        {
+            return SelectIndexMember(time: null, expression, indexSelector, out indexLabel);
+        }
+
+        internal static Expression SelectIndexMember(Expression time, Expression expression, string indexSelector, out string indexLabel)
         {
             Expression selectedIndex;
             if (!string.IsNullOrEmpty(indexSelector))
@@ -88,14 +93,14 @@ internal static Expression SelectIndexMember(Expression expression, string index
             }
             else
             {
-                selectedIndex = Expression.Property(null, typeof(DateTime), nameof(DateTime.Now));
+                selectedIndex = time ?? Expression.Property(null, typeof(DateTime), nameof(DateTime.Now));
                 indexLabel = "Time";
             }
 
             if (selectedIndex.Type == typeof(DateTimeOffset)) selectedIndex = Expression.Property(selectedIndex, nameof(DateTimeOffset.DateTime));
             if (selectedIndex.Type == typeof(DateTime))
             {
-                selectedIndex = Expression.Convert(selectedIndex, typeof(ZedGraph.XDate));
+                selectedIndex = Expression.Convert(selectedIndex, typeof(XDate));
             }
 
             return selectedIndex;

diff --git a/Bonsai.Design.Visualizers/LineGraphVisualizer.cs b/Bonsai.Design.Visualizers/LineGraphVisualizer.cs
@@ -8,7 +8,7 @@ namespace Bonsai.Design.Visualizers
     /// <summary>
     /// Provides a type visualizer to display an object as a line graph.
     /// </summary>
-    public class LineGraphVisualizer : DialogTypeVisualizer
+    public class LineGraphVisualizer : BufferedVisualizer
     {
         LineGraphBuilder.VisualizerController controller;
         LineGraphView view;

diff --git a/Bonsai.Design.Visualizers/RollingGraphBuilder.cs b/Bonsai.Design.Visualizers/RollingGraphBuilder.cs
@@ -101,7 +101,7 @@ internal class VisualizerController
             internal string[] ValueLabels;
             internal SymbolType SymbolType;
             internal float LineWidth;
-            internal Action<object, RollingGraphVisualizer> AddValues;
+            internal Action<DateTime, object, RollingGraphVisualizer> AddValues;
         }
 
         /// <summary>
@@ -113,6 +113,7 @@ public override Expression Build(IEnumerable<Expression> arguments)
         {
             var source = arguments.First();
             var parameterType = source.Type.GetGenericArguments()[0];
+            var timeParameter = Expression.Parameter(typeof(DateTime));
             var valueParameter = Expression.Parameter(typeof(object));
             var viewParameter = Expression.Parameter(typeof(RollingGraphVisualizer));
             var elementVariable = Expression.Variable(parameterType);
@@ -125,7 +126,7 @@ public override Expression Build(IEnumerable<Expression> arguments)
                 LineWidth = LineWidth
             };
 
-            var selectedIndex = GraphHelper.SelectIndexMember(elementVariable, IndexSelector, out Controller.IndexLabel);
+            var selectedIndex = GraphHelper.SelectIndexMember(timeParameter, elementVariable, IndexSelector, out Controller.IndexLabel);
             Controller.IndexType = selectedIndex.Type;
             if (selectedIndex.Type != typeof(double) && selectedIndex.Type != typeof(string))
             {
@@ -136,7 +137,12 @@ public override Expression Build(IEnumerable<Expression> arguments)
             var showBody = Expression.Block(new[] { elementVariable },
                 Expression.Assign(elementVariable, Expression.Convert(valueParameter, parameterType)),
                 Expression.Call(viewParameter, nameof(RollingGraphVisualizer.AddValues), null, selectedIndex, selectedValues));
-            Controller.AddValues = Expression.Lambda<Action<object, RollingGraphVisualizer>>(showBody, valueParameter, viewParameter).Compile();
+            Controller.AddValues = Expression.Lambda<Action<DateTime, object, RollingGraphVisualizer>>(
+                showBody,
+                timeParameter,
+                valueParameter,
+                viewParameter)
+                .Compile();
             return Expression.Call(typeof(RollingGraphBuilder), nameof(Process), new[] { parameterType }, source);
         }
 

diff --git a/Bonsai.Design.Visualizers/RollingGraphVisualizer.cs b/Bonsai.Design.Visualizers/RollingGraphVisualizer.cs
@@ -8,7 +8,7 @@ namespace Bonsai.Design.Visualizers
     /// <summary>
     /// Provides a type visualizer to display an object as a rolling graph.
     /// </summary>
-    public class RollingGraphVisualizer : DialogTypeVisualizer
+    public class RollingGraphVisualizer : BufferedVisualizer
     {
         static readonly TimeSpan TargetElapsedTime = TimeSpan.FromSeconds(1.0 / 30);
         RollingGraphBuilder.VisualizerController controller;
@@ -127,8 +127,13 @@ public override void Load(IServiceProvider provider)
         /// <inheritdoc/>
         public override void Show(object value)
         {
-            var time = DateTime.Now;
-            controller.AddValues(value, this);
+            Show(DateTime.Now, value);
+        }
+
+        /// <inheritdoc/>
+        protected override void Show(DateTime time, object value)
+        {
+            controller.AddValues(time, value, this);
             if ((time - updateTime) > TargetElapsedTime)
             {
                 view.Graph.Invalidate();