blob: a76dc92e811c42be5e6b78ff1384919aac7acd60 [file] [log] [blame]
Martin v. Löwise440e472004-06-01 15:22:42 +00001
2/* Generator object interface */
3
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00004#ifndef Py_LIMITED_API
Martin v. Löwise440e472004-06-01 15:22:42 +00005#ifndef Py_GENOBJECT_H
6#define Py_GENOBJECT_H
7#ifdef __cplusplus
8extern "C" {
9#endif
10
Victor Stinnered488662019-05-20 00:14:57 +020011#include "pystate.h" /* _PyErr_StackItem */
12
Yury Selivanov5376ba92015-06-22 12:19:30 -040013/* _PyGenObject_HEAD defines the initial segment of generator
14 and coroutine objects. */
15#define _PyGenObject_HEAD(prefix) \
16 PyObject_HEAD \
17 /* Note: gi_frame can be NULL if the generator is "finished" */ \
Victor Stinner7c59d7c2020-04-28 16:32:48 +020018 PyFrameObject *prefix##_frame; \
Yury Selivanov5376ba92015-06-22 12:19:30 -040019 /* The code object backing the generator */ \
20 PyObject *prefix##_code; \
21 /* List of weak reference. */ \
22 PyObject *prefix##_weakreflist; \
23 /* Name of the generator. */ \
24 PyObject *prefix##_name; \
25 /* Qualified name of the generator. */ \
Mark Shannonae3087c2017-10-22 22:41:51 +010026 PyObject *prefix##_qualname; \
27 _PyErr_StackItem prefix##_exc_state;
Yury Selivanov5376ba92015-06-22 12:19:30 -040028
Martin v. Löwise440e472004-06-01 15:22:42 +000029typedef struct {
Nick Coghlan76e1bb02012-01-14 16:08:08 +100030 /* The gi_ prefix is intended to remind of generator-iterator. */
Yury Selivanov5376ba92015-06-22 12:19:30 -040031 _PyGenObject_HEAD(gi)
Martin v. Löwise440e472004-06-01 15:22:42 +000032} PyGenObject;
33
34PyAPI_DATA(PyTypeObject) PyGen_Type;
35
36#define PyGen_Check(op) PyObject_TypeCheck(op, &PyGen_Type)
Dong-hee Nad905df72020-02-14 02:37:17 +090037#define PyGen_CheckExact(op) Py_IS_TYPE(op, &PyGen_Type)
Martin v. Löwise440e472004-06-01 15:22:42 +000038
Victor Stinner7c59d7c2020-04-28 16:32:48 +020039PyAPI_FUNC(PyObject *) PyGen_New(PyFrameObject *);
40PyAPI_FUNC(PyObject *) PyGen_NewWithQualName(PyFrameObject *,
Victor Stinner40ee3012014-06-16 15:59:28 +020041 PyObject *name, PyObject *qualname);
Serhiy Storchaka24411f82016-11-06 18:44:42 +020042PyAPI_FUNC(int) _PyGen_SetStopIterationValue(PyObject *);
Nick Coghlanc40bc092012-06-17 15:15:49 +100043PyAPI_FUNC(int) _PyGen_FetchStopIterationValue(PyObject **);
Yury Selivanov833c6262016-10-28 18:48:50 -040044PyAPI_FUNC(PyObject *) _PyGen_Send(PyGenObject *, PyObject *);
Yury Selivanovc724bae2016-03-02 11:30:46 -050045PyObject *_PyGen_yf(PyGenObject *);
Antoine Pitrou58720d62013-08-05 23:26:40 +020046PyAPI_FUNC(void) _PyGen_Finalize(PyObject *self);
47
Yury Selivanov5376ba92015-06-22 12:19:30 -040048#ifndef Py_LIMITED_API
49typedef struct {
50 _PyGenObject_HEAD(cr)
Nathaniel J. Smithfc2f4072018-01-21 06:44:07 -080051 PyObject *cr_origin;
Yury Selivanov5376ba92015-06-22 12:19:30 -040052} PyCoroObject;
53
54PyAPI_DATA(PyTypeObject) PyCoro_Type;
55PyAPI_DATA(PyTypeObject) _PyCoroWrapper_Type;
56
Dong-hee Nad905df72020-02-14 02:37:17 +090057#define PyCoro_CheckExact(op) Py_IS_TYPE(op, &PyCoro_Type)
Yury Selivanov5376ba92015-06-22 12:19:30 -040058PyObject *_PyCoro_GetAwaitableIter(PyObject *o);
Victor Stinner7c59d7c2020-04-28 16:32:48 +020059PyAPI_FUNC(PyObject *) PyCoro_New(PyFrameObject *,
Yury Selivanov5376ba92015-06-22 12:19:30 -040060 PyObject *name, PyObject *qualname);
Yury Selivanoveb636452016-09-08 22:01:51 -070061
62/* Asynchronous Generators */
63
64typedef struct {
65 _PyGenObject_HEAD(ag)
66 PyObject *ag_finalizer;
67
68 /* Flag is set to 1 when hooks set up by sys.set_asyncgen_hooks
69 were called on the generator, to avoid calling them more
70 than once. */
71 int ag_hooks_inited;
72
73 /* Flag is set to 1 when aclose() is called for the first time, or
74 when a StopAsyncIteration exception is raised. */
75 int ag_closed;
Yury Selivanovfc4a0442019-09-29 22:59:11 -070076
77 int ag_running_async;
Yury Selivanoveb636452016-09-08 22:01:51 -070078} PyAsyncGenObject;
79
80PyAPI_DATA(PyTypeObject) PyAsyncGen_Type;
81PyAPI_DATA(PyTypeObject) _PyAsyncGenASend_Type;
82PyAPI_DATA(PyTypeObject) _PyAsyncGenWrappedValue_Type;
83PyAPI_DATA(PyTypeObject) _PyAsyncGenAThrow_Type;
84
Victor Stinner7c59d7c2020-04-28 16:32:48 +020085PyAPI_FUNC(PyObject *) PyAsyncGen_New(PyFrameObject *,
Yury Selivanoveb636452016-09-08 22:01:51 -070086 PyObject *name, PyObject *qualname);
87
Dong-hee Nad905df72020-02-14 02:37:17 +090088#define PyAsyncGen_CheckExact(op) Py_IS_TYPE(op, &PyAsyncGen_Type)
Yury Selivanoveb636452016-09-08 22:01:51 -070089
90PyObject *_PyAsyncGenValueWrapperNew(PyObject *);
91
Yury Selivanov5376ba92015-06-22 12:19:30 -040092#endif
93
94#undef _PyGenObject_HEAD
Martin v. Löwise440e472004-06-01 15:22:42 +000095
96#ifdef __cplusplus
97}
98#endif
99#endif /* !Py_GENOBJECT_H */
Martin v. Löwis4d0d4712010-12-03 20:14:31 +0000100#endif /* Py_LIMITED_API */