summaryrefslogtreecommitdiffhomepage
path: root/dev/kernel/NewKit/Stream.h
blob: dfe1881c395c2187bb34b66b1e054ce3aed2f121 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

/* -------------------------------------------

	Copyright (C) 2024-2025, Amlal El Mahrouss, all rights reserved.

------------------------------------------- */

#pragma once

#include <NewKit/Defines.h>
#include <NewKit/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