LibrePilot/ground/openpilotgcs/src/plugins/gcscontrol/gcscontrolgadget.cpp

/**
 ******************************************************************************
 *
 * @file       GCSControlgadget.cpp
 * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2010.
 * @addtogroup GCSPlugins GCS Plugins
 * @{
 * @addtogroup GCSControlGadgetPlugin GCSControl Gadget Plugin
 * @{
 * @brief A gadget to control the UAV, either from the keyboard or a joystick
 *****************************************************************************/
/*
 * This program 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.
 *
 * This program 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 this program; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */
#include "gcscontrolgadget.h"
#include "gcscontrolgadgetwidget.h"
#include "gcscontrolgadgetconfiguration.h"
#include "extensionsystem/pluginmanager.h"
#include "uavobjectmanager.h"
#include "uavobject.h"
#include <QDebug>

#define JOYSTICK_UPDATE_RATE 50

GCSControlGadget::GCSControlGadget(QString classId, GCSControlGadgetWidget *widget, QWidget *parent, QObject *plugin) :
    IUAVGadget(classId, parent),
    m_widget(widget)
{
    connect(getManualControlCommand(), SIGNAL(objectUpdated(UAVObject *)), this, SLOT(manualControlCommandUpdated(UAVObject *)));
    connect(widget, SIGNAL(sticksChanged(double, double, double, double)), this, SLOT(sticksChangedLocally(double, double, double, double)));
    connect(this, SIGNAL(sticksChangedRemotely(double, double, double, double)), widget, SLOT(updateSticks(double, double, double, double)));

    manualControlCommandUpdated(getManualControlCommand());

    control_sock = new QUdpSocket(this);

    connect(control_sock, SIGNAL(readyRead()), this, SLOT(readUDPCommand()));

    joystickTime.start();
    GCSControlPlugin *pl = dynamic_cast<GCSControlPlugin *>(plugin);
    connect(pl->sdlGamepad, SIGNAL(gamepads(quint8)), this, SLOT(gamepads(quint8)));
    connect(pl->sdlGamepad, SIGNAL(buttonState(ButtonNumber, bool)), this, SLOT(buttonState(ButtonNumber, bool)));
    connect(pl->sdlGamepad, SIGNAL(axesValues(QListInt16)), this, SLOT(axesValues(QListInt16)));
}

GCSControlGadget::~GCSControlGadget()
{
    delete m_widget;
}

void GCSControlGadget::loadConfiguration(IUAVGadgetConfiguration *config)
{
    GCSControlGadgetConfiguration *GCSControlConfig = qobject_cast< GCSControlGadgetConfiguration *>(config);

    QList<int> ql = GCSControlConfig->getChannelsMapping();
    rollChannel     = ql.at(0);
    pitchChannel    = ql.at(1);
    yawChannel      = ql.at(2);
    throttleChannel = ql.at(3);

    // if(control_sock->isOpen())
    // control_sock->close();
    control_sock->bind(GCSControlConfig->getUDPControlHost(), GCSControlConfig->getUDPControlPort(), QUdpSocket::ShareAddress);

    controlsMode = GCSControlConfig->getControlsMode();

    int i;
    for (i = 0; i < 8; i++) {
        buttonSettings[i].ActionID   = GCSControlConfig->getbuttonSettings(i).ActionID;
        buttonSettings[i].FunctionID = GCSControlConfig->getbuttonSettings(i).FunctionID;
        buttonSettings[i].Amount     = GCSControlConfig->getbuttonSettings(i).Amount;
        buttonSettings[i].Amount     = GCSControlConfig->getbuttonSettings(i).Amount;
        channelReverse[i] = GCSControlConfig->getChannelsReverse().at(i);
    }
}

ManualControlCommand *GCSControlGadget::getManualControlCommand()
{
    ExtensionSystem::PluginManager *pm = ExtensionSystem::PluginManager::instance();
    UAVObjectManager *objManager = pm->getObject<UAVObjectManager>();

    return dynamic_cast<ManualControlCommand *>(objManager->getObject(QString("ManualControlCommand")));
}

