#include "messagemodel.h"
-MessageModel::MessageModel(QObject *parent) : QAbstractItemModel(parent) {
-
-
-
-}
+#include "message.h"
+
+#include <QEvent>
+
+class ProcessBufferEvent : public QEvent {
+public:
+ inline ProcessBufferEvent() : QEvent(QEvent::User) {}
+};
-MessageModel::~MessageModel() {
-
-
+MessageModel::MessageModel(QObject *parent)
+ : QAbstractItemModel(parent)
+{
+ QDateTime now = QDateTime::currentDateTime();
+ now.setTimeSpec(Qt::UTC);
+ _nextDayChange.setTimeSpec(Qt::UTC);
+ _nextDayChange.setTime_t(((now.toTime_t() / 86400) + 1) * 86400);
+ _nextDayChange.setTimeSpec(Qt::LocalTime);
+ _dayChangeTimer.setInterval(QDateTime::currentDateTime().secsTo(_nextDayChange) * 1000);
+ _dayChangeTimer.start();
+ connect(&_dayChangeTimer, SIGNAL(timeout()), this, SLOT(changeOfDay()));
}
QVariant MessageModel::data(const QModelIndex &index, int role) const {
- int row = index.row();
- if(row < 0 || row >= _messageList.count()) return QVariant();
+ int row = index.row(); int column = index.column();
+ if(row < 0 || row >= _messageList.count() || column < 0)
+ return QVariant();
+
+ if(role == ColumnTypeRole)
+ return column;
+
return _messageList[row]->data(index.column(), role);
}
bool MessageModel::setData(const QModelIndex &index, const QVariant &value, int role) {
int row = index.row();
- if(row < 0 || row >= _messageList.count()) return false;
- if(_messageList[row]->setData(index.column(), role)) {
- emit dataChanged(index, index); // FIXME make msg emit this (too)
+ if(row < 0 || row >= _messageList.count())
+ return false;
+
+ if(_messageList[row]->setData(index.column(), value, role)) {
+ emit dataChanged(index, index);
return true;
}
+
return false;
}
-void MessageModel::insertMessage(const Message &msg) {
+
+
+bool MessageModel::insertMessage(const Message &msg, bool fakeMsg) {
MsgId id = msg.msgId();
- MessageItem *item = createMessageItem(msg);
- if(id > )
-
+ int idx = indexForId(id);
+ if(!fakeMsg && idx < _messageList.count()) { // check for duplicate
+ if(_messageList[idx]->msgId() == id)
+ return false;
+ }
+
+ MessageModelItem *item = createMessageModelItem(msg);
+ beginInsertRows(QModelIndex(), idx, idx);
+ _messageList.insert(idx, item);
+ endInsertRows();
+ return true;
+}
+
+void MessageModel::insertMessages(const QList<Message> &msglist) {
+ if(msglist.isEmpty())
+ return;
+
+ int processedMsgs = insertMessagesGracefully(msglist);
+ int remainingMsgs = msglist.count() - processedMsgs;
+ if(remainingMsgs > 0) {
+ if(msglist.first().msgId() < msglist.last().msgId()) {
+ // in Order
+ _messageBuffer << msglist.mid(0, remainingMsgs);
+ } else {
+ _messageBuffer << msglist.mid(processedMsgs);
+ }
+ qSort(_messageBuffer);
+ QCoreApplication::postEvent(this, new ProcessBufferEvent());
+ }
+}
+
+void MessageModel::insertMessageGroup(const QList<Message> &msglist) {
+ int idx = indexForId(msglist.first().msgId());
+ if(idx > 0) {
+ int prevIdx = idx - 1;
+ if(_messageList[prevIdx]->msgType() == Message::DayChange
+ && _messageList[prevIdx]->timeStamp() > msglist.value(0).timestamp()) {
+ beginRemoveRows(QModelIndex(), prevIdx, prevIdx);
+ MessageModelItem *oldItem = _messageList.takeAt(prevIdx);
+ delete oldItem;
+ endRemoveRows();
+ idx--;
+ }
+ }
+ int start = idx;
+ int end = idx + msglist.count() - 1;
+
+ Message dayChangeMsg;
+ bool needsDayChangeMsg = false;
+ if(idx < _messageList.count() && _messageList[idx]->msgType() != Message::DayChange) {
+ QDateTime nextTs = _messageList[idx]->timeStamp();
+ QDateTime prevTs = msglist.last().timestamp();
+ nextTs.setTimeSpec(Qt::UTC);
+ prevTs.setTimeSpec(Qt::UTC);
+ uint nextDay = nextTs.toTime_t() / 86400;
+ uint prevDay = prevTs.toTime_t() / 86400;
+ if(nextDay != prevDay) {
+ nextTs.setTime_t(nextDay * 86400);
+ nextTs.setTimeSpec(Qt::LocalTime);
+ dayChangeMsg = Message::ChangeOfDay(nextTs);
+ dayChangeMsg.setMsgId(msglist.last().msgId());
+ needsDayChangeMsg = true;
+ end++;
+ }
+ }
+ beginInsertRows(QModelIndex(), start, end);
+ foreach(Message msg, msglist) {
+ _messageList.insert(idx, createMessageModelItem(msg));
+ idx++;
+ }
+ if(needsDayChangeMsg) {
+ _messageList.insert(idx, createMessageModelItem(dayChangeMsg));
+ }
+ endInsertRows();
+}
+
+int MessageModel::insertMessagesGracefully(const QList<Message> &msglist) {
+ bool inOrder = (msglist.first().msgId() < msglist.last().msgId());
+ // depending on the order we have to traverse from the front to the back or vice versa
+
+ QList<Message> grouplist;
+ MsgId id;
+ MsgId dupeId;
+ int dupeCount = 0;
+ bool fastForward = false;
+ QList<Message>::const_iterator iter;
+ if(inOrder) {
+ iter = msglist.constEnd();
+ iter--; // this op is safe as we've allready passed an empty check
+ } else {
+ iter = msglist.constBegin();
+ }
+
+ int idx = indexForId((*iter).msgId());
+ if(idx >= 0 && !_messageList.isEmpty())
+ dupeId = _messageList[idx]->msgId();
+
+ // we always compare to the previous entry...
+ // if there isn't, we can fastforward to the top
+ if(idx - 1 >= 0)
+ id = _messageList[idx - 1]->msgId();
+ else
+ fastForward = true;
+
+ if((*iter).msgId() != dupeId)
+ grouplist << *iter;
+ else
+ dupeCount++;
+
+ if(!inOrder)
+ iter++;
+
+ if(inOrder) {
+ while(iter != msglist.constBegin()) {
+ iter--;
+
+ if(!fastForward && (*iter).msgId() < id)
+ break;
+
+ if((*iter).msgId() != dupeId) {
+ if(!grouplist.isEmpty()) {
+ QDateTime nextTs = grouplist.value(0).timestamp();
+ QDateTime prevTs = (*iter).timestamp();
+ nextTs.setTimeSpec(Qt::UTC);
+ prevTs.setTimeSpec(Qt::UTC);
+ uint nextDay = nextTs.toTime_t() / 86400;
+ uint prevDay = prevTs.toTime_t() / 86400;
+ if(nextDay != prevDay) {
+ nextTs.setTime_t(nextDay * 86400);
+ nextTs.setTimeSpec(Qt::LocalTime);
+ Message dayChangeMsg = Message::ChangeOfDay(nextTs);
+ dayChangeMsg.setMsgId((*iter).msgId());
+ grouplist.prepend(dayChangeMsg);
+ dupeCount--;
+ }
+ }
+ grouplist.prepend(*iter);
+ } else {
+ dupeCount++;
+ }
+ }
+ } else {
+ while(iter != msglist.constEnd()) {
+ if(!fastForward && (*iter).msgId() < id)
+ break;
+
+ if((*iter).msgId() != dupeId) {
+ if(!grouplist.isEmpty()) {
+ QDateTime nextTs = grouplist.value(0).timestamp();
+ QDateTime prevTs = (*iter).timestamp();
+ nextTs.setTimeSpec(Qt::UTC);
+ prevTs.setTimeSpec(Qt::UTC);
+ uint nextDay = nextTs.toTime_t() / 86400;
+ uint prevDay = prevTs.toTime_t() / 86400;
+ if(nextDay != prevDay) {
+ nextTs.setTime_t(nextDay * 86400);
+ nextTs.setTimeSpec(Qt::LocalTime);
+ Message dayChangeMsg = Message::ChangeOfDay(nextTs);
+ dayChangeMsg.setMsgId((*iter).msgId());
+ grouplist.prepend(dayChangeMsg);
+ dupeCount--;
+ }
+ }
+ grouplist.prepend(*iter);
+ } else {
+ dupeCount++;
+ }
+
+ iter++;
+ }
+ }
+
+ insertMessageGroup(grouplist);
+ return grouplist.count() + dupeCount;
}
+void MessageModel::customEvent(QEvent *event) {
+ if(event->type() != QEvent::User)
+ return;
+
+ event->accept();
+
+ if(_messageBuffer.isEmpty())
+ return;
+
+ int processedMsgs = insertMessagesGracefully(_messageBuffer);
+ int remainingMsgs = _messageBuffer.count() - processedMsgs;
+
+ QList<Message>::iterator removeStart = _messageBuffer.begin() + remainingMsgs;
+ QList<Message>::iterator removeEnd = _messageBuffer.end();
+ _messageBuffer.erase(removeStart, removeEnd);
+
+ if(!_messageBuffer.isEmpty())
+ QCoreApplication::postEvent(this, new ProcessBufferEvent());
+}
+
+void MessageModel::clear() {
+ beginRemoveRows(QModelIndex(), 0, rowCount() - 1);
+ qDeleteAll(_messageList);
+ _messageList.clear();
+ endRemoveRows();
+}
+
+
+
// returns index of msg with given Id or of the next message after that (i.e., the index where we'd insert this msg)
int MessageModel::indexForId(MsgId id) {
- if(!_messageList.count() || id <= _messageList[0]->data(0, MsgIdRole).value<MsgId>()) return 0;
- if(id > _messageList.last()->data(0, MsgIdRole).value<MsgId>()) return _messageList.count();
+ if(_messageList.isEmpty() || id <= _messageList.value(0)->msgId())
+ return 0;
+ if(id > _messageList.last()->msgId())
+ return _messageList.count();
+
// binary search
int start = 0; int end = _messageList.count()-1;
- int idx;
while(1) {
- if(start == end) return start;
- idx = (end + start) / 2;
-
+ if(end - start == 1)
+ return end;
+ int pivot = (end + start) / 2;
+ if(id <= _messageList.value(pivot)->msgId()) end = pivot;
+ else start = pivot;
+ }
+}
+
+void MessageModel::changeOfDay() {
+ _dayChangeTimer.setInterval(86400000);
+ qDebug() << _nextDayChange;
+ if(!_messageList.isEmpty()) {
+ int idx = _messageList.count();
+ while(idx > 0 && _messageList[idx - 1]->timeStamp() > _nextDayChange) {
+ idx--;
+ }
+ beginInsertRows(QModelIndex(), idx, idx);
+ Message dayChangeMsg = Message::ChangeOfDay(_nextDayChange);
+ dayChangeMsg.setMsgId(_messageList[idx - 1]->msgId());
+ _messageList.insert(idx, createMessageModelItem(dayChangeMsg));
+ endInsertRows();
+ }
+ _nextDayChange = _nextDayChange.addSecs(86400);
}
/**********************************************************************************/
+MessageModelItem::MessageModelItem(const Message &msg) :
+ _timestamp(msg.timestamp()),
+ _msgId(msg.msgId()),
+ _bufferId(msg.bufferInfo().bufferId()),
+ _type(msg.type()),
+ _flags(msg.flags())
+{
+}
+
+QVariant MessageModelItem::data(int column, int role) const {
+ if(column < MessageModel::TimestampColumn || column > MessageModel::ContentsColumn)
+ return QVariant();
+
+ switch(role) {
+ case MessageModel::MsgIdRole: return QVariant::fromValue<MsgId>(_msgId);
+ case MessageModel::BufferIdRole: return QVariant::fromValue<BufferId>(_bufferId);
+ case MessageModel::TypeRole: return _type;
+ case MessageModel::FlagsRole: return (int)_flags;
+ case MessageModel::TimestampRole: return _timestamp;
+ default: return QVariant();
+ }
+}
+
+
+// Stuff for later
+bool MessageModelItem::lessThan(const MessageModelItem *m1, const MessageModelItem *m2){
+ return (*m1) < (*m2);
+}
+
+bool MessageModelItem::operator<(const MessageModelItem &other) const {
+ return _msgId < other._msgId;
+}
+
+bool MessageModelItem::operator==(const MessageModelItem &other) const {
+ return _msgId == other._msgId;
+}
+
+bool MessageModelItem::operator>(const MessageModelItem &other) const {
+ return _msgId > other._msgId;
+}
projects / quassel.git / blobdiff