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 <ui/qt/widgets/adaptive_header_view.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) override;
 
private:
    QString _hintText;

    QString _text;

    QLineEdit * _lineEdit;
 
private slots:
    void triggerEntry();
};
 

class TrafficTreeHeaderView : public AdaptiveHeaderView
{
    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 override;

    virtual bool filterAcceptsColumn(int source_column, const QModelIndex &source_parent) const override;

    virtual bool lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const override;
 
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, const QList<int>);

    void handleLayoutChanged(const QList<QPersistentModelIndex>, QAbstractItemModel::LayoutChangeHint);
};
 
#endif // TRAFFIC_TREE_H