...

1 files changed, 44 insertions, 0 deletions

diff --git a/include/cru/common/io/MemoryStream.h b/include/cru/common/io/MemoryStream.h
new file mode 100644
index 00000000..a6243d67
--- /dev/null
+++ b/include/cru/common/io/MemoryStream.h
@@ -0,0 +1,44 @@
+#pragma once
+
+#include "Stream.h"
+
+#include <functional>
+
+namespace cru::io {
+class CRU_BASE_API MemoryStream : public Stream {
+ public:
+  MemoryStream() = default;
+  MemoryStream(
+      std::byte* buffer, Index size, bool read_only = false,
+      std::function<void(std::byte* buffer, Index size)> release_func = {})
+      : buffer_(buffer),
+        size_(size),
+        read_only_(read_only),
+        release_func_(std::move(release_func)) {}
+
+  CRU_DELETE_COPY(MemoryStream)
+  CRU_DELETE_MOVE(MemoryStream)
+
+  ~MemoryStream() override;
+
+ public:
+  bool CanSeek() override;
+  Index Seek(Index offset, SeekOrigin origin = SeekOrigin::Current) override;
+
+  bool CanRead() override;
+  Index Read(std::byte* buffer, Index offset, Index size) override;
+
+  bool CanWrite() override;
+  Index Write(const std::byte* buffer, Index offset, Index size) override;
+
+  std::byte* GetBuffer() const { return buffer_; }
+  Index GetSize() override { return size_; }
+
+ private:
+  std::byte* buffer_ = nullptr;
+  Index size_ = 0;
+  Index position_ = 0;
+  bool read_only_ = false;
+  std::function<void(std::byte* buffer, Index size)> release_func_;
+};
+}  // namespace cru::io