packet_list_model.h

Go to the documentation of this file.

 
#ifndef PACKET_LIST_MODEL_H
#define PACKET_LIST_MODEL_H
 
#include <config.h>
 
#include <stdio.h>
 
#include <epan/packet.h>
 
#include <QAbstractItemModel>
#include <QColor>
#include <QFont>
#include <QVector>
 
#include "packet_list_record.h"
 
#include <epan/cfile.h>

class PacketListModel : public QAbstractItemModel
{
    Q_OBJECT
public:

    enum {
        HEADER_CAN_DISPLAY_STRINGS = Qt::UserRole,
        HEADER_CAN_DISPLAY_DETAILS,
        COLORING_RULE_COLORS,
    };

    explicit PacketListModel(QObject *parent = 0, capture_file *cf = NULL);

    ~PacketListModel();

    void setCaptureFile(capture_file *cf);

    capture_file *captureFile() const { return cap_file_; }

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

    QModelIndex parent(const QModelIndex &) const override;

    PacketListRecord *physicalRecordForFrameNum(int frame_num) const;

    PacketListRecord *physicalRecordAt(int row) const;

    bool isEmpty() const { return physical_rows_.isEmpty(); }

    void clear();

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

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

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

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

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

    int appendPacket(frame_data *fdata);

    void invalidateAllColumnStrings();

    void resetColumns();

    void resetColorized();

    void toggleFrameMark(const QModelIndexList &indices);

    void setFrameMark(const QVector<PacketListRecord *> &records, bool set);

    void toggleFrameIgnore(const QModelIndexList &indices);

    void setFrameIgnore(const QVector<PacketListRecord *> &records, bool set);

    void toggleFrameRefTime(const QModelIndexList &indices);

    void unsetAllFrameRefTime();

    void addFrameComment(const QModelIndexList &indices, const QByteArray &comment);

    void setFrameComment(const QModelIndex &index, const QByteArray &comment, unsigned c_number);

    void deleteFrameComments(const QModelIndexList &indices);

    void deleteAllFrameComments();
 
signals:
    void packetAppended(capture_file *cap_file, frame_data *fdata, qsizetype row);
 
public slots:
    void flushNewRows();
 
private slots:
    void onThemeChanged();
 
private:
    QColor marked_fg_;
    QColor marked_bg_;
    QColor ignored_fg_;
    QColor ignored_bg_;

    void refreshThemeColors();

    capture_file *cap_file_;

    QList<QString> col_names_;

    QVector<PacketListRecord *> physical_rows_;

    int inserted_rows_;
 
};
 
#endif // PACKET_LIST_MODEL_H