From ec08bfac0790d7f5a0031d0c69997756b5230ff6 Mon Sep 17 00:00:00 2001 From: Nick Tiskov Date: Sun, 10 Mar 2013 22:04:05 +0400 Subject: [PATCH] Improve code readability in speed monitor --- src/qtlibtorrent/torrentspeedmonitor.cpp | 64 +++++++++++++----------- 1 file changed, 34 insertions(+), 30 deletions(-) diff --git a/src/qtlibtorrent/torrentspeedmonitor.cpp b/src/qtlibtorrent/torrentspeedmonitor.cpp index a37b229c4..fae0189e9 100644 --- a/src/qtlibtorrent/torrentspeedmonitor.cpp +++ b/src/qtlibtorrent/torrentspeedmonitor.cpp @@ -38,19 +38,25 @@ using namespace libtorrent; +template struct Sample { + Sample(const T down = 0, const T up = 0) : download(down), upload(up) {} + T download; + T upload; +}; + class SpeedSample { public: SpeedSample() {} - void addSample(qint32 speedDL, qint32 speedUL); - QPair average() const; + void addSample(int speedDL, int speedUL); + Sample average() const; void clear(); private: static const int max_samples = 30; private: - QList > m_speedSamples; + QList > m_speedSamples; }; TorrentSpeedMonitor::TorrentSpeedMonitor(QBtSession* session) : @@ -76,31 +82,28 @@ void TorrentSpeedMonitor::run() } while(!m_abort); } -void SpeedSample::addSample(qint32 speedDL, qint32 speedUL) +void SpeedSample::addSample(int speedDL, int speedUL) { - m_speedSamples << qMakePair(speedDL, speedUL); + m_speedSamples << Sample(speedDL, speedUL); if (m_speedSamples.size() > max_samples) m_speedSamples.removeFirst(); } -QPair SpeedSample::average() const +Sample SpeedSample::average() const { if (m_speedSamples.empty()) - return qMakePair(0., 0.); + return Sample(); - qlonglong sumDL = 0, sumUL =0; + qlonglong sumDL = 0; + qlonglong sumUL = 0; - QPair p; - foreach (p, m_speedSamples) { - sumDL += p.first; - sumUL += p.second; + foreach (const Sample& s, m_speedSamples) { + sumDL += s.download; + sumUL += s.upload; } - qreal numSamples = m_speedSamples.size(); - return qMakePair( - (sumDL/numSamples), - (sumUL/numSamples)); - + const qreal numSamples = m_speedSamples.size(); + return Sample(sumDL/numSamples, sumUL/numSamples); } void SpeedSample::clear() @@ -126,10 +129,10 @@ qlonglong TorrentSpeedMonitor::getETA(const QString &hash) const if (h.is_paused() || !m_samples.contains(hash)) return MAX_ETA; - const QPair speed_average = m_samples.value(hash).average(); + const Sample speed_average = m_samples[hash].average(); - if(h.is_seed()) { - if (speed_average.second == 0) + if (h.is_seed()) { + if (!speed_average.upload) return MAX_ETA; bool _unused; @@ -141,13 +144,13 @@ qlonglong TorrentSpeedMonitor::getETA(const QString &hash) const if (realDL <= 0) realDL = h.total_wanted(); - return (realDL * max_ratio - h.all_time_upload()) / speed_average.second; - } else { - if (speed_average.first == 0) - return MAX_ETA; - - return (h.total_wanted() - h.total_done()) / speed_average.first; + return (realDL * max_ratio - h.all_time_upload()) / speed_average.upload; } + + if (!speed_average.download) + return MAX_ETA; + + return (h.total_wanted() - h.total_done()) / speed_average.download; } void TorrentSpeedMonitor::getSamples() @@ -160,12 +163,13 @@ void TorrentSpeedMonitor::getSamples() try { #if LIBTORRENT_VERSION_MINOR > 15 torrent_status st = it->status(0x0); - if (!st.paused) - m_samples[misc::toQString(it->info_hash())].addSample(st.download_payload_rate, st.upload_payload_rate); + if (!st.paused) { #else - if (!it->is_paused()) - m_samples[misc::toQString(it->info_hash())].addSample(it->status().download_payload_rate, it->status().upload_payload_rate); + if (!it->is_paused()) { + torrent_status st = it->status(); #endif + m_samples[misc::toQString(it->info_hash())].addSample(st.download_payload_rate, st.upload_payload_rate); + } } catch(invalid_handle&) {} } }