From 9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0 Mon Sep 17 00:00:00 2001 From: tpearson Date: Wed, 20 Jan 2010 02:37:40 +0000 Subject: Updated to final KDE3 ktorrent release (2.2.6) git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/ktorrent@1077377 283d02a7-25f6-0310-bc7c-ecb5cbfe19da --- plugins/scheduler/bwscheduler.cpp | 282 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 282 insertions(+) create mode 100644 plugins/scheduler/bwscheduler.cpp (limited to 'plugins/scheduler/bwscheduler.cpp') diff --git a/plugins/scheduler/bwscheduler.cpp b/plugins/scheduler/bwscheduler.cpp new file mode 100644 index 0000000..56375b5 --- /dev/null +++ b/plugins/scheduler/bwscheduler.cpp @@ -0,0 +1,282 @@ +/*************************************************************************** + * Copyright (C) 2006 by Ivan Vasić * + * ivasic@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., * + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * + ***************************************************************************/ +#include "bwscheduler.h" +#include "schedulerpluginsettings.h" + +#include +#include + +#include +#include + +#include + +#include +#include + +#include +#include +#include + +#include +#include + +using namespace bt; + +namespace kt +{ + // ****** BWS *---------------------------------------------------------------------- + BWS::BWS() + { + m_schedule = new ScheduleCategory*[7]; + for(int i=0; i<7; ++i) + m_schedule[i] = new ScheduleCategory[24]; + + reset(); + } + + BWS& kt::BWS::operator=(const BWS& b) + { + for(int i=0; i<7; ++i) + for(int j=0; j<24; ++j) + m_schedule[i][j] = b.m_schedule[i][j]; + + for(int i=0; i<3; ++i) + { + download[i] = b.download[i]; + upload[i] = b.upload[i]; + } + + return *this; + } + + int BWS::getDownload(int cat) + { + return download[cat]; + } + + int BWS::getUpload(int cat) + { + return upload[cat]; + } + + void BWS::setDownload(int cat, int val) + { + download[cat] = val; + } + + void BWS::setUpload(int cat, int val) + { + upload[cat] = val; + } + + void BWS::setCategory(int day, int hour, ScheduleCategory val) + { + m_schedule[day][hour] = val; + } + + ScheduleCategory BWS::getCategory(int day, int hour) + { + return m_schedule[day][hour]; + } + + void BWS::reset() + { + for(int i=0; i<7; ++i) + for(int j=0; j<24; ++j) + m_schedule[i][j] = CAT_NORMAL; + + for(int i=0; i<3; ++i) + { + download[i] = 0; + upload[i] = 0; + } + } + + BWS::~BWS() + { + for(int i=0; i<7; ++i) + delete [] m_schedule[i]; + + delete [] m_schedule; + } + + void BWS::debug() + { + for(int i=0; i<7; ++i) + { + Log & lg = Out(); + for(int j=0; j<24; ++j) + lg << m_schedule[i][j]; + lg << endl; + } + } + + // ---- BWScheduler -------------------------------------------------------------------- + + BWScheduler::BWScheduler() + : m_core(0) + { + m_enabled = SchedulerPluginSettings::enableBWS(); + loadSchedule(); + } + + BWScheduler::~BWScheduler() + { + } + + void BWScheduler::setSchedule(const BWS& sch) + { + Out(SYS_SCD|LOG_NOTICE) << "BWS: Setting new schedule..." << endl; + m_schedule = sch; + saveSchedule(); + trigger(); + } + + void BWScheduler::setCoreInterface(CoreInterface* core) + { + m_core = core; + } + + void BWScheduler::trigger() + { + if(!m_enabled) + return; + + QDateTime now = QDateTime::currentDateTime(); + + QString prefix = QString("BWS: %1 :: ").arg(now.toString()); + + int t1 = now.date().dayOfWeek(); + int t2 = now.time().hour(); + ScheduleCategory sch = m_schedule.getCategory(t1-1, t2); + + switch(sch) + { + case CAT_NORMAL: + Out(SYS_SCD|LOG_NOTICE) << prefix << "Switching to NORMAL category" << endl; + Out(SYS_SCD|LOG_NOTICE) << prefix << QString("%1 Up, %2 Down") + .arg(m_core->getMaxUploadSpeed()).arg(m_core->getMaxDownloadSpeed()) << endl; + if(!m_core) + break; + m_core->setPausedState(false); + net::SocketMonitor::setDownloadCap(1024 * m_core->getMaxDownloadSpeed()); + net::SocketMonitor::setUploadCap(1024 * m_core->getMaxUploadSpeed()); + break; + case CAT_FIRST: + Out(SYS_SCD|LOG_NOTICE) << prefix << "Switching to FIRST category" << endl; + Out(SYS_SCD|LOG_NOTICE) << prefix << QString("%1 Up, %2 Down") + .arg(m_schedule.getUpload(0)).arg(m_schedule.getDownload(0)) << endl; + if(!m_core) + break; + m_core->setPausedState(false); + net::SocketMonitor::setDownloadCap(1024 * m_schedule.getDownload(0)); + net::SocketMonitor::setUploadCap(1024 * m_schedule.getUpload(0)); + break; + case CAT_SECOND: + Out(SYS_SCD|LOG_NOTICE) << prefix << "Switching to SECOND category" << endl; + Out(SYS_SCD|LOG_NOTICE) << prefix << QString("%1 Up, %2 Down") + .arg(m_schedule.getUpload(1)).arg(m_schedule.getDownload(1)) << endl; + if(!m_core) + break; + m_core->setPausedState(false); + net::SocketMonitor::setDownloadCap(1024 * m_schedule.getDownload(1)); + net::SocketMonitor::setUploadCap(1024 * m_schedule.getUpload(1)); + break; + case CAT_THIRD: + Out(SYS_SCD|LOG_NOTICE) << prefix << "Switching to THIRD category" << endl; + Out(SYS_SCD|LOG_NOTICE) << prefix << QString("%1 Up, %2 Down") + .arg(m_schedule.getUpload(2)).arg(m_schedule.getDownload(2)) << endl; + if(!m_core) + break; + m_core->setPausedState(false); + net::SocketMonitor::setDownloadCap(1024 * m_schedule.getDownload(2)); + net::SocketMonitor::setUploadCap(1024 * m_schedule.getUpload(2)); + break; + case CAT_OFF: + Out(SYS_SCD|LOG_NOTICE) << prefix << "Switching to OFF" << endl; + + if(!m_core) + break; + m_core->setPausedState(true); + break; + } + } + + void BWScheduler::loadSchedule() + { + QFile file(KGlobal::dirs()->saveLocation("data","ktorrent") + "bwschedule"); + + if(!file.exists()) + return; + + file.open(IO_ReadOnly); + QDataStream stream(&file); + + int tmp; + + //extract category values + for(int i=0; i<3; ++i) + { + stream >> tmp; + m_schedule.setDownload(i, tmp); + stream >> tmp; + m_schedule.setUpload(i, tmp); + } + + //extract schedule + for(int i=0; i<7; ++i) + { + for(int j=0; j<24; ++j) + { + stream >> tmp; + m_schedule.setCategory(i, j, (ScheduleCategory) tmp); + } + } + + file.close(); + } + + void BWScheduler::saveSchedule() + { + QFile file(KGlobal::dirs()->saveLocation("data","ktorrent") + "bwschedule"); + + file.open(IO_WriteOnly); + QDataStream stream(&file); + + for(int i=0; i<3; ++i) + { + stream << m_schedule.getDownload(i); + stream << m_schedule.getUpload(i); + } + + //Now schedule + for(int i=0; i<7; ++i) + for(int j=0; j<24; ++j) + stream << (int) m_schedule.getCategory(i, j); + + file.close(); + } + + void BWScheduler::setEnabled(bool theValue) + { + m_enabled = theValue; + } +} -- cgit v1.2.1