JoystickInput.cc 26.1 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2 3 4 5 6 7 8 9
/*=====================================================================
======================================================================*/

/**
 * @file
 *   @brief Joystick interface
 *
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *   @author Andreas Romer <mavteam@student.ethz.ch>
10
 *   @author Julian Oes <julian@oes.ch
pixhawk's avatar
pixhawk committed
11 12 13 14 15 16 17 18 19
 *
 */

#include "JoystickInput.h"

#include <QDebug>
#include <limits.h>
#include "UAS.h"
#include "UASManager.h"
20
#include "QGC.h"
21
#include <QMutexLocker>
Lorenz Meier's avatar
Lorenz Meier committed
22
#include <QSettings>
23
#include <math.h>
24 25 26
#include <limits>

using namespace std;
pixhawk's avatar
pixhawk committed
27 28 29 30 31 32

/**
 * The coordinate frame of the joystick axis is the aeronautical frame like shown on this image:
 * @image html http://pixhawk.ethz.ch/wiki/_media/standards/body-frame.png Aeronautical frame
 */
JoystickInput::JoystickInput() :
33 34
    sdlJoystickMin(-32768.0f),
    sdlJoystickMax(32767.0f),
35
    isEnabled(false),
36
    isCalibrating(false),
37
    done(false),
38
    uas(NULL),
39 40
    autopilotType(0),
    systemType(0),
41
    uasCanReverse(false),
42 43 44 45 46 47
    rollAxis(-1),
    pitchAxis(-1),
    yawAxis(-1),
    throttleAxis(-1),
    joystickName(""),
    joystickID(-1),
48
    joystickNumButtons(0)
pixhawk's avatar
pixhawk committed
49
{
50 51 52

    // Make sure we initialize with the correct UAS.
    setActiveUAS(UASManager::instance()->getActiveUAS());
53

54 55
    // Start this thread. This allows the Joystick Settings window to work correctly even w/o any UASes connected.
    start();
pixhawk's avatar
pixhawk committed
56 57
}

58 59
JoystickInput::~JoystickInput()
{
60 61
    storeGeneralSettings();
    storeJoystickSettings();
LM's avatar
LM committed
62
    done = true;
Lorenz Meier's avatar
Lorenz Meier committed
63 64
}

65 66 67 68 69 70 71 72
void JoystickInput::loadGeneralSettings()
{
    // Load defaults from settings
    QSettings settings;

    // Deal with settings specific to the JoystickInput
    settings.beginGroup("JOYSTICK_INPUT");
    isEnabled = settings.value("ENABLED", false).toBool();
73
    joystickName = settings.value("JOYSTICK_NAME", "").toString();
74
    mode = (JOYSTICK_MODE)settings.value("JOYSTICK_MODE", JOYSTICK_MODE_ATTITUDE).toInt();
75 76 77
    settings.endGroup();
}

78 79 80 81
/**
 * @brief Restores settings for the current joystick from saved settings file.
 * Assumes that both joystickName & joystickNumAxes are correct.
 */
