1 files changed, 9 insertions, 3 deletions
diff --git a/grapher/Models/Charts/ChartState/ChartState.cs b/grapher/Models/Charts/ChartState/ChartState.cs
index cc334ac..99f44ff 100644
--- a/grapher/Models/Charts/ChartState/ChartState.cs
+++ b/grapher/Models/Charts/ChartState/ChartState.cs
@@ -15,12 +15,14 @@ namespace grapher.Models.Charts.ChartState
             ChartXY sensitivityChart,
             ChartXY velocityChart,
             ChartXY gainChart,
-            AccelData accelData)
+            AccelData accelData,
+            AccelCalculator calculator)
         {
             SensitivityChart = sensitivityChart;
             VelocityChart = velocityChart;
             GainChart = gainChart;
-            AccelData = accelData;
+            Data = accelData;
+            Calculator = calculator;
         }
 
         public ChartXY SensitivityChart { get; }
@@ -29,7 +31,9 @@ namespace grapher.Models.Charts.ChartState
 
         public ChartXY GainChart { get; }
 
-        public AccelData AccelData { get; }
+        public AccelData Data { get; }
+
+        public AccelCalculator Calculator { get; }
 
         public virtual DriverSettings Settings { get; set; }
 
@@ -39,6 +43,8 @@ namespace grapher.Models.Charts.ChartState
 
         public abstract void Activate();
 
+        public abstract void Calculate();
+
         public void DrawLastMovement()
         {
             SensitivityChart.DrawLastMovementValue();