qmlprofiler: Clean up recording state mess

The recording state is a global property of the profiler client, it has
to match the application's recording status as well as possible, and
the user can set an initial target value with a command line parameter.

This way we don't get strange error messages when the application quits
by itself and properly sends all the data before.

Task-number: QTBUG-43066
Change-Id: Id93aa7fb940f870c8f16cea8427c38aab450f864
Reviewed-by: Joerg Bornemann <joerg.bornemann@theqtcompany.com>

4 files changed, 43 insertions, 96 deletions

diff --git a/tools/qmlprofiler/qmlprofilerapplication.cpp b/tools/qmlprofiler/qmlprofilerapplication.cpp
index d3097c267..514299980 100644
--- a/tools/qmlprofiler/qmlprofilerapplication.cpp
+++ b/tools/qmlprofiler/qmlprofilerapplication.cpp
@@ -61,6 +61,7 @@ QmlProfilerApplication::QmlProfilerApplication(int &argc, char **argv) :
     m_port(3768),
     m_verbose(false),
     m_quitAfterSave(false),
+    m_recording(true),
     m_qmlProfilerClient(&m_connection),
     m_v8profilerClient(&m_connection),
     m_connectionAttempts(0),
@@ -75,11 +76,11 @@ QmlProfilerApplication::QmlProfilerApplication(int &argc, char **argv) :
     connect(&m_connection, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(connectionError(QAbstractSocket::SocketError)));
 
     connect(&m_qmlProfilerClient, SIGNAL(enabledChanged()), this, SLOT(traceClientEnabled()));
-    connect(&m_qmlProfilerClient, SIGNAL(recordingChanged(bool)), this, SLOT(recordingChanged()));
     connect(&m_qmlProfilerClient, SIGNAL(range(QQmlProfilerService::RangeType,QQmlProfilerService::BindingType,qint64,qint64,QStringList,QmlEventLocation)),
             &m_profilerData, SLOT(addQmlEvent(QQmlProfilerService::RangeType,QQmlProfilerService::BindingType,qint64,qint64,QStringList,QmlEventLocation)));
     connect(&m_qmlProfilerClient, SIGNAL(traceFinished(qint64)), &m_profilerData, SLOT(setTraceEndTime(qint64)));
     connect(&m_qmlProfilerClient, SIGNAL(traceStarted(qint64)), &m_profilerData, SLOT(setTraceStartTime(qint64)));
+    connect(&m_qmlProfilerClient, SIGNAL(traceStarted(qint64)), this, SLOT(notifyTraceStarted()));
     connect(&m_qmlProfilerClient, SIGNAL(frame(qint64,int,int,int)), &m_profilerData, SLOT(addFrameEvent(qint64,int,int,int)));
     connect(&m_qmlProfilerClient, SIGNAL(sceneGraphFrame(QQmlProfilerService::SceneGraphFrameType,
                                          qint64,qint64,qint64,qint64,qint64,qint64)),
@@ -146,10 +147,13 @@ void QmlProfilerApplication::parseArguments()
                             QLatin1String("port"), QLatin1String("3768"));
     parser.addOption(port);
 
-    QCommandLineOption fromStart(QLatin1String("fromStart"),
-                                 tr("Record as soon as the engine is started. "
-                                    "The default is false."));
-    parser.addOption(fromStart);
+    QCommandLineOption record(QLatin1String("record"),
+                              tr("If set to 'off', don't immediately start recording data when the "
+                                 "QML engine starts, but instead either start the recording "
+                                 "interactively or with the JavaScript console.profile() function. "
+                                 "By default the recording starts immediately."),
+                              QLatin1String("on|off"), QLatin1String("on"));
+    parser.addOption(record);
 
     QCommandLineOption verbose(QStringList() << QLatin1String("verbose"),
                                tr("Print debugging output."));
@@ -181,10 +185,7 @@ void QmlProfilerApplication::parseArguments()
         }
     }
 
-    if (parser.isSet(fromStart)) {
-        m_qmlProfilerClient.setRecording(true);
-        m_v8profilerClient.setRecording(true);
-    }
+    m_recording = (parser.value(record) == QLatin1String("on"));
 
     if (parser.isSet(verbose))
         m_verbose = true;
@@ -244,26 +245,28 @@ void QmlProfilerApplication::userCommand(const QString &command)
         printCommands();
     } else if (cmd == Constants::CMD_RECORD
                || cmd == Constants::CMD_RECORD2) {
-        m_qmlProfilerClient.setRecording(
-                    !m_qmlProfilerClient.isRecording());
-        m_v8profilerClient.setRecording(!m_v8profilerClient.isRecording());
-        m_qmlDataReady = false;
-        m_v8DataReady = false;
+        m_qmlProfilerClient.sendRecordingStatus(!m_recording);
+        m_v8profilerClient.sendRecordingStatus(!m_recording);
     } else if (cmd == Constants::CMD_QUIT
                || cmd == Constants::CMD_QUIT2) {
         print(QLatin1String("Quit"));
-        if (m_qmlProfilerClient.isRecording()) {
+        if (m_recording) {
             m_quitAfterSave = true;
-            m_qmlDataReady = false;
-            m_v8DataReady = false;
-            m_qmlProfilerClient.setRecording(false);
-            m_v8profilerClient.setRecording(false);
+            m_qmlProfilerClient.sendRecordingStatus(false);
+            m_v8profilerClient.sendRecordingStatus(false);
         } else {
             quit();
         }
     }
 }
 
