mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-22 20:40:18 +00:00
78 lines
2.4 KiB
C
78 lines
2.4 KiB
C
#include "pocketpy/interpreter/typeinfo.h"
|
|
#include "pocketpy/common/utils.h"
|
|
|
|
#define CHUNK_SIZE 128
|
|
#define LOG2_CHUNK_SIZE 7
|
|
|
|
void TypeList__ctor(TypeList* self) {
|
|
self->length = 0;
|
|
memset(self->chunks, 0, sizeof(self->chunks));
|
|
}
|
|
|
|
void TypeList__dtor(TypeList* self) {
|
|
for(py_Type t = 0; t < self->length; t++) {
|
|
py_TypeInfo* info = TypeList__get(self, t);
|
|
if(info->magic_1) PK_FREE(info->magic_1);
|
|
}
|
|
for(int i = 0; i < PK_MAX_CHUNK_LENGTH; i++) {
|
|
if(self->chunks[i]) PK_FREE(self->chunks[i]);
|
|
}
|
|
}
|
|
|
|
py_TypeInfo* TypeList__get(TypeList* self, py_Type index) {
|
|
assert(index < self->length);
|
|
int chunk = index >> LOG2_CHUNK_SIZE;
|
|
int offset = index & (CHUNK_SIZE - 1);
|
|
return self->chunks[chunk] + offset;
|
|
}
|
|
|
|
py_TypeInfo* TypeList__emplace(TypeList* self) {
|
|
int chunk = self->length >> LOG2_CHUNK_SIZE;
|
|
int offset = self->length & (CHUNK_SIZE - 1);
|
|
if(self->chunks[chunk] == NULL) {
|
|
if(chunk >= PK_MAX_CHUNK_LENGTH) {
|
|
c11__abort("TypeList__emplace(): max chunk length exceeded");
|
|
}
|
|
self->chunks[chunk] = PK_MALLOC(sizeof(py_TypeInfo) * CHUNK_SIZE);
|
|
memset(self->chunks[chunk], 0, sizeof(py_TypeInfo) * CHUNK_SIZE);
|
|
}
|
|
self->length++;
|
|
return self->chunks[chunk] + offset;
|
|
}
|
|
|
|
void TypeList__apply(TypeList* self, void (*f)(py_TypeInfo*, void*), void* ctx) {
|
|
for(int i = 0; i < self->length; i++) {
|
|
py_TypeInfo* info = TypeList__get(self, i);
|
|
f(info, ctx);
|
|
}
|
|
}
|
|
|
|
py_TValue* TypeList__magic(py_TypeInfo* self, unsigned index) {
|
|
if(index > __xor__) {
|
|
// common magic slots
|
|
return TypeList__magic_common(self, index);
|
|
}
|
|
// uncommon magic slots
|
|
if(self->magic_1 == NULL) {
|
|
self->magic_1 = PK_MALLOC(sizeof(py_TValue) * PK_MAGIC_SLOTS_UNCOMMON_LENGTH);
|
|
memset(self->magic_1, 0, sizeof(py_TValue) * PK_MAGIC_SLOTS_UNCOMMON_LENGTH);
|
|
}
|
|
return self->magic_1 + index;
|
|
}
|
|
|
|
py_TValue* TypeList__magic_readonly_nullable(py_TypeInfo* self, unsigned index) {
|
|
if(index > __xor__) {
|
|
// common magic slots
|
|
py_TValue* slot = TypeList__magic_common(self, index);
|
|
if(py_isnil(slot)) return NULL;
|
|
return slot;
|
|
}
|
|
// uncommon magic slots
|
|
if(self->magic_1 == NULL) return NULL;
|
|
py_TValue* slot = self->magic_1 + index;
|
|
if(py_isnil(slot)) return NULL;
|
|
return slot;
|
|
}
|
|
|
|
#undef CHUNK_SIZE
|
|
#undef LOG2_CHUNK_SIZE |