1 files changed, 129 insertions, 0 deletions
diff --git a/src/plugins/qmltooling/qmldbg_profiler/qqmlenginecontrolservice.cpp b/src/plugins/qmltooling/qmldbg_profiler/qqmlenginecontrolservice.cpp
new file mode 100644
index 0000000000..4f131ac481
--- /dev/null
+++ b/src/plugins/qmltooling/qmldbg_profiler/qqmlenginecontrolservice.cpp
@@ -0,0 +1,129 @@
+/****************************************************************************
+**
+** Copyright (C) 2015 The Qt Company Ltd.
+** Contact: http://www.qt.io/licensing/
+**
+** This file is part of the QtQml module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL21$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see http://www.qt.io/terms-conditions. For further
+** information use the contact form at http://www.qt.io/contact-us.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 or version 3 as published by the Free
+** Software Foundation and appearing in the file LICENSE.LGPLv21 and
+** LICENSE.LGPLv3 included in the packaging of this file. Please review the
+** following information to ensure the GNU Lesser General Public License
+** requirements will be met: https://www.gnu.org/licenses/lgpl.html and
+** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** As a special exception, The Qt Company gives you certain additional
+** rights. These rights are described in The Qt Company LGPL Exception
+** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qqmlenginecontrolservice.h"
+#include <QQmlEngine>
+
+QT_BEGIN_NAMESPACE
+
+const QString QQmlEngineControlService::s_key = QStringLiteral("EngineControl");
+
+QQmlEngineControlService::QQmlEngineControlService(QObject *parent) :
+    QQmlDebugService(s_key, 1, parent)
+{
+}
+
+void QQmlEngineControlService::messageReceived(const QByteArray &message)
+{
+    QMutexLocker lock(&dataMutex);
+    QQmlDebugStream d(message);
+    int command;
+    int engineId;
+    d >> command >> engineId;
+    QQmlEngine *engine = qobject_cast<QQmlEngine *>(objectForId(engineId));
+    if (command == StartWaitingEngine && startingEngines.contains(engine)) {
+        startingEngines.removeOne(engine);
+        emit attachedToEngine(engine);
+    } else if (command == StopWaitingEngine && stoppingEngines.contains(engine)) {
+        stoppingEngines.removeOne(engine);
+        emit detachedFromEngine(engine);
+    }
+}
+
+void QQmlEngineControlService::engineAboutToBeAdded(QQmlEngine *engine)
+{
+    QMutexLocker lock(&dataMutex);
+    if (state() == Enabled) {
+        Q_ASSERT(!stoppingEngines.contains(engine));
+        Q_ASSERT(!startingEngines.contains(engine));
+        startingEngines.append(engine);
+        sendMessage(EngineAboutToBeAdded, engine);
+    } else {
+        emit attachedToEngine(engine);
+    }
+}
+
+void QQmlEngineControlService::engineAboutToBeRemoved(QQmlEngine *engine)
+{
+    QMutexLocker lock(&dataMutex);
+    if (state() == Enabled) {
+        Q_ASSERT(!stoppingEngines.contains(engine));
+        Q_ASSERT(!startingEngines.contains(engine));
+        stoppingEngines.append(engine);
+        sendMessage(EngineAboutToBeRemoved, engine);
+    } else {
+        emit detachedFromEngine(engine);
+    }
+}
+
+void QQmlEngineControlService::engineAdded(QQmlEngine *engine)
+{
+    if (state() == Enabled) {
+        QMutexLocker lock(&dataMutex);
+        Q_ASSERT(!startingEngines.contains(engine));
+        Q_ASSERT(!stoppingEngines.contains(engine));
+        sendMessage(EngineAdded, engine);
+    }
+}
+
+void QQmlEngineControlService::engineRemoved(QQmlEngine *engine)
+{
+    if (state() == Enabled) {
+        QMutexLocker lock(&dataMutex);
+        Q_ASSERT(!startingEngines.contains(engine));
+        Q_ASSERT(!stoppingEngines.contains(engine));
+        sendMessage(EngineRemoved, engine);
+    }
+}
+
+void QQmlEngineControlService::sendMessage(QQmlEngineControlService::MessageType type, QQmlEngine *engine)
+{
+    QByteArray message;
+    QQmlDebugStream d(&message, QIODevice::WriteOnly);
+    d << type << idForObject(engine);
+    emit messageToClient(name(), message);
+}
+
+void QQmlEngineControlService::stateChanged(State)
+{
+    // We flush everything for any kind of state change, to avoid complicated timing issues.
+    QMutexLocker lock(&dataMutex);
+    foreach (QQmlEngine *engine, startingEngines)
+        emit attachedToEngine(engine);
+    startingEngines.clear();
+    foreach (QQmlEngine *engine, stoppingEngines)
+        emit detachedFromEngine(engine);
+    stoppingEngines.clear();
+}
+
+QT_END_NAMESPACE