82
void JoystickInput::loadJoystickSettings()
Lorenz Meier's avatar
Lorenz Meier committed
83
{
84 85
    // Load defaults from settings
    QSettings settings;
86 87

    // Now for the current joystick
88 89 90 91 92 93
    settings.beginGroup(joystickName);
    rollAxis = (settings.value("ROLL_AXIS_MAPPING", -1).toInt());
    pitchAxis = (settings.value("PITCH_AXIS_MAPPING", -1).toInt());
    yawAxis = (settings.value("YAW_AXIS_MAPPING", -1).toInt());
    throttleAxis = (settings.value("THROTTLE_AXIS_MAPPING", -1).toInt());

94 95 96 97
    // Clear out and then restore the (AUTOPILOT, SYSTEM) mapping for joystick settings
    joystickSettings.clear();
    int autopilots = settings.beginReadArray("AUTOPILOTS");
    for (int i = 0; i < autopilots; i++)
98
    {
99 100 101 102
        settings.setArrayIndex(i);
        int autopilotType = settings.value("AUTOPILOT_TYPE", 0).toInt();
        int systems = settings.beginReadArray("SYSTEMS");
        for (int j = 0; j < systems; j++)
103
        {
104 105 106 107 108 109
            settings.setArrayIndex(j);
            int systemType = settings.value("SYSTEM_TYPE", 0).toInt();

            // Now that both the autopilot and system type are available, update some references.
            QMap<int, bool>* joystickAxesInverted = &joystickSettings[autopilotType][systemType].axesInverted;
            QMap<int, bool>* joystickAxesLimited = &joystickSettings[autopilotType][systemType].axesLimited;
110 111
            QMap<int, float>* joystickAxesMinRange = &joystickSettings[autopilotType][systemType].axesMaxRange;
            QMap<int, float>* joystickAxesMaxRange = &joystickSettings[autopilotType][systemType].axesMinRange;
112 113 114 115 116 117 118 119 120 121 122 123
            QMap<int, int>* joystickButtonActions = &joystickSettings[autopilotType][systemType].buttonActions;

            // Read back the joystickAxesInverted QList one element at a time.
            int axesStored = settings.beginReadArray("AXES_INVERTED");
            for (int k = 0; k < axesStored; k++)
            {
                settings.setArrayIndex(k);
                int index = settings.value("INDEX", 0).toInt();
                bool inverted = settings.value("INVERTED", false).toBool();
                joystickAxesInverted->insert(index, inverted);
            }
            settings.endArray();
124

125 126 127 128 129 130 131 132 133 134 135
            // Read back the joystickAxesLimited QList one element at a time.
            axesStored = settings.beginReadArray("AXES_LIMITED");
            for (int k = 0; k < axesStored; k++)
            {
                settings.setArrayIndex(k);
                int index = settings.value("INDEX", 0).toInt();
                bool limited = settings.value("LIMITED", false).toBool();
                joystickAxesLimited->insert(index, limited);
            }
            settings.endArray();

136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157
            // Read back the joystickAxesMinRange QList one element at a time.
            axesStored = settings.beginReadArray("AXES_MIN_RANGE");
            for (int k = 0; k < axesStored; k++)
            {
                settings.setArrayIndex(k);
                int index = settings.value("INDEX", 0).toInt();
                float min = settings.value("MIN_RANGE", false).toFloat();
                joystickAxesMinRange->insert(index, min);
            }
            settings.endArray();

            // Read back the joystickAxesMaxRange QList one element at a time.
            axesStored = settings.beginReadArray("AXES_MAX_RANGE");
            for (int k = 0; k < axesStored; k++)
            {
                settings.setArrayIndex(k);
                int index = settings.value("INDEX", 0).toInt();
                float max = settings.value("MAX_RANGE", false).toFloat();
                joystickAxesMaxRange->insert(index, max);
            }
            settings.endArray();

158 159 160 161 162 163 164 165 166 167
            // Read back the button->action mapping.
            int buttonsStored = settings.beginReadArray("BUTTONS_ACTIONS");
            for (int k = 0; k < buttonsStored; k++)
            {
                settings.setArrayIndex(k);
                int index = settings.value("INDEX", 0).toInt();
                int action = settings.value("ACTION", 0).toInt();
                joystickButtonActions->insert(index, action);
            }
            settings.endArray();
168
        }
169
        settings.endArray();
170 171
    }
    settings.endArray();
172

173
    settings.endGroup();
174 175 176 177 178 179 180 181 182 183

    emit joystickSettingsChanged();
}

void JoystickInput::storeGeneralSettings() const
{
    QSettings settings;
    settings.beginGroup("JOYSTICK_INPUT");
    settings.setValue("ENABLED", isEnabled);
    settings.setValue("JOYSTICK_NAME", joystickName);
184
    settings.setValue("JOYSTICK_MODE", mode);
185
    settings.endGroup();
Lorenz Meier's avatar
Lorenz Meier committed
186 187
}

