blob: 0caf005a9a7a394ff7e3b37a728c266d87a4bb51 [file] [log] [blame]
/*
* Copyright (C) 2021 The Android Open Source Project
* Android BPF library - public API
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
#include <inttypes.h>
#include <log/log.h>
#include <time.h>
#include <sstream>
#include <string>
/**
* An object that can track changes of some value over time, taking into account an additional
* dimension: the object's state. As the tracked value changes, the deltas are distributed
* among the object states in accordance with the time spent in those states.
*/
namespace android {
namespace battery {
typedef uint16_t state_t;
template <class T>
class MultiStateCounter {
uint16_t stateCount;
state_t currentState;
time_t lastStateChangeTimestamp;
T emptyValue;
T lastValue;
time_t lastUpdateTimestamp;
T deltaValue;
bool isEnabled;
struct State {
time_t timeInStateSinceUpdate;
T counter;
};
State* states;
public:
MultiStateCounter(uint16_t stateCount, const T& emptyValue);
virtual ~MultiStateCounter();
void setEnabled(bool enabled, time_t timestamp);
void setState(state_t state, time_t timestamp);
void setValue(state_t state, const T& value);
/**
* Updates the value by distributing the delta from the previously set value
* among states according to their respective time-in-state.
* Returns the delta from the previously set value.
*/
const T& updateValue(const T& value, time_t timestamp);
/**
* Updates the value by distributing the specified increment among states according
* to their respective time-in-state.
*/
void incrementValue(const T& increment, time_t timestamp);
/**
* Adds the specified increment to the value for the current state, without affecting
* the last updated value or timestamp. Ignores partial time-in-state: the entirety of
* the increment is given to the current state.
*/
void addValue(const T& increment);
void reset();
uint16_t getStateCount();
const T& getCount(state_t state);
std::string toString();
private:
/**
* Subtracts previousValue from newValue and returns the result in outValue.
* Returns true iff the combination of previousValue and newValue is valid
* (newValue >= prevValue)
*/
bool delta(const T& previousValue, const T& newValue, T* outValue) const;
/**
* Adds value2 to value1 and stores the result in value1. Denominator is
* guaranteed to be non-zero.
*/
void add(T* value1, const T& value2, const uint64_t numerator,
const uint64_t denominator) const;
std::string valueToString(const T& value) const;
};
// ---------------------- MultiStateCounter Implementation -------------------------
// Since MultiStateCounter is a template, the implementation must be inlined.
template <class T>
MultiStateCounter<T>::MultiStateCounter(uint16_t stateCount, const T& emptyValue)
: stateCount(stateCount),
currentState(0),
lastStateChangeTimestamp(-1),
emptyValue(emptyValue),
lastValue(emptyValue),
lastUpdateTimestamp(-1),
deltaValue(emptyValue),
isEnabled(true) {
states = new State[stateCount];
for (int i = 0; i < stateCount; i++) {
states[i].timeInStateSinceUpdate = 0;
states[i].counter = emptyValue;
}
}
template <class T>
MultiStateCounter<T>::~MultiStateCounter() {
delete[] states;
};
template <class T>
void MultiStateCounter<T>::setEnabled(bool enabled, time_t timestamp) {
if (enabled == isEnabled) {
return;
}
if (!enabled) {
// Confirm the current state for the side-effect of updating the time-in-state
// counter for the current state.
setState(currentState, timestamp);
}
isEnabled = enabled;
if (lastStateChangeTimestamp >= 0) {
lastStateChangeTimestamp = timestamp;
}
}
template <class T>
void MultiStateCounter<T>::setState(state_t state, time_t timestamp) {
if (isEnabled && lastStateChangeTimestamp >= 0) {
if (timestamp >= lastStateChangeTimestamp) {
states[currentState].timeInStateSinceUpdate += timestamp - lastStateChangeTimestamp;
} else {
ALOGE("setState is called with an earlier timestamp: %lu, previous timestamp: %lu\n",
(unsigned long)timestamp, (unsigned long)lastStateChangeTimestamp);
// The accumulated durations have become unreliable. For example, if the timestamp
// sequence was 1000, 2000, 1000, 3000, if we accumulated the positive deltas,
// we would get 4000, which is greater than (last - first). This could lead to
// counts exceeding 100%.
for (int i = 0; i < stateCount; i++) {
states[i].timeInStateSinceUpdate = 0;
}
}
}
currentState = state;
lastStateChangeTimestamp = timestamp;
}
template <class T>
void MultiStateCounter<T>::setValue(state_t state, const T& value) {
states[state].counter = value;
}
template <class T>
const T& MultiStateCounter<T>::updateValue(const T& value, time_t timestamp) {
T* returnValue = &emptyValue;
// If the counter is disabled, we ignore the update, except when the counter got disabled after
// the previous update, in which case we still need to pick up the residual delta.
if (isEnabled || lastUpdateTimestamp < lastStateChangeTimestamp) {
// Confirm the current state for the side-effect of updating the time-in-state
// counter for the current state.
setState(currentState, timestamp);
if (lastUpdateTimestamp >= 0) {
if (timestamp > lastUpdateTimestamp) {
if (delta(lastValue, value, &deltaValue)) {
returnValue = &deltaValue;
time_t timeSinceUpdate = timestamp - lastUpdateTimestamp;
for (int i = 0; i < stateCount; i++) {
time_t timeInState = states[i].timeInStateSinceUpdate;
if (timeInState) {
add(&states[i].counter, deltaValue, timeInState, timeSinceUpdate);
states[i].timeInStateSinceUpdate = 0;
}
}
} else {
std::stringstream str;
str << "updateValue is called with a value " << valueToString(value)
<< ", which is lower than the previous value " << valueToString(lastValue)
<< "\n";
ALOGE("%s", str.str().c_str());
for (int i = 0; i < stateCount; i++) {
states[i].timeInStateSinceUpdate = 0;
}
}
} else if (timestamp < lastUpdateTimestamp) {
ALOGE("updateValue is called with an earlier timestamp: %lu, previous: %lu\n",
(unsigned long)timestamp, (unsigned long)lastUpdateTimestamp);
for (int i = 0; i < stateCount; i++) {
states[i].timeInStateSinceUpdate = 0;
}
}
}
}
lastValue = value;
lastUpdateTimestamp = timestamp;
return *returnValue;
}
template <class T>
void MultiStateCounter<T>::incrementValue(const T& increment, time_t timestamp) {
T newValue = lastValue;
add(&newValue, increment, 1 /* numerator */, 1 /* denominator */);
updateValue(newValue, timestamp);
}
template <class T>
void MultiStateCounter<T>::addValue(const T& value) {
if (!isEnabled) {
return;
}
add(&states[currentState].counter, value, 1 /* numerator */, 1 /* denominator */);
}
template <class T>
void MultiStateCounter<T>::reset() {
lastStateChangeTimestamp = -1;
lastUpdateTimestamp = -1;
for (int i = 0; i < stateCount; i++) {
states[i].timeInStateSinceUpdate = 0;
states[i].counter = emptyValue;
}
}
template <class T>
uint16_t MultiStateCounter<T>::getStateCount() {
return stateCount;
}
template <class T>
const T& MultiStateCounter<T>::getCount(state_t state) {
return states[state].counter;
}
template <class T>
std::string MultiStateCounter<T>::toString() {
std::stringstream str;
str << "[";
for (int i = 0; i < stateCount; i++) {
if (i != 0) {
str << ", ";
}
str << i << ": " << valueToString(states[i].counter);
if (states[i].timeInStateSinceUpdate > 0) {
str << " timeInStateSinceUpdate: " << states[i].timeInStateSinceUpdate;
}
}
str << "]";
if (lastUpdateTimestamp >= 0) {
str << " updated: " << lastUpdateTimestamp;
}
if (lastStateChangeTimestamp >= 0) {
str << " currentState: " << currentState;
if (lastStateChangeTimestamp > lastUpdateTimestamp) {
str << " stateChanged: " << lastStateChangeTimestamp;
}
} else {
str << " currentState: none";
}
if (!isEnabled) {
str << " disabled";
}
return str.str();
}
} // namespace battery
} // namespace android