f53df7da64
Code submissions have continually suffered from formatting inconsistencies that constantly have to be addressed. Using clang-format simplifies this by making code formatting more consistent, and allows automation of the code formatting so that maintainers can focus more on the code itself instead of code formatting.
150 lines
4.4 KiB
C
150 lines
4.4 KiB
C
/******************************************************************************
|
|
Copyright (C) 2017 by Hugh Bailey <jim@obsproject.com>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
******************************************************************************/
|
|
|
|
#include <util/dstr.h>
|
|
#include <util/platform.h>
|
|
|
|
#define NO_REDEFS
|
|
#include "obs-scripting-python-import.h"
|
|
#include "obs-scripting-config.h"
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma warning(disable : 4152)
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#define SO_EXT ".dll"
|
|
#elif __APPLE__
|
|
#define SO_EXT ".dylib"
|
|
#endif
|
|
|
|
bool import_python(const char *python_path)
|
|
{
|
|
struct dstr lib_path;
|
|
bool success = false;
|
|
void *lib;
|
|
|
|
if (!python_path)
|
|
python_path = "";
|
|
|
|
dstr_init_copy(&lib_path, python_path);
|
|
dstr_replace(&lib_path, "\\", "/");
|
|
if (!dstr_is_empty(&lib_path)) {
|
|
dstr_cat(&lib_path, "/");
|
|
}
|
|
dstr_cat(&lib_path, PYTHON_LIB SO_EXT);
|
|
|
|
lib = os_dlopen(lib_path.array);
|
|
if (!lib) {
|
|
blog(LOG_WARNING, "[Python] Could not load library: %s",
|
|
lib_path.array);
|
|
goto fail;
|
|
}
|
|
|
|
#define IMPORT_FUNC(x) \
|
|
do { \
|
|
Import_##x = os_dlsym(lib, #x); \
|
|
if (!Import_##x) { \
|
|
blog(LOG_WARNING, "[Python] Failed to import: %s", \
|
|
#x); \
|
|
goto fail; \
|
|
} \
|
|
} while (false)
|
|
|
|
IMPORT_FUNC(PyType_Ready);
|
|
IMPORT_FUNC(PyObject_GenericGetAttr);
|
|
IMPORT_FUNC(PyObject_IsTrue);
|
|
IMPORT_FUNC(Py_DecRef);
|
|
IMPORT_FUNC(PyObject_Malloc);
|
|
IMPORT_FUNC(PyObject_Free);
|
|
IMPORT_FUNC(PyObject_Init);
|
|
IMPORT_FUNC(PyUnicode_FromFormat);
|
|
IMPORT_FUNC(PyUnicode_Concat);
|
|
IMPORT_FUNC(PyLong_FromVoidPtr);
|
|
IMPORT_FUNC(PyBool_FromLong);
|
|
IMPORT_FUNC(PyGILState_Ensure);
|
|
IMPORT_FUNC(PyGILState_GetThisThreadState);
|
|
IMPORT_FUNC(PyErr_SetString);
|
|
IMPORT_FUNC(PyErr_Occurred);
|
|
IMPORT_FUNC(PyErr_Fetch);
|
|
IMPORT_FUNC(PyErr_Restore);
|
|
IMPORT_FUNC(PyErr_WriteUnraisable);
|
|
IMPORT_FUNC(PyArg_UnpackTuple);
|
|
IMPORT_FUNC(Py_BuildValue);
|
|
IMPORT_FUNC(PyRun_SimpleStringFlags);
|
|
IMPORT_FUNC(PyErr_Print);
|
|
IMPORT_FUNC(Py_SetPythonHome);
|
|
IMPORT_FUNC(Py_Initialize);
|
|
IMPORT_FUNC(Py_Finalize);
|
|
IMPORT_FUNC(Py_IsInitialized);
|
|
IMPORT_FUNC(PyEval_InitThreads);
|
|
IMPORT_FUNC(PyEval_ThreadsInitialized);
|
|
IMPORT_FUNC(PyEval_ReleaseThread);
|
|
IMPORT_FUNC(PySys_SetArgv);
|
|
IMPORT_FUNC(PyImport_ImportModule);
|
|
IMPORT_FUNC(PyObject_CallFunctionObjArgs);
|
|
IMPORT_FUNC(_Py_NotImplementedStruct);
|
|
IMPORT_FUNC(PyExc_TypeError);
|
|
IMPORT_FUNC(PyExc_RuntimeError);
|
|
IMPORT_FUNC(PyObject_GetAttr);
|
|
IMPORT_FUNC(PyUnicode_FromString);
|
|
IMPORT_FUNC(PyDict_GetItemString);
|
|
IMPORT_FUNC(PyDict_SetItemString);
|
|
IMPORT_FUNC(PyCFunction_NewEx);
|
|
IMPORT_FUNC(PyModule_GetDict);
|
|
IMPORT_FUNC(PyModule_GetNameObject);
|
|
IMPORT_FUNC(PyModule_AddObject);
|
|
IMPORT_FUNC(PyModule_AddStringConstant);
|
|
IMPORT_FUNC(PyImport_Import);
|
|
IMPORT_FUNC(PyObject_CallObject);
|
|
IMPORT_FUNC(_Py_FalseStruct);
|
|
IMPORT_FUNC(_Py_TrueStruct);
|
|
IMPORT_FUNC(PyGILState_Release);
|
|
IMPORT_FUNC(PyList_Append);
|
|
IMPORT_FUNC(PySys_GetObject);
|
|
IMPORT_FUNC(PyImport_ReloadModule);
|
|
IMPORT_FUNC(PyObject_GetAttrString);
|
|
IMPORT_FUNC(PyCapsule_New);
|
|
IMPORT_FUNC(PyCapsule_GetPointer);
|
|
IMPORT_FUNC(PyArg_ParseTuple);
|
|
IMPORT_FUNC(PyFunction_Type);
|
|
IMPORT_FUNC(PyObject_SetAttr);
|
|
IMPORT_FUNC(_PyObject_New);
|
|
IMPORT_FUNC(PyCapsule_Import);
|
|
IMPORT_FUNC(PyErr_Clear);
|
|
IMPORT_FUNC(PyObject_Call);
|
|
IMPORT_FUNC(PyList_New);
|
|
IMPORT_FUNC(PyList_Size);
|
|
IMPORT_FUNC(PyList_GetItem);
|
|
IMPORT_FUNC(PyUnicode_AsUTF8String);
|
|
IMPORT_FUNC(PyLong_FromUnsignedLongLong);
|
|
IMPORT_FUNC(PyArg_VaParse);
|
|
IMPORT_FUNC(_Py_NoneStruct);
|
|
|
|
#undef IMPORT_FUNC
|
|
|
|
success = true;
|
|
|
|
fail:
|
|
if (!success && lib)
|
|
os_dlclose(lib);
|
|
|
|
dstr_free(&lib_path);
|
|
|
|
return success;
|
|
}
|