void GCSControlGadget::manualControlCommandUpdated(UAVObject *manualControlCommand)
{
    double roll     = manualControlCommand->getField("Roll")->getDouble();
    double pitch    = manualControlCommand->getField("Pitch")->getDouble();
    double yaw      = manualControlCommand->getField("Yaw")->getDouble();
    double throttle = manualControlCommand->getField("Throttle")->getDouble();

    // necessary against having the wrong joystick profile chosen, which shows weird values
    if (throttle > -1.0 && throttle <= 1.0) {
        // convert ManualControlCommand.Throttle range (0..1) to the widget's throttle stick range (-1..+1)
        throttle = -1.0 + (throttle * 2.0);
    } else {
        // with the safety value (line 206), this helps keep the sticks insde the margins
        if (throttle <= -1.0) {
            throttle = -1.0;
        } else {
            throttle = 1.0;
        }
    }

    // Remap RPYT to left X/Y and right X/Y depending on mode
    switch (controlsMode) {
    case 1:
        // Mode 1: LeftX = Yaw, LeftY = Pitch, RightX = Roll, RightY = Throttle
        emit sticksChangedRemotely(yaw, -pitch, roll, throttle);
        break;
    case 2:
        // Mode 2: LeftX = Yaw, LeftY = Throttle, RightX = Roll, RightY = Pitch
        emit sticksChangedRemotely(yaw, throttle, roll, -pitch);
        break;
    case 3:
        // Mode 3: LeftX = Roll, LeftY = Pitch, RightX = Yaw, RightY = Throttle
        emit sticksChangedRemotely(roll, -pitch, yaw, throttle);
        break;
    case 4:
        // Mode 4: LeftX = Roll, LeftY = Throttle, RightX = Yaw, RightY = Pitch;
        emit sticksChangedRemotely(roll, throttle, yaw, -pitch);
        break;
    }
}

/**
   Update the manual commands - maps depending on mode
 */
