Connection tracker class finished.

[yandex-traffic] / mainwidget.hpp

diff --git a/mainwidget.hpp b/mainwidget.hpp
index 4e29b66..0fb63a2 100644 (file)
--- a/mainwidget.hpp
+++ b/mainwidget.hpp
@@ -3,6 +3,7 @@
 
 #include <QtGui>
 
+#include "traffic.hpp"
 #include "settings.hpp"
 #include "regions.hpp"
 #include "light.hpp"
@@ -12,6 +13,7 @@ class MainWidget : public QWidget
 {
     Q_OBJECT
 private:
+    Traffic _traffic;
     RegionsTable _regions;
     Settings _settings;
 
@@ -24,16 +26,11 @@ public:
     QSize sizeHint () const;
 
 protected:
-    void paintEvent(QPaintEvent *event)
-    {
-        QPainter p(this);
-        p.setBrush(QColor(0, 0, 0, 128));
-        p.setPen(Qt::NoPen);
-        p.drawRoundedRect(rect(), 10, 10);
-        p.end();
-
-        QWidget::paintEvent(event);
-    }
+    void paintEvent (QPaintEvent *event);
+    void timerEvent (QTimerEvent *event);
+
+protected slots:
+    void trafficUpdated ();
 };
 
 #endif /* __MAINWIDGET_H__ */