UASManager.cc 6.49 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2
/*=====================================================================

lm's avatar
lm committed
3
QGroundControl Open Source Ground Control Station
pixhawk's avatar
pixhawk committed
4

lm's avatar
lm committed
5
(c) 2009, 2010 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
pixhawk's avatar
pixhawk committed
6

lm's avatar
lm committed
7
This file is part of the QGROUNDCONTROL project
pixhawk's avatar
pixhawk committed
8

lm's avatar
lm committed
9
    QGROUNDCONTROL is free software: you can redistribute it and/or modify
pixhawk's avatar
pixhawk committed
10 11 12 13
    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.

lm's avatar
lm committed
14
    QGROUNDCONTROL is distributed in the hope that it will be useful,
pixhawk's avatar
pixhawk committed
15 16 17 18 19
    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
lm's avatar
lm committed
20
    along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.
pixhawk's avatar
pixhawk committed
21 22 23 24 25

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

/**
 * @file
26
 *   @brief Implementation of class UASManager
pixhawk's avatar
pixhawk committed
27 28 29 30 31 32 33 34 35
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#include <QList>
#include <QApplication>
#include <QMessageBox>
#include <QTimer>
#include "UAS.h"
36 37 38
#include "UASInterface.h"
#include "UASManager.h"
#include "QGC.h"
pixhawk's avatar
pixhawk committed
39

40 41
UASManager* UASManager::instance()
{
pixhawk's avatar
pixhawk committed
42 43 44 45
    static UASManager* _instance = 0;
    if(_instance == 0) {
        _instance = new UASManager();

pixhawk's avatar
pixhawk committed
46 47
        // Set the application as parent to ensure that this object
        // will be destroyed when the main application exits
pixhawk's avatar
pixhawk committed
48 49 50 51 52 53 54 55 56 57 58 59 60
        _instance->setParent(qApp);
    }
    return _instance;
}

/**
 * @brief Private singleton constructor
 *
 * This class implements the singleton design pattern and has therefore only a private constructor.
 **/
UASManager::UASManager() :
        activeUAS(NULL)
{
61
    systems = QList<UASInterface*>();
pixhawk's avatar
pixhawk committed
62 63 64 65 66
    start(QThread::LowPriority);
}

UASManager::~UASManager()
{
67 68 69 70 71
    // Delete all systems
    foreach (UASInterface* mav, systems)
    {
        delete mav;
    }
pixhawk's avatar
pixhawk committed
72 73 74 75 76
}


void UASManager::run()
{
lm's avatar
lm committed
77 78 79 80 81
//    forever
//    {
//        QGC::SLEEP::msleep(5000);
//    }
    exec();
pixhawk's avatar
pixhawk committed
82 83 84 85
}

void UASManager::addUAS(UASInterface* uas)
{
86 87 88 89 90 91 92 93 94 95 96 97
    // WARNING: The active uas is set here
    // and then announced below. This is necessary
    // to make sure the getActiveUAS() function
    // returns the UAS once the UASCreated() signal
    // is emitted. The code is thus NOT redundant.
    bool firstUAS = false;
    if (activeUAS == NULL)
    {
        firstUAS = true;
        activeUAS = uas;
    }

pixhawk's avatar
pixhawk committed
98
    // Only execute if there is no UAS at this index
99
    if (!systems.contains(uas))
pixhawk's avatar
pixhawk committed
100
    {
101
        systems.append(uas);
102
        connect(uas, SIGNAL(destroyed(QObject*)), this, SLOT(removeUAS(QObject*)));
pixhawk's avatar
pixhawk committed
103 104 105 106
        emit UASCreated(uas);
    }

    // If there is no active UAS yet, set the first one as the active UAS
107
    if (firstUAS)
pixhawk's avatar
pixhawk committed
108
    {
109
        setActiveUAS(uas);
pixhawk's avatar
pixhawk committed
110 111 112
    }
}

