calibrationeditor.h

Go to the documentation of this file.

#ifndef CALIBRATIONEDITOR_H
#define CALIBRATIONEDITOR_H

#include "treemodel.h"
#include "treeitem.h"
#include "tableeditor.h"
#include "baseeditor.h"
#include "global.h"

class AdjustedTreeModel : public ProjectTreeModel {
public:
    AdjustedTreeModel(QObject *parent = 0) : ProjectTreeModel(parent){}
    virtual int columnCount(const QModelIndex &parent = QModelIndex()) const {
        return 1;
    }
    virtual Qt::ItemFlags flags (const QModelIndex & index) const {
        Qt::ItemFlags flags = Qt::ItemIsEnabled;
        if (index.data(ProjectTreeModel::TreeItemPointer).value<TreeItem*>()->data(TreeItem::TreeItemColumnName).toString() == QString("sens_chan_nr"))
            flags = flags | Qt::ItemIsSelectable;
        return flags;
    }
};

#include <QStyledItemDelegate>
#include <QSqlTableModel>
#include <QDateTimeEdit>

class CalibrationDelegate : public QStyledItemDelegate {
public:
    CalibrationDelegate(QObject *parent = 0) : QStyledItemDelegate(parent){}

    QWidget *createEditor(QWidget *parent, const QStyleOptionViewItem &option,
                          const QModelIndex &index) const {
        if (QSqlTableModel *model = dynamic_cast<QSqlTableModel*>(
                const_cast<QAbstractItemModel*>(index.model()))) {
            int timestampColumn = model->record().indexOf("sens_chan_cali_timestamp");
            int formulaColumn = model->record().indexOf("sens_chan_cali_formula");
            int indexColumn = index.column();
            if (indexColumn == timestampColumn) {
                QDateTimeEdit *editor = new QDateTimeEdit(parent);
                editor->setCalendarPopup(true);
                return editor;
            }
            if (indexColumn == formulaColumn) {
                // nog te implementeren...
                // ...een CalibrationFormulaEditor of zo iets
            }
        }
        return QStyledItemDelegate::createEditor(parent, option, index);
    }
};

class CalibrationEditor : public TableBaseEditor {
public:
    CalibrationEditor(QWidget *parent = 0) : TableBaseEditor(parent){
        table()->setItemDelegate(new CalibrationDelegate(table()));
        model()->setTable("tbl_sensor_channel_calibrations");
    }
};

#include <QTreeView>
#include <QSqlIndex>
#include <QLabel>

class ProjectCalibrationEditor : public CalibrationEditor {
    Q_OBJECT
public:
    ProjectCalibrationEditor(QWidget *parent = 0) : CalibrationEditor(parent) {
        setObjectName("ProjectCalibrationEditorEditor");
        setWindowTitle("Calibrations editor, project based");
        treeView = new QTreeView(this);
        treeView->setModel(new AdjustedTreeModel(treeView));

        QSplitter *splitter = new QSplitter(this);
        splitter->addWidget(treeView);
        splitter->addWidget(table());
        splitter->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

        QLabel *note = new QLabel(this);
        note->setTextInteractionFlags(Qt::LinksAccessibleByMouse);
        note->setOpenExternalLinks(true);
        note->setTextFormat(Qt::RichText);
        note->setWordWrap(true);
        note->setText(GlobalConst::CALI_FORMULA_HELP);

        QGridLayout *widgetLayout = new QGridLayout();
        widgetLayout->addWidget(splitter, 0, 0, 1, 1);
        widgetLayout->addWidget(note, 1, 0, 1, 1);

        // main layout setup
        mainLayout()->addLayout(widgetLayout, 0, 0, 1, 1);

        // setup connections
        connect(treeView->selectionModel(), SIGNAL(selectionChanged(QItemSelection,QItemSelection)),
                this, SLOT(onSensorChannelSelection(QItemSelection,QItemSelection)));

        // initialize
        onSensorChannelSelection(QItemSelection(), QItemSelection());

        // setup connections
        connect(table()->itemDelegate(), SIGNAL(closeEditor(QWidget*)), this, SLOT(updateButtons()));
    }

protected slots:
    void onSensorChannelSelection(const QItemSelection &newSelection, const QItemSelection&){
        int tree_id = 0;
        QString filter("(sens_id, sens_chan_nr) IN (SELECT DISTINCT sens_id, sens_chan_nr FROM tbl_trees INNER JOIN tbl_sensor_channel_calibrations USING (sens_id, sens_chan_nr) WHERE tree_id = %1)");
        bool channelIsSelected = isChannelSelected();

        table()->setEnabled(channelIsSelected);
        if (!channelIsSelected) {
            // if there is no selection
            tree_id = 0;
        } else {
            // if there is a selection
            tree_id = newSelection.indexes().first().data(ProjectTreeModel::TreeId).toInt();

            // sanity check
            Q_ASSERT(newSelection.indexes().count() == 1);
            Q_ASSERT(tree_id != 0);
            Q_ASSERT(tree_id != -1);
        }

        // There is a minor problem here: once a non-empty model was selected(),
        // and an other empty is selected afterwards, then the header
        // will display questionmarks.
        model()->setTable(channelIsSelected ? "tbl_sensor_channel_calibrations" : QString());
        model()->setFilter(filter.arg(QString::number(tree_id)));
        model()->setSort(model()->record().indexOf("sens_chan_cali_timestamp"), Qt::AscendingOrder);
        model()->select();

        // hide pk column
        QSqlIndex pk = model()->primaryKey();
        for(int i = 0; i < pk.count() ; i++){
            table()->setColumnHidden(model()->record().indexOf(pk.fieldName(i)), true);
        }

        // hide sens_id & sens_chan_nr columns
        table()->setColumnHidden(model()->record().indexOf("sens_id"), true);
        table()->setColumnHidden(model()->record().indexOf("sens_chan_nr"), true);
    }

