biascalibrationpage.cpp

Go to the documentation of this file.

/*
 * 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, see <http://www.gnu.org/licenses/>
 */

#include <QMessageBox>
#include <QDebug>
#include "biascalibrationpage.h"
#include "ui_biascalibrationpage.h"
#include "setupwizard.h"

BiasCalibrationPage::BiasCalibrationPage(SetupWizard *wizard, QWidget *parent)
    : AbstractWizardPage(wizard, parent)
    , ui(new Ui::BiasCalibrationPage)
    , m_calibrationUtil(nullptr)
{
    ui->setupUi(this);
    connect(ui->levelButton, &QAbstractButton::clicked, this,
            &BiasCalibrationPage::performCalibration);
}

BiasCalibrationPage::~BiasCalibrationPage()
{
    delete ui;
}

bool BiasCalibrationPage::validatePage()
{
    return true;
}

bool BiasCalibrationPage::isComplete() const
{
    return ui->levelButton->isEnabled();
}

void BiasCalibrationPage::enableButtons(bool enable)
{
    ui->levelButton->setEnabled(enable);
    getWizard()->button(QWizard::NextButton)->setEnabled(enable);
    getWizard()->button(QWizard::CancelButton)->setEnabled(enable);
    getWizard()->button(QWizard::BackButton)->setEnabled(enable);
    getWizard()->button(QWizard::CustomButton1)->setEnabled(enable);
    QApplication::processEvents();
}

void BiasCalibrationPage::performCalibration()
{
    if (!getWizard()->getConnectionManager()->isConnected()) {
        QMessageBox msgBox(this);
        msgBox.setText(tr("A flight controller must be connected to your computer to perform bias "
                          "calculations.\nPlease connect your flight controller to your computer "
                          "and try again."));
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
        return;
    }

    // Get the calibration helper object
    if (!m_calibrationUtil) {
        ExtensionSystem::PluginManager *pm = ExtensionSystem::PluginManager::instance();
        m_calibrationUtil = pm->getObject<Calibration>();
        Q_ASSERT(m_calibrationUtil);
        if (!m_calibrationUtil)
            return;
    }

    enableButtons(false);
    ui->progressLabel->setText(QString(tr("Retrieving data...")));

    QTimer *timer = new QTimer(this);
    timer->setSingleShot(true);
    timer->setInterval(30000);

    connect(m_calibrationUtil, &Calibration::levelingProgressChanged, this,
            &BiasCalibrationPage::calibrationProgress);
    connect(m_calibrationUtil, &Calibration::calibrationCompleted, this,
            &BiasCalibrationPage::calibrationDone);
    connect(m_calibrationUtil, &Calibration::calibrationCompleted, timer, &QTimer::stop);
    connect(timer, &QTimer::timeout, this, &BiasCalibrationPage::calibrationTimeout);
    timer->start();
    m_calibrationUtil->doStartBiasAndLeveling();
}

void BiasCalibrationPage::calibrationProgress(int current)
{
    const int total = 100;
    if (ui->levellinProgressBar->maximum() != (int)total) {
        ui->levellinProgressBar->setMaximum((int)total);
    }
    if (ui->levellinProgressBar->value() != (int)current) {
        ui->levellinProgressBar->setValue((int)current);
    }
}

void BiasCalibrationPage::calibrationDone()
{
    disconnect(this, SLOT(calibrationTimeout()));
    stopCalibration();
    emit completeChanged();
}

void BiasCalibrationPage::calibrationTimeout()
{
    stopCalibration();

    QMessageBox msgBox(this);
    msgBox.setText(tr("Calibration timed out"));
    msgBox.setStandardButtons(QMessageBox::Ok);
    msgBox.setDefaultButton(QMessageBox::Ok);
    msgBox.exec();
}

void BiasCalibrationPage::stopCalibration()
{
    if (m_calibrationUtil) {
        disconnect(m_calibrationUtil, &Calibration::levelingProgressChanged, this,
                   &BiasCalibrationPage::calibrationProgress);
        disconnect(m_calibrationUtil, &Calibration::calibrationCompleted, this,
                   &BiasCalibrationPage::calibrationDone);
        ui->progressLabel->setText(QString(tr("<font color='green'>Done!</font>")));
        enableButtons(true);
    }
}