Add more robust analysis error handling (wpilibsuite#342)

Avnzx · Jan 30, 2022 · d6a9741 · d6a9741
1 parent 3b3d640
commit d6a9741
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 10 deletions.
diff --git a/sysid-application/src/main/native/cpp/analysis/AnalysisManager.cpp b/sysid-application/src/main/native/cpp/analysis/AnalysisManager.cpp
@@ -625,6 +625,14 @@ void AnalysisManager::PrepareData() {
 }
 
 AnalysisManager::Gains AnalysisManager::Calculate() {
+  if (m_filteredDatasets.empty()) {
+    throw std::runtime_error(
+        "There is an unresolved issue with the data being used. Please "
+        "double-check the data quality and adjust settings such as units, "
+        "velocity threshold, test duration, etc. to make sure that there are "
+        "no errors with the dataset before attempting to re-calculate gains.");
+  }
+
   WPI_INFO(m_logger, "{}", "Calculating Gains");
   // Calculate feedforward gains from the data.
   auto ffGains = sysid::CalculateFeedforwardGains(

diff --git a/sysid-application/src/main/native/cpp/view/Analyzer.cpp b/sysid-application/src/main/native/cpp/view/Analyzer.cpp
@@ -91,7 +91,7 @@ void Analyzer::Display() {
     if (ImGui::Combo("Dataset", &m_settings.dataset, AnalysisManager::kDatasets,
                      m_type == analysis::kDrivetrain ? 9 : 3)) {
       m_enabled = true;
-      Calculate();
+      RefreshInformation();
       PrepareGraphs();
     }
     ImGui::SameLine(width - ImGui::CalcTextSize("Reset").x);
@@ -151,24 +151,26 @@ void Analyzer::Display() {
         try {
           m_manager->OverrideUnits(m_unit, m_factor);
           m_enabled = true;
-          Calculate();
+          RefreshInformation();
         } catch (const std::exception& e) {
-          ex = true;
-          m_exception = e.what();
+          HandleGeneralError(e);
         }
       }
 
       ImGui::EndPopup();
-      if (ex) {
-        m_errorPopup = true;
-      }
     }
 
     ImGui::SameLine();
     if (ImGui::Button("Reset Units from JSON")) {
-      m_manager->ResetUnitsFromJSON();
-      m_enabled = true;
-      Calculate();
+      try {
+        m_manager->ResetUnitsFromJSON();
+        m_factor = m_manager->GetFactor();
+        m_unit = m_manager->GetUnit();
+        m_enabled = true;
+        RefreshInformation();
+      } catch (const std::exception& e) {
+        HandleGeneralError(e);
+      }
     }
   }