coloring_rules_model.h

Go to the documentation of this file.

 
#ifndef COLORING_RULES_MODEL_H
#define COLORING_RULES_MODEL_H
 
#include <config.h>
 
#include <epan/color_filters.h>
 
#include <ui/qt/models/tree_model_helpers.h>
 
#include <QList>
#include <QColor>
#include <QAbstractTableModel>
#include <QSortFilterProxyModel>

class ColoringRuleItem : public ModelHelperTreeItem<ColoringRuleItem>
{
public:
    ColoringRuleItem(bool disabled, QString name, QString filter, QColor foreground, QColor background, ColoringRuleItem* parent);

    virtual ~ColoringRuleItem();

    ColoringRuleItem(color_filter_t *colorf, ColoringRuleItem* parent);

    ColoringRuleItem(const ColoringRuleItem& item);

    bool disabled_;

    QString name_;

    QString filter_;

    QColor foreground_;

    QColor background_;

    ColoringRuleItem& operator=(ColoringRuleItem& rhs);
 
};

class ColoringRulesModel : public QAbstractItemModel
{
    Q_OBJECT
 
public:
    ColoringRulesModel(QColor defaultForeground, QColor defaultBackground, QObject *parent);

    virtual ~ColoringRulesModel();

    enum ColoringRulesColumn {
        colName = 0,            
        colFilter,              
        colColoringRulesMax     
    };

    void addColor(color_filter_t* colorf);

    void addColor(bool disabled, QString filter, QColor foreground, QColor background);

    bool importColors(QString filename, QString& err);

    bool exportColors(QString filename, QString& err);

    bool writeColors(QString& err);

    Qt::ItemFlags flags(const QModelIndex &index) const override;

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

    bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) 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 &indexItem) const override;
 
    //Drag & drop functionality
    Qt::DropActions supportedDropActions() const override;

    QStringList mimeTypes() const override;

    QMimeData* mimeData(const QModelIndexList &indexes) const override;

    bool dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) override;

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

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

    bool insertRows(int row, int count, const QModelIndex &parent = QModelIndex()) override;

    bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex()) override;

    bool copyRow(int dst_row, int src_row);
 
private:
    void populate();

    struct _GSList *createColorFilterList();

    ColoringRuleItem* root_;
 
    //Save off the conversation colors, do not include in dialog
    struct _GSList *conversation_colors_;

    QColor defaultForeground_;

    QColor defaultBackground_;

    QList<int> dragDropRows_;
};
 
#endif // COLORING_RULES_MODEL_H