1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel IRC Team *
3 * devel@quassel-irc.org *
5 * This program is free software; you can redistribute it and/or modify *
6 * it under the terms of the GNU General Public License as published by *
7 * the Free Software Foundation; either version 2 of the License, or *
8 * (at your option) version 3. *
10 * This program is distributed in the hope that it will be useful, *
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
13 * GNU General Public License for more details. *
15 * You should have received a copy of the GNU General Public License *
16 * along with this program; if not, write to the *
17 * Free Software Foundation, Inc., *
18 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
19 ***************************************************************************/
21 #include "messagemodel.h"
27 class ProcessBufferEvent : public QEvent {
29 inline ProcessBufferEvent() : QEvent(QEvent::User) {}
32 MessageModel::MessageModel(QObject *parent)
33 : QAbstractItemModel(parent)
35 QDateTime now = QDateTime::currentDateTime();
36 now.setTimeSpec(Qt::UTC);
37 _nextDayChange.setTimeSpec(Qt::UTC);
38 _nextDayChange.setTime_t(((now.toTime_t() / 86400) + 1) * 86400);
39 _nextDayChange.setTimeSpec(Qt::LocalTime);
40 _dayChangeTimer.setInterval(QDateTime::currentDateTime().secsTo(_nextDayChange) * 1000);
41 _dayChangeTimer.start();
42 connect(&_dayChangeTimer, SIGNAL(timeout()), this, SLOT(changeOfDay()));
45 QVariant MessageModel::data(const QModelIndex &index, int role) const {
46 int row = index.row(); int column = index.column();
47 if(row < 0 || row >= _messageList.count() || column < 0)
50 if(role == ColumnTypeRole)
53 return _messageList[row]->data(index.column(), role);
56 bool MessageModel::setData(const QModelIndex &index, const QVariant &value, int role) {
57 int row = index.row();
58 if(row < 0 || row >= _messageList.count())
61 if(_messageList[row]->setData(index.column(), value, role)) {
62 emit dataChanged(index, index);
71 bool MessageModel::insertMessage(const Message &msg, bool fakeMsg) {
72 MsgId id = msg.msgId();
73 int idx = indexForId(id);
74 if(!fakeMsg && idx < _messageList.count()) { // check for duplicate
75 if(_messageList[idx]->msgId() == id)
79 MessageModelItem *item = createMessageModelItem(msg);
80 beginInsertRows(QModelIndex(), idx, idx);
81 _messageList.insert(idx, item);
86 void MessageModel::insertMessages(const QList<Message> &msglist) {
90 int processedMsgs = insertMessagesGracefully(msglist);
91 int remainingMsgs = msglist.count() - processedMsgs;
92 if(remainingMsgs > 0) {
93 if(msglist.first().msgId() < msglist.last().msgId()) {
95 _messageBuffer << msglist.mid(0, remainingMsgs);
97 _messageBuffer << msglist.mid(processedMsgs);
99 qSort(_messageBuffer);
100 QCoreApplication::postEvent(this, new ProcessBufferEvent());
104 void MessageModel::insertMessageGroup(const QList<Message> &msglist) {
105 int idx = indexForId(msglist.first().msgId());
107 int prevIdx = idx - 1;
108 if(_messageList[prevIdx]->msgType() == Message::DayChange
109 && _messageList[prevIdx]->timeStamp() > msglist.value(0).timestamp()) {
110 beginRemoveRows(QModelIndex(), prevIdx, prevIdx);
111 MessageModelItem *oldItem = _messageList.takeAt(prevIdx);
117 Message dayChangeMsg;
118 bool needsDayChangeMsg = false;
119 if(idx < _messageList.count() && _messageList[idx]->msgType() != Message::DayChange) {
120 QDateTime nextTs = _messageList[idx]->timeStamp();
121 QDateTime prevTs = msglist.last().timestamp();
122 nextTs.setTimeSpec(Qt::UTC);
123 prevTs.setTimeSpec(Qt::UTC);
124 uint nextDay = nextTs.toTime_t() / 86400;
125 uint prevDay = prevTs.toTime_t() / 86400;
126 if(nextDay != prevDay) {
127 nextTs.setTime_t(nextDay * 86400);
128 nextTs.setTimeSpec(Qt::LocalTime);
129 dayChangeMsg = Message::ChangeOfDay(nextTs);
130 dayChangeMsg.setMsgId(msglist.last().msgId());
131 needsDayChangeMsg = true;
135 int end = idx + msglist.count() - 1;
136 beginInsertRows(QModelIndex(), start, end);
137 foreach(Message msg, msglist) {
138 _messageList.insert(idx, createMessageModelItem(msg));
141 if(needsDayChangeMsg) {
142 _messageList.insert(idx, createMessageModelItem(dayChangeMsg));
147 int MessageModel::insertMessagesGracefully(const QList<Message> &msglist) {
148 bool inOrder = (msglist.first().msgId() < msglist.last().msgId());
149 // depending on the order we have to traverse from the front to the back or vice versa
151 QList<Message> grouplist;
155 bool fastForward = false;
156 QList<Message>::const_iterator iter;
158 iter = msglist.constEnd();
159 iter--; // this op is safe as we've allready passed an empty check
161 iter = msglist.constBegin();
164 int idx = indexForId((*iter).msgId());
165 if(idx >= 0 && !_messageList.isEmpty())
166 dupeId = _messageList[idx]->msgId();
168 // we always compare to the previous entry...
169 // if there isn't, we can fastforward to the top
171 id = _messageList[idx - 1]->msgId();
175 if((*iter).msgId() != dupeId)
184 while(iter != msglist.constBegin()) {
187 if(!fastForward && (*iter).msgId() < id)
190 if((*iter).msgId() != dupeId) {
191 if(!grouplist.isEmpty()) {
192 QDateTime nextTs = grouplist.value(0).timestamp();
193 QDateTime prevTs = (*iter).timestamp();
194 nextTs.setTimeSpec(Qt::UTC);
195 prevTs.setTimeSpec(Qt::UTC);
196 uint nextDay = nextTs.toTime_t() / 86400;
197 uint prevDay = prevTs.toTime_t() / 86400;
198 if(nextDay != prevDay) {
199 nextTs.setTime_t(nextDay * 86400);
200 nextTs.setTimeSpec(Qt::LocalTime);
201 Message dayChangeMsg = Message::ChangeOfDay(nextTs);
202 dayChangeMsg.setMsgId((*iter).msgId());
203 grouplist.prepend(dayChangeMsg);
207 grouplist.prepend(*iter);
213 while(iter != msglist.constEnd()) {
214 if(!fastForward && (*iter).msgId() < id)
217 if((*iter).msgId() != dupeId) {
218 if(!grouplist.isEmpty()) {
219 QDateTime nextTs = grouplist.value(0).timestamp();
220 QDateTime prevTs = (*iter).timestamp();
221 nextTs.setTimeSpec(Qt::UTC);
222 prevTs.setTimeSpec(Qt::UTC);
223 uint nextDay = nextTs.toTime_t() / 86400;
224 uint prevDay = prevTs.toTime_t() / 86400;
225 if(nextDay != prevDay) {
226 nextTs.setTime_t(nextDay * 86400);
227 nextTs.setTimeSpec(Qt::LocalTime);
228 Message dayChangeMsg = Message::ChangeOfDay(nextTs);
229 dayChangeMsg.setMsgId((*iter).msgId());
230 grouplist.prepend(dayChangeMsg);
234 grouplist.prepend(*iter);
243 insertMessageGroup(grouplist);
244 return grouplist.count() + dupeCount;
247 void MessageModel::customEvent(QEvent *event) {
248 if(event->type() != QEvent::User)
253 if(_messageBuffer.isEmpty())
256 int processedMsgs = insertMessagesGracefully(_messageBuffer);
257 int remainingMsgs = _messageBuffer.count() - processedMsgs;
259 QList<Message>::iterator removeStart = _messageBuffer.begin() + remainingMsgs;
260 QList<Message>::iterator removeEnd = _messageBuffer.end();
261 _messageBuffer.erase(removeStart, removeEnd);
263 if(!_messageBuffer.isEmpty())
264 QCoreApplication::postEvent(this, new ProcessBufferEvent());
267 void MessageModel::clear() {
268 beginRemoveRows(QModelIndex(), 0, rowCount() - 1);
269 qDeleteAll(_messageList);
270 _messageList.clear();
276 // returns index of msg with given Id or of the next message after that (i.e., the index where we'd insert this msg)
277 int MessageModel::indexForId(MsgId id) {
278 if(_messageList.isEmpty() || id <= _messageList.value(0)->msgId())
280 if(id > _messageList.last()->msgId())
281 return _messageList.count();
284 int start = 0; int end = _messageList.count()-1;
288 int pivot = (end + start) / 2;
289 if(id <= _messageList.value(pivot)->msgId()) end = pivot;
294 void MessageModel::changeOfDay() {
295 _dayChangeTimer.setInterval(86400000);
296 qDebug() << _nextDayChange;
297 if(!_messageList.isEmpty()) {
298 int idx = _messageList.count();
299 while(idx > 0 && _messageList[idx - 1]->timeStamp() > _nextDayChange) {
302 beginInsertRows(QModelIndex(), idx, idx);
303 Message dayChangeMsg = Message::ChangeOfDay(_nextDayChange);
304 dayChangeMsg.setMsgId(_messageList[idx - 1]->msgId());
305 _messageList.insert(idx, createMessageModelItem(dayChangeMsg));
308 _nextDayChange = _nextDayChange.addSecs(86400);
311 /**********************************************************************************/
313 MessageModelItem::MessageModelItem(const Message &msg) :
314 _timestamp(msg.timestamp()),
316 _bufferId(msg.bufferInfo().bufferId()),
322 QVariant MessageModelItem::data(int column, int role) const {
323 if(column < MessageModel::TimestampColumn || column > MessageModel::ContentsColumn)
327 case MessageModel::MsgIdRole: return QVariant::fromValue<MsgId>(_msgId);
328 case MessageModel::BufferIdRole: return QVariant::fromValue<BufferId>(_bufferId);
329 case MessageModel::TypeRole: return _type;
330 case MessageModel::FlagsRole: return (int)_flags;
331 case MessageModel::TimestampRole: return _timestamp;
332 default: return QVariant();
338 bool MessageModelItem::lessThan(const MessageModelItem *m1, const MessageModelItem *m2){
339 return (*m1) < (*m2);
342 bool MessageModelItem::operator<(const MessageModelItem &other) const {
343 return _msgId < other._msgId;
346 bool MessageModelItem::operator==(const MessageModelItem &other) const {
347 return _msgId == other._msgId;
350 bool MessageModelItem::operator>(const MessageModelItem &other) const {
351 return _msgId > other._msgId;