Set whole or by component with toolstrip item

2 files changed, 91 insertions, 2 deletions

diff --git a/grapher/Models/Options/ApplyOptions.cs b/grapher/Models/Options/ApplyOptions.cs
new file mode 100644
index 0000000..389bf62
--- /dev/null
+++ b/grapher/Models/Options/ApplyOptions.cs
@@ -0,0 +1,85 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using System.Windows.Forms;
+using System.Windows.Forms.DataVisualization.Charting;
+
+namespace grapher.Models.Options
+{
+    public class ApplyOptions
+    {
+        public ApplyOptions(
+            ToolStripMenuItem wholeVectorMenuItem,
+            ToolStripMenuItem byComponentMenuItem)
+        {
+            WholeVectorMenuItem = wholeVectorMenuItem;
+            ByComponentVectorMenuItem = byComponentMenuItem;
+
+            WholeVectorMenuItem.Click += new System.EventHandler(OnWholeClicked);
+            ByComponentVectorMenuItem.Click += new System.EventHandler(OnByComponentClicked);
+
+            WholeVectorMenuItem.CheckedChanged += new System.EventHandler(OnWholeCheckedChange);
+            ByComponentVectorMenuItem.CheckedChanged += new System.EventHandler(OnByComponentCheckedChange);
+
+            IsWhole = false;
+        }
+
+        public ToolStripMenuItem WholeVectorMenuItem { get; }
+
+        public ToolStripMenuItem ByComponentVectorMenuItem { get; }
+
+        public bool IsWhole { get; private set; }
+
+        public void SetActive(bool isWhole)
+        {
+            WholeVectorMenuItem.Checked = isWhole;
+            ByComponentVectorMenuItem.Checked = !isWhole;
+        }
+
+        public void OnWholeClicked(object sender, EventArgs e)
+        {
+            if (!WholeVectorMenuItem.Checked)
+            {
+                WholeVectorMenuItem.Checked = true;
+                ByComponentVectorMenuItem.Checked = false;
+            }
+        }
+
+        public void OnByComponentClicked(object sender, EventArgs e)
+        {
+            if (!ByComponentVectorMenuItem.Checked)
+            {
+                WholeVectorMenuItem.Checked = false;
+                ByComponentVectorMenuItem.Checked = true;
+            }
+        }
+
+        public void OnWholeCheckedChange(object sender, EventArgs e)
+        {
+            if (WholeVectorMenuItem.Checked)
+            {
+                EnableWholeApplication();
+            }
+        }
+
+        public void OnByComponentCheckedChange(object sender, EventArgs e)
+        {
+            if (ByComponentVectorMenuItem.Checked)
+            {
+                EnableWholeApplication();
+            }
+        }
+
+        public void EnableWholeApplication()
+        {
+            IsWhole = true;
+        }
+        public void EnableByComponentApplication()
+        {
+            IsWhole = false;
+        }
+
+    }
+}
diff --git a/grapher/Models/Options/CapOptions.cs b/grapher/Models/Options/CapOptions.cs
index 493561a..3bdf3c6 100644
--- a/grapher/Models/Options/CapOptions.cs
+++ b/grapher/Models/Options/CapOptions.cs
@@ -93,12 +93,16 @@ namespace grapher
                 CapOption.ActiveValueLabels.X.FormatString = GainCapFormatString;
                 CapOption.ActiveValueLabels.X.Prefix = "Gain";
                 CapOption.SetActiveValues(gainCap, gainCap);
+                SensitivityCapCheck.Checked = true;
+                VelocityGainCapCheck.Checked = false;
             }
             else
             {
                 CapOption.ActiveValueLabels.X.FormatString = ActiveValueLabel.DefaultFormatString;
                 CapOption.ActiveValueLabels.X.Prefix = string.Empty;
                 CapOption.SetActiveValues(sensCapX, sensCapY);
+                SensitivityCapCheck.Checked = false;
+                VelocityGainCapCheck.Checked = true;
             }
         }
 
@@ -122,7 +126,7 @@ namespace grapher
 
         void OnSensitivityCapCheckedChange(object sender, EventArgs e)
         {
-            if (SensitivityCapCheck.Checked == true)
+            if (SensitivityCapCheck.Checked)
             {
                 EnableSensitivityCap();
             }
@@ -130,7 +134,7 @@ namespace grapher
 
         void OnVelocityGainCapCheckedChange(object sender, EventArgs e)
         {
-            if (SensitivityCapCheck.Checked == true)
+            if (SensitivityCapCheck.Checked)
             {
                 EnableVelocityGainCap();
             }