Scott Ellis
11 years ago
13 changed files with 618 additions and 9 deletions
@ -0,0 +1,169 @@ |
|||
#include "trafficgraphwidget.h" |
|||
#include "clientmodel.h" |
|||
|
|||
#include <QPainter> |
|||
#include <QColor> |
|||
#include <QTimer> |
|||
|
|||
#include <cmath> |
|||
|
|||
#define DESIRED_SAMPLES 800 |
|||
|
|||
#define XMARGIN 10 |
|||
#define YMARGIN 10 |
|||
|
|||
TrafficGraphWidget::TrafficGraphWidget(QWidget *parent) : |
|||
QWidget(parent), |
|||
timer(0), |
|||
fMax(0.0f), |
|||
nMins(0), |
|||
vSamplesIn(), |
|||
vSamplesOut(), |
|||
nLastBytesIn(0), |
|||
nLastBytesOut(0), |
|||
clientModel(0) |
|||
{ |
|||
timer = new QTimer(this); |
|||
connect(timer, SIGNAL(timeout()), SLOT(updateRates())); |
|||
} |
|||
|
|||
void TrafficGraphWidget::setClientModel(ClientModel *model) |
|||
{ |
|||
clientModel = model; |
|||
if(model) { |
|||
nLastBytesIn = model->getTotalBytesRecv(); |
|||
nLastBytesOut = model->getTotalBytesSent(); |
|||
} |
|||
} |
|||
|
|||
int TrafficGraphWidget::getGraphRangeMins() const |
|||
{ |
|||
return nMins; |
|||
} |
|||
|
|||
void TrafficGraphWidget::paintPath(QPainterPath &path, QQueue<float> &samples) |
|||
{ |
|||
int h = height() - YMARGIN * 2, w = width() - XMARGIN * 2; |
|||
int sampleCount = samples.size(), x = XMARGIN + w, y; |
|||
if(sampleCount > 0) { |
|||
path.moveTo(x, YMARGIN + h); |
|||
for(int i = 0; i < sampleCount; ++i) { |
|||
x = XMARGIN + w - w * i / DESIRED_SAMPLES; |
|||
y = YMARGIN + h - (int)(h * samples.at(i) / fMax); |
|||
path.lineTo(x, y); |
|||
} |
|||
path.lineTo(x, YMARGIN + h); |
|||
} |
|||
} |
|||
|
|||
void TrafficGraphWidget::paintEvent(QPaintEvent *) |
|||
{ |
|||
QPainter painter(this); |
|||
painter.fillRect(rect(), Qt::black); |
|||
|
|||
if(fMax <= 0.0f) return; |
|||
|
|||
QColor axisCol(Qt::gray); |
|||
int h = height() - YMARGIN * 2; |
|||
painter.setPen(axisCol); |
|||
painter.drawLine(XMARGIN, YMARGIN + h, width() - XMARGIN, YMARGIN + h); |
|||
|
|||
// decide what order of magnitude we are
|
|||
int base = floor(log10(fMax)); |
|||
float val = pow(10.0f, base); |
|||
|
|||
const QString units = tr("KB/s"); |
|||
// draw lines
|
|||
painter.setPen(axisCol); |
|||
painter.drawText(XMARGIN, YMARGIN + h - h * val / fMax, QString("%1 %2").arg(val).arg(units)); |
|||
for(float y = val; y < fMax; y += val) { |
|||
int yy = YMARGIN + h - h * y / fMax; |
|||
painter.drawLine(XMARGIN, yy, width() - XMARGIN, yy); |
|||
} |
|||
// if we drew 3 or fewer lines, break them up at the next lower order of magnitude
|
|||
if(fMax / val <= 3.0f) { |
|||
axisCol = axisCol.darker(); |
|||
val = pow(10.0f, base - 1); |
|||
painter.setPen(axisCol); |
|||
painter.drawText(XMARGIN, YMARGIN + h - h * val / fMax, QString("%1 %2").arg(val).arg(units)); |
|||
int count = 1; |
|||
for(float y = val; y < fMax; y += val, count++) { |
|||
// don't overwrite lines drawn above
|
|||
if(count % 10 == 0) |
|||
continue; |
|||
int yy = YMARGIN + h - h * y / fMax; |
|||
painter.drawLine(XMARGIN, yy, width() - XMARGIN, yy); |
|||
} |
|||
} |
|||
|
|||
if(!vSamplesIn.empty()) { |
|||
QPainterPath p; |
|||
paintPath(p, vSamplesIn); |
|||
painter.fillPath(p, QColor(0, 255, 0, 128)); |
|||
painter.setPen(Qt::green); |
|||
painter.drawPath(p); |
|||
} |
|||
if(!vSamplesOut.empty()) { |
|||
QPainterPath p; |
|||
paintPath(p, vSamplesOut); |
|||
painter.fillPath(p, QColor(255, 0, 0, 128)); |
|||
painter.setPen(Qt::red); |
|||
painter.drawPath(p); |
|||
} |
|||
} |
|||
|
|||
void TrafficGraphWidget::updateRates() |
|||
{ |
|||
if(!clientModel) return; |
|||
|
|||
quint64 bytesIn = clientModel->getTotalBytesRecv(), |
|||
bytesOut = clientModel->getTotalBytesSent(); |
|||
float inRate = (bytesIn - nLastBytesIn) / 1024.0f * 1000 / timer->interval(); |
|||
float outRate = (bytesOut - nLastBytesOut) / 1024.0f * 1000 / timer->interval(); |
|||
vSamplesIn.push_front(inRate); |
|||
vSamplesOut.push_front(outRate); |
|||
nLastBytesIn = bytesIn; |
|||
nLastBytesOut = bytesOut; |
|||
|
|||
while(vSamplesIn.size() > DESIRED_SAMPLES) { |
|||
vSamplesIn.pop_back(); |
|||
} |
|||
while(vSamplesOut.size() > DESIRED_SAMPLES) { |
|||
vSamplesOut.pop_back(); |
|||
} |
|||
|
|||
float tmax = 0.0f; |
|||
foreach(float f, vSamplesIn) { |
|||
if(f > tmax) tmax = f; |
|||
} |
|||
foreach(float f, vSamplesOut) { |
|||
if(f > tmax) tmax = f; |
|||
} |
|||
fMax = tmax; |
|||
update(); |
|||
} |
|||
|
|||
void TrafficGraphWidget::setGraphRangeMins(int mins) |
|||
{ |
|||
nMins = mins; |
|||
int msecsPerSample = nMins * 60 * 1000 / DESIRED_SAMPLES; |
|||
timer->stop(); |
|||
timer->setInterval(msecsPerSample); |
|||
|
|||
clear(); |
|||
} |
|||
|
|||
void TrafficGraphWidget::clear() |
|||
{ |
|||
timer->stop(); |
|||
|
|||
vSamplesOut.clear(); |
|||
vSamplesIn.clear(); |
|||
fMax = 0.0f; |
|||
|
|||
if(clientModel) { |
|||
nLastBytesIn = clientModel->getTotalBytesRecv(); |
|||
nLastBytesOut = clientModel->getTotalBytesSent(); |
|||
} |
|||
timer->start(); |
|||
} |
|||
@ -0,0 +1,44 @@ |
|||
#ifndef TRAFFICGRAPHWIDGET_H |
|||
#define TRAFFICGRAPHWIDGET_H |
|||
|
|||
#include <QWidget> |
|||
#include <QQueue> |
|||
|
|||
class ClientModel; |
|||
|
|||
QT_BEGIN_NAMESPACE |
|||
class QPaintEvent; |
|||
class QTimer; |
|||
QT_END_NAMESPACE |
|||
|
|||
class TrafficGraphWidget : public QWidget |
|||
{ |
|||
Q_OBJECT |
|||
|
|||
public: |
|||
explicit TrafficGraphWidget(QWidget *parent = 0); |
|||
void setClientModel(ClientModel *model); |
|||
int getGraphRangeMins() const; |
|||
|
|||
protected: |
|||
void paintEvent(QPaintEvent *); |
|||
|
|||
public slots: |
|||
void updateRates(); |
|||
void setGraphRangeMins(int mins); |
|||
void clear(); |
|||
|
|||
private: |
|||
void paintPath(QPainterPath &path, QQueue<float> &samples); |
|||
|
|||
QTimer *timer; |
|||
float fMax; |
|||
int nMins; |
|||
QQueue<float> vSamplesIn; |
|||
QQueue<float> vSamplesOut; |
|||
quint64 nLastBytesIn; |
|||
quint64 nLastBytesOut; |
|||
ClientModel *clientModel; |
|||
}; |
|||
|
|||
#endif // TRAFFICGRAPHWIDGET_H
|
|||
Loading…
Reference in new issue