From 483926a283e118590da3f9ecfa75a8a4d62143ce Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Wed, 1 Dec 2021 07:15:11 +0100
Subject: Merging upstream version 1.32.0.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 ml/BitRateWindow.h | 170 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 170 insertions(+)
 create mode 100644 ml/BitRateWindow.h

(limited to 'ml/BitRateWindow.h')

diff --git a/ml/BitRateWindow.h b/ml/BitRateWindow.h
new file mode 100644
index 000000000..0d99008b8
--- /dev/null
+++ b/ml/BitRateWindow.h
@@ -0,0 +1,170 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef BIT_RATE_WINDOW_H
+#define BIT_RATE_WINDOW_H
+
+#include "BitBufferCounter.h"
+#include "ml-private.h"
+
+namespace ml {
+
+class BitRateWindow {
+public:
+    enum class State {
+        NotFilled,
+        BelowThreshold,
+        AboveThreshold,
+        Idle
+    };
+
+    using Edge = std::pair<State, State>;
+    using Action = size_t (BitRateWindow::*)(State PrevState, bool NewBit);
+
+private:
+    std::map<Edge, Action> EdgeActions = {
+        // From == To
+        {
+            Edge(State::NotFilled, State::NotFilled),
+            &BitRateWindow::onRoundtripNotFilled,
+        },
+        {
+            Edge(State::BelowThreshold, State::BelowThreshold),
+            &BitRateWindow::onRoundtripBelowThreshold,
+        },
+        {
+            Edge(State::AboveThreshold, State::AboveThreshold),
+            &BitRateWindow::onRoundtripAboveThreshold,
+        },
+        {
+            Edge(State::Idle, State::Idle),
+            &BitRateWindow::onRoundtripIdle,
+        },
+
+
+        // NotFilled => {BelowThreshold, AboveThreshold}
+        {
+            Edge(State::NotFilled, State::BelowThreshold),
+            &BitRateWindow::onNotFilledToBelowThreshold
+        },
+        {
+            Edge(State::NotFilled, State::AboveThreshold),
+            &BitRateWindow::onNotFilledToAboveThreshold
+        },
+
+        // BelowThreshold => AboveThreshold
+        {
+            Edge(State::BelowThreshold, State::AboveThreshold),
+            &BitRateWindow::onBelowToAboveThreshold
+        },
+
+        // AboveThreshold => Idle
+        {
+            Edge(State::AboveThreshold, State::Idle),
+            &BitRateWindow::onAboveThresholdToIdle
+        },
+
+        // Idle => NotFilled
+        {
+            Edge(State::Idle, State::NotFilled),
+            &BitRateWindow::onIdleToNotFilled
+        },
+    };
+
+public:
+    BitRateWindow(size_t MinLength, size_t MaxLength, size_t IdleLength,
+                  size_t SetBitsThreshold) :
+        MinLength(MinLength), MaxLength(MaxLength), IdleLength(IdleLength),
+        SetBitsThreshold(SetBitsThreshold),
+        CurrState(State::NotFilled), CurrLength(0), BBC(MinLength) {}
+
+    std::pair<Edge, size_t> insert(bool Bit);
+
+    void print(std::ostream &OS) const;
+
+private:
+    size_t onRoundtripNotFilled(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength += 1;
+        return CurrLength;
+    }
+
+    size_t onRoundtripBelowThreshold(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength = MinLength;
+        return CurrLength;
+    }
+
+    size_t onRoundtripAboveThreshold(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength += 1;
+        return CurrLength;
+    }
+
+    size_t onRoundtripIdle(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength += 1;
+        return CurrLength;
+    }
+
+    size_t onNotFilledToBelowThreshold(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength = MinLength;
+        return CurrLength;
+    }
+
+    size_t onNotFilledToAboveThreshold(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength += 1;
+        return CurrLength;
+    }
+
+    size_t onBelowToAboveThreshold(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        CurrLength = MinLength;
+        return CurrLength;
+    }
+
+    size_t onAboveThresholdToIdle(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        size_t PrevLength = CurrLength;
+        CurrLength = 1;
+        return PrevLength;
+    }
+
+    size_t onIdleToNotFilled(State PrevState, bool NewBit) {
+        (void) PrevState, (void) NewBit;
+
+        BBC = BitBufferCounter(MinLength);
+        BBC.insert(NewBit);
+
+        CurrLength = 1;
+        return CurrLength;
+    }
+
+private:
+    size_t MinLength;
+    size_t MaxLength;
+    size_t IdleLength;
+    size_t SetBitsThreshold;
+
+    State CurrState;
+    size_t CurrLength;
+    BitBufferCounter BBC;
+};
+
+} // namespace ml
+
+inline std::ostream& operator<<(std::ostream &OS, const ml::BitRateWindow BRW) {
+    BRW.print(OS);
+    return OS;
+}
+
+#endif /* BIT_RATE_WINDOW_H */
-- 
cgit v1.2.3