Files
metadesk/source/base/thread_context.c

87 lines
2.1 KiB
C

#ifdef INTELLISENSE_DIRECTIVES
# include "arena.h"
# include "thread_context.h"
#endif
// Copyright (c) 2024 Epic Games Tools
// Licensed under the MIT license (https://opensource.org/license/mit/)
////////////////////////////////
// NOTE(allen): Thread MD_Context Functions
MD_API_C md_thread_static MD_TCTX* md_tctx_thread_local;
#if ! MD_BUILD_SUPPLEMENTARY_UNIT
MD_API_C md_thread_static MD_TCTX* md_tctx_thread_local = 0;
#endif
void
md_tctx_init_and_equip(MD_TCTX* md_tctx)
{
// md_memory_zero_struct(md_tctx);
MD_Arena** md_arena_ptr = md_tctx->arenas;
for (MD_U64 i = 0; i < md_array_count(md_tctx->arenas); i += 1, md_arena_ptr += 1)
{
if (*md_arena_ptr == md_nullptr)
{
MD_VArena* vm = md_varena_alloc(.reserve_size = MD_VARENA_DEFAULT_RESERVE, .commit_size = MD_VARENA_DEFAULT_COMMIT);
*md_arena_ptr = md_arena_alloc(.backing = md_varena_allocator(vm));
}
}
md_tctx_thread_local = md_tctx;
}
void
md_tctx_init_and_equip_alloc(MD_TCTX* md_tctx, MD_AllocatorInfo ainfo)
{
md_memory_zero_struct(md_tctx);
MD_Arena** md_arena_ptr = md_tctx->arenas;
for (MD_U64 i = 0; i < md_array_count(md_tctx->arenas); i += 1, md_arena_ptr += 1)
{
if (*md_arena_ptr == md_nullptr) {
*md_arena_ptr = md_arena_alloc(.backing = ainfo);
}
}
md_tctx_thread_local = md_tctx;
}
void
md_tctx_release(void) {
for (MD_U64 i = 0; i < md_array_count(md_tctx_thread_local->arenas); i += 1) {
md_arena_release(md_tctx_thread_local->arenas[i]);
}
}
MD_TCTX*
md_tctx_get_equipped(void){
return(md_tctx_thread_local);
}
MD_Arena*
md_tctx_get_scratch(MD_Arena** conflicts, MD_U64 count)
{
MD_TCTX* md_tctx = md_tctx_get_equipped();
MD_Arena* result = 0;
MD_Arena** md_arena_ptr = md_tctx->arenas;
for (MD_U64 i = 0; i < md_array_count(md_tctx->arenas); i += 1, md_arena_ptr += 1)
{
MD_Arena** conflict_ptr = conflicts;
MD_B32 has_conflict = 0;
for (MD_U64 j = 0; j < count; j += 1, conflict_ptr += 1)
{
if (*md_arena_ptr == *conflict_ptr) {
has_conflict = 1;
break;
}
}
if ( ! has_conflict){
result = *md_arena_ptr;
break;
}
}
return(result);
}