traffic_tree.h

Go to the documentation of this file.

 
#ifndef TRAFFIC_TREE_H
#define TRAFFIC_TREE_H
 
#include "config.h"
 
#include <ui/recent.h>
 
#include <ui/qt/models/atap_data_model.h>
#include <ui/qt/filter_action.h>
 
#include <QTreeView>
#include <QMenu>
#include <QHeaderView>
#include <QSortFilterProxyModel>
 
#include <QWidgetAction>
#include <QLineEdit>
#include <QActionGroup>
 
class MenuEditAction : public QWidgetAction
{
    Q_OBJECT
public:
    MenuEditAction(QString text, QString hintText, QObject * parent = nullptr);
 
    QString text() const;
 
protected:
    virtual QWidget * createWidget(QWidget *parent);
 
private:
    QString _hintText;
 
    QString _text;
 
    QLineEdit * _lineEdit;
 
private slots:
    void triggerEntry();
};
 
 
class TrafficTreeHeaderView : public QHeaderView
{
    Q_OBJECT
public:
    TrafficTreeHeaderView(GList ** recentColumnList, QWidget * parent = nullptr);
 
    ~TrafficTreeHeaderView();
 
    void applyRecent();
 
signals:
    void columnsHaveChanged(QList<int> visible);
 
    void filterOnColumn(int column, int filterOn, QString filterText);
 
private:
    GList ** _recentColumnList;
 
    QActionGroup * _actions;
 
    QString _filterText;
 
private slots:
    void headerContextMenu(const QPoint &pos);
 
    void columnTriggered(bool checked = false);
 
    void menuActionTriggered(QAction * act);
 
    void filterColumn(bool checked = false);
};
 
 
class TrafficDataFilterProxy : public QSortFilterProxyModel
{
    Q_OBJECT
public:
 
    enum {
        TRAFFIC_DATA_LESS,
        TRAFFIC_DATA_GREATER,
        TRAFFIC_DATA_EQUAL,
    };
 
    TrafficDataFilterProxy(QObject *parent = nullptr);
 
    void setColumnVisibility(int column, bool visible);
 
    bool columnVisible(int column) const;
 
public slots:
    void filterForColumn(int column, int filterOn, QString filterText);
 
protected:
    virtual bool filterAcceptsRow(int source_row, const QModelIndex &source_parent) const;
 
    virtual bool filterAcceptsColumn(int source_column, const QModelIndex &source_parent) const;
 
    virtual bool lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const;
 
private:
    QList<int> hideColumns_;
 
    int _filterColumn;
 
    int _filterOn;
 
    QString _filterText;
 
    int mapToSourceColumn(int proxyColumn) const;
};
 
 
class TrafficTree : public QTreeView
{
    Q_OBJECT
 
public:
    typedef enum {
        CLIPBOARD_CSV,
        CLIPBOARD_YAML,
        CLIPBOARD_JSON
    } eTrafficTreeClipboard;
 
    TrafficTree(QString baseName, GList ** recentColumnList, QWidget *parent = nullptr);
 
    QMenu * createCopyMenu(QWidget * parent = nullptr);
 
    void applyRecentColumns();
 
    void widenColumnToContents(int column);
 
    virtual void setModel(QAbstractItemModel *model) override;
 
signals:
    void filterAction(QString filter, FilterAction::Action action, FilterAction::ActionType type);
 
    void columnsHaveChanged(QList<int> columns);
 
public slots:
    void tapListenerEnabled(bool enable);
 
    void disableTap();
 
    void columnsChanged(QList<int> columns);
 
private:
    bool _tapEnabled;
 
    int _exportRole;
 
    bool _saveRaw;
 
    QString _baseName;
 
    TrafficTreeHeaderView * _header;
 
    ATapDataModel * dataModel();
 
    QMenu * createActionSubMenu(FilterAction::Action cur_action, QModelIndex idx, bool isConversation);
 
    void copyToClipboard(eTrafficTreeClipboard type);
 
    friend class TrafficTreeHeaderView;
 
private slots:
    void customContextMenu(const QPoint &pos);
 
    void useFilterAction();
 
    void clipboardAction();
 
    void resizeAction();
 
    void toggleSaveRawAction();
 
    void handleDataChanged(const QModelIndex &topLeft, const QModelIndex &bottomRight,
#if QT_VERSION < QT_VERSION_CHECK(6, 0, 0)
        const QVector<int>
#else
        const QList<int>
#endif
        );
 
    void handleLayoutChanged(const QList<QPersistentModelIndex>, QAbstractItemModel::LayoutChangeHint);
};
 
#endif // TRAFFIC_TREE_H