    virtual void updateButtons() {
        CalibrationEditor::updateButtons();
        treeView->setEnabled(!isDirty());
    }

    virtual void onNewButtonPress() {
        CalibrationEditor::onNewButtonPress();
        Q_ASSERT(isChannelSelected());
        Q_ASSERT(table()->isEnabled());

        // fill in foreign key of the currently selected tree_id
        bool ok;
        int tree_id = treeView->selectionModel()->selection().indexes().first().data(ProjectTreeModel::TreeId).toInt(&ok);
        Q_ASSERT(ok);
        QSqlRecord key = Database::execStaticR(QString::number(tree_id).prepend("SELECT DISTINCT sens_id, sens_chan_nr FROM tbl_trees WHERE tree_id = "));
        int sens_id = key.value("sens_id").toInt(&ok);
        Q_ASSERT(ok);
        int sens_chan_nr = key.value("sens_chan_nr").toInt(&ok);
        Q_ASSERT(ok);
        model()->setData(model()->index(model()->rowCount() - 1, model()->record().indexOf("sens_id")), sens_id);
        model()->setData(model()->index(model()->rowCount() - 1, model()->record().indexOf("sens_chan_nr")), sens_chan_nr);
    }

    bool isChannelSelected(){
        Q_ASSERT(treeView->selectionModel()->selection().indexes().count() <= 1);
        return !treeView->selectionModel()->selection().indexes().isEmpty();
    }

private:
    QTreeView *treeView;
};

#include <QSqlRelationalTableModel>
#include <QSqlRelationalDelegate>
#include <QSplitter>
#include <QGroupBox>

class UnitConversionEditor : public TableBaseEditor {
    Q_OBJECT
public:
    UnitConversionEditor(QWidget *parent = 0) : TableBaseEditor(parent) {
        setObjectName("UnitConversionEditor");
        setWindowTitle("Unit conversion editor");
        model()->setTable("tbl_unit_conversions");
        model()->setRelation(model()->record().indexOf("unit_id_from"), QSqlRelation("tbl_units", "unit_id", "unit_symbol"));
        model()->setRelation(model()->record().indexOf("unit_id_to"), QSqlRelation("tbl_units", "unit_id", "unit_symbol"));
        model()->setSort(0, Qt::AscendingOrder);
        model()->select();

        model()->setHeaderData(0, Qt::Horizontal, "From*", Qt::DisplayRole);
        model()->setHeaderData(1, Qt::Horizontal, "To*", Qt::DisplayRole);
        model()->setHeaderData(2, Qt::Horizontal, "Factor*", Qt::DisplayRole);
        model()->setHeaderData(3, Qt::Horizontal, "Offset*", Qt::DisplayRole);

        editor = new QTableView(this);
        editor->setItemDelegate(new QSqlRelationalDelegate());
        editor->setModel(model());

        SqlQueryModel *m = new SqlQueryModel(this);
        m->setQuery(Database::execStatic("SELECT f.unit_symbol AS \"From*\", t.unit_symbol AS \"To*\", unit_conv_factor as \"Factor\", unit_conv_offset AS \"Offset\" "
                                         "FROM vew_unit_conversions AS c INNER JOIN tbl_units AS f ON (unit_id_from = f.unit_id) INNER JOIN tbl_units AS t ON (unit_id_to = t.unit_id) "
                                         "WHERE f.unit_symbol <> t.unit_symbol "
                                         "ORDER BY 1, 2"));
        QTableView *view = new QTableView(this);
        view->setModel(m);

        QGroupBox *eView = new QGroupBox("Editable conversions");
        eView->setLayout(new QHBoxLayout());
        eView->layout()->addWidget(editor);
        QGroupBox *sView = new QGroupBox("Derived conversions");
        sView->setLayout(new QHBoxLayout());
        sView->layout()->addWidget(view);

        QSplitter *splitter = new QSplitter(this);
        splitter->addWidget(sView);
        splitter->addWidget(eView);
        //splitter->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

        // main layout setup
        mainLayout()->addWidget(splitter, 0, 0, 1, 1);

        // setup connections
        connect(editor->itemDelegate(), SIGNAL(closeEditor(QWidget*)), this, SLOT(updateButtons()));
        connect(model(), SIGNAL(queryChanged()), m, SLOT(reQuery()));
    }

protected slots:
private:
    QTableView *editor;
};

#endif // CALIBRATIONEDITOR_H