1071d4279SBram Moolenaar /* vi:set ts=8 sts=4 sw=4: 2071d4279SBram Moolenaar * 3071d4279SBram Moolenaar * VIM - Vi IMproved by Bram Moolenaar 4071d4279SBram Moolenaar * 5071d4279SBram Moolenaar * Do ":help uganda" in Vim to read copying and usage conditions. 6071d4279SBram Moolenaar * Do ":help credits" in Vim to see a list of people who contributed. 7071d4279SBram Moolenaar * See README.txt for an overview of the Vim source code. 8071d4279SBram Moolenaar */ 9071d4279SBram Moolenaar /* 10071d4279SBram Moolenaar * Python extensions by Paul Moore. 11071d4279SBram Moolenaar * Changes for Unix by David Leonard. 12071d4279SBram Moolenaar * 13071d4279SBram Moolenaar * This consists of four parts: 14071d4279SBram Moolenaar * 1. Python interpreter main program 15071d4279SBram Moolenaar * 2. Python output stream: writes output via [e]msg(). 16071d4279SBram Moolenaar * 3. Implementation of the Vim module for Python 17071d4279SBram Moolenaar * 4. Utility functions for handling the interface between Vim and Python. 18071d4279SBram Moolenaar */ 19071d4279SBram Moolenaar 20071d4279SBram Moolenaar #include "vim.h" 21071d4279SBram Moolenaar 22071d4279SBram Moolenaar #include <limits.h> 23071d4279SBram Moolenaar 24071d4279SBram Moolenaar /* Python.h defines _POSIX_THREADS itself (if needed) */ 25071d4279SBram Moolenaar #ifdef _POSIX_THREADS 26071d4279SBram Moolenaar # undef _POSIX_THREADS 27071d4279SBram Moolenaar #endif 28071d4279SBram Moolenaar 29071d4279SBram Moolenaar #if defined(_WIN32) && defined(HAVE_FCNTL_H) 30071d4279SBram Moolenaar # undef HAVE_FCNTL_H 31071d4279SBram Moolenaar #endif 32071d4279SBram Moolenaar 33071d4279SBram Moolenaar #ifdef _DEBUG 34071d4279SBram Moolenaar # undef _DEBUG 35071d4279SBram Moolenaar #endif 36071d4279SBram Moolenaar 37071d4279SBram Moolenaar #ifdef HAVE_STDARG_H 38071d4279SBram Moolenaar # undef HAVE_STDARG_H /* Python's config.h defines it as well. */ 39071d4279SBram Moolenaar #endif 40be2c9ae9SBram Moolenaar #ifdef _POSIX_C_SOURCE 41be2c9ae9SBram Moolenaar # undef _POSIX_C_SOURCE /* pyconfig.h defines it as well. */ 42be2c9ae9SBram Moolenaar #endif 43be2c9ae9SBram Moolenaar #ifdef _XOPEN_SOURCE 44be2c9ae9SBram Moolenaar # undef _XOPEN_SOURCE /* pyconfig.h defines it as well. */ 45be2c9ae9SBram Moolenaar #endif 46071d4279SBram Moolenaar 472c45e945SBram Moolenaar #define PY_SSIZE_T_CLEAN 482c45e945SBram Moolenaar 49071d4279SBram Moolenaar #include <Python.h> 50071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 51071d4279SBram Moolenaar # include "macglue.h" 52071d4279SBram Moolenaar # include <CodeFragments.h> 53071d4279SBram Moolenaar #endif 54071d4279SBram Moolenaar #undef main /* Defined in python.h - aargh */ 55071d4279SBram Moolenaar #undef HAVE_FCNTL_H /* Clash with os_win32.h */ 56071d4279SBram Moolenaar 57170bf1aeSBram Moolenaar static void init_structs(void); 58170bf1aeSBram Moolenaar 59*19e60943SBram Moolenaar /* No-op conversion functions, use with care! */ 60*19e60943SBram Moolenaar #define PyString_AsBytes(obj) (obj) 61*19e60943SBram Moolenaar #define PyString_FreeBytes(obj) 62*19e60943SBram Moolenaar 63071d4279SBram Moolenaar #if !defined(FEAT_PYTHON) && defined(PROTO) 64071d4279SBram Moolenaar /* Use this to be able to generate prototypes without python being used. */ 65e7cb9cf6SBram Moolenaar # define PyObject Py_ssize_t 66e7cb9cf6SBram Moolenaar # define PyThreadState Py_ssize_t 67e7cb9cf6SBram Moolenaar # define PyTypeObject Py_ssize_t 68e7cb9cf6SBram Moolenaar struct PyMethodDef { Py_ssize_t a; }; 69e7cb9cf6SBram Moolenaar # define PySequenceMethods Py_ssize_t 70071d4279SBram Moolenaar #endif 71071d4279SBram Moolenaar 722c45e945SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000 732c45e945SBram Moolenaar # define PyInt Py_ssize_t 742c45e945SBram Moolenaar # define PyInquiry lenfunc 752c45e945SBram Moolenaar # define PyIntArgFunc ssizeargfunc 762c45e945SBram Moolenaar # define PyIntIntArgFunc ssizessizeargfunc 772c45e945SBram Moolenaar # define PyIntObjArgProc ssizeobjargproc 782c45e945SBram Moolenaar # define PyIntIntObjArgProc ssizessizeobjargproc 79e7cb9cf6SBram Moolenaar # define Py_ssize_t_fmt "n" 802c45e945SBram Moolenaar #else 812c45e945SBram Moolenaar # define PyInt int 822c45e945SBram Moolenaar # define PyInquiry inquiry 832c45e945SBram Moolenaar # define PyIntArgFunc intargfunc 842c45e945SBram Moolenaar # define PyIntIntArgFunc intintargfunc 852c45e945SBram Moolenaar # define PyIntObjArgProc intobjargproc 862c45e945SBram Moolenaar # define PyIntIntObjArgProc intintobjargproc 87e7cb9cf6SBram Moolenaar # define Py_ssize_t_fmt "i" 882c45e945SBram Moolenaar #endif 892c45e945SBram Moolenaar 90071d4279SBram Moolenaar /* Parser flags */ 91071d4279SBram Moolenaar #define single_input 256 92071d4279SBram Moolenaar #define file_input 257 93071d4279SBram Moolenaar #define eval_input 258 94071d4279SBram Moolenaar 95071d4279SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x020300F0 96071d4279SBram Moolenaar /* Python 2.3: can invoke ":python" recursively. */ 97071d4279SBram Moolenaar # define PY_CAN_RECURSE 98071d4279SBram Moolenaar #endif 99071d4279SBram Moolenaar 100071d4279SBram Moolenaar # if defined(DYNAMIC_PYTHON) || defined(PROTO) 101071d4279SBram Moolenaar # ifndef DYNAMIC_PYTHON 102e7cb9cf6SBram Moolenaar # define HINSTANCE long_u /* for generating prototypes */ 103071d4279SBram Moolenaar # endif 104071d4279SBram Moolenaar 105fa5d1e63SBram Moolenaar # ifndef WIN3264 106bd5e15fdSBram Moolenaar # include <dlfcn.h> 107bd5e15fdSBram Moolenaar # define FARPROC void* 108bd5e15fdSBram Moolenaar # define HINSTANCE void* 109644d37b8SBram Moolenaar # if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL) 110b61f95c3SBram Moolenaar # define load_dll(n) dlopen((n), RTLD_LAZY) 111b61f95c3SBram Moolenaar # else 112fa5d1e63SBram Moolenaar # define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL) 113b61f95c3SBram Moolenaar # endif 114bd5e15fdSBram Moolenaar # define close_dll dlclose 115bd5e15fdSBram Moolenaar # define symbol_from_dll dlsym 116bd5e15fdSBram Moolenaar # else 117ebbcb824SBram Moolenaar # define load_dll vimLoadLib 118bd5e15fdSBram Moolenaar # define close_dll FreeLibrary 119bd5e15fdSBram Moolenaar # define symbol_from_dll GetProcAddress 120bd5e15fdSBram Moolenaar # endif 121bd5e15fdSBram Moolenaar 122e7cb9cf6SBram Moolenaar /* This makes if_python.c compile without warnings against Python 2.5 123e7cb9cf6SBram Moolenaar * on Win32 and Win64. */ 124e7cb9cf6SBram Moolenaar # undef PyRun_SimpleString 125e7cb9cf6SBram Moolenaar # undef PyArg_Parse 126e7cb9cf6SBram Moolenaar # undef PyArg_ParseTuple 127e7cb9cf6SBram Moolenaar # undef Py_BuildValue 128e7cb9cf6SBram Moolenaar # undef Py_InitModule4 129e7cb9cf6SBram Moolenaar # undef Py_InitModule4_64 130e7cb9cf6SBram Moolenaar 131071d4279SBram Moolenaar /* 132071d4279SBram Moolenaar * Wrapper defines 133071d4279SBram Moolenaar */ 134071d4279SBram Moolenaar # define PyArg_Parse dll_PyArg_Parse 135071d4279SBram Moolenaar # define PyArg_ParseTuple dll_PyArg_ParseTuple 136*19e60943SBram Moolenaar # define PyMem_Free dll_PyMem_Free 137071d4279SBram Moolenaar # define PyDict_SetItemString dll_PyDict_SetItemString 138071d4279SBram Moolenaar # define PyErr_BadArgument dll_PyErr_BadArgument 139071d4279SBram Moolenaar # define PyErr_Clear dll_PyErr_Clear 140071d4279SBram Moolenaar # define PyErr_NoMemory dll_PyErr_NoMemory 141071d4279SBram Moolenaar # define PyErr_Occurred dll_PyErr_Occurred 142071d4279SBram Moolenaar # define PyErr_SetNone dll_PyErr_SetNone 143071d4279SBram Moolenaar # define PyErr_SetString dll_PyErr_SetString 144071d4279SBram Moolenaar # define PyEval_InitThreads dll_PyEval_InitThreads 145071d4279SBram Moolenaar # define PyEval_RestoreThread dll_PyEval_RestoreThread 146071d4279SBram Moolenaar # define PyEval_SaveThread dll_PyEval_SaveThread 147071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE 148071d4279SBram Moolenaar # define PyGILState_Ensure dll_PyGILState_Ensure 149071d4279SBram Moolenaar # define PyGILState_Release dll_PyGILState_Release 150071d4279SBram Moolenaar # endif 151071d4279SBram Moolenaar # define PyInt_AsLong dll_PyInt_AsLong 152071d4279SBram Moolenaar # define PyInt_FromLong dll_PyInt_FromLong 153071d4279SBram Moolenaar # define PyInt_Type (*dll_PyInt_Type) 154071d4279SBram Moolenaar # define PyList_GetItem dll_PyList_GetItem 1550ac9379aSBram Moolenaar # define PyList_Append dll_PyList_Append 156071d4279SBram Moolenaar # define PyList_New dll_PyList_New 157071d4279SBram Moolenaar # define PyList_SetItem dll_PyList_SetItem 158071d4279SBram Moolenaar # define PyList_Size dll_PyList_Size 159071d4279SBram Moolenaar # define PyList_Type (*dll_PyList_Type) 160071d4279SBram Moolenaar # define PyImport_ImportModule dll_PyImport_ImportModule 1610ac9379aSBram Moolenaar # define PyDict_New dll_PyDict_New 162071d4279SBram Moolenaar # define PyDict_GetItemString dll_PyDict_GetItemString 163071d4279SBram Moolenaar # define PyModule_GetDict dll_PyModule_GetDict 164071d4279SBram Moolenaar # define PyRun_SimpleString dll_PyRun_SimpleString 165071d4279SBram Moolenaar # define PyString_AsString dll_PyString_AsString 166071d4279SBram Moolenaar # define PyString_FromString dll_PyString_FromString 167071d4279SBram Moolenaar # define PyString_FromStringAndSize dll_PyString_FromStringAndSize 168071d4279SBram Moolenaar # define PyString_Size dll_PyString_Size 169071d4279SBram Moolenaar # define PyString_Type (*dll_PyString_Type) 170071d4279SBram Moolenaar # define PySys_SetObject dll_PySys_SetObject 171071d4279SBram Moolenaar # define PySys_SetArgv dll_PySys_SetArgv 172071d4279SBram Moolenaar # define PyType_Type (*dll_PyType_Type) 17330fec7bcSBram Moolenaar # define PyType_Ready (*dll_PyType_Ready) 174071d4279SBram Moolenaar # define Py_BuildValue dll_Py_BuildValue 175071d4279SBram Moolenaar # define Py_FindMethod dll_Py_FindMethod 176071d4279SBram Moolenaar # define Py_InitModule4 dll_Py_InitModule4 177644d37b8SBram Moolenaar # define Py_SetPythonHome dll_Py_SetPythonHome 178071d4279SBram Moolenaar # define Py_Initialize dll_Py_Initialize 1790e21a3f6SBram Moolenaar # define Py_Finalize dll_Py_Finalize 1800e21a3f6SBram Moolenaar # define Py_IsInitialized dll_Py_IsInitialized 181071d4279SBram Moolenaar # define _PyObject_New dll__PyObject_New 182071d4279SBram Moolenaar # define _Py_NoneStruct (*dll__Py_NoneStruct) 183071d4279SBram Moolenaar # define PyObject_Init dll__PyObject_Init 184071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000 185071d4279SBram Moolenaar # define PyType_IsSubtype dll_PyType_IsSubtype 186071d4279SBram Moolenaar # endif 187071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000 188071d4279SBram Moolenaar # define PyObject_Malloc dll_PyObject_Malloc 189071d4279SBram Moolenaar # define PyObject_Free dll_PyObject_Free 190071d4279SBram Moolenaar # endif 191071d4279SBram Moolenaar 192071d4279SBram Moolenaar /* 193071d4279SBram Moolenaar * Pointers for dynamic link 194071d4279SBram Moolenaar */ 195071d4279SBram Moolenaar static int(*dll_PyArg_Parse)(PyObject *, char *, ...); 196071d4279SBram Moolenaar static int(*dll_PyArg_ParseTuple)(PyObject *, char *, ...); 197*19e60943SBram Moolenaar static int(*dll_PyMem_Free)(void *); 198071d4279SBram Moolenaar static int(*dll_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item); 199071d4279SBram Moolenaar static int(*dll_PyErr_BadArgument)(void); 200071d4279SBram Moolenaar static void(*dll_PyErr_Clear)(void); 201071d4279SBram Moolenaar static PyObject*(*dll_PyErr_NoMemory)(void); 202071d4279SBram Moolenaar static PyObject*(*dll_PyErr_Occurred)(void); 203071d4279SBram Moolenaar static void(*dll_PyErr_SetNone)(PyObject *); 204071d4279SBram Moolenaar static void(*dll_PyErr_SetString)(PyObject *, const char *); 205071d4279SBram Moolenaar static void(*dll_PyEval_InitThreads)(void); 206071d4279SBram Moolenaar static void(*dll_PyEval_RestoreThread)(PyThreadState *); 207071d4279SBram Moolenaar static PyThreadState*(*dll_PyEval_SaveThread)(void); 208071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE 209071d4279SBram Moolenaar static PyGILState_STATE (*dll_PyGILState_Ensure)(void); 210071d4279SBram Moolenaar static void (*dll_PyGILState_Release)(PyGILState_STATE); 211071d4279SBram Moolenaar #endif 212071d4279SBram Moolenaar static long(*dll_PyInt_AsLong)(PyObject *); 213071d4279SBram Moolenaar static PyObject*(*dll_PyInt_FromLong)(long); 214071d4279SBram Moolenaar static PyTypeObject* dll_PyInt_Type; 2152c45e945SBram Moolenaar static PyObject*(*dll_PyList_GetItem)(PyObject *, PyInt); 2160ac9379aSBram Moolenaar static PyObject*(*dll_PyList_Append)(PyObject *, PyObject *); 2172c45e945SBram Moolenaar static PyObject*(*dll_PyList_New)(PyInt size); 2182c45e945SBram Moolenaar static int(*dll_PyList_SetItem)(PyObject *, PyInt, PyObject *); 2192c45e945SBram Moolenaar static PyInt(*dll_PyList_Size)(PyObject *); 220071d4279SBram Moolenaar static PyTypeObject* dll_PyList_Type; 221071d4279SBram Moolenaar static PyObject*(*dll_PyImport_ImportModule)(const char *); 2220ac9379aSBram Moolenaar static PyObject*(*dll_PyDict_New)(void); 223071d4279SBram Moolenaar static PyObject*(*dll_PyDict_GetItemString)(PyObject *, const char *); 224071d4279SBram Moolenaar static PyObject*(*dll_PyModule_GetDict)(PyObject *); 225071d4279SBram Moolenaar static int(*dll_PyRun_SimpleString)(char *); 226071d4279SBram Moolenaar static char*(*dll_PyString_AsString)(PyObject *); 227071d4279SBram Moolenaar static PyObject*(*dll_PyString_FromString)(const char *); 2282c45e945SBram Moolenaar static PyObject*(*dll_PyString_FromStringAndSize)(const char *, PyInt); 2292c45e945SBram Moolenaar static PyInt(*dll_PyString_Size)(PyObject *); 230071d4279SBram Moolenaar static PyTypeObject* dll_PyString_Type; 231071d4279SBram Moolenaar static int(*dll_PySys_SetObject)(char *, PyObject *); 232071d4279SBram Moolenaar static int(*dll_PySys_SetArgv)(int, char **); 233071d4279SBram Moolenaar static PyTypeObject* dll_PyType_Type; 23430fec7bcSBram Moolenaar static int (*dll_PyType_Ready)(PyTypeObject *type); 235071d4279SBram Moolenaar static PyObject*(*dll_Py_BuildValue)(char *, ...); 236071d4279SBram Moolenaar static PyObject*(*dll_Py_FindMethod)(struct PyMethodDef[], PyObject *, char *); 237071d4279SBram Moolenaar static PyObject*(*dll_Py_InitModule4)(char *, struct PyMethodDef *, char *, PyObject *, int); 238644d37b8SBram Moolenaar static void(*dll_Py_SetPythonHome)(char *home); 239071d4279SBram Moolenaar static void(*dll_Py_Initialize)(void); 2400e21a3f6SBram Moolenaar static void(*dll_Py_Finalize)(void); 2410e21a3f6SBram Moolenaar static int(*dll_Py_IsInitialized)(void); 242071d4279SBram Moolenaar static PyObject*(*dll__PyObject_New)(PyTypeObject *, PyObject *); 243071d4279SBram Moolenaar static PyObject*(*dll__PyObject_Init)(PyObject *, PyTypeObject *); 244071d4279SBram Moolenaar static PyObject* dll__Py_NoneStruct; 245071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000 246071d4279SBram Moolenaar static int (*dll_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *); 247071d4279SBram Moolenaar # endif 248071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000 249071d4279SBram Moolenaar static void* (*dll_PyObject_Malloc)(size_t); 250071d4279SBram Moolenaar static void (*dll_PyObject_Free)(void*); 251071d4279SBram Moolenaar # endif 252071d4279SBram Moolenaar 253071d4279SBram Moolenaar static HINSTANCE hinstPython = 0; /* Instance of python.dll */ 254071d4279SBram Moolenaar 255071d4279SBram Moolenaar /* Imported exception objects */ 256071d4279SBram Moolenaar static PyObject *imp_PyExc_AttributeError; 257071d4279SBram Moolenaar static PyObject *imp_PyExc_IndexError; 258071d4279SBram Moolenaar static PyObject *imp_PyExc_KeyboardInterrupt; 259071d4279SBram Moolenaar static PyObject *imp_PyExc_TypeError; 260071d4279SBram Moolenaar static PyObject *imp_PyExc_ValueError; 261071d4279SBram Moolenaar 262071d4279SBram Moolenaar # define PyExc_AttributeError imp_PyExc_AttributeError 263071d4279SBram Moolenaar # define PyExc_IndexError imp_PyExc_IndexError 264071d4279SBram Moolenaar # define PyExc_KeyboardInterrupt imp_PyExc_KeyboardInterrupt 265071d4279SBram Moolenaar # define PyExc_TypeError imp_PyExc_TypeError 266071d4279SBram Moolenaar # define PyExc_ValueError imp_PyExc_ValueError 267071d4279SBram Moolenaar 268071d4279SBram Moolenaar /* 269071d4279SBram Moolenaar * Table of name to function pointer of python. 270071d4279SBram Moolenaar */ 271071d4279SBram Moolenaar # define PYTHON_PROC FARPROC 272071d4279SBram Moolenaar static struct 273071d4279SBram Moolenaar { 274071d4279SBram Moolenaar char *name; 275071d4279SBram Moolenaar PYTHON_PROC *ptr; 276071d4279SBram Moolenaar } python_funcname_table[] = 277071d4279SBram Moolenaar { 278071d4279SBram Moolenaar {"PyArg_Parse", (PYTHON_PROC*)&dll_PyArg_Parse}, 279071d4279SBram Moolenaar {"PyArg_ParseTuple", (PYTHON_PROC*)&dll_PyArg_ParseTuple}, 280*19e60943SBram Moolenaar {"PyMem_Free", (PYTHON_PROC*)&dll_PyMem_Free}, 281071d4279SBram Moolenaar {"PyDict_SetItemString", (PYTHON_PROC*)&dll_PyDict_SetItemString}, 282071d4279SBram Moolenaar {"PyErr_BadArgument", (PYTHON_PROC*)&dll_PyErr_BadArgument}, 283071d4279SBram Moolenaar {"PyErr_Clear", (PYTHON_PROC*)&dll_PyErr_Clear}, 284071d4279SBram Moolenaar {"PyErr_NoMemory", (PYTHON_PROC*)&dll_PyErr_NoMemory}, 285071d4279SBram Moolenaar {"PyErr_Occurred", (PYTHON_PROC*)&dll_PyErr_Occurred}, 286071d4279SBram Moolenaar {"PyErr_SetNone", (PYTHON_PROC*)&dll_PyErr_SetNone}, 287071d4279SBram Moolenaar {"PyErr_SetString", (PYTHON_PROC*)&dll_PyErr_SetString}, 288071d4279SBram Moolenaar {"PyEval_InitThreads", (PYTHON_PROC*)&dll_PyEval_InitThreads}, 289071d4279SBram Moolenaar {"PyEval_RestoreThread", (PYTHON_PROC*)&dll_PyEval_RestoreThread}, 290071d4279SBram Moolenaar {"PyEval_SaveThread", (PYTHON_PROC*)&dll_PyEval_SaveThread}, 291071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE 292071d4279SBram Moolenaar {"PyGILState_Ensure", (PYTHON_PROC*)&dll_PyGILState_Ensure}, 293071d4279SBram Moolenaar {"PyGILState_Release", (PYTHON_PROC*)&dll_PyGILState_Release}, 294071d4279SBram Moolenaar # endif 295071d4279SBram Moolenaar {"PyInt_AsLong", (PYTHON_PROC*)&dll_PyInt_AsLong}, 296071d4279SBram Moolenaar {"PyInt_FromLong", (PYTHON_PROC*)&dll_PyInt_FromLong}, 297071d4279SBram Moolenaar {"PyInt_Type", (PYTHON_PROC*)&dll_PyInt_Type}, 298071d4279SBram Moolenaar {"PyList_GetItem", (PYTHON_PROC*)&dll_PyList_GetItem}, 2990ac9379aSBram Moolenaar {"PyList_Append", (PYTHON_PROC*)&dll_PyList_Append}, 300071d4279SBram Moolenaar {"PyList_New", (PYTHON_PROC*)&dll_PyList_New}, 301071d4279SBram Moolenaar {"PyList_SetItem", (PYTHON_PROC*)&dll_PyList_SetItem}, 302071d4279SBram Moolenaar {"PyList_Size", (PYTHON_PROC*)&dll_PyList_Size}, 303071d4279SBram Moolenaar {"PyList_Type", (PYTHON_PROC*)&dll_PyList_Type}, 304071d4279SBram Moolenaar {"PyImport_ImportModule", (PYTHON_PROC*)&dll_PyImport_ImportModule}, 305071d4279SBram Moolenaar {"PyDict_GetItemString", (PYTHON_PROC*)&dll_PyDict_GetItemString}, 3060ac9379aSBram Moolenaar {"PyDict_New", (PYTHON_PROC*)&dll_PyDict_New}, 307071d4279SBram Moolenaar {"PyModule_GetDict", (PYTHON_PROC*)&dll_PyModule_GetDict}, 308071d4279SBram Moolenaar {"PyRun_SimpleString", (PYTHON_PROC*)&dll_PyRun_SimpleString}, 309071d4279SBram Moolenaar {"PyString_AsString", (PYTHON_PROC*)&dll_PyString_AsString}, 310071d4279SBram Moolenaar {"PyString_FromString", (PYTHON_PROC*)&dll_PyString_FromString}, 311071d4279SBram Moolenaar {"PyString_FromStringAndSize", (PYTHON_PROC*)&dll_PyString_FromStringAndSize}, 312071d4279SBram Moolenaar {"PyString_Size", (PYTHON_PROC*)&dll_PyString_Size}, 313071d4279SBram Moolenaar {"PyString_Type", (PYTHON_PROC*)&dll_PyString_Type}, 314071d4279SBram Moolenaar {"PySys_SetObject", (PYTHON_PROC*)&dll_PySys_SetObject}, 315071d4279SBram Moolenaar {"PySys_SetArgv", (PYTHON_PROC*)&dll_PySys_SetArgv}, 316071d4279SBram Moolenaar {"PyType_Type", (PYTHON_PROC*)&dll_PyType_Type}, 31730fec7bcSBram Moolenaar {"PyType_Ready", (PYTHON_PROC*)&dll_PyType_Ready}, 318071d4279SBram Moolenaar {"Py_BuildValue", (PYTHON_PROC*)&dll_Py_BuildValue}, 319071d4279SBram Moolenaar {"Py_FindMethod", (PYTHON_PROC*)&dll_Py_FindMethod}, 320e7cb9cf6SBram Moolenaar # if (PY_VERSION_HEX >= 0x02050000) && SIZEOF_SIZE_T != SIZEOF_INT 321e7cb9cf6SBram Moolenaar {"Py_InitModule4_64", (PYTHON_PROC*)&dll_Py_InitModule4}, 322e7cb9cf6SBram Moolenaar # else 323071d4279SBram Moolenaar {"Py_InitModule4", (PYTHON_PROC*)&dll_Py_InitModule4}, 324e7cb9cf6SBram Moolenaar # endif 325644d37b8SBram Moolenaar {"Py_SetPythonHome", (PYTHON_PROC*)&dll_Py_SetPythonHome}, 326071d4279SBram Moolenaar {"Py_Initialize", (PYTHON_PROC*)&dll_Py_Initialize}, 3270e21a3f6SBram Moolenaar {"Py_Finalize", (PYTHON_PROC*)&dll_Py_Finalize}, 3280e21a3f6SBram Moolenaar {"Py_IsInitialized", (PYTHON_PROC*)&dll_Py_IsInitialized}, 329071d4279SBram Moolenaar {"_PyObject_New", (PYTHON_PROC*)&dll__PyObject_New}, 330071d4279SBram Moolenaar {"PyObject_Init", (PYTHON_PROC*)&dll__PyObject_Init}, 331071d4279SBram Moolenaar {"_Py_NoneStruct", (PYTHON_PROC*)&dll__Py_NoneStruct}, 332071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000 333071d4279SBram Moolenaar {"PyType_IsSubtype", (PYTHON_PROC*)&dll_PyType_IsSubtype}, 334071d4279SBram Moolenaar # endif 335071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000 336071d4279SBram Moolenaar {"PyObject_Malloc", (PYTHON_PROC*)&dll_PyObject_Malloc}, 337071d4279SBram Moolenaar {"PyObject_Free", (PYTHON_PROC*)&dll_PyObject_Free}, 338071d4279SBram Moolenaar # endif 339071d4279SBram Moolenaar {"", NULL}, 340071d4279SBram Moolenaar }; 341071d4279SBram Moolenaar 342071d4279SBram Moolenaar /* 343071d4279SBram Moolenaar * Free python.dll 344071d4279SBram Moolenaar */ 345071d4279SBram Moolenaar static void 346071d4279SBram Moolenaar end_dynamic_python(void) 347071d4279SBram Moolenaar { 348071d4279SBram Moolenaar if (hinstPython) 349071d4279SBram Moolenaar { 350bd5e15fdSBram Moolenaar close_dll(hinstPython); 351071d4279SBram Moolenaar hinstPython = 0; 352071d4279SBram Moolenaar } 353071d4279SBram Moolenaar } 354071d4279SBram Moolenaar 355071d4279SBram Moolenaar /* 356071d4279SBram Moolenaar * Load library and get all pointers. 357071d4279SBram Moolenaar * Parameter 'libname' provides name of DLL. 358071d4279SBram Moolenaar * Return OK or FAIL. 359071d4279SBram Moolenaar */ 360071d4279SBram Moolenaar static int 361071d4279SBram Moolenaar python_runtime_link_init(char *libname, int verbose) 362071d4279SBram Moolenaar { 363071d4279SBram Moolenaar int i; 364071d4279SBram Moolenaar 365644d37b8SBram Moolenaar #if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON3) 366b744b2faSBram Moolenaar /* Can't have Python and Python3 loaded at the same time. 367b744b2faSBram Moolenaar * It cause a crash, because RTLD_GLOBAL is needed for 368b744b2faSBram Moolenaar * standard C extension libraries of one or both python versions. */ 3694c3a326cSBram Moolenaar if (python3_loaded()) 3704c3a326cSBram Moolenaar { 371b744b2faSBram Moolenaar EMSG(_("E836: This Vim cannot execute :python after using :py3")); 3724c3a326cSBram Moolenaar return FAIL; 3734c3a326cSBram Moolenaar } 3744c3a326cSBram Moolenaar #endif 3754c3a326cSBram Moolenaar 376071d4279SBram Moolenaar if (hinstPython) 377071d4279SBram Moolenaar return OK; 378bd5e15fdSBram Moolenaar hinstPython = load_dll(libname); 379071d4279SBram Moolenaar if (!hinstPython) 380071d4279SBram Moolenaar { 381071d4279SBram Moolenaar if (verbose) 382071d4279SBram Moolenaar EMSG2(_(e_loadlib), libname); 383071d4279SBram Moolenaar return FAIL; 384071d4279SBram Moolenaar } 385071d4279SBram Moolenaar 386071d4279SBram Moolenaar for (i = 0; python_funcname_table[i].ptr; ++i) 387071d4279SBram Moolenaar { 388bd5e15fdSBram Moolenaar if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython, 389071d4279SBram Moolenaar python_funcname_table[i].name)) == NULL) 390071d4279SBram Moolenaar { 391bd5e15fdSBram Moolenaar close_dll(hinstPython); 392071d4279SBram Moolenaar hinstPython = 0; 393071d4279SBram Moolenaar if (verbose) 394071d4279SBram Moolenaar EMSG2(_(e_loadfunc), python_funcname_table[i].name); 395071d4279SBram Moolenaar return FAIL; 396071d4279SBram Moolenaar } 397071d4279SBram Moolenaar } 398071d4279SBram Moolenaar return OK; 399071d4279SBram Moolenaar } 400071d4279SBram Moolenaar 401071d4279SBram Moolenaar /* 402071d4279SBram Moolenaar * If python is enabled (there is installed python on Windows system) return 403071d4279SBram Moolenaar * TRUE, else FALSE. 404071d4279SBram Moolenaar */ 405071d4279SBram Moolenaar int 406e7cb9cf6SBram Moolenaar python_enabled(int verbose) 407071d4279SBram Moolenaar { 408071d4279SBram Moolenaar return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK; 409071d4279SBram Moolenaar } 410071d4279SBram Moolenaar 411ca8a4dfeSBram Moolenaar /* 412ca8a4dfeSBram Moolenaar * Load the standard Python exceptions - don't import the symbols from the 413071d4279SBram Moolenaar * DLL, as this can cause errors (importing data symbols is not reliable). 414071d4279SBram Moolenaar */ 415071d4279SBram Moolenaar static void 416ca8a4dfeSBram Moolenaar get_exceptions(void) 417071d4279SBram Moolenaar { 418071d4279SBram Moolenaar PyObject *exmod = PyImport_ImportModule("exceptions"); 419071d4279SBram Moolenaar PyObject *exdict = PyModule_GetDict(exmod); 420071d4279SBram Moolenaar imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError"); 421071d4279SBram Moolenaar imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError"); 422071d4279SBram Moolenaar imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt"); 423071d4279SBram Moolenaar imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError"); 424071d4279SBram Moolenaar imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError"); 425071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_AttributeError); 426071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_IndexError); 427071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_KeyboardInterrupt); 428071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_TypeError); 429071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_ValueError); 430071d4279SBram Moolenaar Py_XDECREF(exmod); 431071d4279SBram Moolenaar } 432071d4279SBram Moolenaar #endif /* DYNAMIC_PYTHON */ 433071d4279SBram Moolenaar 434ca8a4dfeSBram Moolenaar static PyObject *BufferNew (buf_T *); 435ca8a4dfeSBram Moolenaar static PyObject *WindowNew(win_T *); 436ca8a4dfeSBram Moolenaar static PyObject *LineToString(const char *); 437ca8a4dfeSBram Moolenaar 438ca8a4dfeSBram Moolenaar static PyTypeObject RangeType; 439ca8a4dfeSBram Moolenaar 440170bf1aeSBram Moolenaar /* 441170bf1aeSBram Moolenaar * Include the code shared with if_python3.c 442170bf1aeSBram Moolenaar */ 443170bf1aeSBram Moolenaar #include "if_py_both.h" 444170bf1aeSBram Moolenaar 445170bf1aeSBram Moolenaar 446071d4279SBram Moolenaar /****************************************************** 447071d4279SBram Moolenaar * Internal function prototypes. 448071d4279SBram Moolenaar */ 449071d4279SBram Moolenaar 450e7cb9cf6SBram Moolenaar static PyInt RangeStart; 451e7cb9cf6SBram Moolenaar static PyInt RangeEnd; 452071d4279SBram Moolenaar 453071d4279SBram Moolenaar static void PythonIO_Flush(void); 454071d4279SBram Moolenaar static int PythonIO_Init(void); 455071d4279SBram Moolenaar static int PythonMod_Init(void); 456071d4279SBram Moolenaar 457071d4279SBram Moolenaar /* Utility functions for the vim/python interface 458071d4279SBram Moolenaar * ---------------------------------------------- 459071d4279SBram Moolenaar */ 460071d4279SBram Moolenaar 461e7cb9cf6SBram Moolenaar static int SetBufferLineList(buf_T *, PyInt, PyInt, PyObject *, PyInt *); 462071d4279SBram Moolenaar 463071d4279SBram Moolenaar 464071d4279SBram Moolenaar /****************************************************** 465071d4279SBram Moolenaar * 1. Python interpreter main program. 466071d4279SBram Moolenaar */ 467071d4279SBram Moolenaar 468071d4279SBram Moolenaar static int initialised = 0; 469071d4279SBram Moolenaar 470071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ 471071d4279SBram Moolenaar typedef PyObject PyThreadState; 4729ba0eb85SBram Moolenaar #endif 473071d4279SBram Moolenaar 4749ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE 4759ba0eb85SBram Moolenaar static PyGILState_STATE pygilstate = PyGILState_UNLOCKED; 4769ba0eb85SBram Moolenaar #else 477071d4279SBram Moolenaar static PyThreadState *saved_python_thread = NULL; 4789ba0eb85SBram Moolenaar #endif 479071d4279SBram Moolenaar 480071d4279SBram Moolenaar /* 481071d4279SBram Moolenaar * Suspend a thread of the Python interpreter, other threads are allowed to 482071d4279SBram Moolenaar * run. 483071d4279SBram Moolenaar */ 484293ee4d4SBram Moolenaar static void 485293ee4d4SBram Moolenaar Python_SaveThread(void) 486071d4279SBram Moolenaar { 4879ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE 4889ba0eb85SBram Moolenaar PyGILState_Release(pygilstate); 4899ba0eb85SBram Moolenaar #else 490071d4279SBram Moolenaar saved_python_thread = PyEval_SaveThread(); 4919ba0eb85SBram Moolenaar #endif 492071d4279SBram Moolenaar } 493071d4279SBram Moolenaar 494071d4279SBram Moolenaar /* 495071d4279SBram Moolenaar * Restore a thread of the Python interpreter, waits for other threads to 496071d4279SBram Moolenaar * block. 497071d4279SBram Moolenaar */ 498293ee4d4SBram Moolenaar static void 499293ee4d4SBram Moolenaar Python_RestoreThread(void) 500071d4279SBram Moolenaar { 5019ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE 5029ba0eb85SBram Moolenaar pygilstate = PyGILState_Ensure(); 5039ba0eb85SBram Moolenaar #else 504071d4279SBram Moolenaar PyEval_RestoreThread(saved_python_thread); 505071d4279SBram Moolenaar saved_python_thread = NULL; 506071d4279SBram Moolenaar #endif 5079ba0eb85SBram Moolenaar } 508071d4279SBram Moolenaar 509071d4279SBram Moolenaar void 510071d4279SBram Moolenaar python_end() 511071d4279SBram Moolenaar { 512a5792f58SBram Moolenaar static int recurse = 0; 513a5792f58SBram Moolenaar 514a5792f58SBram Moolenaar /* If a crash occurs while doing this, don't try again. */ 515a5792f58SBram Moolenaar if (recurse != 0) 516a5792f58SBram Moolenaar return; 517a5792f58SBram Moolenaar 518a5792f58SBram Moolenaar ++recurse; 519a5792f58SBram Moolenaar 520071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON 5210e21a3f6SBram Moolenaar if (hinstPython && Py_IsInitialized()) 5229ba0eb85SBram Moolenaar { 5239ba0eb85SBram Moolenaar Python_RestoreThread(); /* enter python */ 5240e21a3f6SBram Moolenaar Py_Finalize(); 5259ba0eb85SBram Moolenaar } 526071d4279SBram Moolenaar end_dynamic_python(); 5270e21a3f6SBram Moolenaar #else 5280e21a3f6SBram Moolenaar if (Py_IsInitialized()) 5299ba0eb85SBram Moolenaar { 5309ba0eb85SBram Moolenaar Python_RestoreThread(); /* enter python */ 5310e21a3f6SBram Moolenaar Py_Finalize(); 5329ba0eb85SBram Moolenaar } 533071d4279SBram Moolenaar #endif 534a5792f58SBram Moolenaar 535a5792f58SBram Moolenaar --recurse; 536071d4279SBram Moolenaar } 537071d4279SBram Moolenaar 5384c3a326cSBram Moolenaar #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO) 5394c3a326cSBram Moolenaar int 5404c3a326cSBram Moolenaar python_loaded() 5414c3a326cSBram Moolenaar { 5424c3a326cSBram Moolenaar return (hinstPython != 0); 5434c3a326cSBram Moolenaar } 5444c3a326cSBram Moolenaar #endif 5454c3a326cSBram Moolenaar 546071d4279SBram Moolenaar static int 547071d4279SBram Moolenaar Python_Init(void) 548071d4279SBram Moolenaar { 549071d4279SBram Moolenaar if (!initialised) 550071d4279SBram Moolenaar { 551071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON 552071d4279SBram Moolenaar if (!python_enabled(TRUE)) 553071d4279SBram Moolenaar { 554071d4279SBram Moolenaar EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded.")); 555071d4279SBram Moolenaar goto fail; 556071d4279SBram Moolenaar } 557071d4279SBram Moolenaar #endif 558071d4279SBram Moolenaar 559644d37b8SBram Moolenaar #ifdef PYTHON_HOME 560644d37b8SBram Moolenaar Py_SetPythonHome(PYTHON_HOME); 561644d37b8SBram Moolenaar #endif 562644d37b8SBram Moolenaar 563170bf1aeSBram Moolenaar init_structs(); 564170bf1aeSBram Moolenaar 565071d4279SBram Moolenaar #if !defined(MACOS) || defined(MACOS_X_UNIX) 566071d4279SBram Moolenaar Py_Initialize(); 567071d4279SBram Moolenaar #else 568071d4279SBram Moolenaar PyMac_Initialize(); 569071d4279SBram Moolenaar #endif 570071d4279SBram Moolenaar /* initialise threads */ 571071d4279SBram Moolenaar PyEval_InitThreads(); 572071d4279SBram Moolenaar 573071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON 574071d4279SBram Moolenaar get_exceptions(); 575071d4279SBram Moolenaar #endif 576071d4279SBram Moolenaar 577071d4279SBram Moolenaar if (PythonIO_Init()) 578071d4279SBram Moolenaar goto fail; 579071d4279SBram Moolenaar 580071d4279SBram Moolenaar if (PythonMod_Init()) 581071d4279SBram Moolenaar goto fail; 582071d4279SBram Moolenaar 5839774ecc8SBram Moolenaar /* Remove the element from sys.path that was added because of our 5849774ecc8SBram Moolenaar * argv[0] value in PythonMod_Init(). Previously we used an empty 5859774ecc8SBram Moolenaar * string, but dependinding on the OS we then get an empty entry or 5869774ecc8SBram Moolenaar * the current directory in sys.path. */ 5879774ecc8SBram Moolenaar PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)"); 5889774ecc8SBram Moolenaar 589293ee4d4SBram Moolenaar /* the first python thread is vim's, release the lock */ 590071d4279SBram Moolenaar Python_SaveThread(); 591071d4279SBram Moolenaar 592071d4279SBram Moolenaar initialised = 1; 593071d4279SBram Moolenaar } 594071d4279SBram Moolenaar 595071d4279SBram Moolenaar return 0; 596071d4279SBram Moolenaar 597071d4279SBram Moolenaar fail: 598071d4279SBram Moolenaar /* We call PythonIO_Flush() here to print any Python errors. 599071d4279SBram Moolenaar * This is OK, as it is possible to call this function even 600071d4279SBram Moolenaar * if PythonIO_Init() has not completed successfully (it will 601071d4279SBram Moolenaar * not do anything in this case). 602071d4279SBram Moolenaar */ 603071d4279SBram Moolenaar PythonIO_Flush(); 604071d4279SBram Moolenaar return -1; 605071d4279SBram Moolenaar } 606071d4279SBram Moolenaar 607071d4279SBram Moolenaar /* 608071d4279SBram Moolenaar * External interface 609071d4279SBram Moolenaar */ 610071d4279SBram Moolenaar static void 611071d4279SBram Moolenaar DoPythonCommand(exarg_T *eap, const char *cmd) 612071d4279SBram Moolenaar { 6139ba0eb85SBram Moolenaar #ifndef PY_CAN_RECURSE 614071d4279SBram Moolenaar static int recursive = 0; 615071d4279SBram Moolenaar #endif 616071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 617071d4279SBram Moolenaar GrafPtr oldPort; 618071d4279SBram Moolenaar #endif 619071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 620071d4279SBram Moolenaar char *saved_locale; 621071d4279SBram Moolenaar #endif 622071d4279SBram Moolenaar 623071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE 624071d4279SBram Moolenaar if (recursive) 625071d4279SBram Moolenaar { 626071d4279SBram Moolenaar EMSG(_("E659: Cannot invoke Python recursively")); 627071d4279SBram Moolenaar return; 628071d4279SBram Moolenaar } 629071d4279SBram Moolenaar ++recursive; 630071d4279SBram Moolenaar #endif 631071d4279SBram Moolenaar 632071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 633071d4279SBram Moolenaar GetPort(&oldPort); 634071d4279SBram Moolenaar /* Check if the Python library is available */ 635071d4279SBram Moolenaar if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress) 636071d4279SBram Moolenaar goto theend; 637071d4279SBram Moolenaar #endif 638071d4279SBram Moolenaar if (Python_Init()) 639071d4279SBram Moolenaar goto theend; 640071d4279SBram Moolenaar 641071d4279SBram Moolenaar RangeStart = eap->line1; 642071d4279SBram Moolenaar RangeEnd = eap->line2; 643071d4279SBram Moolenaar Python_Release_Vim(); /* leave vim */ 644071d4279SBram Moolenaar 645071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 646071d4279SBram Moolenaar /* Python only works properly when the LC_NUMERIC locale is "C". */ 647071d4279SBram Moolenaar saved_locale = setlocale(LC_NUMERIC, NULL); 648071d4279SBram Moolenaar if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0) 649071d4279SBram Moolenaar saved_locale = NULL; 650071d4279SBram Moolenaar else 651071d4279SBram Moolenaar { 652071d4279SBram Moolenaar /* Need to make a copy, value may change when setting new locale. */ 653071d4279SBram Moolenaar saved_locale = (char *)vim_strsave((char_u *)saved_locale); 654071d4279SBram Moolenaar (void)setlocale(LC_NUMERIC, "C"); 655071d4279SBram Moolenaar } 656071d4279SBram Moolenaar #endif 657071d4279SBram Moolenaar 658071d4279SBram Moolenaar Python_RestoreThread(); /* enter python */ 659071d4279SBram Moolenaar 660071d4279SBram Moolenaar PyRun_SimpleString((char *)(cmd)); 661071d4279SBram Moolenaar 662071d4279SBram Moolenaar Python_SaveThread(); /* leave python */ 663071d4279SBram Moolenaar 664071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 665071d4279SBram Moolenaar if (saved_locale != NULL) 666071d4279SBram Moolenaar { 667071d4279SBram Moolenaar (void)setlocale(LC_NUMERIC, saved_locale); 668071d4279SBram Moolenaar vim_free(saved_locale); 669071d4279SBram Moolenaar } 670071d4279SBram Moolenaar #endif 671071d4279SBram Moolenaar 672071d4279SBram Moolenaar Python_Lock_Vim(); /* enter vim */ 673071d4279SBram Moolenaar PythonIO_Flush(); 674071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 675071d4279SBram Moolenaar SetPort(oldPort); 676071d4279SBram Moolenaar #endif 677071d4279SBram Moolenaar 678071d4279SBram Moolenaar theend: 679071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE 680071d4279SBram Moolenaar --recursive; 681071d4279SBram Moolenaar #endif 682071d4279SBram Moolenaar return; /* keeps lint happy */ 683071d4279SBram Moolenaar } 684071d4279SBram Moolenaar 685071d4279SBram Moolenaar /* 686071d4279SBram Moolenaar * ":python" 687071d4279SBram Moolenaar */ 688071d4279SBram Moolenaar void 689071d4279SBram Moolenaar ex_python(exarg_T *eap) 690071d4279SBram Moolenaar { 691071d4279SBram Moolenaar char_u *script; 692071d4279SBram Moolenaar 693071d4279SBram Moolenaar script = script_get(eap, eap->arg); 694071d4279SBram Moolenaar if (!eap->skip) 695071d4279SBram Moolenaar { 696071d4279SBram Moolenaar if (script == NULL) 697071d4279SBram Moolenaar DoPythonCommand(eap, (char *)eap->arg); 698071d4279SBram Moolenaar else 699071d4279SBram Moolenaar DoPythonCommand(eap, (char *)script); 700071d4279SBram Moolenaar } 701071d4279SBram Moolenaar vim_free(script); 702071d4279SBram Moolenaar } 703071d4279SBram Moolenaar 704071d4279SBram Moolenaar #define BUFFER_SIZE 1024 705071d4279SBram Moolenaar 706071d4279SBram Moolenaar /* 707071d4279SBram Moolenaar * ":pyfile" 708071d4279SBram Moolenaar */ 709071d4279SBram Moolenaar void 710071d4279SBram Moolenaar ex_pyfile(exarg_T *eap) 711071d4279SBram Moolenaar { 712071d4279SBram Moolenaar static char buffer[BUFFER_SIZE]; 713071d4279SBram Moolenaar const char *file = (char *)eap->arg; 714071d4279SBram Moolenaar char *p; 715071d4279SBram Moolenaar 716071d4279SBram Moolenaar /* Have to do it like this. PyRun_SimpleFile requires you to pass a 717071d4279SBram Moolenaar * stdio file pointer, but Vim and the Python DLL are compiled with 718071d4279SBram Moolenaar * different options under Windows, meaning that stdio pointers aren't 719071d4279SBram Moolenaar * compatible between the two. Yuk. 720071d4279SBram Moolenaar * 721071d4279SBram Moolenaar * Put the string "execfile('file')" into buffer. But, we need to 722071d4279SBram Moolenaar * escape any backslashes or single quotes in the file name, so that 723071d4279SBram Moolenaar * Python won't mangle the file name. 724071d4279SBram Moolenaar */ 725071d4279SBram Moolenaar strcpy(buffer, "execfile('"); 726071d4279SBram Moolenaar p = buffer + 10; /* size of "execfile('" */ 727071d4279SBram Moolenaar 728071d4279SBram Moolenaar while (*file && p < buffer + (BUFFER_SIZE - 3)) 729071d4279SBram Moolenaar { 730071d4279SBram Moolenaar if (*file == '\\' || *file == '\'') 731071d4279SBram Moolenaar *p++ = '\\'; 732071d4279SBram Moolenaar *p++ = *file++; 733071d4279SBram Moolenaar } 734071d4279SBram Moolenaar 735071d4279SBram Moolenaar /* If we didn't finish the file name, we hit a buffer overflow */ 736071d4279SBram Moolenaar if (*file != '\0') 737071d4279SBram Moolenaar return; 738071d4279SBram Moolenaar 739071d4279SBram Moolenaar /* Put in the terminating "')" and a null */ 740071d4279SBram Moolenaar *p++ = '\''; 741071d4279SBram Moolenaar *p++ = ')'; 742071d4279SBram Moolenaar *p++ = '\0'; 743071d4279SBram Moolenaar 744071d4279SBram Moolenaar /* Execute the file */ 745071d4279SBram Moolenaar DoPythonCommand(eap, buffer); 746071d4279SBram Moolenaar } 747071d4279SBram Moolenaar 748071d4279SBram Moolenaar /****************************************************** 749071d4279SBram Moolenaar * 2. Python output stream: writes output via [e]msg(). 750071d4279SBram Moolenaar */ 751071d4279SBram Moolenaar 752071d4279SBram Moolenaar /* Implementation functions 753071d4279SBram Moolenaar */ 754071d4279SBram Moolenaar 755071d4279SBram Moolenaar static PyObject * 756071d4279SBram Moolenaar OutputGetattr(PyObject *self, char *name) 757071d4279SBram Moolenaar { 758071d4279SBram Moolenaar if (strcmp(name, "softspace") == 0) 759071d4279SBram Moolenaar return PyInt_FromLong(((OutputObject *)(self))->softspace); 760071d4279SBram Moolenaar 761071d4279SBram Moolenaar return Py_FindMethod(OutputMethods, self, name); 762071d4279SBram Moolenaar } 763071d4279SBram Moolenaar 764071d4279SBram Moolenaar static int 765071d4279SBram Moolenaar OutputSetattr(PyObject *self, char *name, PyObject *val) 766071d4279SBram Moolenaar { 767071d4279SBram Moolenaar if (val == NULL) { 768071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, _("can't delete OutputObject attributes")); 769071d4279SBram Moolenaar return -1; 770071d4279SBram Moolenaar } 771071d4279SBram Moolenaar 772071d4279SBram Moolenaar if (strcmp(name, "softspace") == 0) 773071d4279SBram Moolenaar { 774071d4279SBram Moolenaar if (!PyInt_Check(val)) { 775071d4279SBram Moolenaar PyErr_SetString(PyExc_TypeError, _("softspace must be an integer")); 776071d4279SBram Moolenaar return -1; 777071d4279SBram Moolenaar } 778071d4279SBram Moolenaar 779071d4279SBram Moolenaar ((OutputObject *)(self))->softspace = PyInt_AsLong(val); 780071d4279SBram Moolenaar return 0; 781071d4279SBram Moolenaar } 782071d4279SBram Moolenaar 783071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, _("invalid attribute")); 784071d4279SBram Moolenaar return -1; 785071d4279SBram Moolenaar } 786071d4279SBram Moolenaar 787071d4279SBram Moolenaar /***************/ 788071d4279SBram Moolenaar 789071d4279SBram Moolenaar static int 790071d4279SBram Moolenaar PythonIO_Init(void) 791071d4279SBram Moolenaar { 792071d4279SBram Moolenaar /* Fixups... */ 79321377c8dSBram Moolenaar PyType_Ready(&OutputType); 794071d4279SBram Moolenaar 795170bf1aeSBram Moolenaar return PythonIO_Init_io(); 796071d4279SBram Moolenaar } 797071d4279SBram Moolenaar 798071d4279SBram Moolenaar /****************************************************** 799071d4279SBram Moolenaar * 3. Implementation of the Vim module for Python 800071d4279SBram Moolenaar */ 801071d4279SBram Moolenaar 802071d4279SBram Moolenaar /* Window type - Implementation functions 803071d4279SBram Moolenaar * -------------------------------------- 804071d4279SBram Moolenaar */ 805071d4279SBram Moolenaar 806071d4279SBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_type == &WindowType) 807071d4279SBram Moolenaar 808071d4279SBram Moolenaar static void WindowDestructor(PyObject *); 809071d4279SBram Moolenaar static PyObject *WindowGetattr(PyObject *, char *); 810071d4279SBram Moolenaar 811071d4279SBram Moolenaar /* Buffer type - Implementation functions 812071d4279SBram Moolenaar * -------------------------------------- 813071d4279SBram Moolenaar */ 814071d4279SBram Moolenaar 815071d4279SBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_type == &BufferType) 816071d4279SBram Moolenaar 817071d4279SBram Moolenaar static void BufferDestructor(PyObject *); 818071d4279SBram Moolenaar static PyObject *BufferGetattr(PyObject *, char *); 819071d4279SBram Moolenaar static PyObject *BufferRepr(PyObject *); 820071d4279SBram Moolenaar 8212c45e945SBram Moolenaar static PyInt BufferLength(PyObject *); 8222c45e945SBram Moolenaar static PyObject *BufferItem(PyObject *, PyInt); 8232c45e945SBram Moolenaar static PyObject *BufferSlice(PyObject *, PyInt, PyInt); 8242c45e945SBram Moolenaar static PyInt BufferAssItem(PyObject *, PyInt, PyObject *); 8252c45e945SBram Moolenaar static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *); 826071d4279SBram Moolenaar 827071d4279SBram Moolenaar /* Line range type - Implementation functions 828071d4279SBram Moolenaar * -------------------------------------- 829071d4279SBram Moolenaar */ 830071d4279SBram Moolenaar 831071d4279SBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_type == &RangeType) 832071d4279SBram Moolenaar 8332c45e945SBram Moolenaar static PyInt RangeAssItem(PyObject *, PyInt, PyObject *); 8342c45e945SBram Moolenaar static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *); 835071d4279SBram Moolenaar 836071d4279SBram Moolenaar /* Current objects type - Implementation functions 837071d4279SBram Moolenaar * ----------------------------------------------- 838071d4279SBram Moolenaar */ 839071d4279SBram Moolenaar 840071d4279SBram Moolenaar static PyObject *CurrentGetattr(PyObject *, char *); 841071d4279SBram Moolenaar static int CurrentSetattr(PyObject *, char *, PyObject *); 842071d4279SBram Moolenaar 843071d4279SBram Moolenaar static PySequenceMethods BufferAsSeq = { 8442c45e945SBram Moolenaar (PyInquiry) BufferLength, /* sq_length, len(x) */ 845071d4279SBram Moolenaar (binaryfunc) 0, /* BufferConcat, */ /* sq_concat, x+y */ 8462c45e945SBram Moolenaar (PyIntArgFunc) 0, /* BufferRepeat, */ /* sq_repeat, x*n */ 8472c45e945SBram Moolenaar (PyIntArgFunc) BufferItem, /* sq_item, x[i] */ 8482c45e945SBram Moolenaar (PyIntIntArgFunc) BufferSlice, /* sq_slice, x[i:j] */ 8492c45e945SBram Moolenaar (PyIntObjArgProc) BufferAssItem, /* sq_ass_item, x[i]=v */ 8502c45e945SBram Moolenaar (PyIntIntObjArgProc) BufferAssSlice, /* sq_ass_slice, x[i:j]=v */ 851071d4279SBram Moolenaar }; 852071d4279SBram Moolenaar 853071d4279SBram Moolenaar static PyTypeObject BufferType = { 854071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 855071d4279SBram Moolenaar 0, 856071d4279SBram Moolenaar "buffer", 857071d4279SBram Moolenaar sizeof(BufferObject), 858071d4279SBram Moolenaar 0, 859071d4279SBram Moolenaar 860071d4279SBram Moolenaar (destructor) BufferDestructor, /* tp_dealloc, refcount==0 */ 861071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 862071d4279SBram Moolenaar (getattrfunc) BufferGetattr, /* tp_getattr, x.attr */ 863071d4279SBram Moolenaar (setattrfunc) 0, /* tp_setattr, x.attr=v */ 864071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 865071d4279SBram Moolenaar (reprfunc) BufferRepr, /* tp_repr, `x`, print x */ 866071d4279SBram Moolenaar 867071d4279SBram Moolenaar 0, /* as number */ 868071d4279SBram Moolenaar &BufferAsSeq, /* as sequence */ 869071d4279SBram Moolenaar 0, /* as mapping */ 870071d4279SBram Moolenaar 871071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 872071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 873071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 874071d4279SBram Moolenaar }; 875071d4279SBram Moolenaar 876071d4279SBram Moolenaar /* Buffer object - Implementation 877071d4279SBram Moolenaar */ 878071d4279SBram Moolenaar 879071d4279SBram Moolenaar static PyObject * 880071d4279SBram Moolenaar BufferNew(buf_T *buf) 881071d4279SBram Moolenaar { 882071d4279SBram Moolenaar /* We need to handle deletion of buffers underneath us. 883e344beadSBram Moolenaar * If we add a "b_python_ref" field to the buf_T structure, 884071d4279SBram Moolenaar * then we can get at it in buf_freeall() in vim. We then 885071d4279SBram Moolenaar * need to create only ONE Python object per buffer - if 886071d4279SBram Moolenaar * we try to create a second, just INCREF the existing one 887071d4279SBram Moolenaar * and return it. The (single) Python object referring to 888e344beadSBram Moolenaar * the buffer is stored in "b_python_ref". 889071d4279SBram Moolenaar * Question: what to do on a buf_freeall(). We'll probably 890071d4279SBram Moolenaar * have to either delete the Python object (DECREF it to 891071d4279SBram Moolenaar * zero - a bad idea, as it leaves dangling refs!) or 892071d4279SBram Moolenaar * set the buf_T * value to an invalid value (-1?), which 893071d4279SBram Moolenaar * means we need checks in all access functions... Bah. 894071d4279SBram Moolenaar */ 895071d4279SBram Moolenaar 896071d4279SBram Moolenaar BufferObject *self; 897071d4279SBram Moolenaar 898e344beadSBram Moolenaar if (buf->b_python_ref != NULL) 899071d4279SBram Moolenaar { 900e344beadSBram Moolenaar self = buf->b_python_ref; 901071d4279SBram Moolenaar Py_INCREF(self); 902071d4279SBram Moolenaar } 903071d4279SBram Moolenaar else 904071d4279SBram Moolenaar { 905071d4279SBram Moolenaar self = PyObject_NEW(BufferObject, &BufferType); 906071d4279SBram Moolenaar if (self == NULL) 907071d4279SBram Moolenaar return NULL; 908071d4279SBram Moolenaar self->buf = buf; 909e344beadSBram Moolenaar buf->b_python_ref = self; 910071d4279SBram Moolenaar } 911071d4279SBram Moolenaar 912071d4279SBram Moolenaar return (PyObject *)(self); 913071d4279SBram Moolenaar } 914071d4279SBram Moolenaar 915071d4279SBram Moolenaar static void 916071d4279SBram Moolenaar BufferDestructor(PyObject *self) 917071d4279SBram Moolenaar { 918071d4279SBram Moolenaar BufferObject *this = (BufferObject *)(self); 919071d4279SBram Moolenaar 920071d4279SBram Moolenaar if (this->buf && this->buf != INVALID_BUFFER_VALUE) 921e344beadSBram Moolenaar this->buf->b_python_ref = NULL; 922071d4279SBram Moolenaar 923658ada69SBram Moolenaar Py_DECREF(self); 924071d4279SBram Moolenaar } 925071d4279SBram Moolenaar 926071d4279SBram Moolenaar static PyObject * 927071d4279SBram Moolenaar BufferGetattr(PyObject *self, char *name) 928071d4279SBram Moolenaar { 929071d4279SBram Moolenaar BufferObject *this = (BufferObject *)(self); 930071d4279SBram Moolenaar 931071d4279SBram Moolenaar if (CheckBuffer(this)) 932071d4279SBram Moolenaar return NULL; 933071d4279SBram Moolenaar 934071d4279SBram Moolenaar if (strcmp(name, "name") == 0) 935071d4279SBram Moolenaar return Py_BuildValue("s", this->buf->b_ffname); 936071d4279SBram Moolenaar else if (strcmp(name, "number") == 0) 937e7cb9cf6SBram Moolenaar return Py_BuildValue(Py_ssize_t_fmt, this->buf->b_fnum); 938071d4279SBram Moolenaar else if (strcmp(name,"__members__") == 0) 939071d4279SBram Moolenaar return Py_BuildValue("[ss]", "name", "number"); 940071d4279SBram Moolenaar else 941071d4279SBram Moolenaar return Py_FindMethod(BufferMethods, self, name); 942071d4279SBram Moolenaar } 943071d4279SBram Moolenaar 944071d4279SBram Moolenaar static PyObject * 945071d4279SBram Moolenaar BufferRepr(PyObject *self) 946071d4279SBram Moolenaar { 947555b280fSBram Moolenaar static char repr[100]; 948071d4279SBram Moolenaar BufferObject *this = (BufferObject *)(self); 949071d4279SBram Moolenaar 950071d4279SBram Moolenaar if (this->buf == INVALID_BUFFER_VALUE) 951071d4279SBram Moolenaar { 952e7cb9cf6SBram Moolenaar vim_snprintf(repr, 100, _("<buffer object (deleted) at %p>"), (self)); 953071d4279SBram Moolenaar return PyString_FromString(repr); 954071d4279SBram Moolenaar } 955071d4279SBram Moolenaar else 956071d4279SBram Moolenaar { 957071d4279SBram Moolenaar char *name = (char *)this->buf->b_fname; 958e7cb9cf6SBram Moolenaar PyInt len; 959071d4279SBram Moolenaar 960071d4279SBram Moolenaar if (name == NULL) 961071d4279SBram Moolenaar name = ""; 962071d4279SBram Moolenaar len = strlen(name); 963071d4279SBram Moolenaar 964071d4279SBram Moolenaar if (len > 35) 965071d4279SBram Moolenaar name = name + (35 - len); 966071d4279SBram Moolenaar 967555b280fSBram Moolenaar vim_snprintf(repr, 100, "<buffer %s%s>", len > 35 ? "..." : "", name); 968071d4279SBram Moolenaar 969071d4279SBram Moolenaar return PyString_FromString(repr); 970071d4279SBram Moolenaar } 971071d4279SBram Moolenaar } 972071d4279SBram Moolenaar 973071d4279SBram Moolenaar /******************/ 974071d4279SBram Moolenaar 9752c45e945SBram Moolenaar static PyInt 976071d4279SBram Moolenaar BufferLength(PyObject *self) 977071d4279SBram Moolenaar { 978071d4279SBram Moolenaar /* HOW DO WE SIGNAL AN ERROR FROM THIS FUNCTION? */ 979071d4279SBram Moolenaar if (CheckBuffer((BufferObject *)(self))) 980071d4279SBram Moolenaar return -1; /* ??? */ 981071d4279SBram Moolenaar 982071d4279SBram Moolenaar return (((BufferObject *)(self))->buf->b_ml.ml_line_count); 983071d4279SBram Moolenaar } 984071d4279SBram Moolenaar 985071d4279SBram Moolenaar static PyObject * 9862c45e945SBram Moolenaar BufferItem(PyObject *self, PyInt n) 987071d4279SBram Moolenaar { 988071d4279SBram Moolenaar return RBItem((BufferObject *)(self), n, 1, 989071d4279SBram Moolenaar (int)((BufferObject *)(self))->buf->b_ml.ml_line_count); 990071d4279SBram Moolenaar } 991071d4279SBram Moolenaar 992071d4279SBram Moolenaar static PyObject * 9932c45e945SBram Moolenaar BufferSlice(PyObject *self, PyInt lo, PyInt hi) 994071d4279SBram Moolenaar { 995071d4279SBram Moolenaar return RBSlice((BufferObject *)(self), lo, hi, 1, 996071d4279SBram Moolenaar (int)((BufferObject *)(self))->buf->b_ml.ml_line_count); 997071d4279SBram Moolenaar } 998071d4279SBram Moolenaar 9992c45e945SBram Moolenaar static PyInt 10002c45e945SBram Moolenaar BufferAssItem(PyObject *self, PyInt n, PyObject *val) 1001071d4279SBram Moolenaar { 1002ca8a4dfeSBram Moolenaar return RBAsItem((BufferObject *)(self), n, val, 1, 1003e7cb9cf6SBram Moolenaar (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, 1004071d4279SBram Moolenaar NULL); 1005071d4279SBram Moolenaar } 1006071d4279SBram Moolenaar 10072c45e945SBram Moolenaar static PyInt 10082c45e945SBram Moolenaar BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val) 1009071d4279SBram Moolenaar { 1010*19e60943SBram Moolenaar return RBAsSlice((BufferObject *)(self), lo, hi, val, 1, 1011e7cb9cf6SBram Moolenaar (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, 1012071d4279SBram Moolenaar NULL); 1013071d4279SBram Moolenaar } 1014071d4279SBram Moolenaar 1015071d4279SBram Moolenaar static PySequenceMethods RangeAsSeq = { 10162c45e945SBram Moolenaar (PyInquiry) RangeLength, /* sq_length, len(x) */ 1017071d4279SBram Moolenaar (binaryfunc) 0, /* RangeConcat, */ /* sq_concat, x+y */ 10182c45e945SBram Moolenaar (PyIntArgFunc) 0, /* RangeRepeat, */ /* sq_repeat, x*n */ 10192c45e945SBram Moolenaar (PyIntArgFunc) RangeItem, /* sq_item, x[i] */ 10202c45e945SBram Moolenaar (PyIntIntArgFunc) RangeSlice, /* sq_slice, x[i:j] */ 10212c45e945SBram Moolenaar (PyIntObjArgProc) RangeAssItem, /* sq_ass_item, x[i]=v */ 10222c45e945SBram Moolenaar (PyIntIntObjArgProc) RangeAssSlice, /* sq_ass_slice, x[i:j]=v */ 1023071d4279SBram Moolenaar }; 1024071d4279SBram Moolenaar 1025071d4279SBram Moolenaar /* Line range object - Implementation 1026071d4279SBram Moolenaar */ 1027071d4279SBram Moolenaar 1028071d4279SBram Moolenaar static void 1029071d4279SBram Moolenaar RangeDestructor(PyObject *self) 1030071d4279SBram Moolenaar { 1031071d4279SBram Moolenaar Py_DECREF(((RangeObject *)(self))->buf); 1032658ada69SBram Moolenaar Py_DECREF(self); 1033071d4279SBram Moolenaar } 1034071d4279SBram Moolenaar 1035071d4279SBram Moolenaar static PyObject * 1036071d4279SBram Moolenaar RangeGetattr(PyObject *self, char *name) 1037071d4279SBram Moolenaar { 1038071d4279SBram Moolenaar if (strcmp(name, "start") == 0) 1039e7cb9cf6SBram Moolenaar return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1); 1040071d4279SBram Moolenaar else if (strcmp(name, "end") == 0) 1041e7cb9cf6SBram Moolenaar return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1); 1042071d4279SBram Moolenaar else 1043071d4279SBram Moolenaar return Py_FindMethod(RangeMethods, self, name); 1044071d4279SBram Moolenaar } 1045071d4279SBram Moolenaar 1046071d4279SBram Moolenaar /****************/ 1047071d4279SBram Moolenaar 10482c45e945SBram Moolenaar static PyInt 10492c45e945SBram Moolenaar RangeAssItem(PyObject *self, PyInt n, PyObject *val) 1050071d4279SBram Moolenaar { 1051ca8a4dfeSBram Moolenaar return RBAsItem(((RangeObject *)(self))->buf, n, val, 1052071d4279SBram Moolenaar ((RangeObject *)(self))->start, 1053071d4279SBram Moolenaar ((RangeObject *)(self))->end, 1054071d4279SBram Moolenaar &((RangeObject *)(self))->end); 1055071d4279SBram Moolenaar } 1056071d4279SBram Moolenaar 10572c45e945SBram Moolenaar static PyInt 10582c45e945SBram Moolenaar RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val) 1059071d4279SBram Moolenaar { 1060*19e60943SBram Moolenaar return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val, 1061071d4279SBram Moolenaar ((RangeObject *)(self))->start, 1062071d4279SBram Moolenaar ((RangeObject *)(self))->end, 1063071d4279SBram Moolenaar &((RangeObject *)(self))->end); 1064071d4279SBram Moolenaar } 1065071d4279SBram Moolenaar 1066071d4279SBram Moolenaar /* Buffer list object - Definitions 1067071d4279SBram Moolenaar */ 1068071d4279SBram Moolenaar 1069071d4279SBram Moolenaar typedef struct 1070071d4279SBram Moolenaar { 1071071d4279SBram Moolenaar PyObject_HEAD 1072ca8a4dfeSBram Moolenaar } BufListObject; 1073071d4279SBram Moolenaar 1074071d4279SBram Moolenaar static PySequenceMethods BufListAsSeq = { 10752c45e945SBram Moolenaar (PyInquiry) BufListLength, /* sq_length, len(x) */ 1076071d4279SBram Moolenaar (binaryfunc) 0, /* sq_concat, x+y */ 10772c45e945SBram Moolenaar (PyIntArgFunc) 0, /* sq_repeat, x*n */ 10782c45e945SBram Moolenaar (PyIntArgFunc) BufListItem, /* sq_item, x[i] */ 10792c45e945SBram Moolenaar (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */ 10802c45e945SBram Moolenaar (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */ 10812c45e945SBram Moolenaar (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */ 1082071d4279SBram Moolenaar }; 1083071d4279SBram Moolenaar 1084071d4279SBram Moolenaar static PyTypeObject BufListType = { 1085071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1086071d4279SBram Moolenaar 0, 1087071d4279SBram Moolenaar "buffer list", 1088071d4279SBram Moolenaar sizeof(BufListObject), 1089071d4279SBram Moolenaar 0, 1090071d4279SBram Moolenaar 1091071d4279SBram Moolenaar (destructor) 0, /* tp_dealloc, refcount==0 */ 1092071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1093071d4279SBram Moolenaar (getattrfunc) 0, /* tp_getattr, x.attr */ 1094071d4279SBram Moolenaar (setattrfunc) 0, /* tp_setattr, x.attr=v */ 1095071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1096071d4279SBram Moolenaar (reprfunc) 0, /* tp_repr, `x`, print x */ 1097071d4279SBram Moolenaar 1098071d4279SBram Moolenaar 0, /* as number */ 1099071d4279SBram Moolenaar &BufListAsSeq, /* as sequence */ 1100071d4279SBram Moolenaar 0, /* as mapping */ 1101071d4279SBram Moolenaar 1102071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1103071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1104071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1105071d4279SBram Moolenaar }; 1106071d4279SBram Moolenaar 1107071d4279SBram Moolenaar /* Window object - Definitions 1108071d4279SBram Moolenaar */ 1109071d4279SBram Moolenaar 1110071d4279SBram Moolenaar static struct PyMethodDef WindowMethods[] = { 1111071d4279SBram Moolenaar /* name, function, calling, documentation */ 1112071d4279SBram Moolenaar { NULL, NULL, 0, NULL } 1113071d4279SBram Moolenaar }; 1114071d4279SBram Moolenaar 1115071d4279SBram Moolenaar static PyTypeObject WindowType = { 1116071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1117071d4279SBram Moolenaar 0, 1118071d4279SBram Moolenaar "window", 1119071d4279SBram Moolenaar sizeof(WindowObject), 1120071d4279SBram Moolenaar 0, 1121071d4279SBram Moolenaar 1122071d4279SBram Moolenaar (destructor) WindowDestructor, /* tp_dealloc, refcount==0 */ 1123071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1124071d4279SBram Moolenaar (getattrfunc) WindowGetattr, /* tp_getattr, x.attr */ 1125071d4279SBram Moolenaar (setattrfunc) WindowSetattr, /* tp_setattr, x.attr=v */ 1126071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1127071d4279SBram Moolenaar (reprfunc) WindowRepr, /* tp_repr, `x`, print x */ 1128071d4279SBram Moolenaar 1129071d4279SBram Moolenaar 0, /* as number */ 1130071d4279SBram Moolenaar 0, /* as sequence */ 1131071d4279SBram Moolenaar 0, /* as mapping */ 1132071d4279SBram Moolenaar 1133071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1134071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1135071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1136071d4279SBram Moolenaar }; 1137071d4279SBram Moolenaar 1138071d4279SBram Moolenaar /* Window object - Implementation 1139071d4279SBram Moolenaar */ 1140071d4279SBram Moolenaar 1141071d4279SBram Moolenaar static PyObject * 1142071d4279SBram Moolenaar WindowNew(win_T *win) 1143071d4279SBram Moolenaar { 1144071d4279SBram Moolenaar /* We need to handle deletion of windows underneath us. 1145e344beadSBram Moolenaar * If we add a "w_python_ref" field to the win_T structure, 1146071d4279SBram Moolenaar * then we can get at it in win_free() in vim. We then 1147071d4279SBram Moolenaar * need to create only ONE Python object per window - if 1148071d4279SBram Moolenaar * we try to create a second, just INCREF the existing one 1149071d4279SBram Moolenaar * and return it. The (single) Python object referring to 1150e344beadSBram Moolenaar * the window is stored in "w_python_ref". 1151071d4279SBram Moolenaar * On a win_free() we set the Python object's win_T* field 1152071d4279SBram Moolenaar * to an invalid value. We trap all uses of a window 1153071d4279SBram Moolenaar * object, and reject them if the win_T* field is invalid. 1154071d4279SBram Moolenaar */ 1155071d4279SBram Moolenaar 1156071d4279SBram Moolenaar WindowObject *self; 1157071d4279SBram Moolenaar 1158e344beadSBram Moolenaar if (win->w_python_ref) 1159071d4279SBram Moolenaar { 1160e344beadSBram Moolenaar self = win->w_python_ref; 1161071d4279SBram Moolenaar Py_INCREF(self); 1162071d4279SBram Moolenaar } 1163071d4279SBram Moolenaar else 1164071d4279SBram Moolenaar { 1165071d4279SBram Moolenaar self = PyObject_NEW(WindowObject, &WindowType); 1166071d4279SBram Moolenaar if (self == NULL) 1167071d4279SBram Moolenaar return NULL; 1168071d4279SBram Moolenaar self->win = win; 1169e344beadSBram Moolenaar win->w_python_ref = self; 1170071d4279SBram Moolenaar } 1171071d4279SBram Moolenaar 1172071d4279SBram Moolenaar return (PyObject *)(self); 1173071d4279SBram Moolenaar } 1174071d4279SBram Moolenaar 1175071d4279SBram Moolenaar static void 1176071d4279SBram Moolenaar WindowDestructor(PyObject *self) 1177071d4279SBram Moolenaar { 1178071d4279SBram Moolenaar WindowObject *this = (WindowObject *)(self); 1179071d4279SBram Moolenaar 1180071d4279SBram Moolenaar if (this->win && this->win != INVALID_WINDOW_VALUE) 1181e344beadSBram Moolenaar this->win->w_python_ref = NULL; 1182071d4279SBram Moolenaar 1183658ada69SBram Moolenaar Py_DECREF(self); 1184071d4279SBram Moolenaar } 1185071d4279SBram Moolenaar 1186071d4279SBram Moolenaar static PyObject * 1187071d4279SBram Moolenaar WindowGetattr(PyObject *self, char *name) 1188071d4279SBram Moolenaar { 1189071d4279SBram Moolenaar WindowObject *this = (WindowObject *)(self); 1190071d4279SBram Moolenaar 1191071d4279SBram Moolenaar if (CheckWindow(this)) 1192071d4279SBram Moolenaar return NULL; 1193071d4279SBram Moolenaar 1194071d4279SBram Moolenaar if (strcmp(name, "buffer") == 0) 1195071d4279SBram Moolenaar return (PyObject *)BufferNew(this->win->w_buffer); 1196071d4279SBram Moolenaar else if (strcmp(name, "cursor") == 0) 1197071d4279SBram Moolenaar { 1198071d4279SBram Moolenaar pos_T *pos = &this->win->w_cursor; 1199071d4279SBram Moolenaar 1200071d4279SBram Moolenaar return Py_BuildValue("(ll)", (long)(pos->lnum), (long)(pos->col)); 1201071d4279SBram Moolenaar } 1202071d4279SBram Moolenaar else if (strcmp(name, "height") == 0) 1203071d4279SBram Moolenaar return Py_BuildValue("l", (long)(this->win->w_height)); 1204071d4279SBram Moolenaar #ifdef FEAT_VERTSPLIT 1205071d4279SBram Moolenaar else if (strcmp(name, "width") == 0) 1206071d4279SBram Moolenaar return Py_BuildValue("l", (long)(W_WIDTH(this->win))); 1207071d4279SBram Moolenaar #endif 1208071d4279SBram Moolenaar else if (strcmp(name,"__members__") == 0) 1209071d4279SBram Moolenaar return Py_BuildValue("[sss]", "buffer", "cursor", "height"); 1210071d4279SBram Moolenaar else 1211071d4279SBram Moolenaar return Py_FindMethod(WindowMethods, self, name); 1212071d4279SBram Moolenaar } 1213071d4279SBram Moolenaar 1214071d4279SBram Moolenaar /* Window list object - Definitions 1215071d4279SBram Moolenaar */ 1216071d4279SBram Moolenaar 1217071d4279SBram Moolenaar typedef struct 1218071d4279SBram Moolenaar { 1219071d4279SBram Moolenaar PyObject_HEAD 1220071d4279SBram Moolenaar } 1221071d4279SBram Moolenaar WinListObject; 1222071d4279SBram Moolenaar 1223071d4279SBram Moolenaar static PySequenceMethods WinListAsSeq = { 12242c45e945SBram Moolenaar (PyInquiry) WinListLength, /* sq_length, len(x) */ 1225071d4279SBram Moolenaar (binaryfunc) 0, /* sq_concat, x+y */ 12262c45e945SBram Moolenaar (PyIntArgFunc) 0, /* sq_repeat, x*n */ 12272c45e945SBram Moolenaar (PyIntArgFunc) WinListItem, /* sq_item, x[i] */ 12282c45e945SBram Moolenaar (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */ 12292c45e945SBram Moolenaar (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */ 12302c45e945SBram Moolenaar (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */ 1231071d4279SBram Moolenaar }; 1232071d4279SBram Moolenaar 1233071d4279SBram Moolenaar static PyTypeObject WinListType = { 1234071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1235071d4279SBram Moolenaar 0, 1236071d4279SBram Moolenaar "window list", 1237071d4279SBram Moolenaar sizeof(WinListObject), 1238071d4279SBram Moolenaar 0, 1239071d4279SBram Moolenaar 1240071d4279SBram Moolenaar (destructor) 0, /* tp_dealloc, refcount==0 */ 1241071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1242071d4279SBram Moolenaar (getattrfunc) 0, /* tp_getattr, x.attr */ 1243071d4279SBram Moolenaar (setattrfunc) 0, /* tp_setattr, x.attr=v */ 1244071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1245071d4279SBram Moolenaar (reprfunc) 0, /* tp_repr, `x`, print x */ 1246071d4279SBram Moolenaar 1247071d4279SBram Moolenaar 0, /* as number */ 1248071d4279SBram Moolenaar &WinListAsSeq, /* as sequence */ 1249071d4279SBram Moolenaar 0, /* as mapping */ 1250071d4279SBram Moolenaar 1251071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1252071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1253071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1254071d4279SBram Moolenaar }; 1255071d4279SBram Moolenaar 1256071d4279SBram Moolenaar /* Current items object - Definitions 1257071d4279SBram Moolenaar */ 1258071d4279SBram Moolenaar 1259071d4279SBram Moolenaar typedef struct 1260071d4279SBram Moolenaar { 1261071d4279SBram Moolenaar PyObject_HEAD 1262ca8a4dfeSBram Moolenaar } CurrentObject; 1263071d4279SBram Moolenaar 1264071d4279SBram Moolenaar static PyTypeObject CurrentType = { 1265071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1266071d4279SBram Moolenaar 0, 1267071d4279SBram Moolenaar "current data", 1268071d4279SBram Moolenaar sizeof(CurrentObject), 1269071d4279SBram Moolenaar 0, 1270071d4279SBram Moolenaar 1271071d4279SBram Moolenaar (destructor) 0, /* tp_dealloc, refcount==0 */ 1272071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1273071d4279SBram Moolenaar (getattrfunc) CurrentGetattr, /* tp_getattr, x.attr */ 1274071d4279SBram Moolenaar (setattrfunc) CurrentSetattr, /* tp_setattr, x.attr=v */ 1275071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1276071d4279SBram Moolenaar (reprfunc) 0, /* tp_repr, `x`, print x */ 1277071d4279SBram Moolenaar 1278071d4279SBram Moolenaar 0, /* as number */ 1279071d4279SBram Moolenaar 0, /* as sequence */ 1280071d4279SBram Moolenaar 0, /* as mapping */ 1281071d4279SBram Moolenaar 1282071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1283071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1284071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1285071d4279SBram Moolenaar }; 1286071d4279SBram Moolenaar 1287071d4279SBram Moolenaar /* Current items object - Implementation 1288071d4279SBram Moolenaar */ 1289071d4279SBram Moolenaar static PyObject * 12904bdbbf70SBram Moolenaar CurrentGetattr(PyObject *self UNUSED, char *name) 1291071d4279SBram Moolenaar { 1292071d4279SBram Moolenaar if (strcmp(name, "buffer") == 0) 1293071d4279SBram Moolenaar return (PyObject *)BufferNew(curbuf); 1294071d4279SBram Moolenaar else if (strcmp(name, "window") == 0) 1295071d4279SBram Moolenaar return (PyObject *)WindowNew(curwin); 1296071d4279SBram Moolenaar else if (strcmp(name, "line") == 0) 1297e7cb9cf6SBram Moolenaar return GetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum); 1298071d4279SBram Moolenaar else if (strcmp(name, "range") == 0) 1299071d4279SBram Moolenaar return RangeNew(curbuf, RangeStart, RangeEnd); 1300071d4279SBram Moolenaar else if (strcmp(name,"__members__") == 0) 1301071d4279SBram Moolenaar return Py_BuildValue("[ssss]", "buffer", "window", "line", "range"); 1302071d4279SBram Moolenaar else 1303071d4279SBram Moolenaar { 1304071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, name); 1305071d4279SBram Moolenaar return NULL; 1306071d4279SBram Moolenaar } 1307071d4279SBram Moolenaar } 1308071d4279SBram Moolenaar 1309071d4279SBram Moolenaar static int 13104bdbbf70SBram Moolenaar CurrentSetattr(PyObject *self UNUSED, char *name, PyObject *value) 1311071d4279SBram Moolenaar { 1312071d4279SBram Moolenaar if (strcmp(name, "line") == 0) 1313071d4279SBram Moolenaar { 1314e7cb9cf6SBram Moolenaar if (SetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum, value, NULL) == FAIL) 1315071d4279SBram Moolenaar return -1; 1316071d4279SBram Moolenaar 1317071d4279SBram Moolenaar return 0; 1318071d4279SBram Moolenaar } 1319071d4279SBram Moolenaar else 1320071d4279SBram Moolenaar { 1321071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, name); 1322071d4279SBram Moolenaar return -1; 1323071d4279SBram Moolenaar } 1324071d4279SBram Moolenaar } 1325071d4279SBram Moolenaar 1326071d4279SBram Moolenaar /* External interface 1327071d4279SBram Moolenaar */ 1328071d4279SBram Moolenaar 1329071d4279SBram Moolenaar void 1330071d4279SBram Moolenaar python_buffer_free(buf_T *buf) 1331071d4279SBram Moolenaar { 1332e344beadSBram Moolenaar if (buf->b_python_ref != NULL) 1333071d4279SBram Moolenaar { 1334e344beadSBram Moolenaar BufferObject *bp = buf->b_python_ref; 1335071d4279SBram Moolenaar bp->buf = INVALID_BUFFER_VALUE; 1336e344beadSBram Moolenaar buf->b_python_ref = NULL; 1337071d4279SBram Moolenaar } 1338071d4279SBram Moolenaar } 1339071d4279SBram Moolenaar 1340071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO) 1341071d4279SBram Moolenaar void 1342071d4279SBram Moolenaar python_window_free(win_T *win) 1343071d4279SBram Moolenaar { 1344e344beadSBram Moolenaar if (win->w_python_ref != NULL) 1345071d4279SBram Moolenaar { 1346e344beadSBram Moolenaar WindowObject *wp = win->w_python_ref; 1347071d4279SBram Moolenaar wp->win = INVALID_WINDOW_VALUE; 1348e344beadSBram Moolenaar win->w_python_ref = NULL; 1349071d4279SBram Moolenaar } 1350071d4279SBram Moolenaar } 1351071d4279SBram Moolenaar #endif 1352071d4279SBram Moolenaar 1353071d4279SBram Moolenaar static BufListObject TheBufferList = 1354071d4279SBram Moolenaar { 1355071d4279SBram Moolenaar PyObject_HEAD_INIT(&BufListType) 1356071d4279SBram Moolenaar }; 1357071d4279SBram Moolenaar 1358071d4279SBram Moolenaar static WinListObject TheWindowList = 1359071d4279SBram Moolenaar { 1360071d4279SBram Moolenaar PyObject_HEAD_INIT(&WinListType) 1361071d4279SBram Moolenaar }; 1362071d4279SBram Moolenaar 1363071d4279SBram Moolenaar static CurrentObject TheCurrent = 1364071d4279SBram Moolenaar { 1365071d4279SBram Moolenaar PyObject_HEAD_INIT(&CurrentType) 1366071d4279SBram Moolenaar }; 1367071d4279SBram Moolenaar 1368071d4279SBram Moolenaar static int 1369071d4279SBram Moolenaar PythonMod_Init(void) 1370071d4279SBram Moolenaar { 1371071d4279SBram Moolenaar PyObject *mod; 1372071d4279SBram Moolenaar PyObject *dict; 13739774ecc8SBram Moolenaar /* The special value is removed from sys.path in Python_Init(). */ 13749774ecc8SBram Moolenaar static char *(argv[2]) = {"/must>not&exist/foo", NULL}; 1375071d4279SBram Moolenaar 1376071d4279SBram Moolenaar /* Fixups... */ 137721377c8dSBram Moolenaar PyType_Ready(&BufferType); 137821377c8dSBram Moolenaar PyType_Ready(&RangeType); 137921377c8dSBram Moolenaar PyType_Ready(&WindowType); 138021377c8dSBram Moolenaar PyType_Ready(&BufListType); 138121377c8dSBram Moolenaar PyType_Ready(&WinListType); 138221377c8dSBram Moolenaar PyType_Ready(&CurrentType); 1383071d4279SBram Moolenaar 1384071d4279SBram Moolenaar /* Set sys.argv[] to avoid a crash in warn(). */ 1385071d4279SBram Moolenaar PySys_SetArgv(1, argv); 1386071d4279SBram Moolenaar 1387e7cb9cf6SBram Moolenaar mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION); 1388071d4279SBram Moolenaar dict = PyModule_GetDict(mod); 1389071d4279SBram Moolenaar 1390071d4279SBram Moolenaar VimError = Py_BuildValue("s", "vim.error"); 1391071d4279SBram Moolenaar 1392071d4279SBram Moolenaar PyDict_SetItemString(dict, "error", VimError); 13937df2d662SBram Moolenaar PyDict_SetItemString(dict, "buffers", (PyObject *)(void *)&TheBufferList); 13947df2d662SBram Moolenaar PyDict_SetItemString(dict, "current", (PyObject *)(void *)&TheCurrent); 13957df2d662SBram Moolenaar PyDict_SetItemString(dict, "windows", (PyObject *)(void *)&TheWindowList); 1396071d4279SBram Moolenaar 1397071d4279SBram Moolenaar if (PyErr_Occurred()) 1398071d4279SBram Moolenaar return -1; 1399071d4279SBram Moolenaar 1400071d4279SBram Moolenaar return 0; 1401071d4279SBram Moolenaar } 1402071d4279SBram Moolenaar 1403071d4279SBram Moolenaar /************************************************************************* 1404071d4279SBram Moolenaar * 4. Utility functions for handling the interface between Vim and Python. 1405071d4279SBram Moolenaar */ 1406071d4279SBram Moolenaar 1407071d4279SBram Moolenaar /* Convert a Vim line into a Python string. 1408071d4279SBram Moolenaar * All internal newlines are replaced by null characters. 1409071d4279SBram Moolenaar * 1410071d4279SBram Moolenaar * On errors, the Python exception data is set, and NULL is returned. 1411071d4279SBram Moolenaar */ 1412071d4279SBram Moolenaar static PyObject * 1413071d4279SBram Moolenaar LineToString(const char *str) 1414071d4279SBram Moolenaar { 1415071d4279SBram Moolenaar PyObject *result; 14162c45e945SBram Moolenaar PyInt len = strlen(str); 1417071d4279SBram Moolenaar char *p; 1418071d4279SBram Moolenaar 1419071d4279SBram Moolenaar /* Allocate an Python string object, with uninitialised contents. We 1420071d4279SBram Moolenaar * must do it this way, so that we can modify the string in place 1421071d4279SBram Moolenaar * later. See the Python source, Objects/stringobject.c for details. 1422071d4279SBram Moolenaar */ 1423071d4279SBram Moolenaar result = PyString_FromStringAndSize(NULL, len); 1424071d4279SBram Moolenaar if (result == NULL) 1425071d4279SBram Moolenaar return NULL; 1426071d4279SBram Moolenaar 1427071d4279SBram Moolenaar p = PyString_AsString(result); 1428071d4279SBram Moolenaar 1429071d4279SBram Moolenaar while (*str) 1430071d4279SBram Moolenaar { 1431071d4279SBram Moolenaar if (*str == '\n') 1432071d4279SBram Moolenaar *p = '\0'; 1433071d4279SBram Moolenaar else 1434071d4279SBram Moolenaar *p = *str; 1435071d4279SBram Moolenaar 1436071d4279SBram Moolenaar ++p; 1437071d4279SBram Moolenaar ++str; 1438071d4279SBram Moolenaar } 1439071d4279SBram Moolenaar 1440071d4279SBram Moolenaar return result; 1441071d4279SBram Moolenaar } 1442071d4279SBram Moolenaar 1443071d4279SBram Moolenaar 1444071d4279SBram Moolenaar /* Don't generate a prototype for the next function, it generates an error on 1445071d4279SBram Moolenaar * newer Python versions. */ 1446071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO) 1447071d4279SBram Moolenaar 1448071d4279SBram Moolenaar char * 1449071d4279SBram Moolenaar Py_GetProgramName(void) 1450071d4279SBram Moolenaar { 1451071d4279SBram Moolenaar return "vim"; 1452071d4279SBram Moolenaar } 1453071d4279SBram Moolenaar #endif /* Python 1.4 */ 1454170bf1aeSBram Moolenaar 1455170bf1aeSBram Moolenaar static void 1456170bf1aeSBram Moolenaar init_structs(void) 1457170bf1aeSBram Moolenaar { 1458170bf1aeSBram Moolenaar vim_memset(&OutputType, 0, sizeof(OutputType)); 1459170bf1aeSBram Moolenaar OutputType.tp_name = "message"; 1460170bf1aeSBram Moolenaar OutputType.tp_basicsize = sizeof(OutputObject); 1461170bf1aeSBram Moolenaar OutputType.tp_getattr = OutputGetattr; 1462170bf1aeSBram Moolenaar OutputType.tp_setattr = OutputSetattr; 1463ca8a4dfeSBram Moolenaar 1464ca8a4dfeSBram Moolenaar vim_memset(&RangeType, 0, sizeof(RangeType)); 1465ca8a4dfeSBram Moolenaar RangeType.tp_name = "range"; 1466ca8a4dfeSBram Moolenaar RangeType.tp_basicsize = sizeof(RangeObject); 1467ca8a4dfeSBram Moolenaar RangeType.tp_dealloc = RangeDestructor; 1468ca8a4dfeSBram Moolenaar RangeType.tp_getattr = RangeGetattr; 1469ca8a4dfeSBram Moolenaar RangeType.tp_repr = RangeRepr; 1470ca8a4dfeSBram Moolenaar RangeType.tp_as_sequence = &RangeAsSeq; 1471170bf1aeSBram Moolenaar } 1472