A header with the proper copyright/lincence statement was added into each source...
[toast/confclerk.git] / src / mvc / eventmodel.cpp
1 /*
2  * Copyright (C) 2010 Ixonos Plc.
3  *
4  * This file is part of fosdem-schedule.
5  *
6  * fosdem-schedule is free software: you can redistribute it and/or modify it
7  * under the terms of the GNU General Public License as published by the Free
8  * Software Foundation, either version 2 of the License, or (at your option)
9  * any later version.
10  *
11  * fosdem-schedule is distributed in the hope that it will be useful, but
12  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
14  * more details.
15  *
16  * You should have received a copy of the GNU General Public License along with
17  * fosdem-schedule.  If not, see <http://www.gnu.org/licenses/>.
18  */
19 #include "eventmodel.h"
20 #include <conference.h>
21 #include <track.h>
22 #include <room.h>
23
24 const QString EventModel::COMMA_SEPARATOR = ", ";
25
26 EventModel::EventModel()
27 {
28     mEvents.clear();
29 }
30
31 void EventModel::createTimeGroups()
32 {
33     mGroups.clear();
34     mParents.clear();
35
36     if (mEvents.empty())
37     {
38         return;
39     }
40
41     const int timeSpan = 5400;
42
43     QTime startTime = mEvents.first().start().time();
44     mGroups << EventModel::Group(QString("%1 - %2").arg(startTime.toString("HH:mm"),
45         startTime.addSecs(timeSpan).toString("HH:mm")), 0);
46     QTime nextGroupTime = mEvents.first().start().time().addSecs(timeSpan);
47
48     for (int i=0; i<mEvents.count(); i++)
49     {
50         QTime eventTime = mEvents.at(i).start().time();
51
52         if (nextGroupTime <= eventTime)
53         {
54             mGroups.last().mChildCount = i - mGroups.last().mFirstEventIndex;
55             mGroups << EventModel::Group(QString("%1 - %2").arg(nextGroupTime.toString("HH:mm"),
56                 nextGroupTime.addSecs(timeSpan).toString("HH:mm")), i);
57             nextGroupTime = nextGroupTime.addSecs(timeSpan);
58         }
59
60         // add parent-child relation
61         mParents[mEvents.at(i).id()] = mGroups.count() - 1;
62     }
63
64     mGroups.last().mChildCount = mEvents.count() - mGroups.last().mFirstEventIndex;
65 }
66
67 void EventModel::createTrackGroups() {
68     mGroups.clear();
69     mParents.clear();
70     if (mEvents.empty())
71     {
72         return;
73     }
74     int trackId = mEvents.first().trackId();
75
76     mGroups << EventModel::Group(Track::retrieveTrackName(trackId), 0);
77     int nextTrackId = trackId;
78
79     for (int i=0; i<mEvents.count(); i++)
80     {
81         trackId = mEvents.at(i).trackId();
82         if (nextTrackId != trackId)
83         {
84             mGroups.last().mChildCount = i - mGroups.last().mFirstEventIndex;
85             mGroups << EventModel::Group(Track::retrieveTrackName(trackId), i);
86             nextTrackId = trackId;
87         }
88         // add parent-child relation
89         mParents[mEvents.at(i).id()] = mGroups.count() - 1;
90     }
91     mGroups.last().mChildCount = mEvents.count() - mGroups.last().mFirstEventIndex;
92 }
93
94 void EventModel::createRoomGroups()
95 {
96     mGroups.clear();
97     mParents.clear();
98     if (mEvents.empty())
99     {
100         return;
101     }
102     int roomId = mEvents.first().roomId();
103
104     mGroups << EventModel::Group(Room::retrieveRoomName(roomId), 0);
105     int nextRoomId = roomId;
106
107     QList<Event>::iterator event = mEvents.begin();
108     int i = 0;
109     while (event != mEvents.end())
110     {
111         roomId = event->roomId();
112         if (nextRoomId != roomId)
113         {
114             mGroups.last().mChildCount = i - mGroups.last().mFirstEventIndex;
115             mGroups << EventModel::Group(Room::retrieveRoomName(roomId), i);
116             nextRoomId = roomId;
117         }
118         mParents[event->id()] = mGroups.count() - 1;
119         event++;
120         i++;
121     }
122     mGroups.last().mChildCount = mEvents.count() - mGroups.last().mFirstEventIndex;
123 }
124
125 QVariant EventModel::data(const QModelIndex& index, int role) const
126 {
127     if (index.isValid() && role == Qt::DisplayRole)
128     {
129         if (index.internalId() == 0)
130         {
131             return mGroups.at(index.row()).mTitle;
132         }
133         else //event data
134         {
135             return static_cast<Event*>(index.internalPointer())->id();
136         }
137     }
138
139     return QVariant();
140 }
141
142 QModelIndex EventModel::index(int row, int column, const QModelIndex& parent) const
143 {
144     // TODO: add checks for out of range rows
145
146     if (!parent.isValid())
147     {
148         return createIndex(row, column, 0);
149     }
150     else if (parent.internalId() == 0)
151     {
152         const Group& group = mGroups.at(parent.row());
153         Event* event = const_cast<Event*>(&mEvents.at(row + group.mFirstEventIndex));
154         return createIndex(row, column, reinterpret_cast<void*>(event));
155     }
156     else
157     {
158         return QModelIndex();
159     }
160 }
161
162 QModelIndex EventModel::parent(const QModelIndex & index) const
163 {
164     if (index.isValid())
165     {
166         if (index.internalId() == 0)
167         {
168             return QModelIndex();
169         }
170
171         Event * event = static_cast<Event*>(index.internalPointer());
172
173         return createIndex(mParents[event->id()], 0, 0);
174     }
175
176     return QModelIndex();
177 }
178
179 int EventModel::columnCount(const QModelIndex & parent) const
180 {
181     Q_UNUSED(parent);
182     return 1;
183 }
184
185 int EventModel::rowCount (const QModelIndex & parent) const
186 {
187     if (!parent.isValid())
188     {
189         return mGroups.count();
190     }
191
192     if (parent.internalId() == 0)
193     {
194         return mGroups.at(parent.row()).mChildCount;
195     }
196
197     return 0;
198 }
199
200 void EventModel::clearModel()
201 {
202     for(int i = 0;i < mGroups.count();i++){
203         QModelIndex idx = index(i, 0);
204         Group group = mGroups[i];
205         beginRemoveRows(idx, 0, group.mChildCount - 1);
206         /*bool ok =*/ removeRows(0, group.mChildCount, idx);
207         endRemoveRows();
208         //qDebug() << "removing " << group.mChildCount << " events from group:" << i << idx.data() << ":" << ok;
209     }
210     mEvents.clear();
211 }
212
213 void EventModel::loadEvents(const QDate &aDate, int aConferenceId)
214 {
215     clearModel();
216     // check for existence of the conference in the DB
217     if(Conference::getAll().count())
218     {
219         qDebug() << "Loading Conference Data: [" << Conference::getById(aConferenceId).title() << "] " << aDate;
220         mEvents = Event::getByDate(QDate(aDate.year(), aDate.month(), aDate.day()), aConferenceId, "start");
221     }
222     createTimeGroups();
223 }
224
225 void EventModel::loadFavEvents(const QDate &aDate, int aConferenceId)
226 {
227     clearModel();
228     // check for existence of the conference in the DB
229     if(Conference::getAll().count())
230     {
231         qDebug() << "Loading FAV Conference Data: [" << Conference::getById(aConferenceId).title() << "] " << aDate;
232         mEvents = Event::getFavByDate(QDate(aDate.year(), aDate.month(), aDate.day()), aConferenceId);
233     }
234     createTimeGroups();
235 }
236
237 int EventModel::loadSearchResultEvents(const QDate &aDate, int aConferenceId)
238 {
239     clearModel();
240     // check for existence of the conference in the DB
241     if(Conference::getAll().count())
242     {
243         qDebug() << "Loading search result Data: [" << Conference::getById(aConferenceId).title() << "] " << aDate;
244         try{
245             mEvents = Event::getSearchResultByDate(QDate(aDate.year(), aDate.month(), aDate.day()), aConferenceId, "start");
246         }
247         catch( OrmException &e  ){
248             qDebug() << "Event::getSearchResultByDate failed: " << e.text();
249         }
250         catch(...){
251             qDebug() << "Event::getSearchResultByDate failed";
252         }
253
254     }
255
256     createTimeGroups();
257
258     return mEvents.count();
259 }
260
261 void EventModel::loadEventsByTrack(const QDate &aDate, int aConferenceId)
262 {
263     clearModel();
264     if (Conference::getAll().count())
265     {
266         qDebug() << "Loading Conference Data (by Track): [" << Conference::getById(aConferenceId).title() << "] " << aDate;
267         mEvents = Event::getByDate(QDate(aDate.year(), aDate.month(), aDate.day()), aConferenceId, "xid_track, start");
268     }
269     createTrackGroups();
270 }
271
272 void EventModel::loadEventsByRoom(const QDate &aDate, int aConferenceId)
273 {
274     clearModel();
275     if (Conference::getAll().count())
276     {
277         qDebug() << "Loading Conference Data (by Room): [" << Conference::getById(aConferenceId).title() << "] " << aDate;
278         mEvents = Event::getByDateAndRoom(QDate(aDate.year(), aDate.month(), aDate.day()), aConferenceId);
279     }
280     createRoomGroups();
281 }
282
283 void EventModel::loadNowEvents(int aConferenceId)
284 {
285     clearModel();
286     // check for existence of the conference in the DB
287     if(Conference::getAll().count())
288     {
289         qDebug() << "Loading Conference Data: [" << Conference::getById(aConferenceId).title() << "] scheduled NOW";
290         mEvents = Event::nowEvents(aConferenceId, "start");
291     }
292     createTimeGroups();
293 }
294
295 void EventModel::loadConflictEvents(int aEventId, int aConferenceId)
296 {
297     clearModel();
298     // check for existence of the conference in the DB
299     if(Conference::getAll().count())
300     {
301         qDebug() << "Loading Conference Data: [" << Conference::getById(aConferenceId).title() << "] in conflict with " << aEventId;
302         mEvents = Event::conflictEvents(aEventId, aConferenceId);
303     }
304     createTimeGroups();
305 }
306
307 void EventModel::updateModel(int aEventId)
308 {
309     for(int i=0; i<mEvents.count(); i++)
310     {
311         if(mEvents[i].id() == aEventId)
312             mEvents[i] = Event::getById(aEventId,Conference::activeConference());
313     }
314
315     // find the ModelIndex for given aEventId
316     for(int i=0; i<mGroups.count(); i++)
317     {
318         QModelIndex groupIndex = index(i,0,QModelIndex());
319         for(int j=0; j<mGroups[i].mChildCount; j++)
320         {
321             QModelIndex eventIndex = index(j,0,groupIndex);
322             if(static_cast<Event*>(eventIndex.internalPointer())->id() == aEventId)
323             {
324                 emit(dataChanged(groupIndex,groupIndex));
325                 emit(dataChanged(eventIndex,eventIndex));
326             }
327         }
328     }
329 }
330