Sirikata
|
00001 /* Sirikata 00002 * RateEstimator.hpp 00003 * 00004 * Copyright (c) 2010, Ewen Cheslack-Postava 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions are 00009 * met: 00010 * * Redistributions of source code must retain the above copyright 00011 * notice, this list of conditions and the following disclaimer. 00012 * * Redistributions in binary form must reproduce the above copyright 00013 * notice, this list of conditions and the following disclaimer in 00014 * the documentation and/or other materials provided with the 00015 * distribution. 00016 * * Neither the name of Sirikata nor the names of its contributors may 00017 * be used to endorse or promote products derived from this software 00018 * without specific prior written permission. 00019 * 00020 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 00021 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 00022 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 00023 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 00024 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 00025 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 00026 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00027 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00028 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00029 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00030 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00031 */ 00032 00033 #ifndef _SIRIKATA_RATE_ESTIMATOR_HPP_ 00034 #define _SIRIKATA_RATE_ESTIMATOR_HPP_ 00035 00036 #include <sirikata/core/util/Platform.hpp> 00037 00038 namespace Sirikata { 00039 00041 class RateEstimator { 00042 public: 00043 RateEstimator() 00044 : _value(0), 00045 _t(Time::null()), 00046 _backlog(0) 00047 {} 00048 00049 RateEstimator(double good_guess, const Time& start) 00050 : _value(good_guess), 00051 _t(start), 00052 _backlog(0) 00053 {} 00054 00055 double get()const { 00056 return _value; 00057 } 00058 00059 double get(const Time& t, double K) const { 00060 Duration diff = t - _t; 00061 double dt = diff.toSeconds(); 00062 if (dt<1.0e-9) { 00063 return _value; 00064 } 00065 double blend = exp(-dt/K); 00066 uint32 new_bytes = _backlog; 00067 return _value*blend+(1-blend)*new_bytes/dt; 00068 } 00069 00070 double estimate_rate(const Time& t, uint32 len, double K) { 00071 Duration diff = t - _t; 00072 double dt = diff.toSeconds(); 00073 if (dt<1.0e-9) { 00074 _backlog += len; 00075 return _value; 00076 } 00077 double blend = exp(-dt/K); 00078 uint32 new_bytes = len + _backlog; 00079 _value=_value*blend+(1-blend)*new_bytes/dt; 00080 _t = t; 00081 _backlog = 0; 00082 return _value; 00083 } 00084 private: 00085 double _value; 00086 Time _t; 00087 uint32 _backlog; 00088 }; 00089 00094 class SimpleRateEstimator : public RateEstimator { 00095 public: 00096 SimpleRateEstimator(double K) 00097 : RateEstimator(), 00098 _K(K) 00099 {} 00100 00101 SimpleRateEstimator(double good_guess, const Time& start, double K) 00102 : RateEstimator(good_guess, start), 00103 _K(K) 00104 {} 00105 00106 double estimate_rate(const Time& t, uint32 len) { 00107 return RateEstimator::estimate_rate(t, len, _K); 00108 } 00109 private: 00110 double _K; 00111 }; 00112 00113 } // namespace Sirikata 00114 00115 #endif //_SIRIKATA_RATE_ESTIMATOR_HPP_