mirror of
https://github.com/Ed94/Odin.git
synced 2026-08-06 15:48:51 +00:00
Add thread.run* shorthand procedures; Thread.init_context use new Maybe concept
This commit is contained in:
+40
-2
@@ -1,6 +1,8 @@
|
|||||||
package thread
|
package thread
|
||||||
|
|
||||||
import "core:runtime"
|
import "core:runtime"
|
||||||
|
import "core:os"
|
||||||
|
import "core:sync"
|
||||||
|
|
||||||
Thread_Proc :: #type proc(^Thread);
|
Thread_Proc :: #type proc(^Thread);
|
||||||
|
|
||||||
@@ -10,6 +12,42 @@ Thread :: struct {
|
|||||||
data: rawptr,
|
data: rawptr,
|
||||||
user_index: int,
|
user_index: int,
|
||||||
|
|
||||||
init_context: runtime.Context,
|
init_context: Maybe(runtime.Context),
|
||||||
use_init_context: bool,
|
}
|
||||||
|
|
||||||
|
#assert(size_of(Thread{}.user_index) == size_of(uintptr));
|
||||||
|
|
||||||
|
|
||||||
|
run :: proc(fn: proc(), init_context: Maybe(runtime.Context) = nil, priority := Thread_Priority.Normal) {
|
||||||
|
thread_proc :: proc(t: ^Thread) {
|
||||||
|
fn := cast(proc())t.data;
|
||||||
|
fn();
|
||||||
|
destroy(t);
|
||||||
|
}
|
||||||
|
t := create(thread_proc, priority);
|
||||||
|
t.data = rawptr(fn);
|
||||||
|
t.init_context = init_context;
|
||||||
|
start(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
run_with_data :: proc(fn: proc(data: rawptr), data: rawptr, init_context: Maybe(runtime.Context) = nil, priority := Thread_Priority.Normal) {
|
||||||
|
thread_proc :: proc(t: ^Thread) {
|
||||||
|
fn := cast(proc(rawptr))t.data;
|
||||||
|
data := rawptr(uintptr(t.user_index));
|
||||||
|
fn(data);
|
||||||
|
destroy(t);
|
||||||
|
}
|
||||||
|
t := create(thread_proc, priority);
|
||||||
|
t.data = rawptr(fn);
|
||||||
|
t.user_index = int(uintptr(data));
|
||||||
|
t.init_context = init_context;
|
||||||
|
start(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
run_with_thread_proc :: proc(fn: Thread_Proc, init_context: Maybe(runtime.Context) = nil, priority := Thread_Priority.Normal) {
|
||||||
|
t := create(fn, priority);
|
||||||
|
t.init_context = init_context;
|
||||||
|
start(t);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -51,14 +51,14 @@ create :: proc(procedure: Thread_Proc, priority := Thread_Priority.Normal) -> ^T
|
|||||||
t.start_gate = {};
|
t.start_gate = {};
|
||||||
|
|
||||||
c := context;
|
c := context;
|
||||||
if t.use_init_context {
|
if ic, ok := t.init_context.?; ok {
|
||||||
c = t.init_context;
|
c = ic;
|
||||||
}
|
}
|
||||||
context = c;
|
context = c;
|
||||||
|
|
||||||
t.procedure(t);
|
t.procedure(t);
|
||||||
|
|
||||||
if !t.use_init_context {
|
if t.init_context == nil {
|
||||||
if context.temp_allocator.data == &runtime.global_default_temp_allocator_data {
|
if context.temp_allocator.data == &runtime.global_default_temp_allocator_data {
|
||||||
runtime.default_temp_allocator_destroy(auto_cast context.temp_allocator.data);
|
runtime.default_temp_allocator_destroy(auto_cast context.temp_allocator.data);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,33 +10,33 @@ Thread_Os_Specific :: struct {
|
|||||||
done: bool, // see note in `is_done`
|
done: bool, // see note in `is_done`
|
||||||
}
|
}
|
||||||
|
|
||||||
THREAD_PRIORITY_IDLE :: -15;
|
|
||||||
THREAD_PRIORITY_LOWEST :: -2;
|
|
||||||
THREAD_PRIORITY_BELOW_NORMAL :: -1;
|
|
||||||
THREAD_PRIORITY_NORMAL :: 0;
|
|
||||||
THREAD_PRIORITY_ABOVE_NORMAL :: 1;
|
|
||||||
THREAD_PRIORITY_HIGHEST :: 2;
|
|
||||||
THREAD_PRIORITY_TIME_CRITICAL :: 15;
|
|
||||||
|
|
||||||
Thread_Priority :: enum i32 {
|
Thread_Priority :: enum {
|
||||||
Normal = THREAD_PRIORITY_NORMAL,
|
Normal,
|
||||||
Low = THREAD_PRIORITY_LOWEST,
|
Low,
|
||||||
High = THREAD_PRIORITY_HIGHEST,
|
High,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
_thread_priority_map := map[Thread_Priority]i32{
|
||||||
|
.Normal = 0,
|
||||||
|
.Low = -2,
|
||||||
|
.High = +2,
|
||||||
|
};
|
||||||
|
|
||||||
create :: proc(procedure: Thread_Proc, priority := Thread_Priority.Normal) -> ^Thread {
|
create :: proc(procedure: Thread_Proc, priority := Thread_Priority.Normal) -> ^Thread {
|
||||||
win32_thread_id: u32;
|
win32_thread_id: u32;
|
||||||
|
|
||||||
__windows_thread_entry_proc :: proc "c" (t: ^Thread) -> i32 {
|
__windows_thread_entry_proc :: proc "c" (t: ^Thread) -> i32 {
|
||||||
c := runtime.default_context();
|
context = runtime.default_context();
|
||||||
if t.use_init_context {
|
c := context;
|
||||||
c = t.init_context;
|
if ic, ok := t.init_context.?; ok {
|
||||||
|
c = ic;
|
||||||
}
|
}
|
||||||
context = c;
|
context = c;
|
||||||
|
|
||||||
t.procedure(t);
|
t.procedure(t);
|
||||||
|
|
||||||
if !t.use_init_context {
|
if t.init_context == nil {
|
||||||
if context.temp_allocator.data == &runtime.global_default_temp_allocator_data {
|
if context.temp_allocator.data == &runtime.global_default_temp_allocator_data {
|
||||||
runtime.default_temp_allocator_destroy(auto_cast context.temp_allocator.data);
|
runtime.default_temp_allocator_destroy(auto_cast context.temp_allocator.data);
|
||||||
}
|
}
|
||||||
@@ -58,8 +58,9 @@ create :: proc(procedure: Thread_Proc, priority := Thread_Priority.Normal) -> ^T
|
|||||||
thread.procedure = procedure;
|
thread.procedure = procedure;
|
||||||
thread.win32_thread = win32_thread;
|
thread.win32_thread = win32_thread;
|
||||||
thread.win32_thread_id = win32_thread_id;
|
thread.win32_thread_id = win32_thread_id;
|
||||||
|
thread.init_context = context;
|
||||||
|
|
||||||
ok := win32.set_thread_priority(win32_thread, i32(priority));
|
ok := win32.set_thread_priority(win32_thread, _thread_priority_map[priority]);
|
||||||
assert(ok == true);
|
assert(ok == true);
|
||||||
|
|
||||||
return thread;
|
return thread;
|
||||||
|
|||||||
@@ -1120,7 +1120,6 @@ threading_example :: proc() {
|
|||||||
for in prefix_table {
|
for in prefix_table {
|
||||||
if t := thread.create(worker_proc); t != nil {
|
if t := thread.create(worker_proc); t != nil {
|
||||||
t.init_context = context;
|
t.init_context = context;
|
||||||
t.use_init_context = true;
|
|
||||||
t.user_index = len(threads);
|
t.user_index = len(threads);
|
||||||
append(&threads, t);
|
append(&threads, t);
|
||||||
thread.start(t);
|
thread.start(t);
|
||||||
|
|||||||
Reference in New Issue
Block a user