SessionModelView.cpp

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      GUI/View/Toplevel/SessionModelView.cpp
//! @brief     Implements class SessionModelView
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************
 
#include "GUI/View/Toplevel/SessionModelView.h"
#include "GUI/Model/Model/JobModel.h"
#include "GUI/Model/Model/RealDataModel.h"
#include "GUI/Model/Sample/ItemWithMaterial.h"
#include "GUI/Model/Sample/MaterialItems.h"
#include "GUI/Util/ComboProperty.h"
#include "GUI/View/Project/ProjectManager.h"
#include "GUI/View/Tool/StyleUtils.h"
#include "GUI/View/Toplevel/SessionDecorationModel.h"
#include <QApplication>
#include <QStyledItemDelegate>
#include <QToolBar>
#include <QToolButton>
#include <QTreeView>
#include <QVBoxLayout>
 
namespace {
 
//! The SessionModelDelegate class presents the content of SessionModel items in
//! standard QTreeView. Extents base QItemDelegate with possibility to show/edit
//! our custom QVariant's.
 
class SessionModelDelegate : public QStyledItemDelegate {
public:
    SessionModelDelegate(QObject* parent)
        : QStyledItemDelegate(parent)
    {
    }
 
    void paint(QPainter* painter, const QStyleOptionViewItem& option,
               const QModelIndex& index) const override
    {
        if (index.data().canConvert<ComboProperty>())
            paintCustomLabel(painter, option, index, index.data().value<ComboProperty>().label());
        else
            QStyledItemDelegate::paint(painter, option, index);
    }
 
private:
    void paintCustomLabel(QPainter* painter, const QStyleOptionViewItem& option,
                          const QModelIndex& index, const QString& text) const
    {
        QStyleOptionViewItem opt = option;
        initStyleOption(&opt, index); // calling original method to take into accounts colors etc
        opt.text = displayText(text, option.locale); // by overriding text with ours
        const QWidget* widget = opt.widget;
        QStyle* style = widget ? widget->style() : QApplication::style();
        style->drawControl(QStyle::CE_ItemViewItem, &opt, painter, widget);
    }
};
 
} // namespace
 
SessionModelView::SessionModelView(QWidget* parent, ProjectDocument* document)
    : QWidget(parent)
{
    auto* layout = new QVBoxLayout(this);
    layout->setMargin(0);
 
    auto* tabs = new QTabWidget(this);
    auto* m_expandButton = new QToolButton(this);
    m_expandButton->setText("Expand tree");
    m_expandButton->setIcon(QIcon(":/images/toolbar_expand_collapse_tree.svg"));
    m_expandButton->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
    connect(m_expandButton, &QToolButton::clicked, [=]() {
        auto* tree = static_cast<QTreeView*>(tabs->widget(tabs->currentIndex()));
        tree->expandAll();
        tree->resizeColumnToContents(0);
        tree->resizeColumnToContents(1);
    });
 
    auto* m_collapseButton = new QToolButton(this);
    m_collapseButton->setText("Collapse tree");
    m_collapseButton->setIcon(QIcon(":/images/toolbar_expand_collapse_tree.svg"));
    m_collapseButton->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
    connect(m_collapseButton, &QToolButton::clicked, [=]() {
        auto* tree = static_cast<QTreeView*>(tabs->widget(tabs->currentIndex()));
        tree->collapseAll();
    });
 
    auto* toolbar = new QToolBar;
    toolbar->addWidget(m_expandButton);
    toolbar->addWidget(m_collapseButton);
    layout->addWidget(toolbar);
    layout->addWidget(tabs);
 
    QList<SessionModel*> models{document->realDataModel(), document->jobModel()};
 
    for (auto* model : models) {
        auto* treeView = new QTreeView(this);
        GUI::Util::Style::setPropertyStyle(treeView);
        treeView->setItemDelegate(new SessionModelDelegate(this));
        treeView->setModel(new SessionDecorationModel(treeView, model));
        treeView->setEditTriggers(QAbstractItemView::NoEditTriggers); // read-only!
        tabs->addTab(treeView, model->getModelTag());
    }
}