blueloveTH 2 лет назад
Родитель
Сommit
5cf6267f5b
2 измененных файлов с 8 добавлено и 3 удалено
  1. 7 0
      src/linalg.h
  2. 1 3
      src/pocketpy.h

+ 7 - 0
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 Mat3x3& obj){ return VAR_T(PyMat3x3, obj); }
 inline PyObject* py_var(VM* vm, const PyMat3x3& 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
 }   // namespace pkpy

+ 1 - 3
src/pocketpy.h

@@ -1053,9 +1053,7 @@ inline void VM::post_init(){
         add_module_requests(this);
         add_module_requests(this);
     }
     }
 
 
-    PyObject* linalg = new_module("linalg");
-    PyVec2::register_class(this, linalg);
-    PyMat3x3::register_class(this, linalg);
+    add_module_linalg(this);
 #endif
 #endif
 }
 }