*.c: Use Python's facilities for defining class and static methods.
[pyke] / pyke.h
CommitLineData
c1756f78
MW
1/* -*-c-*-
2 *
3 * Pyke: the Python Kit for Extensions
4 *
5 * (c) 2019 Straylight/Edgeware
6 */
7
8/*----- Licensing notice --------------------------------------------------*
9 *
10 * This file is part of Pyke: the Python Kit for Extensions.
11 *
12 * Pyke is free software: you can redistribute it and/or modify it under
13 * the terms of the GNU General Public License as published by the Free
14 * Software Foundation; either version 2 of the License, or (at your
15 * option) any later version.
16 *
17 * Pyke is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
20 * for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with Pyke. If not, write to the Free Software Foundation, Inc.,
24 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
25 */
26
27#ifndef PYKE_H
28#define PYKE_H
29
30#ifdef __cplusplus
31 extern "C" {
32#endif
33
34/*----- Header files ------------------------------------------------------*/
35
36#define PY_SSIZE_T_CLEAN
37
38#include <Python.h>
39#include <structmember.h>
40
41/*----- Other preliminaries -----------------------------------------------*/
42
43#define NOTHING
44#define COMMA ,
45
46/*----- Symbol visibility -------------------------------------------------*
47 *
48 * This library is very messy regarding symbol namespace. Keep this mess
49 * within our shared-object.
50 */
51
52#define GOBBLE_SEMI extern int notexist
53#if defined(__GNUC__) && defined(__ELF__)
54# define PRIVATE_SYMBOLS _Pragma("GCC visibility push(hidden)") GOBBLE_SEMI
55# define PUBLIC_SYMBOLS _Pragma("GCC visibility pop") GOBBLE_SEMI
56# define EXPORT __attribute__((__visibility__("default")))
57#else
58# define PRIVATE_SYMBOLS GOBBLE_SEMI
59# define PUBLIC_SYMBOLS GOBBLE_SEMI
60# define EXPORT
61#endif
62
63PRIVATE_SYMBOLS;
64
65/*----- Utilities for returning values and exceptions ---------------------*/
66
67/* Returning values. */
68#define RETURN_OBJ(obj) do { Py_INCREF(obj); return (obj); } while (0)
69#define RETURN_NONE RETURN_OBJ(Py_None)
70#define RETURN_NOTIMPL RETURN_OBJ(Py_NotImplemented)
71#define RETURN_TRUE RETURN_OBJ(Py_True)
72#define RETURN_FALSE RETURN_OBJ(Py_False)
73#define RETURN_ME RETURN_OBJ(me)
74
75/* Returning exceptions. (Note that `KeyError' is `MAPERR' here, because
76 * Catacomb has its own kind of `KeyError'.)
77 */
78#define EXCERR(exc, str) do { \
79 PyErr_SetString(exc, str); \
80 goto end; \
81} while (0)
82#define VALERR(str) EXCERR(PyExc_ValueError, str)
83#define OVFERR(str) EXCERR(PyExc_OverflowError, str)
84#define TYERR(str) EXCERR(PyExc_TypeError, str)
85#define IXERR(str) EXCERR(PyExc_IndexError, str)
86#define ZDIVERR(str) EXCERR(PyExc_ZeroDivisionError, str)
87#define SYSERR(str) EXCERR(PyExc_SystemError, str)
88#define NIERR(str) EXCERR(PyExc_NotImplementedError, str)
fcfa1c86 89#define MAPERR(idx) do { \
c1756f78
MW
90 PyErr_SetObject(PyExc_KeyError, idx); \
91 goto end; \
92} while (0)
93#define OSERR(name) do { \
94 PyErr_SetFromErrnoWithFilename(PyExc_OSError, name); \
95 goto end; \
96} while (0)
97
98/* Saving and restoring exceptions. */
99struct excinfo { PyObject *ty, *val, *tb; };
100#define EXCINFO_INIT { 0, 0, 0 }
101#define INIT_EXCINFO(exc) do { \
102 struct excinfo *_exc = (exc); _exc->ty = _exc->val = _exc->tb = 0; \
103} while (0)
104#define RELEASE_EXCINFO(exc) do { \
105 struct excinfo *_exc = (exc); \
106 Py_XDECREF(_exc->ty); _exc->ty = 0; \
107 Py_XDECREF(_exc->val); _exc->val = 0; \
108 Py_XDECREF(_exc->tb); _exc->tb = 0; \
109} while (0)
110#define STASH_EXCINFO(exc) do { \
111 struct excinfo *_exc = (exc); \
112 PyErr_Fetch(&_exc->ty, &_exc->val, &_exc->tb); \
113 PyErr_NormalizeException(&_exc->ty, &_exc->val, &_exc->tb); \
114} while (0)
115#define RESTORE_EXCINFO(exc) do { \
116 struct excinfo *_exc = (exc); \
117 PyErr_Restore(_exc->ty, _exc->val, _exc->tb); \
118 _exc->ty = _exc->val = _exc->tb = 0; \
119} while (0)
120extern void report_lost_exception(struct excinfo *, const char *, ...);
121extern void report_lost_exception_v(struct excinfo *, const char *, va_list);
122extern void stash_exception(struct excinfo *, const char *, ...);
123extern void restore_exception(struct excinfo *, const char *, ...);
124
125/*----- Conversions -------------------------------------------------------*/
126
127/* Define an input conversion (`O&') function: check that the object has
128 * Python type TY, and extract a C pointer to CTY by calling EXT on the
129 * object (which may well be a macro).
130 */
131#define CONVFUNC(ty, cty, ext) \
132 int conv##ty(PyObject *o, void *p) \
133 { \
134 if (!PyObject_TypeCheck(o, ty##_pytype)) \
135 TYERR("wanted a " #ty); \
136 *(cty *)p = ext(o); \
137 return (1); \
138 end: \
139 return (0); \
140 }
141
142/* Input conversion functions for standard kinds of objects, with overflow
143 * checking where applicable.
144 */
145extern int convulong(PyObject *, void *); /* unsigned long */
146extern int convuint(PyObject *, void *); /* unsigned int */
147extern int convszt(PyObject *, void *); /* size_t */
148extern int convbool(PyObject *, void *); /* bool */
149
150/* Output conversions. */
151extern PyObject *getbool(int); /* bool */
152extern PyObject *getulong(unsigned long); /* any kind of unsigned integer */
153
154/*----- Miscellaneous utilities -------------------------------------------*/
155
156#define FREEOBJ(obj) \
157 (((PyObject *)(obj))->ob_type->tp_free((PyObject *)(obj)))
158 /* Actually free OBJ, e.g., in a deallocation function. */
159
160extern PyObject *abstract_pynew(PyTypeObject *, PyObject *, PyObject *);
161 /* A `tp_new' function which refuses to make the object. */
162
dce47d50
MW
163#ifndef CONVERT_CAREFULLY
164# define CONVERT_CAREFULLY(newty, expty, obj) \
165 (!sizeof(*(expty *)0 = (obj)) + (/*unconst*/ newty)(obj))
166 /* Convert OBJ to the type NEWTY, having previously checked that it is
167 * convertible to the expected type EXPTY.
168 *
169 * Because of the way we set up types, we can make many kinds of tables be
170 * `const' which can't usually be so (because Python will want to fiddle
171 * with their reference counts); and, besides, Python's internals are
172 * generally quite bad at being `const'-correct about tables. One frequent
173 * application of this macro, then, is in removing `const' from a type
174 * without sacrificing all type safety. The other common use is in
175 * checking that method function types match up with the signatures
176 * expected in their method definitions.
177 */
178#endif
179
180#define KWLIST CONVERT_CAREFULLY(char **, const char *const *, kwlist)
c1756f78
MW
181 /* Strip `const' qualifiers from the keyword list `kwlist'. Useful when
182 * calling `PyArg_ParseTupleAndKeywords', which isn't `const'-correct.
183 */
184
185/*----- Type definitions --------------------------------------------------*
186 *
187 * Pyke types are defined in a rather unusual way.
188 *
189 * The main code defines a `type skeleton' of type `PyTypeObject',
190 * conventionally named `TY_pytype_skel'. Unlike typical Python type
191 * definitions in extensions, this can (and should) be read-only. Also,
192 * there's no point in setting the `tp_base' pointer here, because the actual
193 * runtime base type object won't, in general, be known at compile time.
194 * Instead, the type skeletons are converted into Python `heap types' by the
195 * `INITTYPE' macro. The main difference is that Python code can add
196 * attributes to heap types, and we make extensive use of this ability.
197 */
198
199extern void *newtype(PyTypeObject */*meta*/,
200 const PyTypeObject */*skel*/, const char */*name*/);
201 /* Make and return a new Python type object, of type META (typically
202 * `PyType_Type', but may be a subclass), filled in from the skeleton SKEL
203 * (null to inherit everything), and named NAME. The caller can mess with
204 * the type object further at this time: call `typeready' when it's set up
205 * properly.
206 */
207
208extern void typeready(PyTypeObject *);
209 /* The type object is now ready to be used. */
210
211extern PyTypeObject *inittype(PyTypeObject */*skel*/,
212 PyTypeObject */*meta*/);
213 /* All-in-one function to construct a working type from a type skeleton
214 * SKEL, with metaclass META. The caller is expected to have filled in the
215 * direct superclass in SKEL->tp_base.
216 */
217
218/* Alias for built-in types, to fit in with Pyke naming conventions. */
219#define root_pytype 0
220#define type_pytype &PyType_Type
221
222#define INITTYPE_META(ty, base, meta) do { \
223 ty##_pytype_skel.tp_base = base##_pytype; \
224 ty##_pytype = inittype(&ty##_pytype_skel, meta##_pytype); \
225} while (0)
226#define INITTYPE(ty, base) INITTYPE_META(ty, base, type)
227 /* Macros to initialize a type from its skeleton. */
228
dce47d50
MW
229/* Macros for filling in `PyMethodDef' tables, ensuring that functions have
230 * the expected signatures.
231 */
7bc403cd
MW
232#define STD_METHOD(decor, func, flags, doc) \
233 { #func, decor(func), METH_VARARGS | flags, doc },
234#define KEYWORD_METHOD(decor, func, flags, doc) \
dce47d50
MW
235 { #func, \
236 CONVERT_CAREFULLY(PyCFunction, PyCFunctionWithKeywords, decor(func)), \
7bc403cd 237 METH_VARARGS | METH_KEYWORDS | flags, \
dce47d50
MW
238 doc },
239
c1756f78
MW
240/* Convenience wrappers for filling in `PyMethodDef' tables, following
241 * Pyke naming convention. Define `METHNAME' locally as
242 *
243 * #define METHNAME(name) foometh_##func
244 *
245 * around the method table.
246 */
7bc403cd
MW
247#define METH(func, doc) STD_METHOD(METHNAME, func, 0, doc)
248#define KWMETH(func, doc) KEYWORD_METHOD(METHNAME, func, 0, doc)
249#define CMTH(func, doc) STD_METHOD(METHNAME, func, METH_CLASS, doc)
250#define KWCMTH(func, doc) KEYWORD_METHOD(METHNAME, func, METH_CLASS, doc)
251#define SMTH(func, doc) STD_METHOD(METHNAME, func, METH_STATIC, doc)
252#define KWSMTH(func, doc) KEYWORD_METHOD(METHNAME, func, METH_STATIC, doc)
c1756f78
MW
253
254/* Convenience wrappers for filling in `PyGetSetDef' tables, following Pyke
255 * naming convention. Define `GETSETNAME' locally as
256 *
257 * #define GETSETNAME(op, name) foo##op##_##func
258 *
259 * around the get/set table.
260 */
261#define GET(func, doc) \
262 { #func, GETSETNAME(get, func), 0, doc },
263#define GETSET(func, doc) \
264 { #func, GETSETNAME(get, func), GETSETNAME(set, func), doc },
265
266/* Convenience wrapper for filling in `PyMemberDef' tables. Define
267 * `MEMBERSTRUCT' locally as
268 *
269 * #define MEMBERSTRUCT foo_pyobj
270 *
271 * around the member table.
272 */
273#define MEMBER(name, ty, f, doc) \
274 { #name, ty, offsetof(MEMBERSTRUCT, name), f, doc },
275
87fa2d56
MW
276/* Wrappers for filling in pointers in a `PyTypeObject' structure, (a)
277 * following Pyke naming convention, and (b) stripping `const' from the types
278 * without losing type safety.
279 */
280#define UNCONST_TYPE_SLOT(type, suffix, op, ty) \
281 CONVERT_CAREFULLY(type *, const type *, op ty##_py##suffix)
282#define PYGETSET(ty) UNCONST_TYPE_SLOT(PyGetSetDef, getset, NOTHING, ty)
283#define PYMETHODS(ty) UNCONST_TYPE_SLOT(PyMethodDef, methods, NOTHING, ty)
284#define PYMEMBERS(ty) UNCONST_TYPE_SLOT(PyMemberDef, members, NOTHING, ty)
285#define PYNUMBER(ty) UNCONST_TYPE_SLOT(PyNumberMethods, number, &, ty)
286#define PYSEQUENCE(ty) UNCONST_TYPE_SLOT(PySequenceMethods, sequence, &, ty)
287#define PYMAPPING(ty) UNCONST_TYPE_SLOT(PyMappingMethods, mapping, &, ty)
288#define PYBUFFER(ty) UNCONST_TYPE_SLOT(PyBufferProcs, buffer, &, ty)
289
c1756f78
MW
290/*----- Populating modules ------------------------------------------------*/
291
292extern PyObject *modname;
293 /* The overall module name. Set this with `PyString_FromString'. */
294
295extern PyObject *home_module;
296 /* The overall module object. */
297
298extern PyObject *mkexc(PyObject */*mod*/, PyObject */*base*/,
87fa2d56 299 const char */*name*/, const PyMethodDef */*methods*/);
c1756f78
MW
300 /* Make and return an exception class called NAME, which will end up in
301 * module MOD (though it is not added at this time). The new class is a
302 * subclass of BASE. Attach the METHODS to it.
303 */
304
305#define INSERT(name, ob) do { \
306 PyObject *_o = (PyObject *)(ob); \
307 Py_INCREF(_o); \
308 PyModule_AddObject(mod, name, _o); \
309} while (0)
310 /* Insert a Python object OB into the module `mod' under the given NAME. */
311
312/* Numeric constants. */
313struct nameval { const char *name; unsigned f; unsigned long value; };
314#define CF_SIGNED 1u
315extern void setconstants(PyObject *, const struct nameval *);
316
317#define INSEXC(name, var, base, meth) \
318 INSERT(name, var = mkexc(mod, base, name, meth))
319 /* Insert an exception class into the module `mod'; other arguments are as
320 * for `mkexc'.
321 */
322
323/*----- Submodules --------------------------------------------------------*
324 *
325 * It's useful to split the Python module up into multiple source files, and
326 * have each one contribute its definitions into the main module.
327 *
328 * Define a list-macro `MODULES' in the master header file naming the
329 * submodules to be processed, and run
330 *
331 * MODULES(DECLARE_MODINIT)
332 *
333 * to declare the interface functions.
334 *
335 * Each submodule FOO defines two functions: `FOO_pyinit' initializes types
336 * (see `INITTYPE' above) and accumulates methods (`addmethods' below), while
337 * `FOO_pyinsert' populates the module with additional definitions
338 * (especially types, though also constants).
339 *
340 * The top-level module initialization should call `INIT_MODULES' before
341 * creating the Python module, and `INSERT_MODULES' afterwards to make
342 * everything work.
343 */
344
345extern void addmethods(const PyMethodDef *);
346extern PyMethodDef *donemethods(void);
347 /* Accumulate method-table fragments, and return the combined table of all
348 * of the fragments.
349 */
350
351#define DECLARE_MODINIT(m) \
352 extern void m##_pyinit(void); \
353 extern void m##_pyinsert(PyObject *);
354 /* Declare submodule interface functions. */
355
356#define DOMODINIT(m) m##_pyinit();
357#define DOMODINSERT(m) m##_pyinsert(mod);
358#define INIT_MODULES do { MODULES(DOMODINIT) } while (0)
359#define INSERT_MODULES do { MODULES(DOMODINSERT) } while (0)
360 /* Top-level dispatch to the various submodules. */
361
362/*----- Generic mapping support -------------------------------------------*/
363
364/* Mapping methods. */
dce47d50 365#define GMAP_METMNAME(func) gmapmeth_##func
7bc403cd
MW
366#define GMAP_METH(func, doc) STD_METHOD(GMAP_METMNAME, func, 0, doc)
367#define GMAP_KWMETH(func, doc) KEYWORD_METHOD(GMAP_METMNAME, func, 0, doc)
c1756f78
MW
368#define GMAP_METHDECL(func, doc) \
369 extern PyObject *gmapmeth_##func(PyObject *, PyObject *);
370#define GMAP_KWMETHDECL(func, doc) \
371 extern PyObject *gmapmeth_##func(PyObject *, PyObject *, PyObject *);
372
373#define GMAP_DOROMETHODS(METH, KWMETH) \
374 METH (has_key, "D.has_key(KEY) -> BOOL") \
375 METH (keys, "D.keys() -> LIST") \
376 METH (values, "D.values() -> LIST") \
377 METH (items, "D.items() -> LIST") \
378 METH (iterkeys, "D.iterkeys() -> ITER") \
379 METH (itervalues, "D.itervalues() -> ITER") \
380 METH (iteritems, "D.iteritems() -> ITER") \
381 KWMETH(get, "D.get(KEY, [default = None]) -> VALUE")
382
383#define GMAP_DOMETHODS(METH, KWMETH) \
384 GMAP_DOROMETHODS(METH, KWMETH) \
385 METH (clear, "D.clear()") \
386 KWMETH(setdefault, "D.setdefault(K, [default = None]) -> VALUE") \
387 KWMETH(pop, "D.pop(KEY, [default = <error>]) -> VALUE") \
388 METH (popitem, "D.popitem() -> (KEY, VALUE)") \
389 METH (update, "D.update(MAP)")
390
391GMAP_DOMETHODS(GMAP_METHDECL, GMAP_KWMETHDECL)
392#define GMAP_ROMETHODS GMAP_DOROMETHODS(GMAP_METH, GMAP_KWMETH)
393#define GMAP_METHODS GMAP_DOMETHODS(GMAP_METH, GMAP_KWMETH)
394
395/* Mapping protocol implementation. */
396extern Py_ssize_t gmap_pysize(PyObject *); /* for `mp_length' */
87fa2d56
MW
397extern const PySequenceMethods gmap_pysequence; /* for `tp_as_sequence' */
398extern const PyMethodDef gmap_pymethods[]; /* all the standard methods */
c1756f78
MW
399
400/*----- That's all, folks -------------------------------------------------*/
401
402#ifdef __cplusplus
403 }
404#endif
405
406#endif