1 files changed, 15 insertions, 1 deletions
diff --git a/qt/mainwindow.hpp b/qt/mainwindow.hpp
index 7f8a8f9f46..16fc563072 100644
--- a/qt/mainwindow.hpp
+++ b/qt/mainwindow.hpp
@@ -5,6 +5,7 @@
 #include "platform/location.hpp"
 #include "platform/location_service.hpp"
 
+#include "std/array.hpp"
 #include "std/unique_ptr.hpp"
 
 #include <QtWidgets/QApplication>
@@ -17,6 +18,7 @@
 class QDockWidget;
 class QPushButton;
 class QLabel;
+class TrafficMode;
 
 namespace search { class Result; }
 
@@ -32,9 +34,12 @@ namespace qt
     QAction * m_clearSelection;
     QAction * m_pSearchAction;
     QAction * m_trafficEnableAction;
+    QAction * m_saveTrafficSampleAction;
+    QAction * m_quitTrafficModeAction;
     DrawWidget * m_pDrawWidget;
 
-    QDockWidget * m_Docks[1];
+    // TODO(mgsergio): Make indexing more informative.
+    array<QDockWidget *, 2> m_Docks;
 
     QPushButton * m_downloadButton;
     QPushButton * m_retryButton;
@@ -43,6 +48,9 @@ namespace qt
 
     unique_ptr<location::LocationService> const m_locationService;
 
+    // This object is managed by Qt memory system.
+    TrafficMode * m_trafficMode = nullptr;
+
     Q_OBJECT
 
   public:
@@ -63,6 +71,9 @@ namespace qt
     void CreateSearchBarAndPanel();
     void CreateCountryStatusControls();
 
+    void CreateTrafficPanel(string const & dataFilePath, string const & sampleFilePath);
+    void DestroyTrafficPanel();
+
 #if defined(Q_WS_WIN)
     /// to handle menu messages
     virtual bool winEvent(MSG * msg, long * result);
@@ -92,5 +103,8 @@ namespace qt
     void OnClearSelection();
 
     void OnTrafficEnabled();
+    void OnOpenTrafficSample();
+    void OnSaveTrafficSample();
+    void OnQuitTrafficMode();
   };
 }