113 114 115 116 117 118 119 120 121 122 123 124 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
void UASManager::removeUAS(QObject* uas)
{
    UASInterface* mav = qobject_cast<UASInterface*>(uas);

    if (mav)
    {
        int listindex = systems.indexOf(mav);

        if (mav == activeUAS)
        {
            if (systems.count() > 1)
            {
                // We only set a new UAS if more than one is present
                if (listindex != 0)
                {
                    // The system to be removed is not at position 1
                    // set position one as new active system
                    setActiveUAS(systems.first());
                }
                else
                {
                    // The system to be removed is at position 1,
                    // select the next system
                    setActiveUAS(systems.at(1));
                }
            }
            else
            {
                // TODO send a null pointer if no UAS is present any more
                // This has to be proberly tested however, since it might
                // crash code parts not handling null pointers correctly.
            }
        }
        systems.removeAt(listindex);
    }
}

lm's avatar
lm committed
150 151
QList<UASInterface*> UASManager::getUASList()
{
152
    return systems;
lm's avatar
lm committed
153 154
}

pixhawk's avatar
pixhawk committed
155 156 157 158 159
UASInterface* UASManager::getActiveUAS()
{
    return activeUAS; ///< Return zero pointer if no UAS has been loaded
}

160 161 162 163 164
UASInterface* UASManager::silentGetActiveUAS()
{
    return activeUAS; ///< Return zero pointer if no UAS has been loaded
}

pixhawk's avatar
pixhawk committed
165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222
bool UASManager::launchActiveUAS()
{
    // If the active UAS is set, execute command
    if (getActiveUAS()) activeUAS->launch();
    return (activeUAS); ///< Returns true if the UAS exists, false else
}

bool UASManager::haltActiveUAS()
{
    // If the active UAS is set, execute command
    if (getActiveUAS()) activeUAS->halt();
    return (activeUAS); ///< Returns true if the UAS exists, false else
}

bool UASManager::continueActiveUAS()
{
    // If the active UAS is set, execute command
    if (getActiveUAS()) activeUAS->go();
    return (activeUAS); ///< Returns true if the UAS exists, false else
}

bool UASManager::returnActiveUAS()
{
    // If the active UAS is set, execute command
    if (getActiveUAS()) activeUAS->home();
    return (activeUAS); ///< Returns true if the UAS exists, false else
}

bool UASManager::stopActiveUAS()
{
    // If the active UAS is set, execute command
    if (getActiveUAS()) activeUAS->emergencySTOP();
    return (activeUAS); ///< Returns true if the UAS exists, false else
}

bool UASManager::killActiveUAS()
{
    if (getActiveUAS()) activeUAS->emergencyKILL();
    return (activeUAS);
}

bool UASManager::shutdownActiveUAS()
{
    if (getActiveUAS()) activeUAS->shutdown();
    return (activeUAS);
}

void UASManager::configureActiveUAS()
{
    UASInterface* actUAS = getActiveUAS();
    if(actUAS)
    {
        // Do something
    }
}

UASInterface* UASManager::getUASForId(int id)
{
223 224 225 226 227 228 229 230 231 232 233 234
    UASInterface* system = NULL;

    foreach(UASInterface* sys, systems)
    {
        if (sys->getUASID() == id)
        {
            system = sys;
        }
    }

    // Return NULL if not found
    return system;
pixhawk's avatar
pixhawk committed
235 236
}

pixhawk's avatar
pixhawk committed
237
void UASManager::setActiveUAS(UASInterface* uas)
pixhawk's avatar
pixhawk committed
238
{
pixhawk's avatar
pixhawk committed
239 240 241
    if (uas != NULL)
    {
        activeUASMutex.lock();
242 243 244 245 246
        if (activeUAS != NULL)
        {
            emit activeUASStatusChanged(activeUAS, false);
            emit activeUASStatusChanged(activeUAS->getUASID(), false);
        }
pixhawk's avatar
pixhawk committed
247 248
        activeUAS = uas;
        activeUASMutex.unlock();
pixhawk's avatar
pixhawk committed
249

250
        activeUAS->setSelected();
pixhawk's avatar
pixhawk committed
251
        emit activeUASSet(uas);
252
        emit activeUASSet(uas->getUASID());
253
        emit activeUASSetListIndex(systems.indexOf(uas));
254 255
        emit activeUASStatusChanged(uas, true);
        emit activeUASStatusChanged(uas->getUASID(), true);
pixhawk's avatar
pixhawk committed
256
    }
pixhawk's avatar
pixhawk committed
257 258
}