188
void JoystickInput::storeJoystickSettings() const
Lorenz Meier's avatar
Lorenz Meier committed
189
{
190 191 192 193 194 195 196
    // Store settings
    QSettings settings;
    settings.beginGroup(joystickName);
    settings.setValue("ROLL_AXIS_MAPPING", rollAxis);
    settings.setValue("PITCH_AXIS_MAPPING", pitchAxis);
    settings.setValue("YAW_AXIS_MAPPING", yawAxis);
    settings.setValue("THROTTLE_AXIS_MAPPING", throttleAxis);
197 198 199 200
    settings.beginWriteArray("AUTOPILOTS");
    QMapIterator<int, QMap<int, JoystickSettings> > i(joystickSettings);
    int autopilotIndex = 0;
    while (i.hasNext())
201
    {
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221
        i.next();
        settings.setArrayIndex(autopilotIndex++);

        int autopilotType = i.key();
        settings.setValue("AUTOPILOT_TYPE", autopilotType);

        settings.beginWriteArray("SYSTEMS");
        QMapIterator<int, JoystickSettings> j(i.value());
        int systemIndex = 0;
        while (j.hasNext())
        {
            j.next();
            settings.setArrayIndex(systemIndex++);

            int systemType = j.key();
            settings.setValue("SYSTEM_TYPE", systemType);

            // Now that both the autopilot and system type are available, update some references.
            QMapIterator<int, bool> joystickAxesInverted(joystickSettings[autopilotType][systemType].axesInverted);
            QMapIterator<int, bool> joystickAxesLimited(joystickSettings[autopilotType][systemType].axesLimited);
222 223
            QMapIterator<int, float> joystickAxesMinRange(joystickSettings[autopilotType][systemType].axesMaxRange);
            QMapIterator<int, float> joystickAxesMaxRange(joystickSettings[autopilotType][systemType].axesMinRange);
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
            QMapIterator<int, int> joystickButtonActions(joystickSettings[autopilotType][systemType].buttonActions);

            settings.beginWriteArray("AXES_INVERTED");
            int k = 0;
            while (joystickAxesInverted.hasNext())
            {
                joystickAxesInverted.next();
                int inverted = joystickAxesInverted.value();
                // Only save this axes' inversion status if it's not the default
                if (inverted)
                {
                    settings.setArrayIndex(k++);
                    int index = joystickAxesInverted.key();
                    settings.setValue("INDEX", index);
                    settings.setValue("INVERTED", inverted);
                }
            }
            settings.endArray();

243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
            settings.beginWriteArray("AXES_MIN_RANGE");
            k = 0;
            while (joystickAxesMinRange.hasNext())
            {
                joystickAxesMinRange.next();
                float min = joystickAxesMinRange.value();
                settings.setArrayIndex(k++);
                int index = joystickAxesMinRange.key();
                settings.setValue("INDEX", index);
                settings.setValue("MIN_RANGE", min);
            }
            settings.endArray();

            settings.beginWriteArray("AXES_MAX_RANGE");
            k = 0;
            while (joystickAxesMaxRange.hasNext())
            {
                joystickAxesMaxRange.next();
                float max = joystickAxesMaxRange.value();
                settings.setArrayIndex(k++);
                int index = joystickAxesMaxRange.key();
                settings.setValue("INDEX", index);
                settings.setValue("MAX_RANGE", max);
            }
            settings.endArray();

269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301
            settings.beginWriteArray("AXES_LIMITED");
            k = 0;
            while (joystickAxesLimited.hasNext())
            {
                joystickAxesLimited.next();
                int limited = joystickAxesLimited.value();
                if (limited)
                {
                    settings.setArrayIndex(k++);
                    int index = joystickAxesLimited.key();
                    settings.setValue("INDEX", index);
                    settings.setValue("LIMITED", limited);
                }
            }
            settings.endArray();

            settings.beginWriteArray("BUTTONS_ACTIONS");
            k = 0;
            while (joystickButtonActions.hasNext())
            {
                joystickButtonActions.next();
                int action = joystickButtonActions.value();
                if (action != -1)
                {
                    settings.setArrayIndex(k++);
                    int index = joystickButtonActions.key();
                    settings.setValue("INDEX", index);
                    settings.setValue("ACTION", action);
                }
            }
            settings.endArray();
        }
        settings.endArray(); // SYSTEMS
302
    }
303
    settings.endArray(); // AUTOPILOTS
304
    settings.endGroup();
305 306
}

