2023-04-15 11:34:17 +08:00

98 lines
3.0 KiB
C

#include "_thread.h"
#include "PikaVM.h"
#include "TinyObj.h"
typedef struct pika_thread_info {
Arg* function;
Arg* args;
pika_platform_thread_t* thread;
} pika_thread_info;
static void _thread_func(void* arg) {
pika_debug("waiting for first lock");
while (1) {
if (_VM_is_first_lock()) {
break;
}
//! This May break the thread
// if (_VMEvent_getVMCnt() <= 0) {
// break;
// }
pika_debug("VM num %d", _VMEvent_getVMCnt());
pika_platform_thread_delay();
}
pika_debug("thread start");
pika_GIL_ENTER();
PikaObj* ctx = New_TinyObj(NULL);
pika_thread_info* info = (pika_thread_info*)arg;
if (NULL != info->args) {
obj_setArg(ctx, "args", info->args);
}
obj_setArg(ctx, "thread", info->function);
if (NULL == info->args) {
/* clang-format off */
PIKA_PYTHON(
thread()
)
/* clang-format on */
const uint8_t bytes[] = {
0x04, 0x00, 0x00, 0x00, /* instruct array size */
0x00, 0x82, 0x01, 0x00, /* instruct array */
0x08, 0x00, 0x00, 0x00, /* const pool size */
0x00, 0x74, 0x68, 0x72, 0x65, 0x61, 0x64, 0x00, /* const pool */
};
pikaVM_runByteCode(ctx, (uint8_t*)bytes);
} else {
/* clang-format off */
PIKA_PYTHON(
thread(*args)
)
/* clang-format on */
const uint8_t bytes[] = {
0x0c, 0x00, 0x00, 0x00, /* instruct array size */
0x20, 0x81, 0x01, 0x00, 0x10, 0x08, 0x06, 0x00, 0x00, 0x02, 0x08,
0x00,
/* instruct array */
0x0f, 0x00, 0x00, 0x00, /* const pool size */
0x00, 0x61, 0x72, 0x67, 0x73, 0x00, 0x2a, 0x00, 0x74, 0x68, 0x72,
0x65, 0x61, 0x64, 0x00, /* const pool */
};
pikaVM_runByteCode(ctx, (uint8_t*)bytes);
}
obj_deinit(ctx);
arg_deinit(info->function);
if (NULL != info->args) {
arg_deinit(info->args);
}
pika_debug("thread exiting");
pika_platform_thread_t* thread = info->thread;
pikaFree(info, sizeof(pika_thread_info));
pika_GIL_EXIT();
#if PIKA_FREERTOS_ENABLE
pikaFree(thread, sizeof(pika_platform_thread_t));
pika_platform_thread_exit(NULL);
#else
pika_platform_thread_exit(thread);
#endif
}
int PikaStdData_Tuple_len(PikaObj* self);
void _thread_start_new_thread(PikaObj* self, Arg* function, Arg* args_) {
pika_thread_info* info =
(pika_thread_info*)pikaMalloc(sizeof(pika_thread_info));
pika_platform_memset(info, 0, sizeof(pika_thread_info));
info->function = arg_copy(function);
PikaObj* tuple = arg_getPtr(args_);
size_t tuple_size = PikaStdData_Tuple_len(tuple);
if (tuple_size > 0) {
info->args = arg_copy(args_);
}
_VM_lock_init();
info->thread = pika_platform_thread_init(
"pika_thread", _thread_func, info, PIKA_THREAD_STACK_SIZE,
PIKA_THREAD_PRIO, PIKA_THREAD_TICK);
}