From 2a88e790df52a87b101916416de2ee553ef1e77e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20St=C4=99pie=C5=84?= Date: Wed, 21 Sep 2011 01:16:23 +0200 Subject: [PATCH] New PieceAvailabilityBar. --- src/properties/pieceavailabilitybar.cpp | 225 ++++++++++++++++++++++++ src/properties/pieceavailabilitybar.h | 95 ++++------ src/properties/properties.pri | 3 +- 3 files changed, 257 insertions(+), 66 deletions(-) create mode 100644 src/properties/pieceavailabilitybar.cpp diff --git a/src/properties/pieceavailabilitybar.cpp b/src/properties/pieceavailabilitybar.cpp new file mode 100644 index 000000000..358c0ebf6 --- /dev/null +++ b/src/properties/pieceavailabilitybar.cpp @@ -0,0 +1,225 @@ +/* + * Bittorrent Client using Qt4 and libtorrent. + * Copyright (C) 2006 Christophe Dumez + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * In addition, as a special exception, the copyright holders give permission to + * link this program with the OpenSSL project's "OpenSSL" library (or with + * modified versions of it that use the same license as the "OpenSSL" library), + * and distribute the linked executables. You must obey the GNU General Public + * License in all respects for all of the code used other than "OpenSSL". If you + * modify file(s), you may extend this exception to your version of the file(s), + * but you are not obligated to do so. If you do not wish to do so, delete this + * exception statement from your version. + * + * Contact : chris@qbittorrent.org + */ + +#include "pieceavailabilitybar.h" + +//#include + +PieceAvailabilityBar::PieceAvailabilityBar(QWidget *parent) : + QWidget(parent) +{ + setFixedHeight(BAR_HEIGHT); + + bg_color = 0xffffff; + piece_color = 0x0000ff; + + updatePieceColors(); +} + +std::vector PieceAvailabilityBar::intToFloatVector(const std::vector &vecin, int reqSize) +{ + std::vector result(reqSize, 0.0); + + if (vecin.size() == 0) + return result; + + const float ratio = vecin.size() / (float)reqSize; + + const int maxElement = *std::max_element(vecin.begin(), vecin.end()); + + // qMax bacause in normalization we don't want divide by 0 + // if maxElement == 0 check will be disabled please enable this line: + // const int maxElement = qMax(*std::max_element(avail.begin(), avail.end()), 1); + + if (maxElement == 0) + return result; + + // simple linear transformation algorithm + // for example: + // image.x(0) = pieces.x(0.0 >= x < 1.7) + // image.x(1) = pieces.x(1.7 >= x < 3.4) + + for (int x = 0; x < reqSize; ++x) { + + // don't use previously calculated value "ratio" here!!! + // float cannot save irrational number like 7/9, if this number will be rounded up by std::ceil + // give you x2 == pieces.size(), and index out of range: pieces[x2] + // this code is safe, so keep that in mind when you try optimize more. + // tested with size = 3000000ul + + // R - real + const float fromR = (x * vecin.size()) / (float)reqSize; + const float toR = ((x + 1) * vecin.size()) / (float)reqSize; + + // C - integer + int fromC = fromR;// std::floor not needed + int toC = std::ceil(toR); + + // position in pieces table + // libtorrent::bitfield::m_size is unsigned int(31 bits), so qlonglong is not needed + // tested with size = 3000000ul + int x2 = fromC; + + // little speed up for really big pieces table, 10K+ size + const int toCMinusOne = toC - 1; + + // value in returned vector + float value = 0; + + // case when calculated range is (15.2 >= x < 15.7) + if (x2 == toCMinusOne) { + if (vecin[x2]) { + value += (toR - fromR) * vecin[x2]; + } + ++x2; + } + // case when (15.2 >= x < 17.8) + else { + // subcase (15.2 >= x < 16) + if (x2 != fromR) { + if (vecin[x2]) { + value += (1.0 - (fromR - fromC)) * vecin[x2]; + } + ++x2; + } + + // subcase (16 >= x < 17) + for (; x2 < toCMinusOne; ++x2) { + if (vecin[x2]) { + value += vecin[x2]; + } + } + + // subcase (17 >= x < 17.8) + if (x2 == toCMinusOne) { + if (vecin[x2]) { + value += (1.0 - (toC - toR)) * vecin[x2]; + } + ++x2; + } + } + + // normalization <0, 1> + value /= ratio * maxElement; + + // float precision sometimes gives > 1, bacause in not possible to store irrational numbers + value = qMin(value, (float)1.0); + + result[x] = value; + } + + return result; +} + +int PieceAvailabilityBar::mixTwoColors(int &rgb1, int &rgb2, float ratio) +{ + int r1 = qRed(rgb1); + int g1 = qGreen(rgb1); + int b1 = qBlue(rgb1); + + int r2 = qRed(rgb2); + int g2 = qGreen(rgb2); + int b2 = qBlue(rgb2); + + float ratio_n = 1.0 - ratio; + int r = (r1 * ratio_n) + (r2 * ratio); + int g = (g1 * ratio_n) + (g2 * ratio); + int b = (b1 * ratio_n) + (b2 * ratio); + + return qRgb(r, g, b); +} + +void PieceAvailabilityBar::updateImage() +{ +// qDebug() << "updateImageAv"; + QImage image2(width() - 2, 1, QImage::Format_RGB888); + + if (pieces.empty()) { + image2.fill(0xffffff); + image = image2; + update(); + return; + } + + std::vector scaled_pieces = intToFloatVector(pieces, image2.width()); + + // filling image + for (unsigned int x = 0; x < scaled_pieces.size(); ++x) + { + float pieces2_val = scaled_pieces.at(x); + image2.setPixel(x, 0, piece_colors[pieces2_val * 255]); + } + image = image2; +} + +void PieceAvailabilityBar::setAvailability(const std::vector& avail) +{ + pieces.assign(avail.begin(), avail.end()); + + updateImage(); + update(); +} + +void PieceAvailabilityBar::updatePieceColors() +{ + piece_colors = std::vector(256); + for (int i = 0; i < 256; ++i) { + float ratio = (i / 255.0); + piece_colors[i] = mixTwoColors(bg_color, piece_color, ratio); + } +} + +void PieceAvailabilityBar::clear() +{ + image = QImage(); + update(); +} + +void PieceAvailabilityBar::paintEvent(QPaintEvent *) +{ + QPainter painter(this); + QRect imageRect(1, 1, width() - 2, height() - 2); + if (image.isNull()) + { + painter.setBrush(Qt::white); + painter.drawRect(imageRect); + } + else + { + if (image.width() != imageRect.width()) + updateImage(); + painter.drawImage(imageRect, image); + } + QPainterPath border; + border.addRect(0, 0, width() - 1, height() - 1); + + painter.setPen(palette().color(QPalette::Dark)); + painter.drawPath(border); +} diff --git a/src/properties/pieceavailabilitybar.h b/src/properties/pieceavailabilitybar.h index 278cb8ce1..6c568fe9e 100644 --- a/src/properties/pieceavailabilitybar.h +++ b/src/properties/pieceavailabilitybar.h @@ -33,85 +33,50 @@ #include #include -#include -#include -#include +#include #include #include #define BAR_HEIGHT 18 class PieceAvailabilityBar: public QWidget { - Q_OBJECT - Q_DISABLE_COPY(PieceAvailabilityBar) + Q_OBJECT + Q_DISABLE_COPY(PieceAvailabilityBar) private: - QPixmap pixmap; + QImage image; + + // I used values, bacause it should be possible to change colors in runtime + + // background color + int bg_color; + // complete piece color + int piece_color; + // buffered 256 levels gradient from bg_color to piece_color + std::vector piece_colors; + + // last used int vector, uses to better resize redraw + // TODO: make a diff pieces to new pieces and update only changed pixels, speedup when update > 20x faster + std::vector pieces; + + // scale int vector to float vector + std::vector intToFloatVector(const std::vector &vecin, int reqSize); + + // mix two colors by light model, ratio <0, 1> + int mixTwoColors(int &rgb1, int &rgb2, float ratio); + // draw new image and replace actual image + void updateImage(); public: - PieceAvailabilityBar(QWidget *parent): QWidget(parent) { - setFixedHeight(BAR_HEIGHT); - } + PieceAvailabilityBar(QWidget *parent); - void setAvailability(const std::vector& avail) { - if(avail.empty()) { - // Empty bar - QPixmap pix = QPixmap(1, 1); - pix.fill(); - pixmap = pix; - } else { - // Reduce the number of pieces before creating the pixmap - // otherwise it can crash when there are too many pieces - const qulonglong nb_pieces = avail.size(); - const uint w = width(); - if(nb_pieces > w) { - const qulonglong ratio = floor(nb_pieces/(double)w); - std::vector scaled_avail; - scaled_avail.reserve(ceil(nb_pieces/(double)ratio)); - for(qulonglong i=0; i& avail); + void updatePieceColors(); + void clear(); protected: - void paintEvent(QPaintEvent *) { - if(pixmap.isNull()) return; - QPainter painter(this); - painter.drawPixmap(rect(), pixmap); - } + void paintEvent(QPaintEvent *); -private: - void updatePixmap(const std::vector avail) { - const int max = *std::max_element(avail.begin(), avail.end()); - if(max == 0) { - QPixmap pix = QPixmap(1, 1); - pix.fill(); - pixmap = pix; - return; - } - QPixmap pix = QPixmap(avail.size(), 1); - //pix.fill(); - QPainter painter(&pix); - for(uint i=0; i < avail.size(); ++i) { - const uint rg = 0xff - (0xff * avail[i]/max); - painter.setPen(QColor(rg, rg, 0xff)); - painter.drawPoint(i,0); - } - pixmap = pix; - } }; #endif // PIECEAVAILABILITYBAR_H diff --git a/src/properties/properties.pri b/src/properties/properties.pri index 58ed4ae5d..29db6bb0a 100644 --- a/src/properties/properties.pri +++ b/src/properties/properties.pri @@ -19,4 +19,5 @@ SOURCES += $$PWD/propertieswidget.cpp \ $$PWD/peerlistwidget.cpp \ $$PWD/trackerlist.cpp \ $$PWD/proptabbar.cpp \ - properties/downloadedpiecesbar.cpp + properties/downloadedpiecesbar.cpp \ + properties/pieceavailabilitybar.cpp