ChatScene now properly react on aboutToRemoveRows(), which should improve stability...

[quassel.git] / src / client / messagemodel.cpp

diff --git a/src/client/messagemodel.cpp b/src/client/messagemodel.cpp
index ea31117..940dda2 100644 (file)
--- a/src/client/messagemodel.cpp
+++ b/src/client/messagemodel.cpp
@@ -17,3 +17,105 @@
  *   Free Software Foundation, Inc.,                                       *
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
+
+#include "messagemodel.h"
+
+#include "message.h"
+
+MessageModel::MessageModel(QObject *parent)
+  : QAbstractItemModel(parent)
+{
+}
+
+QVariant MessageModel::data(const QModelIndex &index, int role) const {
+  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(), value, role)) {
+    emit dataChanged(index, index);
+    return true;
+  }
+
+  return false;
+}
+
+bool MessageModel::insertMessage(const Message &msg, bool fakeMsg) {
+  MsgId id = msg.msgId();
+  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;
+  // FIXME make this more efficient by grouping msgs
+  foreach(Message msg, msglist)
+    insertMessage(msg);
+}
+
+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.isEmpty() || id <= _messageList.value(0)->data(0, MsgIdRole).value<MsgId>()) return 0;
+  if(id > _messageList.last()->data(0, MsgIdRole).value<MsgId>()) return _messageList.count();
+  // binary search
+  int start = 0; int end = _messageList.count()-1;
+  while(1) {
+    if(end - start == 1) return end;
+    int pivot = (end + start) / 2;
+    if(id <= _messageList.value(pivot)->data(0, MsgIdRole).value<MsgId>()) end = pivot;
+    else start = pivot;
+  }
+}
+
+/**********************************************************************************/
+
+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();
+  }
+}
+