pixhawk's avatar
pixhawk committed
307 308
void JoystickInput::setActiveUAS(UASInterface* uas)
{
309 310 311 312 313 314
    // Do nothing if the UAS hasn't changed.
    if (uas == this->uas)
    {
        return;
    }

pixhawk's avatar
pixhawk committed
315 316
    // Only connect / disconnect is the UAS is of a controllable UAS class
    UAS* tmp = 0;
LM's avatar
LM committed
317 318
    if (this->uas)
    {
pixhawk's avatar
pixhawk committed
319
        tmp = dynamic_cast<UAS*>(this->uas);
LM's avatar
LM committed
320 321
        if(tmp)
        {
322
            disconnect(this, SIGNAL(joystickChanged(float,float,float,float,qint8,qint8,quint16,quint8)), tmp, SLOT(setExternalControlSetpoint(float,float,float,float,qint8,qint8,quint16,quint8)));
323
            disconnect(this, SIGNAL(actionTriggered(int)), tmp, SLOT(triggerAction(int)));
pixhawk's avatar
pixhawk committed
324
        }
325
        uasCanReverse = false;
pixhawk's avatar
pixhawk committed
326 327
    }

328 329 330 331 332 333
    // Save any settings for the last UAS
    if (joystickID > -1)
    {
        storeJoystickSettings();
    }

pixhawk's avatar
pixhawk committed
334 335
    this->uas = uas;

336
    if (this->uas && (tmp = dynamic_cast<UAS*>(this->uas)))
LM's avatar
LM committed
337
    {
338 339
        connect(this, SIGNAL(joystickChanged(float,float,float,float,qint8,qint8,quint16,quint8)), tmp, SLOT(setExternalControlSetpoint(float,float,float,float,qint8,qint8,quint16,quint8)));
        qDebug() << "connected joystick";
340 341 342 343 344 345 346 347
        connect(this, SIGNAL(actionTriggered(int)), tmp, SLOT(triggerAction(int)));
        uasCanReverse = tmp->systemCanReverse();

        // Update the joystick settings for a new UAS.
        autopilotType = uas->getAutopilotType();
        systemType = uas->getSystemType();
    }

348 349 350 351
    // Make sure any UI elements know we've updated the UAS. The UASManager signal is re-emitted here so that UI elements know to
    // update their UAS-specific UI.
    emit activeUASSet(uas);

352 353 354 355
    // Load any joystick-specific settings now that the UAS has changed.
    if (joystickID > -1)
    {
        loadJoystickSettings();
356
    }
pixhawk's avatar
pixhawk committed
357 358
}

359 360 361 362 363 364
void JoystickInput::setEnabled(bool enabled)
{
    this->isEnabled = enabled;
    storeJoystickSettings();
}

