add sniffer

2016-07-07 16:47:43 +02:00
parent e9d3b718c9
commit 314ea0769d
10 changed files with 862 additions and 0 deletions
@@ -0,0 +1,267 @@
+#include <QDebug>
+#include "sniffermodel.h"
+#include "snifferwindow.h"
+
+
+SnifferModel::SnifferModel(QObject *parent)
+    : QAbstractItemModel(parent),
+      mFilter(false)
+{
+}
+
+SnifferModel::~SnifferModel()
+{
+    clear();
+}
+
+
+int SnifferModel::columnCount(const QModelIndex &parent) const
+{
+    return parent.isValid() ? 0 : tc::LAST+1;
+}
+
+
+int SnifferModel::rowCount(const QModelIndex &parent) const
+{
+    const QMap<quint32, SnifferItem*>& map = mFilter ? mFilters : mMap;
+    return parent.isValid() ? 0 : map.size();
+}
+
+
+QVariant SnifferModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid())
+        return QVariant();
+
+    const QMap<quint32, SnifferItem*>& map = mFilter ? mFilters : mMap;
+
+    if(index.row()<=map.size())
+    {
+        quint64 key = map.keys()[index.row()];
+        SnifferItem* item = map[key];
+
+        int col = index.column();
+
+        switch(role)
+        {
+            case Qt::DisplayRole:
+            {
+                switch(col)
+                {
+                    case tc::DELTA:
+                        return QString::number(item->getDelta(), 'f');
+                    case tc::ID:
+                        return QString("%1").arg(item->getId(), 5, 16, QLatin1Char(' ')).toUpper();
+                    default:
+                        break;
+                }
+                if(tc::DATA_0<=col && col <=tc::DATA_7)
+                {
+                    int data = item->getData(col-tc::DATA_0);
+                    if(data>=0)
+                        return QString("%1").arg(data, 2, 16, QLatin1Char('0')).toUpper();
+                }
+                break;
+            }
+            case Qt::BackgroundRole:
+            {
+                if(tc::ID==col)
+                {
+                    if(item->elapsed()>4000)
+                        return QBrush(Qt::red);
+                }
+                else if(tc::DATA_0<=col && col<=tc::DATA_7)
+                {
+                    dc change = item->dataChange(col-tc::DATA_0);
+                    switch(change)
+                    {
+                        case dc::INC:
+                            return QBrush(Qt::green);
+                        case dc::DEC:
+                            return QBrush(Qt::red);
+                        default:
+                            break;
+                    }
+                }
+                break;
+            }
+        }
+    }
+
+    return QVariant();
+}
+
+
+Qt::ItemFlags SnifferModel::flags(const QModelIndex &index) const
+{
+    if (!index.isValid())
+        return 0;
+
+    return QAbstractItemModel::flags(index);
+}
+
+
+QVariant SnifferModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+    if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
+    {
+        switch(section)
+        {
+            case tc::DELTA:
+                return QString("Delta");
+            case tc::ID:
+                return QString("ID");
+            default:
+                break;
+        }
+        if(tc::DATA_0<=section && section <=tc::DATA_7)
+            return QString::number(section-tc::DATA_0);
+    }
+
+    return QVariant();
+}
+
+
+QModelIndex SnifferModel::index(int row, int column, const QModelIndex &parent)
+            const
+{
+    if (parent.isValid())
+        return QModelIndex();
+
+    const QMap<quint32, SnifferItem*>& map = mFilter ? mFilters : mMap;
+
+    if(column>tc::LAST || row>=map.size())
+        return QModelIndex();
+
+    return createIndex(row, column);
+}
+
+
+QModelIndex SnifferModel::parent(const QModelIndex &) const
+{
+    return QModelIndex();
+}
+
+
+void SnifferModel::clear()
+{
+    beginResetModel();
+    qDeleteAll(mMap);
+    mMap.clear();
+    mFilters.clear();
+    endResetModel();
+
+    mFilter = false;
+}
+
+
+void SnifferModel::refresh()
+{
+    QMap<quint32, SnifferItem*>::iterator i;
+    QVector<quint32> toRemove;
+    SnifferItem* item;
+
+    /* update markers */
+
+
+    for (i = mMap.begin(); i != mMap.end(); ++i)
+    {
+        i.value()->updateMarker();
+        if(i.value()->elapsed()>5000)
+            toRemove.append(i.key());
+    }
+
+    if(toRemove.size())
+    {
+        beginResetModel();
+        foreach(quint32 id, toRemove)
+        {
+            /* remove element */
+            item = mMap.take(id);
+            mFilters.remove(id);
+            delete item;
+            /* send notification */
+            emit idChange(id, false);
+        }
+        endResetModel();
+    }
+    else
+        /* refresh data */
+        dataChanged(createIndex(0, 0),
+                    createIndex(rowCount()-1, columnCount()-1), QVector<int>(Qt::DisplayRole));
+}
+
+
+void SnifferModel::filter(fltType pType, int pId)
+{
+    beginResetModel();
+    switch(pType)
+    {
+        case fltType::NONE:
+            /* erase everything */
+            mFilter = true;
+            mFilters.clear();
+            break;
+        case fltType::ADD:
+            /* add filter to list */
+            mFilter = true;
+            mFilters[pId] = mMap[pId];
+            break;
+        case fltType::REMOVE:
+            /* remove filter */
+            if(!mFilter)
+                mFilters = mMap;
+            mFilter = true;
+            mFilters.remove(pId);
+            break;
+        case fltType::ALL:
+            /* stop filtering */
+            mFilter = false;
+            mFilters.clear();
+            break;
+    }
+    endResetModel();
+}
+
+
+/***********************************************/
+/**********         slots       ****************/
+/***********************************************/
+
+void SnifferModel::update(CANConnection*, QVector<CANFrame>& pFrames)
+{
+    foreach(const CANFrame& frame, pFrames)
+    {
+        if(!mMap.contains(frame.ID))
+        {
+            int index = std::distance(mMap.begin(), mMap.lowerBound(frame.ID));
+            /* add the frame */
+            beginInsertRows(QModelIndex(), index, index);
+            mMap[frame.ID] = new SnifferItem(frame);
+            endInsertRows();
+
+            emit idChange(frame.ID, true);
+        }
+        else
+            //updateData
+            mMap[frame.ID]->update(frame);
+    }
+}
+
+void SnifferModel::notch()
+{
+    QMap<quint32, SnifferItem*>& map = mFilter ? mFilters : mMap;
+
+    foreach(SnifferItem* item, map)
+        item->notch(true);
+}
+
+void SnifferModel::unNotch()
+{
+    QMap<quint32, SnifferItem*>& map = mFilter ? mFilters : mMap;
+
+    foreach(SnifferItem* item, map)
+        item->notch(false);
+}
+
+