1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// lua.py - Use Lua in Python programs
/* Copyright 2012-2023 Bas Wijnen <wijnen@debian.org> {{{
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
* }}} */
#include "module.h"
// Create new Function from value at top of stack.
PyObject *Function_create(Lua *context) { // {{{
Function *self = (Function *)(function_type->tp_alloc(function_type, 0));
if (!self)
return NULL;
self->lua = (Lua *)context;
Py_INCREF((PyObject *)(self->lua));
// Wrap function at top of lua stack, which must have been pushed before calling this.
self->id = luaL_ref(self->lua->state, LUA_REGISTRYINDEX);
return (PyObject *)self;
}; // }}}
// Destructor.
void Function_dealloc(Function *self) { // {{{
luaL_unref(self->lua->state, LUA_REGISTRYINDEX, self->id);
Py_DECREF(self->lua);
function_type->tp_free((PyObject *)self);
} // }}}
PyObject *Function_call(Function *self, PyObject *args, PyObject *keywords) { // {{{
// Parse keep_single argument. {{{
bool keep_single = false;
if (keywords) {
PyObject *kw = PyDict_GetItemString(keywords, "keep_single"); // Returns borrowed reference.
Py_ssize_t num_keywords = PyDict_Size(keywords);
if (kw) {
if (!PyBool_Check(kw)) {
PyErr_SetString(PyExc_ValueError, "keep_single argument must be of bool type");
return NULL;
}
keep_single = kw == Py_True;
num_keywords -= 1;
}
if (num_keywords > 0) {
PyErr_SetString(PyExc_ValueError, "only keep_single is supported as a keyword argument");
return NULL;
}
}
// }}}
int pos = lua_gettop(self->lua->state);
// Push target function to stack.
lua_rawgeti(self->lua->state, LUA_REGISTRYINDEX, self->id);
// Push arguments to stack. {{{
assert(PyTuple_Check(args));
Py_ssize_t nargs = PyTuple_Size(args);
for (Py_ssize_t a = 0; a < nargs; ++a) {
PyObject *arg = PyTuple_GetItem(args, a); // Borrowed reference.
Lua_push(self->lua, arg);
}
// }}}
// Call function.
lua_call(self->lua->state, nargs, LUA_MULTRET);
// Build return tuple. {{{
Py_ssize_t size = lua_gettop(self->lua->state) - pos;
if (!keep_single && size < 2) {
if (size == 0)
Py_RETURN_NONE;
// Size is 1.
return Lua_to_python(self->lua, -1);
}
PyObject *ret = PyTuple_New(size);
for (int i = 0; i < size; ++i) {
PyObject *value = Lua_to_python(self->lua, -size + i); // New reference.
PyTuple_SET_ITEM(ret, i, value); // Steal reference.
}
// }}}
lua_settop(self->lua->state, pos);
return ret;
} // }}}
PyObject *Function_repr(PyObject *self) { // {{{
char str[100];
snprintf(str, sizeof(str), "<lua.Function@%p>", self);
return PyUnicode_DecodeUTF8(str, strlen(str), NULL);
} // }}}
// }}}
// Python-accessible methods.
//PyMethodDef function_methods[] = { // {{{
// {"__call__", (PyCFunction)Function_call, METH_VARARGS | METH_KEYWORDS, "Call the Lua function"},
// {NULL, NULL, 0, NULL}
//}; // }}}
PyTypeObject *function_type;
// vim: set foldmethod=marker :
|
