0777c1861e
While these warnings point to real errors in the code, PyMethodDef are an exception where functions with different numbers of arguments are all cast to the same function type.
88 lines
2.3 KiB
C++
88 lines
2.3 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup pythonintern
|
|
*
|
|
* This file inserts an exit callback into Python's 'atexit' module.
|
|
* Without this `sys.exit()` can crash because blender is not properly closing
|
|
* resources.
|
|
*/
|
|
|
|
#include <Python.h>
|
|
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "bpy.h" /* own include */
|
|
#include "bpy_capi_utils.h"
|
|
|
|
#include "WM_api.h"
|
|
|
|
static PyObject *bpy_atexit(PyObject * /*self*/, PyObject * /*args*/, PyObject * /*kw*/)
|
|
{
|
|
/* close down enough of blender at least not to crash */
|
|
bContext *C = BPY_context_get();
|
|
|
|
WM_exit_ex(C, false, false);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
#if (defined(__GNUC__) && !defined(__clang__))
|
|
# pragma GCC diagnostic push
|
|
# pragma GCC diagnostic ignored "-Wcast-function-type"
|
|
#endif
|
|
|
|
static PyMethodDef meth_bpy_atexit = {"bpy_atexit", (PyCFunction)bpy_atexit, METH_NOARGS, nullptr};
|
|
|
|
#if (defined(__GNUC__) && !defined(__clang__))
|
|
# pragma GCC diagnostic pop
|
|
#endif
|
|
|
|
static PyObject *func_bpy_atregister = nullptr; /* borrowed reference, `atexit` holds. */
|
|
|
|
static void atexit_func_call(const char *func_name, PyObject *atexit_func_arg)
|
|
{
|
|
/* NOTE(@ideasman42): no error checking, if any of these fail we'll get a crash
|
|
* this is intended, but if its problematic it could be changed. */
|
|
|
|
PyObject *atexit_mod = PyImport_ImportModuleLevel("atexit", nullptr, nullptr, nullptr, 0);
|
|
PyObject *atexit_func = PyObject_GetAttrString(atexit_mod, func_name);
|
|
PyObject *args = PyTuple_New(1);
|
|
PyObject *ret;
|
|
|
|
PyTuple_SET_ITEM(args, 0, atexit_func_arg);
|
|
Py_INCREF(atexit_func_arg); /* only incref so we don't dec'ref along with 'args' */
|
|
|
|
ret = PyObject_CallObject(atexit_func, args);
|
|
|
|
Py_DECREF(atexit_mod);
|
|
Py_DECREF(atexit_func);
|
|
Py_DECREF(args);
|
|
|
|
if (ret) {
|
|
Py_DECREF(ret);
|
|
}
|
|
else { /* should never happen */
|
|
PyErr_Print();
|
|
}
|
|
}
|
|
|
|
void BPY_atexit_register()
|
|
{
|
|
/* atexit module owns this new function reference */
|
|
BLI_assert(func_bpy_atregister == nullptr);
|
|
|
|
func_bpy_atregister = (PyObject *)PyCFunction_New(&meth_bpy_atexit, nullptr);
|
|
atexit_func_call("register", func_bpy_atregister);
|
|
}
|
|
|
|
void BPY_atexit_unregister()
|
|
{
|
|
BLI_assert(func_bpy_atregister != nullptr);
|
|
|
|
atexit_func_call("unregister", func_bpy_atregister);
|
|
func_bpy_atregister = nullptr; /* don't really need to set but just in case */
|
|
}
|