void GCSControlGadget::sticksChangedLocally(double leftX, double leftY, double rightX, double rightY)
{
    ManualControlCommand *manualControlCommand = getManualControlCommand();
    double oldRoll     = manualControlCommand->getField("Roll")->getDouble();
    double oldPitch    = manualControlCommand->getField("Pitch")->getDouble();
    double oldYaw      = manualControlCommand->getField("Yaw")->getDouble();
    double oldThrottle = manualControlCommand->getField("Throttle")->getDouble();

    double newRoll     = 0.0;
    double newPitch    = 0.0;
    double newYaw      = 0.0;
    double newThrottle = 0.0;

    // Remap left X/Y and right X/Y to RPYT depending on mode
    switch (controlsMode) {
    case 1:
        // Mode 1: LeftX = Yaw, LeftY = Pitch, RightX = Roll, RightY = Throttle
        newRoll     = rightX;
        newPitch    = -leftY;
        newYaw      = leftX;
        newThrottle = rightY;
        break;
    case 2:
        // Mode 2: LeftX = Yaw, LeftY = Throttle, RightX = Roll, RightY = Pitch
        newRoll     = rightX;
        newPitch    = -rightY;
        newYaw      = leftX;
        newThrottle = leftY;
        break;
    case 3:
        // Mode 3: LeftX = Roll, LeftY = Pitch, RightX = Yaw, RightY = Throttle
        newRoll     = leftX;
        newPitch    = -leftY;
        newYaw      = rightX;
        newThrottle = rightY;
        break;
    case 4:
        // Mode 4: LeftX = Roll, LeftY = Throttle, RightX = Yaw, RightY = Pitch;
        newRoll     = leftX;
        newPitch    = -rightY;
        newYaw      = rightX;
        newThrottle = leftY;
        break;
    }

    // check if buttons have control over this axis... if so don't update it
    int buttonRollControl     = 0;
    int buttonPitchControl    = 0;
    int buttonYawControl      = 0;
    int buttonThrottleControl = 0;
    for (int i = 0; i < 8; i++) {
        if ((buttonSettings[i].FunctionID == 1) && ((buttonSettings[i].ActionID == 1) || (buttonSettings[i].ActionID == 2))) {
            buttonRollControl = 1;
        }
        if ((buttonSettings[i].FunctionID == 2) && ((buttonSettings[i].ActionID == 1) || (buttonSettings[i].ActionID == 2))) {
            buttonPitchControl = 1;
        }
        if ((buttonSettings[i].FunctionID == 3) && ((buttonSettings[i].ActionID == 1) || (buttonSettings[i].ActionID == 2))) {
            buttonYawControl = 1;
        }
        if ((buttonSettings[i].FunctionID == 4) && ((buttonSettings[i].ActionID == 1) || (buttonSettings[i].ActionID == 2))) {
            buttonThrottleControl = 1;
        }
    }

    // if we are not in local gcs control mode, ignore the joystick input
    if (((GCSControlGadgetWidget *)m_widget)->getGCSControl() == false || ((GCSControlGadgetWidget *)m_widget)->getUDPControl()) {
        return;
    }

    // if (newThrottle != oldThrottle) {
    // convert widget's throttle stick range (-1..+1) to ManualControlCommand.Throttle range (0..1)
    newThrottle = (newThrottle + 1.0) / 2.0;

    // safety value to stop the motors from spinning at 0% throttle
    if (newThrottle <= 0.01) {
        newThrottle = -1;
    }
    // }


    if ((newThrottle != oldThrottle) || (newPitch != oldPitch) || (newYaw != oldYaw) || (newRoll != oldRoll)) {
        if (buttonRollControl == 0) {
            manualControlCommand->getField("Roll")->setDouble(newRoll);
        }
        if (buttonPitchControl == 0) {
            manualControlCommand->getField("Pitch")->setDouble(newPitch);
        }
        if (buttonYawControl == 0) {
            manualControlCommand->getField("Yaw")->setDouble(newYaw);
        }
        if (buttonThrottleControl == 0) {
            manualControlCommand->getField("Throttle")->setDouble(newThrottle);
            manualControlCommand->getField("Thrust")->setDouble(newThrottle);
        }
        manualControlCommand->getField("Connected")->setValue("True");
        manualControlCommand->updated();
    }
}

void GCSControlGadget::gamepads(quint8 count)
{
    Q_UNUSED(count);

// sdlGamepad.setGamepad(0);
// sdlGamepad.setTickRate(JOYSTICK_UPDATE_RATE);
}

void GCSControlGadget::readUDPCommand()
{
    double pitch    = 0.0;
    double yaw      = 0.0;
    double roll     = 0.0;
    double throttle = 0.0;

    while (control_sock->hasPendingDatagrams()) {
        QByteArray datagram;
        datagram.resize(control_sock->pendingDatagramSize());
        control_sock->readDatagram(datagram.data(), datagram.size());
        QDataStream readData(datagram);
        bool badPack = false;
        int state    = 0;
        while (!readData.atEnd() && !badPack) {
            double buffer;
            readData >> buffer;
            switch (state) {
            case 0:
                if (buffer == 42) {
                    state = 1;
                } else {
                    state   = 0;
                    badPack = true;
                }
                break;
            case 1:
                pitch    = buffer;
                state    = 2;
                break;
            case 2:
                yaw      = buffer;
                state    = 3;
                break;
            case 3:
                roll     = buffer;
                state    = 4;
                break;
            case 4:
                throttle = buffer;
                state    = 5;
                break;
            case 5:
                if (buffer != 36 || !readData.atEnd()) {
                    badPack = true;
                }
                break;
            }
        }
        if (!badPack && ((GCSControlGadgetWidget *)m_widget)->getUDPControl()) {
            ManualControlCommand *manualControlCommand = getManualControlCommand();
            bool update = false;

            if (pitch != manualControlCommand->getField("Pitch")->getDouble()) {
                manualControlCommand->getField("Pitch")->setDouble(constrain(pitch));
                update = true;
            }
            if (yaw != manualControlCommand->getField("Yaw")->getDouble()) {
                manualControlCommand->getField("Yaw")->setDouble(constrain(yaw));
                update = true;
            }
            if (roll != manualControlCommand->getField("Roll")->getDouble()) {
                manualControlCommand->getField("Roll")->setDouble(constrain(roll));
                update = true;
            }
            if (throttle != manualControlCommand->getField("Throttle")->getDouble()) {
                manualControlCommand->getField("Throttle")->setDouble(constrain(throttle));
                manualControlCommand->getField("Thrust")->setDouble(constrain(throttle));
                update = true;
            }
            if (update) {
                manualControlCommand->getField("Connected")->setValue("True");
                manualControlCommand->updated();
            }
        }
    }

    qDebug() << "Pitch: " << pitch << " Yaw: " << yaw << " Roll: " << roll << " Throttle: " << throttle;
}

