/*************************************************************************** * Copyright (C) 2005-2006 Gao Xianchao * * 2007 Gao Xianchao gnap_an linux_lyb ahlongxp * * * * 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. * ***************************************************************************/ /* * Author: gxcoo * Create date: 2006-01-06 19:03 */ #ifndef RATEMEASURE_H_ #define RATEMEASURE_H_ #include "BitStorm.h" #include <list> #define MaxPriorityLevel 5 typedef struct { IRateMeasureClient* client; bool remove; }TRateMeasureCtx; typedef std::list<TRateMeasureCtx> TRateMeasureClientList; class CRateMeasure : public IRateMeasure { public: CRateMeasure(); virtual ~CRateMeasure(); private: TRateMeasureClientList _clientList; unsigned int _upSpeed; unsigned int _downSpeed; unsigned int _lastTick; unsigned int _upCount; unsigned int _downCount; bool _writeBlocked; bool _readBlocked; private: void resetAllPriority(); void scheduleUpload(); void scheduleDownload(); void blockReadAll(bool block); void blockWriteAll(bool block); public: void addClient(IRateMeasureClient* client); void removeClient(IRateMeasureClient* client); void update(); void setUploadSpeed(unsigned int sp); void setDownloadSpeed(unsigned int sp); unsigned int getUploadSpeed(); unsigned int getDownloadSpeed(); }; #endif /*RATEMEASURE_H_*/