summaryrefslogtreecommitdiffhomepage
path: root/src/kernel/NeKit/Stream.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/kernel/NeKit/Stream.h')
-rw-r--r--src/kernel/NeKit/Stream.h45
1 files changed, 45 insertions, 0 deletions
diff --git a/src/kernel/NeKit/Stream.h b/src/kernel/NeKit/Stream.h
new file mode 100644
index 00000000..1a53e7f0
--- /dev/null
+++ b/src/kernel/NeKit/Stream.h
@@ -0,0 +1,45 @@
+
+/* ========================================
+
+ Copyright (C) 2024-2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.
+
+======================================== */
+
+#pragma once
+
+#include <NeKit/Defines.h>
+#include <NeKit/Ref.h>
+
+namespace Kernel {
+template <typename StreamTrait, typename Kind>
+class Stream final {
+ public:
+ explicit Stream(Ref<Stream> ref) : fStream(ref) {}
+
+ ~Stream() = default;
+
+ Stream& operator=(const Stream&) = default;
+ Stream(const Stream&) = default;
+
+ template <typename Data>
+ friend Stream<StreamTrait, Kind>& operator>>(Stream<StreamTrait, Kind>& Ks, Ref<Data>& Buf) {
+ Ks.fKind = Ks.fStream->In(Buf);
+ return *Ks;
+ }
+
+ template <typename Data>
+ friend Stream<StreamTrait, Kind>& operator<<(Stream<StreamTrait, Kind>& Ks, Ref<Data>& Buf) {
+ Ks.fKind = Buf;
+ Ks.fStream->Out(Buf.Leak());
+ return *Ks;
+ }
+
+ Ref<StreamTrait>& AsStreamTrait() { return fStream; }
+
+ Ref<Kind>& AsType() { return fKind; }
+
+ private:
+ Ref<StreamTrait> fStream;
+ Ref<Kind> fKind;
+};
+} // namespace Kernel