summaryrefslogtreecommitdiffstats
path: root/plugins/stats/PeerMonitor.cc
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/stats/PeerMonitor.cc')
-rw-r--r--plugins/stats/PeerMonitor.cc187
1 files changed, 187 insertions, 0 deletions
diff --git a/plugins/stats/PeerMonitor.cc b/plugins/stats/PeerMonitor.cc
new file mode 100644
index 0000000..a4cf449
--- /dev/null
+++ b/plugins/stats/PeerMonitor.cc
@@ -0,0 +1,187 @@
+/***************************************************************************
+ * Copyright © 2007 by Krzysztof Kundzicz *
+ * athantor@gmail.com *
+ * *
+ * 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., *
+ * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
+ ***************************************************************************/
+
+#include "PeerMonitor.h"
+
+namespace kt {
+
+PeerMonitor::PeerMonitor(kt::TorrentInterface * pTi, std::map<bt::SHA1Hash, PeerMonitor *> * pM) : kt::MonitorInterface(), QObject(), pmTorIface(pTi), pmPeerMMgr(pM)
+{
+
+}
+
+PeerMonitor::~PeerMonitor()
+{
+}
+
+void PeerMonitor::peerAdded (kt::PeerInterface *peer)
+{
+ QMutexLocker lock(&mtx);
+
+ mPeers.push_back( peer );
+}
+
+void PeerMonitor::peerRemoved (kt::PeerInterface *peer)
+{
+
+ QMutexLocker lock(&mtx);
+
+ data_t::iterator it = std::find(mPeers.begin(), mPeers.end(), peer);
+
+ if(it != mPeers.end())
+ {
+ mPeers.erase(it);
+ // *it = 0;
+ }
+
+}
+
+void PeerMonitor::downloadStarted (kt::ChunkDownloadInterface *)
+{
+
+}
+
+void PeerMonitor::downloadRemoved (kt::ChunkDownloadInterface *)
+{
+
+}
+
+void PeerMonitor::stopped ()
+{
+ QMutexLocker lock(&mtx);
+
+ std::fill(mPeers.begin(), mPeers.end(), static_cast<PeerInterface*>( 0 ) );
+// mPeers.clear();
+}
+
+void PeerMonitor::destroyed ()
+{
+ if(pmPeerMMgr -> find(pmTorIface -> getInfoHash()) != pmPeerMMgr -> end() )
+ {
+ pmTorIface -> setMonitor(0);
+ pmPeerMMgr -> erase(pmTorIface -> getInfoHash());
+ delete this;
+ }
+
+}
+
+double PeerMonitor::LeechersUpSpeed()
+{
+ QMutexLocker lock(&mtx);
+
+ double spd = 0.0;
+
+ //without it'll segfault/SIGABRT on stop as in meantime the iterator from
+ // mPeers will be invalidated
+
+ for( data_t::const_iterator it = mPeers.begin(); it != mPeers.end(); it++)
+ {
+ if((it != mPeers.end()) && *it && ( (*it) -> getStats().perc_of_file < 100.0) )
+ {
+ spd += (*it) -> getStats().download_rate;
+ }
+ }
+
+ return spd;
+}
+
+double PeerMonitor::LeechersDownSpeed()
+{
+ QMutexLocker lock(&mtx);
+
+ double spd = 0.0;
+
+
+ for( data_t::const_iterator it = mPeers.begin(); it != mPeers.end(); it++)
+ {
+ if((it != mPeers.end()) && *it && ( (*it) -> getStats().perc_of_file < 100.0) )
+ {
+ spd += (*it) -> getStats().upload_rate;
+ }
+ }
+
+ return spd;
+
+}
+
+double PeerMonitor::SeedersUpSpeed()
+{
+ QMutexLocker lock(&mtx);
+
+ double spd = 0.0;
+
+
+ for( data_t::const_iterator it = mPeers.begin(); it != mPeers.end(); it++)
+ {
+ if((it != mPeers.end()) && *it && ( (*it) -> getStats().perc_of_file == 100.0) )
+ {
+ spd += (*it) -> getStats().download_rate;
+ }
+
+ }
+
+ return spd;
+
+}
+
+uint64_t PeerMonitor::GetLeechers()
+{
+ QMutexLocker lock(&mtx);
+
+ uint64_t l = 0;
+
+
+ for( data_t::const_iterator it = mPeers.begin(); it != mPeers.end(); it++)
+ {
+ if((it != mPeers.end()) && *it && ( (*it) -> getStats().perc_of_file != 100.0) )
+ {
+ l++;
+ }
+
+ }
+
+ return l;
+}
+
+uint64_t PeerMonitor::GetSeeders()
+{
+ QMutexLocker lock(&mtx);
+
+ uint64_t s = 0;
+
+
+ for( data_t::const_iterator it = mPeers.begin(); it != mPeers.end(); it++)
+ {
+ if((it != mPeers.end()) && *it && ( (*it) -> getStats().perc_of_file == 100) )
+ {
+ s++;
+ }
+
+ }
+
+ return s;
+}
+
+kt::TorrentInterface * PeerMonitor::GetTorIface() const
+{
+ return pmTorIface;
+}
+
+} //NS end