#include <math.h>

#include "MT_Statistics.h"
#include "audio/Audio_Interface.h"
#include "helper/HL_Log.h"
#define LOG_SUBSYSTEM "Statistics"

using namespace MT;

void JitterStatistics::process(jrtplib::RTPPacket* packet, int rate)
{
    uint32_t      arrival = 0;
    int           d = 0;
    uint32_t      transit = 0;

    uint64_t    current_time = TimeHelper::getTimestamp();
    int         interarrival_time_ms = current_time - mPrevRxTimestamp;

    // get the 'arrival time' of this packet as measured in 'timestamp units' and offset
    //  to match the timestamp range in this stream
    arrival = interarrival_time_ms * (rate / 1000);

    if (mPrevArrival == 0)
        arrival = packet->GetTimestamp();
    else
        arrival += mPrevArrival;;

    mPrevArrival = packet->GetTimestamp();

    transit = arrival - packet->GetTimestamp();
    jrtplib::RTPTime receiveTime = packet->GetReceiveTime();

    d = transit - mPrevTransit;
    mPrevTransit = transit;
    if (d < 0)
        d = -d;
    mJitter += (1.0/16.0) * ((double)d - mJitter);

    mPrevRxTimestamp = current_time;
    if (mMaxJitter < mJitter)
        mMaxJitter = mJitter;


    uint32_t timestamp = packet->GetTimestamp();

    if (!mLastJitter.is_initialized())
    {
        mReceiveTime = receiveTime;
        mReceiveTimestamp = timestamp;
        mLastJitter = 0.0;
    }
    else
    {
        double receiveDelta = receiveTime.GetDouble() - mReceiveTime.GetDouble();
        double timestampDelta = double(timestamp - mReceiveTimestamp);

        if (timestampDelta == 0.0)
            // Skip current packet silently. Most probably it is error in RTP stream like duplicated packet.
            return;

        double delta = receiveDelta / timestampDelta;
        if (fabs(delta) > mMaxDelta)
            mMaxDelta = fabs(delta);

        mLastJitter = mLastJitter.value() + (fabs(delta) - mLastJitter.value()) / 16.0;
        mReceiveTime = receiveTime;
        mReceiveTimestamp = timestamp;

        mJitter.process(mLastJitter.value());
    }
}


// ---------------------------- Statistics ------------------------------------


Statistics::Statistics()
    :mReceived(0), mSent(0), mReceivedRtp(0), mSentRtp(0),
      mReceivedRtcp(0), mSentRtcp(0), mDuplicatedRtp(0), mOldRtp(0), mIllegalRtp(0),
      mPacketLoss(0), mJitter(0.0), mAudioTime(0), mSsrc(0)
{
#if defined(USE_AMR_CODEC)
    mBitrateSwitchCounter = 0;
#endif

    memset(mLoss, 0, sizeof mLoss);

    // It is to keep track of statistics instance via grep | wc -l
    //ICELogDebug(<< "Create statistics instance.");
}

Statistics::~Statistics()
{
}

void Statistics::reset()
{
    mReceived = 0;
    mSent = 0;
    mReceivedRtp = 0;
    mSentRtp = 0;
    mReceivedRtcp = 0;
    mSentRtcp = 0;
    mDuplicatedRtp = 0;
    mOldRtp = 0;
    mPacketLoss = 0;
    mIllegalRtp = 0;
    mJitter = 0.0;
    mAudioTime = 0;
    memset(mLoss, 0, sizeof mLoss);
}

/*
double calculate_mos_g711(double ppl, double burstr, int version) {
double r;
double bpl = 8.47627; //mos = -4.23836 + 0.29873 * r - 0.00416744 * r * r + 0.0000209855 * r * r * r;
double mos;

if(ppl == 0 or burstr == 0) {
return 4.5;
}

if(ppl > 0.5) {
return 1;
}

switch(version) {
case 1:
case 2:
default:
// this mos is calculated for G.711 and PLC
bpl = 17.2647;
r = 93.2062077233 - 95.0 * (ppl*100/(ppl*100/burstr + bpl));
mos = 2.06405 + 0.031738 * r - 0.000356641 * r * r + 2.93143 * pow(10,-6) * r * r * r;
if(mos < 1)
return 1;
if(mos > 4.5)
return 4.5;
}

return mos;
}


double calculate_mos(double ppl, double burstr, int codec, unsigned int received) {
if(codec == PAYLOAD_G729) {
if(opt_mos_g729) {
if(received < 100) {
return 3.92;
}
return (double)mos_g729((long double)ppl, (long double)burstr);
} else {
if(received < 100) {
return 4.5;
}
return calculate_mos_g711(ppl, burstr, 2);
}
} else {
if(received < 100) {
return 4.5;
}
return calculate_mos_g711(ppl, burstr, 2);
}
}

*/

