ComponentProxyModel.h

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      GUI/coregui/Models/ComponentProxyModel.h
//! @brief     Defines class ComponentProxyModel
//!
//! @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)
//
//  ************************************************************************************************
 
#ifndef BORNAGAIN_GUI_COREGUI_MODELS_COMPONENTPROXYMODEL_H
#define BORNAGAIN_GUI_COREGUI_MODELS_COMPONENTPROXYMODEL_H
 
#include "GUI/coregui/Models/ProxyModelStrategy.h"
#include <QAbstractProxyModel>
#include <QMap>
#include <QPersistentModelIndex>
#include <memory>
 
class SessionModel;
class ProxyModelStrategy;
 
//! Proxy model to adjust SessionModel for component editor (right bottom corner of SampleView
//! and similar).
//! The model hides all SessionItems which are not PropertyItems.
//! The model hides GroupPropertyItem children and shows grand-children of currently selected item
//! one level up.
 
class ComponentProxyModel : public QAbstractProxyModel {
    Q_OBJECT
 
    friend class ProxyModelStrategy;
 
public:
    ComponentProxyModel(QObject* parent = nullptr);
 
    void setSessionModel(SessionModel* model);
 
    void setRootIndex(const QModelIndex& sourceRootIndex);
 
    void setProxyStrategy(ProxyModelStrategy* strategy);
 
    QModelIndex mapToSource(const QModelIndex& proxyIndex) const;
    QModelIndex mapFromSource(const QModelIndex& sourceIndex) const;
 
    QModelIndex index(int row, int column, const QModelIndex& parent = {}) const;
    QModelIndex parent(const QModelIndex& child) const;
    int rowCount(const QModelIndex& parent = {}) const;
    int columnCount(const QModelIndex& parent = {}) const;
 
    bool hasChildren(const QModelIndex& parent) const;
 
private slots:
    void sourceDataChanged(const QModelIndex& topLeft, const QModelIndex& bottomRight,
                           const QVector<int>& roles = {});
    void sourceRowsInserted(const QModelIndex& parent, int start, int end);
    void sourceRowsRemoved(const QModelIndex& parent, int start, int end);
 
private:
    void buildModelMap();
    void updateModelMap();
 
    SessionModel* m_model;
    std::unique_ptr<ProxyModelStrategy> m_proxyStrategy;
};
 
#endif // BORNAGAIN_GUI_COREGUI_MODELS_COMPONENTPROXYMODEL_H