+#include "chatline.h"
+#include "chatlinemodel.h"
+#include "chatview.h"
+#include "contextmenuactionprovider.h"
+#include "mainwin.h"
+#include "qtui.h"
+#include "qtuistyle.h"
+
+ChatItem::ChatItem(const QRectF &boundingRect, ChatLine *parent)
+ : _parent(parent),
+ _boundingRect(boundingRect),
+ _selectionMode(NoSelection),
+ _selectionStart(-1),
+ _cachedLayout(0)
+{
+}
+
+
+ChatItem::~ChatItem()
+{
+ delete _cachedLayout;
+}
+
+
+ChatLine *ChatItem::chatLine() const
+{
+ return _parent;
+}
+
+
+ChatScene *ChatItem::chatScene() const
+{
+ return chatLine()->chatScene();
+}
+
+
+ChatView *ChatItem::chatView() const
+{
+ return chatScene()->chatView();
+}
+
+
+const QAbstractItemModel *ChatItem::model() const
+{
+ return chatLine()->model();
+}
+
+
+int ChatItem::row() const
+{
+ return chatLine()->row();
+}
+
+
+QPointF ChatItem::mapToLine(const QPointF &p) const
+{
+ return p + pos();
+}
+
+
+QPointF ChatItem::mapFromLine(const QPointF &p) const
+{
+ return p - pos();
+}
+
+
+// relative to the ChatLine
+QPointF ChatItem::mapToScene(const QPointF &p) const
+{
+ return chatLine()->mapToScene(p /* + pos() */);
+}
+
+
+QPointF ChatItem::mapFromScene(const QPointF &p) const
+{
+ return chatLine()->mapFromScene(p) /* - pos() */;
+}
+
+
+QVariant ChatItem::data(int role) const
+{
+ QModelIndex index = model()->index(row(), column());
+ if (!index.isValid()) {
+ qWarning() << "ChatItem::data(): model index is invalid!" << index;
+ return QVariant();
+ }
+ return model()->data(index, role);
+}
+
+
+QTextLayout *ChatItem::layout() const
+{
+ if (_cachedLayout)
+ return _cachedLayout;
+
+ _cachedLayout = new QTextLayout;
+ initLayout(_cachedLayout);
+ chatView()->setHasCache(chatLine());
+ return _cachedLayout;
+}
+
+
+void ChatItem::clearCache()
+{
+ delete _cachedLayout;
+ _cachedLayout = 0;
+}
+
+
+void ChatItem::initLayoutHelper(QTextLayout *layout, QTextOption::WrapMode wrapMode, Qt::Alignment alignment) const
+{
+ Q_ASSERT(layout);
+
+ layout->setText(data(MessageModel::DisplayRole).toString());
+
+ QTextOption option;
+ option.setWrapMode(wrapMode);
+ option.setAlignment(alignment);
+ layout->setTextOption(option);
+
+ QList<QTextLayout::FormatRange> formatRanges
+ = QtUi::style()->toTextLayoutList(formatList(), layout->text().length(), data(ChatLineModel::MsgLabelRole).toUInt());
+ layout->setAdditionalFormats(formatRanges);
+}
+
+
+void ChatItem::initLayout(QTextLayout *layout) const
+{
+ initLayoutHelper(layout, QTextOption::NoWrap);
+ doLayout(layout);
+}
+
+
+void ChatItem::doLayout(QTextLayout *layout) const
+{
+ layout->beginLayout();
+ QTextLine line = layout->createLine();
+ if (line.isValid()) {
+ line.setLineWidth(width());
+ line.setPosition(QPointF(0, 0));
+ }
+ layout->endLayout();
+}
+
+
+UiStyle::FormatList ChatItem::formatList() const
+{
+ return data(MessageModel::FormatRole).value<UiStyle::FormatList>();
+}
+
+
+qint16 ChatItem::posToCursor(const QPointF &posInLine) const
+{
+ QPointF pos = mapFromLine(posInLine);
+ if (pos.y() > height())
+ return data(MessageModel::DisplayRole).toString().length();
+ if (pos.y() < 0)
+ return 0;
+
+ 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);
+ }
+ }
+ return 0;
+}
+
+
+void ChatItem::paintBackground(QPainter *painter)
+{
+ QVariant bgBrush;
+ if (_selectionMode == FullSelection)
+ bgBrush = data(ChatLineModel::SelectedBackgroundRole);
+ else
+ bgBrush = data(ChatLineModel::BackgroundRole);
+ if (bgBrush.isValid())
+ painter->fillRect(boundingRect(), bgBrush.value<QBrush>());
+}
+
+
+// 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);
+ painter->save();
+ painter->setClipRect(boundingRect());
+ paintBackground(painter);
+
+ layout()->draw(painter, pos(), additionalFormats(), boundingRect());
+
+ // layout()->draw(painter, QPointF(0,0), formats, boundingRect());
+
+ // Debuging Stuff
+ // uncomment partially or all of the following stuff:
+ //
+ // 0) alternativ painter color for debug stuff
+// if(row() % 2)
+// painter->setPen(Qt::red);
+// else
+// painter->setPen(Qt::blue);
+// 1) draw wordwrap points in the first line
+// if(column() == 2) {
+// ChatLineModel::WrapList wrapList = data(ChatLineModel::WrapListRole).value<ChatLineModel::WrapList>();
+// foreach(ChatLineModel::Word word, wrapList) {
+// if(word.endX > width())
+// break;
+// painter->drawLine(word.endX, 0, word.endX, height());
+// }
+// }
+// 2) draw MsgId over the time column
+// if(column() == 0) {
+// QString msgIdString = QString::number(data(MessageModel::MsgIdRole).value<MsgId>().toInt());
+// QPointF bottomPoint = boundingRect().bottomLeft();
+// bottomPoint.ry() -= 2;
+// painter->drawText(bottomPoint, msgIdString);
+// }
+// 3) draw bounding rect
+// painter->drawRect(_boundingRect.adjusted(0, 0, -1, -1));
+
+ painter->restore();
+}
+
+
+void ChatItem::overlayFormat(UiStyle::FormatList &fmtList, int start, int end, quint32 overlayFmt) const
+{
+ for (int i = 0; i < fmtList.count(); i++) {
+ int fmtStart = fmtList.at(i).first;
+ int fmtEnd = (i < fmtList.count()-1 ? fmtList.at(i+1).first : data(MessageModel::DisplayRole).toString().length());
+
+ if (fmtEnd <= start)
+ continue;
+ if (fmtStart >= end)
+ break;
+
+ // split the format if necessary
+ if (fmtStart < start) {
+ fmtList.insert(i, fmtList.at(i));
+ fmtList[++i].first = start;
+ }
+ if (end < fmtEnd) {
+ fmtList.insert(i, fmtList.at(i));
+ fmtList[i+1].first = end;
+ }
+
+ fmtList[i].second |= overlayFmt;
+ }
+}
+
+
+QVector<QTextLayout::FormatRange> ChatItem::additionalFormats() const
+{
+ return selectionFormats();
+}
+
+
+QVector<QTextLayout::FormatRange> ChatItem::selectionFormats() const
+{
+ if (!hasSelection())
+ return QVector<QTextLayout::FormatRange>();
+
+ int start, end;
+ if (_selectionMode == FullSelection) {
+ start = 0;
+ end = data(MessageModel::DisplayRole).toString().length();
+ }
+ else {
+ start = qMin(_selectionStart, _selectionEnd);
+ end = qMax(_selectionStart, _selectionEnd);
+ }
+
+ UiStyle::FormatList fmtList = formatList();
+
+ while (fmtList.count() > 1 && fmtList.at(1).first <= start)
+ fmtList.removeFirst();
+
+ fmtList.first().first = start;
+
+ while (fmtList.count() > 1 && fmtList.last().first >= end)
+ fmtList.removeLast();
+
+ return QtUi::style()->toTextLayoutList(fmtList, end, UiStyle::Selected|data(ChatLineModel::MsgLabelRole).toUInt()).toVector();
+}
+
+
+bool ChatItem::hasSelection() const
+{
+ if (_selectionMode == NoSelection)
+ return false;
+ if (_selectionMode == FullSelection)
+ return true;
+ // partial
+ return _selectionStart != _selectionEnd;
+}
+
+
+QString ChatItem::selection() const
+{
+ if (_selectionMode == FullSelection)
+ return data(MessageModel::DisplayRole).toString();
+ if (_selectionMode == PartialSelection)
+ return data(MessageModel::DisplayRole).toString().mid(qMin(_selectionStart, _selectionEnd), qAbs(_selectionStart - _selectionEnd));
+ return QString();
+}
+
+
+void ChatItem::setSelection(SelectionMode mode, qint16 start, qint16 end)
+{
+ _selectionMode = mode;
+ _selectionStart = start;
+ _selectionEnd = end;
+ chatLine()->update();
+}
+
+
+void ChatItem::setFullSelection()
+{
+ if (_selectionMode != FullSelection) {
+ _selectionMode = FullSelection;
+ chatLine()->update();
+ }
+}
+
+
+void ChatItem::clearSelection()
+{
+ if (_selectionMode != NoSelection) {
+ _selectionMode = NoSelection;
+ chatLine()->update();
+ }
+}
+
+
+void ChatItem::continueSelecting(const QPointF &pos)
+{
+ _selectionMode = PartialSelection;
+ _selectionEnd = posToCursor(pos);
+ chatLine()->update();
+}
+
+
+bool ChatItem::isPosOverSelection(const QPointF &pos) const
+{
+ if (_selectionMode == FullSelection)
+ return true;
+ if (_selectionMode == PartialSelection) {
+ int cursor = posToCursor(pos);
+ return cursor >= qMin(_selectionStart, _selectionEnd) && cursor <= qMax(_selectionStart, _selectionEnd);
+ }
+ return false;
+}
+
+
+QList<QRectF> ChatItem::findWords(const QString &searchWord, Qt::CaseSensitivity caseSensitive)
+{
+ QList<QRectF> resultList;
+ const QAbstractItemModel *model_ = model();
+ if (!model_)
+ return resultList;