pixhawk's avatar
pixhawk committed
365 366
void JoystickInput::init()
{
367
    // Initialize SDL Joystick support and detect number of joysticks.
368
    if (SDL_InitSubSystem(SDL_INIT_JOYSTICK | SDL_INIT_NOPARACHUTE) < 0) {
pixhawk's avatar
pixhawk committed
369 370 371 372
        printf("Couldn't initialize SimpleDirectMediaLayer: %s\n", SDL_GetError());
    }

    // Enumerate joysticks and select one
373
    numJoysticks = SDL_NumJoysticks();
pixhawk's avatar
pixhawk committed
374

375 376 377
    // If no joysticks are connected, there's nothing we can do, so just keep
    // going back to sleep every second unless the program quits.
    while (!numJoysticks && !done)
LM's avatar
LM committed
378
    {
379
        QGC::SLEEP::sleep(1);
pixhawk's avatar
pixhawk committed
380
    }
381 382 383 384
    if (done)
    {
        return;
    }
pixhawk's avatar
pixhawk committed
385

386 387 388 389
    // Now that we've detected a joystick, load in the joystick-agnostic settings.
    loadGeneralSettings();

    // Enumerate all found devices
390
    qDebug() << QString("%1 Input devices found:").arg(numJoysticks);
391
    int activeJoystick = 0;
392
    for(int i=0; i < numJoysticks; i++ )
LM's avatar
LM committed
393
    {
394
        QString name = SDL_JoystickName(i);
395
        qDebug() << QString("\t%1").arg(name);
396 397 398 399 400 401 402 403

        // If we've matched this joystick to what was last opened, note it.
        // Note: The way this is implemented the LAST joystick of a given name will be opened.
        if (name == joystickName)
        {
            activeJoystick = i;
        }

404
        SDL_Joystick* x = SDL_JoystickOpen(i);
405 406
        qDebug() << QString("\tNumber of Axes: %1").arg(QString::number(SDL_JoystickNumAxes(x)));
        qDebug() << QString("\tNumber of Buttons: %1").arg(QString::number(SDL_JoystickNumButtons(x)));
407
        SDL_JoystickClose(x);
pixhawk's avatar
pixhawk committed
408 409
    }

410 411 412
    // Set the active joystick based on name, if a joystick was found in the saved settings, otherwise
    // default to the first one.
    setActiveJoystick(activeJoystick);
413

414
    // Now make sure we know what the current UAS is and track changes to it.
415
    setActiveUAS(UASManager::instance()->getActiveUAS());
416
    connect(UASManager::instance(), SIGNAL(activeUASSet(UASInterface*)), this, SLOT(setActiveUAS(UASInterface*)));
pixhawk's avatar
pixhawk committed
417
}
418

419 420 421 422
void JoystickInput::shutdown()
{
    done = true;
}
pixhawk's avatar
pixhawk committed
423 424 425

/**
 * @brief Execute the Joystick process
426 427
 * Note that the SDL procedure is polled. This is because connecting and disconnecting while the event checker is running
 * fails as of SDL 1.2. It is therefore much easier to just poll for the joystick we want to sample.
pixhawk's avatar
pixhawk committed
428 429 430 431 432
 */
