telemetryschedulergadgetwidget.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/>
 */

#ifndef TELEMETRYSCHEDULERGADGETWIDGET_H_
#define TELEMETRYSCHEDULERGADGETWIDGET_H_

#include <QMap>
#include <QSpinBox>
#include <QTableView>
#include <QStandardItemModel>
#include <QItemDelegate>
#include <QLabel>

#include "uavobjectutil/uavobjectutilmanager.h"
#include "extensionsystem/pluginmanager.h"
#include "uavobjects/uavobject.h"

class Ui_TelemetryScheduler;
class Ui_Metadata_Dialog;
class QFrozenTableViewWithCopyPaste;
class SchedulerModel;

class TelemetrySchedulerGadgetWidget : public QWidget
{
    Q_OBJECT

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

signals:

protected slots:

private slots:
    void saveTelemetryToFile();
    void loadTelemetryFromFile();

    QList<UAVMetaObject *> applySchedule();
    void saveSchedule();
    void onSavedSchedule(bool);
    void onCompletedMetadataWrite(bool);
    void onCompletedMetadataSave(int, bool);
    void updateCurrentColumn(UAVObject *);
    void dataModel_itemChanged(int col);
    void dataModel_itemChanged(QStandardItem *item);
    void addTelemetryColumn();
    void removeTelemetryColumn();
    void changeVerticalHeader(int);
    void changeHorizontalHeader(int);
    void customMenuRequested(QPoint pos);
    void uavoPresentOnHardwareChanged(UAVDataObject *);
    void onHideNotPresent(bool);

private:
    int stripMs(QVariant rate_ms);
    QList<UAVMetaObject *> metaObjectsToSave;
    void importTelemetryConfiguration(const QString &fileName);
    UAVObjectUtilManager *getObjectUtilManager();
    UAVObjectManager *getObjectManager();

    Ui_TelemetryScheduler *m_telemetryeditor;

    UAVObjectManager *objManager;
    QString filename;

    QMap<QString, UAVObject::Metadata> defaultMdata;
    QMap<UAVDataObject *, int> uavoIndex;

    QStringList columnHeaders;

    SchedulerModel *schedulerModel;
    QFrozenTableViewWithCopyPaste *telemetryScheduleView;
    QStandardItemModel *frozenModel;
};

class SchedulerModel : public QStandardItemModel
{
public:
    SchedulerModel(int rows, int cols, QObject *parent = nullptr)
        : QStandardItemModel(rows, cols, parent)
    {
    }

    ~SchedulerModel() {}

    // The first two columns are not editable, but all other columns are
    Qt::ItemFlags flags(const QModelIndex &index) const
    {
        if (index.column() == 0 || index.column() == 1)
            return nullptr;
        else
            return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
    }
};

class QFrozenTableViewWithCopyPaste : public QTableView
{
    Q_OBJECT

public:
    QFrozenTableViewWithCopyPaste(QAbstractItemModel *model);
    ~QFrozenTableViewWithCopyPaste();

    QStandardItemModel *getFrozenModel() { return frozenModel; }
    QTableView *getFrozenTableView() { return frozenTableView; }
    void setHorizontalHeaderItem(int column, QStandardItem *item);
    bool removeColumns(int column, int count, const QModelIndex &parent = QModelIndex());

    void fixGeometry(int value);

protected:
    virtual void keyPressEvent(QKeyEvent *event);

    virtual void resizeEvent(QResizeEvent *event);
    void scrollTo(const QModelIndex &index, ScrollHint hint = EnsureVisible);

private slots:
    void updateSectionWidth(int logicalIndex, int, int newSize);
    void updateSectionHeight(int logicalIndex, int, int newSize);

private:
    void copy();
    void paste();
    void deleteCells();

    void updateFrozenTableGeometry(int verticalHeaderWidth);
    void init();
    QTableView *frozenTableView;
    QStandardItemModel *frozenModel;
};

class SpinBoxDelegate : public QItemDelegate
{
    Q_OBJECT

public:
    SpinBoxDelegate(QObject *parent = nullptr);

    QWidget *createEditor(QWidget *parent, const QStyleOptionViewItem &option,
                          const QModelIndex &index) const;

    void setEditorData(QWidget *editor, const QModelIndex &index) const;
    void setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const;

    void updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option,
                              const QModelIndex &index) const;
};

#endif /* TELEMETRYSCHEDULERGADGETWIDGET_H_ */