git.iko.soy/unity8
on e225248 help sign in
unity8/plugins/LightDM/SessionsModel.cpp
90 lines 2.8 kB history raw
1/*
2 * Copyright (C) 2015 Canonical, Ltd.
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; version 3.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public License
14 * along with this program. If not, see <http://www.gnu.org/licenses/>.
15 *
16 */
18#include "SessionsModel.h"
19#include <QtCore/QFile>
20#include <QtCore/QSortFilterProxyModel>
21QHash<int, QByteArray> SessionsModel::roleNames() const
23 return m_roleNames;
26int SessionsModel::rowCount(const QModelIndex& parent) const
28 return m_model->rowCount(parent);
31// Used for testing
32QList<QUrl> SessionsModel::iconSearchDirectories() const
34 return m_iconSearchDirectories;
37QModelIndex SessionsModel::sessionIndex(QString sessionName) const
39 for (int i = 0; i < rowCount(QModelIndex()); i ++) {
40 QModelIndex index = m_model->index(i, Qt::DisplayRole);
41 QString aSession(this->data(index, Qt::DisplayRole).toString());
42 if (!QString::compare(aSession, sessionName, Qt::CaseInsensitive)) {
43 return index;
44 }
45 }
46 return QModelIndex();
49QUrl SessionsModel::iconUrl(QString sessionName) const
51 Q_FOREACH(const QUrl& searchDirectory, m_iconSearchDirectories)
52 {
53 // This is an established icon naming convention
54 QString iconUrl = searchDirectory.toString(QUrl::StripTrailingSlash) +
55 "/" + sessionName.toLower() + "_badge.png";
56 QFile iconFile(iconUrl);
57 if (iconFile.exists()) {
58 return QUrl(iconUrl);
59 }
60 }
62 // FIXME make this smarter
63 return QUrl("./graphics/session_icons/unknown_badge.png");
66QVariant SessionsModel::data(const QModelIndex& index, int role) const
68 switch (role) {
69 case SessionsModel::IconRole:
70 return QVariant(iconUrl(m_model->data(index, Qt::DisplayRole).toString()));
71 default:
72 return m_model->data(index, role);
73 }
76SessionsModel::SessionsModel(QObject* parent)
77 : UnitySortFilterProxyModelQML(parent)
79 // Add a custom IconRole that isn't in either of the lightdm implementations
80 m_model = new QLightDM::SessionsModel(this);
81 m_roleNames = m_model->roleNames();
82 m_roleNames[IconRole] = "icon_url";
84 setModel(m_model);
85 setSourceModel(m_model);
86 setSortCaseSensitivity(Qt::CaseInsensitive);
87 setSortLocaleAware(true);
88 setSortRole(Qt::DisplayRole);
89 sort(0);