void JoystickInput::run()
{
    init();

433
    forever
LM's avatar
LM committed
434 435 436
    {
        if (done)
        {
437 438 439
            done = false;
            exit();
            return;
LM's avatar
LM committed
440
        }
pixhawk's avatar
pixhawk committed
441

442 443
        // Poll the joystick for new values.
        SDL_JoystickUpdate();
pixhawk's avatar
pixhawk committed
444

445
        // Emit all necessary signals for all axes.
446
        for (int i = 0; i < joystickNumAxes; i++)
447 448 449
        {
            // First emit the uncalibrated values for each axis based on their ID.
            // This is generally not used for controlling a vehicle, but a UI representation, so it being slightly off is fine.
450
            // Here we map the joystick axis value into the initial range of [0:1].
451
            float axisValue = SDL_JoystickGetAxis(joystick, i);
452 453 454 455 456 457 458 459 460 461 462 463 464 465 466

            // during calibration save min and max values
            if (isCalibrating)
            {
                if (joystickSettings[autopilotType][systemType].axesMinRange.value(i) > axisValue)
                {
                    joystickSettings[autopilotType][systemType].axesMinRange[i] = axisValue;
                }

                if (joystickSettings[autopilotType][systemType].axesMaxRange.value(i) < axisValue)
                {
                    joystickSettings[autopilotType][systemType].axesMaxRange[i] = axisValue;
                }
            }

467
            if (joystickSettings[autopilotType][systemType].axesInverted[i])
468
            {
469 470 471
                axisValue = (axisValue - joystickSettings[autopilotType][systemType].axesMinRange.value(i)) /
                        (joystickSettings[autopilotType][systemType].axesMaxRange.value(i) -
                         joystickSettings[autopilotType][systemType].axesMinRange.value(i));
472 473 474
            }
            else
            {
475 476 477
                axisValue = (axisValue - joystickSettings[autopilotType][systemType].axesMaxRange.value(i)) /
                        (joystickSettings[autopilotType][systemType].axesMinRange.value(i) -
                         joystickSettings[autopilotType][systemType].axesMaxRange.value(i));
478
            }
479
            axisValue = 1.0f - axisValue;
480

481
            // For non-throttle axes or if the UAS can reverse, go ahead and convert this into the range [-1:1].
482 483 484 485 486

            //if (uasCanReverse || throttleAxis != i)
            // don't take into account if UAS can reverse. This means to reverse position but not throttle
            // therefore deactivated for now
            if (throttleAxis != i)
487 488
            {
                axisValue = axisValue * 2.0f - 1.0f;
489
            }
490
            // Otherwise if this vehicle can only go forward, scale it to [0:1].
491
            else if (throttleAxis == i && joystickSettings[autopilotType][systemType].axesLimited.value(i))
492 493 494 495 496
            {
                if (axisValue < 0.0f)
                {
                    axisValue = 0.0f;
                }
497
            }
498 499 500 501

            // Bound rounding errors
            if (axisValue > 1.0f) axisValue = 1.0f;
            if (axisValue < -1.0f) axisValue = -1.0f;
502 503 504 505 506
            if (joystickAxes[i] != axisValue)
            {
                joystickAxes[i] = axisValue;
                emit axisValueChanged(i, axisValue);
            }
507
        }
pixhawk's avatar
pixhawk committed
508 509

        // Build up vectors describing the hat position
510
        int hatPosition = SDL_JoystickGetHat(joystick, 0);
511
        qint8 newYHat = 0;
512 513
        if ((SDL_HAT_UP & hatPosition) > 0) newYHat = 1;
        if ((SDL_HAT_DOWN & hatPosition) > 0) newYHat = -1;
514
        qint8 newXHat = 0;
515 516 517 518 519 520
        if ((SDL_HAT_LEFT & hatPosition) > 0) newXHat = -1;
        if ((SDL_HAT_RIGHT & hatPosition) > 0) newXHat = 1;
        if (newYHat != yHat || newXHat != xHat)
        {
            xHat = newXHat;
            yHat = newYHat;
521
            emit hatDirectionChanged(newXHat, newYHat);
522
        }
pixhawk's avatar
pixhawk committed
523

524
        // Emit signals for each button individually
525
        for (int i = 0; i < joystickNumButtons; i++)
LM's avatar
LM committed
526
        {
527
            // If the button was down, but now it's up, trigger a buttonPressed event
528
            quint16 lastButtonState = joystickButtons & (1 << i);
529
            if (SDL_JoystickGetButton(joystick, i) && !lastButtonState)
LM's avatar
LM committed
530
            {
pixhawk's avatar
pixhawk committed
531
                emit buttonPressed(i);
532
                joystickButtons |= 1 << i;
533 534 535 536
            }
            else if (!SDL_JoystickGetButton(joystick, i) && lastButtonState)
            {
                emit buttonReleased(i);
537 538 539 540
                if (isEnabled && joystickSettings[autopilotType][systemType].buttonActions.contains(i))
                {
                    emit actionTriggered(joystickSettings[autopilotType][systemType].buttonActions.value(i));
                }
541
                joystickButtons &= ~(1 << i);
pixhawk's avatar
pixhawk committed
542 543
            }
        }
544 545

        // Now signal an update for all UI together.
546 547
        if (isEnabled)
        {
548 549 550 551
            float roll = rollAxis > -1?joystickAxes[rollAxis]:numeric_limits<float>::quiet_NaN();
            float pitch = pitchAxis > -1?joystickAxes[pitchAxis]:numeric_limits<float>::quiet_NaN();
            float yaw = yawAxis > -1?joystickAxes[yawAxis]:numeric_limits<float>::quiet_NaN();
            float throttle = throttleAxis > -1?joystickAxes[throttleAxis]:numeric_limits<float>::quiet_NaN();
552
            emit joystickChanged(roll, pitch, yaw, throttle, xHat, yHat, joystickButtons, mode);
553
        }
pixhawk's avatar
pixhawk committed
554

555 556
        // Sleep, update rate of joystick is approx. 25 Hz (1000 ms / 25 = 40 ms)
        QGC::SLEEP::msleep(40);
pixhawk's avatar
pixhawk committed
557 558
    }
}
559

