ground/gcs/src/plugins/setupwizard/setupwizard.h

   1 /**
   2  ******************************************************************************
   3  *
   4  * @file       setupwizard.h
   5  * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2012.
   6  * @addtogroup GCSPlugins GCS Plugins
   7  * @{
   8  * @addtogroup Setup Wizard  Plugin
   9  * @{
  10  * @brief A Wizards to make the initial setup easy for everyone.
  11  *****************************************************************************/
  12 /*
  13  * This program is free software; you can redistribute it and/or modify
  14  * it under the terms of the GNU General Public License as published by
  15  * the Free Software Foundation; either version 3 of the License, or
  16  * (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful, but
  19  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  20  * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
  21  * for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License along
  24  * with this program; if not, write to the Free Software Foundation, Inc.,
  25  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  26  */
  27
  28 #ifndef SETUPWIZARD_H
  29 #define SETUPWIZARD_H
  30
  31 #include <QWizard>
  32 #include "biascalibrationutil.h"
  33 #include <coreplugin/icore.h>
  34 #include <coreplugin/connectionmanager.h>
  35 #include "vehicleconfigurationsource.h"
  36 #include "vehicleconfigurationhelper.h"
  37
  38 class SetupWizard : public QWizard, public VehicleConfigurationSource {
  39     Q_OBJECT
  40
  41 public:
  42     SetupWizard(QWidget *parent = 0);
  43     ~SetupWizard();
  44     int nextId() const;
  45
  46     void setControllerType(SetupWizard::CONTROLLER_TYPE type)
  47     {
  48         m_controllerType = type;
  49     }
  50     SetupWizard::CONTROLLER_TYPE getControllerType() const
  51     {
  52         return m_controllerType;
  53     }
  54
  55     void setVehicleType(SetupWizard::VEHICLE_TYPE type)
  56     {
  57         m_vehicleType = type;
  58     }
  59     SetupWizard::VEHICLE_TYPE getVehicleType() const
  60     {
  61         return m_vehicleType;
  62     }
  63
  64     void setVehicleSubType(SetupWizard::VEHICLE_SUB_TYPE type)
  65     {
  66         m_vehicleSubType = type;
  67     }
  68     SetupWizard::VEHICLE_SUB_TYPE getVehicleSubType() const
  69     {
  70         return m_vehicleSubType;
  71     }
  72
  73     void setInputType(SetupWizard::INPUT_TYPE type)
  74     {
  75         m_inputType = type;
  76     }
  77     SetupWizard::INPUT_TYPE getInputType() const
  78     {
  79         return m_inputType;
  80     }
  81
  82     void setEscType(SetupWizard::ESC_TYPE type)
  83     {
  84         m_escType = type;
  85     }
  86     SetupWizard::ESC_TYPE getEscType() const
  87     {
  88         return m_escType;
  89     }
  90
  91     void setServoType(SetupWizard::SERVO_TYPE type)
  92     {
  93         m_servoType = type;
  94     }
  95     SetupWizard::SERVO_TYPE getServoType() const
  96     {
  97         return m_servoType;
  98     }
  99
 100     void setAirspeedType(SetupWizard::AIRSPEED_TYPE setting)
 101     {
 102         m_airspeedType = setting;
 103     }
 104     SetupWizard::AIRSPEED_TYPE getAirspeedType() const
 105     {
 106         return m_airspeedType;
 107     }
 108
 109
 110     void setGpsType(SetupWizard::GPS_TYPE setting)
 111     {
 112         m_gpsType = setting;
 113     }
 114     SetupWizard::GPS_TYPE getGpsType() const
 115     {
 116         return m_gpsType;
 117     }
 118
 119     void setRadioSetting(SetupWizard::RADIO_SETTING setting)
 120     {
 121         m_radioSetting = setting;
 122     }
 123     SetupWizard::RADIO_SETTING getRadioSetting() const
 124     {
 125         return m_radioSetting;
 126     }
 127
 128     void setVehicleTemplate(QJsonObject *templ)
 129     {
 130         m_vehicleTemplate = templ;
 131     }
 132     QJsonObject *getVehicleTemplate() const
 133     {
 134         return m_vehicleTemplate;
 135     }
 136
 137     void setLevellingBias(accelGyroBias bias)
 138     {
 139         m_calibrationBias = bias; m_calibrationPerformed = true;
 140     }
 141     bool isCalibrationPerformed() const
 142     {
 143         return m_calibrationPerformed;
 144     }
 145     accelGyroBias getCalibrationBias() const
 146     {
 147         return m_calibrationBias;
 148     }
 149
 150     void setActuatorSettings(QList<actuatorChannelSettings> actuatorSettings)
 151     {
 152         m_actuatorSettings = actuatorSettings;
 153     }
 154     bool isMotorCalibrationPerformed() const
 155     {
 156         return m_motorCalibrationPerformed;
 157     }
 158     QList<actuatorChannelSettings> getActuatorSettings() const
 159     {
 160         return m_actuatorSettings;
 161     }
 162
 163     void setRestartNeeded(bool needed)
 164     {
 165         m_restartNeeded = needed;
 166     }
 167     bool isRestartNeeded() const
 168     {
 169         return m_restartNeeded;
 170     }
 171
 172     QString getSummaryText();
 173
 174     Core::ConnectionManager *getConnectionManager()
 175     {
 176         if (!m_connectionManager) {
 177             m_connectionManager = Core::ICore::instance()->connectionManager();
 178             Q_ASSERT(m_connectionManager);
 179         }
 180         return m_connectionManager;
 181     }
 182     void reboot() const;
 183
 184 private slots:
 185     void customBackClicked();
 186     void pageChanged(int currId);
 187 private:
 188     enum { PAGE_START, PAGE_CONTROLLER, PAGE_VEHICLES, PAGE_MULTI, PAGE_FIXEDWING,
 189            PAGE_AIRSPEED, PAGE_GPS, PAGE_HELI, PAGE_SURFACE, PAGE_INPUT, PAGE_ESC, PAGE_SERVO,
 190            PAGE_BIAS_CALIBRATION, PAGE_ESC_CALIBRATION, PAGE_REVO_CALIBRATION, PAGE_OUTPUT_CALIBRATION,
 191            PAGE_SAVE, PAGE_SUMMARY, PAGE_NOTYETIMPLEMENTED, PAGE_AIRFRAME_INITIAL_TUNING,
 192            PAGE_REBOOT, PAGE_END, PAGE_UPDATE };
 193     void createPages();
 194     bool saveHardwareSettings() const;
 195     bool canAutoUpdate() const;
 196
 197     CONTROLLER_TYPE m_controllerType;
 198     VEHICLE_TYPE m_vehicleType;
 199     VEHICLE_SUB_TYPE m_vehicleSubType;
 200     INPUT_TYPE m_inputType;
 201     ESC_TYPE m_escType;
 202     SERVO_TYPE m_servoType;
 203     AIRSPEED_TYPE m_airspeedType;
 204     GPS_TYPE m_gpsType;
 205     RADIO_SETTING m_radioSetting;
 206
 207     QJsonObject *m_vehicleTemplate;
 208
 209     bool m_calibrationPerformed;
 210     accelGyroBias m_calibrationBias;
 211
 212     bool m_motorCalibrationPerformed;
 213     QList<actuatorChannelSettings> m_actuatorSettings;
 214
 215     bool m_restartNeeded;
 216
 217     bool m_back;
 218
 219     Core::ConnectionManager *m_connectionManager;
 220 };
 221
 222 #endif // SETUPWIZARD_H