From af3755bf919ffacba430c2fe86ad00ead1b61f88 Mon Sep 17 00:00:00 2001
From: Christophe Dumez <chris@qbittorrent.org>
Date: Thu, 18 Mar 2010 22:58:32 +0000
Subject: [PATCH] "Downloaded pieces" bar now displays in yellow the pieces
 being downloaded

---
 src/downloadedpiecesbar.h | 42 +++++++++++++++++++++++++++++----------
 src/propertieswidget.cpp  |  4 +++-
 src/qtorrenthandle.cpp    | 10 ++++++++++
 src/qtorrenthandle.h      |  1 +
 4 files changed, 45 insertions(+), 12 deletions(-)

diff --git a/src/downloadedpiecesbar.h b/src/downloadedpiecesbar.h
index 9dba245c6..8a5ee9d38 100644
--- a/src/downloadedpiecesbar.h
+++ b/src/downloadedpiecesbar.h
@@ -53,34 +53,49 @@ public:
     setFixedHeight(BAR_HEIGHT);
   }
 
-  void setProgress(bitfield pieces) {
+  void setProgress(const bitfield &pieces, const bitfield &downloading_pieces) {
     if(pieces.empty()) {
       // Empty bar
       QPixmap pix = QPixmap(1, 1);
       pix.fill();
       pixmap = pix;
     } else {
-      int nb_pieces = pieces.size();
+      const int nb_pieces = pieces.size();
       // Reduce the number of pieces before creating the pixmap
       // otherwise it can crash when there are too many pieces
       if(nb_pieces > width()) {
         int ratio = floor(nb_pieces/(double)width());
-        QVector<bool> scaled_pieces;
+        std::vector<bool> scaled_pieces;
+        std::vector<bool> scaled_downloading;
         for(int i=0; i<nb_pieces; i+= ratio) {
           bool have = true;
           for(int j=i; j<qMin(i+ratio, nb_pieces); ++j) {
             if(!pieces[i]) { have = false; break; }
           }
-          scaled_pieces << have;
+          scaled_pieces.push_back(have);
+          if(have) {
+            scaled_downloading.push_back(false);
+          } else {
+            bool downloading = false;
+            for(int j=i; j<qMin(i+ratio, nb_pieces); ++j) {
+              if(downloading_pieces[i]) { downloading = true; break; }
+            }
+            scaled_downloading.push_back(downloading);
+          }
         }
         QPixmap pix = QPixmap(scaled_pieces.size(), 1);
         pix.fill();
         QPainter painter(&pix);
-        for(int i=0; i<scaled_pieces.size(); ++i) {
-          if(scaled_pieces[i])
+        for(uint i=0; i<scaled_pieces.size(); ++i) {
+          if(scaled_pieces[i]) {
             painter.setPen(Qt::blue);
-          else
-            painter.setPen(Qt::white);
+          } else {
+            if(scaled_downloading[i]) {
+              painter.setPen(Qt::yellow);
+            } else {
+              painter.setPen(Qt::white);
+            }
+          }
           painter.drawPoint(i,0);
         }
         pixmap = pix;
@@ -89,10 +104,15 @@ public:
         pix.fill();
         QPainter painter(&pix);
         for(uint i=0; i<pieces.size(); ++i) {
-          if(pieces[i])
+          if(pieces[i]) {
             painter.setPen(Qt::blue);
-          else
-            painter.setPen(Qt::white);
+          } else {
+            if(downloading_pieces[i]) {
+              painter.setPen(Qt::yellow);
+            } else {
+              painter.setPen(Qt::white);
+            }
+          }
           painter.drawPoint(i,0);
         }
         pixmap = pix;
diff --git a/src/propertieswidget.cpp b/src/propertieswidget.cpp
index 56f90a7cc..2ad4eb9be 100644
--- a/src/propertieswidget.cpp
+++ b/src/propertieswidget.cpp
@@ -338,7 +338,9 @@ void PropertiesWidget::loadDynamicData() {
       if(!h.is_seed()) {
         showPiecesDownloaded(true);
         // Downloaded pieces
-        downloaded_pieces->setProgress(h.pieces());
+        bitfield bf(h.get_torrent_info().num_pieces(), 0);
+        h.downloading_pieces(bf);
+        downloaded_pieces->setProgress(h.pieces(), bf);
         // Pieces availability
         if(h.has_metadata() && !h.is_paused() && !h.is_queued() && !h.is_checking()) {
           showPiecesAvailability(true);
diff --git a/src/qtorrenthandle.cpp b/src/qtorrenthandle.cpp
index 3509333a7..4d16ed451 100644
--- a/src/qtorrenthandle.cpp
+++ b/src/qtorrenthandle.cpp
@@ -451,6 +451,16 @@ bool QTorrentHandle::has_error() const {
   return h.status().error.empty();
 }
 
+void QTorrentHandle::downloading_pieces(bitfield &bf) const {
+  Q_ASSERT(h.is_valid());
+  std::vector<partial_piece_info> queue;
+  h.get_download_queue(queue);
+  for(std::vector<partial_piece_info>::iterator it=queue.begin(); it!= queue.end(); it++) {
+    bf.set_bit(it->piece_index);
+  }
+  return;
+}
+
 //
 // Setters
 //
diff --git a/src/qtorrenthandle.h b/src/qtorrenthandle.h
index 86c9e7922..64ce17d5f 100644
--- a/src/qtorrenthandle.h
+++ b/src/qtorrenthandle.h
@@ -129,6 +129,7 @@ class QTorrentHandle {
     bool first_last_piece_first() const;
     QString root_path() const;
     bool has_error() const;
+    void downloading_pieces(bitfield &bf) const;
 
     //
     // Setters