#include "PeriodicHousekeepingHelper.h"

#include "../datapoollocal/LocalPoolDataSetBase.h"
#include <cmath>

PeriodicHousekeepingHelper::PeriodicHousekeepingHelper(
        LocalPoolDataSetBase* owner): owner(owner) {}


void PeriodicHousekeepingHelper::initialize(float collectionInterval,
        dur_millis_t minimumPeriodicInterval, bool isDiagnostics,
        uint8_t nonDiagIntervalFactor) {
    this->isDiagnostics = isDiagnostics;
    this->minimumPeriodicInterval = minimumPeriodicInterval;
    this->nonDiagIntervalFactor = nonDiagIntervalFactor;
    collectionIntervalTicks = intervalSecondsToIntervalTicks(collectionInterval);
    /* This will cause a checkOpNecessary call to be true immediately. I think it's okay
	if a HK packet is generated immediately instead of waiting one generation cycle. */
    internalTickCounter = collectionIntervalTicks;
}

float PeriodicHousekeepingHelper::getCollectionIntervalInSeconds() const {
    return intervalTicksToSeconds(collectionIntervalTicks);
}

bool PeriodicHousekeepingHelper::checkOpNecessary() {
    if(internalTickCounter >= collectionIntervalTicks) {
        internalTickCounter = 1;
        return true;
    }
    internalTickCounter++;
    return false;
}

uint32_t PeriodicHousekeepingHelper::intervalSecondsToIntervalTicks(
        float collectionIntervalSeconds) {
    /* Avoid division by zero */
    if(minimumPeriodicInterval == 0) {
        if(isDiagnostics) {
            /* Perform operation each cycle */
            return 1;
        }
        else {
            return nonDiagIntervalFactor;
        }
    }
    else {
        dur_millis_t intervalInMs = collectionIntervalSeconds * 1000;
        uint32_t divisor = minimumPeriodicInterval;
        if(not isDiagnostics) {
            /* We need to multiply the divisor because non-diagnostics only
            allow a multiple of the minimum periodic interval */
            divisor *= nonDiagIntervalFactor;
        }
        uint32_t ticks = std::ceil(static_cast<float>(intervalInMs) / divisor);
        if(not isDiagnostics) {
            /* Now we need to multiply the calculated ticks with the factor as as well
            because the minimum tick count to generate a non-diagnostic is
            the factor.

            Example calculation for non-diagnostic with
            0.4 second interval and 0.2 second task interval.
            Resultant tick count of 5 is equal to operation each second.

            Examle calculation for non-diagnostic with  2.0 second interval and 0.2 second
            task interval.
            Resultant tick count of 10 is equal to operatin every 2 seconds.

            Example calculation for diagnostic with 0.4 second interval and 0.3
            second task interval. Resulting tick count of 2 is equal to operation
            every 0.6 seconds. */
            ticks *= nonDiagIntervalFactor;
        }
        return ticks;
    }
}

float PeriodicHousekeepingHelper::intervalTicksToSeconds(
        uint32_t collectionInterval) const {
    /* Number of ticks times the minimum interval is in milliseconds, so we divide by 1000 to get
    the value in seconds */
    return static_cast<float>(collectionInterval * minimumPeriodicInterval / 1000.0);
}

void PeriodicHousekeepingHelper::changeCollectionInterval(
        float newIntervalSeconds) {
    collectionIntervalTicks = intervalSecondsToIntervalTicks(newIntervalSeconds);
}