diff options
Diffstat (limited to 'src/kernel/NeKit/Stream.h')
| -rw-r--r-- | src/kernel/NeKit/Stream.h | 45 |
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 |
