uploadergadgetwidget.h

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/>
 *
 * Additional note on redistribution: The copyright and license notices above
 * must be maintained in each individual source file that is a derivative work
 * of this source file; otherwise redistribution is prohibited.
 */

#ifndef UPLOADERGADGETWIDGET_H
#define UPLOADERGADGETWIDGET_H

#include <QPointer>
#include "tl_dfu.h"
#include <coreplugin/iboardtype.h>
#include "uploader_global.h"
#include "uavobjectutil/devicedescriptorstruct.h"
#include "uavobjectutil/uavobjectutilmanager.h"
#include "uavtalk/telemetrymanager.h"
#include "coreplugin/connectionmanager.h"
#include "uavsettingsimportexport/uavsettingsimportexportmanager.h"
#include "upgradeassistantdialog.h"

class QNetworkAccessManager;
class Ui_UploaderWidget;

using namespace tl_dfu;

namespace uploader {

typedef enum { STATUSICON_OK, STATUSICON_RUNNING, STATUSICON_FAIL, STATUSICON_INFO } StatusIcon;

class UPLOADER_EXPORT UploaderGadgetWidget : public QWidget
{
    Q_OBJECT

public:
    UploaderGadgetWidget(QWidget *parent = nullptr);
    ~UploaderGadgetWidget();

    bool active() const;

signals:
    void newBoardSeen(deviceInfo board, deviceDescriptorStruct device);
    void enteredLoader();
private slots:
    void onAutopilotConnect();
    void onAutopilotDisconnect();
    void onAutopilotReady();
    void onIAPUpdated();
    void onLoadFirmwareButtonClick();
    void onFlashButtonClick();
    void onRescueButtonClick();
    void onExportButtonClick();
    void onBootloaderDetected();
    void onBootloaderRemoved();
    void onRescueTimer();
    void onStatusUpdate(QString, int);
    void onPartitionSave();
    void onPartitionFlash();
    void onPartitionErase();
    void onBootButtonClick();
    void openHelp();
    void onDeviceDiscovered();

private:
    void FirmwareOnDeviceClear(bool clear);
    void FirmwareLoadedClear(bool clear);
    void PartitionBrowserClear();
    void DeviceInformationClear();
    void DeviceInformationUpdate(deviceInfo board);
    void FirmwareOnDeviceUpdate(deviceDescriptorStruct firmware, QString crc);
    void FirmwareLoadedUpdate(QByteArray firmwareArray);
    QString LoadFirmwareFileDialog(QString);
    uploader::UploaderStatus getUploaderStatus() const;
    void setUploaderStatus(const uploader::UploaderStatus &value);
    void CheckAutopilotReady();
    bool CheckInBootloaderState();
    /* XXX TODO: make capitalization consistent */
    void setStatusInfo(QString str, uploader::StatusIcon ic);
    QString getImagePath(QString boardName, QString imageType = QString("fw"));
    bool FirmwareLoadFromFile(QString filename, QByteArray *contents);
    bool FirmwareCheckForUpdate(deviceDescriptorStruct device);
    void triggerPartitionDownload(int index);
    void haltOrReset(bool halting);
    void startRescue();
    bool tradeSettingsWithCloud(QString srcRelease, QString ancestor, bool upgrading = false,
                                QByteArray *settingsOut = NULL);
    int isCloudReleaseAvailable(QString srcRelease);

    bool saveSettings(const QByteArray &settingsDump);

    bool askIfShouldContinue();
    bool downloadSettings();
    void stepChangeAndDelay(QEventLoop &loop, int delayMs,
                            UpgradeAssistantDialog::UpgradeAssistantStep step);
    void doUpgradeOperation(bool blankFC, tl_dfu::device &dev);
    void upgradeError(QString why);
    bool flashFirmware(QByteArray &firmwareImage);
    bool haveSettingsPart() const;

    Ui_UploaderWidget *m_widget;

    UpgradeAssistantDialog m_dialog;

    bool telemetryConnected;
    bool iapUpdated;

    QByteArray loadedFile;
    QByteArray settingsDump;

    DFUObject dfu;
    USBSignalFilter *usbFilterBL;
    USBSignalFilter *usbFilterUP;
    ExtensionSystem::PluginManager *pm;
    TelemetryManager *telMngr;
    UAVObjectUtilManager *utilMngr;
    Core::ConnectionManager *conMngr;
    QNetworkAccessManager *netMngr;
    UAVSettingsImportExportManager *importMngr;

    FirmwareIAPObj *firmwareIap;
    deviceInfo currentBoard;
    QString ignoredRev;

    uploader::UploaderStatus uploaderStatus;
    QByteArray tempArray;
    QTimer rescueTimer;
    bool upgraderActive = false;

    const QString exportUrl = QString("http://dronin-autotown.appspot.com/convert");
    const QString hasRevUrl = QString("http://dronin-autotown.appspot.com/uavos/%1");
};
}
#endif // UPLOADERGADGETWIDGET_H