distribution_model.h

Go to the documentation of this file.

 
#ifndef DISTRIBUTION_MODEL_H
#define DISTRIBUTION_MODEL_H
 
#include <config.h>
 
#include <vector>
 
#include <QAbstractTableModel>
#include <QHash>
#include <QList>
#include <QString>
#include <QStringList>
#include <QVariant>
 
#include <epan/proto.h>

struct DistributionField
{
    enum SortKind {
        SortSigned,     
        SortUnsigned,   
        SortIPv4,       
        SortIPv6,       
        SortDouble,     
        SortString      
    };
 
    QString name;           
    QList<int> hf_indices;  
    SortKind kind = SortString; 
};

class DistributionModel : public QAbstractTableModel
{
    Q_OBJECT
 
public:
    using SortKind = DistributionField::SortKind;

    static constexpr int maxFields = 4;

    enum { SortRole = Qt::UserRole + 1 };

    explicit DistributionModel(QObject *parent = nullptr);

    static bool supportsField(const header_field_info *hfinfo);

    static SortKind sortKind(enum ftenum type);

    static QString parseFields(const QStringList &names, QList<DistributionField> &fields);

    void setFields(const QList<DistributionField> &fields);

    void clear();

    bool addPacket(proto_tree *tree);

    void publish();

    void setResolveValues(bool resolve);

    bool hasFields() const { return !fields_.isEmpty(); }

    bool hasResult() const { return has_result_; }

    int countColumn() const;

    int percentColumn() const;

    QString filterExpression(int row) const;

    QList<QVariant> rowData(int row) const;

    quint64 count(int row) const;

    QList<int> topRows(int n) const;

    QString rowLabel(int row) const;

    QStringList fieldNames() const;

    int distinctValues() const { return published_; }

    quint64 packets() const { return published_packets_; }

    quint64 observations() const { return published_observations_; }

    double entropy() const { return entropy_; }
 
    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    int columnCount(const QModelIndex &parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
 
signals:
    void summaryChanged();
 
private:
    struct Row {
        QStringList dfilter;    
        QStringList display;    
        QStringList label;      
        QList<QVariant> sort;   
        quint64 count;          
    };
 
    QString shownText(const Row &row, int field) const;
    double percent(const Row &row) const;
    void resetCounts();
 
    QList<DistributionField> fields_;
    std::vector<Row> rows_;
    QHash<QStringList, int> index_;
    int published_;
    quint64 packets_;
    quint64 observations_;
    quint64 published_packets_;
    quint64 published_observations_;
    double entropy_;
    bool has_result_;
    bool resolve_values_;
};
 
#endif // DISTRIBUTION_MODEL_H