From 5cf6267f5b94ca0c41ab7cb73f66cbbfbb11ec6e Mon Sep 17 00:00:00 2001 From: blueloveTH Date: Fri, 12 May 2023 14:04:59 +0800 Subject: [PATCH] ... --- src/linalg.h | 7 +++++++ src/pocketpy.h | 4 +--- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/src/linalg.h b/src/linalg.h index 543c11ca..802ec484 100644 --- a/src/linalg.h +++ b/src/linalg.h @@ -694,4 +694,11 @@ inline PyObject* py_var(VM* vm, const PyVec3& obj){ return VAR_T(PyVec3, obj);} inline PyObject* py_var(VM* vm, const Mat3x3& obj){ return VAR_T(PyMat3x3, obj); } inline PyObject* py_var(VM* vm, const PyMat3x3& obj){ return VAR_T(PyMat3x3, obj); } +inline void add_module_linalg(VM* vm){ + PyObject* linalg = vm->new_module("linalg"); + PyVec2::register_class(vm, linalg); + PyVec3::register_class(vm, linalg); + PyMat3x3::register_class(vm, linalg); +} + } // namespace pkpy \ No newline at end of file diff --git a/src/pocketpy.h b/src/pocketpy.h index 870a85b2..36c89a06 100644 --- a/src/pocketpy.h +++ b/src/pocketpy.h @@ -1053,9 +1053,7 @@ inline void VM::post_init(){ add_module_requests(this); } - PyObject* linalg = new_module("linalg"); - PyVec2::register_class(this, linalg); - PyMat3x3::register_class(this, linalg); + add_module_linalg(this); #endif }