560 561
void JoystickInput::setActiveJoystick(int id)
{
562
    // If we already had a joystick, close that one before opening a new one.
563 564
    if (joystick && SDL_JoystickOpened(joystickID))
    {
565
        storeJoystickSettings();
566 567 568 569 570
        SDL_JoystickClose(joystick);
        joystick = NULL;
        joystickID = -1;
    }

571
    joystickID = id;
572
    joystick = SDL_JoystickOpen(joystickID);
573
    if (joystick && SDL_JoystickOpened(joystickID))
574
    {
575
        // Update joystick configuration.
576
        joystickName = QString(SDL_JoystickName(joystickID));
577 578 579
        joystickNumButtons = SDL_JoystickNumButtons(joystick);
        joystickNumAxes = SDL_JoystickNumAxes(joystick);

580
        // Restore saved settings for this joystick.
581
        loadJoystickSettings();
582

583 584
        // Update cached joystick axes values.
        // Also emit any signals for currently-triggering events
585
        joystickAxes.clear();
586
        for (int i = 0; i < joystickNumAxes; i++)
587
        {
588
            joystickAxes.append(numeric_limits<float>::quiet_NaN());
589
        }
590 591 592

        // Update cached joystick button values.
        // Emit signals for any button events.
593 594 595 596 597 598
        joystickButtons = 0;
    }
    else
    {
        joystickNumButtons = 0;
        joystickNumAxes = 0;
599
    }
600

601 602 603
    // Specify that a new joystick has been selected, so that any UI elements can update.
    emit newJoystickSelected();
    // And then trigger an update of this new UI.
604
    emit joystickSettingsChanged();
605 606
}

607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638
void JoystickInput::setCalibrating(bool active)
{
    if (active)
    {
        setEnabled(false);
        isCalibrating = true;

        // set range small so that limits can be re-found
        for (int i = 0; i < joystickNumAxes; i++)
        {
            joystickSettings[autopilotType][systemType].axesMinRange[i] = -10.0f;
            joystickSettings[autopilotType][systemType].axesMaxRange[i] = 10.0f;
        }

    } else {

        // store calibration values
        storeJoystickSettings();

        qDebug() << "Calibration result:";
        for (int i = 0; i < joystickNumAxes; i++)
        {
            qDebug() << i << ": " <<
                        joystickSettings[autopilotType][systemType].axesMinRange[i] <<
                        " - " <<
                        joystickSettings[autopilotType][systemType].axesMaxRange[i];
        }
        setEnabled(true);
        isCalibrating = false;
    }
}

