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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
// This contains the implementation of the Q_ENUMS and Q_FLAGS support.
//
// Copyright (c) 2016 Riverbank Computing Limited <info@riverbankcomputing.com>
//
// This file is part of PyQt5.
//
// This file may be used under the terms of the GNU General Public License
// version 3.0 as published by the Free Software Foundation and appearing in
// the file LICENSE included in the packaging of this file. Please review the
// following information to ensure the GNU General Public License version 3.0
// requirements will be met: http://www.gnu.org/copyleft/gpl.html.
//
// If you do not wish to use this file under the terms of the GPL version 3.0
// then you may purchase a commercial license. For more information contact
// info@riverbankcomputing.com.
//
// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
// WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
#include <Python.h>
#include <frameobject.h>
#include <QMultiHash>
#include "qpycore_chimera.h"
#include "qpycore_enums_flags.h"
#include "sipAPIQtCore.h"
// Forward declarations.
static PyObject *parse_enums_flags(PyObject *args, bool flags);
// The enums and flags defined in each frame.
static QMultiHash<const PyFrameObject *, EnumsFlags> enums_flags_hash;
// Add the given Q_ENUMS() arguments to the current enums/flags hash.
PyObject *qpycore_Enums(PyObject *args)
{
return parse_enums_flags(args, false);
}
// Add the given Q_FLAGS() arguments to the current enums/flags hash.
PyObject *qpycore_Flags(PyObject *args)
{
return parse_enums_flags(args, true);
}
// Add the given Q_ENUMS() or Q_FLAGS() arguments to the current enums/flags
// hash.
static PyObject *parse_enums_flags(PyObject *args, bool flags)
{
PyFrameObject *frame = PyEval_GetFrame();
// We need the frame we were called from, not the current one.
if (frame)
frame = frame->f_back;
if (!frame)
{
PyErr_SetString(PyExc_RuntimeError, "no current frame");
return 0;
}
// Each argument is a separate enum/flag.
for (Py_ssize_t i = 0; i < PyTuple_GET_SIZE(args); ++i)
{
PyObject *arg = PyTuple_GET_ITEM(args, i);
// Check the argument's type is type.
if (!PyType_Check(arg))
{
PyErr_Format(PyExc_TypeError,
"arguments to %s() must be type objects",
(flags ? "Q_FLAGS" : "Q_ENUMS"));
return 0;
}
// Create the basic enum/flag.
EnumsFlags enums_flags(((PyTypeObject *)arg)->tp_name, flags);
// Go through the type dictionary looking for int attributes.
Py_ssize_t pos = 0;
PyObject *key, *value;
while (PyDict_Next(((PyTypeObject *)arg)->tp_dict, &pos, &key, &value))
{
PyErr_Clear();
int i_value = SIPLong_AsLong(value);
if (PyErr_Occurred())
continue;
const char *s_key = sipString_AsASCIIString(&key);
if (!s_key)
continue;
enums_flags.keys.insert(s_key, i_value);
Py_DECREF(key);
}
enums_flags_hash.insert(frame, enums_flags);
Chimera::registerPyEnum(arg);
}
// Make sure there are no exceptions left after failed value conversions.
PyErr_Clear();
Py_INCREF(Py_None);
return Py_None;
}
// Return the current enums/flags list.
QList<EnumsFlags> qpycore_get_enums_flags_list()
{
PyFrameObject *frame = PyEval_GetFrame();
QList<EnumsFlags> enums_flags_list = enums_flags_hash.values(frame);
enums_flags_hash.remove(frame);
return enums_flags_list;
}
|