+void QmlProfilerApplication::notifyTraceStarted()
+{
+    // Synchronize to server state. It doesn't hurt to do this multiple times in a row for
+    // different traces. There is no symmetric event to "Complete" after all.
+    m_recording = true;
+}
+
 void QmlProfilerApplication::run()
 {
     if (m_runMode == LaunchMode) {
@@ -311,17 +314,9 @@ void QmlProfilerApplication::tryToConnect()
 void QmlProfilerApplication::connected()
 {
     m_connectTimer.stop();
-    print(QString(QLatin1String("Connected to host:port %1:%2."
-                                "Wait for profile data or type a command"
-                                "(type 'help'' to show list of commands).")
-                  ).arg(m_hostName).arg((m_port)));
-    QString recordingStatus(QLatin1String("Recording Status: %1"));
-    if (!m_qmlProfilerClient.isRecording() &&
-            !m_v8profilerClient.isRecording())
-        recordingStatus = recordingStatus.arg(QLatin1String("Off"));
-    else
-        recordingStatus = recordingStatus.arg(QLatin1String("On"));
-    print(recordingStatus);
+    print(tr("Connected to host:port %1:%2. Wait for profile data or type a command (type 'help' "
+             "to show list of commands).").arg(m_hostName).arg((m_port)));
+    print(tr("Recording Status: %1").arg(m_recording ? tr("on") : tr("off")));
 }
 
 void QmlProfilerApplication::connectionStateChanged(
@@ -352,7 +347,7 @@ void QmlProfilerApplication::processFinished()
     if (m_process->exitStatus() == QProcess::NormalExit) {
         logStatus(QString("Process exited (%1).").arg(m_process->exitCode()));
 
-        if (m_qmlProfilerClient.isRecording()) {
+        if (m_recording) {
             logError("Process exited while recording, last trace is lost!");
             exit(2);
         } else {
@@ -369,8 +364,8 @@ void QmlProfilerApplication::traceClientEnabled()
     logStatus("Trace client is attached.");
     // blocked server is waiting for recording message from both clients
     // once the last one is connected, both messages should be sent
-    m_qmlProfilerClient.sendRecordingStatus();
-    m_v8profilerClient.sendRecordingStatus();
+    m_qmlProfilerClient.sendRecordingStatus(m_recording);
+    m_v8profilerClient.sendRecordingStatus(m_recording);
 }
 
 void QmlProfilerApplication::profilerClientEnabled()
@@ -379,30 +374,26 @@ void QmlProfilerApplication::profilerClientEnabled()
 
     // blocked server is waiting for recording message from both clients
     // once the last one is connected, both messages should be sent
-    m_qmlProfilerClient.sendRecordingStatus();
-    m_v8profilerClient.sendRecordingStatus();
+    m_qmlProfilerClient.sendRecordingStatus(m_recording);
+    m_v8profilerClient.sendRecordingStatus(m_recording);
 }
 
 void QmlProfilerApplication::traceFinished()
 {
+    m_recording = false; // only on "Complete" we know that the trace is really finished.
     const QString fileName = traceFileName();
 
     if (m_profilerData.save(fileName))
         print(QString("Saving trace to %1.").arg(fileName));
 
+    // after saving, reset the flags
+    m_qmlDataReady = false;
+    m_v8DataReady = false;
+
     if (m_quitAfterSave)
         quit();
 }
 
-void QmlProfilerApplication::recordingChanged()
-{
-    if (m_qmlProfilerClient.isRecording()) {
-        print(QLatin1String("Recording is on."));
-    } else {
-        print(QLatin1String("Recording is off."));
-    }
-}
-
 void QmlProfilerApplication::print(const QString &line)
 {
     QTextStream err(stderr);
@@ -429,8 +420,6 @@ void QmlProfilerApplication::qmlComplete()
     if (m_v8profilerClient.state() != QQmlDebugClient::Enabled ||
             m_v8DataReady) {
         m_profilerData.complete();
-        // once complete is sent, reset the flag
-        m_qmlDataReady = false;
     }
 }
 
@@ -440,7 +429,5 @@ void QmlProfilerApplication::v8Complete()
     if (m_qmlProfilerClient.state() != QQmlDebugClient::Enabled ||
             m_qmlDataReady) {
         m_profilerData.complete();
-        // once complete is sent, reset the flag
-        m_v8DataReady = false;
     }
 }
diff --git a/tools/qmlprofiler/qmlprofilerapplication.h b/tools/qmlprofiler/qmlprofilerapplication.h
index aa63f54dc..1b1131e33 100644
--- a/tools/qmlprofiler/qmlprofilerapplication.h
+++ b/tools/qmlprofiler/qmlprofilerapplication.h
@@ -53,6 +53,7 @@ public:
 
 public slots:
     void userCommand(const QString &command);
+    void notifyTraceStarted();
 
 private slots:
     void run();
@@ -66,7 +67,6 @@ private slots:
     void traceClientEnabled();
     void profilerClientEnabled();
     void traceFinished();
-    void recordingChanged();
 
     void print(const QString &line);
     void logError(const QString &error);
@@ -94,6 +94,7 @@ private:
     quint16 m_port;
     bool m_verbose;
     bool m_quitAfterSave;
+    bool m_recording;
 
     QQmlDebugConnection m_connection;
     QmlProfilerClient m_qmlProfilerClient;
diff --git a/tools/qmlprofiler/qmlprofilerclient.cpp b/tools/qmlprofiler/qmlprofilerclient.cpp
index 711ddbd86..b752a7594 100644
--- a/tools/qmlprofiler/qmlprofilerclient.cpp
+++ b/tools/qmlprofiler/qmlprofilerclient.cpp
@@ -39,17 +39,12 @@
 ProfilerClient::ProfilerClient(const QString &clientName,
                              QQmlDebugConnection *client)
     : QQmlDebugClient(clientName, client),
-      m_recording(false),
       m_enabled(false)
 {
 }
 
 ProfilerClient::~ProfilerClient()
 {
-    //Disable profiling if started by client
-    //Profiling data will be lost!!
-    if (isRecording())
-        setRecording(false);
 }
 
 void ProfilerClient::clearData()
@@ -62,29 +57,6 @@ bool ProfilerClient::isEnabled() const
     return m_enabled;
 }
 
-void ProfilerClient::sendRecordingStatus()
-{
-}
-
-bool ProfilerClient::isRecording() const
-{
-    return m_recording;
-}
-
-void ProfilerClient::setRecording(bool v)
-{
-    if (v == m_recording)
-        return;
-
-    m_recording = v;
-
-    if (state() == Enabled) {
-        sendRecordingStatus();
-    }
-
-    emit recordingChanged(v);
-}
-
 void ProfilerClient::stateChanged(State status)
 {
     if ((m_enabled && status != Enabled) ||
@@ -135,11 +107,11 @@ void QmlProfilerClient::clearData()
     ProfilerClient::clearData();
 }
 
-void QmlProfilerClient::sendRecordingStatus()
+void QmlProfilerClient::sendRecordingStatus(bool record)
 {
     QByteArray ba;
     QDataStream stream(&ba, QIODevice::WriteOnly);
-    stream << isRecording();
+    stream << record;
     sendMessage(ba);
 }
 
@@ -296,19 +268,14 @@ V8ProfilerClient::~V8ProfilerClient()
 {
 }
 
-void V8ProfilerClient::sendRecordingStatus()
+void V8ProfilerClient::sendRecordingStatus(bool record)
 {
     QByteArray ba;
     QDataStream stream(&ba, QIODevice::WriteOnly);
     QByteArray cmd("V8PROFILER");
-    QByteArray option("");
+    QByteArray option(record ? "start" : "stop");
     QByteArray title("");
 
-    if (m_recording) {
-        option = "start";
-    } else {
-        option = "stop";
-    }
     stream << cmd << option << title;
     sendMessage(ba);
 }
diff --git a/tools/qmlprofiler/qmlprofilerclient.h b/tools/qmlprofiler/qmlprofilerclient.h
index 2d9c382ff..84da96197 100644
--- a/tools/qmlprofiler/qmlprofilerclient.h
+++ b/tools/qmlprofiler/qmlprofilerclient.h
@@ -44,25 +44,18 @@ class ProfilerClient : public QQmlDebugClient
     Q_OBJECT
 
     Q_PROPERTY(bool enabled READ isEnabled NOTIFY enabledChanged)
-    Q_PROPERTY(bool recording READ isRecording WRITE setRecording
-               NOTIFY recordingChanged)
-
 public:
     ProfilerClient(const QString &clientName,
                   QQmlDebugConnection *client);
     ~ProfilerClient();
 
     bool isEnabled() const;
-    bool isRecording() const;
 
 public slots:
-    void setRecording(bool);
     virtual void clearData();
-    virtual void sendRecordingStatus();
 
 signals:
     void complete();
-    void recordingChanged(bool arg);
     void enabledChanged();
     void cleared();
 
@@ -70,7 +63,6 @@ protected:
     virtual void stateChanged(State);
 
 protected:
-    bool m_recording;
     bool m_enabled;
 };
 
@@ -84,7 +76,7 @@ public:
 
 public slots:
     void clearData();
-    void sendRecordingStatus();
+    void sendRecordingStatus(bool record);
 
 signals:
     void traceFinished( qint64 time );
@@ -125,7 +117,7 @@ public:
     ~V8ProfilerClient();
 
 public slots:
-    void sendRecordingStatus();
+    void sendRecordingStatus(bool record);
 
 signals:
     void range(int depth, const QString &function, const QString &filename,