Reworking the wordwrap stuff.

[quassel.git] / src / qtui / chatlinemodelitem.cpp

diff --git a/src/qtui/chatlinemodelitem.cpp b/src/qtui/chatlinemodelitem.cpp
index 59f7b5c..38663bb 100644 (file)
--- a/src/qtui/chatlinemodelitem.cpp
+++ b/src/qtui/chatlinemodelitem.cpp
@@ -18,6 +18,9 @@
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include <QFontMetrics>
+#include <QTextBoundaryFinder>
+
 #include "chatlinemodelitem.h"
 #include "chatlinemodel.h"
 #include "qtui.h"
@@ -34,6 +37,7 @@ ChatLineModelItem::ChatLineModelItem(const Message &msg) : MessageModelItem(msg)
   _sender.formatList = m.sender.formatList;
   _contents.formatList = m.contents.formatList;
 
+  computeWrapList();
 }
 
 
@@ -48,8 +52,13 @@ QVariant ChatLineModelItem::data(int column, int role) const {
   }
 
   switch(role) {
-    case ChatLineModel::DisplayRole: return part->plainText;
-    case ChatLineModel::FormatRole:  return QVariant::fromValue<UiStyle::FormatList>(part->formatList);
+    case ChatLineModel::DisplayRole:
+      return part->plainText;
+    case ChatLineModel::FormatRole:
+      return QVariant::fromValue<UiStyle::FormatList>(part->formatList);
+    case ChatLineModel::WrapListRole:
+      if(column != ChatLineModel::ContentsColumn) return QVariant();
+      return QVariant::fromValue<ChatLineModel::WrapList>(_wrapList);
   }
 
   return MessageModelItem::data(column, role);
@@ -58,3 +67,68 @@ QVariant ChatLineModelItem::data(int column, int role) const {
 bool ChatLineModelItem::setData(int column, const QVariant &value, int role) {
   return false;
 }
+
+// compute the width of a text snippet
+qreal ChatLineModelItem::snippetWidth(int start, int end, QFontMetricsF *&metrics, int &formatListIdx, int &formatEnd) {
+  qreal width = 0;
+  while(start < end) {
+    if(formatEnd <= start) {
+      formatListIdx++;
+      formatEnd = _contents.formatList.count() > formatListIdx+1 ? _contents.formatList[formatListIdx+1].first
+                                                                 : _contents.plainText.length();
+      metrics = QtUi::style()->fontMetrics(_contents.formatList[formatListIdx].second);
+      Q_ASSERT(formatEnd > start);
+    }
+    int i = qMin(end, formatEnd);
+    width += metrics->width(_contents.plainText.mid(start, i - start));
+    start = i;
+  }
+  return width;
+}
+
+void ChatLineModelItem::computeWrapList() {
+  enum Mode { SearchStart, SearchEnd };
+
+  QList<ChatLineModel::Word> wplist;  // use a temp list which we'll later copy into a QVector for efficiency
+  QTextBoundaryFinder finder(QTextBoundaryFinder::Word, _contents.plainText);
+  int idx, oldidx;
+  qreal pxpos = 0;
+  int flistidx = -1;
+  int fmtend = -1;
+  bool wordStart = false; bool wordEnd = false;
+  QFontMetricsF *metrics = 0;
+  Mode mode = SearchEnd;
+  ChatLineModel::Word word;
+  word.start = 0;
+  do {
+    idx = finder.toNextBoundary();
+    if(idx < 0) idx = _contents.plainText.length();
+    wordStart = finder.boundaryReasons().testFlag(QTextBoundaryFinder::StartWord);
+    wordEnd = finder.boundaryReasons().testFlag(QTextBoundaryFinder::EndWord);
+
+    //qDebug() << wordStart << wordEnd << _contents.plainText.left(idx) << _contents.plainText.mid(idx);
+
+    if(mode == SearchEnd || !wordStart && wordEnd) {
+      if(wordStart || !wordEnd) continue;
+      oldidx = idx;
+      mode = SearchStart;
+      continue;
+    }
+    // mode == SearchStart
+    word.width = snippetWidth(word.start, oldidx, metrics, flistidx, fmtend);
+    word.trailing = snippetWidth(oldidx, idx, metrics, flistidx, fmtend);
+    wplist.append(word);
+
+    if(wordStart) {
+      word.start = idx;
+      mode = SearchEnd;
+    }
+  } while(finder.isAtBoundary());
+
+  // A QVector needs less space than a QList
+  _wrapList.resize(wplist.count());
+  for(int i = 0; i < wplist.count(); i++) {
+    _wrapList[i] = wplist.at(i);
+  }
+}
+