Skip to content

Commit

Permalink
Modified x axes to display forecast data from min to max x value
Browse files Browse the repository at this point in the history
  • Loading branch information
ncguilbeault committed Jun 4, 2024
1 parent a0a496e commit 597791a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/Bonsai.ML.Visualizers/ForecastVisualizer.cs
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -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();
Expand All @@ -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));
Expand All @@ -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);
}
}

Expand Down

0 comments on commit 597791a

Please sign in to comment.