639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
void JoystickInput::setAxisMapping(int axis, JOYSTICK_INPUT_MAPPING newMapping)
{
    switch (newMapping)
    {
        case JOYSTICK_INPUT_MAPPING_ROLL:
            rollAxis = axis;
            break;
        case JOYSTICK_INPUT_MAPPING_PITCH:
            pitchAxis = axis;
            break;
        case JOYSTICK_INPUT_MAPPING_YAW:
            yawAxis = axis;
            break;
        case JOYSTICK_INPUT_MAPPING_THROTTLE:
            throttleAxis = axis;
            break;
        case JOYSTICK_INPUT_MAPPING_NONE:
        default:
            if (rollAxis == axis)
            {
                rollAxis = -1;
            }
            if (pitchAxis == axis)
            {
                pitchAxis = -1;
            }
            if (yawAxis == axis)
            {
                yawAxis = -1;
            }
            if (throttleAxis == axis)
            {
                throttleAxis = -1;
672
                joystickSettings[autopilotType][systemType].axesLimited.remove(axis);
673 674 675
            }
            break;
    }
676
    storeJoystickSettings();
677 678 679 680
}

void JoystickInput::setAxisInversion(int axis, bool inverted)
{
681
    if (axis < joystickNumAxes)
682
    {
683 684
        joystickSettings[autopilotType][systemType].axesInverted[axis] = inverted;
        storeJoystickSettings();
685 686 687
    }
}

688 689
void JoystickInput::setAxisRangeLimit(int axis, bool limitRange)
{
690
    if (axis < joystickNumAxes)
691
    {
692 693
        joystickSettings[autopilotType][systemType].axesLimited[axis] = limitRange;
        storeJoystickSettings();
694 695 696
    }
}

697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714
void JoystickInput::setAxisRangeLimitMin(int axis, float min)
{
    if (axis < joystickNumAxes)
    {
        joystickSettings[autopilotType][systemType].axesMinRange[axis] = min;
        storeJoystickSettings();
    }
}

void JoystickInput::setAxisRangeLimitMax(int axis, float max)
{
    if (axis < joystickNumAxes)
    {
        joystickSettings[autopilotType][systemType].axesMaxRange[axis] = max;
        storeJoystickSettings();
    }
}

715
void JoystickInput::setButtonAction(int button, int action)
716
{
717
    if (button < joystickNumButtons)
718
    {
719 720 721 722 723 724 725 726 727 728
        joystickSettings[autopilotType][systemType].buttonActions[button] = action;
        storeJoystickSettings();
    }
}

float JoystickInput::getCurrentValueForAxis(int axis) const
{
    if (axis < joystickNumAxes)
    {
        return joystickAxes.at(axis);
729 730
    }
    return 0.0f;
731
}
732

733
bool JoystickInput::getInvertedForAxis(int axis) const
734
{
735
    if (axis < joystickNumAxes)
736
    {
737
        return joystickSettings[autopilotType][systemType].axesInverted.value(axis);
738 739 740 741
    }
    return false;
}

742
bool JoystickInput::getRangeLimitForAxis(int axis) const
743
{
744
    if (axis < joystickNumAxes)
745
    {
746
        return joystickSettings[autopilotType][systemType].axesLimited.value(axis);
747 748 749
    }
    return false;
}
750

751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768
float JoystickInput::getAxisRangeLimitMinForAxis(int axis) const
{
    if (axis < joystickNumAxes)
    {
        return joystickSettings[autopilotType][systemType].axesMinRange.value(axis);
    }
    return sdlJoystickMin;
}

float JoystickInput::getAxisRangeLimitMaxForAxis(int axis) const
{
    if (axis < joystickNumAxes)
    {
        return joystickSettings[autopilotType][systemType].axesMaxRange.value(axis);
    }
    return sdlJoystickMax;
}

769 770 771 772 773 774 775 776
int JoystickInput::getActionForButton(int button) const
{
    if (button < joystickNumButtons && joystickSettings[autopilotType][systemType].buttonActions.contains(button))
    {
        return joystickSettings[autopilotType][systemType].buttonActions.value(button);
    }
    return -1;
}