neobytes/src/qt/trafficgraphdata.cpp

#include "trafficgraphdata.h"

const int TrafficGraphData::RangeMinutes[] = {5,10,15,30,60,120,180,360,720,1440};
const int TrafficGraphData::DESIRED_DATA_SAMPLES = 800;
const int TrafficGraphData::DesiredQueueSizes[] = {
        TrafficGraphData::DESIRED_DATA_SAMPLES,     //Range_5m
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_10m
        TrafficGraphData::DESIRED_DATA_SAMPLES*2/3, //Range_15m
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_30m
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_1h
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_2h
        TrafficGraphData::DESIRED_DATA_SAMPLES*2/3, //Range_3h
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_6h
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_12h
        TrafficGraphData::DESIRED_DATA_SAMPLES/2,   //Range_24h
    } ;

const int TrafficGraphData::SMALLEST_SAMPLE_PERIOD =
        TrafficGraphData::RangeMinutes[TrafficGraphData::Range_5m] * 60 * 1000 / TrafficGraphData::DESIRED_DATA_SAMPLES;

TrafficGraphData::TrafficGraphData(GraphRange range)
    :currentGraphRange(range),
    currentSampleCounter(0),
    nLastBytesIn(0),
    nLastBytesOut(0)
{
}

void TrafficGraphData::tryAddingSampleToStash(GraphRange range)
{
    SampleQueue& queue = sampleMap[range];
    if (queue.size() > DesiredQueueSizes[range]){
           sampleStash[range].push_front(queue.at(DesiredQueueSizes[range]));
    }
}

void TrafficGraphData::tryUpdateNextWithLast2Samples(GraphRange range, GraphRange nextRange)
{
    SampleQueue& queue = sampleMap[range];
    if (queue.size() == DesiredQueueSizes[range] + 2){
        update(nextRange, queue.takeLast() + queue.takeLast());
    }
}

void TrafficGraphData::tryUpdateNextWithLast3Samples(GraphRange range, GraphRange nextRange)
{
    SampleQueue& stashQueue = sampleStash[range];
    if (stashQueue.size() == 3){
        update(nextRange, stashQueue.takeLast() + stashQueue.takeLast() + stashQueue.takeLast());
    }
}

void TrafficGraphData::setLastBytes(quint64 nLastBytesIn, quint64 nLastBytesOut)
{
    this->nLastBytesIn = nLastBytesIn;
    this->nLastBytesOut = nLastBytesOut;
}

bool TrafficGraphData::update(quint64 totalBytesRecv, quint64 totalBytesSent)
{
    float inRate = totalBytesRecv - nLastBytesIn ;
    float outRate = totalBytesSent - nLastBytesOut;
    nLastBytesIn = totalBytesRecv;
    nLastBytesOut = totalBytesSent;
    return update(TrafficSample(inRate,outRate));
}

bool TrafficGraphData::update(const TrafficSample& trafficSample)
{
    update(Range_5m, trafficSample);

    currentSampleCounter++;

    if (RangeMinutes[currentGraphRange]/RangeMinutes[Range_5m] == currentSampleCounter){
        currentSampleCounter = 0;
        return true;
    }
    return false;
}

void TrafficGraphData::update(GraphRange range, const TrafficSample& trafficSample)
{
    SampleQueue& queue = sampleMap[range];
    queue.push_front(trafficSample);

    switch(range){
        case Range_5m:
                tryAddingSampleToStash(Range_5m);
                tryUpdateNextWithLast2Samples(Range_5m, Range_10m);
                tryUpdateNextWithLast3Samples(Range_5m, Range_15m);
                return;
        case Range_15m:
                tryUpdateNextWithLast2Samples(Range_15m, Range_30m);
                return;
        case Range_30m:
                tryUpdateNextWithLast2Samples(Range_30m, Range_1h);
                return;
        case Range_1h:
                tryAddingSampleToStash(Range_1h);
                tryUpdateNextWithLast2Samples(Range_1h, Range_2h);
                tryUpdateNextWithLast3Samples(Range_1h, Range_3h);
                return;
        case Range_3h:
                tryUpdateNextWithLast2Samples(Range_3h, Range_6h);
                return;
        case Range_6h:
                tryUpdateNextWithLast2Samples( Range_6h, Range_12h);
                return;
        case Range_12h:
                tryUpdateNextWithLast2Samples(Range_12h, Range_24h);
                return;
        default:
            if (queue.size() > DesiredQueueSizes[range])
                queue.removeLast();
            return;
    }
}