double GCSControlGadget::constrain(double value)
{
    if (value < -1) {
        return -1;
    }
    if (value > 1) {
        return 1;
    }
    return value;
}

void GCSControlGadget::buttonState(ButtonNumber number, bool pressed)
{
    if ((buttonSettings[number].ActionID > 0) && (buttonSettings[number].FunctionID > 0) && (pressed)) { // this button is configured
        ExtensionSystem::PluginManager *pm  = ExtensionSystem::PluginManager::instance();
        UAVObjectManager *objManager = pm->getObject<UAVObjectManager>();
        UAVDataObject *manualControlCommand = dynamic_cast<UAVDataObject *>(objManager->getObject(QString("ManualControlCommand")));
        bool currentCGSControl = ((GCSControlGadgetWidget *)m_widget)->getGCSControl();
        bool currentUDPControl = ((GCSControlGadgetWidget *)m_widget)->getUDPControl();

        switch (buttonSettings[number].ActionID) {
        case 1: // increase
            if (currentCGSControl) {
                switch (buttonSettings[number].FunctionID) {
                case 1: // Roll
                    manualControlCommand->getField("Roll")->setValue(bound(manualControlCommand->getField("Roll")->getValue().toDouble() + buttonSettings[number].Amount));
                    break;
                case 2: // Pitch
                    manualControlCommand->getField("Pitch")->setValue(bound(manualControlCommand->getField("Pitch")->getValue().toDouble() + buttonSettings[number].Amount));
                    break;
                case 3: // Yaw
                    manualControlCommand->getField("Yaw")->setValue(wrap(manualControlCommand->getField("Yaw")->getValue().toDouble() + buttonSettings[number].Amount));
                    break;
                case 4: // Throttle
                    manualControlCommand->getField("Throttle")->setValue(bound(manualControlCommand->getField("Throttle")->getValue().toDouble() + buttonSettings[number].Amount));
                    manualControlCommand->getField("Thrust")->setValue(bound(manualControlCommand->getField("Thrust")->getValue().toDouble() + buttonSettings[number].Amount));
                    break;
                }
            }
            break;
        case 2: // decrease
            if (currentCGSControl) {
                switch (buttonSettings[number].FunctionID) {
                case 1: // Roll
                    manualControlCommand->getField("Roll")->setValue(bound(manualControlCommand->getField("Roll")->getValue().toDouble() - buttonSettings[number].Amount));
                    break;
                case 2: // Pitch
                    manualControlCommand->getField("Pitch")->setValue(bound(manualControlCommand->getField("Pitch")->getValue().toDouble() - buttonSettings[number].Amount));
                    break;
                case 3: // Yaw
                    manualControlCommand->getField("Yaw")->setValue(wrap(manualControlCommand->getField("Yaw")->getValue().toDouble() - buttonSettings[number].Amount));
                    break;
                case 4: // Throttle
                    manualControlCommand->getField("Throttle")->setValue(bound(manualControlCommand->getField("Throttle")->getValue().toDouble() - buttonSettings[number].Amount));
                    manualControlCommand->getField("Thrust")->setValue(bound(manualControlCommand->getField("Thrust")->getValue().toDouble() - buttonSettings[number].Amount));
                    break;
                }
            }
            break;
        case 3: // toggle
            switch (buttonSettings[number].FunctionID) {
            case 1: // Armed
                if (currentCGSControl) {
                    if (((GCSControlGadgetWidget *)m_widget)->getArmed()) {
                        ((GCSControlGadgetWidget *)m_widget)->setArmed(false);
                    } else {
                        ((GCSControlGadgetWidget *)m_widget)->setArmed(true);
                    }
                }
                break;
            case 2: // GCS Control
                    // Toggle the GCS Control checkbox, its built in signalling will handle the update to OP
                ((GCSControlGadgetWidget *)m_widget)->setGCSControl(!currentCGSControl);

                break;
            case 3: // UDP Control
                if (currentCGSControl) {
                    ((GCSControlGadgetWidget *)m_widget)->setUDPControl(!currentUDPControl);
                }

                break;
            }

            break;
        }

        manualControlCommand->getField("Connected")->setValue("True");
        manualControlCommand->updated();
    }
    // buttonSettings[number].ActionID NIDT
    // buttonSettings[number].FunctionID -RPYTAC
    // buttonSettings[number].Amount
}

