UASControlWidget.cc 4.66 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
/*=====================================================================

PIXHAWK Micro Air Vehicle Flying Robotics Toolkit

(c) 2009, 2010 PIXHAWK PROJECT  <http://pixhawk.ethz.ch>

This file is part of the PIXHAWK project

    PIXHAWK is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    PIXHAWK is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with PIXHAWK. If not, see <http://www.gnu.org/licenses/>.

======================================================================*/

/**
 * @file
 *   @brief Definition of widget controlling one MAV
 *
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#include <QString>
#include <QTimer>
#include <QLabel>
#include <QFileDialog>
#include <QDebug>
#include <QProcess>

#include <MG.h>
#include "UASControlWidget.h"
#include <UASManager.h>
#include <UAS.h>
//#include <mavlink.h>

UASControlWidget::UASControlWidget(QWidget *parent) : QWidget(parent),
        uas(NULL)
{
    ui.setupUi(this);

    connect(UASManager::instance(), SIGNAL(activeUASSet(UASInterface*)), this, SLOT(setUAS(UASInterface*)));
    ui.modeComboBox->insertItem(MAV_MODE_LOCKED, "MODE LOCKED");
    ui.modeComboBox->insertItem(MAV_MODE_MANUAL, "MODE MANUAL");
    ui.modeComboBox->insertItem(MAV_MODE_GUIDED, "MODE GUIDED");
    ui.modeComboBox->insertItem(MAV_MODE_AUTO, "MODE AUTO");
    ui.modeComboBox->insertItem(MAV_MODE_TEST1, "MODE TEST1");
    ui.modeComboBox->insertItem(MAV_MODE_TEST2, "MODE TEST2");
    ui.modeComboBox->insertItem(MAV_MODE_TEST3, "MODE TEST3");
}

void UASControlWidget::setUAS(UASInterface* uas)
{
    if (this->uas != NULL)
    {
        disconnect(ui.controlButton, SIGNAL(clicked()), uas, SLOT(enable_motors()));
        disconnect(ui.liftoffButton, SIGNAL(clicked()), uas, SLOT(launch()));
        disconnect(ui.landButton, SIGNAL(clicked()), uas, SLOT(home()));
        disconnect(ui.shutdownButton, SIGNAL(clicked()), uas, SLOT(shutdown()));
68 69
        disconnect(ui.modeComboBox, SIGNAL(activated(int)), this, SLOT(setMode(int)));
        disconnect(ui.setModeButton, SIGNAL(clicked()), this, SLOT(transmitMode()));
pixhawk's avatar
pixhawk committed
70 71 72 73 74 75 76 77 78
    }
    else
    {
        // Connect user interface controls
        connect(ui.controlButton, SIGNAL(clicked()), this, SLOT(cycleContextButton()));
        connect(ui.liftoffButton, SIGNAL(clicked()), uas, SLOT(launch()));
        connect(ui.landButton, SIGNAL(clicked()), uas, SLOT(home()));
        connect(ui.shutdownButton, SIGNAL(clicked()), uas, SLOT(shutdown()));
        connect(ui.modeComboBox, SIGNAL(activated(int)), this, SLOT(setMode(int)));
79
        connect(ui.setModeButton, SIGNAL(clicked()), this, SLOT(transmitMode()));
pixhawk's avatar
pixhawk committed
80 81 82 83 84 85 86 87 88 89 90 91 92 93
        ui.modeComboBox->insertItem(0, "Select..");

        ui.controlStatusLabel->setText(tr("Connected to ") + uas->getUASName());

        this->uas = uas;
    }
}

UASControlWidget::~UASControlWidget() {

}

void UASControlWidget::setMode(int mode)
{
94
    // Adapt context button mode
pixhawk's avatar
pixhawk committed
95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
    switch (mode)
    {
        case MAV_MODE_LOCKED:
        break;
        case MAV_MODE_MANUAL:
        break;
        case MAV_MODE_GUIDED:
        break;
        case MAV_MODE_AUTO:
        break;
        case MAV_MODE_TEST1:
        break;
        case MAV_MODE_TEST2:
        break;
        case MAV_MODE_TEST3:
        break;
    }
pixhawk's avatar
pixhawk committed
112 113

    // Set mode on system
114 115 116 117 118 119 120 121 122 123 124
    if (mode >= MAV_MODE_LOCKED && mode <= MAV_MODE_TEST3)
    {
        uasMode = mode;
        ui.modeComboBox->setCurrentIndex(mode);
    }
    qDebug() << "SET MODE REQUESTED" << mode;
}

void UASControlWidget::transmitMode()
{
    this->uas->setMode(uasMode);
pixhawk's avatar
pixhawk committed
125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
}

void UASControlWidget::cycleContextButton()
{
    //switch(uas->getMode());
    static int state = 0;

    UAS* mav = dynamic_cast<UAS*>(this->uas);
    if (mav)
    {

        switch (state)
        {
        case 0:
            ui.controlButton->setText(tr("Stop Engine"));
            mav->setMode(MAV_MODE_MANUAL);
            mav->enable_motors();
            state++;
            break;
        case 1:
            ui.controlButton->setText(tr("Activate Engine"));
            mav->setMode(MAV_MODE_LOCKED);
            mav->disable_motors();
            state = 0;
            break;
        case 2:
            //ui.controlButton->setText(tr("Force Landing"));
            ui.controlButton->setText(tr("KILL VEHICLE"));
            break;
        }
    }

}