Kernel/DoubleBuffer.h


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

#pragma once

#include <AK/Types.h>
#include <AK/Vector.h>
#include <Kernel/Lock.h>

class DoubleBuffer {
public:
    DoubleBuffer()
        : m_write_buffer(&m_buffer1)
        , m_read_buffer(&m_buffer2)
    {
    }

    ssize_t write(const u8*, ssize_t);
    ssize_t read(u8*, ssize_t);

    bool is_empty() const { return m_empty; }

    // FIXME: Isn't this racy? What if we get interrupted between getting the buffer pointer and dereferencing it?
    ssize_t bytes_in_write_buffer() const { return (ssize_t)m_write_buffer->size(); }

private:
    void flip();
    void compute_emptiness();

    Vector<u8>* m_write_buffer { nullptr };
    Vector<u8>* m_read_buffer { nullptr };
    Vector<u8> m_buffer1;
    Vector<u8> m_buffer2;
    ssize_t m_read_buffer_index { 0 };
    bool m_empty { true };
    Lock m_lock { "DoubleBuffer" };
};