@@ -35,7 +35,7 @@
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();
@@ -431,6 +433,59 @@ void TopLevelWindowModel::onSurfacesAddedToWorkspace(const std::shared_ptr<miral
436+void TopLevelWindowModel::onSurfacesAboutToBeRemovedFromWorkspace(const std::shared_ptr<miral::Workspace> &workspace,
437+ const QVector<unity::shell::application::MirSurfaceInterface *> surfaces)
439+ if (workspace != m_workspace->workspace()) return;
443+ for (auto iter = surfaces.constBegin(); iter != surfaces.constEnd();) {
444+ auto surface = *iter;
447+ // Do removals in adjacent blocks.
448+ start = end = indexOf(surface);
450+ // probably a child surface
451+ m_allSurfaces.remove(surface);
454+ while(iter != surfaces.constEnd()) {
455+ int index = indexOf(*iter);
456+ if (index != end+1) {
463+ if (m_modelState == IdleState) {
464+ beginRemoveRows(QModelIndex(), start, end);
465+ m_modelState = RemovingState;
467+ Q_ASSERT(m_modelState == ResettingState);
468+ // No point in signaling anything if we're resetting the whole model
471+ for (int index = start; index <= end; index++) {
472+ auto window = m_windowModel[start].window;
473+ window->setSurface(nullptr);
474+ window->setFocused(false);
476+ m_windowModel.removeAt(start);
477+ m_allSurfaces.remove(window->surface());
480+ if (m_modelState == RemovingState) {
482+ Q_EMIT countChanged();
483+ Q_EMIT listChanged();
484+ m_modelState = IdleState;
434void TopLevelWindowModel::removeAt(int index)
489void TopLevelWindowModel::removeAt(int index)
436 if (m_modelState == IdleState) {
491 if (m_modelState == IdleState) {