ChatScene speed improvement. This might even fix the dreaded CPU bug!
[quassel.git] / src / qtui / chatitem.cpp
index 7d52c5a..7dd7aba 100644 (file)
@@ -20,6 +20,7 @@
 
 #include <QApplication>
 #include <QClipboard>
+#include <QDesktopServices>
 #include <QFontMetrics>
 #include <QGraphicsSceneMouseEvent>
 #include <QPainter>
 #include "chatlinemodel.h"
 #include "qtui.h"
 
-ChatItem::ChatItem(const QPersistentModelIndex &index_, QGraphicsItem *parent) : QGraphicsItem(parent), _index(index_) {
-  _fontMetrics = QtUi::style()->fontMetrics(data(ChatLineModel::FormatRole).value<UiStyle::FormatList>().at(0).second);
-  _layout = 0;
-  _lines = 0;
-  _selectionStart = -1;
+ChatItem::ChatItem(ChatLineModel::ColumnType col, QAbstractItemModel *model, QGraphicsItem *parent)
+  : QGraphicsItem(parent),
+    _fontMetrics(0),
+    _selectionMode(NoSelection),
+    _selectionStart(-1),
+    _layout(0)
+{
+  Q_ASSERT(model);
+  QModelIndex index = model->index(row(), col);
+  _fontMetrics = QtUi::style()->fontMetrics(model->data(index, ChatLineModel::FormatRole).value<UiStyle::FormatList>().at(0).second);
+  setAcceptHoverEvents(true);
+  setZValue(20);
 }
 
 ChatItem::~ChatItem() {
@@ -42,31 +50,27 @@ ChatItem::~ChatItem() {
 }
 
 QVariant ChatItem::data(int role) const {
-  if(!_index.isValid()) {
-    qWarning() << "ChatItem::data(): Model index is invalid!" << _index;
+  QModelIndex index = model()->index(row(), column());
+  if(!index.isValid()) {
+    qWarning() << "ChatItem::data(): model index is invalid!" << index;
     return QVariant();
   }
-  return _index.data(role);
+  return model()->data(index, role);
 }
 
-int ChatItem::setWidth(int w) {
+qreal ChatItem::setGeometry(qreal w, qreal h) {
   if(w == _boundingRect.width()) return _boundingRect.height();
   prepareGeometryChange();
   _boundingRect.setWidth(w);
-  int h = heightForWidth(w);
+  if(h < 0) h = computeHeight();
+  //if(h < 0) h = fontMetrics()->lineSpacing(); // only contents can be multi-line
   _boundingRect.setHeight(h);
   if(haveLayout()) updateLayout();
   return h;
 }
 
-int ChatItem::heightForWidth(int width) {
-  if(data(ChatLineModel::ColumnTypeRole).toUInt() != ChatLineModel::ContentsColumn)
-    return fontMetrics()->lineSpacing(); // only contents can be multi-line
-
-  _lines = 1;
-  WrapColumnFinder finder(this);
-  while(finder.nextWrapColumn() > 0) _lines++;
-  return _lines * fontMetrics()->lineSpacing();
+qreal ChatItem::computeHeight() {
+  return fontMetrics()->lineSpacing(); // only contents can be multi-line
 }
 
 QTextLayout *ChatItem::createLayout(QTextOption::WrapMode wrapMode, Qt::Alignment alignment) {
@@ -84,50 +88,16 @@ QTextLayout *ChatItem::createLayout(QTextOption::WrapMode wrapMode, Qt::Alignmen
 }
 
 void ChatItem::updateLayout() {
-  switch(data(ChatLineModel::ColumnTypeRole).toUInt()) {
-    case ChatLineModel::TimestampColumn:
-      if(!haveLayout()) _layout = createLayout(QTextOption::WrapAnywhere, Qt::AlignLeft);
-      // fallthrough
-    case ChatLineModel::SenderColumn:
-      if(!haveLayout()) _layout = createLayout(QTextOption::WrapAnywhere, Qt::AlignRight);
-      _layout->beginLayout();
-      {
-        QTextLine line = _layout->createLine();
-        if(line.isValid()) {
-          line.setLineWidth(width());
-          line.setPosition(QPointF(0, fontMetrics()->leading()));
-        }
-        _layout->endLayout();
-      }
-      break;
-    case ChatLineModel::ContentsColumn: {
-      if(!haveLayout()) _layout = createLayout(QTextOption::WrapAnywhere);
-
-      // Now layout
-      ChatLineModel::WrapList wrapList = data(ChatLineModel::WrapListRole).value<ChatLineModel::WrapList>();
-      if(!wrapList.count()) return; // empty chatitem
-      int wordidx = 0;
-      ChatLineModel::Word word = wrapList.at(0);
-
-      qreal h = 0;
-      WrapColumnFinder finder(this);
-      _layout->beginLayout();
-      forever {
-        QTextLine line = _layout->createLine();
-        if (!line.isValid())
-          break;
-
-        int col = finder.nextWrapColumn();
-        line.setNumColumns(col >= 0 ? col - line.textStart() : _layout->text().length());
-
-        h += fontMetrics()->leading();
-        line.setPosition(QPointF(0, h));
-        h += line.height();
-      }
-      _layout->endLayout();
-    }
-    break;
+  if(!haveLayout())
+    setLayout(createLayout(QTextOption::WrapAnywhere, Qt::AlignLeft));
+
+  layout()->beginLayout();
+  QTextLine line = layout()->createLine();
+  if(line.isValid()) {
+    line.setLineWidth(width());
+    line.setPosition(QPointF(0,0));
   }
+  layout()->endLayout();
 }
 
 void ChatItem::clearLayout() {
@@ -135,18 +105,40 @@ void ChatItem::clearLayout() {
   _layout = 0;
 }
 
+// NOTE: This is not the most time-efficient implementation, but it saves space by not caching unnecessary data
+//       This is a deliberate trade-off. (-> selectFmt creation, data() call)
 void ChatItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget) {
   Q_UNUSED(option); Q_UNUSED(widget);
   if(!haveLayout()) updateLayout();
-  _layout->draw(painter, QPointF(0,0), QVector<QTextLayout::FormatRange>(), boundingRect());
+  painter->setClipRect(boundingRect()); // no idea why QGraphicsItem clipping won't work
+  //if(_selectionMode == FullSelection) {
+    //painter->save();
+    //painter->fillRect(boundingRect(), QApplication::palette().brush(QPalette::Highlight));
+    //painter->restore();
+  //}
+  QVector<QTextLayout::FormatRange> formats;
+  if(_selectionMode != NoSelection) {
+    QTextLayout::FormatRange selectFmt;
+    selectFmt.format.setForeground(QApplication::palette().brush(QPalette::HighlightedText));
+    selectFmt.format.setBackground(QApplication::palette().brush(QPalette::Highlight));
+    if(_selectionMode == PartialSelection) {
+      selectFmt.start = qMin(_selectionStart, _selectionEnd);
+      selectFmt.length = qAbs(_selectionStart - _selectionEnd);
+    } else { // FullSelection
+      selectFmt.start = 0;
+      selectFmt.length = data(MessageModel::DisplayRole).toString().length();
+    }
+    formats.append(selectFmt);
+  }
+  layout()->draw(painter, QPointF(0,0), formats, boundingRect());
 }
 
-int ChatItem::posToCursor(const QPointF &pos) {
+qint16 ChatItem::posToCursor(const QPointF &pos) {
   if(pos.y() > height()) return data(MessageModel::DisplayRole).toString().length();
   if(pos.y() < 0) return 0;
   if(!haveLayout()) updateLayout();
-  for(int l = _layout->lineCount() - 1; l >= 0; l--) {
-    QTextLine line = _layout->lineAt(l);
+  for(int l = layout()->lineCount() - 1; l >= 0; l--) {
+    QTextLine line = layout()->lineAt(l);
     if(pos.y() >= line.y()) {
       return line.xToCursor(pos.x(), QTextLine::CursorOnCharacter);
     }
@@ -154,41 +146,57 @@ int ChatItem::posToCursor(const QPointF &pos) {
   return 0;
 }
 
-void ChatItem::clearSelection() {
-  if(_selectionStart >= 0) {
-    QList<QTextLayout::FormatRange> formats = _layout->additionalFormats();
-    formats.removeLast();
-    _layout->setAdditionalFormats(formats);
-    _selectionStart = -1;
-    updateLayout();
+void ChatItem::setFullSelection() {
+  if(_selectionMode != FullSelection) {
+    _selectionMode = FullSelection;
     update();
   }
 }
 
-void ChatItem::mouseMoveEvent(QGraphicsSceneMouseEvent *event) {
-  int selectionEnd = posToCursor(event->pos());
-  QList<QTextLayout::FormatRange> formats = _layout->additionalFormats();
-  formats.last().start = qMin(_selectionStart, selectionEnd);
-  formats.last().length = qMax(_selectionStart, selectionEnd) - formats.last().start;
-  _layout->setAdditionalFormats(formats);
-  updateLayout();
+void ChatItem::clearSelection() {
+  _selectionMode = NoSelection;
+  update();
+}
+
+void ChatItem::continueSelecting(const QPointF &pos) {
+  _selectionMode = PartialSelection;
+  _selectionEnd = posToCursor(pos);
   update();
 }
 
+QList<QRectF> ChatItem::findWords(const QString &searchWord, Qt::CaseSensitivity caseSensitive) {
+  QList<QRectF> resultList;
+  const QAbstractItemModel *model_ = model();
+  if(!model_)
+    return resultList;
+
+  QString plainText = model_->data(model_->index(row(), column()), MessageModel::DisplayRole).toString();
+  QList<int> indexList;
+  int searchIdx = plainText.indexOf(searchWord, 0, caseSensitive);
+  while(searchIdx != -1) {
+    indexList << searchIdx;
+    searchIdx = plainText.indexOf(searchWord, searchIdx + 1, caseSensitive);
+  }
+
+  if(!haveLayout())
+    updateLayout();
+
+  foreach(int idx, indexList) {
+    QTextLine line = layout()->lineForTextPosition(idx);
+    qreal x = line.cursorToX(idx);
+    qreal width = line.cursorToX(idx + searchWord.count()) - x;
+    qreal height = fontMetrics()->lineSpacing();
+    qreal y = height * line.lineNumber();
+    resultList << QRectF(x, y, width, height);
+  }
+  return resultList;
+}
+
 void ChatItem::mousePressEvent(QGraphicsSceneMouseEvent *event) {
-  if(event->buttons() & Qt::LeftButton) {
-    if(!haveLayout()) updateLayout();
+  if(event->buttons() == Qt::LeftButton) {
     chatScene()->setSelectingItem(this);
-    _selectionStart = posToCursor(event->pos());
-    QList<QTextLayout::FormatRange> formats = _layout->additionalFormats();
-    QTextLayout::FormatRange selectFmt;
-    QPalette pal = QApplication::palette();
-    selectFmt.format.setForeground(pal.brush(QPalette::HighlightedText));
-    selectFmt.format.setBackground(pal.brush(QPalette::Highlight));
-    selectFmt.length = 0;
-    formats.append(selectFmt);
-    _layout->setAdditionalFormats(formats);
-    updateLayout();
+    _selectionStart = _selectionEnd = posToCursor(event->pos());
+    _selectionMode = NoSelection; // will be set to PartialSelection by mouseMoveEvent
     update();
     event->accept();
   } else {
@@ -196,12 +204,31 @@ void ChatItem::mousePressEvent(QGraphicsSceneMouseEvent *event) {
   }
 }
 
+void ChatItem::mouseMoveEvent(QGraphicsSceneMouseEvent *event) {
+  if(event->buttons() == Qt::LeftButton) {
+    if(contains(event->pos())) {
+      qint16 end = posToCursor(event->pos());
+      if(end != _selectionEnd) {
+        _selectionEnd = end;
+        _selectionMode = (_selectionStart != _selectionEnd ? PartialSelection : NoSelection);
+        update();
+      }
+    } else {
+      setFullSelection();
+      chatScene()->startGlobalSelection(this, event->pos());
+    }
+    event->accept();
+  } else {
+    event->ignore();
+  }
+}
+
 void ChatItem::mouseReleaseEvent(QGraphicsSceneMouseEvent *event) {
-  if(_selectionStart >= 0) {
-    int selectionEnd = posToCursor(event->pos());
+  if(_selectionMode != NoSelection && !event->buttons() & Qt::LeftButton) {
+    _selectionEnd = posToCursor(event->pos());
     QString selection
-        = data(MessageModel::DisplayRole).toString().mid(qMin(_selectionStart, selectionEnd), qAbs(_selectionStart - selectionEnd));
-    QApplication::clipboard()->setText(selection, QClipboard::Clipboard);  // TODO configure where selections should go
+        = data(MessageModel::DisplayRole).toString().mid(qMin(_selectionStart, _selectionEnd), qAbs(_selectionStart - _selectionEnd));
+    chatScene()->putToClipboard(selection);
     event->accept();
   } else {
     event->ignore();
@@ -210,20 +237,118 @@ void ChatItem::mouseReleaseEvent(QGraphicsSceneMouseEvent *event) {
 
 /*************************************************************************************************/
 
-ChatItem::WrapColumnFinder::WrapColumnFinder(ChatItem *_item) : item(_item) {
-  wrapList = item->data(ChatLineModel::WrapListRole).value<ChatLineModel::WrapList>();
-  wordidx = 0;
-  layout = 0;
-  lastwrapcol = 0;
-  lastwrappos = 0;
-  w = 0;
+/*************************************************************************************************/
+
+void SenderChatItem::updateLayout() {
+  if(!haveLayout()) setLayout(createLayout(QTextOption::WrapAnywhere, Qt::AlignRight));
+  ChatItem::updateLayout();
+}
+
+/*************************************************************************************************/
+
+ContentsChatItem::ContentsChatItem(QAbstractItemModel *model, QGraphicsItem *parent) : ChatItem(column(), model, parent),
+  _layoutData(0)
+{
+
+}
+
+ContentsChatItem::~ContentsChatItem() {
+  delete _layoutData;
+}
+
+qreal ContentsChatItem::computeHeight() {
+  int lines = 1;
+  WrapColumnFinder finder(this);
+  while(finder.nextWrapColumn() > 0) lines++;
+  return lines * fontMetrics()->lineSpacing();
+}
+
+void ContentsChatItem::setLayout(QTextLayout *layout) {
+  if(!_layoutData)
+    _layoutData = new LayoutData;
+  _layoutData->layout = layout;
+}
+
+void ContentsChatItem::clearLayout() {
+  delete _layoutData;
+  _layoutData = 0;
+}
+
+void ContentsChatItem::updateLayout() {
+  if(!haveLayout()) setLayout(createLayout(QTextOption::WrapAnywhere));
+
+  // Now layout
+  ChatLineModel::WrapList wrapList = data(ChatLineModel::WrapListRole).value<ChatLineModel::WrapList>();
+  if(!wrapList.count()) return; // empty chatitem
+
+  qreal h = 0;
+  WrapColumnFinder finder(this);
+  layout()->beginLayout();
+  forever {
+    QTextLine line = layout()->createLine();
+    if(!line.isValid())
+      break;
+
+    int col = finder.nextWrapColumn();
+    line.setNumColumns(col >= 0 ? col - line.textStart() : layout()->text().length());
+    line.setPosition(QPointF(0, h));
+    h += line.height() + fontMetrics()->leading();
+  }
+  layout()->endLayout();
+}
+
+void ContentsChatItem::mouseDoubleClickEvent(QGraphicsSceneMouseEvent *event) {
+  // FIXME dirty and fast hack to make http:// urls klickable
+
+  QRegExp regex("\\b([hf]t{1,2}ps?://[^\\s]+)\\b");
+  QString str = data(ChatLineModel::DisplayRole).toString();
+  int idx = posToCursor(event->pos());
+  int mi = 0;
+  do {
+    mi = regex.indexIn(str, mi);
+    if(mi < 0) break;
+    if(idx >= mi && idx < mi + regex.matchedLength()) {
+      QDesktopServices::openUrl(QUrl(regex.capturedTexts()[1]));
+      break;
+    }
+    mi += regex.matchedLength();
+  } while(mi >= 0);
+  event->accept();
+}
+
+void ContentsChatItem::hoverEnterEvent(QGraphicsSceneHoverEvent *event) {
+  //qDebug() << (void*)this << "entering";
+  event->ignore();
+}
+
+void ContentsChatItem::hoverLeaveEvent(QGraphicsSceneHoverEvent *event) {
+  //qDebug() << (void*)this << "leaving";
+  event->ignore();
+}
+
+void ContentsChatItem::hoverMoveEvent(QGraphicsSceneHoverEvent *event) {
+  //qDebug() << (void*)this << event->pos();
+  event->ignore();
+}
+
+/*************************************************************************************************/
+
+ContentsChatItem::WrapColumnFinder::WrapColumnFinder(ChatItem *_item)
+  : item(_item),
+    layout(0),
+    wrapList(item->data(ChatLineModel::WrapListRole).value<ChatLineModel::WrapList>()),
+    wordidx(0),
+    lastwrapcol(0),
+    lastwrappos(0),
+    w(0)
+{
 }
 
-ChatItem::WrapColumnFinder::~WrapColumnFinder() {
+ContentsChatItem::WrapColumnFinder::~WrapColumnFinder() {
   delete layout;
 }
 
-int ChatItem::WrapColumnFinder::nextWrapColumn() {
+qint16 ContentsChatItem::WrapColumnFinder::nextWrapColumn() {
   while(wordidx < wrapList.count()) {
     w += wrapList.at(wordidx).width;
     if(w >= item->width()) {