mirrors/Odin
1
0
Fork 0
mirror of https://github.com/odin-lang/Odin.git synced 2025-12-30 18:02:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

move to a growing queue

Browse Source
This commit is contained in:
Colin Davidson
2024-07-13 23:16:22 -07:00
parent 8cebc7cc0c
commit cdede4928c
2 changed files with 99 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/threading.cpp
View File

@@ -46,6 +46,18 @@ typedef struct WorkerTask {
void *data;
} WorkerTask;
typedef struct TaskRingBuffer {
std::atomic<ssize_t> size;
std::atomic<WorkerTask *> buffer;
} TaskRingBuffer;
typedef struct TaskQueue {
std::atomic<ssize_t> top;
std::atomic<ssize_t> bottom;
std::atomic<TaskRingBuffer *> ring;
} TaskQueue;
struct Thread {
#if defined(GB_SYSTEM_WINDOWS)
void *win32_handle;
@@ -54,12 +66,9 @@ struct Thread {
#endif
isize idx;
WorkerTask *queue;
size_t capacity;
std::atomic<uint64_t> head_and_tail;
isize stack_size;
struct TaskQueue queue;
struct ThreadPool *pool;
};
@@ -551,6 +560,18 @@ gb_internal void *internal_thread_proc(void *arg) {
}
#endif
TaskRingBuffer *taskring_init(ssize_t size) {
TaskRingBuffer *ring = (TaskRingBuffer *)gb_alloc(heap_allocator(), sizeof(TaskRingBuffer));
ring->size = size;
ring->buffer = (WorkerTask *)gb_alloc_array(heap_allocator(), WorkerTask, ring->size);
return ring;
}
void thread_queue_destroy(TaskQueue *q) {
gb_free(heap_allocator(), (*q->ring).buffer);
gb_free(heap_allocator(), q->ring);
}
gb_internal void thread_init(ThreadPool *pool, Thread *t, isize idx) {
gb_zero_item(t);
#if defined(GB_SYSTEM_WINDOWS)
@@ -559,14 +580,12 @@ gb_internal void thread_init(ThreadPool *pool, Thread *t, isize idx) {
t->posix_handle = 0;
#endif
t->capacity = 1 << 14; // must be a power of 2
t->queue = gb_alloc_array(heap_allocator(), WorkerTask, t->capacity);
t->head_and_tail = 0;
// Size must be a power of 2
t->queue.ring = taskring_init(1 << 14);
t->pool = pool;
t->idx = idx;
}
gb_internal void thread_init_and_start(ThreadPool *pool, Thread *t, isize idx) {
thread_init(pool, t, idx);
isize stack_size = 0;
@@ -598,7 +617,7 @@ gb_internal void thread_join_and_destroy(Thread *t) {
t->posix_handle = 0;
#endif
gb_free(heap_allocator(), t->queue);
thread_queue_destroy(&t->queue);
}
gb_internal void thread_set_name(Thread *t, char const *name) {