/**
 * @file llviewerthrottle.cpp
 * @brief LLViewerThrottle class implementation
 *
 * $LicenseInfo:firstyear=2002&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2010, Linden Research, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License only.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
 * $/LicenseInfo$
 */

#include "llviewerprecompiledheaders.h"
#include "llviewerthrottle.h"

#include "llviewercontrol.h"
#include "message.h"
#include "llagent.h"
#include "llframetimer.h"
#include "llviewerstats.h"
#include "lldatapacker.h"

using namespace LLOldEvents;

// consts

// The viewer is allowed to set the under-the-hood bandwidth to 50%
// greater than the prefs UI shows, under the assumption that the
// viewer won't receive all the different message types at once.
// I didn't design this, don't know who did. JC
const F32 MAX_FRACTIONAL = 1.5f;
const F32 MIN_FRACTIONAL = 0.2f;

const F32 MIN_BANDWIDTH = 50.f;
const F32 MAX_BANDWIDTH = 6000.f;
const F32 STEP_FRACTIONAL = 0.1f;
const LLUnit<F32, LLUnits::Percent> TIGHTEN_THROTTLE_THRESHOLD(3.0f); // packet loss % per s
const LLUnit<F32, LLUnits::Percent> EASE_THROTTLE_THRESHOLD(0.5f); // packet loss % per s
const F32 DYNAMIC_UPDATE_DURATION = 5.0f; // seconds

LLViewerThrottle gViewerThrottle;

// static
const std:: string LLViewerThrottle::sNames[TC_EOF] = {
                            "Resend",
                            "Land",
                            "Wind",
                            "Cloud",
                            "Task",
                            "Texture",
                            "Asset"
                            };


// Bandwidth settings for different bit rates, they're interpolated/extrapolated.
//                                Resend Land Wind Cloud Task Texture Asset
const F32 BW_PRESET_50[TC_EOF]   = {   5,  10,   3,   3,  10,  10,   9 };
const F32 BW_PRESET_300[TC_EOF]  = {  30,  40,   9,   9,  86,  86,  40 };
const F32 BW_PRESET_500[TC_EOF]  = {  50,  70,  14,  14, 136, 136,  80 };
const F32 BW_PRESET_1000[TC_EOF] = { 100, 100,  20,  20, 310, 310, 140 };

LLViewerThrottleGroup::LLViewerThrottleGroup()
{
    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        mThrottles[i] = 0.f;
    }
    mThrottleTotal = 0.f;
}


LLViewerThrottleGroup::LLViewerThrottleGroup(const F32 settings[TC_EOF])
{
    mThrottleTotal = 0.f;
    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        mThrottles[i] = settings[i];
        mThrottleTotal += settings[i];
    }
}


LLViewerThrottleGroup LLViewerThrottleGroup::operator*(const F32 frac) const
{
    LLViewerThrottleGroup res;
    res.mThrottleTotal = 0.f;

    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        res.mThrottles[i] = mThrottles[i] * frac;
        res.mThrottleTotal += res.mThrottles[i];
    }

    return res;
}


LLViewerThrottleGroup LLViewerThrottleGroup::operator+(const LLViewerThrottleGroup &b) const
{
    LLViewerThrottleGroup res;
    res.mThrottleTotal = 0.f;

    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        res.mThrottles[i] = mThrottles[i] + b.mThrottles[i];
        res.mThrottleTotal += res.mThrottles[i];
    }

    return res;
}


LLViewerThrottleGroup LLViewerThrottleGroup::operator-(const LLViewerThrottleGroup &b) const
{
    LLViewerThrottleGroup res;
    res.mThrottleTotal = 0.f;

    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        res.mThrottles[i] = mThrottles[i] - b.mThrottles[i];
        res.mThrottleTotal += res.mThrottles[i];
    }

    return res;
}


void LLViewerThrottleGroup::sendToSim() const
{
    LL_INFOS() << "Sending throttle settings, total BW " << mThrottleTotal << LL_ENDL;
    LLMessageSystem* msg = gMessageSystem;

    msg->newMessageFast(_PREHASH_AgentThrottle);
    msg->nextBlockFast(_PREHASH_AgentData);
    msg->addUUIDFast(_PREHASH_AgentID, gAgent.getID());
    msg->addUUIDFast(_PREHASH_SessionID, gAgent.getSessionID());
    msg->addU32Fast(_PREHASH_CircuitCode, msg->mOurCircuitCode);

    msg->nextBlockFast(_PREHASH_Throttle);
    msg->addU32Fast(_PREHASH_GenCounter, 0);

    // Pack up the throttle data
    U8 tmp[64];
    LLDataPackerBinaryBuffer dp(tmp, MAX_THROTTLE_SIZE);
    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        //sim wants BPS, not KBPS
        dp.packF32(mThrottles[i] * 1024.0f, "Throttle");
    }
    S32 len = dp.getCurrentSize();
    msg->addBinaryDataFast(_PREHASH_Throttles, tmp, len);

    gAgent.sendReliableMessage();
}


void LLViewerThrottleGroup::dump()
{
    S32 i;
    for (i = 0; i < TC_EOF; i++)
    {
        LL_DEBUGS("Throttle") << LLViewerThrottle::sNames[i] << ": " << mThrottles[i] << LL_ENDL;
    }
    LL_DEBUGS("Throttle") << "Total: " << mThrottleTotal << LL_ENDL;
}

