Connection tracker class finished.

[yandex-traffic] / mainwidget.hpp

diff --git a/mainwidget.hpp b/mainwidget.hpp
index e7c791f..0fb63a2 100644 (file)
--- a/mainwidget.hpp
+++ b/mainwidget.hpp
@@ -3,48 +3,34 @@
 
 #include <QtGui>
 
-#include <settings.hpp>
-#include <regions.hpp>
+#include "traffic.hpp"
+#include "settings.hpp"
+#include "regions.hpp"
+#include "light.hpp"
 
 
-class MainWidget : public QLabel
+class MainWidget : public QWidget
 {
     Q_OBJECT
 private:
+    Traffic _traffic;
     RegionsTable _regions;
     Settings _settings;
 
-public:
-    MainWidget ()
-        : QLabel ()
-    {
-        const RegionInfo *ri;
-
-        setAlignment(Qt::AlignCenter);
-#ifdef Q_WS_MAEMO_5
-        setAttribute(Qt::WA_TranslucentBackground);
-#endif
-        ri = _regions.lookup (_settings.regions ()[0]);
-
-        setText (_settings.regions ().join (", ") + ", " + QString (ri ? ri->name () : ""));
-    }
+    // Widgets
+    TrafficLight *_light;
+    QLabel *_label;
 
-    QSize sizeHint() const
-    {
-        return QLabel::sizeHint() + QSize (20, 0);
-    }
+public:
+    MainWidget ();
+    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();
-
-        QLabel::paintEvent(event);
-    }
+    void paintEvent (QPaintEvent *event);
+    void timerEvent (QTimerEvent *event);
+
+protected slots:
+    void trafficUpdated ();
 };
 
 #endif /* __MAINWIDGET_H__ */