plugins/Unity/categories.cpp · unity8

git.iko.soy/unity8

on ed85b03 help sign in

unity8/plugins/Unity/categories.cpp

247 lines 8.9 kB history raw

1/*

3 *

4 * Authors:

5 * Michał Sawicz <michal.sawicz@canonical.com>

6 * Michal Hruby <michal.hruby@canonical.com>

7 *

8 * This program is free software; you can redistribute it and/or modify

9 * it under the terms of the GNU General Public License as published by

10 * the Free Software Foundation; version 3.

11 *

12 * This program is distributed in the hope that it will be useful,

13 * but WITHOUT ANY WARRANTY; without even the implied warranty of

14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

15 * GNU General Public License for more details.

16 *

17 * You should have received a copy of the GNU General Public License

18 * along with this program. If not, see <http://www.gnu.org/licenses/>.

19 */

21// self

22#include "categories.h"

23#include "categoryresults.h"

24#include <QDebug>

26// TODO: use something from libunity once it's public

27enum CategoryColumn {

28 ID,

29 DISPLAY_NAME,

30 ICON_HINT,

31 RENDERER_NAME,

32 HINTS

33};

35Categories::Categories(QObject* parent)

36 : DeeListModel(parent)

37{

38 m_roles[Categories::RoleCategoryId] = "categoryId";

39 m_roles[Categories::RoleName] = "name";

40 m_roles[Categories::RoleIcon] = "icon";

41 m_roles[Categories::RoleRenderer] = "renderer";

42 m_roles[Categories::RoleContentType] = "contentType";

43 m_roles[Categories::RoleProgressSource] = "progressSource";

44 m_roles[Categories::RoleHints] = "hints";

45 m_roles[Categories::RoleResults] = "results";

46 m_roles[Categories::RoleCount] = "count";

48 // TODO This should not be needed but accumulatting the count changes

49 // makes the visualization more stable and also makes crashes on fast

50 // change of the search term harder to reproduce

51 m_timer.setSingleShot(true);

52 m_timer.setInterval(50);

53 connect(&m_timer, &QTimer::timeout, this, &Categories::onEmitCountChanged);

54}

56DeeListModel*

57Categories::getResults(int index) const

58{

59 if (!m_results.contains(index)) {

60 CategoryResults* results = new CategoryResults(const_cast<Categories*>(this));

61 results->setCategoryIndex(index);

62 connect(results, &DeeListModel::countChanged, this, &Categories::onCountChanged);

64 unsigned categoryIndex = static_cast<unsigned>(index);

65 auto unity_results = m_unityScope->GetResultsForCategory(categoryIndex);

66 if (unity_results) {

67 results->setModel(unity_results->model());

68 } else {

69 // No results model returned by unity core; this can be the case when the global

70 // results model of this scope became invalid in unity core. Don't set backend

71 // model in DeeListModel - it will still beahve properly as an empty model. Since

72 // we're connected to the results model change signal, we'll reset category models

73 // when results model is set again.

74 qWarning() << "No results model for category" << categoryIndex;

75 }

77 m_results.insert(index, results);

78 }

80 return m_results[index];

81}

83void Categories::onCategoriesModelChanged(unity::glib::Object<DeeModel> model)

84{

85 m_updatedCategories.clear();

86 // FIXME: this might destroy the renderer view and re-create it, optimize?

87 Q_FOREACH(DeeListModel* model, m_results) {

88 delete model;

89 }

90 m_results.clear();

91 setModel(model);

92}

95void Categories::onScopeResultsModelChanged(unity::glib::Object<DeeModel> /* model */)

96{

97 // if scope results model changes, we need to reset all category models

98 onCategoriesModelChanged(m_unityScope->categories()->model());

99}

100

101void

102Categories::setUnityScope(const unity::dash::Scope::Ptr& scope)

103{

104 m_unityScope = scope;

105

106 // no need to call this, we'll get notified

107 //setModel(m_unityScope->categories()->model());

108

109 m_categoriesChangedConnection.disconnect();

110 m_categoriesChangedConnection =

111 m_unityScope->categories()->model.changed.connect(sigc::mem_fun(this, &Categories::onCategoriesModelChanged));

112

113 m_resultshangedConnection.disconnect();

114 m_resultshangedConnection = m_unityScope->results()->model.changed.connect(sigc::mem_fun(this, &Categories::onScopeResultsModelChanged));

115}

116

117void

118Categories::onCountChanged()

119{

120 CategoryResults* results = qobject_cast<CategoryResults*>(sender());

121 if (results) {

122 m_updatedCategories << results->categoryIndex();

123 m_timer.start();

124 }

125}

126

127void Categories::onRowCountChanged()

