fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-05-01 03:48:06 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

util: add thread-safe version of idalloc

Browse source 
For buffer IDs in u_threaded_context.

Reviewed-By: Mike Blumenkrantz <michael.blumenkrantz@gmail.com>
Reviewed-by: Rob Clark <robdclark@chromium.org>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/10662>
This commit is contained in:
Marek Olšák 2021-05-05 20:09:56 -04:00 • committed by Marge Bot
parent 6fed423ade
commit 3828081c92
2 changed files with 100 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
src/util/u_idalloc.c
View file

@ -108,3 +108,70 @@ util_idalloc_reserve(struct util_idalloc *buf, unsigned id)
assert((buf->data[id / 32] & (1u << (id % 32))) == 0);
buf->data[id / 32] |= 1u << (id % 32);
}
void
util_idalloc_mt_init(struct util_idalloc_mt *buf,
unsigned initial_num_elements, bool skip_zero)
{
simple_mtx_init(&buf->mutex, mtx_plain);
util_idalloc_init(&buf->buf);
buf->skip_zero = skip_zero;
if (initial_num_elements)
util_idalloc_resize(&buf->buf, initial_num_elements);
if (skip_zero) {
ASSERTED unsigned zero = util_idalloc_alloc(&buf->buf);
assert(zero == 0);
}
}
/* Callback for drivers using u_threaded_context (abbreviated as tc). */
void
util_idalloc_mt_init_tc(struct util_idalloc_mt *buf)
{
util_idalloc_mt_init(buf, 1 << 16, true);
}
void
util_idalloc_mt_fini(struct util_idalloc_mt *buf)
{
util_idalloc_fini(&buf->buf);
simple_mtx_destroy(&buf->mutex);
}
void
util_idalloc_mt_resize(struct util_idalloc_mt *buf, unsigned new_num_elements)
{
simple_mtx_lock(&buf->mutex);
util_idalloc_resize(&buf->buf, new_num_elements);
simple_mtx_unlock(&buf->mutex);
}
unsigned
util_idalloc_mt_alloc(struct util_idalloc_mt *buf)
{
simple_mtx_lock(&buf->mutex);
unsigned id = util_idalloc_alloc(&buf->buf);
simple_mtx_unlock(&buf->mutex);
return id;
}
void
util_idalloc_mt_free(struct util_idalloc_mt *buf, unsigned id)
{
if (id == 0 && buf->skip_zero)
return;
simple_mtx_lock(&buf->mutex);
util_idalloc_free(&buf->buf, id);
simple_mtx_unlock(&buf->mutex);
}
void
util_idalloc_mt_reserve(struct util_idalloc_mt *buf, unsigned id)
{
simple_mtx_lock(&buf->mutex);
util_idalloc_reserve(&buf->buf, id);
simple_mtx_unlock(&buf->mutex);
}

33
src/util/u_idalloc.h
View file

@ -29,6 +29,8 @@
#define U_IDALLOC_H
#include <inttypes.h>
#include <stdbool.h>
#include "simple_mtx.h"
#ifdef __cplusplus
extern "C" {
@ -59,6 +61,37 @@ util_idalloc_free(struct util_idalloc *buf, unsigned id);
void
util_idalloc_reserve(struct util_idalloc *buf, unsigned id);
/* Thread-safe variant. */
struct util_idalloc_mt {
struct util_idalloc buf;
simple_mtx_t mutex;
bool skip_zero;
};
void
util_idalloc_mt_init(struct util_idalloc_mt *buf,
unsigned initial_num_elements, bool skip_zero);
void
util_idalloc_mt_init_tc(struct util_idalloc_mt *buf);
void
util_idalloc_mt_fini(struct util_idalloc_mt *buf);
void
util_idalloc_mt_resize(struct util_idalloc_mt *buf, unsigned new_num_elements);
unsigned
util_idalloc_mt_alloc(struct util_idalloc_mt *buf);
void
util_idalloc_mt_free(struct util_idalloc_mt *buf, unsigned id);
void
util_idalloc_mt_reserve(struct util_idalloc_mt *buf, unsigned id);
#ifdef __cplusplus
}
#endif