X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fqtui%2Fchatviewsearchcontroller.cpp;h=8f3423d54fd10fe8a74b04c856edb2fbc88c0ba9;hp=1635d90970183a4921ac76e4f238e8aabf79baf9;hb=f04db2cb802b1296ca739c823495930c71d3b4ab;hpb=49f966fd9c04f3aa3149be73bb61b8e1781ce1c2

diff --git a/src/qtui/chatviewsearchcontroller.cpp b/src/qtui/chatviewsearchcontroller.cpp
index 1635d909..8f3423d5 100644
--- a/src/qtui/chatviewsearchcontroller.cpp
+++ b/src/qtui/chatviewsearchcontroller.cpp
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-08 by the Quassel Project                          *
+ *   Copyright (C) 2005-2013 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
@@ -15,7 +15,7 @@
  *   You should have received a copy of the GNU General Public License     *
  *   along with this program; if not, write to the                         *
  *   Free Software Foundation, Inc.,                                       *
- *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+ *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  ***************************************************************************/
 
 #include "chatviewsearchcontroller.h"
@@ -24,12 +24,13 @@
 #include <QPainter>
 
 #include "chatitem.h"
+#include "chatline.h"
 #include "chatlinemodel.h"
 #include "chatscene.h"
 #include "messagemodel.h"
 
 ChatViewSearchController::ChatViewSearchController(QObject *parent)
-  : QObject(parent),
+    : QObject(parent),
     _scene(0),
     _currentHighlight(0),
     _caseSensitive(false),
@@ -39,192 +40,354 @@ ChatViewSearchController::ChatViewSearchController(QObject *parent)
 {
 }
 
