From a7fdc594403d7157a8d3c3ffd1dd4cbe99fd3746 Mon Sep 17 00:00:00 2001 From: Sam Vervaeck Date: Sun, 14 May 2023 21:20:44 +0200 Subject: [PATCH] Add missing Stream.hpp --- include/bolt/Stream.hpp | 76 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 76 insertions(+) create mode 100644 include/bolt/Stream.hpp diff --git a/include/bolt/Stream.hpp b/include/bolt/Stream.hpp new file mode 100644 index 000000000..76dfb7c4d --- /dev/null +++ b/include/bolt/Stream.hpp @@ -0,0 +1,76 @@ + +#pragma once + +#include +#include + +namespace bolt { + + template + class Stream { + public: + + virtual T get() = 0; + virtual T peek(std::size_t Offset = 0) = 0; + + virtual ~Stream() {} + + }; + + template + class VectorStream : public Stream { + public: + + using value_type = T; + + ContainerT& Data; + value_type Sentry; + std::size_t Offset; + + VectorStream(ContainerT& Data, value_type Sentry, std::size_t Offset = 0): + Data(Data), Sentry(Sentry), Offset(Offset) {} + + value_type get() override { + return Offset < Data.size() ? Data[Offset++] : Sentry; + } + + value_type peek(std::size_t Offset2) override { + auto I = Offset + Offset2; + return I < Data.size() ? Data[I] : Sentry; + } + + }; + + template + class BufferedStream : public Stream { + + std::deque Buffer; + + protected: + + virtual T read() = 0; + + public: + + using value_type = T; + + value_type get() override { + if (Buffer.empty()) { + return read(); + } else { + auto Keep = Buffer.front(); + Buffer.pop_front(); + return Keep; + } + } + + value_type peek(std::size_t Offset = 0) override { + while (Buffer.size() <= Offset) { + Buffer.push_back(read()); + } + return Buffer[Offset]; + } + + }; + +}