Refactor and simplify ChatItem, implement full selection, use qreal where appropriate
[quassel.git] / src / qtui / chatitem.cpp
index 6bf77a4..5d1bcb7 100644 (file)
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include <QApplication>
+#include <QClipboard>
 #include <QFontMetrics>
 #include <QGraphicsSceneMouseEvent>
-#include <QTextCursor>
-#include <QTextDocument>
-
-#include <QtGui>
+#include <QPainter>
+#include <QPalette>
+#include <QTextLayout>
 
 #include "chatitem.h"
 #include "chatlinemodel.h"
 #include "qtui.h"
 
 ChatItem::ChatItem(const QPersistentModelIndex &index_, QGraphicsItem *parent) : QGraphicsItem(parent), _index(index_) {
-  QFontMetricsF *metrics = QtUi::style()->fontMetrics(data(ChatLineModel::FormatRole).value<UiStyle::FormatList>().at(0).second);
-  _lineHeight = metrics->lineSpacing();
+  _fontMetrics = QtUi::style()->fontMetrics(data(ChatLineModel::FormatRole).value<UiStyle::FormatList>().at(0).second);
+  _layout = 0;
+  _lines = 0;
+  _selectionStart = -1;
+  _selectionMode = NoSelection;
 }
 
 ChatItem::~ChatItem() {
-
+  delete _layout;
 }
 
 QVariant ChatItem::data(int role) const {
@@ -46,125 +50,216 @@ QVariant ChatItem::data(int role) const {
   return _index.data(role);
 }
 
-/*
-QRectF ChatItem::boundingRect() const {
-  return QRectF(0, 0, _width, _height);
-}
-*/
-
-void ChatItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget) {
-  Q_UNUSED(option); Q_UNUSED(widget);
-
-  painter->drawText(boundingRect(), data(MessageModel::DisplayRole).toString());
-  painter->setPen(Qt::DotLine);
-  painter->drawRect(boundingRect());
-}
-
-int ChatItem::setWidth(int w) {
+qreal ChatItem::setWidth(qreal w) {
   if(w == _boundingRect.width()) return _boundingRect.height();
-  int h = heightForWidth(w);
+  prepareGeometryChange();
   _boundingRect.setWidth(w);
+  qreal h = computeHeight();
   _boundingRect.setHeight(h);
+  if(haveLayout()) updateLayout();
   return h;
 }
 
-int ChatItem::heightForWidth(int width) {
+qreal ChatItem::computeHeight() {
   if(data(ChatLineModel::ColumnTypeRole).toUInt() != ChatLineModel::ContentsColumn)
-    return _lineHeight; // only contents can be multi-line
-
-  QVariantList wrapList = data(ChatLineModel::WrapListRole).toList();
-  int lines = 1;
-  int offset = 0;
-  for(int i = 0; i < wrapList.count(); i+=2) {
-    if(wrapList.at(i+1).toUInt() - offset < width) continue;
-    lines++;
-    if(i > 0) {
-      if(offset < wrapList.at(i-1).toUInt()) offset = wrapList.at(i-1).toUInt();
-      else offset += width;
-    } else {
-      offset += width;
-    }
-    i-=2;
-  }
-  return lines * _lineHeight;
+    return fontMetrics()->lineSpacing(); // only contents can be multi-line
+
+  _lines = 1;
+  WrapColumnFinder finder(this);
+  while(finder.nextWrapColumn() > 0) _lines++;
+  return _lines * fontMetrics()->lineSpacing();
 }
 
-/*
+QTextLayout *ChatItem::createLayout(QTextOption::WrapMode wrapMode, Qt::Alignment alignment) {
+  QTextLayout *layout = new QTextLayout(data(MessageModel::DisplayRole).toString());
+
+  QTextOption option;
+  option.setWrapMode(wrapMode);
+  option.setAlignment(alignment);
+  layout->setTextOption(option);
 
-void ChatItem::setTextOption(const QTextOption &option) {
-  _textOption = option;
-  layout();
+  QList<QTextLayout::FormatRange> formatRanges
+         = QtUi::style()->toTextLayoutList(data(MessageModel::FormatRole).value<UiStyle::FormatList>(), layout->text().length());
+  layout->setAdditionalFormats(formatRanges);
+  return layout;
 }
 
-QTextOption ChatItem::textOption() const {
-  return _textOption;
+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,0));
+        }
+        _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
+
+      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();
+    }
+    break;
+  }
 }
 
-QString ChatItem::text() const {
-  return _layout.text();
+void ChatItem::clearLayout() {
+  delete _layout;
+  _layout = 0;
 }
 
-void ChatItem::setText(const UiStyle::StyledText &text) {
-  _layout.setText(text.text);
-  _layout.setAdditionalFormats(text.formatList);
-  layout();
+// 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();
+  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();
+  } // TODO: add selection format here
+  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());
 }
 
-void ChatItem::layout() {
-  if(!_layout.additionalFormats().count()) return; // no text set
-  if(_width <= 0) return;
-  prepareGeometryChange();
-  QFontMetrics metrics(_layout.additionalFormats()[0].format.font());
-  int leading = metrics.leading();
-  int height = 0;
-  _layout.setTextOption(textOption());
-  _layout.beginLayout();
-  while(1) {
-    QTextLine line = _layout.createLine();
-    if(!line.isValid()) break;
-    line.setLineWidth(_width);
-    if(textOption().wrapMode() != QTextOption::NoWrap && line.naturalTextWidth() > _width) {
-      // word did not fit, we need to wrap it in the middle
-      // this is a workaround for Qt failing to handle WrapAtWordBoundaryOrAnywhere correctly
-      QTextOption::WrapMode mode = textOption().wrapMode();
-      textOption().setWrapMode(QTextOption::WrapAnywhere);
-      _layout.setTextOption(textOption());
-      line.setLineWidth(_width);
-      textOption().setWrapMode(mode);
-      _layout.setTextOption(textOption());
+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);
+    if(pos.y() >= line.y()) {
+      return line.xToCursor(pos.x(), QTextLine::CursorOnCharacter);
     }
-    height += leading;
-    line.setPosition(QPoint(0, height));
-    height += line.height();
   }
-  _layout.endLayout();
-  update();
-}    QDateTime _timestamp;
-    MsgId _msgId;
+  return 0;
+}
 
+void ChatItem::setFullSelection() {
+  _selectionMode = FullSelection;
+  update();
+}
 
-QRectF ChatItem::boundingRect() const {
-  return _layout.boundingRect();
+void ChatItem::clearSelection() {
+  _selectionMode = NoSelection;
+  update();
 }
 
-void ChatItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget) {
-  Q_UNUSED(option); Q_UNUSED(widget);
-  _layout.draw(painter, QPointF(0, 0));
+void ChatItem::mousePressEvent(QGraphicsSceneMouseEvent *event) {
+  if(event->buttons() & Qt::LeftButton) {
+    chatScene()->setSelectingItem(this);  // removes earlier selection if exists
+    _selectionStart = _selectionEnd = posToCursor(event->pos());
+    _selectionMode = PartialSelection;
+    event->accept();
+  } else {
+    event->ignore();
+  }
+}
 
+void ChatItem::mouseMoveEvent(QGraphicsSceneMouseEvent *event) {
+  if(contains(event->pos())) {
+    _selectionEnd = posToCursor(event->pos());
+    update();
+  } else {
+    setFullSelection();
+    ungrabMouse();
+    chatScene()->startGlobalSelection(this);
+  }
 }
-*/
 
-/*
-void ChatItem::mouseMoveEvent ( QGraphicsSceneMouseEvent * event ) {
-  qDebug() << (void*)this << "moving" << event->pos();
-  if(event->pos().y() < 0) {
-    QTextCursor cursor(document());
-    //cursor.insertText("foo");
-    //cursor.select(QTextCursor::Document);
+void ChatItem::mouseReleaseEvent(QGraphicsSceneMouseEvent *event) {
+  if(_selectionMode != NoSelection) {
+    _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
+    event->accept();
+  } else {
     event->ignore();
-  } else QGraphicsTextItem::mouseMoveEvent(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;
+}
 
+ChatItem::WrapColumnFinder::~WrapColumnFinder() {
+  delete layout;
+}
 
+qint16 ChatItem::WrapColumnFinder::nextWrapColumn() {
+  while(wordidx < wrapList.count()) {
+    w += wrapList.at(wordidx).width;
+    if(w >= item->width()) {
+      if(lastwrapcol >= wrapList.at(wordidx).start) {
+        // first word, and it doesn't fit
+        if(!line.isValid()) {
+          layout = item->createLayout(QTextOption::NoWrap);
+          layout->beginLayout();
+          line = layout->createLine();
+          line.setLineWidth(item->width());
+          layout->endLayout();
+        }
+        int idx = line.xToCursor(lastwrappos + item->width(), QTextLine::CursorOnCharacter);
+        qreal x = line.cursorToX(idx, QTextLine::Trailing);
+        w = w - wrapList.at(wordidx).width - (x - lastwrappos);
+        lastwrappos = x;
+        lastwrapcol = idx;
+        return idx;
+      }
+      // not the first word, so just wrap before this
+      lastwrapcol = wrapList.at(wordidx).start;
+      lastwrappos = lastwrappos + w - wrapList.at(wordidx).width;
+      w = 0;
+      return lastwrapcol;
+    }
+    w += wrapList.at(wordidx).trailing;
+    wordidx++;
+  }
+  return -1;
+}