Display and update message processing progress
[quassel.git] / src / qtui / chatscene.cpp
index 8d830b6..43ff248 100644 (file)
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include <QApplication>
+#include <QClipboard>
 #include <QGraphicsSceneMouseEvent>
 #include <QPersistentModelIndex>
 
 #include "buffer.h"
 #include "chatitem.h"
+#include "chatline.h"
 #include "chatlinemodelitem.h"
 #include "chatscene.h"
 #include "columnhandleitem.h"
@@ -37,11 +40,15 @@ ChatScene::ChatScene(QAbstractItemModel *model, const QString &idString, QObject
   _model(model)
 {
   _width = 0;
+  _selectingItem = 0;
+  _isSelecting = false;
+  _selectionStart = -1;
   connect(this, SIGNAL(sceneRectChanged(const QRectF &)), this, SLOT(rectChanged(const QRectF &)));
 
   connect(model, SIGNAL(rowsInserted(const QModelIndex &, int, int)), this, SLOT(rowsInserted(const QModelIndex &, int, int)));
+  connect(model, SIGNAL(modelAboutToBeReset()), this, SLOT(modelReset()));
   for(int i = 0; i < model->rowCount(); i++) {
-    ChatLine *line = new ChatLine(model->index(i, 0));
+    ChatLine *line = new ChatLine(i, model);
     _lines.append(line);
     addItem(line);
   }
@@ -71,7 +78,6 @@ ChatScene::ChatScene(QAbstractItemModel *model, const QString &idString, QObject
 
 ChatScene::~ChatScene() {
 
-
 }
 
 void ChatScene::rowsInserted(const QModelIndex &index, int start, int end) {
@@ -83,7 +89,7 @@ void ChatScene::rowsInserted(const QModelIndex &index, int start, int end) {
   qreal y = 0;
   if(_width && start > 0) y = _lines.value(start - 1)->y() + _lines.value(start - 1)->height();
   for(int i = start; i <= end; i++) {
-    ChatLine *line = new ChatLine(model()->index(i, 0));
+    ChatLine *line = new ChatLine(i, model());
     _lines.insert(i, line);
     addItem(line);
     if(_width > 0) {
@@ -91,6 +97,11 @@ void ChatScene::rowsInserted(const QModelIndex &index, int start, int end) {
       h += line->setGeometry(_width, firstColHandlePos, secondColHandlePos);
     }
   }
+  // update existing items
+  for(int i = end+1; i < _lines.count(); i++) {
+    _lines[i]->setRow(i);
+  }
+  
   if(h > 0) {
     _height += h;
     for(int i = end+1; i < _lines.count(); i++) {
@@ -101,6 +112,15 @@ void ChatScene::rowsInserted(const QModelIndex &index, int start, int end) {
   }
 }
 
+void ChatScene::modelReset() {
+  foreach(ChatLine *line, _lines) {
+    removeItem(line);
+    delete line;
+  }
+  _lines.clear();
+  setSceneRect(QRectF(0, 0, _width, 0));
+}
+
 void ChatScene::setWidth(qreal w) {
   _width = w;
   _height = 0;
@@ -139,3 +159,119 @@ void ChatScene::handlePositionChanged(qreal xpos) {
   // width() should be the same for both handles, so just use firstColHandle regardless
   update(qMin(oldx, xpos) - firstColHandle->width()/2, 0, qMax(oldx, xpos) + firstColHandle->width()/2, height());
 }
+
+void ChatScene::setSelectingItem(ChatItem *item) {
+  if(_selectingItem) _selectingItem->clearSelection();
+  _selectingItem = item;
+}
+
+void ChatScene::startGlobalSelection(ChatItem *item, const QPointF &itemPos) {
+  _selectionStart = _selectionEnd = item->row();
+  _selectionStartCol = _selectionMinCol = item->column();
+  _isSelecting = true;
+  _lines[_selectionStart]->setSelected(true, (ChatLineModel::ColumnType)_selectionMinCol);
+  updateSelection(item->mapToScene(itemPos));
+}
+
+void ChatScene::updateSelection(const QPointF &pos) {
+  // This is somewhat hacky... we look at the contents item that is at the cursor's y position (ignoring x), since
+  // it has the full height. From this item, we can then determine the row index and hence the ChatLine.
+  ChatItem *contentItem = static_cast<ChatItem *>(itemAt(QPointF(secondColHandlePos + secondColHandle->width()/2, pos.y())));
+  if(!contentItem) return;
+
+  int curRow = contentItem->row();
+  int curColumn;
+  if(pos.x() > secondColHandlePos + secondColHandle->width()/2) curColumn = ChatLineModel::ContentsColumn;
+  else if(pos.x() > firstColHandlePos) curColumn = ChatLineModel::SenderColumn;
+  else curColumn = ChatLineModel::TimestampColumn;
+
+  ChatLineModel::ColumnType minColumn = (ChatLineModel::ColumnType)qMin(curColumn, _selectionStartCol);
+  if(minColumn != _selectionMinCol) {
+    _selectionMinCol = minColumn;
+    for(int l = qMin(_selectionStart, _selectionEnd); l <= qMax(_selectionStart, _selectionEnd); l++) {
+      _lines[l]->setSelected(true, minColumn);
+    }
+  }
+
+  if(curRow > _selectionEnd && curRow > _selectionStart) {  // select further towards bottom
+    for(int l = _selectionEnd + 1; l <= curRow; l++) {
+      _lines[l]->setSelected(true, minColumn);
+    }
+  } else if(curRow > _selectionEnd && curRow <= _selectionStart) { // deselect towards bottom
+    for(int l = _selectionEnd; l < curRow; l++) {
+      _lines[l]->setSelected(false);
+    }
+  } else if(curRow < _selectionEnd && curRow >= _selectionStart) {
+    for(int l = _selectionEnd; l > curRow; l--) {
+      _lines[l]->setSelected(false);
+    }
+  } else if(curRow < _selectionEnd && curRow < _selectionStart) {
+    for(int l = _selectionEnd - 1; l >= curRow; l--) {
+      _lines[l]->setSelected(true, minColumn);
+    }
+  }
+  _selectionEnd = curRow;
+
+  if(curRow == _selectionStart && minColumn == ChatLineModel::ContentsColumn) {
+    _lines[curRow]->setSelected(false);
+    _isSelecting = false;
+    _selectingItem->continueSelecting(_selectingItem->mapFromScene(pos));
+  }
+}
+
+void ChatScene::mouseMoveEvent(QGraphicsSceneMouseEvent *event) {
+  if(_isSelecting && event->buttons() & Qt::LeftButton) {
+    updateSelection(event->scenePos());
+    event->accept();
+  } else {
+    QGraphicsScene::mouseMoveEvent(event);
+  }
+}
+
+void ChatScene::mousePressEvent(QGraphicsSceneMouseEvent *event) {
+  if(event->buttons() & Qt::LeftButton && _selectionStart >= 0) {
+    for(int l = qMin(_selectionStart, _selectionEnd); l <= qMax(_selectionStart, _selectionEnd); l++) {
+      _lines[l]->setSelected(false);
+    }
+    _selectionStart = -1;
+    event->accept();
+  } else {
+    QGraphicsScene::mousePressEvent(event);
+  }
+}
+
+void ChatScene::mouseReleaseEvent(QGraphicsSceneMouseEvent *event) {
+  if(_isSelecting) {
+#   ifdef Q_WS_X11
+      QApplication::clipboard()->setText(selectionToString(), QClipboard::Selection);
+#   endif
+//# else
+      QApplication::clipboard()->setText(selectionToString());
+//# endif
+    _isSelecting = false;
+    event->accept();
+  } else {
+    QGraphicsScene::mouseReleaseEvent(event);
+  }
+}
+
+//!\brief Convert current selection to human-readable string.
+QString ChatScene::selectionToString() const {
+  //TODO Make selection format configurable!
+  if(!_isSelecting) return QString();
+  int start = qMin(_selectionStart, _selectionEnd);
+  int end = qMax(_selectionStart, _selectionEnd);
+  if(start < 0 || end >= _lines.count()) {
+    qDebug() << "Invalid selection range:" << start << end;
+    return QString();
+  }
+  QString result;
+  for(int l = start; l <= end; l++) {
+    if(_selectionMinCol == ChatLineModel::TimestampColumn)
+      result += _lines[l]->item(ChatLineModel::TimestampColumn).data(MessageModel::DisplayRole).toString() + " ";
+    if(_selectionMinCol <= ChatLineModel::SenderColumn)
+      result += _lines[l]->item(ChatLineModel::SenderColumn).data(MessageModel::DisplayRole).toString() + " ";
+    result += _lines[l]->item(ChatLineModel::ContentsColumn).data(MessageModel::DisplayRole).toString() + "\n";
+  }
+  return result;
+}