blob: 0fac23c1528a0b44ebcc0d0b42096ed140c8f0ad (
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
|
// Copyright 2024-2026, Amlal El Mahrouss (amlal@nekernel.org)
// Licensed under the Apache License, Version 2.0 (see LICENSE file)
// Official repository: https://github.com/ne-foss-org/nekernel
#ifndef NEKIT_STREAM_H
#define NEKIT_STREAM_H
#include <CompilerKit/CompilerKit.h>
#include <NeKit/Config.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;
NE_COPY_DEFAULT(Stream)
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 = Ks.fStream->Out(Buf.Leak());
return *Ks;
}
Ref<StreamTrait>& AsStreamTrait() { return fStream; }
Ref<Kind>& AsType() { return fKind; }
private:
Ref<StreamTrait> fStream;
Ref<Kind> fKind;
};
template <class ST, class Kind>
using UnbufferedStream = Stream<ST, Kind>;
template <class BST, class BKind>
using BufferedStream = Stream<BST, BKind>;
} // namespace Kernel
#endif
|