@@ -29,13 +29,13 @@
30#include "Window.h"
30#include "Window.h"
31#include "Workspace.h"
31#include "Workspace.h"
32−#include "windowmanagementpolicy.h"
32+#include "wmpolicyinterface.h"
35#include <qtmir/mirserverapplication.h>
35#include <qtmir/mirserverapplication.h>
36#include <qtmir/windowmodelnotifier.h>
36#include <qtmir/windowmodelnotifier.h>
38−Q_LOGGING_CATEGORY(TOPLEVELWINDOWMODEL, "toplevelwindowmodel")
38+Q_LOGGING_CATEGORY(TOPLEVELWINDOWMODEL, "toplevelwindowmodel", QtInfoMsg)
40#define DEBUG_MSG qCDebug(TOPLEVELWINDOWMODEL).nospace().noquote() << __func__
40#define DEBUG_MSG qCDebug(TOPLEVELWINDOWMODEL).nospace().noquote() << __func__
41#define INFO_MSG qCInfo(TOPLEVELWINDOWMODEL).nospace().noquote() << __func__
41#define INFO_MSG qCInfo(TOPLEVELWINDOWMODEL).nospace().noquote() << __func__
@@ -109,6 +109,8 @@ void TopLevelWindowModel::setSurfaceManager(unityapi::SurfaceManagerInterface *s
110 connect(m_surfaceManager, &unityapi::SurfaceManagerInterface::surfacesAddedToWorkspace,
110 connect(m_surfaceManager, &unityapi::SurfaceManagerInterface::surfacesAddedToWorkspace,
111 this, &TopLevelWindowModel::onSurfacesAddedToWorkspace);
111 this, &TopLevelWindowModel::onSurfacesAddedToWorkspace);
112+ connect(m_surfaceManager, &unityapi::SurfaceManagerInterface::surfacesAboutToBeRemovedFromWorkspace,
113+ this, &TopLevelWindowModel::onSurfacesAboutToBeRemovedFromWorkspace);
113 if (m_workspace && m_applicationManager) {
115 if (m_workspace && m_applicationManager) {
114 refreshWindows();
116 refreshWindows();
@@ -397,6 +399,7 @@ void TopLevelWindowModel::onSurfaceCreated(unityapi::MirSurfaceInterface */*surf
397void TopLevelWindowModel::onSurfacesAddedToWorkspace(const std::shared_ptr<miral::Workspace>& workspace,
399void TopLevelWindowModel::onSurfacesAddedToWorkspace(const std::shared_ptr<miral::Workspace>& workspace,
398 const QVector<unity::shell::application::MirSurfaceInterface*> surfaces)
400 const QVector<unity::shell::application::MirSurfaceInterface*> surfaces)
402+ if (!m_workspace || !m_applicationManager) return;
400 if (workspace != m_workspace->workspace()) return;
403 if (workspace != m_workspace->workspace()) return;
402 Q_FOREACH(auto surface, surfaces) {
405 Q_FOREACH(auto surface, surfaces) {
@@ -431,6 +434,59 @@ void TopLevelWindowModel::onSurfacesAddedToWorkspace(const std::shared_ptr<miral
437+void TopLevelWindowModel::onSurfacesAboutToBeRemovedFromWorkspace(const std::shared_ptr<miral::Workspace> &workspace,
438+ const QVector<unity::shell::application::MirSurfaceInterface *> surfaces)
440+ if (workspace != m_workspace->workspace()) return;
444+ for (auto iter = surfaces.constBegin(); iter != surfaces.constEnd();) {
445+ auto surface = *iter;
448+ // Do removals in adjacent blocks.
449+ start = end = indexOf(surface);
451+ // probably a child surface
452+ m_allSurfaces.remove(surface);
455+ while(iter != surfaces.constEnd()) {
456+ int index = indexOf(*iter);
457+ if (index != end+1) {
464+ if (m_modelState == IdleState) {
465+ beginRemoveRows(QModelIndex(), start, end);
466+ m_modelState = RemovingState;
468+ Q_ASSERT(m_modelState == ResettingState);
469+ // No point in signaling anything if we're resetting the whole model
472+ for (int index = start; index <= end; index++) {
473+ auto window = m_windowModel[start].window;
474+ window->setSurface(nullptr);
475+ window->setFocused(false);
477+ m_windowModel.removeAt(start);
478+ m_allSurfaces.remove(window->surface());
481+ if (m_modelState == RemovingState) {
483+ Q_EMIT countChanged();
484+ Q_EMIT listChanged();
485+ m_modelState = IdleState;
434void TopLevelWindowModel::removeAt(int index)
490void TopLevelWindowModel::removeAt(int index)
436 if (m_modelState == IdleState) {
492 if (m_modelState == IdleState) {
@@ -774,7 +830,8 @@ void TopLevelWindowModel::refreshWindows()
774 m_windowModel.clear();
830 m_windowModel.clear();
775 m_allSurfaces.clear();
831 m_allSurfaces.clear();
833+ if (!m_workspace || !m_applicationManager) return;
777− WindowManagementPolicy::instance()->forEachWindowInWorkspace(m_workspace->workspace(), [this](const miral::Window &window) {
834+ WMPolicyInterface::instance()->forEachWindowInWorkspace(m_workspace->workspace(), [this](const miral::Window &window) {
778 auto surface = m_surfaceManager->surfaceFor(window);
835 auto surface = m_surfaceManager->surfaceFor(window);
779 if (surface) {
836 if (surface) {
780 if (surface->parentSurface()) {
837 if (surface->parentSurface()) {