-
Notifications
You must be signed in to change notification settings - Fork 220
Expand file tree
/
Copy pathRingBuffer.h
More file actions
75 lines (62 loc) · 1.29 KB
/
RingBuffer.h
File metadata and controls
75 lines (62 loc) · 1.29 KB
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
// SPDX-License-Identifier: MIT
/**
Copyright (c) 2015 - 2022 Beckhoff Automation GmbH & Co. KG
*/
#pragma once
#include <cassert>
#include <cstdint>
#include <memory>
struct RingBuffer {
friend struct RingBufferTransaction;
RingBuffer(size_t N)
: dataSize(N + 1)
, data(new uint8_t[N + 1])
, write(data.get())
, read(data.get())
{
}
size_t BytesFree() const
{
return (write < read) ? read - write - 1 :
dataSize - 1 - (write - read);
}
size_t BytesAvailable() const
{
return dataSize - BytesFree() - 1;
}
size_t WriteChunk() const
{
return (write < read) ? read - write - 1 :
data.get() + dataSize - write -
(data.get() == read);
}
void Write(size_t n)
{
assert(n <= BytesFree());
write = Increment(write, n);
}
template <class T> T ReadFromLittleEndian()
{
T result = 0;
for (size_t i = 0; i < sizeof(T); ++i) {
result += (((T)(*read)) << (8 * i));
read = Increment(read, 1);
}
return result;
}
void Read(size_t n)
{
assert(n <= BytesAvailable());
read = Increment(read, n);
}
private:
const size_t dataSize;
const std::unique_ptr<uint8_t[]> data;
inline uint8_t *Increment(const uint8_t *ptr, size_t n)
{
return data.get() + ((ptr - data.get() + n) % dataSize);
}
public:
uint8_t *write;
const uint8_t *read;
};