TreeModel.hpp

#pragma once
 
#include <Gui/RaGui.hpp>
 
#include <memory>
#include <vector>
 
#include <QAbstractItemModel>
#include <QVariant>
 
 
namespace Ra {
namespace Gui {
 
class RA_GUI_API TreeItem
{
  public:
    TreeItem() {}
    virtual ~TreeItem() {}
 
    TreeItem( const TreeItem& )            = delete;
    TreeItem& operator=( const TreeItem& ) = delete;
 
    virtual std::string getName() const = 0;
 
    virtual bool isValid() const = 0;
 
    virtual bool isSelectable() const = 0;
 
 
    int getIndexInParent() const;
 
    bool isChecked() const { return m_checked; }
    void setChecked( bool checked = true ) { m_checked = checked; }
  public:
    TreeItem* m_parent { nullptr };
 
    std::vector<std::unique_ptr<TreeItem>> m_children;
  private:
    bool m_checked { true };
};
 
class RA_GUI_API TreeModel : public QAbstractItemModel
{
    Q_OBJECT
 
  public:
    TreeModel( QObject* parent = nullptr ) : QAbstractItemModel( parent ) {}
 
 
    int rowCount( const QModelIndex& parent = QModelIndex() ) const override;
 
    int columnCount( const QModelIndex& parent = QModelIndex() ) const override {
        CORE_UNUSED( parent );
        return 1;
    }
 
    QVariant data( const QModelIndex& index, int role ) const override;
 
    bool setData( const QModelIndex& index, const QVariant& value, int role ) override;
 
    QVariant headerData( int section,
                         Qt::Orientation orientation,
                         int role = Qt::DisplayRole ) const override;
 
    QModelIndex
    index( int row, int column, const QModelIndex& parent = QModelIndex() ) const override;
 
    QModelIndex parent( const QModelIndex& child ) const override;
 
    Qt::ItemFlags flags( const QModelIndex& index ) const override;
 
 
    void rebuildModel();
 
  signals:
    void modelRebuilt();
 
  protected:
    virtual std::string getHeaderString() const = 0;
 
    virtual void buildModel() = 0;
 
    TreeItem* getItem( const QModelIndex& index ) const;
 
    bool findInTree( const TreeItem* item ) const;
 
    void printModel() const;
 
    void setAllItemsChecked( bool checked = true );
 
    void setItemChecked( const QModelIndex& index, bool checked = true );
 
  protected:
    std::unique_ptr<TreeItem> m_rootItem;
};
} // namespace Gui
} // namespace Ra