class LLBPSListener : public LLSimpleListener
{
public:
    virtual bool handleEvent(LLPointer<LLEvent> event, const LLSD& userdata)
    {
        gViewerThrottle.setMaxBandwidth((F32) event->getValue().asReal()*1024);
        return true;
    }
};

LLViewerThrottle::LLViewerThrottle() :
    mMaxBandwidth(0.f),
    mCurrentBandwidth(0.f),
    mThrottleFrac(1.f)
{
    // Need to be pushed on in bandwidth order
    mPresets.push_back(LLViewerThrottleGroup(BW_PRESET_50));
    mPresets.push_back(LLViewerThrottleGroup(BW_PRESET_300));
    mPresets.push_back(LLViewerThrottleGroup(BW_PRESET_500));
    mPresets.push_back(LLViewerThrottleGroup(BW_PRESET_1000));
}


void LLViewerThrottle::setMaxBandwidth(F32 kbits_per_second, BOOL from_event)
{
    if (!from_event)
    {
        gSavedSettings.setF32("ThrottleBandwidthKBPS", kbits_per_second);
    }
    gViewerThrottle.load();

    if (gAgent.getRegion())
    {
        gViewerThrottle.sendToSim();
    }
}

void LLViewerThrottle::load()
{
    mMaxBandwidth = gSavedSettings.getF32("ThrottleBandwidthKBPS")*1024;
    resetDynamicThrottle();
    mCurrent.dump();
}


void LLViewerThrottle::save() const
{
    gSavedSettings.setF32("ThrottleBandwidthKBPS", mMaxBandwidth/1024);
}


void LLViewerThrottle::sendToSim() const
{
    mCurrent.sendToSim();
}


LLViewerThrottleGroup LLViewerThrottle::getThrottleGroup(const F32 bandwidth_kbps)
{
    //Clamp the bandwidth users can set.
    F32 set_bandwidth = llclamp(bandwidth_kbps, MIN_BANDWIDTH, MAX_BANDWIDTH);

    S32 count = mPresets.size();

    S32 i;
    for (i = 0; i < count; i++)
    {
        if (mPresets[i].getTotal() > set_bandwidth)
        {
            break;
        }
    }

    if (i == 0)
    {
        // We return the minimum if it's less than the minimum
        return mPresets[0];
    }
    else if (i == count)
    {
        // Higher than the highest preset, we extrapolate out based on the
        // last two presets.  This allows us to keep certain throttle channels from
        // growing in bandwidth
        F32 delta_bw = set_bandwidth - mPresets[count-1].getTotal();
        LLViewerThrottleGroup delta_throttle = mPresets[count - 1] - mPresets[count - 2];
        F32 delta_total = delta_throttle.getTotal();
        F32 delta_frac = delta_bw / delta_total;
        delta_throttle = delta_throttle * delta_frac;
        return mPresets[count-1] + delta_throttle;
    }
    else
    {
        // In between two presets, just interpolate
        F32 delta_bw = set_bandwidth - mPresets[i - 1].getTotal();
        LLViewerThrottleGroup delta_throttle = mPresets[i] - mPresets[i - 1];
        F32 delta_total = delta_throttle.getTotal();
        F32 delta_frac = delta_bw / delta_total;
        delta_throttle = delta_throttle * delta_frac;
        return mPresets[i - 1] + delta_throttle;
    }
}


// static
void LLViewerThrottle::resetDynamicThrottle()
{
    mThrottleFrac = MAX_FRACTIONAL;

    mCurrentBandwidth = mMaxBandwidth*MAX_FRACTIONAL;
    mCurrent = getThrottleGroup(mCurrentBandwidth / 1024.0f);
}

void LLViewerThrottle::updateDynamicThrottle()
{
    if (mUpdateTimer.getElapsedTimeF32() < DYNAMIC_UPDATE_DURATION)
    {
        return;
    }
    mUpdateTimer.reset();

    LLUnit<F32, LLUnits::Percent> mean_packets_lost = LLViewerStats::instance().getRecording().getMean(LLStatViewer::PACKETS_LOST_PERCENT);
    if (mean_packets_lost > TIGHTEN_THROTTLE_THRESHOLD)
    {
        if (mThrottleFrac <= MIN_FRACTIONAL || mCurrentBandwidth / 1024.0f <= MIN_BANDWIDTH)
        {
            return;
        }
        mThrottleFrac -= STEP_FRACTIONAL;
        mThrottleFrac = llmax(MIN_FRACTIONAL, mThrottleFrac);
        mCurrentBandwidth = mMaxBandwidth * mThrottleFrac;
        mCurrent = getThrottleGroup(mCurrentBandwidth / 1024.0f);
        mCurrent.sendToSim();
        LL_INFOS() << "Tightening network throttle to " << mCurrentBandwidth << LL_ENDL;
    }
    else if (mean_packets_lost <= EASE_THROTTLE_THRESHOLD)
    {
        if (mThrottleFrac >= MAX_FRACTIONAL || mCurrentBandwidth / 1024.0f >= MAX_BANDWIDTH)
        {
            return;
        }
        mThrottleFrac += STEP_FRACTIONAL;
        mThrottleFrac = llmin(MAX_FRACTIONAL, mThrottleFrac);
        mCurrentBandwidth = mMaxBandwidth * mThrottleFrac;
        mCurrent = getThrottleGroup(mCurrentBandwidth/1024.0f);
        mCurrent.sendToSim();
        LL_INFOS() << "Easing network throttle to " << mCurrentBandwidth << LL_ENDL;
    }
}