void Statistics::calculateBurstr(double* burstr, double* lossr) const
{
    int lost = 0;
    int bursts = 0;
    for (int i = 0; i < 128; i++)
    {
        lost += i * mLoss[i];
        bursts += mLoss[i];
    }

    if (lost < 5)
    {
        // ignore such small packet loss
        *lossr = *burstr = 0;
        return;
    }

    if (mReceivedRtp > 0 && bursts > 0)
    {
        *burstr = (double)((double)lost / (double)bursts) / (double)(1.0 / (1.0 - (double)lost / (double)mReceivedRtp));
        if (*burstr < 0)
            *burstr = -*burstr;
        else if (*burstr < 1)
            *burstr = 1;
    }
    else
        *burstr = 0;
    //printf("total loss: %d\n", lost);
    if (mReceivedRtp > 0)
        *lossr = (double)((double)lost / (double)mReceivedRtp);
    else
        *lossr = 0;
}

double Statistics::calculateMos(double maximalMos) const
{
    // calculate lossrate and burst rate
    double burstr, lossr;
    calculateBurstr(&burstr, &lossr);

    double r;
    double bpl = 8.47627; //mos = -4.23836 + 0.29873 * r - 0.00416744 * r * r + 0.0000209855 * r * r * r;
    double mos;

    if (mReceivedRtp < 100)
        return 0.0;

    if (lossr == 0 || burstr == 0)
    {
        return maximalMos;
    }

    if (lossr > 0.5)
        return 1;

    bpl = 17.2647;
    r = 93.2062077233 - 95.0 * (lossr * 100 / (lossr * 100 / burstr + bpl));
    mos = 2.06405 + 0.031738 * r - 0.000356641 * r * r + 2.93143 * pow(10, -6) * r * r * r;
    if (mos < 1)
        return 1;

    if (mos > maximalMos)
        return maximalMos;

    return mos;
}

Statistics& Statistics::operator += (const Statistics& src)
{
    mReceived += src.mReceived;
    mSent += src.mSent;
    mReceivedRtp += src.mReceivedRtp;
    mSentRtp += src.mSentRtp;
    mReceivedRtcp += src.mReceivedRtcp;
    mSentRtcp += src.mSentRtcp;
    mDuplicatedRtp += src.mDuplicatedRtp;
    mOldRtp += src.mOldRtp;
    mPacketLoss += src.mPacketLoss;
    mAudioTime += src.mAudioTime;

    for (auto codecStat: src.mCodecCount)
    {
        if (mCodecCount.find(codecStat.first) == mCodecCount.end())
            mCodecCount[codecStat.first] = codecStat.second;
        else
            mCodecCount[codecStat.first] += codecStat.second;
    }

    mJitter = src.mJitter;
    mRttDelay = src.mRttDelay;
    if (!src.mCodecName.empty())
        mCodecName = src.mCodecName;

    // Find minimal
    if (mFirstRtpTime.is_initialized())
    {
        if (src.mFirstRtpTime.is_initialized())
        {
            if (mFirstRtpTime.value() > src.mFirstRtpTime.value())
                mFirstRtpTime = src.mFirstRtpTime;
        }
    }
    else
        if (src.mFirstRtpTime.is_initialized())
            mFirstRtpTime = src.mFirstRtpTime;

#if defined(USE_AMR_CODEC)
    mBitrateSwitchCounter += src.mBitrateSwitchCounter;
#endif
    mRemotePeer = src.mRemotePeer;
    mSsrc = src.mSsrc;

    return *this;
}

Statistics& Statistics::operator -= (const Statistics& src)
{
    mReceived -= src.mReceived;
    mSent -= src.mSent;
    mReceivedRtp -= src.mReceivedRtp;
    mIllegalRtp -= src.mIllegalRtp;
    mSentRtp -= src.mSentRtp;
    mReceivedRtcp -= src.mReceivedRtcp;
    mSentRtcp -= src.mSentRtcp;
    mDuplicatedRtp -= src.mDuplicatedRtp;
    mOldRtp -= src.mOldRtp;
    mPacketLoss -= src.mPacketLoss;
    mAudioTime -= src.mAudioTime;
    for (auto codecStat: src.mCodecCount)
    {
        if (mCodecCount.find(codecStat.first) != mCodecCount.end())
            mCodecCount[codecStat.first] -= codecStat.second;
    }

    return *this;
}


std::string Statistics::toShortString() const
{
    std::ostringstream oss;
    oss << "Received: " << mReceivedRtp
        << ", lost: " << mPacketLoss
        << ", sent: " << mSentRtp;

    return oss.str();
}