summaryrefslogtreecommitdiff
path: root/src/kitbuffer.c
blob: 5c492ea4b039ecaf39565701252c86fc074e95d1 (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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include "kitchensink/internal/kitbuffer.h"

#include <stdlib.h>
#include <assert.h>

Kit_Buffer* Kit_CreateBuffer(unsigned int size, Kit_BufferFreeCallback free_cb) {
    Kit_Buffer *b = calloc(1, sizeof(Kit_Buffer));
    if(b == NULL) {
        return NULL;
    }
    b->size = size;
    b->free_cb = free_cb;
    b->data = calloc(size, sizeof(void*));
    if(b->data == NULL) {
        free(b);
        return NULL;
    }
    return b;
}

void Kit_DestroyBuffer(Kit_Buffer *buffer) {
    if(buffer == NULL) return;
    Kit_ClearBuffer(buffer);
    free(buffer->data);
    free(buffer);
}

void Kit_ClearBuffer(Kit_Buffer *buffer) {
    void *data;
    while((data = Kit_ReadBuffer(buffer)) != NULL) {
        buffer->free_cb(data);
    }
}

void* Kit_ReadBuffer(Kit_Buffer *buffer) {
    assert(buffer != NULL);
    if(buffer->read_p < buffer->write_p) {
        void *out = buffer->data[buffer->read_p % buffer->size];
        buffer->data[buffer->read_p % buffer->size] = NULL;
        buffer->read_p++;
        if(buffer->read_p >= buffer->size) {
            buffer->read_p = buffer->read_p % buffer->size;
            buffer->write_p = buffer->write_p % buffer->size;
        }
        return out;
    }
    return NULL;
}

KIT_LOCAL void* Kit_PeekBuffer(const Kit_Buffer *buffer) {
    assert(buffer != NULL);
    return buffer->data[buffer->read_p % buffer->size];
}

KIT_LOCAL void Kit_AdvanceBuffer(Kit_Buffer *buffer) {
    assert(buffer != NULL);
    if(buffer->read_p < buffer->write_p) {
        buffer->data[buffer->read_p % buffer->size] = NULL;
        buffer->read_p++;
        if(buffer->read_p >= buffer->size) {
            buffer->read_p = buffer->read_p % buffer->size;
            buffer->write_p = buffer->write_p % buffer->size;
        }
    }
}

int Kit_WriteBuffer(Kit_Buffer *buffer, void *ptr) {
    assert(buffer != NULL);
    assert(ptr != NULL);

    if(!Kit_IsBufferFull(buffer)) {
        buffer->data[buffer->write_p % buffer->size] = ptr;
        buffer->write_p++;
        return 0;
    }
    return 1;
}

int Kit_IsBufferFull(const Kit_Buffer *buffer) {
    int len = buffer->write_p - buffer->read_p;
    int k = (len >= buffer->size);
    return k;
}