1 files changed, 17 insertions, 5 deletions
diff --git a/tools/qmldebugger/canvasframerate.h b/tools/qmldebugger/canvasframerate.h
index aa275aa..6c1ff9f 100644
--- a/tools/qmldebugger/canvasframerate.h
+++ b/tools/qmldebugger/canvasframerate.h
@@ -1,31 +1,43 @@
 #ifndef CANVASFRAMERATE_H
 #define CANVASFRAMERATE_H
 
-#include <QWidget>
+#include <QtCore/qpointer.h>
+#include <QtGui/qwidget.h>
+
+#include <QtDeclarative/qmldebugclient.h>
 
 QT_BEGIN_NAMESPACE
 
-class QmlDebugConnection;
 class QTabWidget;
 class QSpinBox;
+class QCheckBox;
+
+class CanvasFrameRatePlugin;
+
 class CanvasFrameRate : public QWidget
 {
     Q_OBJECT
 public:
-    CanvasFrameRate(QmlDebugConnection *, QWidget *parent = 0);
+    CanvasFrameRate(QWidget *parent = 0);
+    
+    void reset(QmlDebugConnection *conn);
 
     void setSizeHint(const QSize &);
     virtual QSize sizeHint() const;
 
 private slots:
     void newTab();
-    void stateChanged(int);
+    void enabledStateChanged(int);
+    void connectionStateChanged(QAbstractSocket::SocketState state);
 
 private:
+    void handleConnected(QmlDebugConnection *conn);
+
     QTabWidget *m_tabs;
     QSpinBox *m_spin;
-    QObject *m_plugin;
+    CanvasFrameRatePlugin *m_plugin;
     QSize m_sizeHint;
+    QCheckBox *m_enabledCheckBox;
 };
 
 QT_END_NAMESPACE