You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
283 lines
7.1 KiB
283 lines
7.1 KiB
/***************************************************************************
|
|
* 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 <torrent/globals.h>
|
|
#include <torrent/queuemanager.h>
|
|
|
|
#include <interfaces/coreinterface.h>
|
|
#include <interfaces/torrentinterface.h>
|
|
|
|
#include <net/socketmonitor.h>
|
|
|
|
#include <util/constants.h>
|
|
#include <util/log.h>
|
|
|
|
#include <tqdatetime.h>
|
|
#include <tqfile.h>
|
|
#include <tqptrlist.h>
|
|
|
|
#include <kglobal.h>
|
|
#include <kstandarddirs.h>
|
|
|
|
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;
|
|
|
|
TQDateTime now = TQDateTime::currentDateTime();
|
|
|
|
TQString prefix = TQString("BWS: %1 :: ").tqarg(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 << TQString("%1 Up, %2 Down")
|
|
.tqarg(m_core->getMaxUploadSpeed()).tqarg(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 << TQString("%1 Up, %2 Down")
|
|
.tqarg(m_schedule.getUpload(0)).tqarg(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 << TQString("%1 Up, %2 Down")
|
|
.tqarg(m_schedule.getUpload(1)).tqarg(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 << TQString("%1 Up, %2 Down")
|
|
.tqarg(m_schedule.getUpload(2)).tqarg(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()
|
|
{
|
|
TQFile file(KGlobal::dirs()->saveLocation("data","ktorrent") + "bwschedule");
|
|
|
|
if(!file.exists())
|
|
return;
|
|
|
|
file.open(IO_ReadOnly);
|
|
TQDataStream 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()
|
|
{
|
|
TQFile file(KGlobal::dirs()->saveLocation("data","ktorrent") + "bwschedule");
|
|
|
|
file.open(IO_WriteOnly);
|
|
TQDataStream 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;
|
|
}
|
|
}
|