128{

129 QAbstractItemModel* model = qobject_cast<QAbstractItemModel*>(sender());

130 // find the corresponding category index

131 for (auto iter = m_overriddenCategories.begin(); iter != m_overriddenCategories.end(); ++iter) {

132 if (iter.value() == model) {

133 for (int i = 0; i < rowCount(); i++) {

134 auto id = data(index(i), RoleCategoryId).toString();

135 if (id != iter.key()) continue;

136 QVector<int> roles;

137 roles.append(RoleCount);

138 QModelIndex changedIndex = index(i);

139 Q_EMIT dataChanged(changedIndex, changedIndex, roles);

140 break;

141 }

142 }

143 }

144}

145

146void

147Categories::onEmitCountChanged()

148{

149 QVector<int> roles;

150 roles.append(Categories::RoleCount);

151 Q_FOREACH(int categoryIndex, m_updatedCategories) {

152 if (!m_results.contains(categoryIndex)) continue;

153 QModelIndex changedIndex = index(categoryIndex);

154 Q_EMIT dataChanged(changedIndex, changedIndex, roles);

155 }

156 m_updatedCategories.clear();

157}

158

159QHash<int, QByteArray>

160Categories::roleNames() const

161{

162 return m_roles;

163}

164

165void Categories::onOverrideModelDestroyed()

166{

167 QObject* model = sender();

168 auto iter = m_overriddenCategories.begin();

169 while (iter != m_overriddenCategories.end()) {

170 if (iter.value() == model) {

171 iter = m_overriddenCategories.erase(iter);

172 continue;

173 }

174 ++iter;

175 }

176}

177

178void Categories::overrideResults(const QString& categoryId, QAbstractItemModel* model)

179{

180 m_overriddenCategories[categoryId] = model;

181 // watch the model

182 connect(model, &QObject::destroyed, this, &Categories::onOverrideModelDestroyed);

183 connect(model, &QAbstractItemModel::rowsInserted, this, &Categories::onRowCountChanged);

184 connect(model, &QAbstractItemModel::rowsRemoved, this, &Categories::onRowCountChanged);

185 connect(model, &QAbstractItemModel::modelReset, this, &Categories::onRowCountChanged);

186

187 // emit the dataChanged signal if the category is already in the model

188 for (int i = 0; i < rowCount(); i++) {

189 auto id = data(index(i), RoleCategoryId).toString();

190 if (id != categoryId) continue;

191 QVector<int> roles;

192 roles.append(RoleCount);

193 roles.append(RoleResults);

194 QModelIndex changedIndex = index(i);

195 Q_EMIT dataChanged(changedIndex, changedIndex, roles);

196 break;

197 }

198}

199

200QVariant

201Categories::data(const QModelIndex& index, int role) const

202{

203 if (!index.isValid()) {

204 return QVariant();

205 }

206

207 switch (role) {

208 case RoleCategoryId:

209 return DeeListModel::data(index, CategoryColumn::ID);

210 case RoleName:

211 return DeeListModel::data(index, CategoryColumn::DISPLAY_NAME);

212 case RoleIcon:

213 return DeeListModel::data(index, CategoryColumn::ICON_HINT);

214 case RoleRenderer:

215 return DeeListModel::data(index, CategoryColumn::RENDERER_NAME);

216 case RoleContentType:

217 {

218 auto hints = DeeListModel::data(index, CategoryColumn::HINTS).toHash();

219 return hints.contains("content-type") ? hints["content-type"] : QVariant(QString("default"));

220 }

221 case RoleProgressSource:

222 {

223 auto hints = DeeListModel::data(index, CategoryColumn::HINTS).toHash();

224 return hints.contains("progress-source") ? hints["progress-source"] : QVariant();

225 }

226 case RoleHints:

227 return DeeListModel::data(index, CategoryColumn::HINTS);

228 case RoleResults:

229 if (m_overriddenCategories.size() > 0)

230 {

231 auto id = DeeListModel::data(index, CategoryColumn::ID).toString();

232 if (m_overriddenCategories.find(id) != m_overriddenCategories.end())

233 return QVariant::fromValue(m_overriddenCategories[id]);

234 }

235 return QVariant::fromValue(getResults(index.row()));

236 case RoleCount:

237 if (m_overriddenCategories.size() > 0)

238 {

239 auto id = DeeListModel::data(index, CategoryColumn::ID).toString();

240 if (m_overriddenCategories.find(id) != m_overriddenCategories.end())

241 return QVariant::fromValue(m_overriddenCategories[id]->rowCount());

242 }

243 return QVariant::fromValue(getResults(index.row())->rowCount());

244 default:

245 return QVariant();

246 }

247}