-void ChatViewSearchController::setSearchString(const QString &searchString) {
-  QString oldSearchString = _searchString;
-  _searchString = searchString;
-  if(_scene) {
-    if(!searchString.startsWith(oldSearchString) || oldSearchString.isEmpty()) {
-      // we can't reuse our all findings... cler the scene and do it all over
-      updateHighlights();
-    } else {
-      // reuse all findings
-      updateHighlights(true);
+
+void ChatViewSearchController::setSearchString(const QString &searchString)
+{
+    QString oldSearchString = _searchString;
+    _searchString = searchString;
+    if (_scene) {
+        if (!searchString.startsWith(oldSearchString) || oldSearchString.isEmpty()) {
+            // we can't reuse our all findings... cler the scene and do it all over
+            updateHighlights();
+        }
+        else {
+            // reuse all findings
+            updateHighlights(true);
+        }
     }
-  }
 }
 
- void ChatViewSearchController::setScene(ChatScene *scene) {
-  Q_ASSERT(scene);
-  if(scene == _scene)
-    return;
 
-  if(_scene) {
-    disconnect(_scene, 0, this, 0);
-    qDeleteAll(_highlightItems);
-    _highlightItems.clear();
-  }
+void ChatViewSearchController::setScene(ChatScene *scene)
+{
+    Q_ASSERT(scene);
+    if (scene == _scene)
+        return;
+
+    if (_scene) {
+        disconnect(_scene, 0, this, 0);
+        disconnect(Client::messageModel(), 0, this, 0);
+        qDeleteAll(_highlightItems);
+        _highlightItems.clear();
+    }
 
-  _scene = scene;
-  if(!scene)
-    return;
+    _scene = scene;
+    if (!scene)
+        return;
+
+    connect(_scene, SIGNAL(destroyed()), this, SLOT(sceneDestroyed()));
+    connect(_scene, SIGNAL(layoutChanged()), this, SLOT(repositionHighlights()));
+    connect(Client::messageModel(), SIGNAL(finishedBacklogFetch(BufferId)), this, SLOT(updateHighlights()));
+    updateHighlights();
+}
 
-  connect(_scene, SIGNAL(destroyed()), this, SLOT(sceneDestroyed()));
-  updateHighlights();
- }
 
-void ChatViewSearchController::highlightNext() {
-  if(_highlightItems.isEmpty())
-    return;
+void ChatViewSearchController::highlightNext()
+{
+    if (_highlightItems.isEmpty())
+        return;
 
-  if(_currentHighlight < _highlightItems.count()) {
-    _highlightItems.at(_currentHighlight)->setHighlighted(false);
-  }
+    if (_currentHighlight < _highlightItems.count()) {
+        _highlightItems.at(_currentHighlight)->setHighlighted(false);
+    }
 
-  _currentHighlight++;
-  if(_currentHighlight >= _highlightItems.count())
-    _currentHighlight = 0;
-  _highlightItems.at(_currentHighlight)->setHighlighted(true);
-  emit newCurrentHighlight(_highlightItems.at(_currentHighlight));
+    _currentHighlight++;
+    if (_currentHighlight >= _highlightItems.count())
+        _currentHighlight = 0;
+    _highlightItems.at(_currentHighlight)->setHighlighted(true);
+    emit newCurrentHighlight(_highlightItems.at(_currentHighlight));
 }
 
-void ChatViewSearchController::highlightPrev() {
-  if(_highlightItems.isEmpty())
-    return;
 
-  if(_currentHighlight < _highlightItems.count()) {
-    _highlightItems.at(_currentHighlight)->setHighlighted(false);
-  }
+void ChatViewSearchController::highlightPrev()
+{
+    if (_highlightItems.isEmpty())
+        return;
+
+    if (_currentHighlight < _highlightItems.count()) {
+        _highlightItems.at(_currentHighlight)->setHighlighted(false);
+    }
 
-  _currentHighlight--;
-  if(_currentHighlight < 0)
-    _currentHighlight = _highlightItems.count() - 1;
-  _highlightItems.at(_currentHighlight)->setHighlighted(true);
-  emit newCurrentHighlight(_highlightItems.at(_currentHighlight));
+    _currentHighlight--;
+    if (_currentHighlight < 0)
+        _currentHighlight = _highlightItems.count() - 1;
+    _highlightItems.at(_currentHighlight)->setHighlighted(true);
+    emit newCurrentHighlight(_highlightItems.at(_currentHighlight));
 }
 
-void ChatViewSearchController::updateHighlights(bool reuse) {
-  if(!_scene)
-    return;
 
-  QAbstractItemModel *model = _scene->model();
-  Q_ASSERT(model);
+void ChatViewSearchController::updateHighlights(bool reuse)
+{
+    if (!_scene)
+        return;
+
+    if (reuse) {
+        QSet<ChatLine *> chatLines;
+        foreach(SearchHighlightItem *highlightItem, _highlightItems) {
+            ChatLine *line = qgraphicsitem_cast<ChatLine *>(highlightItem->parentItem());
+            if (line)
+                chatLines << line;
+        }
+        foreach(ChatLine *line, QList<ChatLine *>(chatLines.toList())) {
+            updateHighlights(line);
+        }
+    }
+    else {
+        QPointF oldHighlightPos;
+        if (!_highlightItems.isEmpty() && _currentHighlight < _highlightItems.count()) {
+            oldHighlightPos = _highlightItems[_currentHighlight]->scenePos();
+        }
+        qDeleteAll(_highlightItems);
+        _highlightItems.clear();
+        Q_ASSERT(_highlightItems.isEmpty());
+
+        if (searchString().isEmpty() || !(_searchSenders || _searchMsgs))
+            return;
+
+        checkMessagesForHighlight();
+
+        if (!_highlightItems.isEmpty()) {
+            if (!oldHighlightPos.isNull()) {
+                int start = 0; int end = _highlightItems.count() - 1;
+                QPointF startPos;
+                QPointF endPos;
+                while (1) {
+                    startPos = _highlightItems[start]->scenePos();
+                    endPos = _highlightItems[end]->scenePos();
+                    if (startPos == oldHighlightPos) {
+                        _currentHighlight = start;
+                        break;
+                    }
+                    if (endPos == oldHighlightPos) {
+                        _currentHighlight = end;
+                        break;
+                    }
+                    if (end - start == 1) {
+                        _currentHighlight = start;
+                        break;
+                    }
+                    int pivot = (end + start) / 2;
+                    QPointF pivotPos = _highlightItems[pivot]->scenePos();
+                    if (startPos.y() == endPos.y()) {
+                        if (oldHighlightPos.x() <= pivotPos.x())
+                            end = pivot;
+                        else
+                            start = pivot;
+                    }
+                    else {
+                        if (oldHighlightPos.y() <= pivotPos.y())
+                            end = pivot;
+                        else
+                            start = pivot;
+                    }
+                }
+            }
+            else {
+                _currentHighlight = _highlightItems.count() - 1;
+            }
+            _highlightItems[_currentHighlight]->setHighlighted(true);
+            emit newCurrentHighlight(_highlightItems[_currentHighlight]);
+        }
+    }
+}
 
 
-  QList<ChatLine *> chatLines;
-  if(reuse) {
-    foreach(SearchHighlightItem *highlightItem, _highlightItems) {
-      ChatLine *line = dynamic_cast<ChatLine *>(highlightItem->parentItem());
-      if(!line || chatLines.contains(line))
-	continue;
-      chatLines << line;
+void ChatViewSearchController::checkMessagesForHighlight(int start, int end)
+{
+    QAbstractItemModel *model = _scene->model();
+    Q_ASSERT(model);
+
+    if (end == -1) {
+        end = model->rowCount() - 1;
+        if (end == -1)
+            return;
     }
-  }
 
-  qDeleteAll(_highlightItems);
-  _highlightItems.clear();
-  Q_ASSERT(_highlightItems.isEmpty());
+    QModelIndex index;
+    for (int row = start; row <= end; row++) {
+        if (_searchOnlyRegularMsgs) {
+            index = model->index(row, 0);
+            if (!checkType((Message::Type)index.data(MessageModel::TypeRole).toInt()))
+                continue;
+        }
+        highlightLine(_scene->chatLine(row));
+    }
+}
 
-  if(searchString().isEmpty() || !(_searchSenders || _searchMsgs))
-    return;
 
-  if(reuse) {
-    QModelIndex index;
-    foreach(ChatLine *line, chatLines) {
-      if(_searchOnlyRegularMsgs) {
-	index = model->index(line->row(), 0);
-	if(!checkType((Message::Type)index.data(MessageModel::TypeRole).toInt()))
-	  continue;
-      }
-      highlightLine(line);
+void ChatViewSearchController::updateHighlights(ChatLine *line)
+{
+    QList<ChatItem *> checkItems;
+    if (_searchSenders)
+        checkItems << line->item(MessageModel::SenderColumn);
+
+    if (_searchMsgs)
+        checkItems << line->item(MessageModel::ContentsColumn);
+
+    QHash<quint64, QHash<quint64, QRectF> > wordRects;
+    foreach(ChatItem *item, checkItems) {
+        foreach(QRectF wordRect, item->findWords(searchString(), caseSensitive())) {
+            wordRects[(quint64)(wordRect.x() + item->x())][(quint64)(wordRect.y())] = wordRect;
+        }
     }
-  } else {
-    // we have to crawl through the data
-    QModelIndex index;
-    QString plainText;
-    int rowCount = model->rowCount();
-    for(int row = 0; row < rowCount; row++) {
-      ChatLine *line = _scene->chatLine(row);
-
-      if(_searchOnlyRegularMsgs) {
-	index = model->index(row, 0);
-	if(!checkType((Message::Type)index.data(MessageModel::TypeRole).toInt()))
-	  continue;
-      }
-      highlightLine(line);
+
+    bool deleteAll = false;
+    QAbstractItemModel *model = _scene->model();
+    Q_ASSERT(model);
+    if (_searchOnlyRegularMsgs) {
+        QModelIndex index = model->index(line->row(), 0);
+        if (!checkType((Message::Type)index.data(MessageModel::TypeRole).toInt()))
+            deleteAll = true;
+    }
+
+    foreach(QGraphicsItem *child, line->childItems()) {
+        SearchHighlightItem *highlightItem = qgraphicsitem_cast<SearchHighlightItem *>(child);
+        if (!highlightItem)
+            continue;
+
+        if (!deleteAll && wordRects.contains((quint64)(highlightItem->pos().x())) && wordRects[(quint64)(highlightItem->pos().x())].contains((quint64)(highlightItem->pos().y()))) {
+            QRectF &wordRect = wordRects[(quint64)(highlightItem->pos().x())][(quint64)(highlightItem->pos().y())];
+            highlightItem->updateGeometry(wordRect.width(), wordRect.height());
+        }
+        else {
+            int pos = _highlightItems.indexOf(highlightItem);
+            if (pos == _currentHighlight) {
+                highlightPrev();
+            }
+            else if (pos < _currentHighlight) {
+                _currentHighlight--;
+            }
+
+            _highlightItems.removeAt(pos);
+            delete highlightItem;
+        }
     }
-  }
+}
+
+
+void ChatViewSearchController::highlightLine(ChatLine *line)
+{
+    QList<ChatItem *> checkItems;
+    if (_searchSenders)
+        checkItems << line->item(MessageModel::SenderColumn);
+
+    if (_searchMsgs)
+        checkItems << line->item(MessageModel::ContentsColumn);
+
+    foreach(ChatItem *item, checkItems) {
+        foreach(QRectF wordRect, item->findWords(searchString(), caseSensitive())) {
+            _highlightItems << new SearchHighlightItem(wordRect.adjusted(item->x(), 0, item->x(), 0), line);
+        }
+    }
+}
 
-  if(!_highlightItems.isEmpty()) {
-    _highlightItems.last()->setHighlighted(true);
-    _currentHighlight = _highlightItems.count() - 1;
-    emit newCurrentHighlight(_highlightItems.last());
-  }
+
+void ChatViewSearchController::repositionHighlights()
+{
+    QSet<ChatLine *> chatLines;
+    foreach(SearchHighlightItem *item, _highlightItems) {
+        ChatLine *line = qgraphicsitem_cast<ChatLine *>(item->parentItem());
+        if (line)
+            chatLines << line;
+    }
+    QList<ChatLine *> chatLineList(chatLines.toList());
+    foreach(ChatLine *line, chatLineList) {
+        repositionHighlights(line);
+    }
 }
 
-void ChatViewSearchController::highlightLine(ChatLine *line) {
-  QList<ChatItem *> checkItems;
-  if(_searchSenders)
-    checkItems << &(line->item(MessageModel::SenderColumn));
 
-  if(_searchMsgs)
-    checkItems << &(line->item(MessageModel::ContentsColumn));
+void ChatViewSearchController::repositionHighlights(ChatLine *line)
+{
+    QList<SearchHighlightItem *> searchHighlights;
+    foreach(QGraphicsItem *child, line->childItems()) {
+        SearchHighlightItem *highlightItem = qgraphicsitem_cast<SearchHighlightItem *>(child);
+        if (highlightItem)
+            searchHighlights << highlightItem;
+    }
+
+    if (searchHighlights.isEmpty())
+        return;
+
+    QList<QPointF> wordPos;
+    if (_searchSenders) {
+        foreach(QRectF wordRect, line->senderItem()->findWords(searchString(), caseSensitive())) {
+            wordPos << QPointF(wordRect.x() + line->senderItem()->x(), wordRect.y());
+        }
+    }
+    if (_searchMsgs) {
+        foreach(QRectF wordRect, line->contentsItem()->findWords(searchString(), caseSensitive())) {
+            wordPos << QPointF(wordRect.x() + line->contentsItem()->x(), wordRect.y());
+        }
+    }
+
+    qSort(searchHighlights.begin(), searchHighlights.end(), SearchHighlightItem::firstInLine);
 
-  foreach(ChatItem *item, checkItems) {
-    foreach(QRectF wordRect, item->findWords(searchString(), caseSensitive())) {
-      _highlightItems << new SearchHighlightItem(wordRect.adjusted(item->x(), 0, item->x(), 0), line);
+    Q_ASSERT(wordPos.count() == searchHighlights.count());
+    for (int i = 0; i < searchHighlights.count(); i++) {
+        searchHighlights.at(i)->setPos(wordPos.at(i));
     }
-  }
 }
 
-void ChatViewSearchController::sceneDestroyed() {
-  // WARNING: don't call any methods on scene!
-  _scene = 0;
-  // the items will be automatically deleted when the scene is destroyed
-  // so we just have to clear the list;
-  _highlightItems.clear();
+
+void ChatViewSearchController::sceneDestroyed()
+{
+    // WARNING: don't call any methods on scene!
+    _scene = 0;
+    // the items will be automatically deleted when the scene is destroyed
+    // so we just have to clear the list;
+    _highlightItems.clear();
 }
 
-void ChatViewSearchController::setCaseSensitive(bool caseSensitive) {
-  if(_caseSensitive == caseSensitive)
-    return;
 
-  _caseSensitive = caseSensitive;
+void ChatViewSearchController::setCaseSensitive(bool caseSensitive)
+{
+    if (_caseSensitive == caseSensitive)
+        return;
+
+    _caseSensitive = caseSensitive;
 
-  // we can reuse the original search results if the new search
-  // parameters are a restriction of the original one
-  updateHighlights(caseSensitive);
+    // we can reuse the original search results if the new search
+    // parameters are a restriction of the original one
+    updateHighlights(caseSensitive);
 }
 
-void ChatViewSearchController::setSearchSenders(bool searchSenders) {
-  if(_searchSenders == searchSenders)
-    return;
 
-  _searchSenders = searchSenders;
-  // we can reuse the original search results if the new search
-  // parameters are a restriction of the original one
-  updateHighlights(!searchSenders);
+void ChatViewSearchController::setSearchSenders(bool searchSenders)
+{
+    if (_searchSenders == searchSenders)
+        return;
+
+    _searchSenders = searchSenders;
+    // we can reuse the original search results if the new search
+    // parameters are a restriction of the original one
+    updateHighlights(!searchSenders);
 }
 
-void ChatViewSearchController::setSearchMsgs(bool searchMsgs) {
-  if(_searchMsgs == searchMsgs)
-    return;
 
-  _searchMsgs = searchMsgs;
+void ChatViewSearchController::setSearchMsgs(bool searchMsgs)
+{
+    if (_searchMsgs == searchMsgs)
+        return;
+
+    _searchMsgs = searchMsgs;
 
-  // we can reuse the original search results if the new search
-  // parameters are a restriction of the original one
-  updateHighlights(!searchMsgs);
+    // we can reuse the original search results if the new search
+    // parameters are a restriction of the original one
+    updateHighlights(!searchMsgs);
 }
 
-void ChatViewSearchController::setSearchOnlyRegularMsgs(bool searchOnlyRegularMsgs) {
-  if(_searchOnlyRegularMsgs == searchOnlyRegularMsgs)
-    return;
 
-  _searchOnlyRegularMsgs = searchOnlyRegularMsgs;
+void ChatViewSearchController::setSearchOnlyRegularMsgs(bool searchOnlyRegularMsgs)
+{
+    if (_searchOnlyRegularMsgs == searchOnlyRegularMsgs)
+        return;
 
-  // we can reuse the original search results if the new search
-  // parameters are a restriction of the original one
-  updateHighlights(searchOnlyRegularMsgs);
+    _searchOnlyRegularMsgs = searchOnlyRegularMsgs;
+
+    // we can reuse the original search results if the new search
+    // parameters are a restriction of the original one
+    updateHighlights(searchOnlyRegularMsgs);
 }
 
 
@@ -232,45 +395,68 @@ void ChatViewSearchController::setSearchOnlyRegularMsgs(bool searchOnlyRegularMs
 //  SearchHighlightItem
 // ==================================================
 SearchHighlightItem::SearchHighlightItem(QRectF wordRect, QGraphicsItem *parent)
-  : QObject(),
+    : QObject(),
     QGraphicsItem(parent),
     _highlighted(false),
-    _alpha(100),
+    _alpha(70),
     _timeLine(150)
 {
-  setPos(wordRect.x(), wordRect.y());
-  qreal sizedelta = wordRect.height() * 0.1;
-  _boundingRect = QRectF(-sizedelta, -sizedelta, wordRect.width() + 2 * sizedelta, wordRect.height() + 2 * sizedelta);
+    setPos(wordRect.x(), wordRect.y());
+    updateGeometry(wordRect.width(), wordRect.height());
 
-  connect(&_timeLine, SIGNAL(valueChanged(qreal)), this, SLOT(updateHighlight(qreal)));
+    connect(&_timeLine, SIGNAL(valueChanged(qreal)), this, SLOT(updateHighlight(qreal)));
 }
 
-void SearchHighlightItem::setHighlighted(bool highlighted) {
-  _highlighted = highlighted;
 
-  if(highlighted)
-    _timeLine.setDirection(QTimeLine::Forward);
-  else
-    _timeLine.setDirection(QTimeLine::Backward);
+void SearchHighlightItem::setHighlighted(bool highlighted)
+{
+    _highlighted = highlighted;
 
-  if(_timeLine.state() != QTimeLine::Running)
-    _timeLine.start();
+    if (highlighted)
+        _timeLine.setDirection(QTimeLine::Forward);
+    else
+        _timeLine.setDirection(QTimeLine::Backward);
 
-  update();
+    if (_timeLine.state() != QTimeLine::Running)
+        _timeLine.start();
+
+    update();
 }
 
-void SearchHighlightItem::updateHighlight(qreal value) {
-  _alpha = 100 + 155 * value;
-  update();
+
+void SearchHighlightItem::updateHighlight(qreal value)
+{
+    _alpha = 70 + (int)(80 * value);
+    update();
 }
 
-void SearchHighlightItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget) {
-  Q_UNUSED(option);
-  Q_UNUSED(widget);
 
-  painter->setPen(QPen(QColor(0, 0, 0, _alpha), 1.5));
-  painter->setBrush(QColor(254, 237, 45, _alpha));
-  painter->setRenderHints(QPainter::Antialiasing);
-  qreal radius = boundingRect().height() * 0.30;
-  painter->drawRoundedRect(boundingRect(), radius, radius);
+void SearchHighlightItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)
+{
+    Q_UNUSED(option);
+    Q_UNUSED(widget);
+
+    painter->setPen(QPen(QColor(0, 0, 0), 1.5));
+    painter->setBrush(QColor(254, 237, 45, _alpha));
+    painter->setRenderHints(QPainter::Antialiasing);
+    qreal radius = boundingRect().height() * 0.30;
+    painter->drawRoundedRect(boundingRect(), radius, radius);
+}
+
+
+void SearchHighlightItem::updateGeometry(qreal width, qreal height)
+{
+    prepareGeometryChange();
+    qreal sizedelta = height * 0.1;
+    _boundingRect = QRectF(-sizedelta, -sizedelta, width + 2 * sizedelta, height + 2 * sizedelta);
+    update();
+}
+
+
+bool SearchHighlightItem::firstInLine(QGraphicsItem *item1, QGraphicsItem *item2)
+{
+    if (item1->pos().y() != item2->pos().y())
+        return item1->pos().y() < item2->pos().y();
+    else
+        return item1->pos().x() < item2->pos().x();
 }