void GCSControlGadget::axesValues(QListInt16 values)
{
    int chMax = values.length();

    if (rollChannel >= chMax || pitchChannel >= chMax ||
        yawChannel >= chMax || throttleChannel >= chMax) {
        qDebug() << "GCSControl: configuration is inconsistent with current joystick! Aborting update.";
        return;
    }

    double rValue = (rollChannel > -1) ? values[rollChannel] : 0;
    double pValue = (pitchChannel > -1) ? values[pitchChannel] : 0;
    double yValue = (yawChannel > -1) ? values[yawChannel] : 0;
    double tValue = (throttleChannel > -1) ? values[throttleChannel] : 0;
    double max    = 32767;

    if (rollChannel > -1) {
        if (channelReverse[rollChannel] == true) {
            rValue = -rValue;
        }
    }
    if (pitchChannel > -1) {
        if (channelReverse[pitchChannel] == true) {
            pValue = -pValue;
        }
    }
    if (yawChannel > -1) {
        if (channelReverse[yawChannel] == true) {
            yValue = -yValue;
        }
    }
    if (throttleChannel > -1) {
        if (channelReverse[throttleChannel] == true) {
            tValue = -tValue;
        }
    }


    if (joystickTime.elapsed() > JOYSTICK_UPDATE_RATE) {
        joystickTime.restart();
        // Remap RPYT to left X/Y and right X/Y depending on mode
        // Mode 1: LeftX = Yaw, LeftY = Pitch, RightX = Roll, RightY = Throttle
        // Mode 2: LeftX = Yaw, LeftY = THrottle, RightX = Roll, RightY = Pitch
        // Mode 3: LeftX = Roll, LeftY = Pitch, RightX = Yaw, RightY = Throttle
        // Mode 4: LeftX = Roll, LeftY = Throttle, RightX = Yaw, RightY = Pitch;
        switch (controlsMode) {
        case 1:
            sticksChangedLocally(yValue / max, -pValue / max, rValue / max, -tValue / max);
            break;
        case 2:
            sticksChangedLocally(yValue / max, -tValue / max, rValue / max, -pValue / max);
            break;
        case 3:
            sticksChangedLocally(rValue / max, -pValue / max, yValue / max, -tValue / max);
            break;
        case 4:
            sticksChangedLocally(rValue / max, -tValue / max, yValue / max, -pValue / max);
            break;
        }
    }
}


double GCSControlGadget::bound(double input)
{
    if (input > 1.0) {
        return 1.0;
    }
    if (input < -1.0) {
        return -1.0;
    }
    return input;
}

double GCSControlGadget::wrap(double input)
{
    while (input > 1.0) {
        input -= 2.0;
    }
    while (input < -1.0) {
        input += 2.0;
    }
    return input;
}