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 5919e60943SBram Moolenaar /* No-op conversion functions, use with care! */ 6019e60943SBram Moolenaar #define PyString_AsBytes(obj) (obj) 6119e60943SBram Moolenaar #define PyString_FreeBytes(obj) 6219e60943SBram 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 13619e60943SBram 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 *, ...); 19719e60943SBram 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}, 28019e60943SBram 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 { 371*9dc93ae4SBram Moolenaar if (verbose) 372b744b2faSBram Moolenaar EMSG(_("E836: This Vim cannot execute :python after using :py3")); 3734c3a326cSBram Moolenaar return FAIL; 3744c3a326cSBram Moolenaar } 3754c3a326cSBram Moolenaar #endif 3764c3a326cSBram Moolenaar 377071d4279SBram Moolenaar if (hinstPython) 378071d4279SBram Moolenaar return OK; 379bd5e15fdSBram Moolenaar hinstPython = load_dll(libname); 380071d4279SBram Moolenaar if (!hinstPython) 381071d4279SBram Moolenaar { 382071d4279SBram Moolenaar if (verbose) 383071d4279SBram Moolenaar EMSG2(_(e_loadlib), libname); 384071d4279SBram Moolenaar return FAIL; 385071d4279SBram Moolenaar } 386071d4279SBram Moolenaar 387071d4279SBram Moolenaar for (i = 0; python_funcname_table[i].ptr; ++i) 388071d4279SBram Moolenaar { 389bd5e15fdSBram Moolenaar if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython, 390071d4279SBram Moolenaar python_funcname_table[i].name)) == NULL) 391071d4279SBram Moolenaar { 392bd5e15fdSBram Moolenaar close_dll(hinstPython); 393071d4279SBram Moolenaar hinstPython = 0; 394071d4279SBram Moolenaar if (verbose) 395071d4279SBram Moolenaar EMSG2(_(e_loadfunc), python_funcname_table[i].name); 396071d4279SBram Moolenaar return FAIL; 397071d4279SBram Moolenaar } 398071d4279SBram Moolenaar } 399071d4279SBram Moolenaar return OK; 400071d4279SBram Moolenaar } 401071d4279SBram Moolenaar 402071d4279SBram Moolenaar /* 403071d4279SBram Moolenaar * If python is enabled (there is installed python on Windows system) return 404071d4279SBram Moolenaar * TRUE, else FALSE. 405071d4279SBram Moolenaar */ 406071d4279SBram Moolenaar int 407e7cb9cf6SBram Moolenaar python_enabled(int verbose) 408071d4279SBram Moolenaar { 409071d4279SBram Moolenaar return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK; 410071d4279SBram Moolenaar } 411071d4279SBram Moolenaar 412ca8a4dfeSBram Moolenaar /* 413ca8a4dfeSBram Moolenaar * Load the standard Python exceptions - don't import the symbols from the 414071d4279SBram Moolenaar * DLL, as this can cause errors (importing data symbols is not reliable). 415071d4279SBram Moolenaar */ 416071d4279SBram Moolenaar static void 417ca8a4dfeSBram Moolenaar get_exceptions(void) 418071d4279SBram Moolenaar { 419071d4279SBram Moolenaar PyObject *exmod = PyImport_ImportModule("exceptions"); 420071d4279SBram Moolenaar PyObject *exdict = PyModule_GetDict(exmod); 421071d4279SBram Moolenaar imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError"); 422071d4279SBram Moolenaar imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError"); 423071d4279SBram Moolenaar imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt"); 424071d4279SBram Moolenaar imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError"); 425071d4279SBram Moolenaar imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError"); 426071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_AttributeError); 427071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_IndexError); 428071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_KeyboardInterrupt); 429071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_TypeError); 430071d4279SBram Moolenaar Py_XINCREF(imp_PyExc_ValueError); 431071d4279SBram Moolenaar Py_XDECREF(exmod); 432071d4279SBram Moolenaar } 433071d4279SBram Moolenaar #endif /* DYNAMIC_PYTHON */ 434071d4279SBram Moolenaar 435ca8a4dfeSBram Moolenaar static PyObject *BufferNew (buf_T *); 436ca8a4dfeSBram Moolenaar static PyObject *WindowNew(win_T *); 437ca8a4dfeSBram Moolenaar static PyObject *LineToString(const char *); 438ca8a4dfeSBram Moolenaar 439ca8a4dfeSBram Moolenaar static PyTypeObject RangeType; 440ca8a4dfeSBram Moolenaar 441170bf1aeSBram Moolenaar /* 442170bf1aeSBram Moolenaar * Include the code shared with if_python3.c 443170bf1aeSBram Moolenaar */ 444170bf1aeSBram Moolenaar #include "if_py_both.h" 445170bf1aeSBram Moolenaar 446170bf1aeSBram Moolenaar 447071d4279SBram Moolenaar /****************************************************** 448071d4279SBram Moolenaar * Internal function prototypes. 449071d4279SBram Moolenaar */ 450071d4279SBram Moolenaar 451e7cb9cf6SBram Moolenaar static PyInt RangeStart; 452e7cb9cf6SBram Moolenaar static PyInt RangeEnd; 453071d4279SBram Moolenaar 454071d4279SBram Moolenaar static void PythonIO_Flush(void); 455071d4279SBram Moolenaar static int PythonIO_Init(void); 456071d4279SBram Moolenaar static int PythonMod_Init(void); 457071d4279SBram Moolenaar 458071d4279SBram Moolenaar /* Utility functions for the vim/python interface 459071d4279SBram Moolenaar * ---------------------------------------------- 460071d4279SBram Moolenaar */ 461071d4279SBram Moolenaar 462e7cb9cf6SBram Moolenaar static int SetBufferLineList(buf_T *, PyInt, PyInt, PyObject *, PyInt *); 463071d4279SBram Moolenaar 464071d4279SBram Moolenaar 465071d4279SBram Moolenaar /****************************************************** 466071d4279SBram Moolenaar * 1. Python interpreter main program. 467071d4279SBram Moolenaar */ 468071d4279SBram Moolenaar 469071d4279SBram Moolenaar static int initialised = 0; 470071d4279SBram Moolenaar 471071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ 472071d4279SBram Moolenaar typedef PyObject PyThreadState; 4739ba0eb85SBram Moolenaar #endif 474071d4279SBram Moolenaar 4759ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE 4769ba0eb85SBram Moolenaar static PyGILState_STATE pygilstate = PyGILState_UNLOCKED; 4779ba0eb85SBram Moolenaar #else 478071d4279SBram Moolenaar static PyThreadState *saved_python_thread = NULL; 4799ba0eb85SBram Moolenaar #endif 480071d4279SBram Moolenaar 481071d4279SBram Moolenaar /* 482071d4279SBram Moolenaar * Suspend a thread of the Python interpreter, other threads are allowed to 483071d4279SBram Moolenaar * run. 484071d4279SBram Moolenaar */ 485293ee4d4SBram Moolenaar static void 486293ee4d4SBram Moolenaar Python_SaveThread(void) 487071d4279SBram Moolenaar { 4889ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE 4899ba0eb85SBram Moolenaar PyGILState_Release(pygilstate); 4909ba0eb85SBram Moolenaar #else 491071d4279SBram Moolenaar saved_python_thread = PyEval_SaveThread(); 4929ba0eb85SBram Moolenaar #endif 493071d4279SBram Moolenaar } 494071d4279SBram Moolenaar 495071d4279SBram Moolenaar /* 496071d4279SBram Moolenaar * Restore a thread of the Python interpreter, waits for other threads to 497071d4279SBram Moolenaar * block. 498071d4279SBram Moolenaar */ 499293ee4d4SBram Moolenaar static void 500293ee4d4SBram Moolenaar Python_RestoreThread(void) 501071d4279SBram Moolenaar { 5029ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE 5039ba0eb85SBram Moolenaar pygilstate = PyGILState_Ensure(); 5049ba0eb85SBram Moolenaar #else 505071d4279SBram Moolenaar PyEval_RestoreThread(saved_python_thread); 506071d4279SBram Moolenaar saved_python_thread = NULL; 507071d4279SBram Moolenaar #endif 5089ba0eb85SBram Moolenaar } 509071d4279SBram Moolenaar 510071d4279SBram Moolenaar void 511071d4279SBram Moolenaar python_end() 512071d4279SBram Moolenaar { 513a5792f58SBram Moolenaar static int recurse = 0; 514a5792f58SBram Moolenaar 515a5792f58SBram Moolenaar /* If a crash occurs while doing this, don't try again. */ 516a5792f58SBram Moolenaar if (recurse != 0) 517a5792f58SBram Moolenaar return; 518a5792f58SBram Moolenaar 519a5792f58SBram Moolenaar ++recurse; 520a5792f58SBram Moolenaar 521071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON 5220e21a3f6SBram Moolenaar if (hinstPython && Py_IsInitialized()) 5239ba0eb85SBram Moolenaar { 5249ba0eb85SBram Moolenaar Python_RestoreThread(); /* enter python */ 5250e21a3f6SBram Moolenaar Py_Finalize(); 5269ba0eb85SBram Moolenaar } 527071d4279SBram Moolenaar end_dynamic_python(); 5280e21a3f6SBram Moolenaar #else 5290e21a3f6SBram Moolenaar if (Py_IsInitialized()) 5309ba0eb85SBram Moolenaar { 5319ba0eb85SBram Moolenaar Python_RestoreThread(); /* enter python */ 5320e21a3f6SBram Moolenaar Py_Finalize(); 5339ba0eb85SBram Moolenaar } 534071d4279SBram Moolenaar #endif 535a5792f58SBram Moolenaar 536a5792f58SBram Moolenaar --recurse; 537071d4279SBram Moolenaar } 538071d4279SBram Moolenaar 5394c3a326cSBram Moolenaar #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO) 5404c3a326cSBram Moolenaar int 5414c3a326cSBram Moolenaar python_loaded() 5424c3a326cSBram Moolenaar { 5434c3a326cSBram Moolenaar return (hinstPython != 0); 5444c3a326cSBram Moolenaar } 5454c3a326cSBram Moolenaar #endif 5464c3a326cSBram Moolenaar 547071d4279SBram Moolenaar static int 548071d4279SBram Moolenaar Python_Init(void) 549071d4279SBram Moolenaar { 550071d4279SBram Moolenaar if (!initialised) 551071d4279SBram Moolenaar { 552071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON 553071d4279SBram Moolenaar if (!python_enabled(TRUE)) 554071d4279SBram Moolenaar { 555071d4279SBram Moolenaar EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded.")); 556071d4279SBram Moolenaar goto fail; 557071d4279SBram Moolenaar } 558071d4279SBram Moolenaar #endif 559071d4279SBram Moolenaar 560644d37b8SBram Moolenaar #ifdef PYTHON_HOME 561644d37b8SBram Moolenaar Py_SetPythonHome(PYTHON_HOME); 562644d37b8SBram Moolenaar #endif 563644d37b8SBram Moolenaar 564170bf1aeSBram Moolenaar init_structs(); 565170bf1aeSBram Moolenaar 566071d4279SBram Moolenaar #if !defined(MACOS) || defined(MACOS_X_UNIX) 567071d4279SBram Moolenaar Py_Initialize(); 568071d4279SBram Moolenaar #else 569071d4279SBram Moolenaar PyMac_Initialize(); 570071d4279SBram Moolenaar #endif 571071d4279SBram Moolenaar /* initialise threads */ 572071d4279SBram Moolenaar PyEval_InitThreads(); 573071d4279SBram Moolenaar 574071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON 575071d4279SBram Moolenaar get_exceptions(); 576071d4279SBram Moolenaar #endif 577071d4279SBram Moolenaar 578071d4279SBram Moolenaar if (PythonIO_Init()) 579071d4279SBram Moolenaar goto fail; 580071d4279SBram Moolenaar 581071d4279SBram Moolenaar if (PythonMod_Init()) 582071d4279SBram Moolenaar goto fail; 583071d4279SBram Moolenaar 5849774ecc8SBram Moolenaar /* Remove the element from sys.path that was added because of our 5859774ecc8SBram Moolenaar * argv[0] value in PythonMod_Init(). Previously we used an empty 5869774ecc8SBram Moolenaar * string, but dependinding on the OS we then get an empty entry or 5879774ecc8SBram Moolenaar * the current directory in sys.path. */ 5889774ecc8SBram Moolenaar PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)"); 5899774ecc8SBram Moolenaar 590293ee4d4SBram Moolenaar /* the first python thread is vim's, release the lock */ 591071d4279SBram Moolenaar Python_SaveThread(); 592071d4279SBram Moolenaar 593071d4279SBram Moolenaar initialised = 1; 594071d4279SBram Moolenaar } 595071d4279SBram Moolenaar 596071d4279SBram Moolenaar return 0; 597071d4279SBram Moolenaar 598071d4279SBram Moolenaar fail: 599071d4279SBram Moolenaar /* We call PythonIO_Flush() here to print any Python errors. 600071d4279SBram Moolenaar * This is OK, as it is possible to call this function even 601071d4279SBram Moolenaar * if PythonIO_Init() has not completed successfully (it will 602071d4279SBram Moolenaar * not do anything in this case). 603071d4279SBram Moolenaar */ 604071d4279SBram Moolenaar PythonIO_Flush(); 605071d4279SBram Moolenaar return -1; 606071d4279SBram Moolenaar } 607071d4279SBram Moolenaar 608071d4279SBram Moolenaar /* 609071d4279SBram Moolenaar * External interface 610071d4279SBram Moolenaar */ 611071d4279SBram Moolenaar static void 612071d4279SBram Moolenaar DoPythonCommand(exarg_T *eap, const char *cmd) 613071d4279SBram Moolenaar { 6149ba0eb85SBram Moolenaar #ifndef PY_CAN_RECURSE 615071d4279SBram Moolenaar static int recursive = 0; 616071d4279SBram Moolenaar #endif 617071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 618071d4279SBram Moolenaar GrafPtr oldPort; 619071d4279SBram Moolenaar #endif 620071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 621071d4279SBram Moolenaar char *saved_locale; 622071d4279SBram Moolenaar #endif 623071d4279SBram Moolenaar 624071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE 625071d4279SBram Moolenaar if (recursive) 626071d4279SBram Moolenaar { 627071d4279SBram Moolenaar EMSG(_("E659: Cannot invoke Python recursively")); 628071d4279SBram Moolenaar return; 629071d4279SBram Moolenaar } 630071d4279SBram Moolenaar ++recursive; 631071d4279SBram Moolenaar #endif 632071d4279SBram Moolenaar 633071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 634071d4279SBram Moolenaar GetPort(&oldPort); 635071d4279SBram Moolenaar /* Check if the Python library is available */ 636071d4279SBram Moolenaar if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress) 637071d4279SBram Moolenaar goto theend; 638071d4279SBram Moolenaar #endif 639071d4279SBram Moolenaar if (Python_Init()) 640071d4279SBram Moolenaar goto theend; 641071d4279SBram Moolenaar 642071d4279SBram Moolenaar RangeStart = eap->line1; 643071d4279SBram Moolenaar RangeEnd = eap->line2; 644071d4279SBram Moolenaar Python_Release_Vim(); /* leave vim */ 645071d4279SBram Moolenaar 646071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 647071d4279SBram Moolenaar /* Python only works properly when the LC_NUMERIC locale is "C". */ 648071d4279SBram Moolenaar saved_locale = setlocale(LC_NUMERIC, NULL); 649071d4279SBram Moolenaar if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0) 650071d4279SBram Moolenaar saved_locale = NULL; 651071d4279SBram Moolenaar else 652071d4279SBram Moolenaar { 653071d4279SBram Moolenaar /* Need to make a copy, value may change when setting new locale. */ 654071d4279SBram Moolenaar saved_locale = (char *)vim_strsave((char_u *)saved_locale); 655071d4279SBram Moolenaar (void)setlocale(LC_NUMERIC, "C"); 656071d4279SBram Moolenaar } 657071d4279SBram Moolenaar #endif 658071d4279SBram Moolenaar 659071d4279SBram Moolenaar Python_RestoreThread(); /* enter python */ 660071d4279SBram Moolenaar 661071d4279SBram Moolenaar PyRun_SimpleString((char *)(cmd)); 662071d4279SBram Moolenaar 663071d4279SBram Moolenaar Python_SaveThread(); /* leave python */ 664071d4279SBram Moolenaar 665071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 666071d4279SBram Moolenaar if (saved_locale != NULL) 667071d4279SBram Moolenaar { 668071d4279SBram Moolenaar (void)setlocale(LC_NUMERIC, saved_locale); 669071d4279SBram Moolenaar vim_free(saved_locale); 670071d4279SBram Moolenaar } 671071d4279SBram Moolenaar #endif 672071d4279SBram Moolenaar 673071d4279SBram Moolenaar Python_Lock_Vim(); /* enter vim */ 674071d4279SBram Moolenaar PythonIO_Flush(); 675071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX) 676071d4279SBram Moolenaar SetPort(oldPort); 677071d4279SBram Moolenaar #endif 678071d4279SBram Moolenaar 679071d4279SBram Moolenaar theend: 680071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE 681071d4279SBram Moolenaar --recursive; 682071d4279SBram Moolenaar #endif 683071d4279SBram Moolenaar return; /* keeps lint happy */ 684071d4279SBram Moolenaar } 685071d4279SBram Moolenaar 686071d4279SBram Moolenaar /* 687071d4279SBram Moolenaar * ":python" 688071d4279SBram Moolenaar */ 689071d4279SBram Moolenaar void 690071d4279SBram Moolenaar ex_python(exarg_T *eap) 691071d4279SBram Moolenaar { 692071d4279SBram Moolenaar char_u *script; 693071d4279SBram Moolenaar 694071d4279SBram Moolenaar script = script_get(eap, eap->arg); 695071d4279SBram Moolenaar if (!eap->skip) 696071d4279SBram Moolenaar { 697071d4279SBram Moolenaar if (script == NULL) 698071d4279SBram Moolenaar DoPythonCommand(eap, (char *)eap->arg); 699071d4279SBram Moolenaar else 700071d4279SBram Moolenaar DoPythonCommand(eap, (char *)script); 701071d4279SBram Moolenaar } 702071d4279SBram Moolenaar vim_free(script); 703071d4279SBram Moolenaar } 704071d4279SBram Moolenaar 705071d4279SBram Moolenaar #define BUFFER_SIZE 1024 706071d4279SBram Moolenaar 707071d4279SBram Moolenaar /* 708071d4279SBram Moolenaar * ":pyfile" 709071d4279SBram Moolenaar */ 710071d4279SBram Moolenaar void 711071d4279SBram Moolenaar ex_pyfile(exarg_T *eap) 712071d4279SBram Moolenaar { 713071d4279SBram Moolenaar static char buffer[BUFFER_SIZE]; 714071d4279SBram Moolenaar const char *file = (char *)eap->arg; 715071d4279SBram Moolenaar char *p; 716071d4279SBram Moolenaar 717071d4279SBram Moolenaar /* Have to do it like this. PyRun_SimpleFile requires you to pass a 718071d4279SBram Moolenaar * stdio file pointer, but Vim and the Python DLL are compiled with 719071d4279SBram Moolenaar * different options under Windows, meaning that stdio pointers aren't 720071d4279SBram Moolenaar * compatible between the two. Yuk. 721071d4279SBram Moolenaar * 722071d4279SBram Moolenaar * Put the string "execfile('file')" into buffer. But, we need to 723071d4279SBram Moolenaar * escape any backslashes or single quotes in the file name, so that 724071d4279SBram Moolenaar * Python won't mangle the file name. 725071d4279SBram Moolenaar */ 726071d4279SBram Moolenaar strcpy(buffer, "execfile('"); 727071d4279SBram Moolenaar p = buffer + 10; /* size of "execfile('" */ 728071d4279SBram Moolenaar 729071d4279SBram Moolenaar while (*file && p < buffer + (BUFFER_SIZE - 3)) 730071d4279SBram Moolenaar { 731071d4279SBram Moolenaar if (*file == '\\' || *file == '\'') 732071d4279SBram Moolenaar *p++ = '\\'; 733071d4279SBram Moolenaar *p++ = *file++; 734071d4279SBram Moolenaar } 735071d4279SBram Moolenaar 736071d4279SBram Moolenaar /* If we didn't finish the file name, we hit a buffer overflow */ 737071d4279SBram Moolenaar if (*file != '\0') 738071d4279SBram Moolenaar return; 739071d4279SBram Moolenaar 740071d4279SBram Moolenaar /* Put in the terminating "')" and a null */ 741071d4279SBram Moolenaar *p++ = '\''; 742071d4279SBram Moolenaar *p++ = ')'; 743071d4279SBram Moolenaar *p++ = '\0'; 744071d4279SBram Moolenaar 745071d4279SBram Moolenaar /* Execute the file */ 746071d4279SBram Moolenaar DoPythonCommand(eap, buffer); 747071d4279SBram Moolenaar } 748071d4279SBram Moolenaar 749071d4279SBram Moolenaar /****************************************************** 750071d4279SBram Moolenaar * 2. Python output stream: writes output via [e]msg(). 751071d4279SBram Moolenaar */ 752071d4279SBram Moolenaar 753071d4279SBram Moolenaar /* Implementation functions 754071d4279SBram Moolenaar */ 755071d4279SBram Moolenaar 756071d4279SBram Moolenaar static PyObject * 757071d4279SBram Moolenaar OutputGetattr(PyObject *self, char *name) 758071d4279SBram Moolenaar { 759071d4279SBram Moolenaar if (strcmp(name, "softspace") == 0) 760071d4279SBram Moolenaar return PyInt_FromLong(((OutputObject *)(self))->softspace); 761071d4279SBram Moolenaar 762071d4279SBram Moolenaar return Py_FindMethod(OutputMethods, self, name); 763071d4279SBram Moolenaar } 764071d4279SBram Moolenaar 765071d4279SBram Moolenaar static int 766071d4279SBram Moolenaar OutputSetattr(PyObject *self, char *name, PyObject *val) 767071d4279SBram Moolenaar { 768071d4279SBram Moolenaar if (val == NULL) { 769071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, _("can't delete OutputObject attributes")); 770071d4279SBram Moolenaar return -1; 771071d4279SBram Moolenaar } 772071d4279SBram Moolenaar 773071d4279SBram Moolenaar if (strcmp(name, "softspace") == 0) 774071d4279SBram Moolenaar { 775071d4279SBram Moolenaar if (!PyInt_Check(val)) { 776071d4279SBram Moolenaar PyErr_SetString(PyExc_TypeError, _("softspace must be an integer")); 777071d4279SBram Moolenaar return -1; 778071d4279SBram Moolenaar } 779071d4279SBram Moolenaar 780071d4279SBram Moolenaar ((OutputObject *)(self))->softspace = PyInt_AsLong(val); 781071d4279SBram Moolenaar return 0; 782071d4279SBram Moolenaar } 783071d4279SBram Moolenaar 784071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, _("invalid attribute")); 785071d4279SBram Moolenaar return -1; 786071d4279SBram Moolenaar } 787071d4279SBram Moolenaar 788071d4279SBram Moolenaar /***************/ 789071d4279SBram Moolenaar 790071d4279SBram Moolenaar static int 791071d4279SBram Moolenaar PythonIO_Init(void) 792071d4279SBram Moolenaar { 793071d4279SBram Moolenaar /* Fixups... */ 79421377c8dSBram Moolenaar PyType_Ready(&OutputType); 795071d4279SBram Moolenaar 796170bf1aeSBram Moolenaar return PythonIO_Init_io(); 797071d4279SBram Moolenaar } 798071d4279SBram Moolenaar 799071d4279SBram Moolenaar /****************************************************** 800071d4279SBram Moolenaar * 3. Implementation of the Vim module for Python 801071d4279SBram Moolenaar */ 802071d4279SBram Moolenaar 803071d4279SBram Moolenaar /* Window type - Implementation functions 804071d4279SBram Moolenaar * -------------------------------------- 805071d4279SBram Moolenaar */ 806071d4279SBram Moolenaar 807071d4279SBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_type == &WindowType) 808071d4279SBram Moolenaar 809071d4279SBram Moolenaar static void WindowDestructor(PyObject *); 810071d4279SBram Moolenaar static PyObject *WindowGetattr(PyObject *, char *); 811071d4279SBram Moolenaar 812071d4279SBram Moolenaar /* Buffer type - Implementation functions 813071d4279SBram Moolenaar * -------------------------------------- 814071d4279SBram Moolenaar */ 815071d4279SBram Moolenaar 816071d4279SBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_type == &BufferType) 817071d4279SBram Moolenaar 818071d4279SBram Moolenaar static void BufferDestructor(PyObject *); 819071d4279SBram Moolenaar static PyObject *BufferGetattr(PyObject *, char *); 820071d4279SBram Moolenaar static PyObject *BufferRepr(PyObject *); 821071d4279SBram Moolenaar 8222c45e945SBram Moolenaar static PyInt BufferLength(PyObject *); 8232c45e945SBram Moolenaar static PyObject *BufferItem(PyObject *, PyInt); 8242c45e945SBram Moolenaar static PyObject *BufferSlice(PyObject *, PyInt, PyInt); 8252c45e945SBram Moolenaar static PyInt BufferAssItem(PyObject *, PyInt, PyObject *); 8262c45e945SBram Moolenaar static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *); 827071d4279SBram Moolenaar 828071d4279SBram Moolenaar /* Line range type - Implementation functions 829071d4279SBram Moolenaar * -------------------------------------- 830071d4279SBram Moolenaar */ 831071d4279SBram Moolenaar 832071d4279SBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_type == &RangeType) 833071d4279SBram Moolenaar 8342c45e945SBram Moolenaar static PyInt RangeAssItem(PyObject *, PyInt, PyObject *); 8352c45e945SBram Moolenaar static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *); 836071d4279SBram Moolenaar 837071d4279SBram Moolenaar /* Current objects type - Implementation functions 838071d4279SBram Moolenaar * ----------------------------------------------- 839071d4279SBram Moolenaar */ 840071d4279SBram Moolenaar 841071d4279SBram Moolenaar static PyObject *CurrentGetattr(PyObject *, char *); 842071d4279SBram Moolenaar static int CurrentSetattr(PyObject *, char *, PyObject *); 843071d4279SBram Moolenaar 844071d4279SBram Moolenaar static PySequenceMethods BufferAsSeq = { 8452c45e945SBram Moolenaar (PyInquiry) BufferLength, /* sq_length, len(x) */ 846071d4279SBram Moolenaar (binaryfunc) 0, /* BufferConcat, */ /* sq_concat, x+y */ 8472c45e945SBram Moolenaar (PyIntArgFunc) 0, /* BufferRepeat, */ /* sq_repeat, x*n */ 8482c45e945SBram Moolenaar (PyIntArgFunc) BufferItem, /* sq_item, x[i] */ 8492c45e945SBram Moolenaar (PyIntIntArgFunc) BufferSlice, /* sq_slice, x[i:j] */ 8502c45e945SBram Moolenaar (PyIntObjArgProc) BufferAssItem, /* sq_ass_item, x[i]=v */ 8512c45e945SBram Moolenaar (PyIntIntObjArgProc) BufferAssSlice, /* sq_ass_slice, x[i:j]=v */ 852071d4279SBram Moolenaar }; 853071d4279SBram Moolenaar 854071d4279SBram Moolenaar static PyTypeObject BufferType = { 855071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 856071d4279SBram Moolenaar 0, 857071d4279SBram Moolenaar "buffer", 858071d4279SBram Moolenaar sizeof(BufferObject), 859071d4279SBram Moolenaar 0, 860071d4279SBram Moolenaar 861071d4279SBram Moolenaar (destructor) BufferDestructor, /* tp_dealloc, refcount==0 */ 862071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 863071d4279SBram Moolenaar (getattrfunc) BufferGetattr, /* tp_getattr, x.attr */ 864071d4279SBram Moolenaar (setattrfunc) 0, /* tp_setattr, x.attr=v */ 865071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 866071d4279SBram Moolenaar (reprfunc) BufferRepr, /* tp_repr, `x`, print x */ 867071d4279SBram Moolenaar 868071d4279SBram Moolenaar 0, /* as number */ 869071d4279SBram Moolenaar &BufferAsSeq, /* as sequence */ 870071d4279SBram Moolenaar 0, /* as mapping */ 871071d4279SBram Moolenaar 872071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 873071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 874071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 875071d4279SBram Moolenaar }; 876071d4279SBram Moolenaar 877071d4279SBram Moolenaar /* Buffer object - Implementation 878071d4279SBram Moolenaar */ 879071d4279SBram Moolenaar 880071d4279SBram Moolenaar static PyObject * 881071d4279SBram Moolenaar BufferNew(buf_T *buf) 882071d4279SBram Moolenaar { 883071d4279SBram Moolenaar /* We need to handle deletion of buffers underneath us. 884e344beadSBram Moolenaar * If we add a "b_python_ref" field to the buf_T structure, 885071d4279SBram Moolenaar * then we can get at it in buf_freeall() in vim. We then 886071d4279SBram Moolenaar * need to create only ONE Python object per buffer - if 887071d4279SBram Moolenaar * we try to create a second, just INCREF the existing one 888071d4279SBram Moolenaar * and return it. The (single) Python object referring to 889e344beadSBram Moolenaar * the buffer is stored in "b_python_ref". 890071d4279SBram Moolenaar * Question: what to do on a buf_freeall(). We'll probably 891071d4279SBram Moolenaar * have to either delete the Python object (DECREF it to 892071d4279SBram Moolenaar * zero - a bad idea, as it leaves dangling refs!) or 893071d4279SBram Moolenaar * set the buf_T * value to an invalid value (-1?), which 894071d4279SBram Moolenaar * means we need checks in all access functions... Bah. 895071d4279SBram Moolenaar */ 896071d4279SBram Moolenaar 897071d4279SBram Moolenaar BufferObject *self; 898071d4279SBram Moolenaar 899e344beadSBram Moolenaar if (buf->b_python_ref != NULL) 900071d4279SBram Moolenaar { 901e344beadSBram Moolenaar self = buf->b_python_ref; 902071d4279SBram Moolenaar Py_INCREF(self); 903071d4279SBram Moolenaar } 904071d4279SBram Moolenaar else 905071d4279SBram Moolenaar { 906071d4279SBram Moolenaar self = PyObject_NEW(BufferObject, &BufferType); 907071d4279SBram Moolenaar if (self == NULL) 908071d4279SBram Moolenaar return NULL; 909071d4279SBram Moolenaar self->buf = buf; 910e344beadSBram Moolenaar buf->b_python_ref = self; 911071d4279SBram Moolenaar } 912071d4279SBram Moolenaar 913071d4279SBram Moolenaar return (PyObject *)(self); 914071d4279SBram Moolenaar } 915071d4279SBram Moolenaar 916071d4279SBram Moolenaar static void 917071d4279SBram Moolenaar BufferDestructor(PyObject *self) 918071d4279SBram Moolenaar { 919071d4279SBram Moolenaar BufferObject *this = (BufferObject *)(self); 920071d4279SBram Moolenaar 921071d4279SBram Moolenaar if (this->buf && this->buf != INVALID_BUFFER_VALUE) 922e344beadSBram Moolenaar this->buf->b_python_ref = NULL; 923071d4279SBram Moolenaar 924658ada69SBram Moolenaar Py_DECREF(self); 925071d4279SBram Moolenaar } 926071d4279SBram Moolenaar 927071d4279SBram Moolenaar static PyObject * 928071d4279SBram Moolenaar BufferGetattr(PyObject *self, char *name) 929071d4279SBram Moolenaar { 930071d4279SBram Moolenaar BufferObject *this = (BufferObject *)(self); 931071d4279SBram Moolenaar 932071d4279SBram Moolenaar if (CheckBuffer(this)) 933071d4279SBram Moolenaar return NULL; 934071d4279SBram Moolenaar 935071d4279SBram Moolenaar if (strcmp(name, "name") == 0) 936071d4279SBram Moolenaar return Py_BuildValue("s", this->buf->b_ffname); 937071d4279SBram Moolenaar else if (strcmp(name, "number") == 0) 938e7cb9cf6SBram Moolenaar return Py_BuildValue(Py_ssize_t_fmt, this->buf->b_fnum); 939071d4279SBram Moolenaar else if (strcmp(name,"__members__") == 0) 940071d4279SBram Moolenaar return Py_BuildValue("[ss]", "name", "number"); 941071d4279SBram Moolenaar else 942071d4279SBram Moolenaar return Py_FindMethod(BufferMethods, self, name); 943071d4279SBram Moolenaar } 944071d4279SBram Moolenaar 945071d4279SBram Moolenaar static PyObject * 946071d4279SBram Moolenaar BufferRepr(PyObject *self) 947071d4279SBram Moolenaar { 948555b280fSBram Moolenaar static char repr[100]; 949071d4279SBram Moolenaar BufferObject *this = (BufferObject *)(self); 950071d4279SBram Moolenaar 951071d4279SBram Moolenaar if (this->buf == INVALID_BUFFER_VALUE) 952071d4279SBram Moolenaar { 953e7cb9cf6SBram Moolenaar vim_snprintf(repr, 100, _("<buffer object (deleted) at %p>"), (self)); 954071d4279SBram Moolenaar return PyString_FromString(repr); 955071d4279SBram Moolenaar } 956071d4279SBram Moolenaar else 957071d4279SBram Moolenaar { 958071d4279SBram Moolenaar char *name = (char *)this->buf->b_fname; 959e7cb9cf6SBram Moolenaar PyInt len; 960071d4279SBram Moolenaar 961071d4279SBram Moolenaar if (name == NULL) 962071d4279SBram Moolenaar name = ""; 963071d4279SBram Moolenaar len = strlen(name); 964071d4279SBram Moolenaar 965071d4279SBram Moolenaar if (len > 35) 966071d4279SBram Moolenaar name = name + (35 - len); 967071d4279SBram Moolenaar 968555b280fSBram Moolenaar vim_snprintf(repr, 100, "<buffer %s%s>", len > 35 ? "..." : "", name); 969071d4279SBram Moolenaar 970071d4279SBram Moolenaar return PyString_FromString(repr); 971071d4279SBram Moolenaar } 972071d4279SBram Moolenaar } 973071d4279SBram Moolenaar 974071d4279SBram Moolenaar /******************/ 975071d4279SBram Moolenaar 9762c45e945SBram Moolenaar static PyInt 977071d4279SBram Moolenaar BufferLength(PyObject *self) 978071d4279SBram Moolenaar { 979071d4279SBram Moolenaar /* HOW DO WE SIGNAL AN ERROR FROM THIS FUNCTION? */ 980071d4279SBram Moolenaar if (CheckBuffer((BufferObject *)(self))) 981071d4279SBram Moolenaar return -1; /* ??? */ 982071d4279SBram Moolenaar 983071d4279SBram Moolenaar return (((BufferObject *)(self))->buf->b_ml.ml_line_count); 984071d4279SBram Moolenaar } 985071d4279SBram Moolenaar 986071d4279SBram Moolenaar static PyObject * 9872c45e945SBram Moolenaar BufferItem(PyObject *self, PyInt n) 988071d4279SBram Moolenaar { 989071d4279SBram Moolenaar return RBItem((BufferObject *)(self), n, 1, 990071d4279SBram Moolenaar (int)((BufferObject *)(self))->buf->b_ml.ml_line_count); 991071d4279SBram Moolenaar } 992071d4279SBram Moolenaar 993071d4279SBram Moolenaar static PyObject * 9942c45e945SBram Moolenaar BufferSlice(PyObject *self, PyInt lo, PyInt hi) 995071d4279SBram Moolenaar { 996071d4279SBram Moolenaar return RBSlice((BufferObject *)(self), lo, hi, 1, 997071d4279SBram Moolenaar (int)((BufferObject *)(self))->buf->b_ml.ml_line_count); 998071d4279SBram Moolenaar } 999071d4279SBram Moolenaar 10002c45e945SBram Moolenaar static PyInt 10012c45e945SBram Moolenaar BufferAssItem(PyObject *self, PyInt n, PyObject *val) 1002071d4279SBram Moolenaar { 1003ca8a4dfeSBram Moolenaar return RBAsItem((BufferObject *)(self), n, val, 1, 1004e7cb9cf6SBram Moolenaar (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, 1005071d4279SBram Moolenaar NULL); 1006071d4279SBram Moolenaar } 1007071d4279SBram Moolenaar 10082c45e945SBram Moolenaar static PyInt 10092c45e945SBram Moolenaar BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val) 1010071d4279SBram Moolenaar { 101119e60943SBram Moolenaar return RBAsSlice((BufferObject *)(self), lo, hi, val, 1, 1012e7cb9cf6SBram Moolenaar (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, 1013071d4279SBram Moolenaar NULL); 1014071d4279SBram Moolenaar } 1015071d4279SBram Moolenaar 1016071d4279SBram Moolenaar static PySequenceMethods RangeAsSeq = { 10172c45e945SBram Moolenaar (PyInquiry) RangeLength, /* sq_length, len(x) */ 1018071d4279SBram Moolenaar (binaryfunc) 0, /* RangeConcat, */ /* sq_concat, x+y */ 10192c45e945SBram Moolenaar (PyIntArgFunc) 0, /* RangeRepeat, */ /* sq_repeat, x*n */ 10202c45e945SBram Moolenaar (PyIntArgFunc) RangeItem, /* sq_item, x[i] */ 10212c45e945SBram Moolenaar (PyIntIntArgFunc) RangeSlice, /* sq_slice, x[i:j] */ 10222c45e945SBram Moolenaar (PyIntObjArgProc) RangeAssItem, /* sq_ass_item, x[i]=v */ 10232c45e945SBram Moolenaar (PyIntIntObjArgProc) RangeAssSlice, /* sq_ass_slice, x[i:j]=v */ 1024071d4279SBram Moolenaar }; 1025071d4279SBram Moolenaar 1026071d4279SBram Moolenaar /* Line range object - Implementation 1027071d4279SBram Moolenaar */ 1028071d4279SBram Moolenaar 1029071d4279SBram Moolenaar static void 1030071d4279SBram Moolenaar RangeDestructor(PyObject *self) 1031071d4279SBram Moolenaar { 1032071d4279SBram Moolenaar Py_DECREF(((RangeObject *)(self))->buf); 1033658ada69SBram Moolenaar Py_DECREF(self); 1034071d4279SBram Moolenaar } 1035071d4279SBram Moolenaar 1036071d4279SBram Moolenaar static PyObject * 1037071d4279SBram Moolenaar RangeGetattr(PyObject *self, char *name) 1038071d4279SBram Moolenaar { 1039071d4279SBram Moolenaar if (strcmp(name, "start") == 0) 1040e7cb9cf6SBram Moolenaar return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1); 1041071d4279SBram Moolenaar else if (strcmp(name, "end") == 0) 1042e7cb9cf6SBram Moolenaar return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1); 1043071d4279SBram Moolenaar else 1044071d4279SBram Moolenaar return Py_FindMethod(RangeMethods, self, name); 1045071d4279SBram Moolenaar } 1046071d4279SBram Moolenaar 1047071d4279SBram Moolenaar /****************/ 1048071d4279SBram Moolenaar 10492c45e945SBram Moolenaar static PyInt 10502c45e945SBram Moolenaar RangeAssItem(PyObject *self, PyInt n, PyObject *val) 1051071d4279SBram Moolenaar { 1052ca8a4dfeSBram Moolenaar return RBAsItem(((RangeObject *)(self))->buf, n, val, 1053071d4279SBram Moolenaar ((RangeObject *)(self))->start, 1054071d4279SBram Moolenaar ((RangeObject *)(self))->end, 1055071d4279SBram Moolenaar &((RangeObject *)(self))->end); 1056071d4279SBram Moolenaar } 1057071d4279SBram Moolenaar 10582c45e945SBram Moolenaar static PyInt 10592c45e945SBram Moolenaar RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val) 1060071d4279SBram Moolenaar { 106119e60943SBram Moolenaar return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val, 1062071d4279SBram Moolenaar ((RangeObject *)(self))->start, 1063071d4279SBram Moolenaar ((RangeObject *)(self))->end, 1064071d4279SBram Moolenaar &((RangeObject *)(self))->end); 1065071d4279SBram Moolenaar } 1066071d4279SBram Moolenaar 1067071d4279SBram Moolenaar /* Buffer list object - Definitions 1068071d4279SBram Moolenaar */ 1069071d4279SBram Moolenaar 1070071d4279SBram Moolenaar typedef struct 1071071d4279SBram Moolenaar { 1072071d4279SBram Moolenaar PyObject_HEAD 1073ca8a4dfeSBram Moolenaar } BufListObject; 1074071d4279SBram Moolenaar 1075071d4279SBram Moolenaar static PySequenceMethods BufListAsSeq = { 10762c45e945SBram Moolenaar (PyInquiry) BufListLength, /* sq_length, len(x) */ 1077071d4279SBram Moolenaar (binaryfunc) 0, /* sq_concat, x+y */ 10782c45e945SBram Moolenaar (PyIntArgFunc) 0, /* sq_repeat, x*n */ 10792c45e945SBram Moolenaar (PyIntArgFunc) BufListItem, /* sq_item, x[i] */ 10802c45e945SBram Moolenaar (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */ 10812c45e945SBram Moolenaar (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */ 10822c45e945SBram Moolenaar (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */ 1083071d4279SBram Moolenaar }; 1084071d4279SBram Moolenaar 1085071d4279SBram Moolenaar static PyTypeObject BufListType = { 1086071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1087071d4279SBram Moolenaar 0, 1088071d4279SBram Moolenaar "buffer list", 1089071d4279SBram Moolenaar sizeof(BufListObject), 1090071d4279SBram Moolenaar 0, 1091071d4279SBram Moolenaar 1092071d4279SBram Moolenaar (destructor) 0, /* tp_dealloc, refcount==0 */ 1093071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1094071d4279SBram Moolenaar (getattrfunc) 0, /* tp_getattr, x.attr */ 1095071d4279SBram Moolenaar (setattrfunc) 0, /* tp_setattr, x.attr=v */ 1096071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1097071d4279SBram Moolenaar (reprfunc) 0, /* tp_repr, `x`, print x */ 1098071d4279SBram Moolenaar 1099071d4279SBram Moolenaar 0, /* as number */ 1100071d4279SBram Moolenaar &BufListAsSeq, /* as sequence */ 1101071d4279SBram Moolenaar 0, /* as mapping */ 1102071d4279SBram Moolenaar 1103071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1104071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1105071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1106071d4279SBram Moolenaar }; 1107071d4279SBram Moolenaar 1108071d4279SBram Moolenaar /* Window object - Definitions 1109071d4279SBram Moolenaar */ 1110071d4279SBram Moolenaar 1111071d4279SBram Moolenaar static struct PyMethodDef WindowMethods[] = { 1112071d4279SBram Moolenaar /* name, function, calling, documentation */ 1113071d4279SBram Moolenaar { NULL, NULL, 0, NULL } 1114071d4279SBram Moolenaar }; 1115071d4279SBram Moolenaar 1116071d4279SBram Moolenaar static PyTypeObject WindowType = { 1117071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1118071d4279SBram Moolenaar 0, 1119071d4279SBram Moolenaar "window", 1120071d4279SBram Moolenaar sizeof(WindowObject), 1121071d4279SBram Moolenaar 0, 1122071d4279SBram Moolenaar 1123071d4279SBram Moolenaar (destructor) WindowDestructor, /* tp_dealloc, refcount==0 */ 1124071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1125071d4279SBram Moolenaar (getattrfunc) WindowGetattr, /* tp_getattr, x.attr */ 1126071d4279SBram Moolenaar (setattrfunc) WindowSetattr, /* tp_setattr, x.attr=v */ 1127071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1128071d4279SBram Moolenaar (reprfunc) WindowRepr, /* tp_repr, `x`, print x */ 1129071d4279SBram Moolenaar 1130071d4279SBram Moolenaar 0, /* as number */ 1131071d4279SBram Moolenaar 0, /* as sequence */ 1132071d4279SBram Moolenaar 0, /* as mapping */ 1133071d4279SBram Moolenaar 1134071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1135071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1136071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1137071d4279SBram Moolenaar }; 1138071d4279SBram Moolenaar 1139071d4279SBram Moolenaar /* Window object - Implementation 1140071d4279SBram Moolenaar */ 1141071d4279SBram Moolenaar 1142071d4279SBram Moolenaar static PyObject * 1143071d4279SBram Moolenaar WindowNew(win_T *win) 1144071d4279SBram Moolenaar { 1145071d4279SBram Moolenaar /* We need to handle deletion of windows underneath us. 1146e344beadSBram Moolenaar * If we add a "w_python_ref" field to the win_T structure, 1147071d4279SBram Moolenaar * then we can get at it in win_free() in vim. We then 1148071d4279SBram Moolenaar * need to create only ONE Python object per window - if 1149071d4279SBram Moolenaar * we try to create a second, just INCREF the existing one 1150071d4279SBram Moolenaar * and return it. The (single) Python object referring to 1151e344beadSBram Moolenaar * the window is stored in "w_python_ref". 1152071d4279SBram Moolenaar * On a win_free() we set the Python object's win_T* field 1153071d4279SBram Moolenaar * to an invalid value. We trap all uses of a window 1154071d4279SBram Moolenaar * object, and reject them if the win_T* field is invalid. 1155071d4279SBram Moolenaar */ 1156071d4279SBram Moolenaar 1157071d4279SBram Moolenaar WindowObject *self; 1158071d4279SBram Moolenaar 1159e344beadSBram Moolenaar if (win->w_python_ref) 1160071d4279SBram Moolenaar { 1161e344beadSBram Moolenaar self = win->w_python_ref; 1162071d4279SBram Moolenaar Py_INCREF(self); 1163071d4279SBram Moolenaar } 1164071d4279SBram Moolenaar else 1165071d4279SBram Moolenaar { 1166071d4279SBram Moolenaar self = PyObject_NEW(WindowObject, &WindowType); 1167071d4279SBram Moolenaar if (self == NULL) 1168071d4279SBram Moolenaar return NULL; 1169071d4279SBram Moolenaar self->win = win; 1170e344beadSBram Moolenaar win->w_python_ref = self; 1171071d4279SBram Moolenaar } 1172071d4279SBram Moolenaar 1173071d4279SBram Moolenaar return (PyObject *)(self); 1174071d4279SBram Moolenaar } 1175071d4279SBram Moolenaar 1176071d4279SBram Moolenaar static void 1177071d4279SBram Moolenaar WindowDestructor(PyObject *self) 1178071d4279SBram Moolenaar { 1179071d4279SBram Moolenaar WindowObject *this = (WindowObject *)(self); 1180071d4279SBram Moolenaar 1181071d4279SBram Moolenaar if (this->win && this->win != INVALID_WINDOW_VALUE) 1182e344beadSBram Moolenaar this->win->w_python_ref = NULL; 1183071d4279SBram Moolenaar 1184658ada69SBram Moolenaar Py_DECREF(self); 1185071d4279SBram Moolenaar } 1186071d4279SBram Moolenaar 1187071d4279SBram Moolenaar static PyObject * 1188071d4279SBram Moolenaar WindowGetattr(PyObject *self, char *name) 1189071d4279SBram Moolenaar { 1190071d4279SBram Moolenaar WindowObject *this = (WindowObject *)(self); 1191071d4279SBram Moolenaar 1192071d4279SBram Moolenaar if (CheckWindow(this)) 1193071d4279SBram Moolenaar return NULL; 1194071d4279SBram Moolenaar 1195071d4279SBram Moolenaar if (strcmp(name, "buffer") == 0) 1196071d4279SBram Moolenaar return (PyObject *)BufferNew(this->win->w_buffer); 1197071d4279SBram Moolenaar else if (strcmp(name, "cursor") == 0) 1198071d4279SBram Moolenaar { 1199071d4279SBram Moolenaar pos_T *pos = &this->win->w_cursor; 1200071d4279SBram Moolenaar 1201071d4279SBram Moolenaar return Py_BuildValue("(ll)", (long)(pos->lnum), (long)(pos->col)); 1202071d4279SBram Moolenaar } 1203071d4279SBram Moolenaar else if (strcmp(name, "height") == 0) 1204071d4279SBram Moolenaar return Py_BuildValue("l", (long)(this->win->w_height)); 1205071d4279SBram Moolenaar #ifdef FEAT_VERTSPLIT 1206071d4279SBram Moolenaar else if (strcmp(name, "width") == 0) 1207071d4279SBram Moolenaar return Py_BuildValue("l", (long)(W_WIDTH(this->win))); 1208071d4279SBram Moolenaar #endif 1209071d4279SBram Moolenaar else if (strcmp(name,"__members__") == 0) 1210071d4279SBram Moolenaar return Py_BuildValue("[sss]", "buffer", "cursor", "height"); 1211071d4279SBram Moolenaar else 1212071d4279SBram Moolenaar return Py_FindMethod(WindowMethods, self, name); 1213071d4279SBram Moolenaar } 1214071d4279SBram Moolenaar 1215071d4279SBram Moolenaar /* Window list object - Definitions 1216071d4279SBram Moolenaar */ 1217071d4279SBram Moolenaar 1218071d4279SBram Moolenaar typedef struct 1219071d4279SBram Moolenaar { 1220071d4279SBram Moolenaar PyObject_HEAD 1221071d4279SBram Moolenaar } 1222071d4279SBram Moolenaar WinListObject; 1223071d4279SBram Moolenaar 1224071d4279SBram Moolenaar static PySequenceMethods WinListAsSeq = { 12252c45e945SBram Moolenaar (PyInquiry) WinListLength, /* sq_length, len(x) */ 1226071d4279SBram Moolenaar (binaryfunc) 0, /* sq_concat, x+y */ 12272c45e945SBram Moolenaar (PyIntArgFunc) 0, /* sq_repeat, x*n */ 12282c45e945SBram Moolenaar (PyIntArgFunc) WinListItem, /* sq_item, x[i] */ 12292c45e945SBram Moolenaar (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */ 12302c45e945SBram Moolenaar (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */ 12312c45e945SBram Moolenaar (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */ 1232071d4279SBram Moolenaar }; 1233071d4279SBram Moolenaar 1234071d4279SBram Moolenaar static PyTypeObject WinListType = { 1235071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1236071d4279SBram Moolenaar 0, 1237071d4279SBram Moolenaar "window list", 1238071d4279SBram Moolenaar sizeof(WinListObject), 1239071d4279SBram Moolenaar 0, 1240071d4279SBram Moolenaar 1241071d4279SBram Moolenaar (destructor) 0, /* tp_dealloc, refcount==0 */ 1242071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1243071d4279SBram Moolenaar (getattrfunc) 0, /* tp_getattr, x.attr */ 1244071d4279SBram Moolenaar (setattrfunc) 0, /* tp_setattr, x.attr=v */ 1245071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1246071d4279SBram Moolenaar (reprfunc) 0, /* tp_repr, `x`, print x */ 1247071d4279SBram Moolenaar 1248071d4279SBram Moolenaar 0, /* as number */ 1249071d4279SBram Moolenaar &WinListAsSeq, /* as sequence */ 1250071d4279SBram Moolenaar 0, /* as mapping */ 1251071d4279SBram Moolenaar 1252071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1253071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1254071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1255071d4279SBram Moolenaar }; 1256071d4279SBram Moolenaar 1257071d4279SBram Moolenaar /* Current items object - Definitions 1258071d4279SBram Moolenaar */ 1259071d4279SBram Moolenaar 1260071d4279SBram Moolenaar typedef struct 1261071d4279SBram Moolenaar { 1262071d4279SBram Moolenaar PyObject_HEAD 1263ca8a4dfeSBram Moolenaar } CurrentObject; 1264071d4279SBram Moolenaar 1265071d4279SBram Moolenaar static PyTypeObject CurrentType = { 1266071d4279SBram Moolenaar PyObject_HEAD_INIT(0) 1267071d4279SBram Moolenaar 0, 1268071d4279SBram Moolenaar "current data", 1269071d4279SBram Moolenaar sizeof(CurrentObject), 1270071d4279SBram Moolenaar 0, 1271071d4279SBram Moolenaar 1272071d4279SBram Moolenaar (destructor) 0, /* tp_dealloc, refcount==0 */ 1273071d4279SBram Moolenaar (printfunc) 0, /* tp_print, print x */ 1274071d4279SBram Moolenaar (getattrfunc) CurrentGetattr, /* tp_getattr, x.attr */ 1275071d4279SBram Moolenaar (setattrfunc) CurrentSetattr, /* tp_setattr, x.attr=v */ 1276071d4279SBram Moolenaar (cmpfunc) 0, /* tp_compare, x>y */ 1277071d4279SBram Moolenaar (reprfunc) 0, /* tp_repr, `x`, print x */ 1278071d4279SBram Moolenaar 1279071d4279SBram Moolenaar 0, /* as number */ 1280071d4279SBram Moolenaar 0, /* as sequence */ 1281071d4279SBram Moolenaar 0, /* as mapping */ 1282071d4279SBram Moolenaar 1283071d4279SBram Moolenaar (hashfunc) 0, /* tp_hash, dict(x) */ 1284071d4279SBram Moolenaar (ternaryfunc) 0, /* tp_call, x() */ 1285071d4279SBram Moolenaar (reprfunc) 0, /* tp_str, str(x) */ 1286071d4279SBram Moolenaar }; 1287071d4279SBram Moolenaar 1288071d4279SBram Moolenaar /* Current items object - Implementation 1289071d4279SBram Moolenaar */ 1290071d4279SBram Moolenaar static PyObject * 12914bdbbf70SBram Moolenaar CurrentGetattr(PyObject *self UNUSED, char *name) 1292071d4279SBram Moolenaar { 1293071d4279SBram Moolenaar if (strcmp(name, "buffer") == 0) 1294071d4279SBram Moolenaar return (PyObject *)BufferNew(curbuf); 1295071d4279SBram Moolenaar else if (strcmp(name, "window") == 0) 1296071d4279SBram Moolenaar return (PyObject *)WindowNew(curwin); 1297071d4279SBram Moolenaar else if (strcmp(name, "line") == 0) 1298e7cb9cf6SBram Moolenaar return GetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum); 1299071d4279SBram Moolenaar else if (strcmp(name, "range") == 0) 1300071d4279SBram Moolenaar return RangeNew(curbuf, RangeStart, RangeEnd); 1301071d4279SBram Moolenaar else if (strcmp(name,"__members__") == 0) 1302071d4279SBram Moolenaar return Py_BuildValue("[ssss]", "buffer", "window", "line", "range"); 1303071d4279SBram Moolenaar else 1304071d4279SBram Moolenaar { 1305071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, name); 1306071d4279SBram Moolenaar return NULL; 1307071d4279SBram Moolenaar } 1308071d4279SBram Moolenaar } 1309071d4279SBram Moolenaar 1310071d4279SBram Moolenaar static int 13114bdbbf70SBram Moolenaar CurrentSetattr(PyObject *self UNUSED, char *name, PyObject *value) 1312071d4279SBram Moolenaar { 1313071d4279SBram Moolenaar if (strcmp(name, "line") == 0) 1314071d4279SBram Moolenaar { 1315e7cb9cf6SBram Moolenaar if (SetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum, value, NULL) == FAIL) 1316071d4279SBram Moolenaar return -1; 1317071d4279SBram Moolenaar 1318071d4279SBram Moolenaar return 0; 1319071d4279SBram Moolenaar } 1320071d4279SBram Moolenaar else 1321071d4279SBram Moolenaar { 1322071d4279SBram Moolenaar PyErr_SetString(PyExc_AttributeError, name); 1323071d4279SBram Moolenaar return -1; 1324071d4279SBram Moolenaar } 1325071d4279SBram Moolenaar } 1326071d4279SBram Moolenaar 1327071d4279SBram Moolenaar /* External interface 1328071d4279SBram Moolenaar */ 1329071d4279SBram Moolenaar 1330071d4279SBram Moolenaar void 1331071d4279SBram Moolenaar python_buffer_free(buf_T *buf) 1332071d4279SBram Moolenaar { 1333e344beadSBram Moolenaar if (buf->b_python_ref != NULL) 1334071d4279SBram Moolenaar { 1335e344beadSBram Moolenaar BufferObject *bp = buf->b_python_ref; 1336071d4279SBram Moolenaar bp->buf = INVALID_BUFFER_VALUE; 1337e344beadSBram Moolenaar buf->b_python_ref = NULL; 1338071d4279SBram Moolenaar } 1339071d4279SBram Moolenaar } 1340071d4279SBram Moolenaar 1341071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO) 1342071d4279SBram Moolenaar void 1343071d4279SBram Moolenaar python_window_free(win_T *win) 1344071d4279SBram Moolenaar { 1345e344beadSBram Moolenaar if (win->w_python_ref != NULL) 1346071d4279SBram Moolenaar { 1347e344beadSBram Moolenaar WindowObject *wp = win->w_python_ref; 1348071d4279SBram Moolenaar wp->win = INVALID_WINDOW_VALUE; 1349e344beadSBram Moolenaar win->w_python_ref = NULL; 1350071d4279SBram Moolenaar } 1351071d4279SBram Moolenaar } 1352071d4279SBram Moolenaar #endif 1353071d4279SBram Moolenaar 1354071d4279SBram Moolenaar static BufListObject TheBufferList = 1355071d4279SBram Moolenaar { 1356071d4279SBram Moolenaar PyObject_HEAD_INIT(&BufListType) 1357071d4279SBram Moolenaar }; 1358071d4279SBram Moolenaar 1359071d4279SBram Moolenaar static WinListObject TheWindowList = 1360071d4279SBram Moolenaar { 1361071d4279SBram Moolenaar PyObject_HEAD_INIT(&WinListType) 1362071d4279SBram Moolenaar }; 1363071d4279SBram Moolenaar 1364071d4279SBram Moolenaar static CurrentObject TheCurrent = 1365071d4279SBram Moolenaar { 1366071d4279SBram Moolenaar PyObject_HEAD_INIT(&CurrentType) 1367071d4279SBram Moolenaar }; 1368071d4279SBram Moolenaar 1369071d4279SBram Moolenaar static int 1370071d4279SBram Moolenaar PythonMod_Init(void) 1371071d4279SBram Moolenaar { 1372071d4279SBram Moolenaar PyObject *mod; 1373071d4279SBram Moolenaar PyObject *dict; 13749774ecc8SBram Moolenaar /* The special value is removed from sys.path in Python_Init(). */ 13759774ecc8SBram Moolenaar static char *(argv[2]) = {"/must>not&exist/foo", NULL}; 1376071d4279SBram Moolenaar 1377071d4279SBram Moolenaar /* Fixups... */ 137821377c8dSBram Moolenaar PyType_Ready(&BufferType); 137921377c8dSBram Moolenaar PyType_Ready(&RangeType); 138021377c8dSBram Moolenaar PyType_Ready(&WindowType); 138121377c8dSBram Moolenaar PyType_Ready(&BufListType); 138221377c8dSBram Moolenaar PyType_Ready(&WinListType); 138321377c8dSBram Moolenaar PyType_Ready(&CurrentType); 1384071d4279SBram Moolenaar 1385071d4279SBram Moolenaar /* Set sys.argv[] to avoid a crash in warn(). */ 1386071d4279SBram Moolenaar PySys_SetArgv(1, argv); 1387071d4279SBram Moolenaar 1388e7cb9cf6SBram Moolenaar mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION); 1389071d4279SBram Moolenaar dict = PyModule_GetDict(mod); 1390071d4279SBram Moolenaar 1391071d4279SBram Moolenaar VimError = Py_BuildValue("s", "vim.error"); 1392071d4279SBram Moolenaar 1393071d4279SBram Moolenaar PyDict_SetItemString(dict, "error", VimError); 13947df2d662SBram Moolenaar PyDict_SetItemString(dict, "buffers", (PyObject *)(void *)&TheBufferList); 13957df2d662SBram Moolenaar PyDict_SetItemString(dict, "current", (PyObject *)(void *)&TheCurrent); 13967df2d662SBram Moolenaar PyDict_SetItemString(dict, "windows", (PyObject *)(void *)&TheWindowList); 1397071d4279SBram Moolenaar 1398071d4279SBram Moolenaar if (PyErr_Occurred()) 1399071d4279SBram Moolenaar return -1; 1400071d4279SBram Moolenaar 1401071d4279SBram Moolenaar return 0; 1402071d4279SBram Moolenaar } 1403071d4279SBram Moolenaar 1404071d4279SBram Moolenaar /************************************************************************* 1405071d4279SBram Moolenaar * 4. Utility functions for handling the interface between Vim and Python. 1406071d4279SBram Moolenaar */ 1407071d4279SBram Moolenaar 1408071d4279SBram Moolenaar /* Convert a Vim line into a Python string. 1409071d4279SBram Moolenaar * All internal newlines are replaced by null characters. 1410071d4279SBram Moolenaar * 1411071d4279SBram Moolenaar * On errors, the Python exception data is set, and NULL is returned. 1412071d4279SBram Moolenaar */ 1413071d4279SBram Moolenaar static PyObject * 1414071d4279SBram Moolenaar LineToString(const char *str) 1415071d4279SBram Moolenaar { 1416071d4279SBram Moolenaar PyObject *result; 14172c45e945SBram Moolenaar PyInt len = strlen(str); 1418071d4279SBram Moolenaar char *p; 1419071d4279SBram Moolenaar 1420071d4279SBram Moolenaar /* Allocate an Python string object, with uninitialised contents. We 1421071d4279SBram Moolenaar * must do it this way, so that we can modify the string in place 1422071d4279SBram Moolenaar * later. See the Python source, Objects/stringobject.c for details. 1423071d4279SBram Moolenaar */ 1424071d4279SBram Moolenaar result = PyString_FromStringAndSize(NULL, len); 1425071d4279SBram Moolenaar if (result == NULL) 1426071d4279SBram Moolenaar return NULL; 1427071d4279SBram Moolenaar 1428071d4279SBram Moolenaar p = PyString_AsString(result); 1429071d4279SBram Moolenaar 1430071d4279SBram Moolenaar while (*str) 1431071d4279SBram Moolenaar { 1432071d4279SBram Moolenaar if (*str == '\n') 1433071d4279SBram Moolenaar *p = '\0'; 1434071d4279SBram Moolenaar else 1435071d4279SBram Moolenaar *p = *str; 1436071d4279SBram Moolenaar 1437071d4279SBram Moolenaar ++p; 1438071d4279SBram Moolenaar ++str; 1439071d4279SBram Moolenaar } 1440071d4279SBram Moolenaar 1441071d4279SBram Moolenaar return result; 1442071d4279SBram Moolenaar } 1443071d4279SBram Moolenaar 1444071d4279SBram Moolenaar 1445071d4279SBram Moolenaar /* Don't generate a prototype for the next function, it generates an error on 1446071d4279SBram Moolenaar * newer Python versions. */ 1447071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO) 1448071d4279SBram Moolenaar 1449071d4279SBram Moolenaar char * 1450071d4279SBram Moolenaar Py_GetProgramName(void) 1451071d4279SBram Moolenaar { 1452071d4279SBram Moolenaar return "vim"; 1453071d4279SBram Moolenaar } 1454071d4279SBram Moolenaar #endif /* Python 1.4 */ 1455170bf1aeSBram Moolenaar 1456170bf1aeSBram Moolenaar static void 1457170bf1aeSBram Moolenaar init_structs(void) 1458170bf1aeSBram Moolenaar { 1459170bf1aeSBram Moolenaar vim_memset(&OutputType, 0, sizeof(OutputType)); 1460170bf1aeSBram Moolenaar OutputType.tp_name = "message"; 1461170bf1aeSBram Moolenaar OutputType.tp_basicsize = sizeof(OutputObject); 1462170bf1aeSBram Moolenaar OutputType.tp_getattr = OutputGetattr; 1463170bf1aeSBram Moolenaar OutputType.tp_setattr = OutputSetattr; 1464ca8a4dfeSBram Moolenaar 1465ca8a4dfeSBram Moolenaar vim_memset(&RangeType, 0, sizeof(RangeType)); 1466ca8a4dfeSBram Moolenaar RangeType.tp_name = "range"; 1467ca8a4dfeSBram Moolenaar RangeType.tp_basicsize = sizeof(RangeObject); 1468ca8a4dfeSBram Moolenaar RangeType.tp_dealloc = RangeDestructor; 1469ca8a4dfeSBram Moolenaar RangeType.tp_getattr = RangeGetattr; 1470ca8a4dfeSBram Moolenaar RangeType.tp_repr = RangeRepr; 1471ca8a4dfeSBram Moolenaar RangeType.tp_as_sequence = &RangeAsSeq; 1472170bf1aeSBram Moolenaar } 1473