From 597791af7c99012d9cec7ab00ec7f93256e5a981 Mon Sep 17 00:00:00 2001
From: ncguilbeault <ncguilbeault@gmail.com>
Date: Fri, 24 May 2024 22:47:56 +0100
Subject: [PATCH] Modified x axes to display forecast data from min to max x
 value

---
 src/Bonsai.ML.Visualizers/ForecastVisualizer.cs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Bonsai.ML.Visualizers/ForecastVisualizer.cs b/src/Bonsai.ML.Visualizers/ForecastVisualizer.cs
index d35e034..99cfa8f 100644
--- a/src/Bonsai.ML.Visualizers/ForecastVisualizer.cs
+++ b/src/Bonsai.ML.Visualizers/ForecastVisualizer.cs
@@ -5,9 +5,7 @@
 using Bonsai.Design;
 using Bonsai.ML.Visualizers;
 using Bonsai.ML.LinearDynamicalSystems.Kinematics;
-using System.Drawing;
 using OxyPlot;
-using OxyPlot.Series;
 using System.Reactive;
 using System.Linq;
 using System.Reactive.Linq;
@@ -91,6 +89,7 @@ protected override void ShowBuffer(IList<Timestamped<object>> values)
             var latestForecast = values.Last();
             var timestamp = latestForecast.Timestamp;
             var forecast = (Forecast)latestForecast.Value;
+            var futureTime = timestamp;
 
             List<Timestamped<object>> positionX = new();
             List<Timestamped<object>> positionY = new();
@@ -101,7 +100,7 @@ protected override void ShowBuffer(IList<Timestamped<object>> values)
 
             foreach (var forecastResult in forecast.ForecastResults)
             {
-                var futureTime = timestamp + forecastResult.Timestep;
+                futureTime = timestamp + forecastResult.Timestep;
                 positionX.Add(new Timestamped<object>(forecastResult.KinematicState.Position.X, futureTime));
                 positionY.Add(new Timestamped<object>(forecastResult.KinematicState.Position.Y, futureTime));
                 velocityX.Add(new Timestamped<object>(forecastResult.KinematicState.Velocity.X, futureTime));
@@ -119,6 +118,7 @@ protected override void ShowBuffer(IList<Timestamped<object>> values)
                 item.Visualizer.Plot.ResetLineSeries(item.Visualizer.lineSeries);
                 item.Visualizer.Plot.ResetAreaSeries(item.Visualizer.areaSeries);
                 item.Visualizer.ShowDataBuffer(item.Data);
+                item.Visualizer.Plot.SetAxes(minTime: timestamp.DateTime, maxTime: futureTime.DateTime);
             }
         }