More wordwrap fixes.
[quassel.git] / src / qtui / chatlinemodelitem.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-08 by the Quassel Project                          *
3  *   devel@quassel-irc.org                                                 *
4  *                                                                         *
5  *   This program is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU General Public License as published by  *
7  *   the Free Software Foundation; either version 2 of the License, or     *
8  *   (at your option) version 3.                                           *
9  *                                                                         *
10  *   This program is distributed in the hope that it will be useful,       *
11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
13  *   GNU General Public License for more details.                          *
14  *                                                                         *
15  *   You should have received a copy of the GNU General Public License     *
16  *   along with this program; if not, write to the                         *
17  *   Free Software Foundation, Inc.,                                       *
18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
19  ***************************************************************************/
20
21 #include <QFontMetrics>
22 #include <QTextBoundaryFinder>
23
24 #include "chatlinemodelitem.h"
25 #include "chatlinemodel.h"
26 #include "qtui.h"
27 #include "uistyle.h"
28
29 ChatLineModelItem::ChatLineModelItem(const Message &msg) : MessageModelItem(msg) {
30   QtUiStyle::StyledMessage m = QtUi::style()->styleMessage(msg);
31
32   _timestamp.plainText = m.timestamp.plainText;
33   _sender.plainText = m.sender.plainText;
34   _contents.plainText = m.contents.plainText;
35
36   _timestamp.formatList = m.timestamp.formatList;
37   _sender.formatList = m.sender.formatList;
38   _contents.formatList = m.contents.formatList;
39
40   computeWrapList();
41 }
42
43
44 QVariant ChatLineModelItem::data(int column, int role) const {
45   const ChatLinePart *part;
46
47   switch(column) {
48     case ChatLineModel::TimestampColumn: part = &_timestamp; break;
49     case ChatLineModel::SenderColumn:    part = &_sender; break;
50     case ChatLineModel::ContentsColumn:      part = &_contents; break;
51     default: return MessageModelItem::data(column, role);
52   }
53
54   switch(role) {
55     case ChatLineModel::DisplayRole:
56       return part->plainText;
57     case ChatLineModel::FormatRole:
58       return QVariant::fromValue<UiStyle::FormatList>(part->formatList);
59     case ChatLineModel::WrapListRole:
60       if(column != ChatLineModel::ContentsColumn) return QVariant();
61       return QVariant::fromValue<ChatLineModel::WrapList>(_wrapList);
62   }
63
64   return MessageModelItem::data(column, role);
65 }
66
67 bool ChatLineModelItem::setData(int column, const QVariant &value, int role) {
68   return false;
69 }
70
71 // compute the width of a text snippet
72 qreal ChatLineModelItem::snippetWidth(int start, int end, QFontMetricsF *&metrics, int &formatListIdx, int &formatEnd) {
73   qreal width = 0;
74   while(start < end) {
75     if(formatEnd <= start) {
76       formatListIdx++;
77       formatEnd = _contents.formatList.count() > formatListIdx+1 ? _contents.formatList[formatListIdx+1].first
78                                                                  : _contents.plainText.length();
79       metrics = QtUi::style()->fontMetrics(_contents.formatList[formatListIdx].second);
80       Q_ASSERT(formatEnd > start);
81     }
82     int i = qMin(end, formatEnd);
83     width += metrics->width(_contents.plainText.mid(start, i - start));
84     start = i;
85   }
86   return width;
87 }
88
89 void ChatLineModelItem::computeWrapList() {
90   enum Mode { SearchStart, SearchEnd };
91
92   QList<ChatLineModel::Word> wplist;  // use a temp list which we'll later copy into a QVector for efficiency
93   QTextBoundaryFinder finder(QTextBoundaryFinder::Word, _contents.plainText);
94   int idx, oldidx;
95   qreal pxpos = 0;
96   int flistidx = -1;
97   int fmtend = -1;
98   bool wordStart = false; bool wordEnd = false;
99   QFontMetricsF *metrics = 0;
100   Mode mode = SearchEnd;
101   ChatLineModel::Word word;
102   word.start = 0;
103   do {
104     idx = finder.toNextBoundary();
105     if(idx < 0) idx = _contents.plainText.length();
106     wordStart = finder.boundaryReasons().testFlag(QTextBoundaryFinder::StartWord);
107     wordEnd = finder.boundaryReasons().testFlag(QTextBoundaryFinder::EndWord);
108
109     //qDebug() << wordStart << wordEnd << _contents.plainText.left(idx) << _contents.plainText.mid(idx);
110
111     if(mode == SearchEnd || !wordStart && wordEnd) {
112       if(wordStart || !wordEnd) continue;
113       oldidx = idx;
114       mode = SearchStart;
115       continue;
116     }
117     // mode == SearchStart
118     word.width = snippetWidth(word.start, oldidx, metrics, flistidx, fmtend);
119     word.trailing = snippetWidth(oldidx, idx, metrics, flistidx, fmtend);
120     wplist.append(word);
121
122     if(wordStart) {
123       word.start = idx;
124       mode = SearchEnd;
125     }
126   } while(finder.isAtBoundary());
127
128   // A QVector needs less space than a QList
129   _wrapList.resize(wplist.count());
130   for(int i = 0; i < wplist.count(); i++) {
131     _wrapList[i] = wplist.at(i);
132   }
133 }
134