Mercurial > vim
annotate src/if_python.c @ 2384:aeea25941392 vim73
Temporary solution for crashing when using both :py and :py3: disallow both in
one session.
author | Bram Moolenaar <bram@vim.org> |
---|---|
date | Sat, 24 Jul 2010 15:42:14 +0200 |
parents | f414e5a4b40e |
children | 76f0c4918f5c |
rev | line source |
---|---|
7 | 1 /* vi:set ts=8 sts=4 sw=4: |
2 * | |
3 * VIM - Vi IMproved by Bram Moolenaar | |
4 * | |
5 * Do ":help uganda" in Vim to read copying and usage conditions. | |
6 * Do ":help credits" in Vim to see a list of people who contributed. | |
7 * See README.txt for an overview of the Vim source code. | |
8 */ | |
9 /* | |
10 * Python extensions by Paul Moore. | |
11 * Changes for Unix by David Leonard. | |
12 * | |
13 * This consists of four parts: | |
14 * 1. Python interpreter main program | |
15 * 2. Python output stream: writes output via [e]msg(). | |
16 * 3. Implementation of the Vim module for Python | |
17 * 4. Utility functions for handling the interface between Vim and Python. | |
18 */ | |
19 | |
20 #include "vim.h" | |
21 | |
22 #include <limits.h> | |
23 | |
24 /* Python.h defines _POSIX_THREADS itself (if needed) */ | |
25 #ifdef _POSIX_THREADS | |
26 # undef _POSIX_THREADS | |
27 #endif | |
28 | |
2250
1bac28a53fae
Add the conceal patch from Vince Negri.
Bram Moolenaar <bram@vim.org>
parents:
2100
diff
changeset
|
29 #if defined(_WIN32) && defined(HAVE_FCNTL_H) |
7 | 30 # undef HAVE_FCNTL_H |
31 #endif | |
32 | |
33 #ifdef _DEBUG | |
34 # undef _DEBUG | |
35 #endif | |
36 | |
37 #ifdef HAVE_STDARG_H | |
38 # undef HAVE_STDARG_H /* Python's config.h defines it as well. */ | |
39 #endif | |
1991 | 40 #ifdef _POSIX_C_SOURCE |
41 # undef _POSIX_C_SOURCE /* pyconfig.h defines it as well. */ | |
42 #endif | |
43 #ifdef _XOPEN_SOURCE | |
44 # undef _XOPEN_SOURCE /* pyconfig.h defines it as well. */ | |
45 #endif | |
7 | 46 |
1594 | 47 #define PY_SSIZE_T_CLEAN |
48 | |
7 | 49 #include <Python.h> |
50 #if defined(MACOS) && !defined(MACOS_X_UNIX) | |
51 # include "macglue.h" | |
52 # include <CodeFragments.h> | |
53 #endif | |
54 #undef main /* Defined in python.h - aargh */ | |
55 #undef HAVE_FCNTL_H /* Clash with os_win32.h */ | |
56 | |
57 #if !defined(FEAT_PYTHON) && defined(PROTO) | |
58 /* Use this to be able to generate prototypes without python being used. */ | |
1607 | 59 # define PyObject Py_ssize_t |
60 # define PyThreadState Py_ssize_t | |
61 # define PyTypeObject Py_ssize_t | |
62 struct PyMethodDef { Py_ssize_t a; }; | |
63 # define PySequenceMethods Py_ssize_t | |
7 | 64 #endif |
65 | |
1594 | 66 #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000 |
67 # define PyInt Py_ssize_t | |
68 # define PyInquiry lenfunc | |
69 # define PyIntArgFunc ssizeargfunc | |
70 # define PyIntIntArgFunc ssizessizeargfunc | |
71 # define PyIntObjArgProc ssizeobjargproc | |
72 # define PyIntIntObjArgProc ssizessizeobjargproc | |
1607 | 73 # define Py_ssize_t_fmt "n" |
1594 | 74 #else |
75 # define PyInt int | |
76 # define PyInquiry inquiry | |
77 # define PyIntArgFunc intargfunc | |
78 # define PyIntIntArgFunc intintargfunc | |
79 # define PyIntObjArgProc intobjargproc | |
80 # define PyIntIntObjArgProc intintobjargproc | |
1607 | 81 # define Py_ssize_t_fmt "i" |
1594 | 82 #endif |
83 | |
7 | 84 /* Parser flags */ |
85 #define single_input 256 | |
86 #define file_input 257 | |
87 #define eval_input 258 | |
88 | |
89 #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x020300F0 | |
90 /* Python 2.3: can invoke ":python" recursively. */ | |
91 # define PY_CAN_RECURSE | |
92 #endif | |
93 | |
94 #if defined(DYNAMIC_PYTHON) || defined(PROTO) | |
95 # ifndef DYNAMIC_PYTHON | |
1607 | 96 # define HINSTANCE long_u /* for generating prototypes */ |
7 | 97 # endif |
98 | |
2374
f414e5a4b40e
Fix: "import termios" doesn't work with dynamically loaded Python. (James
Bram Moolenaar <bram@vim.org>
parents:
2329
diff
changeset
|
99 #ifndef WIN3264 |
2329
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
100 # include <dlfcn.h> |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
101 # define FARPROC void* |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
102 # define HINSTANCE void* |
2374
f414e5a4b40e
Fix: "import termios" doesn't work with dynamically loaded Python. (James
Bram Moolenaar <bram@vim.org>
parents:
2329
diff
changeset
|
103 # define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL) |
2329
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
104 # define close_dll dlclose |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
105 # define symbol_from_dll dlsym |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
106 #else |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
107 # define load_dll LoadLibrary |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
108 # define close_dll FreeLibrary |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
109 # define symbol_from_dll GetProcAddress |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
110 #endif |
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
111 |
1607 | 112 /* This makes if_python.c compile without warnings against Python 2.5 |
113 * on Win32 and Win64. */ | |
114 #undef PyRun_SimpleString | |
115 #undef PyArg_Parse | |
116 #undef PyArg_ParseTuple | |
117 #undef Py_BuildValue | |
118 #undef Py_InitModule4 | |
119 #undef Py_InitModule4_64 | |
120 | |
7 | 121 /* |
122 * Wrapper defines | |
123 */ | |
124 # define PyArg_Parse dll_PyArg_Parse | |
125 # define PyArg_ParseTuple dll_PyArg_ParseTuple | |
126 # define PyDict_SetItemString dll_PyDict_SetItemString | |
127 # define PyErr_BadArgument dll_PyErr_BadArgument | |
128 # define PyErr_Clear dll_PyErr_Clear | |
129 # define PyErr_NoMemory dll_PyErr_NoMemory | |
130 # define PyErr_Occurred dll_PyErr_Occurred | |
131 # define PyErr_SetNone dll_PyErr_SetNone | |
132 # define PyErr_SetString dll_PyErr_SetString | |
133 # define PyEval_InitThreads dll_PyEval_InitThreads | |
134 # define PyEval_RestoreThread dll_PyEval_RestoreThread | |
135 # define PyEval_SaveThread dll_PyEval_SaveThread | |
136 # ifdef PY_CAN_RECURSE | |
137 # define PyGILState_Ensure dll_PyGILState_Ensure | |
138 # define PyGILState_Release dll_PyGILState_Release | |
139 # endif | |
140 # define PyInt_AsLong dll_PyInt_AsLong | |
141 # define PyInt_FromLong dll_PyInt_FromLong | |
142 # define PyInt_Type (*dll_PyInt_Type) | |
143 # define PyList_GetItem dll_PyList_GetItem | |
637 | 144 # define PyList_Append dll_PyList_Append |
7 | 145 # define PyList_New dll_PyList_New |
146 # define PyList_SetItem dll_PyList_SetItem | |
147 # define PyList_Size dll_PyList_Size | |
148 # define PyList_Type (*dll_PyList_Type) | |
149 # define PyImport_ImportModule dll_PyImport_ImportModule | |
637 | 150 # define PyDict_New dll_PyDict_New |
7 | 151 # define PyDict_GetItemString dll_PyDict_GetItemString |
152 # define PyModule_GetDict dll_PyModule_GetDict | |
153 # define PyRun_SimpleString dll_PyRun_SimpleString | |
154 # define PyString_AsString dll_PyString_AsString | |
155 # define PyString_FromString dll_PyString_FromString | |
156 # define PyString_FromStringAndSize dll_PyString_FromStringAndSize | |
157 # define PyString_Size dll_PyString_Size | |
158 # define PyString_Type (*dll_PyString_Type) | |
159 # define PySys_SetObject dll_PySys_SetObject | |
160 # define PySys_SetArgv dll_PySys_SetArgv | |
161 # define PyType_Type (*dll_PyType_Type) | |
162 # define Py_BuildValue dll_Py_BuildValue | |
163 # define Py_FindMethod dll_Py_FindMethod | |
164 # define Py_InitModule4 dll_Py_InitModule4 | |
165 # define Py_Initialize dll_Py_Initialize | |
242 | 166 # define Py_Finalize dll_Py_Finalize |
167 # define Py_IsInitialized dll_Py_IsInitialized | |
7 | 168 # define _PyObject_New dll__PyObject_New |
169 # define _Py_NoneStruct (*dll__Py_NoneStruct) | |
170 # define PyObject_Init dll__PyObject_Init | |
171 # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000 | |
172 # define PyType_IsSubtype dll_PyType_IsSubtype | |
173 # endif | |
174 # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000 | |
175 # define PyObject_Malloc dll_PyObject_Malloc | |
176 # define PyObject_Free dll_PyObject_Free | |
177 # endif | |
178 | |
179 /* | |
180 * Pointers for dynamic link | |
181 */ | |
182 static int(*dll_PyArg_Parse)(PyObject *, char *, ...); | |
183 static int(*dll_PyArg_ParseTuple)(PyObject *, char *, ...); | |
184 static int(*dll_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item); | |
185 static int(*dll_PyErr_BadArgument)(void); | |
186 static void(*dll_PyErr_Clear)(void); | |
187 static PyObject*(*dll_PyErr_NoMemory)(void); | |
188 static PyObject*(*dll_PyErr_Occurred)(void); | |
189 static void(*dll_PyErr_SetNone)(PyObject *); | |
190 static void(*dll_PyErr_SetString)(PyObject *, const char *); | |
191 static void(*dll_PyEval_InitThreads)(void); | |
192 static void(*dll_PyEval_RestoreThread)(PyThreadState *); | |
193 static PyThreadState*(*dll_PyEval_SaveThread)(void); | |
194 # ifdef PY_CAN_RECURSE | |
195 static PyGILState_STATE (*dll_PyGILState_Ensure)(void); | |
196 static void (*dll_PyGILState_Release)(PyGILState_STATE); | |
197 #endif | |
198 static long(*dll_PyInt_AsLong)(PyObject *); | |
199 static PyObject*(*dll_PyInt_FromLong)(long); | |
200 static PyTypeObject* dll_PyInt_Type; | |
1594 | 201 static PyObject*(*dll_PyList_GetItem)(PyObject *, PyInt); |
637 | 202 static PyObject*(*dll_PyList_Append)(PyObject *, PyObject *); |
1594 | 203 static PyObject*(*dll_PyList_New)(PyInt size); |
204 static int(*dll_PyList_SetItem)(PyObject *, PyInt, PyObject *); | |
205 static PyInt(*dll_PyList_Size)(PyObject *); | |
7 | 206 static PyTypeObject* dll_PyList_Type; |
207 static PyObject*(*dll_PyImport_ImportModule)(const char *); | |
637 | 208 static PyObject*(*dll_PyDict_New)(void); |
7 | 209 static PyObject*(*dll_PyDict_GetItemString)(PyObject *, const char *); |
210 static PyObject*(*dll_PyModule_GetDict)(PyObject *); | |
211 static int(*dll_PyRun_SimpleString)(char *); | |
212 static char*(*dll_PyString_AsString)(PyObject *); | |
213 static PyObject*(*dll_PyString_FromString)(const char *); | |
1594 | 214 static PyObject*(*dll_PyString_FromStringAndSize)(const char *, PyInt); |
215 static PyInt(*dll_PyString_Size)(PyObject *); | |
7 | 216 static PyTypeObject* dll_PyString_Type; |
217 static int(*dll_PySys_SetObject)(char *, PyObject *); | |
218 static int(*dll_PySys_SetArgv)(int, char **); | |
219 static PyTypeObject* dll_PyType_Type; | |
220 static PyObject*(*dll_Py_BuildValue)(char *, ...); | |
221 static PyObject*(*dll_Py_FindMethod)(struct PyMethodDef[], PyObject *, char *); | |
222 static PyObject*(*dll_Py_InitModule4)(char *, struct PyMethodDef *, char *, PyObject *, int); | |
223 static void(*dll_Py_Initialize)(void); | |
242 | 224 static void(*dll_Py_Finalize)(void); |
225 static int(*dll_Py_IsInitialized)(void); | |
7 | 226 static PyObject*(*dll__PyObject_New)(PyTypeObject *, PyObject *); |
227 static PyObject*(*dll__PyObject_Init)(PyObject *, PyTypeObject *); | |
228 static PyObject* dll__Py_NoneStruct; | |
229 # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000 | |
230 static int (*dll_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *); | |
231 # endif | |
232 # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000 | |
233 static void* (*dll_PyObject_Malloc)(size_t); | |
234 static void (*dll_PyObject_Free)(void*); | |
235 # endif | |
236 | |
237 static HINSTANCE hinstPython = 0; /* Instance of python.dll */ | |
238 | |
239 /* Imported exception objects */ | |
240 static PyObject *imp_PyExc_AttributeError; | |
241 static PyObject *imp_PyExc_IndexError; | |
242 static PyObject *imp_PyExc_KeyboardInterrupt; | |
243 static PyObject *imp_PyExc_TypeError; | |
244 static PyObject *imp_PyExc_ValueError; | |
245 | |
246 # define PyExc_AttributeError imp_PyExc_AttributeError | |
247 # define PyExc_IndexError imp_PyExc_IndexError | |
248 # define PyExc_KeyboardInterrupt imp_PyExc_KeyboardInterrupt | |
249 # define PyExc_TypeError imp_PyExc_TypeError | |
250 # define PyExc_ValueError imp_PyExc_ValueError | |
251 | |
252 /* | |
253 * Table of name to function pointer of python. | |
254 */ | |
255 # define PYTHON_PROC FARPROC | |
256 static struct | |
257 { | |
258 char *name; | |
259 PYTHON_PROC *ptr; | |
260 } python_funcname_table[] = | |
261 { | |
262 {"PyArg_Parse", (PYTHON_PROC*)&dll_PyArg_Parse}, | |
263 {"PyArg_ParseTuple", (PYTHON_PROC*)&dll_PyArg_ParseTuple}, | |
264 {"PyDict_SetItemString", (PYTHON_PROC*)&dll_PyDict_SetItemString}, | |
265 {"PyErr_BadArgument", (PYTHON_PROC*)&dll_PyErr_BadArgument}, | |
266 {"PyErr_Clear", (PYTHON_PROC*)&dll_PyErr_Clear}, | |
267 {"PyErr_NoMemory", (PYTHON_PROC*)&dll_PyErr_NoMemory}, | |
268 {"PyErr_Occurred", (PYTHON_PROC*)&dll_PyErr_Occurred}, | |
269 {"PyErr_SetNone", (PYTHON_PROC*)&dll_PyErr_SetNone}, | |
270 {"PyErr_SetString", (PYTHON_PROC*)&dll_PyErr_SetString}, | |
271 {"PyEval_InitThreads", (PYTHON_PROC*)&dll_PyEval_InitThreads}, | |
272 {"PyEval_RestoreThread", (PYTHON_PROC*)&dll_PyEval_RestoreThread}, | |
273 {"PyEval_SaveThread", (PYTHON_PROC*)&dll_PyEval_SaveThread}, | |
274 # ifdef PY_CAN_RECURSE | |
275 {"PyGILState_Ensure", (PYTHON_PROC*)&dll_PyGILState_Ensure}, | |
276 {"PyGILState_Release", (PYTHON_PROC*)&dll_PyGILState_Release}, | |
277 # endif | |
278 {"PyInt_AsLong", (PYTHON_PROC*)&dll_PyInt_AsLong}, | |
279 {"PyInt_FromLong", (PYTHON_PROC*)&dll_PyInt_FromLong}, | |
280 {"PyInt_Type", (PYTHON_PROC*)&dll_PyInt_Type}, | |
281 {"PyList_GetItem", (PYTHON_PROC*)&dll_PyList_GetItem}, | |
637 | 282 {"PyList_Append", (PYTHON_PROC*)&dll_PyList_Append}, |
7 | 283 {"PyList_New", (PYTHON_PROC*)&dll_PyList_New}, |
284 {"PyList_SetItem", (PYTHON_PROC*)&dll_PyList_SetItem}, | |
285 {"PyList_Size", (PYTHON_PROC*)&dll_PyList_Size}, | |
286 {"PyList_Type", (PYTHON_PROC*)&dll_PyList_Type}, | |
287 {"PyImport_ImportModule", (PYTHON_PROC*)&dll_PyImport_ImportModule}, | |
288 {"PyDict_GetItemString", (PYTHON_PROC*)&dll_PyDict_GetItemString}, | |
637 | 289 {"PyDict_New", (PYTHON_PROC*)&dll_PyDict_New}, |
7 | 290 {"PyModule_GetDict", (PYTHON_PROC*)&dll_PyModule_GetDict}, |
291 {"PyRun_SimpleString", (PYTHON_PROC*)&dll_PyRun_SimpleString}, | |
292 {"PyString_AsString", (PYTHON_PROC*)&dll_PyString_AsString}, | |
293 {"PyString_FromString", (PYTHON_PROC*)&dll_PyString_FromString}, | |
294 {"PyString_FromStringAndSize", (PYTHON_PROC*)&dll_PyString_FromStringAndSize}, | |
295 {"PyString_Size", (PYTHON_PROC*)&dll_PyString_Size}, | |
296 {"PyString_Type", (PYTHON_PROC*)&dll_PyString_Type}, | |
297 {"PySys_SetObject", (PYTHON_PROC*)&dll_PySys_SetObject}, | |
298 {"PySys_SetArgv", (PYTHON_PROC*)&dll_PySys_SetArgv}, | |
299 {"PyType_Type", (PYTHON_PROC*)&dll_PyType_Type}, | |
300 {"Py_BuildValue", (PYTHON_PROC*)&dll_Py_BuildValue}, | |
301 {"Py_FindMethod", (PYTHON_PROC*)&dll_Py_FindMethod}, | |
1607 | 302 # if (PY_VERSION_HEX >= 0x02050000) && SIZEOF_SIZE_T != SIZEOF_INT |
303 {"Py_InitModule4_64", (PYTHON_PROC*)&dll_Py_InitModule4}, | |
304 # else | |
7 | 305 {"Py_InitModule4", (PYTHON_PROC*)&dll_Py_InitModule4}, |
1607 | 306 # endif |
7 | 307 {"Py_Initialize", (PYTHON_PROC*)&dll_Py_Initialize}, |
242 | 308 {"Py_Finalize", (PYTHON_PROC*)&dll_Py_Finalize}, |
309 {"Py_IsInitialized", (PYTHON_PROC*)&dll_Py_IsInitialized}, | |
7 | 310 {"_PyObject_New", (PYTHON_PROC*)&dll__PyObject_New}, |
311 {"PyObject_Init", (PYTHON_PROC*)&dll__PyObject_Init}, | |
312 {"_Py_NoneStruct", (PYTHON_PROC*)&dll__Py_NoneStruct}, | |
313 # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000 | |
314 {"PyType_IsSubtype", (PYTHON_PROC*)&dll_PyType_IsSubtype}, | |
315 # endif | |
316 # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000 | |
317 {"PyObject_Malloc", (PYTHON_PROC*)&dll_PyObject_Malloc}, | |
318 {"PyObject_Free", (PYTHON_PROC*)&dll_PyObject_Free}, | |
319 # endif | |
320 {"", NULL}, | |
321 }; | |
322 | |
323 /* | |
324 * Free python.dll | |
325 */ | |
326 static void | |
327 end_dynamic_python(void) | |
328 { | |
329 if (hinstPython) | |
330 { | |
2329
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
331 close_dll(hinstPython); |
7 | 332 hinstPython = 0; |
333 } | |
334 } | |
335 | |
336 /* | |
337 * Load library and get all pointers. | |
338 * Parameter 'libname' provides name of DLL. | |
339 * Return OK or FAIL. | |
340 */ | |
341 static int | |
342 python_runtime_link_init(char *libname, int verbose) | |
343 { | |
344 int i; | |
345 | |
2384
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
346 #if defined(UNIX) && defined(FEAT_PYTHON3) |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
347 /* Can't have Python and Python3 loaded at the same time, it may cause a |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
348 * crash. */ |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
349 if (python3_loaded()) |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
350 { |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
351 EMSG(_("E999: Python: Cannot use :py and :py3 in one session")); |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
352 return FAIL; |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
353 } |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
354 #endif |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
355 |
7 | 356 if (hinstPython) |
357 return OK; | |
2329
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
358 hinstPython = load_dll(libname); |
7 | 359 if (!hinstPython) |
360 { | |
361 if (verbose) | |
362 EMSG2(_(e_loadlib), libname); | |
363 return FAIL; | |
364 } | |
365 | |
366 for (i = 0; python_funcname_table[i].ptr; ++i) | |
367 { | |
2329
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
368 if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython, |
7 | 369 python_funcname_table[i].name)) == NULL) |
370 { | |
2329
ad2889f48843
Added support for Python 3. (Roland Puntaier)
Bram Moolenaar <bram@vim.org>
parents:
2311
diff
changeset
|
371 close_dll(hinstPython); |
7 | 372 hinstPython = 0; |
373 if (verbose) | |
374 EMSG2(_(e_loadfunc), python_funcname_table[i].name); | |
375 return FAIL; | |
376 } | |
377 } | |
378 return OK; | |
379 } | |
380 | |
381 /* | |
382 * If python is enabled (there is installed python on Windows system) return | |
383 * TRUE, else FALSE. | |
384 */ | |
385 int | |
1607 | 386 python_enabled(int verbose) |
7 | 387 { |
388 return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK; | |
389 } | |
390 | |
391 /* Load the standard Python exceptions - don't import the symbols from the | |
392 * DLL, as this can cause errors (importing data symbols is not reliable). | |
393 */ | |
394 static void get_exceptions __ARGS((void)); | |
395 | |
396 static void | |
397 get_exceptions() | |
398 { | |
399 PyObject *exmod = PyImport_ImportModule("exceptions"); | |
400 PyObject *exdict = PyModule_GetDict(exmod); | |
401 imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError"); | |
402 imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError"); | |
403 imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt"); | |
404 imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError"); | |
405 imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError"); | |
406 Py_XINCREF(imp_PyExc_AttributeError); | |
407 Py_XINCREF(imp_PyExc_IndexError); | |
408 Py_XINCREF(imp_PyExc_KeyboardInterrupt); | |
409 Py_XINCREF(imp_PyExc_TypeError); | |
410 Py_XINCREF(imp_PyExc_ValueError); | |
411 Py_XDECREF(exmod); | |
412 } | |
413 #endif /* DYNAMIC_PYTHON */ | |
414 | |
415 /****************************************************** | |
416 * Internal function prototypes. | |
417 */ | |
418 | |
419 static void DoPythonCommand(exarg_T *, const char *); | |
1607 | 420 static PyInt RangeStart; |
421 static PyInt RangeEnd; | |
7 | 422 |
423 static void PythonIO_Flush(void); | |
424 static int PythonIO_Init(void); | |
425 static int PythonMod_Init(void); | |
426 | |
427 /* Utility functions for the vim/python interface | |
428 * ---------------------------------------------- | |
429 */ | |
1607 | 430 static PyObject *GetBufferLine(buf_T *, PyInt); |
1594 | 431 static PyObject *GetBufferLineList(buf_T *, PyInt, PyInt); |
7 | 432 |
1607 | 433 static int SetBufferLine(buf_T *, PyInt, PyObject *, PyInt *); |
434 static int SetBufferLineList(buf_T *, PyInt, PyInt, PyObject *, PyInt *); | |
435 static int InsertBufferLines(buf_T *, PyInt, PyObject *, PyInt *); | |
7 | 436 |
437 static PyObject *LineToString(const char *); | |
438 static char *StringToLine(PyObject *); | |
439 | |
440 static int VimErrorCheck(void); | |
441 | |
442 #define PyErr_SetVim(str) PyErr_SetString(VimError, str) | |
443 | |
444 /****************************************************** | |
445 * 1. Python interpreter main program. | |
446 */ | |
447 | |
448 static int initialised = 0; | |
449 | |
450 #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ | |
451 typedef PyObject PyThreadState; | |
323 | 452 #endif |
453 | |
454 #ifdef PY_CAN_RECURSE | |
455 static PyGILState_STATE pygilstate = PyGILState_UNLOCKED; | |
456 #else | |
36 | 457 static PyThreadState *saved_python_thread = NULL; |
323 | 458 #endif |
7 | 459 |
460 /* | |
461 * Suspend a thread of the Python interpreter, other threads are allowed to | |
462 * run. | |
463 */ | |
36 | 464 static void |
465 Python_SaveThread(void) | |
7 | 466 { |
323 | 467 #ifdef PY_CAN_RECURSE |
468 PyGILState_Release(pygilstate); | |
469 #else | |
7 | 470 saved_python_thread = PyEval_SaveThread(); |
323 | 471 #endif |
7 | 472 } |
473 | |
474 /* | |
475 * Restore a thread of the Python interpreter, waits for other threads to | |
476 * block. | |
477 */ | |
36 | 478 static void |
479 Python_RestoreThread(void) | |
7 | 480 { |
323 | 481 #ifdef PY_CAN_RECURSE |
482 pygilstate = PyGILState_Ensure(); | |
483 #else | |
7 | 484 PyEval_RestoreThread(saved_python_thread); |
485 saved_python_thread = NULL; | |
323 | 486 #endif |
7 | 487 } |
488 | |
489 /* | |
490 * obtain a lock on the Vim data structures | |
491 */ | |
492 static void Python_Lock_Vim(void) | |
493 { | |
494 } | |
495 | |
496 /* | |
497 * release a lock on the Vim data structures | |
498 */ | |
499 static void Python_Release_Vim(void) | |
500 { | |
501 } | |
502 | |
503 void | |
504 python_end() | |
505 { | |
557 | 506 static int recurse = 0; |
507 | |
508 /* If a crash occurs while doing this, don't try again. */ | |
509 if (recurse != 0) | |
510 return; | |
511 | |
512 ++recurse; | |
513 | |
7 | 514 #ifdef DYNAMIC_PYTHON |
242 | 515 if (hinstPython && Py_IsInitialized()) |
323 | 516 { |
856 | 517 Python_RestoreThread(); /* enter python */ |
518 Py_Finalize(); | |
323 | 519 } |
7 | 520 end_dynamic_python(); |
242 | 521 #else |
522 if (Py_IsInitialized()) | |
323 | 523 { |
856 | 524 Python_RestoreThread(); /* enter python */ |
525 Py_Finalize(); | |
323 | 526 } |
7 | 527 #endif |
557 | 528 |
529 --recurse; | |
7 | 530 } |
531 | |
2384
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
532 #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO) |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
533 int |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
534 python_loaded() |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
535 { |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
536 return (hinstPython != 0); |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
537 } |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
538 #endif |
aeea25941392
Temporary solution for crashing when using both :py and :py3: disallow both in
Bram Moolenaar <bram@vim.org>
parents:
2374
diff
changeset
|
539 |
7 | 540 static int |
541 Python_Init(void) | |
542 { | |
543 if (!initialised) | |
544 { | |
545 #ifdef DYNAMIC_PYTHON | |
546 if (!python_enabled(TRUE)) | |
547 { | |
548 EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded.")); | |
549 goto fail; | |
550 } | |
551 #endif | |
552 | |
553 #if !defined(MACOS) || defined(MACOS_X_UNIX) | |
554 Py_Initialize(); | |
555 #else | |
556 PyMac_Initialize(); | |
557 #endif | |
558 /* initialise threads */ | |
559 PyEval_InitThreads(); | |
560 | |
561 #ifdef DYNAMIC_PYTHON | |
562 get_exceptions(); | |
563 #endif | |
564 | |
565 if (PythonIO_Init()) | |
566 goto fail; | |
567 | |
568 if (PythonMod_Init()) | |
569 goto fail; | |
570 | |
1748 | 571 /* Remove the element from sys.path that was added because of our |
572 * argv[0] value in PythonMod_Init(). Previously we used an empty | |
573 * string, but dependinding on the OS we then get an empty entry or | |
574 * the current directory in sys.path. */ | |
575 PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)"); | |
576 | |
36 | 577 /* the first python thread is vim's, release the lock */ |
7 | 578 Python_SaveThread(); |
579 | |
580 initialised = 1; | |
581 } | |
582 | |
583 return 0; | |
584 | |
585 fail: | |
586 /* We call PythonIO_Flush() here to print any Python errors. | |
587 * This is OK, as it is possible to call this function even | |
588 * if PythonIO_Init() has not completed successfully (it will | |
589 * not do anything in this case). | |
590 */ | |
591 PythonIO_Flush(); | |
592 return -1; | |
593 } | |
594 | |
595 /* | |
596 * External interface | |
597 */ | |
598 static void | |
599 DoPythonCommand(exarg_T *eap, const char *cmd) | |
600 { | |
323 | 601 #ifndef PY_CAN_RECURSE |
7 | 602 static int recursive = 0; |
603 #endif | |
604 #if defined(MACOS) && !defined(MACOS_X_UNIX) | |
605 GrafPtr oldPort; | |
606 #endif | |
607 #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) | |
608 char *saved_locale; | |
609 #endif | |
610 | |
611 #ifndef PY_CAN_RECURSE | |
612 if (recursive) | |
613 { | |
614 EMSG(_("E659: Cannot invoke Python recursively")); | |
615 return; | |
616 } | |
617 ++recursive; | |
618 #endif | |
619 | |
620 #if defined(MACOS) && !defined(MACOS_X_UNIX) | |
621 GetPort(&oldPort); | |
622 /* Check if the Python library is available */ | |
623 if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress) | |
624 goto theend; | |
625 #endif | |
626 if (Python_Init()) | |
627 goto theend; | |
628 | |
629 RangeStart = eap->line1; | |
630 RangeEnd = eap->line2; | |
631 Python_Release_Vim(); /* leave vim */ | |
632 | |
633 #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) | |
634 /* Python only works properly when the LC_NUMERIC locale is "C". */ | |
635 saved_locale = setlocale(LC_NUMERIC, NULL); | |
636 if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0) | |
637 saved_locale = NULL; | |
638 else | |
639 { | |
640 /* Need to make a copy, value may change when setting new locale. */ | |
641 saved_locale = (char *)vim_strsave((char_u *)saved_locale); | |
642 (void)setlocale(LC_NUMERIC, "C"); | |
643 } | |
644 #endif | |
645 | |
646 Python_RestoreThread(); /* enter python */ | |
647 | |
648 PyRun_SimpleString((char *)(cmd)); | |
649 | |
650 Python_SaveThread(); /* leave python */ | |
651 | |
652 #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) | |
653 if (saved_locale != NULL) | |
654 { | |
655 (void)setlocale(LC_NUMERIC, saved_locale); | |
656 vim_free(saved_locale); | |
657 } | |
658 #endif | |
659 | |
660 Python_Lock_Vim(); /* enter vim */ | |
661 PythonIO_Flush(); | |
662 #if defined(MACOS) && !defined(MACOS_X_UNIX) | |
663 SetPort(oldPort); | |
664 #endif | |
665 | |
666 theend: | |
667 #ifndef PY_CAN_RECURSE | |
668 --recursive; | |
669 #endif | |
670 return; /* keeps lint happy */ | |
671 } | |
672 | |
673 /* | |
674 * ":python" | |
675 */ | |
676 void | |
677 ex_python(exarg_T *eap) | |
678 { | |
679 char_u *script; | |
680 | |
681 script = script_get(eap, eap->arg); | |
682 if (!eap->skip) | |
683 { | |
684 if (script == NULL) | |
685 DoPythonCommand(eap, (char *)eap->arg); | |
686 else | |
687 DoPythonCommand(eap, (char *)script); | |
688 } | |
689 vim_free(script); | |
690 } | |
691 | |
692 #define BUFFER_SIZE 1024 | |
693 | |
694 /* | |
695 * ":pyfile" | |
696 */ | |
697 void | |
698 ex_pyfile(exarg_T *eap) | |
699 { | |
700 static char buffer[BUFFER_SIZE]; | |
701 const char *file = (char *)eap->arg; | |
702 char *p; | |
703 | |
704 /* Have to do it like this. PyRun_SimpleFile requires you to pass a | |
705 * stdio file pointer, but Vim and the Python DLL are compiled with | |
706 * different options under Windows, meaning that stdio pointers aren't | |
707 * compatible between the two. Yuk. | |
708 * | |
709 * Put the string "execfile('file')" into buffer. But, we need to | |
710 * escape any backslashes or single quotes in the file name, so that | |
711 * Python won't mangle the file name. | |
712 */ | |
713 strcpy(buffer, "execfile('"); | |
714 p = buffer + 10; /* size of "execfile('" */ | |
715 | |
716 while (*file && p < buffer + (BUFFER_SIZE - 3)) | |
717 { | |
718 if (*file == '\\' || *file == '\'') | |
719 *p++ = '\\'; | |
720 *p++ = *file++; | |
721 } | |
722 | |
723 /* If we didn't finish the file name, we hit a buffer overflow */ | |
724 if (*file != '\0') | |
725 return; | |
726 | |
727 /* Put in the terminating "')" and a null */ | |
728 *p++ = '\''; | |
729 *p++ = ')'; | |
730 *p++ = '\0'; | |
731 | |
732 /* Execute the file */ | |
733 DoPythonCommand(eap, buffer); | |
734 } | |
735 | |
736 /****************************************************** | |
737 * 2. Python output stream: writes output via [e]msg(). | |
738 */ | |
739 | |
740 /* Implementation functions | |
741 */ | |
742 | |
743 static PyObject *OutputGetattr(PyObject *, char *); | |
744 static int OutputSetattr(PyObject *, char *, PyObject *); | |
745 | |
746 static PyObject *OutputWrite(PyObject *, PyObject *); | |
747 static PyObject *OutputWritelines(PyObject *, PyObject *); | |
748 | |
749 typedef void (*writefn)(char_u *); | |
1607 | 750 static void writer(writefn fn, char_u *str, PyInt n); |
7 | 751 |
752 /* Output object definition | |
753 */ | |
754 | |
755 typedef struct | |
756 { | |
757 PyObject_HEAD | |
758 long softspace; | |
759 long error; | |
760 } OutputObject; | |
761 | |
762 static struct PyMethodDef OutputMethods[] = { | |
763 /* name, function, calling, documentation */ | |
764 {"write", OutputWrite, 1, "" }, | |
765 {"writelines", OutputWritelines, 1, "" }, | |
766 { NULL, NULL, 0, NULL } | |
767 }; | |
768 | |
769 static PyTypeObject OutputType = { | |
770 PyObject_HEAD_INIT(0) | |
771 0, | |
772 "message", | |
773 sizeof(OutputObject), | |
774 0, | |
775 | |
776 (destructor) 0, | |
777 (printfunc) 0, | |
778 (getattrfunc) OutputGetattr, | |
779 (setattrfunc) OutputSetattr, | |
780 (cmpfunc) 0, | |
781 (reprfunc) 0, | |
782 | |
783 0, /* as number */ | |
784 0, /* as sequence */ | |
785 0, /* as mapping */ | |
786 | |
787 (hashfunc) 0, | |
788 (ternaryfunc) 0, | |
789 (reprfunc) 0 | |
790 }; | |
791 | |
792 /*************/ | |
793 | |
794 static PyObject * | |
795 OutputGetattr(PyObject *self, char *name) | |
796 { | |
797 if (strcmp(name, "softspace") == 0) | |
798 return PyInt_FromLong(((OutputObject *)(self))->softspace); | |
799 | |
800 return Py_FindMethod(OutputMethods, self, name); | |
801 } | |
802 | |
803 static int | |
804 OutputSetattr(PyObject *self, char *name, PyObject *val) | |
805 { | |
806 if (val == NULL) { | |
807 PyErr_SetString(PyExc_AttributeError, _("can't delete OutputObject attributes")); | |
808 return -1; | |
809 } | |
810 | |
811 if (strcmp(name, "softspace") == 0) | |
812 { | |
813 if (!PyInt_Check(val)) { | |
814 PyErr_SetString(PyExc_TypeError, _("softspace must be an integer")); | |
815 return -1; | |
816 } | |
817 | |
818 ((OutputObject *)(self))->softspace = PyInt_AsLong(val); | |
819 return 0; | |
820 } | |
821 | |
822 PyErr_SetString(PyExc_AttributeError, _("invalid attribute")); | |
823 return -1; | |
824 } | |
825 | |
826 /*************/ | |
827 | |
828 static PyObject * | |
829 OutputWrite(PyObject *self, PyObject *args) | |
830 { | |
831 int len; | |
832 char *str; | |
833 int error = ((OutputObject *)(self))->error; | |
834 | |
835 if (!PyArg_ParseTuple(args, "s#", &str, &len)) | |
836 return NULL; | |
837 | |
838 Py_BEGIN_ALLOW_THREADS | |
839 Python_Lock_Vim(); | |
840 writer((writefn)(error ? emsg : msg), (char_u *)str, len); | |
841 Python_Release_Vim(); | |
842 Py_END_ALLOW_THREADS | |
843 | |
844 Py_INCREF(Py_None); | |
845 return Py_None; | |
846 } | |
847 | |
848 static PyObject * | |
849 OutputWritelines(PyObject *self, PyObject *args) | |
850 { | |
1594 | 851 PyInt n; |
852 PyInt i; | |
7 | 853 PyObject *list; |
854 int error = ((OutputObject *)(self))->error; | |
855 | |
856 if (!PyArg_ParseTuple(args, "O", &list)) | |
857 return NULL; | |
858 Py_INCREF(list); | |
859 | |
860 if (!PyList_Check(list)) { | |
861 PyErr_SetString(PyExc_TypeError, _("writelines() requires list of strings")); | |
862 Py_DECREF(list); | |
863 return NULL; | |
864 } | |
865 | |
866 n = PyList_Size(list); | |
867 | |
868 for (i = 0; i < n; ++i) | |
869 { | |
870 PyObject *line = PyList_GetItem(list, i); | |
871 char *str; | |
1607 | 872 PyInt len; |
7 | 873 |
874 if (!PyArg_Parse(line, "s#", &str, &len)) { | |
875 PyErr_SetString(PyExc_TypeError, _("writelines() requires list of strings")); | |
876 Py_DECREF(list); | |
877 return NULL; | |
878 } | |
879 | |
880 Py_BEGIN_ALLOW_THREADS | |
881 Python_Lock_Vim(); | |
882 writer((writefn)(error ? emsg : msg), (char_u *)str, len); | |
883 Python_Release_Vim(); | |
884 Py_END_ALLOW_THREADS | |
885 } | |
886 | |
887 Py_DECREF(list); | |
888 Py_INCREF(Py_None); | |
889 return Py_None; | |
890 } | |
891 | |
892 /* Output buffer management | |
893 */ | |
894 | |
895 static char_u *buffer = NULL; | |
1607 | 896 static PyInt buffer_len = 0; |
897 static PyInt buffer_size = 0; | |
7 | 898 |
899 static writefn old_fn = NULL; | |
900 | |
901 static void | |
1607 | 902 buffer_ensure(PyInt n) |
7 | 903 { |
1607 | 904 PyInt new_size; |
7 | 905 char_u *new_buffer; |
906 | |
907 if (n < buffer_size) | |
908 return; | |
909 | |
910 new_size = buffer_size; | |
911 while (new_size < n) | |
912 new_size += 80; | |
913 | |
914 if (new_size != buffer_size) | |
915 { | |
916 new_buffer = alloc((unsigned)new_size); | |
917 if (new_buffer == NULL) | |
918 return; | |
919 | |
920 if (buffer) | |
921 { | |
922 memcpy(new_buffer, buffer, buffer_len); | |
923 vim_free(buffer); | |
924 } | |
925 | |
926 buffer = new_buffer; | |
927 buffer_size = new_size; | |
928 } | |
929 } | |
930 | |
931 static void | |
932 PythonIO_Flush(void) | |
933 { | |
934 if (old_fn && buffer_len) | |
935 { | |
936 buffer[buffer_len] = 0; | |
937 old_fn(buffer); | |
938 } | |
939 | |
940 buffer_len = 0; | |
941 } | |
942 | |
943 static void | |
1607 | 944 writer(writefn fn, char_u *str, PyInt n) |
7 | 945 { |
946 char_u *ptr; | |
947 | |
948 if (fn != old_fn && old_fn != NULL) | |
949 PythonIO_Flush(); | |
950 | |
951 old_fn = fn; | |
952 | |
953 while (n > 0 && (ptr = memchr(str, '\n', n)) != NULL) | |
954 { | |
1607 | 955 PyInt len = ptr - str; |
7 | 956 |
957 buffer_ensure(buffer_len + len + 1); | |
958 | |
959 memcpy(buffer + buffer_len, str, len); | |
960 buffer_len += len; | |
961 buffer[buffer_len] = 0; | |
962 fn(buffer); | |
963 str = ptr + 1; | |
964 n -= len + 1; | |
965 buffer_len = 0; | |
966 } | |
967 | |
968 /* Put the remaining text into the buffer for later printing */ | |
969 buffer_ensure(buffer_len + n + 1); | |
970 memcpy(buffer + buffer_len, str, n); | |
971 buffer_len += n; | |
972 } | |
973 | |
974 /***************/ | |
975 | |
976 static OutputObject Output = | |
977 { | |
978 PyObject_HEAD_INIT(&OutputType) | |
979 0, | |
980 0 | |
981 }; | |
982 | |
983 static OutputObject Error = | |
984 { | |
985 PyObject_HEAD_INIT(&OutputType) | |
986 0, | |
987 1 | |
988 }; | |
989 | |
990 static int | |
991 PythonIO_Init(void) | |
992 { | |
993 /* Fixups... */ | |
994 OutputType.ob_type = &PyType_Type; | |
995 | |
135 | 996 PySys_SetObject("stdout", (PyObject *)(void *)&Output); |
997 PySys_SetObject("stderr", (PyObject *)(void *)&Error); | |
7 | 998 |
999 if (PyErr_Occurred()) | |
1000 { | |
1001 EMSG(_("E264: Python: Error initialising I/O objects")); | |
1002 return -1; | |
1003 } | |
1004 | |
1005 return 0; | |
1006 } | |
1007 | |
1008 /****************************************************** | |
1009 * 3. Implementation of the Vim module for Python | |
1010 */ | |
1011 | |
1012 /* Vim module - Implementation functions | |
1013 * ------------------------------------- | |
1014 */ | |
1015 | |
1016 static PyObject *VimError; | |
1017 | |
1018 static PyObject *VimCommand(PyObject *, PyObject *); | |
1019 static PyObject *VimEval(PyObject *, PyObject *); | |
1020 | |
1021 /* Window type - Implementation functions | |
1022 * -------------------------------------- | |
1023 */ | |
1024 | |
1025 typedef struct | |
1026 { | |
1027 PyObject_HEAD | |
1028 win_T *win; | |
1029 } | |
1030 WindowObject; | |
1031 | |
1032 #define INVALID_WINDOW_VALUE ((win_T *)(-1)) | |
1033 | |
1034 #define WindowType_Check(obj) ((obj)->ob_type == &WindowType) | |
1035 | |
1036 static PyObject *WindowNew(win_T *); | |
1037 | |
1038 static void WindowDestructor(PyObject *); | |
1039 static PyObject *WindowGetattr(PyObject *, char *); | |
1040 static int WindowSetattr(PyObject *, char *, PyObject *); | |
1041 static PyObject *WindowRepr(PyObject *); | |
1042 | |
1043 /* Buffer type - Implementation functions | |
1044 * -------------------------------------- | |
1045 */ | |
1046 | |
1047 typedef struct | |
1048 { | |
1049 PyObject_HEAD | |
1050 buf_T *buf; | |
1051 } | |
1052 BufferObject; | |
1053 | |
1054 #define INVALID_BUFFER_VALUE ((buf_T *)(-1)) | |
1055 | |
1056 #define BufferType_Check(obj) ((obj)->ob_type == &BufferType) | |
1057 | |
1058 static PyObject *BufferNew (buf_T *); | |
1059 | |
1060 static void BufferDestructor(PyObject *); | |
1061 static PyObject *BufferGetattr(PyObject *, char *); | |
1062 static PyObject *BufferRepr(PyObject *); | |
1063 | |
1594 | 1064 static PyInt BufferLength(PyObject *); |
1065 static PyObject *BufferItem(PyObject *, PyInt); | |
1066 static PyObject *BufferSlice(PyObject *, PyInt, PyInt); | |
1067 static PyInt BufferAssItem(PyObject *, PyInt, PyObject *); | |
1068 static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *); | |
7 | 1069 |
1070 static PyObject *BufferAppend(PyObject *, PyObject *); | |
1071 static PyObject *BufferMark(PyObject *, PyObject *); | |
1072 static PyObject *BufferRange(PyObject *, PyObject *); | |
1073 | |
1074 /* Line range type - Implementation functions | |
1075 * -------------------------------------- | |
1076 */ | |
1077 | |
1078 typedef struct | |
1079 { | |
1080 PyObject_HEAD | |
1081 BufferObject *buf; | |
1607 | 1082 PyInt start; |
1083 PyInt end; | |
7 | 1084 } |
1085 RangeObject; | |
1086 | |
1087 #define RangeType_Check(obj) ((obj)->ob_type == &RangeType) | |
1088 | |
1607 | 1089 static PyObject *RangeNew(buf_T *, PyInt, PyInt); |
7 | 1090 |
1091 static void RangeDestructor(PyObject *); | |
1092 static PyObject *RangeGetattr(PyObject *, char *); | |
1093 static PyObject *RangeRepr(PyObject *); | |
1094 | |
1594 | 1095 static PyInt RangeLength(PyObject *); |
1096 static PyObject *RangeItem(PyObject *, PyInt); | |
1097 static PyObject *RangeSlice(PyObject *, PyInt, PyInt); | |
1098 static PyInt RangeAssItem(PyObject *, PyInt, PyObject *); | |
1099 static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *); | |
7 | 1100 |
1101 static PyObject *RangeAppend(PyObject *, PyObject *); | |
1102 | |
1103 /* Window list type - Implementation functions | |
1104 * ------------------------------------------- | |
1105 */ | |
1106 | |
1594 | 1107 static PyInt WinListLength(PyObject *); |
1108 static PyObject *WinListItem(PyObject *, PyInt); | |
7 | 1109 |
1110 /* Buffer list type - Implementation functions | |
1111 * ------------------------------------------- | |
1112 */ | |
1113 | |
1594 | 1114 static PyInt BufListLength(PyObject *); |
1115 static PyObject *BufListItem(PyObject *, PyInt); | |
7 | 1116 |
1117 /* Current objects type - Implementation functions | |
1118 * ----------------------------------------------- | |
1119 */ | |
1120 | |
1121 static PyObject *CurrentGetattr(PyObject *, char *); | |
1122 static int CurrentSetattr(PyObject *, char *, PyObject *); | |
1123 | |
1124 /* Vim module - Definitions | |
1125 */ | |
1126 | |
1127 static struct PyMethodDef VimMethods[] = { | |
1128 /* name, function, calling, documentation */ | |
1607 | 1129 {"command", VimCommand, 1, "Execute a Vim ex-mode command" }, |
1130 {"eval", VimEval, 1, "Evaluate an expression using Vim evaluator" }, | |
7 | 1131 { NULL, NULL, 0, NULL } |
1132 }; | |
1133 | |
1134 /* Vim module - Implementation | |
1135 */ | |
1136 static PyObject * | |
1887 | 1137 VimCommand(PyObject *self UNUSED, PyObject *args) |
7 | 1138 { |
1139 char *cmd; | |
1140 PyObject *result; | |
1141 | |
1142 if (!PyArg_ParseTuple(args, "s", &cmd)) | |
1143 return NULL; | |
1144 | |
1145 PyErr_Clear(); | |
1146 | |
1147 Py_BEGIN_ALLOW_THREADS | |
1148 Python_Lock_Vim(); | |
1149 | |
1150 do_cmdline_cmd((char_u *)cmd); | |
1151 update_screen(VALID); | |
1152 | |
1153 Python_Release_Vim(); | |
1154 Py_END_ALLOW_THREADS | |
1155 | |
1156 if (VimErrorCheck()) | |
1157 result = NULL; | |
1158 else | |
1159 result = Py_None; | |
1160 | |
1161 Py_XINCREF(result); | |
1162 return result; | |
1163 } | |
1164 | |
1681 | 1165 #ifdef FEAT_EVAL |
634 | 1166 /* |
1167 * Function to translate a typval_T into a PyObject; this will recursively | |
1168 * translate lists/dictionaries into their Python equivalents. | |
1169 * | |
1607 | 1170 * The depth parameter is to avoid infinite recursion, set it to 1 when |
634 | 1171 * you call VimToPython. |
1172 */ | |
1173 static PyObject * | |
1174 VimToPython(typval_T *our_tv, int depth, PyObject *lookupDict) | |
1175 { | |
1176 PyObject *result; | |
1177 PyObject *newObj; | |
1178 char ptrBuf[NUMBUFLEN]; | |
1179 | |
1180 /* Avoid infinite recursion */ | |
1181 if (depth > 100) | |
1182 { | |
856 | 1183 Py_INCREF(Py_None); |
1184 result = Py_None; | |
1185 return result; | |
634 | 1186 } |
1187 | |
1188 /* Check if we run into a recursive loop. The item must be in lookupDict | |
1189 * then and we can use it again. */ | |
1786 | 1190 if ((our_tv->v_type == VAR_LIST && our_tv->vval.v_list != NULL) |
1191 || (our_tv->v_type == VAR_DICT && our_tv->vval.v_dict != NULL)) | |
1192 { | |
1193 sprintf(ptrBuf, PRINTF_DECIMAL_LONG_U, | |
2311
ccda151dde4e
Support completion for ":find". (Nazri Ramliy)
Bram Moolenaar <bram@vim.org>
parents:
2250
diff
changeset
|
1194 our_tv->v_type == VAR_LIST ? (long_u)our_tv->vval.v_list |
ccda151dde4e
Support completion for ":find". (Nazri Ramliy)
Bram Moolenaar <bram@vim.org>
parents:
2250
diff
changeset
|
1195 : (long_u)our_tv->vval.v_dict); |
1786 | 1196 result = PyDict_GetItemString(lookupDict, ptrBuf); |
1197 if (result != NULL) | |
1198 { | |
1199 Py_INCREF(result); | |
1200 return result; | |
1201 } | |
1202 } | |
1203 | |
1204 if (our_tv->v_type == VAR_STRING) | |
634 | 1205 { |
856 | 1206 result = Py_BuildValue("s", our_tv->vval.v_string); |
634 | 1207 } |
1208 else if (our_tv->v_type == VAR_NUMBER) | |
1209 { | |
856 | 1210 char buf[NUMBUFLEN]; |
634 | 1211 |
1212 /* For backwards compatibility numbers are stored as strings. */ | |
856 | 1213 sprintf(buf, "%ld", (long)our_tv->vval.v_number); |
1214 result = Py_BuildValue("s", buf); | |
634 | 1215 } |
1681 | 1216 # ifdef FEAT_FLOAT |
1594 | 1217 else if (our_tv->v_type == VAR_FLOAT) |
1218 { | |
1219 char buf[NUMBUFLEN]; | |
1220 | |
1221 sprintf(buf, "%f", our_tv->vval.v_float); | |
1222 result = Py_BuildValue("s", buf); | |
1223 } | |
1681 | 1224 # endif |
634 | 1225 else if (our_tv->v_type == VAR_LIST) |
1226 { | |
856 | 1227 list_T *list = our_tv->vval.v_list; |
1228 listitem_T *curr; | |
1229 | |
1230 result = PyList_New(0); | |
634 | 1231 |
1232 if (list != NULL) | |
1233 { | |
1786 | 1234 PyDict_SetItemString(lookupDict, ptrBuf, result); |
1235 | |
634 | 1236 for (curr = list->lv_first; curr != NULL; curr = curr->li_next) |
1237 { | |
1238 newObj = VimToPython(&curr->li_tv, depth + 1, lookupDict); | |
1239 PyList_Append(result, newObj); | |
1240 Py_DECREF(newObj); | |
1241 } | |
1242 } | |
1243 } | |
1244 else if (our_tv->v_type == VAR_DICT) | |
1245 { | |
856 | 1246 result = PyDict_New(); |
634 | 1247 |
1248 if (our_tv->vval.v_dict != NULL) | |
1249 { | |
1250 hashtab_T *ht = &our_tv->vval.v_dict->dv_hashtab; | |
1607 | 1251 long_u todo = ht->ht_used; |
634 | 1252 hashitem_T *hi; |
1253 dictitem_T *di; | |
1254 | |
1786 | 1255 PyDict_SetItemString(lookupDict, ptrBuf, result); |
1256 | |
634 | 1257 for (hi = ht->ht_array; todo > 0; ++hi) |
1258 { | |
1259 if (!HASHITEM_EMPTY(hi)) | |
1260 { | |
1261 --todo; | |
1262 | |
1263 di = dict_lookup(hi); | |
1264 newObj = VimToPython(&di->di_tv, depth + 1, lookupDict); | |
1265 PyDict_SetItemString(result, (char *)hi->hi_key, newObj); | |
1266 Py_DECREF(newObj); | |
1267 } | |
1268 } | |
1269 } | |
1270 } | |
1271 else | |
1272 { | |
856 | 1273 Py_INCREF(Py_None); |
1274 result = Py_None; | |
634 | 1275 } |
1276 | |
1277 return result; | |
1278 } | |
1681 | 1279 #endif |
634 | 1280 |
7 | 1281 static PyObject * |
1887 | 1282 VimEval(PyObject *self UNUSED, PyObject *args) |
7 | 1283 { |
1284 #ifdef FEAT_EVAL | |
1285 char *expr; | |
634 | 1286 typval_T *our_tv; |
7 | 1287 PyObject *result; |
634 | 1288 PyObject *lookup_dict; |
7 | 1289 |
1290 if (!PyArg_ParseTuple(args, "s", &expr)) | |
1291 return NULL; | |
1292 | |
1293 Py_BEGIN_ALLOW_THREADS | |
1294 Python_Lock_Vim(); | |
634 | 1295 our_tv = eval_expr((char_u *)expr, NULL); |
1296 | |
7 | 1297 Python_Release_Vim(); |
1298 Py_END_ALLOW_THREADS | |
1299 | |
634 | 1300 if (our_tv == NULL) |
7 | 1301 { |
1302 PyErr_SetVim(_("invalid expression")); | |
1303 return NULL; | |
1304 } | |
1305 | |
634 | 1306 /* Convert the Vim type into a Python type. Create a dictionary that's |
1307 * used to check for recursive loops. */ | |
1308 lookup_dict = PyDict_New(); | |
1309 result = VimToPython(our_tv, 1, lookup_dict); | |
1310 Py_DECREF(lookup_dict); | |
1311 | |
7 | 1312 |
1313 Py_BEGIN_ALLOW_THREADS | |
1314 Python_Lock_Vim(); | |
634 | 1315 free_tv(our_tv); |
7 | 1316 Python_Release_Vim(); |
1317 Py_END_ALLOW_THREADS | |
1318 | |
1319 return result; | |
1320 #else | |
1321 PyErr_SetVim(_("expressions disabled at compile time")); | |
1322 return NULL; | |
1323 #endif | |
1324 } | |
1325 | |
1326 /* Common routines for buffers and line ranges | |
1327 * ------------------------------------------- | |
1328 */ | |
1329 static int | |
1330 CheckBuffer(BufferObject *this) | |
1331 { | |
1332 if (this->buf == INVALID_BUFFER_VALUE) | |
1333 { | |
1334 PyErr_SetVim(_("attempt to refer to deleted buffer")); | |
1335 return -1; | |
1336 } | |
1337 | |
1338 return 0; | |
1339 } | |
1340 | |
1341 static PyObject * | |
1607 | 1342 RBItem(BufferObject *self, PyInt n, PyInt start, PyInt end) |
7 | 1343 { |
1344 if (CheckBuffer(self)) | |
1345 return NULL; | |
1346 | |
1347 if (n < 0 || n > end - start) | |
1348 { | |
1349 PyErr_SetString(PyExc_IndexError, _("line number out of range")); | |
1350 return NULL; | |
1351 } | |
1352 | |
1353 return GetBufferLine(self->buf, n+start); | |
1354 } | |
1355 | |
1356 static PyObject * | |
1607 | 1357 RBSlice(BufferObject *self, PyInt lo, PyInt hi, PyInt start, PyInt end) |
7 | 1358 { |
1594 | 1359 PyInt size; |
7 | 1360 |
1361 if (CheckBuffer(self)) | |
1362 return NULL; | |
1363 | |
1364 size = end - start + 1; | |
1365 | |
1366 if (lo < 0) | |
1367 lo = 0; | |
1368 else if (lo > size) | |
1369 lo = size; | |
1370 if (hi < 0) | |
1371 hi = 0; | |
1372 if (hi < lo) | |
1373 hi = lo; | |
1374 else if (hi > size) | |
1375 hi = size; | |
1376 | |
1377 return GetBufferLineList(self->buf, lo+start, hi+start); | |
1378 } | |
1379 | |
1594 | 1380 static PyInt |
1607 | 1381 RBAssItem(BufferObject *self, PyInt n, PyObject *val, PyInt start, PyInt end, PyInt *new_end) |
7 | 1382 { |
1607 | 1383 PyInt len_change; |
7 | 1384 |
1385 if (CheckBuffer(self)) | |
1386 return -1; | |
1387 | |
1388 if (n < 0 || n > end - start) | |
1389 { | |
1390 PyErr_SetString(PyExc_IndexError, _("line number out of range")); | |
1391 return -1; | |
1392 } | |
1393 | |
1394 if (SetBufferLine(self->buf, n+start, val, &len_change) == FAIL) | |
1395 return -1; | |
1396 | |
1397 if (new_end) | |
1398 *new_end = end + len_change; | |
1399 | |
1400 return 0; | |
1401 } | |
1402 | |
1594 | 1403 static PyInt |
1607 | 1404 RBAssSlice(BufferObject *self, PyInt lo, PyInt hi, PyObject *val, PyInt start, PyInt end, PyInt *new_end) |
7 | 1405 { |
1607 | 1406 PyInt size; |
1407 PyInt len_change; | |
7 | 1408 |
1409 /* Self must be a valid buffer */ | |
1410 if (CheckBuffer(self)) | |
1411 return -1; | |
1412 | |
1413 /* Sort out the slice range */ | |
1414 size = end - start + 1; | |
1415 | |
1416 if (lo < 0) | |
1417 lo = 0; | |
1418 else if (lo > size) | |
1419 lo = size; | |
1420 if (hi < 0) | |
1421 hi = 0; | |
1422 if (hi < lo) | |
1423 hi = lo; | |
1424 else if (hi > size) | |
1425 hi = size; | |
1426 | |
1427 if (SetBufferLineList(self->buf, lo+start, hi+start, val, &len_change) == FAIL) | |
1428 return -1; | |
1429 | |
1430 if (new_end) | |
1431 *new_end = end + len_change; | |
1432 | |
1433 return 0; | |
1434 } | |
1435 | |
1436 static PyObject * | |
1607 | 1437 RBAppend(BufferObject *self, PyObject *args, PyInt start, PyInt end, PyInt *new_end) |
7 | 1438 { |
1439 PyObject *lines; | |
1607 | 1440 PyInt len_change; |
1441 PyInt max; | |
1442 PyInt n; | |
7 | 1443 |
1444 if (CheckBuffer(self)) | |
1445 return NULL; | |
1446 | |
1447 max = n = end - start + 1; | |
1448 | |
1607 | 1449 if (!PyArg_ParseTuple(args, "O|" Py_ssize_t_fmt, &lines, &n)) |
7 | 1450 return NULL; |
1451 | |
1452 if (n < 0 || n > max) | |
1453 { | |
1454 PyErr_SetString(PyExc_ValueError, _("line number out of range")); | |
1455 return NULL; | |
1456 } | |
1457 | |
1458 if (InsertBufferLines(self->buf, n + start - 1, lines, &len_change) == FAIL) | |
1459 return NULL; | |
1460 | |
1461 if (new_end) | |
1462 *new_end = end + len_change; | |
1463 | |
1464 Py_INCREF(Py_None); | |
1465 return Py_None; | |
1466 } | |
1467 | |
1468 | |
1469 /* Buffer object - Definitions | |
1470 */ | |
1471 | |
1472 static struct PyMethodDef BufferMethods[] = { | |
1473 /* name, function, calling, documentation */ | |
1607 | 1474 {"append", BufferAppend, 1, "Append data to Vim buffer" }, |
1475 {"mark", BufferMark, 1, "Return (row,col) representing position of named mark" }, | |
1476 {"range", BufferRange, 1, "Return a range object which represents the part of the given buffer between line numbers s and e" }, | |
7 | 1477 { NULL, NULL, 0, NULL } |
1478 }; | |
1479 | |
1480 static PySequenceMethods BufferAsSeq = { | |
1594 | 1481 (PyInquiry) BufferLength, /* sq_length, len(x) */ |
7 | 1482 (binaryfunc) 0, /* BufferConcat, */ /* sq_concat, x+y */ |
1594 | 1483 (PyIntArgFunc) 0, /* BufferRepeat, */ /* sq_repeat, x*n */ |
1484 (PyIntArgFunc) BufferItem, /* sq_item, x[i] */ | |
1485 (PyIntIntArgFunc) BufferSlice, /* sq_slice, x[i:j] */ | |
1486 (PyIntObjArgProc) BufferAssItem, /* sq_ass_item, x[i]=v */ | |
1487 (PyIntIntObjArgProc) BufferAssSlice, /* sq_ass_slice, x[i:j]=v */ | |
7 | 1488 }; |
1489 | |
1490 static PyTypeObject BufferType = { | |
1491 PyObject_HEAD_INIT(0) | |
1492 0, | |
1493 "buffer", | |
1494 sizeof(BufferObject), | |
1495 0, | |
1496 | |
1497 (destructor) BufferDestructor, /* tp_dealloc, refcount==0 */ | |
1498 (printfunc) 0, /* tp_print, print x */ | |
1499 (getattrfunc) BufferGetattr, /* tp_getattr, x.attr */ | |
1500 (setattrfunc) 0, /* tp_setattr, x.attr=v */ | |
1501 (cmpfunc) 0, /* tp_compare, x>y */ | |
1502 (reprfunc) BufferRepr, /* tp_repr, `x`, print x */ | |
1503 | |
1504 0, /* as number */ | |
1505 &BufferAsSeq, /* as sequence */ | |
1506 0, /* as mapping */ | |
1507 | |
1508 (hashfunc) 0, /* tp_hash, dict(x) */ | |
1509 (ternaryfunc) 0, /* tp_call, x() */ | |
1510 (reprfunc) 0, /* tp_str, str(x) */ | |
1511 }; | |
1512 | |
1513 /* Buffer object - Implementation | |
1514 */ | |
1515 | |
1516 static PyObject * | |
1517 BufferNew(buf_T *buf) | |
1518 { | |
1519 /* We need to handle deletion of buffers underneath us. | |
502 | 1520 * If we add a "b_python_ref" field to the buf_T structure, |
7 | 1521 * then we can get at it in buf_freeall() in vim. We then |
1522 * need to create only ONE Python object per buffer - if | |
1523 * we try to create a second, just INCREF the existing one | |
1524 * and return it. The (single) Python object referring to | |
502 | 1525 * the buffer is stored in "b_python_ref". |
7 | 1526 * Question: what to do on a buf_freeall(). We'll probably |
1527 * have to either delete the Python object (DECREF it to | |
1528 * zero - a bad idea, as it leaves dangling refs!) or | |
1529 * set the buf_T * value to an invalid value (-1?), which | |
1530 * means we need checks in all access functions... Bah. | |
1531 */ | |
1532 | |
1533 BufferObject *self; | |
1534 | |
502 | 1535 if (buf->b_python_ref != NULL) |
7 | 1536 { |
502 | 1537 self = buf->b_python_ref; |
7 | 1538 Py_INCREF(self); |
1539 } | |
1540 else | |
1541 { | |
1542 self = PyObject_NEW(BufferObject, &BufferType); | |
1543 if (self == NULL) | |
1544 return NULL; | |
1545 self->buf = buf; | |
502 | 1546 buf->b_python_ref = self; |
7 | 1547 } |
1548 | |
1549 return (PyObject *)(self); | |
1550 } | |
1551 | |
1552 static void | |
1553 BufferDestructor(PyObject *self) | |
1554 { | |
1555 BufferObject *this = (BufferObject *)(self); | |
1556 | |
1557 if (this->buf && this->buf != INVALID_BUFFER_VALUE) | |
502 | 1558 this->buf->b_python_ref = NULL; |
7 | 1559 |
986 | 1560 Py_DECREF(self); |
7 | 1561 } |
1562 | |
1563 static PyObject * | |
1564 BufferGetattr(PyObject *self, char *name) | |
1565 { | |
1566 BufferObject *this = (BufferObject *)(self); | |
1567 | |
1568 if (CheckBuffer(this)) | |
1569 return NULL; | |
1570 | |
1571 if (strcmp(name, "name") == 0) | |
1607 | 1572 return Py_BuildValue("s", this->buf->b_ffname); |
7 | 1573 else if (strcmp(name, "number") == 0) |
1607 | 1574 return Py_BuildValue(Py_ssize_t_fmt, this->buf->b_fnum); |
7 | 1575 else if (strcmp(name,"__members__") == 0) |
1576 return Py_BuildValue("[ss]", "name", "number"); | |
1577 else | |
1578 return Py_FindMethod(BufferMethods, self, name); | |
1579 } | |
1580 | |
1581 static PyObject * | |
1582 BufferRepr(PyObject *self) | |
1583 { | |
274 | 1584 static char repr[100]; |
7 | 1585 BufferObject *this = (BufferObject *)(self); |
1586 | |
1587 if (this->buf == INVALID_BUFFER_VALUE) | |
1588 { | |
1607 | 1589 vim_snprintf(repr, 100, _("<buffer object (deleted) at %p>"), (self)); |
7 | 1590 return PyString_FromString(repr); |
1591 } | |
1592 else | |
1593 { | |
1594 char *name = (char *)this->buf->b_fname; | |
1607 | 1595 PyInt len; |
7 | 1596 |
1597 if (name == NULL) | |
1598 name = ""; | |
1599 len = strlen(name); | |
1600 | |
1601 if (len > 35) | |
1602 name = name + (35 - len); | |
1603 | |
274 | 1604 vim_snprintf(repr, 100, "<buffer %s%s>", len > 35 ? "..." : "", name); |
7 | 1605 |
1606 return PyString_FromString(repr); | |
1607 } | |
1608 } | |
1609 | |
1610 /******************/ | |
1611 | |
1594 | 1612 static PyInt |
7 | 1613 BufferLength(PyObject *self) |
1614 { | |
1615 /* HOW DO WE SIGNAL AN ERROR FROM THIS FUNCTION? */ | |
1616 if (CheckBuffer((BufferObject *)(self))) | |
1617 return -1; /* ??? */ | |
1618 | |
1619 return (((BufferObject *)(self))->buf->b_ml.ml_line_count); | |
1620 } | |
1621 | |
1622 static PyObject * | |
1594 | 1623 BufferItem(PyObject *self, PyInt n) |
7 | 1624 { |
1625 return RBItem((BufferObject *)(self), n, 1, | |
1626 (int)((BufferObject *)(self))->buf->b_ml.ml_line_count); | |
1627 } | |
1628 | |
1629 static PyObject * | |
1594 | 1630 BufferSlice(PyObject *self, PyInt lo, PyInt hi) |
7 | 1631 { |
1632 return RBSlice((BufferObject *)(self), lo, hi, 1, | |
1633 (int)((BufferObject *)(self))->buf->b_ml.ml_line_count); | |
1634 } | |
1635 | |
1594 | 1636 static PyInt |
1637 BufferAssItem(PyObject *self, PyInt n, PyObject *val) | |
7 | 1638 { |
1639 return RBAssItem((BufferObject *)(self), n, val, 1, | |
1607 | 1640 (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, |
7 | 1641 NULL); |
1642 } | |
1643 | |
1594 | 1644 static PyInt |
1645 BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val) | |
7 | 1646 { |
1647 return RBAssSlice((BufferObject *)(self), lo, hi, val, 1, | |
1607 | 1648 (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, |
7 | 1649 NULL); |
1650 } | |
1651 | |
1652 static PyObject * | |
1653 BufferAppend(PyObject *self, PyObject *args) | |
1654 { | |
1655 return RBAppend((BufferObject *)(self), args, 1, | |
1607 | 1656 (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, |
7 | 1657 NULL); |
1658 } | |
1659 | |
1660 static PyObject * | |
1661 BufferMark(PyObject *self, PyObject *args) | |
1662 { | |
1663 pos_T *posp; | |
1664 char mark; | |
1665 buf_T *curbuf_save; | |
1666 | |
1667 if (CheckBuffer((BufferObject *)(self))) | |
1668 return NULL; | |
1669 | |
1670 if (!PyArg_ParseTuple(args, "c", &mark)) | |
1671 return NULL; | |
1672 | |
1673 curbuf_save = curbuf; | |
1674 curbuf = ((BufferObject *)(self))->buf; | |
1675 posp = getmark(mark, FALSE); | |
1676 curbuf = curbuf_save; | |
1677 | |
1678 if (posp == NULL) | |
1679 { | |
1680 PyErr_SetVim(_("invalid mark name")); | |
1681 return NULL; | |
1682 } | |
1683 | |
1684 /* Ckeck for keyboard interrupt */ | |
1685 if (VimErrorCheck()) | |
1686 return NULL; | |
1687 | |
1688 if (posp->lnum <= 0) | |
1689 { | |
1690 /* Or raise an error? */ | |
1691 Py_INCREF(Py_None); | |
1692 return Py_None; | |
1693 } | |
1694 | |
1695 return Py_BuildValue("(ll)", (long)(posp->lnum), (long)(posp->col)); | |
1696 } | |
1697 | |
1698 static PyObject * | |
1699 BufferRange(PyObject *self, PyObject *args) | |
1700 { | |
1607 | 1701 PyInt start; |
1702 PyInt end; | |
7 | 1703 |
1704 if (CheckBuffer((BufferObject *)(self))) | |
1705 return NULL; | |
1706 | |
1607 | 1707 if (!PyArg_ParseTuple(args, Py_ssize_t_fmt Py_ssize_t_fmt, &start, &end)) |
7 | 1708 return NULL; |
1709 | |
1710 return RangeNew(((BufferObject *)(self))->buf, start, end); | |
1711 } | |
1712 | |
1713 /* Line range object - Definitions | |
1714 */ | |
1715 | |
1716 static struct PyMethodDef RangeMethods[] = { | |
1717 /* name, function, calling, documentation */ | |
1607 | 1718 {"append", RangeAppend, 1, "Append data to the Vim range" }, |
7 | 1719 { NULL, NULL, 0, NULL } |
1720 }; | |
1721 | |
1722 static PySequenceMethods RangeAsSeq = { | |
1594 | 1723 (PyInquiry) RangeLength, /* sq_length, len(x) */ |
7 | 1724 (binaryfunc) 0, /* RangeConcat, */ /* sq_concat, x+y */ |
1594 | 1725 (PyIntArgFunc) 0, /* RangeRepeat, */ /* sq_repeat, x*n */ |
1726 (PyIntArgFunc) RangeItem, /* sq_item, x[i] */ | |
1727 (PyIntIntArgFunc) RangeSlice, /* sq_slice, x[i:j] */ | |
1728 (PyIntObjArgProc) RangeAssItem, /* sq_ass_item, x[i]=v */ | |
1729 (PyIntIntObjArgProc) RangeAssSlice, /* sq_ass_slice, x[i:j]=v */ | |
7 | 1730 }; |
1731 | |
1732 static PyTypeObject RangeType = { | |
1733 PyObject_HEAD_INIT(0) | |
1734 0, | |
1735 "range", | |
1736 sizeof(RangeObject), | |
1737 0, | |
1738 | |
1739 (destructor) RangeDestructor, /* tp_dealloc, refcount==0 */ | |
1740 (printfunc) 0, /* tp_print, print x */ | |
1741 (getattrfunc) RangeGetattr, /* tp_getattr, x.attr */ | |
1742 (setattrfunc) 0, /* tp_setattr, x.attr=v */ | |
1743 (cmpfunc) 0, /* tp_compare, x>y */ | |
1744 (reprfunc) RangeRepr, /* tp_repr, `x`, print x */ | |
1745 | |
1746 0, /* as number */ | |
1747 &RangeAsSeq, /* as sequence */ | |
1748 0, /* as mapping */ | |
1749 | |
1750 (hashfunc) 0, /* tp_hash, dict(x) */ | |
1751 (ternaryfunc) 0, /* tp_call, x() */ | |
1752 (reprfunc) 0, /* tp_str, str(x) */ | |
1753 }; | |
1754 | |
1755 /* Line range object - Implementation | |
1756 */ | |
1757 | |
1758 static PyObject * | |
1607 | 1759 RangeNew(buf_T *buf, PyInt start, PyInt end) |
7 | 1760 { |
1761 BufferObject *bufr; | |
1762 RangeObject *self; | |
1763 self = PyObject_NEW(RangeObject, &RangeType); | |
1764 if (self == NULL) | |
1765 return NULL; | |
1766 | |
1767 bufr = (BufferObject *)BufferNew(buf); | |
1768 if (bufr == NULL) | |
1769 { | |
986 | 1770 Py_DECREF(self); |
7 | 1771 return NULL; |
1772 } | |
1773 Py_INCREF(bufr); | |
1774 | |
1775 self->buf = bufr; | |
1776 self->start = start; | |
1777 self->end = end; | |
1778 | |
1779 return (PyObject *)(self); | |
1780 } | |
1781 | |
1782 static void | |
1783 RangeDestructor(PyObject *self) | |
1784 { | |
1785 Py_DECREF(((RangeObject *)(self))->buf); | |
986 | 1786 Py_DECREF(self); |
7 | 1787 } |
1788 | |
1789 static PyObject * | |
1790 RangeGetattr(PyObject *self, char *name) | |
1791 { | |
1792 if (strcmp(name, "start") == 0) | |
1607 | 1793 return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1); |
7 | 1794 else if (strcmp(name, "end") == 0) |
1607 | 1795 return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1); |
7 | 1796 else |
1797 return Py_FindMethod(RangeMethods, self, name); | |
1798 } | |
1799 | |
1800 static PyObject * | |
1801 RangeRepr(PyObject *self) | |
1802 { | |
274 | 1803 static char repr[100]; |
7 | 1804 RangeObject *this = (RangeObject *)(self); |
1805 | |
1806 if (this->buf->buf == INVALID_BUFFER_VALUE) | |
1807 { | |
1607 | 1808 vim_snprintf(repr, 100, "<range object (for deleted buffer) at %p>", |
1809 (self)); | |
7 | 1810 return PyString_FromString(repr); |
1811 } | |
1812 else | |
1813 { | |
1814 char *name = (char *)this->buf->buf->b_fname; | |
1815 int len; | |
1816 | |
1817 if (name == NULL) | |
1818 name = ""; | |
1668 | 1819 len = (int)strlen(name); |
7 | 1820 |
1821 if (len > 45) | |
1822 name = name + (45 - len); | |
1823 | |
274 | 1824 vim_snprintf(repr, 100, "<range %s%s (%d:%d)>", |
7 | 1825 len > 45 ? "..." : "", name, |
1826 this->start, this->end); | |
1827 | |
1828 return PyString_FromString(repr); | |
1829 } | |
1830 } | |
1831 | |
1832 /****************/ | |
1833 | |
1594 | 1834 static PyInt |
7 | 1835 RangeLength(PyObject *self) |
1836 { | |
1837 /* HOW DO WE SIGNAL AN ERROR FROM THIS FUNCTION? */ | |
1838 if (CheckBuffer(((RangeObject *)(self))->buf)) | |
1839 return -1; /* ??? */ | |
1840 | |
1841 return (((RangeObject *)(self))->end - ((RangeObject *)(self))->start + 1); | |
1842 } | |
1843 | |
1844 static PyObject * | |
1594 | 1845 RangeItem(PyObject *self, PyInt n) |
7 | 1846 { |
1847 return RBItem(((RangeObject *)(self))->buf, n, | |
1848 ((RangeObject *)(self))->start, | |
1849 ((RangeObject *)(self))->end); | |
1850 } | |
1851 | |
1852 static PyObject * | |
1594 | 1853 RangeSlice(PyObject *self, PyInt lo, PyInt hi) |
7 | 1854 { |
1855 return RBSlice(((RangeObject *)(self))->buf, lo, hi, | |
1856 ((RangeObject *)(self))->start, | |
1857 ((RangeObject *)(self))->end); | |
1858 } | |
1859 | |
1594 | 1860 static PyInt |
1861 RangeAssItem(PyObject *self, PyInt n, PyObject *val) | |
7 | 1862 { |
1863 return RBAssItem(((RangeObject *)(self))->buf, n, val, | |
1864 ((RangeObject *)(self))->start, | |
1865 ((RangeObject *)(self))->end, | |
1866 &((RangeObject *)(self))->end); | |
1867 } | |
1868 | |
1594 | 1869 static PyInt |
1870 RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val) | |
7 | 1871 { |
1872 return RBAssSlice(((RangeObject *)(self))->buf, lo, hi, val, | |
1873 ((RangeObject *)(self))->start, | |
1874 ((RangeObject *)(self))->end, | |
1875 &((RangeObject *)(self))->end); | |
1876 } | |
1877 | |
1878 static PyObject * | |
1879 RangeAppend(PyObject *self, PyObject *args) | |
1880 { | |
1881 return RBAppend(((RangeObject *)(self))->buf, args, | |
1882 ((RangeObject *)(self))->start, | |
1883 ((RangeObject *)(self))->end, | |
1884 &((RangeObject *)(self))->end); | |
1885 } | |
1886 | |
1887 /* Buffer list object - Definitions | |
1888 */ | |
1889 | |
1890 typedef struct | |
1891 { | |
1892 PyObject_HEAD | |
1893 } | |
1894 BufListObject; | |
1895 | |
1896 static PySequenceMethods BufListAsSeq = { | |
1594 | 1897 (PyInquiry) BufListLength, /* sq_length, len(x) */ |
7 | 1898 (binaryfunc) 0, /* sq_concat, x+y */ |
1594 | 1899 (PyIntArgFunc) 0, /* sq_repeat, x*n */ |
1900 (PyIntArgFunc) BufListItem, /* sq_item, x[i] */ | |
1901 (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */ | |
1902 (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */ | |
1903 (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */ | |
7 | 1904 }; |
1905 | |
1906 static PyTypeObject BufListType = { | |
1907 PyObject_HEAD_INIT(0) | |
1908 0, | |
1909 "buffer list", | |
1910 sizeof(BufListObject), | |
1911 0, | |
1912 | |
1913 (destructor) 0, /* tp_dealloc, refcount==0 */ | |
1914 (printfunc) 0, /* tp_print, print x */ | |
1915 (getattrfunc) 0, /* tp_getattr, x.attr */ | |
1916 (setattrfunc) 0, /* tp_setattr, x.attr=v */ | |
1917 (cmpfunc) 0, /* tp_compare, x>y */ | |
1918 (reprfunc) 0, /* tp_repr, `x`, print x */ | |
1919 | |
1920 0, /* as number */ | |
1921 &BufListAsSeq, /* as sequence */ | |
1922 0, /* as mapping */ | |
1923 | |
1924 (hashfunc) 0, /* tp_hash, dict(x) */ | |
1925 (ternaryfunc) 0, /* tp_call, x() */ | |
1926 (reprfunc) 0, /* tp_str, str(x) */ | |
1927 }; | |
1928 | |
1929 /* Buffer list object - Implementation | |
1930 */ | |
1931 | |
1594 | 1932 static PyInt |
1887 | 1933 BufListLength(PyObject *self UNUSED) |
7 | 1934 { |
1935 buf_T *b = firstbuf; | |
1607 | 1936 PyInt n = 0; |
7 | 1937 |
1938 while (b) | |
1939 { | |
1940 ++n; | |
1941 b = b->b_next; | |
1942 } | |
1943 | |
1944 return n; | |
1945 } | |
1946 | |
1947 static PyObject * | |
1887 | 1948 BufListItem(PyObject *self UNUSED, PyInt n) |
7 | 1949 { |
1950 buf_T *b; | |
1951 | |
1952 for (b = firstbuf; b; b = b->b_next, --n) | |
1953 { | |
1954 if (n == 0) | |
1955 return BufferNew(b); | |
1956 } | |
1957 | |
1958 PyErr_SetString(PyExc_IndexError, _("no such buffer")); | |
1959 return NULL; | |
1960 } | |
1961 | |
1962 /* Window object - Definitions | |
1963 */ | |
1964 | |
1965 static struct PyMethodDef WindowMethods[] = { | |
1966 /* name, function, calling, documentation */ | |
1967 { NULL, NULL, 0, NULL } | |
1968 }; | |
1969 | |
1970 static PyTypeObject WindowType = { | |
1971 PyObject_HEAD_INIT(0) | |
1972 0, | |
1973 "window", | |
1974 sizeof(WindowObject), | |
1975 0, | |
1976 | |
1977 (destructor) WindowDestructor, /* tp_dealloc, refcount==0 */ | |
1978 (printfunc) 0, /* tp_print, print x */ | |
1979 (getattrfunc) WindowGetattr, /* tp_getattr, x.attr */ | |
1980 (setattrfunc) WindowSetattr, /* tp_setattr, x.attr=v */ | |
1981 (cmpfunc) 0, /* tp_compare, x>y */ | |
1982 (reprfunc) WindowRepr, /* tp_repr, `x`, print x */ | |
1983 | |
1984 0, /* as number */ | |
1985 0, /* as sequence */ | |
1986 0, /* as mapping */ | |
1987 | |
1988 (hashfunc) 0, /* tp_hash, dict(x) */ | |
1989 (ternaryfunc) 0, /* tp_call, x() */ | |
1990 (reprfunc) 0, /* tp_str, str(x) */ | |
1991 }; | |
1992 | |
1993 /* Window object - Implementation | |
1994 */ | |
1995 | |
1996 static PyObject * | |
1997 WindowNew(win_T *win) | |
1998 { | |
1999 /* We need to handle deletion of windows underneath us. | |
502 | 2000 * If we add a "w_python_ref" field to the win_T structure, |
7 | 2001 * then we can get at it in win_free() in vim. We then |
2002 * need to create only ONE Python object per window - if | |
2003 * we try to create a second, just INCREF the existing one | |
2004 * and return it. The (single) Python object referring to | |
502 | 2005 * the window is stored in "w_python_ref". |
7 | 2006 * On a win_free() we set the Python object's win_T* field |
2007 * to an invalid value. We trap all uses of a window | |
2008 * object, and reject them if the win_T* field is invalid. | |
2009 */ | |
2010 | |
2011 WindowObject *self; | |
2012 | |
502 | 2013 if (win->w_python_ref) |
7 | 2014 { |
502 | 2015 self = win->w_python_ref; |
7 | 2016 Py_INCREF(self); |
2017 } | |
2018 else | |
2019 { | |
2020 self = PyObject_NEW(WindowObject, &WindowType); | |
2021 if (self == NULL) | |
2022 return NULL; | |
2023 self->win = win; | |
502 | 2024 win->w_python_ref = self; |
7 | 2025 } |
2026 | |
2027 return (PyObject *)(self); | |
2028 } | |
2029 | |
2030 static void | |
2031 WindowDestructor(PyObject *self) | |
2032 { | |
2033 WindowObject *this = (WindowObject *)(self); | |
2034 | |
2035 if (this->win && this->win != INVALID_WINDOW_VALUE) | |
502 | 2036 this->win->w_python_ref = NULL; |
7 | 2037 |
986 | 2038 Py_DECREF(self); |
7 | 2039 } |
2040 | |
2041 static int | |
2042 CheckWindow(WindowObject *this) | |
2043 { | |
2044 if (this->win == INVALID_WINDOW_VALUE) | |
2045 { | |
2046 PyErr_SetVim(_("attempt to refer to deleted window")); | |
2047 return -1; | |
2048 } | |
2049 | |
2050 return 0; | |
2051 } | |
2052 | |
2053 static PyObject * | |
2054 WindowGetattr(PyObject *self, char *name) | |
2055 { | |
2056 WindowObject *this = (WindowObject *)(self); | |
2057 | |
2058 if (CheckWindow(this)) | |
2059 return NULL; | |
2060 | |
2061 if (strcmp(name, "buffer") == 0) | |
2062 return (PyObject *)BufferNew(this->win->w_buffer); | |
2063 else if (strcmp(name, "cursor") == 0) | |
2064 { | |
2065 pos_T *pos = &this->win->w_cursor; | |
2066 | |
2067 return Py_BuildValue("(ll)", (long)(pos->lnum), (long)(pos->col)); | |
2068 } | |
2069 else if (strcmp(name, "height") == 0) | |
2070 return Py_BuildValue("l", (long)(this->win->w_height)); | |
2071 #ifdef FEAT_VERTSPLIT | |
2072 else if (strcmp(name, "width") == 0) | |
2073 return Py_BuildValue("l", (long)(W_WIDTH(this->win))); | |
2074 #endif | |
2075 else if (strcmp(name,"__members__") == 0) | |
2076 return Py_BuildValue("[sss]", "buffer", "cursor", "height"); | |
2077 else | |
2078 return Py_FindMethod(WindowMethods, self, name); | |
2079 } | |
2080 | |
2081 static int | |
2082 WindowSetattr(PyObject *self, char *name, PyObject *val) | |
2083 { | |
2084 WindowObject *this = (WindowObject *)(self); | |
2085 | |
2086 if (CheckWindow(this)) | |
2087 return -1; | |
2088 | |
2089 if (strcmp(name, "buffer") == 0) | |
2090 { | |
2091 PyErr_SetString(PyExc_TypeError, _("readonly attribute")); | |
2092 return -1; | |
2093 } | |
2094 else if (strcmp(name, "cursor") == 0) | |
2095 { | |
2096 long lnum; | |
2097 long col; | |
1971 | 2098 long len; |
7 | 2099 |
2100 if (!PyArg_Parse(val, "(ll)", &lnum, &col)) | |
2101 return -1; | |
2102 | |
2103 if (lnum <= 0 || lnum > this->win->w_buffer->b_ml.ml_line_count) | |
2104 { | |
2105 PyErr_SetVim(_("cursor position outside buffer")); | |
2106 return -1; | |
2107 } | |
2108 | |
2109 /* Check for keyboard interrupts */ | |
2110 if (VimErrorCheck()) | |
2111 return -1; | |
2112 | |
1971 | 2113 /* When column is out of range silently correct it. */ |
2100
7d121c69f540
updated for version 7.2.383
Bram Moolenaar <bram@zimbu.org>
parents:
1991
diff
changeset
|
2114 len = (long)STRLEN(ml_get_buf(this->win->w_buffer, lnum, FALSE)); |
1971 | 2115 if (col > len) |
2116 col = len; | |
7 | 2117 |
2118 this->win->w_cursor.lnum = lnum; | |
2119 this->win->w_cursor.col = col; | |
1971 | 2120 #ifdef FEAT_VIRTUALEDIT |
2121 this->win->w_cursor.coladd = 0; | |
2122 #endif | |
7 | 2123 update_screen(VALID); |
2124 | |
2125 return 0; | |
2126 } | |
2127 else if (strcmp(name, "height") == 0) | |
2128 { | |
2129 int height; | |
2130 win_T *savewin; | |
2131 | |
2132 if (!PyArg_Parse(val, "i", &height)) | |
2133 return -1; | |
2134 | |
2135 #ifdef FEAT_GUI | |
2136 need_mouse_correct = TRUE; | |
2137 #endif | |
2138 savewin = curwin; | |
2139 curwin = this->win; | |
2140 win_setheight(height); | |
2141 curwin = savewin; | |
2142 | |
2143 /* Check for keyboard interrupts */ | |
2144 if (VimErrorCheck()) | |
2145 return -1; | |
2146 | |
2147 return 0; | |
2148 } | |
2149 #ifdef FEAT_VERTSPLIT | |
2150 else if (strcmp(name, "width") == 0) | |
2151 { | |
2152 int width; | |
2153 win_T *savewin; | |
2154 | |
2155 if (!PyArg_Parse(val, "i", &width)) | |
2156 return -1; | |
2157 | |
2158 #ifdef FEAT_GUI | |
2159 need_mouse_correct = TRUE; | |
2160 #endif | |
2161 savewin = curwin; | |
2162 curwin = this->win; | |
2163 win_setwidth(width); | |
2164 curwin = savewin; | |
2165 | |
2166 /* Check for keyboard interrupts */ | |
2167 if (VimErrorCheck()) | |
2168 return -1; | |
2169 | |
2170 return 0; | |
2171 } | |
2172 #endif | |
2173 else | |
2174 { | |
2175 PyErr_SetString(PyExc_AttributeError, name); | |
2176 return -1; | |
2177 } | |
2178 } | |
2179 | |
2180 static PyObject * | |
2181 WindowRepr(PyObject *self) | |
2182 { | |
274 | 2183 static char repr[100]; |
7 | 2184 WindowObject *this = (WindowObject *)(self); |
2185 | |
2186 if (this->win == INVALID_WINDOW_VALUE) | |
2187 { | |
1607 | 2188 vim_snprintf(repr, 100, _("<window object (deleted) at %p>"), (self)); |
7 | 2189 return PyString_FromString(repr); |
2190 } | |
2191 else | |
2192 { | |
2193 int i = 0; | |
2194 win_T *w; | |
2195 | |
2196 for (w = firstwin; w != NULL && w != this->win; w = W_NEXT(w)) | |
2197 ++i; | |
2198 | |
2199 if (w == NULL) | |
1607 | 2200 vim_snprintf(repr, 100, _("<window object (unknown) at %p>"), |
2201 (self)); | |
7 | 2202 else |
274 | 2203 vim_snprintf(repr, 100, _("<window %d>"), i); |
7 | 2204 |
2205 return PyString_FromString(repr); | |
2206 } | |
2207 } | |
2208 | |
2209 /* Window list object - Definitions | |
2210 */ | |
2211 | |
2212 typedef struct | |
2213 { | |
2214 PyObject_HEAD | |
2215 } | |
2216 WinListObject; | |
2217 | |
2218 static PySequenceMethods WinListAsSeq = { | |
1594 | 2219 (PyInquiry) WinListLength, /* sq_length, len(x) */ |
7 | 2220 (binaryfunc) 0, /* sq_concat, x+y */ |
1594 | 2221 (PyIntArgFunc) 0, /* sq_repeat, x*n */ |
2222 (PyIntArgFunc) WinListItem, /* sq_item, x[i] */ | |
2223 (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */ | |
2224 (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */ | |
2225 (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */ | |
7 | 2226 }; |
2227 | |
2228 static PyTypeObject WinListType = { | |
2229 PyObject_HEAD_INIT(0) | |
2230 0, | |
2231 "window list", | |
2232 sizeof(WinListObject), | |
2233 0, | |
2234 | |
2235 (destructor) 0, /* tp_dealloc, refcount==0 */ | |
2236 (printfunc) 0, /* tp_print, print x */ | |
2237 (getattrfunc) 0, /* tp_getattr, x.attr */ | |
2238 (setattrfunc) 0, /* tp_setattr, x.attr=v */ | |
2239 (cmpfunc) 0, /* tp_compare, x>y */ | |
2240 (reprfunc) 0, /* tp_repr, `x`, print x */ | |
2241 | |
2242 0, /* as number */ | |
2243 &WinListAsSeq, /* as sequence */ | |
2244 0, /* as mapping */ | |
2245 | |
2246 (hashfunc) 0, /* tp_hash, dict(x) */ | |
2247 (ternaryfunc) 0, /* tp_call, x() */ | |
2248 (reprfunc) 0, /* tp_str, str(x) */ | |
2249 }; | |
2250 | |
2251 /* Window list object - Implementation | |
2252 */ | |
1594 | 2253 static PyInt |
1887 | 2254 WinListLength(PyObject *self UNUSED) |
7 | 2255 { |
2256 win_T *w = firstwin; | |
1607 | 2257 PyInt n = 0; |
7 | 2258 |
671 | 2259 while (w != NULL) |
7 | 2260 { |
2261 ++n; | |
2262 w = W_NEXT(w); | |
2263 } | |
2264 | |
2265 return n; | |
2266 } | |
2267 | |
2268 static PyObject * | |
1887 | 2269 WinListItem(PyObject *self UNUSED, PyInt n) |
7 | 2270 { |
2271 win_T *w; | |
2272 | |
671 | 2273 for (w = firstwin; w != NULL; w = W_NEXT(w), --n) |
7 | 2274 if (n == 0) |
2275 return WindowNew(w); | |
2276 | |
2277 PyErr_SetString(PyExc_IndexError, _("no such window")); | |
2278 return NULL; | |
2279 } | |
2280 | |
2281 /* Current items object - Definitions | |
2282 */ | |
2283 | |
2284 typedef struct | |
2285 { | |
2286 PyObject_HEAD | |
2287 } | |
2288 CurrentObject; | |
2289 | |
2290 static PyTypeObject CurrentType = { | |
2291 PyObject_HEAD_INIT(0) | |
2292 0, | |
2293 "current data", | |
2294 sizeof(CurrentObject), | |
2295 0, | |
2296 | |
2297 (destructor) 0, /* tp_dealloc, refcount==0 */ | |
2298 (printfunc) 0, /* tp_print, print x */ | |
2299 (getattrfunc) CurrentGetattr, /* tp_getattr, x.attr */ | |
2300 (setattrfunc) CurrentSetattr, /* tp_setattr, x.attr=v */ | |
2301 (cmpfunc) 0, /* tp_compare, x>y */ | |
2302 (reprfunc) 0, /* tp_repr, `x`, print x */ | |
2303 | |
2304 0, /* as number */ | |
2305 0, /* as sequence */ | |
2306 0, /* as mapping */ | |
2307 | |
2308 (hashfunc) 0, /* tp_hash, dict(x) */ | |
2309 (ternaryfunc) 0, /* tp_call, x() */ | |
2310 (reprfunc) 0, /* tp_str, str(x) */ | |
2311 }; | |
2312 | |
2313 /* Current items object - Implementation | |
2314 */ | |
2315 static PyObject * | |
1887 | 2316 CurrentGetattr(PyObject *self UNUSED, char *name) |
7 | 2317 { |
2318 if (strcmp(name, "buffer") == 0) | |
2319 return (PyObject *)BufferNew(curbuf); | |
2320 else if (strcmp(name, "window") == 0) | |
2321 return (PyObject *)WindowNew(curwin); | |
2322 else if (strcmp(name, "line") == 0) | |
1607 | 2323 return GetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum); |
7 | 2324 else if (strcmp(name, "range") == 0) |
2325 return RangeNew(curbuf, RangeStart, RangeEnd); | |
2326 else if (strcmp(name,"__members__") == 0) | |
2327 return Py_BuildValue("[ssss]", "buffer", "window", "line", "range"); | |
2328 else | |
2329 { | |
2330 PyErr_SetString(PyExc_AttributeError, name); | |
2331 return NULL; | |
2332 } | |
2333 } | |
2334 | |
2335 static int | |
1887 | 2336 CurrentSetattr(PyObject *self UNUSED, char *name, PyObject *value) |
7 | 2337 { |
2338 if (strcmp(name, "line") == 0) | |
2339 { | |
1607 | 2340 if (SetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum, value, NULL) == FAIL) |
7 | 2341 return -1; |
2342 | |
2343 return 0; | |
2344 } | |
2345 else | |
2346 { | |
2347 PyErr_SetString(PyExc_AttributeError, name); | |
2348 return -1; | |
2349 } | |
2350 } | |
2351 | |
2352 /* External interface | |
2353 */ | |
2354 | |
2355 void | |
2356 python_buffer_free(buf_T *buf) | |
2357 { | |
502 | 2358 if (buf->b_python_ref != NULL) |
7 | 2359 { |
502 | 2360 BufferObject *bp = buf->b_python_ref; |
7 | 2361 bp->buf = INVALID_BUFFER_VALUE; |
502 | 2362 buf->b_python_ref = NULL; |
7 | 2363 } |
2364 } | |
2365 | |
2366 #if defined(FEAT_WINDOWS) || defined(PROTO) | |
2367 void | |
2368 python_window_free(win_T *win) | |
2369 { | |
502 | 2370 if (win->w_python_ref != NULL) |
7 | 2371 { |
502 | 2372 WindowObject *wp = win->w_python_ref; |
7 | 2373 wp->win = INVALID_WINDOW_VALUE; |
502 | 2374 win->w_python_ref = NULL; |
7 | 2375 } |
2376 } | |
2377 #endif | |
2378 | |
2379 static BufListObject TheBufferList = | |
2380 { | |
2381 PyObject_HEAD_INIT(&BufListType) | |
2382 }; | |
2383 | |
2384 static WinListObject TheWindowList = | |
2385 { | |
2386 PyObject_HEAD_INIT(&WinListType) | |
2387 }; | |
2388 | |
2389 static CurrentObject TheCurrent = | |
2390 { | |
2391 PyObject_HEAD_INIT(&CurrentType) | |
2392 }; | |
2393 | |
2394 static int | |
2395 PythonMod_Init(void) | |
2396 { | |
2397 PyObject *mod; | |
2398 PyObject *dict; | |
1748 | 2399 /* The special value is removed from sys.path in Python_Init(). */ |
2400 static char *(argv[2]) = {"/must>not&exist/foo", NULL}; | |
7 | 2401 |
2402 /* Fixups... */ | |
2403 BufferType.ob_type = &PyType_Type; | |
2404 RangeType.ob_type = &PyType_Type; | |
2405 WindowType.ob_type = &PyType_Type; | |
2406 BufListType.ob_type = &PyType_Type; | |
2407 WinListType.ob_type = &PyType_Type; | |
2408 CurrentType.ob_type = &PyType_Type; | |
2409 | |
2410 /* Set sys.argv[] to avoid a crash in warn(). */ | |
2411 PySys_SetArgv(1, argv); | |
2412 | |
1607 | 2413 mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION); |
7 | 2414 dict = PyModule_GetDict(mod); |
2415 | |
2416 VimError = Py_BuildValue("s", "vim.error"); | |
2417 | |
2418 PyDict_SetItemString(dict, "error", VimError); | |
135 | 2419 PyDict_SetItemString(dict, "buffers", (PyObject *)(void *)&TheBufferList); |
2420 PyDict_SetItemString(dict, "current", (PyObject *)(void *)&TheCurrent); | |
2421 PyDict_SetItemString(dict, "windows", (PyObject *)(void *)&TheWindowList); | |
7 | 2422 |
2423 if (PyErr_Occurred()) | |
2424 return -1; | |
2425 | |
2426 return 0; | |
2427 } | |
2428 | |
2429 /************************************************************************* | |
2430 * 4. Utility functions for handling the interface between Vim and Python. | |
2431 */ | |
2432 | |
2433 /* Get a line from the specified buffer. The line number is | |
2434 * in Vim format (1-based). The line is returned as a Python | |
2435 * string object. | |
2436 */ | |
2437 static PyObject * | |
1607 | 2438 GetBufferLine(buf_T *buf, PyInt n) |
7 | 2439 { |
2440 return LineToString((char *)ml_get_buf(buf, (linenr_T)n, FALSE)); | |
2441 } | |
2442 | |
2443 /* Get a list of lines from the specified buffer. The line numbers | |
2444 * are in Vim format (1-based). The range is from lo up to, but not | |
2445 * including, hi. The list is returned as a Python list of string objects. | |
2446 */ | |
2447 static PyObject * | |
1594 | 2448 GetBufferLineList(buf_T *buf, PyInt lo, PyInt hi) |
7 | 2449 { |
1594 | 2450 PyInt i; |
2451 PyInt n = hi - lo; | |
7 | 2452 PyObject *list = PyList_New(n); |
2453 | |
2454 if (list == NULL) | |
2455 return NULL; | |
2456 | |
2457 for (i = 0; i < n; ++i) | |
2458 { | |
2459 PyObject *str = LineToString((char *)ml_get_buf(buf, (linenr_T)(lo+i), FALSE)); | |
2460 | |
2461 /* Error check - was the Python string creation OK? */ | |
2462 if (str == NULL) | |
2463 { | |
2464 Py_DECREF(list); | |
2465 return NULL; | |
2466 } | |
2467 | |
2468 /* Set the list item */ | |
2469 if (PyList_SetItem(list, i, str)) | |
2470 { | |
2471 Py_DECREF(str); | |
2472 Py_DECREF(list); | |
2473 return NULL; | |
2474 } | |
2475 } | |
2476 | |
2477 /* The ownership of the Python list is passed to the caller (ie, | |
2478 * the caller should Py_DECREF() the object when it is finished | |
2479 * with it). | |
2480 */ | |
2481 | |
2482 return list; | |
2483 } | |
2484 | |
2485 /* | |
2486 * Check if deleting lines made the cursor position invalid. | |
2487 * Changed the lines from "lo" to "hi" and added "extra" lines (negative if | |
2488 * deleted). | |
2489 */ | |
2490 static void | |
1607 | 2491 py_fix_cursor(linenr_T lo, linenr_T hi, linenr_T extra) |
7 | 2492 { |
2493 if (curwin->w_cursor.lnum >= lo) | |
2494 { | |
2495 /* Adjust the cursor position if it's in/after the changed | |
2496 * lines. */ | |
2497 if (curwin->w_cursor.lnum >= hi) | |
2498 { | |
2499 curwin->w_cursor.lnum += extra; | |
2500 check_cursor_col(); | |
2501 } | |
2502 else if (extra < 0) | |
2503 { | |
2504 curwin->w_cursor.lnum = lo; | |
2505 check_cursor(); | |
2506 } | |
1083 | 2507 else |
2508 check_cursor_col(); | |
7 | 2509 changed_cline_bef_curs(); |
2510 } | |
2511 invalidate_botline(); | |
2512 } | |
2513 | |
2514 /* Replace a line in the specified buffer. The line number is | |
2515 * in Vim format (1-based). The replacement line is given as | |
2516 * a Python string object. The object is checked for validity | |
2517 * and correct format. Errors are returned as a value of FAIL. | |
2518 * The return value is OK on success. | |
2519 * If OK is returned and len_change is not NULL, *len_change | |
2520 * is set to the change in the buffer length. | |
2521 */ | |
2522 static int | |
1607 | 2523 SetBufferLine(buf_T *buf, PyInt n, PyObject *line, PyInt *len_change) |
7 | 2524 { |
2525 /* First of all, we check the thpe of the supplied Python object. | |
2526 * There are three cases: | |
2527 * 1. NULL, or None - this is a deletion. | |
2528 * 2. A string - this is a replacement. | |
2529 * 3. Anything else - this is an error. | |
2530 */ | |
2531 if (line == Py_None || line == NULL) | |
2532 { | |
2533 buf_T *savebuf = curbuf; | |
2534 | |
2535 PyErr_Clear(); | |
2536 curbuf = buf; | |
2537 | |
2538 if (u_savedel((linenr_T)n, 1L) == FAIL) | |
2539 PyErr_SetVim(_("cannot save undo information")); | |
2540 else if (ml_delete((linenr_T)n, FALSE) == FAIL) | |
2541 PyErr_SetVim(_("cannot delete line")); | |
2542 else | |
2543 { | |
2544 if (buf == curwin->w_buffer) | |
1607 | 2545 py_fix_cursor((linenr_T)n, (linenr_T)n + 1, (linenr_T)-1); |
1929 | 2546 deleted_lines_mark((linenr_T)n, 1L); |
7 | 2547 } |
2548 | |
2549 curbuf = savebuf; | |
2550 | |
2551 if (PyErr_Occurred() || VimErrorCheck()) | |
2552 return FAIL; | |
2553 | |
2554 if (len_change) | |
2555 *len_change = -1; | |
2556 | |
2557 return OK; | |
2558 } | |
2559 else if (PyString_Check(line)) | |
2560 { | |
2561 char *save = StringToLine(line); | |
2562 buf_T *savebuf = curbuf; | |
2563 | |
2564 if (save == NULL) | |
2565 return FAIL; | |
2566 | |
2567 /* We do not need to free "save" if ml_replace() consumes it. */ | |
2568 PyErr_Clear(); | |
2569 curbuf = buf; | |
2570 | |
2571 if (u_savesub((linenr_T)n) == FAIL) | |
2572 { | |
2573 PyErr_SetVim(_("cannot save undo information")); | |
2574 vim_free(save); | |
2575 } | |
2576 else if (ml_replace((linenr_T)n, (char_u *)save, FALSE) == FAIL) | |
2577 { | |
2578 PyErr_SetVim(_("cannot replace line")); | |
2579 vim_free(save); | |
2580 } | |
2581 else | |
2582 changed_bytes((linenr_T)n, 0); | |
2583 | |
2584 curbuf = savebuf; | |
2585 | |
1083 | 2586 /* Check that the cursor is not beyond the end of the line now. */ |
2587 if (buf == curwin->w_buffer) | |
2588 check_cursor_col(); | |
2589 | |
7 | 2590 if (PyErr_Occurred() || VimErrorCheck()) |
2591 return FAIL; | |
2592 | |
2593 if (len_change) | |
2594 *len_change = 0; | |
2595 | |
2596 return OK; | |
2597 } | |
2598 else | |
2599 { | |
2600 PyErr_BadArgument(); | |
2601 return FAIL; | |
2602 } | |
2603 } | |
2604 | |
2605 /* Replace a range of lines in the specified buffer. The line numbers are in | |
2606 * Vim format (1-based). The range is from lo up to, but not including, hi. | |
2607 * The replacement lines are given as a Python list of string objects. The | |
2608 * list is checked for validity and correct format. Errors are returned as a | |
2609 * value of FAIL. The return value is OK on success. | |
2610 * If OK is returned and len_change is not NULL, *len_change | |
2611 * is set to the change in the buffer length. | |
2612 */ | |
2613 static int | |
1607 | 2614 SetBufferLineList(buf_T *buf, PyInt lo, PyInt hi, PyObject *list, PyInt *len_change) |
7 | 2615 { |
2616 /* First of all, we check the thpe of the supplied Python object. | |
2617 * There are three cases: | |
2618 * 1. NULL, or None - this is a deletion. | |
2619 * 2. A list - this is a replacement. | |
2620 * 3. Anything else - this is an error. | |
2621 */ | |
2622 if (list == Py_None || list == NULL) | |
2623 { | |
1594 | 2624 PyInt i; |
1607 | 2625 PyInt n = (int)(hi - lo); |
7 | 2626 buf_T *savebuf = curbuf; |
2627 | |
2628 PyErr_Clear(); | |
2629 curbuf = buf; | |
2630 | |
2631 if (u_savedel((linenr_T)lo, (long)n) == FAIL) | |
2632 PyErr_SetVim(_("cannot save undo information")); | |
2633 else | |
2634 { | |
2635 for (i = 0; i < n; ++i) | |
2636 { | |
2637 if (ml_delete((linenr_T)lo, FALSE) == FAIL) | |
2638 { | |
2639 PyErr_SetVim(_("cannot delete line")); | |
2640 break; | |
2641 } | |
2642 } | |
2643 if (buf == curwin->w_buffer) | |
1607 | 2644 py_fix_cursor((linenr_T)lo, (linenr_T)hi, (linenr_T)-n); |
1929 | 2645 deleted_lines_mark((linenr_T)lo, (long)i); |
7 | 2646 } |
2647 | |
2648 curbuf = savebuf; | |
2649 | |
2650 if (PyErr_Occurred() || VimErrorCheck()) | |
2651 return FAIL; | |
2652 | |
2653 if (len_change) | |
2654 *len_change = -n; | |
2655 | |
2656 return OK; | |
2657 } | |
2658 else if (PyList_Check(list)) | |
2659 { | |
1594 | 2660 PyInt i; |
2661 PyInt new_len = PyList_Size(list); | |
2662 PyInt old_len = hi - lo; | |
1607 | 2663 PyInt extra = 0; /* lines added to text, can be negative */ |
7 | 2664 char **array; |
2665 buf_T *savebuf; | |
2666 | |
2667 if (new_len == 0) /* avoid allocating zero bytes */ | |
2668 array = NULL; | |
2669 else | |
2670 { | |
2671 array = (char **)alloc((unsigned)(new_len * sizeof(char *))); | |
2672 if (array == NULL) | |
2673 { | |
2674 PyErr_NoMemory(); | |
2675 return FAIL; | |
2676 } | |
2677 } | |
2678 | |
2679 for (i = 0; i < new_len; ++i) | |
2680 { | |
2681 PyObject *line = PyList_GetItem(list, i); | |
2682 | |
2683 array[i] = StringToLine(line); | |
2684 if (array[i] == NULL) | |
2685 { | |
2686 while (i) | |
2687 vim_free(array[--i]); | |
2688 vim_free(array); | |
2689 return FAIL; | |
2690 } | |
2691 } | |
2692 | |
2693 savebuf = curbuf; | |
2694 | |
2695 PyErr_Clear(); | |
2696 curbuf = buf; | |
2697 | |
2698 if (u_save((linenr_T)(lo-1), (linenr_T)hi) == FAIL) | |
2699 PyErr_SetVim(_("cannot save undo information")); | |
2700 | |
2701 /* If the size of the range is reducing (ie, new_len < old_len) we | |
2702 * need to delete some old_len. We do this at the start, by | |
2703 * repeatedly deleting line "lo". | |
2704 */ | |
2705 if (!PyErr_Occurred()) | |
2706 { | |
2707 for (i = 0; i < old_len - new_len; ++i) | |
2708 if (ml_delete((linenr_T)lo, FALSE) == FAIL) | |
2709 { | |
2710 PyErr_SetVim(_("cannot delete line")); | |
2711 break; | |
2712 } | |
2713 extra -= i; | |
2714 } | |
2715 | |
2716 /* For as long as possible, replace the existing old_len with the | |
2717 * new old_len. This is a more efficient operation, as it requires | |
2718 * less memory allocation and freeing. | |
2719 */ | |
2720 if (!PyErr_Occurred()) | |
2721 { | |
2722 for (i = 0; i < old_len && i < new_len; ++i) | |
2723 if (ml_replace((linenr_T)(lo+i), (char_u *)array[i], FALSE) | |
2724 == FAIL) | |
2725 { | |
2726 PyErr_SetVim(_("cannot replace line")); | |
2727 break; | |
2728 } | |
2729 } | |
2730 else | |
2731 i = 0; | |
2732 | |
2733 /* Now we may need to insert the remaining new old_len. If we do, we | |
2734 * must free the strings as we finish with them (we can't pass the | |
2735 * responsibility to vim in this case). | |
2736 */ | |
2737 if (!PyErr_Occurred()) | |
2738 { | |
2739 while (i < new_len) | |
2740 { | |
2741 if (ml_append((linenr_T)(lo + i - 1), | |
2742 (char_u *)array[i], 0, FALSE) == FAIL) | |
2743 { | |
2744 PyErr_SetVim(_("cannot insert line")); | |
2745 break; | |
2746 } | |
2747 vim_free(array[i]); | |
2748 ++i; | |
2749 ++extra; | |
2750 } | |
2751 } | |
2752 | |
2753 /* Free any left-over old_len, as a result of an error */ | |
2754 while (i < new_len) | |
2755 { | |
2756 vim_free(array[i]); | |
2757 ++i; | |
2758 } | |
2759 | |
2760 /* Free the array of old_len. All of its contents have now | |
2761 * been dealt with (either freed, or the responsibility passed | |
2762 * to vim. | |
2763 */ | |
2764 vim_free(array); | |
2765 | |
2766 /* Adjust marks. Invalidate any which lie in the | |
2767 * changed range, and move any in the remainder of the buffer. | |
2768 */ | |
2769 mark_adjust((linenr_T)lo, (linenr_T)(hi - 1), | |
2770 (long)MAXLNUM, (long)extra); | |
2771 changed_lines((linenr_T)lo, 0, (linenr_T)hi, (long)extra); | |
2772 | |
2773 if (buf == curwin->w_buffer) | |
1607 | 2774 py_fix_cursor((linenr_T)lo, (linenr_T)hi, (linenr_T)extra); |
7 | 2775 |
2776 curbuf = savebuf; | |
2777 | |
2778 if (PyErr_Occurred() || VimErrorCheck()) | |
2779 return FAIL; | |
2780 | |
2781 if (len_change) | |
2782 *len_change = new_len - old_len; | |
2783 | |
2784 return OK; | |
2785 } | |
2786 else | |
2787 { | |
2788 PyErr_BadArgument(); | |
2789 return FAIL; | |
2790 } | |
2791 } | |
2792 | |
2793 /* Insert a number of lines into the specified buffer after the specifed line. | |
2794 * The line number is in Vim format (1-based). The lines to be inserted are | |
2795 * given as a Python list of string objects or as a single string. The lines | |
2796 * to be added are checked for validity and correct format. Errors are | |
2797 * returned as a value of FAIL. The return value is OK on success. | |
2798 * If OK is returned and len_change is not NULL, *len_change | |
2799 * is set to the change in the buffer length. | |
2800 */ | |
2801 static int | |
1607 | 2802 InsertBufferLines(buf_T *buf, PyInt n, PyObject *lines, PyInt *len_change) |
7 | 2803 { |
2804 /* First of all, we check the type of the supplied Python object. | |
2805 * It must be a string or a list, or the call is in error. | |
2806 */ | |
2807 if (PyString_Check(lines)) | |
2808 { | |
2809 char *str = StringToLine(lines); | |
2810 buf_T *savebuf; | |
2811 | |
2812 if (str == NULL) | |
2813 return FAIL; | |
2814 | |
2815 savebuf = curbuf; | |
2816 | |
2817 PyErr_Clear(); | |
2818 curbuf = buf; | |
2819 | |
2820 if (u_save((linenr_T)n, (linenr_T)(n+1)) == FAIL) | |
2821 PyErr_SetVim(_("cannot save undo information")); | |
2822 else if (ml_append((linenr_T)n, (char_u *)str, 0, FALSE) == FAIL) | |
2823 PyErr_SetVim(_("cannot insert line")); | |
2824 else | |
2825 appended_lines_mark((linenr_T)n, 1L); | |
2826 | |
2827 vim_free(str); | |
2828 curbuf = savebuf; | |
2829 update_screen(VALID); | |
2830 | |
2831 if (PyErr_Occurred() || VimErrorCheck()) | |
2832 return FAIL; | |
2833 | |
2834 if (len_change) | |
2835 *len_change = 1; | |
2836 | |
2837 return OK; | |
2838 } | |
2839 else if (PyList_Check(lines)) | |
2840 { | |
1594 | 2841 PyInt i; |
2842 PyInt size = PyList_Size(lines); | |
7 | 2843 char **array; |
2844 buf_T *savebuf; | |
2845 | |
2846 array = (char **)alloc((unsigned)(size * sizeof(char *))); | |
2847 if (array == NULL) | |
2848 { | |
2849 PyErr_NoMemory(); | |
2850 return FAIL; | |
2851 } | |
2852 | |
2853 for (i = 0; i < size; ++i) | |
2854 { | |
2855 PyObject *line = PyList_GetItem(lines, i); | |
2856 array[i] = StringToLine(line); | |
2857 | |
2858 if (array[i] == NULL) | |
2859 { | |
2860 while (i) | |
2861 vim_free(array[--i]); | |
2862 vim_free(array); | |
2863 return FAIL; | |
2864 } | |
2865 } | |
2866 | |
2867 savebuf = curbuf; | |
2868 | |
2869 PyErr_Clear(); | |
2870 curbuf = buf; | |
2871 | |
2872 if (u_save((linenr_T)n, (linenr_T)(n + 1)) == FAIL) | |
2873 PyErr_SetVim(_("cannot save undo information")); | |
2874 else | |
2875 { | |
2876 for (i = 0; i < size; ++i) | |
2877 { | |
2878 if (ml_append((linenr_T)(n + i), | |
2879 (char_u *)array[i], 0, FALSE) == FAIL) | |
2880 { | |
2881 PyErr_SetVim(_("cannot insert line")); | |
2882 | |
2883 /* Free the rest of the lines */ | |
2884 while (i < size) | |
2885 vim_free(array[i++]); | |
2886 | |
2887 break; | |
2888 } | |
2889 vim_free(array[i]); | |
2890 } | |
2891 if (i > 0) | |
2892 appended_lines_mark((linenr_T)n, (long)i); | |
2893 } | |
2894 | |
2895 /* Free the array of lines. All of its contents have now | |
2896 * been freed. | |
2897 */ | |
2898 vim_free(array); | |
2899 | |
2900 curbuf = savebuf; | |
2901 update_screen(VALID); | |
2902 | |
2903 if (PyErr_Occurred() || VimErrorCheck()) | |
2904 return FAIL; | |
2905 | |
2906 if (len_change) | |
2907 *len_change = size; | |
2908 | |
2909 return OK; | |
2910 } | |
2911 else | |
2912 { | |
2913 PyErr_BadArgument(); | |
2914 return FAIL; | |
2915 } | |
2916 } | |
2917 | |
2918 /* Convert a Vim line into a Python string. | |
2919 * All internal newlines are replaced by null characters. | |
2920 * | |
2921 * On errors, the Python exception data is set, and NULL is returned. | |
2922 */ | |
2923 static PyObject * | |
2924 LineToString(const char *str) | |
2925 { | |
2926 PyObject *result; | |
1594 | 2927 PyInt len = strlen(str); |
7 | 2928 char *p; |
2929 | |
2930 /* Allocate an Python string object, with uninitialised contents. We | |
2931 * must do it this way, so that we can modify the string in place | |
2932 * later. See the Python source, Objects/stringobject.c for details. | |
2933 */ | |
2934 result = PyString_FromStringAndSize(NULL, len); | |
2935 if (result == NULL) | |
2936 return NULL; | |
2937 | |
2938 p = PyString_AsString(result); | |
2939 | |
2940 while (*str) | |
2941 { | |
2942 if (*str == '\n') | |
2943 *p = '\0'; | |
2944 else | |
2945 *p = *str; | |
2946 | |
2947 ++p; | |
2948 ++str; | |
2949 } | |
2950 | |
2951 return result; | |
2952 } | |
2953 | |
2954 /* Convert a Python string into a Vim line. | |
2955 * | |
2956 * The result is in allocated memory. All internal nulls are replaced by | |
2957 * newline characters. It is an error for the string to contain newline | |
2958 * characters. | |
2959 * | |
2960 * On errors, the Python exception data is set, and NULL is returned. | |
2961 */ | |
2962 static char * | |
2963 StringToLine(PyObject *obj) | |
2964 { | |
2965 const char *str; | |
2966 char *save; | |
1594 | 2967 PyInt len; |
2968 PyInt i; | |
20 | 2969 char *p; |
7 | 2970 |
2971 if (obj == NULL || !PyString_Check(obj)) | |
2972 { | |
2973 PyErr_BadArgument(); | |
2974 return NULL; | |
2975 } | |
2976 | |
2977 str = PyString_AsString(obj); | |
2978 len = PyString_Size(obj); | |
2979 | |
20 | 2980 /* |
2981 * Error checking: String must not contain newlines, as we | |
7 | 2982 * are replacing a single line, and we must replace it with |
2983 * a single line. | |
20 | 2984 * A trailing newline is removed, so that append(f.readlines()) works. |
7 | 2985 */ |
20 | 2986 p = memchr(str, '\n', len); |
2987 if (p != NULL) | |
7 | 2988 { |
20 | 2989 if (p == str + len - 1) |
2990 --len; | |
2991 else | |
2992 { | |
2993 PyErr_SetVim(_("string cannot contain newlines")); | |
2994 return NULL; | |
2995 } | |
7 | 2996 } |
2997 | |
2998 /* Create a copy of the string, with internal nulls replaced by | |
2999 * newline characters, as is the vim convention. | |
3000 */ | |
3001 save = (char *)alloc((unsigned)(len+1)); | |
3002 if (save == NULL) | |
3003 { | |
3004 PyErr_NoMemory(); | |
3005 return NULL; | |
3006 } | |
3007 | |
3008 for (i = 0; i < len; ++i) | |
3009 { | |
3010 if (str[i] == '\0') | |
3011 save[i] = '\n'; | |
3012 else | |
3013 save[i] = str[i]; | |
3014 } | |
3015 | |
3016 save[i] = '\0'; | |
3017 | |
3018 return save; | |
3019 } | |
3020 | |
3021 /* Check to see whether a Vim error has been reported, or a keyboard | |
3022 * interrupt has been detected. | |
3023 */ | |
3024 static int | |
3025 VimErrorCheck(void) | |
3026 { | |
3027 if (got_int) | |
3028 { | |
3029 PyErr_SetNone(PyExc_KeyboardInterrupt); | |
3030 return 1; | |
3031 } | |
3032 else if (did_emsg && !PyErr_Occurred()) | |
3033 { | |
3034 PyErr_SetNone(VimError); | |
3035 return 1; | |
3036 } | |
3037 | |
3038 return 0; | |
3039 } | |
3040 | |
3041 | |
3042 /* Don't generate a prototype for the next function, it generates an error on | |
3043 * newer Python versions. */ | |
3044 #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO) | |
3045 | |
3046 char * | |
3047 Py_GetProgramName(void) | |
3048 { | |
3049 return "vim"; | |
3050 } | |
3051 #endif /* Python 1.4 */ |