From e3acbf3b0b89e2c0ef1458f59b3306c38380d07b Mon Sep 17 00:00:00 2001
From: Don Gagne <don@thegagnes.com>
Date: Wed, 24 Dec 2014 23:15:04 -0800
Subject: [PATCH] Setup prereq and armed checks

---
 src/VehicleSetup/SetupView.cc | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/VehicleSetup/SetupView.cc b/src/VehicleSetup/SetupView.cc
index 3718e290e..9149f1db0 100644
--- a/src/VehicleSetup/SetupView.cc
+++ b/src/VehicleSetup/SetupView.cc
@@ -33,6 +33,7 @@
 #include "ParameterEditor.h"
 #include "SetupWidgetHolder.h"
 #include "MainWindow.h"
+#include "QGCMessageBox.h"
 
 #include <QQmlError>
 #include <QQmlContext>
@@ -117,6 +118,11 @@ void SetupView::_setConnectedView(void)
 
 void SetupView::_firmwareButtonClicked(void)
 {
+    if (_uasCurrent->isArmed()) {
+        QGCMessageBox::warning("Setup", "Firmware Update cannot be performed while vehicle is armed.");
+        return;
+    }
+    
     SetupWidgetHolder* dialog = new SetupWidgetHolder(MainWindow::instance());
     dialog->setModal(true);
     dialog->setWindowTitle("Firmware Upgrade");
@@ -139,9 +145,20 @@ void SetupView::_parametersButtonClicked(void)
 
 void SetupView::_setupButtonClicked(const QVariant& component)
 {
+    if (_uasCurrent->isArmed()) {
+        QGCMessageBox::warning("Setup", "Setup cannot be performed while vehicle is armed.");
+        return;
+    }
+
     VehicleComponent* vehicle = qobject_cast<VehicleComponent*>(component.value<QObject*>());
     Q_ASSERT(vehicle);
     
+    QString setupPrereq = vehicle->prerequisiteSetup();
+    if (!setupPrereq.isEmpty()) {
+        QGCMessageBox::warning("Setup", QString("%1 setup must be completed prior to %2 setup.").arg(setupPrereq).arg(vehicle->name()));
+        return;
+    }
+    
     SetupWidgetHolder dialog(MainWindow::instance());
     dialog.setModal(true);
     dialog.setWindowTitle(vehicle->name());
-- 
2.22.0