ktorrent/plugins/scheduler/bwscheduler.cpp

/***************************************************************************
 *   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::tqcurrentDateTime();

		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;
	}
}