dissector_tables_model.h

Go to the documentation of this file.

 
#ifndef DISSECTOR_TABLES_MODEL_H
#define DISSECTOR_TABLES_MODEL_H
 
#include <config.h>
 
#include <ui/qt/models/tree_model_helpers.h>
 
#include <QSortFilterProxyModel>

class DissectorTablesItem : public ModelHelperTreeItem<DissectorTablesItem>
{
public:
    DissectorTablesItem(QString tableName, QString dissectorDescription, DissectorTablesItem* parent);

    virtual ~DissectorTablesItem();

    QString tableName() const {return tableName_;}

    QString dissectorDescription() const {return dissectorDescription_;}

    virtual bool lessThan(DissectorTablesItem &right) const;
 
protected:
    QString tableName_;

    QString dissectorDescription_;
};

class DissectorTablesModel : public QAbstractItemModel
{
    Q_OBJECT
 
public:
    explicit DissectorTablesModel(QObject * parent = Q_NULLPTR);

    virtual ~DissectorTablesModel();

    enum DissectorTablesColumn {
        colTableName = 0,           
        colDissectorDescription,    
        colLast                     
    };

    QModelIndex index(int row, int column,
                      const QModelIndex &parent = QModelIndex()) const override;

    QModelIndex parent(const QModelIndex &index) const override;

    QVariant data(const QModelIndex &index, int role) const override;

    int rowCount(const QModelIndex &parent = QModelIndex()) const override;

    int columnCount(const QModelIndex &parent = QModelIndex()) const override;

    void populate();
 
private:
    DissectorTablesItem* root_;
};

class DissectorTablesProxyModel : public QSortFilterProxyModel
{
    Q_OBJECT
 
public:
    enum SearchType
    {
        EveryWhere = 0x0,       
        OnlyName = 0x1,         
        OnlyDescription = 0x2,  
    };
    Q_ENUM(SearchType)
    Q_DECLARE_FLAGS(SearchTypes, SearchType)
 
    
    explicit DissectorTablesProxyModel(QObject * parent = Q_NULLPTR);

    virtual bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const override;

    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;

    void adjustHeader(const QModelIndex &currentIndex);

    void setFilter(const QString& filter, SearchTypes type);
 
protected:
    bool lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const override;
 
private:

    QString tableName_;

    QString dissectorDescription_;

    QString filter_;
 
    DissectorTablesProxyModel::SearchTypes type_;
};
 
#endif // DISSECTOR_TABLES_MODEL_H