Introduce WrapColumnFinder.
[quassel.git] / src / qtui / chatitem.cpp
index 1e08605..0920f77 100644 (file)
 
 #include <QFontMetrics>
 #include <QGraphicsSceneMouseEvent>
-#include <QTextCursor>
-#include <QTextDocument>
-
-#include <QtGui>
+#include <QPainter>
+#include <QTextLayout>
 
 #include "chatitem.h"
+#include "chatlinemodel.h"
+#include "qtui.h"
 
 ChatItem::ChatItem(const QPersistentModelIndex &index_, QGraphicsItem *parent) : QGraphicsItem(parent), _index(index_) {
-  //if(_wrapMode == WordWrap) {
-  //  setFlags(QGraphicsItem::ItemClipsToShape, true);
-  //}
+  _fontMetrics = QtUi::style()->fontMetrics(data(ChatLineModel::FormatRole).value<UiStyle::FormatList>().at(0).second);
+  _layout = 0;
+  _lines = 0;
 }
 
 ChatItem::~ChatItem() {
@@ -39,49 +39,142 @@ ChatItem::~ChatItem() {
 
 QVariant ChatItem::data(int role) const {
   if(!_index.isValid()) {
-    qWarning() << "ChatItem::data(): Model index is invalid!";
+    qWarning() << "ChatItem::data(): Model index is invalid!" << _index;
     return QVariant();
   }
   return _index.data(role);
 }
 
-QRectF ChatItem::boundingRect() const {
-  return QRectF(0, 0, 500,20);
+int ChatItem::setWidth(int w) {
+  w -= 10;
+  if(w == _boundingRect.width()) return _boundingRect.height();
+  _boundingRect.setWidth(w);
+  int h = heightForWidth(w);
+  _boundingRect.setHeight(h);
+  if(haveLayout()) updateLayout();
+  return h;
 }
 
-void ChatItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget) {
-  Q_UNUSED(option); Q_UNUSED(widget);
+int ChatItem::heightForWidth(int width) {
+  if(data(ChatLineModel::ColumnTypeRole).toUInt() != ChatLineModel::ContentsColumn)
+    return fontMetrics()->lineSpacing(); // only contents can be multi-line
 
-  painter->drawRect(boundingRect());
-  painter->drawText(boundingRect(), data(MessageModel::DisplayRole).toString());
+  _lines = 1;
+  QTextLine line;
+  WrapColumnFinder finder(this);
+  while(finder.nextWrapColumn(line) > 0) _lines++;
+  return _lines * fontMetrics()->lineSpacing();
+}
+
+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;
+}
 
-/*
-void ChatItem::setWidth(int w) {
-  _width = w;
-  layout();
+int ChatItem::WrapColumnFinder::nextWrapColumn(QTextLine &line) {
+  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();
+          layout->beginLayout();
+          line = layout->createLine();
+          line.setLineWidth(item->width());
+          layout->endLayout();
+        }
+        int idx = line.xToCursor((line.textStart() ? 0 : lastwrappos) + item->width());
+        qreal x = line.cursorToX(idx);
+        idx += line.textStart();
+        w = w - x - wrapList.at(wordidx).width;
+        if(line.textStart()) lastwrappos += x;
+        else 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;
 }
 
-void ChatItem::setTextOption(const QTextOption &option) {
-  _textOption = option;
-  layout();
+QTextLayout *ChatItem::createLayout() {
+  QTextLayout *layout = new QTextLayout(data(MessageModel::DisplayRole).toString());
+
+  QTextOption option;
+  option.setWrapMode(QTextOption::WrapAnywhere);
+  layout->setTextOption(option);
+
+  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() {
+  if(!haveLayout()) _layout = createLayout();
+
+  // 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);
+    line.setNumColumns(col >= 0 ? col - line.textStart() : _layout->text().length());
+
+    h += fontMetrics()->leading();
+    line.setPosition(QPointF(0, h));
+    h += line.height();
+  }
+  _layout->endLayout();
 }
 
-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.formats);
-  layout();
+//int ChatItem::findNextWrapColumn(
+
+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->drawRect(boundingRect());
+  int width = 0;
+  QVariantList wrapList = data(ChatLineModel::WrapListRole).toList();
+  for(int i = 2; i < wrapList.count(); i+=2) {
+    QRect r(wrapList[i-1].toUInt(), 0, wrapList[i+1].toUInt() - wrapList[i-1].toUInt(), fontMetrics()->lineSpacing());
+    painter->drawRect(r);
+  }
 }
 
+
+/*
 void ChatItem::layout() {
   if(!_layout.additionalFormats().count()) return; // no text set
   if(_width <= 0) return;
@@ -137,6 +230,3 @@ void ChatItem::mouseMoveEvent ( QGraphicsSceneMouseEvent * event ) {
   } else QGraphicsTextItem::mouseMoveEvent(event);
 }
 */
-
-
-