dbselectView.h

Go to the documentation of this file.

#pragma once
 
// gemc
#include "gdetectorConstruction.h"
#include "gbase.h"
 
// qt
#include <QWidget>
#include <QTreeView>
#include <QStandardItem>
#include <QLabel>
#include <QPushButton>
#include <QComboBox>
#include <QStyledItemDelegate>
 
// c++
#include <sqlite3.h>
#include <string>
#include <vector>
 
 
class DBSelectView : public QWidget, public GBase<DBSelectView> {
    Q_OBJECT
public:
    explicit DBSelectView(const std::shared_ptr<GOptions>& gopts,
                          GDetectorConstruction* dc,
                          QWidget* parent = nullptr);
 
    ~DBSelectView() override { if (db) sqlite3_close(db); }
 
    SystemList get_gsystems();
 
private:
    void setupUI();
 
    void loadExperiments();
 
    void loadSystemsForExperiment(QStandardItem* experimentItem);
 
    QStringList getAvailableVariations(const std::string& system) const;
 
    QStringList getAvailableRuns(const std::string& system);
 
    int getGeometryCount(const std::string& system, const std::string& variation, int run) const;
 
    void updateSystemItemAppearance(QStandardItem* systemItem);
 
    bool systemAvailable(const std::string& system, const std::string& variation, int run);
 
    void updateExperimentHeader();
 
    void updateModifiedUI();
 
    QIcon createStatusIcon(const QColor& color);
 
    bool isGeometryTableValid() const;
 
    void applyGSystemSelections();
 
private:
    bool modified = false;
 
    QLabel* titleLabel = nullptr;
 
    QPushButton* reloadButton = nullptr;
 
    sqlite3* db = nullptr;
 
    std::string dbhost;
 
    std::string experiment;
 
    QTreeView* experimentTree = nullptr;
 
    QStandardItemModel* experimentModel = nullptr;
 
    QLabel* experimentHeaderLabel = nullptr;
 
    bool m_ignoreItemChange = false;
 
    GDetectorConstruction* gDetectorConstruction = nullptr;
 
    std::shared_ptr<GOptions> gopt;
 
private slots:
    void onItemChanged(QStandardItem* item);
 
public slots:
    void reload_geometry();
};
 
 
//
// A custom delegate for drop-down editing (for both variations and runs).
//
class ComboDelegate : public QStyledItemDelegate {
    Q_OBJECT
public:
    explicit ComboDelegate(QObject* parent = nullptr) : QStyledItemDelegate(parent) {}
 
    QWidget* createEditor(QWidget* parent,
                          const QStyleOptionViewItem& option,
                          const QModelIndex& index) const override {
        Q_UNUSED(option);
        Q_UNUSED(index);
        auto editor = new QComboBox(parent);
        return editor;
    }
 
    void setEditorData(QWidget* editor, const QModelIndex& index) const override {
        auto combo = qobject_cast<QComboBox*>(editor);
        if (!combo)
            return;
 
        // Retrieve list from UserRole.
        QVariant var     = index.model()->data(index, Qt::UserRole);
        QStringList opts = var.toStringList();
 
        combo->clear();
        combo->addItems(opts);
 
        QString currentText = index.model()->data(index, Qt::EditRole).toString();
        int idx = combo->findText(currentText);
        if (idx >= 0)
            combo->setCurrentIndex(idx);
    }
 
    void setModelData(QWidget* editor, QAbstractItemModel* model,
                      const QModelIndex& index) const override {
        auto combo = qobject_cast<QComboBox*>(editor);
        if (!combo)
            return;
 
        QString value = combo->currentText();
        model->setData(index, value, Qt::EditRole);
    }
 
    void updateEditorGeometry(QWidget* editor,
                              const QStyleOptionViewItem& option,
                              const QModelIndex& index) const override {
        Q_UNUSED(index);
        editor->setGeometry(option.rect);
    }
};