File indexing completed on 2024-05-12 17:07:45

 /*
  *   SPDX-FileCopyrightText: 2021 Aleix Pol <apol@kde.org>
  *   SPDX-FileCopyrightText: 2021 Devin Lin <devin@kde.org>
  *
  *   SPDX-License-Identifier: GPL-2.0-or-later
  */
 
 #include "displaysmodel.h"
 
 #include <QGuiApplication>
 
 DisplaysModel::DisplaysModel(QObject *parent)
     : QAbstractListModel(parent)
 {
     if (!QGuiApplication::platformName().startsWith(QLatin1String("wayland"), Qt::CaseInsensitive)) {
         return;
     }
 
     using namespace KWayland::Client;
     ConnectionThread *connection = ConnectionThread::fromApplication(this);
 
     if (!connection) {
         return;
     }
 
     auto *registry = new Registry(this);
     registry->create(connection);
 
     connect(registry, &Registry::outputAnnounced, this, [this, registry](quint32 name, quint32 version) {
         createOutput(registry->bindOutput(name, version));
     });
     connect(registry, &Registry::plasmaWindowManagementAnnounced, this, [this, registry](quint32 name, quint32 version) {
         m_windowManagement = registry->createPlasmaWindowManagement(name, version, this);
     });
 
     registry->setup();
     connection->roundtrip();
 }
 
 QHash<int, QByteArray> DisplaysModel::roleNames() const
 {
     return {
         {Model, "modelName"},
         {Geometry, "geometry"},
         {Position, "position"},
         {Output, "output"},
     };
 }
 
 void DisplaysModel::createOutput(wl_output *output)
 {
     auto newOutput = new KWayland::Client::Output(this);
     connect(newOutput, &KWayland::Client::Output::removed, this, [this, newOutput] {
         auto i = m_outputs.indexOf(newOutput);
         Q_ASSERT(i >= 0);
         beginRemoveRows({}, i, i);
         m_outputs.removeAt(i);
         endRemoveRows();
     });
     newOutput->setup(output);
     beginInsertRows({}, m_outputs.count(), m_outputs.count());
     m_outputs.append(newOutput);
     endInsertRows();
 }
 
 void DisplaysModel::sendWindowToOutput(const QString &uuid, KWayland::Client::Output *output)
 {
     const auto windows = m_windowManagement->windows();
     for (auto w : windows) {
         if (w->uuid() == uuid) {
             w->sendToOutput(output);
         }
     }
 }
 
 int DisplaysModel::rowCount(const QModelIndex &parent) const
 {
     return parent.isValid() ? 0 : m_outputs.count();
 }
 
 QVariant DisplaysModel::data(const QModelIndex &index, int role) const
 {
     if (index.row() >= m_outputs.count()) {
         return {};
     }
 
     auto o = m_outputs[index.row()];
     switch (role) {
     case Model:
         return o->model();
     case Geometry:
         return o->geometry();
     case Position:
         return o->globalPosition();
     case Output:
         return QVariant::fromValue<QObject *>(o);
     }
     return {};
 }