void TrafficGraphData::switchRange(GraphRange newRange)
{
    currentGraphRange = newRange;
    currentSampleCounter = 0;
}

TrafficGraphData::SampleQueue TrafficGraphData::sumEach2Samples(const SampleQueue& rangeQueue)
{
    SampleQueue result;
    int i = rangeQueue.size() - 1;

    while (i - 1 >= 0){
        result.push_front(rangeQueue.at(i)+ rangeQueue.at(i-1));
        i -= 2;
    }
    return result;
}

TrafficGraphData::SampleQueue TrafficGraphData::sumEach3Samples(const SampleQueue& rangeQueue, GraphRange range)
{
    SampleQueue result;
    int lastUnusedSample = std::min(rangeQueue.size()-1 ,DESIRED_DATA_SAMPLES-1);

    // use stash first
    SampleQueue& stashQueue = sampleStash[range];
    TrafficSample sum(0,0);
    for (int i = 0; i < stashQueue.size(); i++){
       sum+=stashQueue.at(i);
    }
    int toFullSample = 3 - stashQueue.size();
    if (toFullSample > rangeQueue.size())
        return result;

    // append to stash data to create whole sample
    for (int i = 0; i < toFullSample; i++){
        sum+=rangeQueue.at(lastUnusedSample);
        lastUnusedSample--;
    }
    result.push_front(sum);

    while (lastUnusedSample - 2 >= 0){
        result.push_front(rangeQueue.at(lastUnusedSample) + rangeQueue.at(lastUnusedSample-1) + rangeQueue.at(lastUnusedSample-2));
        lastUnusedSample -= 3;
    }
    return result;
}

TrafficGraphData::SampleQueue TrafficGraphData::getRangeQueue(GraphRange range)
{
    switch(range){
        case Range_5m:
            return sampleMap[Range_5m];
        case Range_10m:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_5m));
                queue.append(sampleMap[Range_10m]);
                return queue;
            }
        case Range_15m:
            {
                SampleQueue queue = sumEach3Samples(getRangeQueue(Range_5m), Range_5m);
                queue.append(sampleMap[Range_15m]);
                return queue;
            }
        case Range_30m:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_15m));
                queue.append(sampleMap[Range_30m]);
                return queue;
            }
        case Range_1h:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_30m));
                queue.append(sampleMap[Range_1h]);
                return queue;
            }
        case Range_2h:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_1h));
                queue.append(sampleMap[Range_2h]);
                return queue;
            }
        case Range_3h:
            {
                SampleQueue queue = sumEach3Samples(getRangeQueue(Range_1h),Range_1h);
                queue.append(sampleMap[Range_3h]);
                return queue;
            }
        case Range_6h:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_3h));
                queue.append(sampleMap[Range_6h]);
                return queue;
            }
        case Range_12h:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_6h));
                queue.append(sampleMap[Range_12h]);
                return queue;
            }
        case Range_24h:
            {
                SampleQueue queue = sumEach2Samples(getRangeQueue(Range_12h));
                queue.append(sampleMap[Range_24h]);
                return queue;
            };
        default:
            return SampleQueue();

    }
}

TrafficGraphData::SampleQueue TrafficGraphData::getCurrentRangeQueue()
{
    SampleQueue newQueue;
    getRangeQueue(currentGraphRange).mid(0,DESIRED_DATA_SAMPLES).swap(newQueue);
    return newQueue;
}


float TrafficGraphData::converSampletoBandwith(float dataAmount){
    // to base range
    float result  = dataAmount / RangeMinutes[currentGraphRange] * RangeMinutes[TrafficGraphData::Range_5m];
    // to B/s
    result = result * 1000 / SMALLEST_SAMPLE_PERIOD;
    // to KB/s
    result = result / 1024;

    return result;

}

TrafficGraphData::SampleQueue TrafficGraphData::getCurrentRangeQueueWithAverageBandwidth(){
    SampleQueue newQueue;
    getRangeQueue(currentGraphRange).mid(0,DESIRED_DATA_SAMPLES).swap(newQueue);
    for(auto& sample : newQueue){
        sample.in = converSampletoBandwith(sample.in);
        sample.out = converSampletoBandwith(sample.out);
    }
    return newQueue;
}

void TrafficGraphData::clear()
{
    sampleMap.clear();
    sampleMap.clear();
    currentSampleCounter = 0;
    nLastBytesIn = 0;
    nLastBytesOut = 0;
}