plugins/Wizard/timezonemodel.cpp · unity8

git.iko.soy/unity8

on a91dd21 help sign in

unity8/plugins/Wizard/timezonemodel.cpp

212 lines 6.0 kB history raw

1/*

3 *

4 * This program is free software: you can redistribute it and/or modify it

5 * under the terms of the GNU General Public License version 3, as published

6 * by the Free Software Foundation.

7 *

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

9 * WITHOUT ANY WARRANTY; without even the implied warranties of

10 * MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR

11 * PURPOSE. See the GNU General Public License for more details.

12 *

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

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

15 */

17#include <QDebug>

19#include <glib.h>

20#include <glib-object.h>

21#include <timezonemap/tz.h>

23#include "LocalePlugin.h"

24#include "timezonemodel.h"

27TimeZoneLocationModel::TimeZoneLocationModel(QObject *parent):

28 QAbstractListModel(parent),

29 m_workerThread(new TimeZonePopulateWorker())

30{

31 qRegisterMetaType<TzLocation>();

33 m_roleNames[Qt::DisplayRole] = "displayName";

34 m_roleNames[TimeZoneRole] = "timeZone";

35 m_roleNames[CityRole] = "city";

36 m_roleNames[CountryRole] = "country";

38 QObject::connect(m_workerThread,

39 &TimeZonePopulateWorker::resultReady,

40 this,

41 &TimeZoneLocationModel::processModelResult);

42 QObject::connect(m_workerThread,

43 &TimeZonePopulateWorker::finished,

44 this,

45 &TimeZoneLocationModel::store);

46 QObject::connect(m_workerThread,

47 &TimeZonePopulateWorker::finished,

48 m_workerThread,

49 &QObject::deleteLater);

51 init();

52}

54void TimeZoneLocationModel::init()

55{

56 beginResetModel();

57 m_workerThread->start();

58}

60void TimeZoneLocationModel::store()

61{

62 m_workerThread = nullptr;

63 endResetModel();

64}

66void TimeZoneLocationModel::processModelResult(const TzLocation &location)

67{

68 m_locations.append(location);

69}

71int TimeZoneLocationModel::rowCount(const QModelIndex &parent) const

72{

73 if (parent.isValid())

74 return 0;

75 return m_locations.count();

76}

78QVariant TimeZoneLocationModel::data(const QModelIndex &index, int role) const

79{

80 if (index.row() >= m_locations.count() || index.row() < 0)

81 return QVariant();

83 const TzLocation tz = m_locations.at(index.row());

85 const QString country(tz.full_country.isEmpty() ? tz.country : tz.full_country);

87 switch (role) {

88 case Qt::DisplayRole:

89 if (!tz.state.isEmpty())

90 return QStringLiteral("%1, %2, %3").arg(tz.city).arg(tz.state).arg(country);

91 else

92 return QStringLiteral("%1, %2").arg(tz.city).arg(country);

93 case SimpleRole:

94 return QStringLiteral("%1, %2").arg(tz.city).arg(country);

95 case TimeZoneRole:

96 return tz.timezone;

97 case CountryRole:

98 return tz.country;

99 case CityRole:

100 return tz.city;

101 default:

102 qWarning() << Q_FUNC_INFO << "Unknown role";

103 return QVariant();

104 }

105}

106

107QHash<int, QByteArray> TimeZoneLocationModel::roleNames() const

108{

109 return m_roleNames;

110}

111

112void TimeZonePopulateWorker::run()

113{

114 buildCityMap();

115}

116

117void TimeZonePopulateWorker::buildCityMap()

118{

119 TzDB *tzdb = tz_load_db();

120 GPtrArray *tz_locations = tz_get_locations(tzdb);

121

122 TimeZoneLocationModel::TzLocation tmpTz;

123

124 for (guint i = 0; i < tz_locations->len; ++i) {

125 auto tmp = static_cast<CcTimezoneLocation *>(g_ptr_array_index(tz_locations, i));

126 gchar *en_name, *country, *zone, *state, *full_country;

127 g_object_get (tmp, "en_name", &en_name,

128 "country", &country,

129 "zone", &zone,

130 "state", &state,

131 "full_country", &full_country,

132 nullptr);

133 // There are empty entries in the DB

134 if (g_strcmp0(en_name, "") != 0) {

135 tmpTz.city = en_name;

136 tmpTz.country = country;

137 tmpTz.timezone = zone;

138 tmpTz.state = state;

139 tmpTz.full_country = full_country;

140

141 Q_EMIT (resultReady(tmpTz));

142 }

143 g_free (en_name);

144 g_free (country);

145 g_free (zone);

146 g_free (state);

147 g_free (full_country);

148 }

149

150 g_ptr_array_free (tz_locations, TRUE);

151 tz_db_free(tzdb);

152}

153

154

155TimeZoneFilterModel::TimeZoneFilterModel(QObject *parent)

156 : QSortFilterProxyModel(parent)

157{

158 setDynamicSortFilter(false);

159 setSortLocaleAware(true);

160 setSortRole(TimeZoneLocationModel::CityRole);

161 m_stringMatcher.setCaseSensitivity(Qt::CaseInsensitive);

162 sort(0);

163}

164

165bool TimeZoneFilterModel::filterAcceptsRow(int row, const QModelIndex &parentIndex) const

166{

167 if (!sourceModel()) {

168 return true;

169 }

170

171 if (!m_filter.isEmpty()) { // filtering by freeform text input, cf setFilter(QString)

172 const QString city = sourceModel()->index(row, 0, parentIndex).data(TimeZoneLocationModel::CityRole).toString();

173

174 if (m_stringMatcher.indexIn(city) == 0) { // match at the beginning of the city name

175 return true;

176 }

177 } else if (!m_country.isEmpty()) { // filter by country code

178 const QString countryCode = sourceModel()->index(row, 0, parentIndex).data(TimeZoneLocationModel::CountryRole).toString();

179 return m_country.compare(countryCode, Qt::CaseInsensitive) == 0;

180 }

181

182 return false;

183}

184

185QString TimeZoneFilterModel::filter() const

186{

187 return m_filter;

188}

189

190void TimeZoneFilterModel::setFilter(const QString &filter)

191{

192 if (filter != m_filter) {

193 m_filter = filter;

194 m_stringMatcher.setPattern(m_filter);

195 Q_EMIT filterChanged();

196 invalidate();

197 }

198}

199

200QString TimeZoneFilterModel::country() const

201{

202 return m_country;

203}

204

205void TimeZoneFilterModel::setCountry(const QString &country)

206{

207 if (m_country == country)

208 return;

209

210 m_country = country;

211 Q_EMIT countryChanged(country);

212}