|
@@ -0,0 +1,417 @@
|
|
1
|
+#include "python_rpntoken.h"
|
|
2
|
+
|
|
3
|
+static PyMethodDef RPNToken_methods[] = {
|
|
4
|
+ {"from_str", (PyCFunction)rpntoken_from_str, METH_CLASS | METH_O,
|
|
5
|
+ "Return a new RPNToken subclass instance from string"},
|
|
6
|
+ {NULL} //
|
|
7
|
+};
|
|
8
|
+
|
|
9
|
+PyTypeObject RPNTokenType = {
|
|
10
|
+ PyVarObject_HEAD_INIT(NULL, 0)
|
|
11
|
+ .tp_name = "pyrpn.tokens.Token",
|
|
12
|
+ .tp_doc = "Abstract class for RPN expression tokens",
|
|
13
|
+ .tp_basicsize = sizeof(RPNToken_t),
|
|
14
|
+ .tp_itemsize = 0,
|
|
15
|
+ .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
16
|
+ .tp_init = rpntoken_init,
|
|
17
|
+ .tp_new = PyType_GenericNew,
|
|
18
|
+ .tp_str = rpntoken_str,
|
|
19
|
+ .tp_repr = rpntoken_repr,
|
|
20
|
+ .tp_methods = RPNToken_methods,
|
|
21
|
+};
|
|
22
|
+
|
|
23
|
+static PyMethodDef RPNTokenOp_methods[] = {
|
|
24
|
+ {"opcode_max", (PyCFunction)rpntokenop_opcode_max,
|
|
25
|
+ METH_STATIC | METH_NOARGS,
|
|
26
|
+ "Return the maximum valid value for an opcode"},
|
|
27
|
+ {"chr", (PyCFunction)rpntokenop_opchr, METH_NOARGS,
|
|
28
|
+ "Return the single char representation of the operand"},
|
|
29
|
+ {"str", (PyCFunction)rpntokenop_opstr, METH_NOARGS,
|
|
30
|
+ "Return the string (multi-char) representation of the operand"},
|
|
31
|
+ {NULL} //
|
|
32
|
+};
|
|
33
|
+
|
|
34
|
+static PyMemberDef RPNTokenOp_members[] = {
|
|
35
|
+ {"opcode", T_BYTE, offsetof(RPNTokenOp_t, super.value.op_n), READONLY,
|
|
36
|
+ "The number representing the operand"},
|
|
37
|
+ /*
|
|
38
|
+ {"chr", T_CHAR, offsetof(RPNTokenOp_t, super.value.op->chr), READONLY,
|
|
39
|
+ "The single char representation of the operand"},
|
|
40
|
+ {"str", T_STRING, offsetof(RPNTokenOp_t, super.value.op->str), READONLY,
|
|
41
|
+ "The str representation of the operand"},
|
|
42
|
+ */
|
|
43
|
+ {NULL} //
|
|
44
|
+};
|
|
45
|
+
|
|
46
|
+PyTypeObject RPNTokenOpType = {
|
|
47
|
+ PyVarObject_HEAD_INIT(NULL, 0)
|
|
48
|
+ .tp_base = &RPNTokenType,
|
|
49
|
+ .tp_name = "pyrpn.tokens.Operand",
|
|
50
|
+ .tp_doc = "RPN expression operand token",
|
|
51
|
+ .tp_basicsize = sizeof(RPNTokenOp_t),
|
|
52
|
+ .tp_itemsize = 0,
|
|
53
|
+ .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
54
|
+ .tp_init = rpntokenop_init,
|
|
55
|
+ .tp_members = RPNTokenOp_members,
|
|
56
|
+ .tp_methods = RPNTokenOp_methods,
|
|
57
|
+};
|
|
58
|
+
|
|
59
|
+static PyMemberDef RPNTokenArg_members[] = {
|
|
60
|
+ {"argno", T_ULONG, offsetof(RPNTokenOp_t, super.value.arg_n), READONLY,
|
|
61
|
+ "The argument number"},
|
|
62
|
+ {NULL} //
|
|
63
|
+};
|
|
64
|
+
|
|
65
|
+PyTypeObject RPNTokenArgType = {
|
|
66
|
+ PyVarObject_HEAD_INIT(NULL, 0)
|
|
67
|
+ .tp_base = &RPNTokenType,
|
|
68
|
+ .tp_name = "pyrpn.tokens.Argument",
|
|
69
|
+ .tp_doc = "RPN expression argument token",
|
|
70
|
+ .tp_basicsize = sizeof(RPNTokenArg_t),
|
|
71
|
+ .tp_itemsize = 0,
|
|
72
|
+ .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
73
|
+ .tp_init = rpntokenarg_init,
|
|
74
|
+ .tp_members = RPNTokenArg_members,
|
|
75
|
+};
|
|
76
|
+
|
|
77
|
+static PyMemberDef RPNTokenVal_members[] = {
|
|
78
|
+ {"value", T_ULONG, offsetof(RPNTokenOp_t, super.value.value), READONLY,
|
|
79
|
+ "The immediate value"},
|
|
80
|
+ {NULL} //
|
|
81
|
+};
|
|
82
|
+
|
|
83
|
+PyTypeObject RPNTokenValType = {
|
|
84
|
+ PyVarObject_HEAD_INIT(NULL, 0)
|
|
85
|
+ .tp_base = &RPNTokenType,
|
|
86
|
+ .tp_name = "pyrpn.tokens.Value",
|
|
87
|
+ .tp_doc = "RPN expression value token",
|
|
88
|
+ .tp_basicsize = sizeof(RPNTokenVal_t),
|
|
89
|
+ .tp_itemsize = 0,
|
|
90
|
+ .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
91
|
+ .tp_init = rpntokenval_init,
|
|
92
|
+ .tp_members = RPNTokenVal_members,
|
|
93
|
+};
|
|
94
|
+
|
|
95
|
+PyModuleDef rpntokens_module = {
|
|
96
|
+ PyModuleDef_HEAD_INIT,
|
|
97
|
+ .m_name = "pyrpn.tokens",
|
|
98
|
+ .m_doc = "RPN expression tokens classes",
|
|
99
|
+ .m_size = -1,
|
|
100
|
+ .m_methods = NULL,
|
|
101
|
+ .m_clear = NULL,
|
|
102
|
+};
|
|
103
|
+
|
|
104
|
+
|
|
105
|
+PyObject* rpntokens_module_init(void)
|
|
106
|
+{
|
|
107
|
+ PyObject *mod;
|
|
108
|
+
|
|
109
|
+ mod = PyModule_Create(&rpntokens_module);
|
|
110
|
+ if(!mod) { return NULL; }
|
|
111
|
+
|
|
112
|
+ if(PyType_Ready(&RPNTokenType) < 0
|
|
113
|
+ || PyType_Ready(&RPNTokenOpType) < 0 \
|
|
114
|
+ || PyType_Ready(&RPNTokenValType) < 0 \
|
|
115
|
+ || PyType_Ready(&RPNTokenArgType) < 0)
|
|
116
|
+ {
|
|
117
|
+ return NULL;
|
|
118
|
+ }
|
|
119
|
+ Py_INCREF(&RPNTokenType);
|
|
120
|
+ Py_INCREF(&RPNTokenOpType);
|
|
121
|
+ Py_INCREF(&RPNTokenValType);
|
|
122
|
+ Py_INCREF(&RPNTokenArgType);
|
|
123
|
+
|
|
124
|
+ if( PyModule_AddObject(mod, "Token",
|
|
125
|
+ (PyObject*)&RPNTokenType) < 0 \
|
|
126
|
+ || PyModule_AddObject(mod, "Operand",
|
|
127
|
+ (PyObject*)&RPNTokenOpType) < 0 \
|
|
128
|
+ || PyModule_AddObject(mod, "Value",
|
|
129
|
+ (PyObject*)&RPNTokenValType) < 0 \
|
|
130
|
+ || PyModule_AddObject(mod, "Argument",
|
|
131
|
+ (PyObject*)&RPNTokenArgType) < 0)
|
|
132
|
+ {
|
|
133
|
+ Py_DECREF(&RPNTokenType);
|
|
134
|
+ Py_DECREF(&RPNTokenOpType);
|
|
135
|
+ Py_DECREF(&RPNTokenValType);
|
|
136
|
+ Py_DECREF(&RPNTokenArgType);
|
|
137
|
+ return NULL;
|
|
138
|
+ }
|
|
139
|
+
|
|
140
|
+ return mod;
|
|
141
|
+}
|
|
142
|
+
|
|
143
|
+PyObject* rpntoken_from_str(PyObject *cls, PyObject *arg)
|
|
144
|
+{
|
|
145
|
+ if(!PyUnicode_Check(arg))
|
|
146
|
+ {
|
|
147
|
+ PyErr_SetString(PyExc_TypeError, "Expected argument to be a str");
|
|
148
|
+ return NULL;
|
|
149
|
+ }
|
|
150
|
+ PyObject *bytes_str = PyUnicode_AsASCIIString(arg);
|
|
151
|
+ if(PyErr_Occurred())
|
|
152
|
+ {
|
|
153
|
+ return NULL;
|
|
154
|
+ }
|
|
155
|
+ const char *str = PyBytes_AS_STRING(bytes_str);
|
|
156
|
+ rpn_token_t token;
|
|
157
|
+
|
|
158
|
+ char err_str[64];
|
|
159
|
+ if(rpn_tokenize(str, &token, err_str) < 0)
|
|
160
|
+ {
|
|
161
|
+ Py_DECREF(bytes_str);
|
|
162
|
+ PyObject *ascii = PyObject_ASCII(arg);
|
|
163
|
+ PyErr_Format(PyExc_ValueError, "Unable to parse %s", ascii);
|
|
164
|
+ Py_DECREF(ascii);
|
|
165
|
+ return NULL;
|
|
166
|
+ }
|
|
167
|
+ Py_DECREF(bytes_str);
|
|
168
|
+
|
|
169
|
+ return rpntoken_from_token(&token);
|
|
170
|
+}
|
|
171
|
+
|
|
172
|
+
|
|
173
|
+PyObject* rpntoken_from_token(const rpn_token_t *token)
|
|
174
|
+{
|
|
175
|
+ PyTypeObject *type;
|
|
176
|
+
|
|
177
|
+ switch(token->type)
|
|
178
|
+ {
|
|
179
|
+ case RPN_op:
|
|
180
|
+ type = &RPNTokenOpType;
|
|
181
|
+ break;
|
|
182
|
+ case RPN_val:
|
|
183
|
+ type = &RPNTokenValType;
|
|
184
|
+ break;
|
|
185
|
+ case RPN_arg:
|
|
186
|
+ type = &RPNTokenArgType;
|
|
187
|
+ break;
|
|
188
|
+ default:
|
|
189
|
+ PyErr_SetString(PyExc_RuntimeError,
|
|
190
|
+ "Unrecognized parsed token");
|
|
191
|
+ return NULL;
|
|
192
|
+ }
|
|
193
|
+
|
|
194
|
+ PyObject *ret = PyObject_CallMethod((PyObject*)type, "__new__", "O", type);
|
|
195
|
+ if(PyErr_Occurred())
|
|
196
|
+ {
|
|
197
|
+ return NULL;
|
|
198
|
+ }
|
|
199
|
+ // Any child type should work here since struct begins with super
|
|
200
|
+ RPNToken_t *_ret = (RPNToken_t*)ret;
|
|
201
|
+ memcpy(&_ret->value, token, sizeof(*token));
|
|
202
|
+ return ret;
|
|
203
|
+
|
|
204
|
+}
|
|
205
|
+
|
|
206
|
+
|
|
207
|
+int rpntoken_init(PyObject *_self, PyObject *args, PyObject *kwds)
|
|
208
|
+{
|
|
209
|
+ PyErr_SetString(PyExc_NotImplementedError, "Abstract class");
|
|
210
|
+ return -1;
|
|
211
|
+}
|
|
212
|
+
|
|
213
|
+
|
|
214
|
+PyObject* rpntoken_repr(PyObject *_self)
|
|
215
|
+{
|
|
216
|
+ RPNToken_t *self = (RPNToken_t*)_self;
|
|
217
|
+
|
|
218
|
+
|
|
219
|
+ int needed_sz = rpn_token_snprintf(&self->value, NULL, 0);
|
|
220
|
+ if(needed_sz < 0)
|
|
221
|
+ {
|
|
222
|
+ PyErr_Format(PyExc_RuntimeError, "Error : %s", strerror(errno));
|
|
223
|
+ return NULL;
|
|
224
|
+ }
|
|
225
|
+ char str[needed_sz+1];
|
|
226
|
+ rpn_token_snprintf(&self->value, str, needed_sz+1);
|
|
227
|
+
|
|
228
|
+ PyTypeObject *tp = Py_TYPE(_self);
|
|
229
|
+ PyObject *tp_name = PyType_GetName(tp);
|
|
230
|
+ PyObject *bytes_str = PyUnicode_AsASCIIString(tp_name);
|
|
231
|
+ Py_DECREF(tp_name);
|
|
232
|
+ const char *typename = PyBytes_AS_STRING(bytes_str);
|
|
233
|
+
|
|
234
|
+ needed_sz = snprintf(NULL, 0, "<%s '%s'>", typename, str);
|
|
235
|
+ char res_str[needed_sz+1];
|
|
236
|
+ needed_sz = snprintf(res_str, needed_sz+1, "<%s '%s'>", typename, str);
|
|
237
|
+ Py_DECREF(bytes_str);
|
|
238
|
+
|
|
239
|
+ return PyUnicode_FromString(res_str);
|
|
240
|
+}
|
|
241
|
+
|
|
242
|
+PyObject* rpntoken_str(PyObject *_self)
|
|
243
|
+{
|
|
244
|
+ RPNToken_t *self = (RPNToken_t*)_self;
|
|
245
|
+
|
|
246
|
+ int needed_sz = rpn_token_snprintf(&self->value, NULL, 0);
|
|
247
|
+ if(needed_sz < 0)
|
|
248
|
+ {
|
|
249
|
+ PyErr_Format(PyExc_RuntimeError, "Error : %s", strerror(errno));
|
|
250
|
+ return NULL;
|
|
251
|
+ }
|
|
252
|
+ char str[needed_sz+1];
|
|
253
|
+ rpn_token_snprintf(&self->value, str, needed_sz+1);
|
|
254
|
+ return PyUnicode_FromString(str);
|
|
255
|
+}
|
|
256
|
+
|
|
257
|
+
|
|
258
|
+int rpntokenop_init(PyObject *_self, PyObject *args, PyObject *kwds)
|
|
259
|
+{
|
|
260
|
+ RPNTokenOp_t *self = (RPNTokenOp_t*)_self;
|
|
261
|
+ PyObject *pyop = NULL;
|
|
262
|
+ char *names[] = {"op", NULL};
|
|
263
|
+
|
|
264
|
+
|
|
265
|
+ if(!PyArg_ParseTupleAndKeywords(args, kwds, "O:RPNTokenOP.__init__",
|
|
266
|
+ names,
|
|
267
|
+ &pyop))
|
|
268
|
+ {
|
|
269
|
+ return -1;
|
|
270
|
+ }
|
|
271
|
+
|
|
272
|
+ Py_INCREF(pyop);
|
|
273
|
+
|
|
274
|
+ if(PyLong_Check(pyop))
|
|
275
|
+ {
|
|
276
|
+ //opcode given ?
|
|
277
|
+ long opcode = PyLong_AsLong(pyop);
|
|
278
|
+ if(PyErr_Occurred()) { return -1; }
|
|
279
|
+ if(opcode < 0)
|
|
280
|
+ {
|
|
281
|
+ PyErr_SetString(PyExc_ValueError, "Opcode cannot be negative");
|
|
282
|
+ return -1;
|
|
283
|
+ }
|
|
284
|
+ else if (opcode >= rpn_op_sz())
|
|
285
|
+ {
|
|
286
|
+ PyErr_Format(PyExc_ValueError,
|
|
287
|
+ "Maximum opcode is %ld but %ld given",
|
|
288
|
+ rpn_op_sz()-1, opcode);
|
|
289
|
+ return -1;
|
|
290
|
+ }
|
|
291
|
+ self->super.value.op_n = opcode;
|
|
292
|
+ self->super.value.op = rpn_op_from_opcode(opcode);
|
|
293
|
+ }
|
|
294
|
+ else if(PyUnicode_Check(pyop))
|
|
295
|
+ {
|
|
296
|
+ PyObject *bytes_str = PyUnicode_AsASCIIString(pyop);
|
|
297
|
+ if(PyErr_Occurred())
|
|
298
|
+ {
|
|
299
|
+ return -1;
|
|
300
|
+ }
|
|
301
|
+ const char *token_str = PyBytes_AS_STRING(bytes_str);
|
|
302
|
+ char err_str[64];
|
|
303
|
+
|
|
304
|
+ if(rpn_tokenize(token_str, &(self->super.value), err_str) < 0)
|
|
305
|
+ {
|
|
306
|
+ Py_DECREF(bytes_str);
|
|
307
|
+ PyObject *ascii = PyObject_ASCII(pyop);
|
|
308
|
+ PyErr_Format(PyExc_ValueError, "Unrecognized token '%s' : %s",
|
|
309
|
+ ascii, err_str);
|
|
310
|
+ Py_DECREF(ascii);
|
|
311
|
+ goto err;
|
|
312
|
+ }
|
|
313
|
+ Py_DECREF(bytes_str);
|
|
314
|
+ if(self->super.value.type != RPN_op)
|
|
315
|
+ {
|
|
316
|
+ PyErr_SetString(PyExc_TypeError, "Decoded token is not an operand");
|
|
317
|
+ goto err;
|
|
318
|
+ }
|
|
319
|
+ }
|
|
320
|
+ else
|
|
321
|
+ {
|
|
322
|
+ PyErr_SetString(PyExc_TypeError, "Given argument is neither str neither int");
|
|
323
|
+ goto err;
|
|
324
|
+
|
|
325
|
+ }
|
|
326
|
+
|
|
327
|
+ Py_DECREF(pyop);
|
|
328
|
+ return 0;
|
|
329
|
+
|
|
330
|
+err:
|
|
331
|
+ if(pyop)
|
|
332
|
+ {
|
|
333
|
+ Py_DECREF(pyop);
|
|
334
|
+ }
|
|
335
|
+ return -1;
|
|
336
|
+}
|
|
337
|
+
|
|
338
|
+PyObject *rpntokenop_opcode_max(PyObject* Py_UNUSED(_static))
|
|
339
|
+{
|
|
340
|
+ return PyLong_FromSize_t(rpn_op_sz()-1);
|
|
341
|
+}
|
|
342
|
+
|
|
343
|
+PyObject *rpntokenop_opchr(PyObject *_self, PyObject* Py_UNUSED(_null))
|
|
344
|
+{
|
|
345
|
+ RPNTokenOp_t *self = (RPNTokenOp_t*)_self;
|
|
346
|
+ if(self->super.value.op->chr == '\0')
|
|
347
|
+ {
|
|
348
|
+ Py_RETURN_NONE;
|
|
349
|
+ }
|
|
350
|
+ char buf[2];
|
|
351
|
+ buf[1] = '\0';
|
|
352
|
+ buf[0] = self->super.value.op->chr;
|
|
353
|
+ return PyUnicode_FromString(buf);
|
|
354
|
+}
|
|
355
|
+
|
|
356
|
+
|
|
357
|
+PyObject *rpntokenop_opstr(PyObject *_self, PyObject* Py_UNUSED(_null))
|
|
358
|
+{
|
|
359
|
+ RPNTokenOp_t *self = (RPNTokenOp_t*)_self;
|
|
360
|
+ if(!self->super.value.op->str)
|
|
361
|
+ {
|
|
362
|
+ Py_RETURN_NONE;
|
|
363
|
+ }
|
|
364
|
+ Py_RETURN_NONE;
|
|
365
|
+ return PyUnicode_FromString(self->super.value.op->str);
|
|
366
|
+}
|
|
367
|
+
|
|
368
|
+int rpntokenval_init(PyObject *_self, PyObject *args, PyObject *kwds)
|
|
369
|
+{
|
|
370
|
+ RPNTokenOp_t *self = (RPNTokenOp_t*)_self;
|
|
371
|
+ char *names[] = {"value", NULL};
|
|
372
|
+ PyObject *arg = NULL;
|
|
373
|
+
|
|
374
|
+ if(!PyArg_ParseTupleAndKeywords(args, kwds, "O:RPNTokenVal.__init__",
|
|
375
|
+ names, &arg))
|
|
376
|
+ {
|
|
377
|
+ return -1;
|
|
378
|
+ }
|
|
379
|
+ if(!PyLong_Check(arg))
|
|
380
|
+ {
|
|
381
|
+ PyErr_SetString(PyExc_TypeError, "Expected integer as argument");
|
|
382
|
+ return -1;
|
|
383
|
+ }
|
|
384
|
+ self->super.value.value = PyLong_AsUnsignedLong(arg);
|
|
385
|
+ if(PyErr_Occurred())
|
|
386
|
+ {
|
|
387
|
+ return -1;
|
|
388
|
+ }
|
|
389
|
+ self->super.value.type = RPN_val;
|
|
390
|
+ return 0;
|
|
391
|
+}
|
|
392
|
+
|
|
393
|
+int rpntokenarg_init(PyObject *_self, PyObject *args, PyObject *kwds)
|
|
394
|
+{
|
|
395
|
+ RPNTokenOp_t *self = (RPNTokenOp_t*)_self;
|
|
396
|
+ char *names[] = {"argno", NULL};
|
|
397
|
+ PyObject *arg = NULL;
|
|
398
|
+
|
|
399
|
+ if(!PyArg_ParseTupleAndKeywords(args, kwds, "O:RPNTokenArg.__init__",
|
|
400
|
+ names, &arg))
|
|
401
|
+ {
|
|
402
|
+ return -1;
|
|
403
|
+ }
|
|
404
|
+ if(!PyLong_Check(arg))
|
|
405
|
+ {
|
|
406
|
+ PyErr_SetString(PyExc_TypeError, "Expected integer as argument");
|
|
407
|
+ return -1;
|
|
408
|
+ }
|
|
409
|
+ self->super.value.arg_n = PyLong_AsUnsignedLong(arg);
|
|
410
|
+ if(PyErr_Occurred())
|
|
411
|
+ {
|
|
412
|
+ return -1;
|
|
413
|
+ }
|
|
414
|
+ self->super.value.type = RPN_arg;
|
|
415
|
+ return 0;
|
|
416
|
+}
|
|
417
|
+
|