]> sigrok.org Git - libsigrokdecode.git/blame - type_decoder.c
usb_request: Remove code duplication, save end sample in central location
[libsigrokdecode.git] / type_decoder.c
CommitLineData
d0a0ed03 1/*
50bd5d25 2 * This file is part of the libsigrokdecode project.
d0a0ed03
BV
3 *
4 * Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
5 *
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 */
19
36784362 20#include <config.h>
f6c7eade
MC
21#include "libsigrokdecode-internal.h" /* First, so we avoid a _POSIX_C_SOURCE warning. */
22#include "libsigrokdecode.h"
dbeaab27 23#include <inttypes.h>
d0a0ed03 24
17475b09
UH
25typedef struct {
26 PyObject_HEAD
27} srd_Decoder;
28
201a85a8
DE
29/* This is only used for nicer srd_dbg() output.
30 */
31static const char *output_type_name(unsigned int idx)
32{
33 static const char names[][16] = {
34 "OUTPUT_ANN",
35 "OUTPUT_PYTHON",
36 "OUTPUT_BINARY",
37 "OUTPUT_META",
38 "(invalid)"
39 };
40 return names[MIN(idx, G_N_ELEMENTS(names) - 1)];
41}
58572aed 42
4f75f1c1 43static int convert_annotation(struct srd_decoder_inst *di, PyObject *obj,
0b922460 44 struct srd_proto_data *pdata)
d0a0ed03
BV
45{
46 PyObject *py_tmp;
47 struct srd_pd_output *pdo;
0b922460 48 struct srd_proto_data_annotation *pda;
280d554c 49 int ann_class;
0b922460 50 char **ann_text;
514b2edc
UH
51 PyGILState_STATE gstate;
52
53 gstate = PyGILState_Ensure();
d0a0ed03 54
280d554c 55 /* Should be a list of [annotation class, [string, ...]]. */
0679f5bf 56 if (!PyList_Check(obj)) {
201a85a8 57 srd_err("Protocol decoder %s submitted an annotation that"
0679f5bf 58 " is not a list", di->decoder->name);
514b2edc 59 goto err;
d0a0ed03
BV
60 }
61
361fdcaa 62 /* Should have 2 elements. */
d0a0ed03 63 if (PyList_Size(obj) != 2) {
c9bfccc6 64 srd_err("Protocol decoder %s submitted annotation list with "
4916b173 65 "%zd elements instead of 2", di->decoder->name,
c9bfccc6 66 PyList_Size(obj));
514b2edc 67 goto err;
d0a0ed03
BV
68 }
69
c9bfccc6
UH
70 /*
71 * The first element should be an integer matching a previously
280d554c 72 * registered annotation class.
c9bfccc6 73 */
d0a0ed03
BV
74 py_tmp = PyList_GetItem(obj, 0);
75 if (!PyLong_Check(py_tmp)) {
c9bfccc6
UH
76 srd_err("Protocol decoder %s submitted annotation list, but "
77 "first element was not an integer.", di->decoder->name);
514b2edc 78 goto err;
d0a0ed03 79 }
280d554c
UH
80 ann_class = PyLong_AsLong(py_tmp);
81 if (!(pdo = g_slist_nth_data(di->decoder->annotations, ann_class))) {
d906d3f9 82 srd_err("Protocol decoder %s submitted data to unregistered "
280d554c 83 "annotation class %d.", di->decoder->name, ann_class);
514b2edc 84 goto err;
d0a0ed03 85 }
d0a0ed03 86
361fdcaa 87 /* Second element must be a list. */
d0a0ed03
BV
88 py_tmp = PyList_GetItem(obj, 1);
89 if (!PyList_Check(py_tmp)) {
d906d3f9 90 srd_err("Protocol decoder %s submitted annotation list, but "
c9bfccc6 91 "second element was not a list.", di->decoder->name);
514b2edc 92 goto err;
d0a0ed03 93 }
62a2b15c 94 if (py_strseq_to_char(py_tmp, &ann_text) != SRD_OK) {
d906d3f9 95 srd_err("Protocol decoder %s submitted annotation list, but "
c9bfccc6 96 "second element was malformed.", di->decoder->name);
514b2edc 97 goto err;
d0a0ed03
BV
98 }
99
077fa8ac 100 pda = g_malloc(sizeof(struct srd_proto_data_annotation));
280d554c 101 pda->ann_class = ann_class;
0b922460
BV
102 pda->ann_text = ann_text;
103 pdata->data = pda;
104
514b2edc
UH
105 PyGILState_Release(gstate);
106
d0a0ed03 107 return SRD_OK;
514b2edc
UH
108
109err:
110 PyGILState_Release(gstate);
111
112 return SRD_ERR_PYTHON;
d0a0ed03
BV
113}
114
d75d8a7c
BV
115static int convert_binary(struct srd_decoder_inst *di, PyObject *obj,
116 struct srd_proto_data *pdata)
117{
118 struct srd_proto_data_binary *pdb;
119 PyObject *py_tmp;
120 Py_ssize_t size;
121 int bin_class;
122 char *class_name, *buf;
514b2edc
UH
123 PyGILState_STATE gstate;
124
125 gstate = PyGILState_Ensure();
d75d8a7c 126
2824e811
UH
127 /* Should be a list of [binary class, bytes]. */
128 if (!PyList_Check(obj)) {
129 srd_err("Protocol decoder %s submitted non-list for SRD_OUTPUT_BINARY.",
201a85a8 130 di->decoder->name);
514b2edc 131 goto err;
d75d8a7c
BV
132 }
133
134 /* Should have 2 elements. */
2824e811
UH
135 if (PyList_Size(obj) != 2) {
136 srd_err("Protocol decoder %s submitted SRD_OUTPUT_BINARY list "
4916b173 137 "with %zd elements instead of 2", di->decoder->name,
d75d8a7c 138 PyList_Size(obj));
514b2edc 139 goto err;
d75d8a7c
BV
140 }
141
142 /* The first element should be an integer. */
2824e811 143 py_tmp = PyList_GetItem(obj, 0);
d75d8a7c 144 if (!PyLong_Check(py_tmp)) {
2824e811 145 srd_err("Protocol decoder %s submitted SRD_OUTPUT_BINARY list, "
d75d8a7c 146 "but first element was not an integer.", di->decoder->name);
514b2edc 147 goto err;
d75d8a7c
BV
148 }
149 bin_class = PyLong_AsLong(py_tmp);
150 if (!(class_name = g_slist_nth_data(di->decoder->binary, bin_class))) {
151 srd_err("Protocol decoder %s submitted SRD_OUTPUT_BINARY with "
152 "unregistered binary class %d.", di->decoder->name, bin_class);
514b2edc 153 goto err;
d75d8a7c
BV
154 }
155
156 /* Second element should be bytes. */
2824e811 157 py_tmp = PyList_GetItem(obj, 1);
d75d8a7c 158 if (!PyBytes_Check(py_tmp)) {
2824e811 159 srd_err("Protocol decoder %s submitted SRD_OUTPUT_BINARY list, "
d75d8a7c 160 "but second element was not bytes.", di->decoder->name);
514b2edc 161 goto err;
d75d8a7c
BV
162 }
163
164 /* Consider an empty set of bytes a bug. */
165 if (PyBytes_Size(py_tmp) == 0) {
166 srd_err("Protocol decoder %s submitted SRD_OUTPUT_BINARY "
167 "with empty data set.", di->decoder->name);
514b2edc 168 goto err;
d75d8a7c
BV
169 }
170
077fa8ac 171 pdb = g_malloc(sizeof(struct srd_proto_data_binary));
d75d8a7c 172 if (PyBytes_AsStringAndSize(py_tmp, &buf, &size) == -1)
514b2edc
UH
173 goto err;
174
175 PyGILState_Release(gstate);
176
d75d8a7c
BV
177 pdb->bin_class = bin_class;
178 pdb->size = size;
179 if (!(pdb->data = g_try_malloc(pdb->size)))
180 return SRD_ERR_MALLOC;
181 memcpy((void *)pdb->data, (const void *)buf, pdb->size);
182 pdata->data = pdb;
183
184 return SRD_OK;
514b2edc
UH
185
186err:
187 PyGILState_Release(gstate);
188
189 return SRD_ERR_PYTHON;
d75d8a7c
BV
190}
191
7ee0c40b
BV
192static int convert_meta(struct srd_proto_data *pdata, PyObject *obj)
193{
194 long long intvalue;
195 double dvalue;
514b2edc
UH
196 PyGILState_STATE gstate;
197
198 gstate = PyGILState_Ensure();
7ee0c40b
BV
199
200 if (pdata->pdo->meta_type == G_VARIANT_TYPE_INT64) {
201 if (!PyLong_Check(obj)) {
202 PyErr_Format(PyExc_TypeError, "This output was registered "
201a85a8 203 "as 'int', but something else was passed.");
514b2edc 204 goto err;
7ee0c40b
BV
205 }
206 intvalue = PyLong_AsLongLong(obj);
207 if (PyErr_Occurred())
514b2edc 208 goto err;
7ee0c40b
BV
209 pdata->data = g_variant_new_int64(intvalue);
210 } else if (pdata->pdo->meta_type == G_VARIANT_TYPE_DOUBLE) {
211 if (!PyFloat_Check(obj)) {
212 PyErr_Format(PyExc_TypeError, "This output was registered "
201a85a8 213 "as 'float', but something else was passed.");
514b2edc 214 goto err;
7ee0c40b
BV
215 }
216 dvalue = PyFloat_AsDouble(obj);
217 if (PyErr_Occurred())
514b2edc 218 goto err;
7ee0c40b
BV
219 pdata->data = g_variant_new_double(dvalue);
220 }
221
514b2edc
UH
222 PyGILState_Release(gstate);
223
7ee0c40b 224 return SRD_OK;
514b2edc
UH
225
226err:
227 PyGILState_Release(gstate);
228
229 return SRD_ERR_PYTHON;
7ee0c40b
BV
230}
231
d0a0ed03
BV
232static PyObject *Decoder_put(PyObject *self, PyObject *args)
233{
234 GSList *l;
4f75f1c1 235 PyObject *py_data, *py_res;
a8b72b05 236 struct srd_decoder_inst *di, *next_di;
d0a0ed03 237 struct srd_pd_output *pdo;
8a9f60b1 238 struct srd_proto_data pdata;
d0a0ed03
BV
239 uint64_t start_sample, end_sample;
240 int output_id;
2994587f 241 struct srd_pd_callback *cb;
514b2edc
UH
242 PyGILState_STATE gstate;
243
244 gstate = PyGILState_Ensure();
d0a0ed03 245
a8b72b05 246 if (!(di = srd_inst_find_by_obj(NULL, self))) {
58572aed 247 /* Shouldn't happen. */
7a1712c4 248 srd_dbg("put(): self instance not found.");
514b2edc 249 goto err;
58572aed 250 }
d0a0ed03 251
c9bfccc6 252 if (!PyArg_ParseTuple(args, "KKiO", &start_sample, &end_sample,
3d14e7c9 253 &output_id, &py_data)) {
c9bfccc6
UH
254 /*
255 * This throws an exception, but by returning NULL here we let
256 * Python raise it. This results in a much better trace in
257 * controller.c on the decode() method call.
258 */
514b2edc 259 goto err;
c9bfccc6 260 }
d0a0ed03
BV
261
262 if (!(l = g_slist_nth(di->pd_output, output_id))) {
d906d3f9 263 srd_err("Protocol decoder %s submitted invalid output ID %d.",
c9bfccc6 264 di->decoder->name, output_id);
514b2edc 265 goto err;
d0a0ed03
BV
266 }
267 pdo = l->data;
268
1c2b0d0b 269 srd_spew("Instance %s put %" PRIu64 "-%" PRIu64 " %s on oid %d.",
a8b72b05 270 di->inst_id, start_sample, end_sample,
201a85a8 271 output_type_name(pdo->output_type), output_id);
58572aed 272
8a9f60b1
UH
273 pdata.start_sample = start_sample;
274 pdata.end_sample = end_sample;
275 pdata.pdo = pdo;
276 pdata.data = NULL;
d0a0ed03
BV
277
278 switch (pdo->output_type) {
279 case SRD_OUTPUT_ANN:
280 /* Annotations are only fed to callbacks. */
32cfb920 281 if ((cb = srd_pd_output_callback_find(di->sess, pdo->output_type))) {
d75d8a7c 282 /* Convert from PyDict to srd_proto_data_annotation. */
8a9f60b1 283 if (convert_annotation(di, py_data, &pdata) != SRD_OK) {
d0a0ed03
BV
284 /* An error was already logged. */
285 break;
286 }
514b2edc 287 Py_BEGIN_ALLOW_THREADS
8a9f60b1 288 cb->cb(&pdata, cb->cb_data);
514b2edc 289 Py_END_ALLOW_THREADS
d0a0ed03
BV
290 }
291 break;
f2a5df42 292 case SRD_OUTPUT_PYTHON:
d0a0ed03
BV
293 for (l = di->next_di; l; l = l->next) {
294 next_di = l->data;
4916b173 295 srd_spew("Sending %" PRIu64 "-%" PRIu64 " to instance %s",
3d14e7c9 296 start_sample, end_sample, next_di->inst_id);
c9bfccc6 297 if (!(py_res = PyObject_CallMethod(
3d14e7c9
BV
298 next_di->py_inst, "decode", "KKO", start_sample,
299 end_sample, py_data))) {
201a85a8 300 srd_exception_catch("Calling %s decode() failed",
3d14e7c9 301 next_di->inst_id);
d0a0ed03
BV
302 }
303 Py_XDECREF(py_res);
304 }
3d14e7c9
BV
305 if ((cb = srd_pd_output_callback_find(di->sess, pdo->output_type))) {
306 /* Frontends aren't really supposed to get Python
307 * callbacks, but it's useful for testing. */
8a9f60b1
UH
308 pdata.data = py_data;
309 cb->cb(&pdata, cb->cb_data);
3d14e7c9 310 }
d0a0ed03
BV
311 break;
312 case SRD_OUTPUT_BINARY:
d75d8a7c
BV
313 if ((cb = srd_pd_output_callback_find(di->sess, pdo->output_type))) {
314 /* Convert from PyDict to srd_proto_data_binary. */
8a9f60b1 315 if (convert_binary(di, py_data, &pdata) != SRD_OK) {
d75d8a7c
BV
316 /* An error was already logged. */
317 break;
318 }
514b2edc 319 Py_BEGIN_ALLOW_THREADS
8a9f60b1 320 cb->cb(&pdata, cb->cb_data);
514b2edc 321 Py_END_ALLOW_THREADS
d75d8a7c 322 }
d0a0ed03 323 break;
7ee0c40b
BV
324 case SRD_OUTPUT_META:
325 if ((cb = srd_pd_output_callback_find(di->sess, pdo->output_type))) {
326 /* Annotations need converting from PyObject. */
8a9f60b1 327 if (convert_meta(&pdata, py_data) != SRD_OK) {
7ee0c40b
BV
328 /* An exception was already set up. */
329 break;
330 }
514b2edc 331 Py_BEGIN_ALLOW_THREADS
8a9f60b1 332 cb->cb(&pdata, cb->cb_data);
514b2edc 333 Py_END_ALLOW_THREADS
7ee0c40b
BV
334 }
335 break;
d0a0ed03 336 default:
d906d3f9 337 srd_err("Protocol decoder %s submitted invalid output type %d.",
c9bfccc6 338 di->decoder->name, pdo->output_type);
d0a0ed03
BV
339 break;
340 }
341
514b2edc
UH
342 PyGILState_Release(gstate);
343
d0a0ed03 344 Py_RETURN_NONE;
514b2edc
UH
345
346err:
347 PyGILState_Release(gstate);
348
349 return NULL;
d0a0ed03
BV
350}
351
7ee0c40b
BV
352static PyObject *Decoder_register(PyObject *self, PyObject *args,
353 PyObject *kwargs)
d0a0ed03 354{
a8b72b05 355 struct srd_decoder_inst *di;
7ee0c40b
BV
356 struct srd_pd_output *pdo;
357 PyObject *py_new_output_id;
358 PyTypeObject *meta_type_py;
359 const GVariantType *meta_type_gv;
360 int output_type;
361 char *proto_id, *meta_name, *meta_descr;
362 char *keywords[] = {"output_type", "proto_id", "meta", NULL};
514b2edc
UH
363 PyGILState_STATE gstate;
364
365 gstate = PyGILState_Ensure();
7ee0c40b
BV
366
367 meta_type_py = NULL;
368 meta_type_gv = NULL;
369 meta_name = meta_descr = NULL;
d0a0ed03 370
a8b72b05 371 if (!(di = srd_inst_find_by_obj(NULL, self))) {
d0a0ed03 372 PyErr_SetString(PyExc_Exception, "decoder instance not found");
514b2edc 373 goto err;
d0a0ed03
BV
374 }
375
7ee0c40b
BV
376 /* Default to instance id, which defaults to class id. */
377 proto_id = di->inst_id;
378 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|s(Oss)", keywords,
379 &output_type, &proto_id,
380 &meta_type_py, &meta_name, &meta_descr)) {
511e2123 381 /* Let Python raise this exception. */
514b2edc 382 goto err;
d0a0ed03
BV
383 }
384
7ee0c40b
BV
385 /* Check if the meta value's type is supported. */
386 if (output_type == SRD_OUTPUT_META) {
387 if (meta_type_py == &PyLong_Type)
388 meta_type_gv = G_VARIANT_TYPE_INT64;
389 else if (meta_type_py == &PyFloat_Type)
390 meta_type_gv = G_VARIANT_TYPE_DOUBLE;
391 else {
201a85a8 392 PyErr_Format(PyExc_TypeError, "Unsupported type.");
514b2edc 393 goto err;
7ee0c40b
BV
394 }
395 }
396
397 srd_dbg("Instance %s creating new output type %d for %s.",
398 di->inst_id, output_type, proto_id);
399
077fa8ac 400 pdo = g_malloc(sizeof(struct srd_pd_output));
7ee0c40b
BV
401
402 /* pdo_id is just a simple index, nothing is deleted from this list anyway. */
403 pdo->pdo_id = g_slist_length(di->pd_output);
404 pdo->output_type = output_type;
405 pdo->di = di;
406 pdo->proto_id = g_strdup(proto_id);
407
408 if (output_type == SRD_OUTPUT_META) {
409 pdo->meta_type = meta_type_gv;
410 pdo->meta_name = g_strdup(meta_name);
411 pdo->meta_descr = g_strdup(meta_descr);
412 }
413
414 di->pd_output = g_slist_append(di->pd_output, pdo);
415 py_new_output_id = Py_BuildValue("i", pdo->pdo_id);
416
514b2edc
UH
417 PyGILState_Release(gstate);
418
7ee0c40b 419 return py_new_output_id;
514b2edc
UH
420
421err:
422 PyGILState_Release(gstate);
423
424 return NULL;
7ee0c40b
BV
425}
426
21dfd91d
UH
427static int get_term_type(const char *v)
428{
429 switch (v[0]) {
430 case 'h':
431 return SRD_TERM_HIGH;
432 case 'l':
433 return SRD_TERM_LOW;
434 case 'r':
435 return SRD_TERM_RISING_EDGE;
436 case 'f':
437 return SRD_TERM_FALLING_EDGE;
438 case 'e':
439 return SRD_TERM_EITHER_EDGE;
440 case 'n':
441 return SRD_TERM_NO_EDGE;
442 }
443
444 return -1;
445}
446
447/**
448 * Get the pin values at the current sample number.
449 *
450 * @param di The decoder instance to use. Must not be NULL.
451 * The number of channels must be >= 1.
452 *
453 * @return A newly allocated PyTuple containing the pin values at the
454 * current sample number.
455 */
456static PyObject *get_current_pinvalues(const struct srd_decoder_inst *di)
457{
458 int i;
459 uint8_t sample;
460 const uint8_t *sample_pos;
461 int byte_offset, bit_offset;
462 PyObject *py_pinvalues;
514b2edc
UH
463 PyGILState_STATE gstate;
464
465 gstate = PyGILState_Ensure();
21dfd91d
UH
466
467 if (!di) {
468 srd_err("Invalid decoder instance.");
514b2edc 469 PyGILState_Release(gstate);
21dfd91d
UH
470 return NULL;
471 }
472
473 py_pinvalues = PyTuple_New(di->dec_num_channels);
474
475 for (i = 0; i < di->dec_num_channels; i++) {
476 /* A channelmap value of -1 means "unused optional channel". */
477 if (di->dec_channelmap[i] == -1) {
478 /* Value of unused channel is 0xff, instead of 0 or 1. */
479 PyTuple_SetItem(py_pinvalues, i, PyLong_FromLong(0xff));
480 } else {
4564e8e5 481 sample_pos = di->inbuf + ((di->abs_cur_samplenum - di->abs_start_samplenum) * di->data_unitsize);
21dfd91d
UH
482 byte_offset = di->dec_channelmap[i] / 8;
483 bit_offset = di->dec_channelmap[i] % 8;
484 sample = *(sample_pos + byte_offset) & (1 << bit_offset) ? 1 : 0;
485 PyTuple_SetItem(py_pinvalues, i, PyLong_FromLong(sample));
486 }
487 }
488
514b2edc
UH
489 PyGILState_Release(gstate);
490
21dfd91d
UH
491 return py_pinvalues;
492}
493
494/**
495 * Create a list of terms in the specified condition.
496 *
497 * If there are no terms in the condition, 'term_list' will be NULL.
498 *
499 * @param py_dict A Python dict containing terms. Must not be NULL.
500 * @param term_list Pointer to a GSList which will be set to the newly
501 * created list of terms. Must not be NULL.
502 *
503 * @return SRD_OK upon success, a negative error code otherwise.
504 */
505static int create_term_list(PyObject *py_dict, GSList **term_list)
506{
507 Py_ssize_t pos = 0;
508 PyObject *py_key, *py_value;
509 struct srd_term *term;
510 uint64_t num_samples_to_skip;
511 char *term_str;
514b2edc 512 PyGILState_STATE gstate;
21dfd91d
UH
513
514 if (!py_dict || !term_list)
515 return SRD_ERR_ARG;
516
517 /* "Create" an empty GSList of terms. */
518 *term_list = NULL;
519
514b2edc
UH
520 gstate = PyGILState_Ensure();
521
21dfd91d
UH
522 /* Iterate over all items in the current dict. */
523 while (PyDict_Next(py_dict, &pos, &py_key, &py_value)) {
524 /* Check whether the current key is a string or a number. */
525 if (PyLong_Check(py_key)) {
526 /* The key is a number. */
527 /* TODO: Check if the number is a valid channel. */
528 /* Get the value string. */
529 if ((py_pydictitem_as_str(py_dict, py_key, &term_str)) != SRD_OK) {
530 srd_err("Failed to get the value.");
514b2edc 531 goto err;
21dfd91d
UH
532 }
533 term = g_malloc0(sizeof(struct srd_term));
534 term->type = get_term_type(term_str);
535 term->channel = PyLong_AsLong(py_key);
536 g_free(term_str);
537 } else if (PyUnicode_Check(py_key)) {
538 /* The key is a string. */
539 /* TODO: Check if it's "skip". */
540 if ((py_pydictitem_as_long(py_dict, py_key, &num_samples_to_skip)) != SRD_OK) {
541 srd_err("Failed to get number of samples to skip.");
514b2edc 542 goto err;
21dfd91d
UH
543 }
544 term = g_malloc0(sizeof(struct srd_term));
545 term->type = SRD_TERM_SKIP;
546 term->num_samples_to_skip = num_samples_to_skip;
547 term->num_samples_already_skipped = 0;
548 } else {
549 srd_err("Term key is neither a string nor a number.");
514b2edc 550 goto err;
21dfd91d
UH
551 }
552
553 /* Add the term to the list of terms. */
554 *term_list = g_slist_append(*term_list, term);
555 }
556
514b2edc
UH
557 PyGILState_Release(gstate);
558
21dfd91d 559 return SRD_OK;
514b2edc
UH
560
561err:
562 PyGILState_Release(gstate);
563
564 return SRD_ERR;
21dfd91d
UH
565}
566
567/**
568 * Replace the current condition list with the new one.
569 *
570 * @param self TODO. Must not be NULL.
571 * @param args TODO. Must not be NULL.
572 *
573 * @retval SRD_OK The new condition list was set successfully.
574 * @retval SRD_ERR There was an error setting the new condition list.
575 * The contents of di->condition_list are undefined.
576 * @retval 9999 TODO.
577 */
578static int set_new_condition_list(PyObject *self, PyObject *args)
579{
580 struct srd_decoder_inst *di;
581 GSList *term_list;
582 PyObject *py_conditionlist, *py_conds, *py_dict;
583 int i, num_conditions, ret;
514b2edc 584 PyGILState_STATE gstate;
21dfd91d
UH
585
586 if (!self || !args)
587 return SRD_ERR_ARG;
588
514b2edc
UH
589 gstate = PyGILState_Ensure();
590
21dfd91d
UH
591 /* Get the decoder instance. */
592 if (!(di = srd_inst_find_by_obj(NULL, self))) {
593 PyErr_SetString(PyExc_Exception, "decoder instance not found");
514b2edc 594 goto err;
21dfd91d
UH
595 }
596
04383ea8
GS
597 /*
598 * Return an error condition from .wait() when termination is
599 * requested, such that decode() will terminate.
600 */
601 if (di->want_wait_terminate) {
602 srd_dbg("%s: %s: Skip (want_term).", di->inst_id, __func__);
514b2edc 603 goto err;
04383ea8
GS
604 }
605
6e19f325
GS
606 /*
607 * Parse the argument of self.wait() into 'py_conds', and check
608 * the data type. The argument is optional, None is assumed in
609 * its absence. None or an empty dict or an empty list mean that
610 * there is no condition, and the next available sample shall
611 * get returned to the caller.
612 */
613 py_conds = Py_None;
614 if (!PyArg_ParseTuple(args, "|O", &py_conds)) {
21dfd91d 615 /* Let Python raise this exception. */
514b2edc 616 goto err;
21dfd91d 617 }
6e19f325
GS
618 if (py_conds == Py_None) {
619 /* 'py_conds' is None. */
514b2edc 620 goto ret_9999;
6e19f325 621 } else if (PyList_Check(py_conds)) {
21dfd91d
UH
622 /* 'py_conds' is a list. */
623 py_conditionlist = py_conds;
624 num_conditions = PyList_Size(py_conditionlist);
625 if (num_conditions == 0)
514b2edc 626 goto ret_9999; /* The PD invoked self.wait([]). */
066fbafd 627 Py_IncRef(py_conditionlist);
21dfd91d
UH
628 } else if (PyDict_Check(py_conds)) {
629 /* 'py_conds' is a dict. */
630 if (PyDict_Size(py_conds) == 0)
514b2edc 631 goto ret_9999; /* The PD invoked self.wait({}). */
21dfd91d
UH
632 /* Make a list and put the dict in there for convenience. */
633 py_conditionlist = PyList_New(1);
01f15bdf 634 Py_IncRef(py_conds);
21dfd91d
UH
635 PyList_SetItem(py_conditionlist, 0, py_conds);
636 num_conditions = 1;
637 } else {
638 srd_err("Condition list is neither a list nor a dict.");
514b2edc 639 goto err;
21dfd91d
UH
640 }
641
642 /* Free the old condition list. */
643 condition_list_free(di);
644
645 ret = SRD_OK;
646
647 /* Iterate over the conditions, set di->condition_list accordingly. */
648 for (i = 0; i < num_conditions; i++) {
649 /* Get a condition (dict) from the condition list. */
650 py_dict = PyList_GetItem(py_conditionlist, i);
651 if (!PyDict_Check(py_dict)) {
652 srd_err("Condition is not a dict.");
653 ret = SRD_ERR;
654 break;
655 }
656
657 /* Create the list of terms in this condition. */
658 if ((ret = create_term_list(py_dict, &term_list)) < 0)
659 break;
660
661 /* Add the new condition to the PD instance's condition list. */
662 di->condition_list = g_slist_append(di->condition_list, term_list);
663 }
664
665 Py_DecRef(py_conditionlist);
666
514b2edc
UH
667 PyGILState_Release(gstate);
668
21dfd91d 669 return ret;
514b2edc
UH
670
671err:
672 PyGILState_Release(gstate);
673
674 return SRD_ERR;
675
676ret_9999:
677 PyGILState_Release(gstate);
678
679 return 9999;
21dfd91d
UH
680}
681
96e1cee7
GS
682/**
683 * Create a SKIP condition list for condition-less .wait() calls.
684 *
685 * @param di Decoder instance.
686 * @param count Number of samples to skip.
687 *
688 * @retval SRD_OK The new condition list was set successfully.
689 * @retval SRD_ERR There was an error setting the new condition list.
690 * The contents of di->condition_list are undefined.
691 *
692 * This routine is a reduced and specialized version of the @ref
693 * set_new_condition_list() and @ref create_term_list() routines which
694 * gets invoked when .wait() was called without specifications for
695 * conditions. This minor duplication of the SKIP term list creation
696 * simplifies the logic and avoids the creation of expensive Python
697 * objects with "constant" values which the caller did not pass in the
698 * first place. It results in maximum sharing of match handling code
699 * paths.
700 */
701static int set_skip_condition(struct srd_decoder_inst *di, uint64_t count)
702{
703 struct srd_term *term;
704 GSList *term_list;
705
706 condition_list_free(di);
707 term = g_malloc0(sizeof(*term));
708 term->type = SRD_TERM_SKIP;
709 term->num_samples_to_skip = count;
710 term->num_samples_already_skipped = 0;
711 term_list = g_slist_append(NULL, term);
712 di->condition_list = g_slist_append(di->condition_list, term_list);
713
714 return SRD_OK;
715}
716
21dfd91d
UH
717static PyObject *Decoder_wait(PyObject *self, PyObject *args)
718{
719 int ret;
96e1cee7 720 uint64_t skip_count;
21dfd91d
UH
721 unsigned int i;
722 gboolean found_match;
723 struct srd_decoder_inst *di;
724 PyObject *py_pinvalues, *py_matched;
514b2edc 725 PyGILState_STATE gstate;
21dfd91d
UH
726
727 if (!self || !args)
728 return NULL;
729
514b2edc
UH
730 gstate = PyGILState_Ensure();
731
21dfd91d
UH
732 if (!(di = srd_inst_find_by_obj(NULL, self))) {
733 PyErr_SetString(PyExc_Exception, "decoder instance not found");
514b2edc 734 PyGILState_Release(gstate);
21dfd91d
UH
735 Py_RETURN_NONE;
736 }
737
738 ret = set_new_condition_list(self, args);
04383ea8
GS
739 if (ret < 0) {
740 srd_dbg("%s: %s: Aborting wait().", di->inst_id, __func__);
514b2edc 741 goto err;
04383ea8 742 }
21dfd91d 743 if (ret == 9999) {
96e1cee7
GS
744 /*
745 * Empty condition list, automatic match. Arrange for the
746 * execution of regular match handling code paths such that
747 * the next available sample is returned to the caller.
748 * Make sure to skip one sample when "anywhere within the
749 * stream", yet make sure to not skip sample number 0.
750 */
751 if (di->abs_cur_samplenum)
752 skip_count = 1;
753 else if (!di->condition_list)
754 skip_count = 0;
755 else
756 skip_count = 1;
757 ret = set_skip_condition(di, skip_count);
758 if (ret < 0) {
759 srd_dbg("%s: %s: Cannot setup condition-less wait().",
760 di->inst_id, __func__);
514b2edc 761 goto err;
96e1cee7 762 }
21dfd91d
UH
763 }
764
765 while (1) {
514b2edc
UH
766
767 Py_BEGIN_ALLOW_THREADS
768
04383ea8 769 /* Wait for new samples to process, or termination request. */
21dfd91d 770 g_mutex_lock(&di->data_mutex);
04383ea8 771 while (!di->got_new_samples && !di->want_wait_terminate)
21dfd91d
UH
772 g_cond_wait(&di->got_new_samples_cond, &di->data_mutex);
773
04383ea8
GS
774 /*
775 * Check whether any of the current condition(s) match.
776 * Arrange for termination requests to take a code path which
777 * won't find new samples to process, pretends to have processed
778 * previously stored samples, and returns to the main thread,
779 * while the termination request still gets signalled.
780 */
781 found_match = FALSE;
21dfd91d
UH
782 ret = process_samples_until_condition_match(di, &found_match);
783
514b2edc
UH
784 Py_END_ALLOW_THREADS
785
21dfd91d
UH
786 /* If there's a match, set self.samplenum etc. and return. */
787 if (found_match) {
788 /* Set self.samplenum to the (absolute) sample number that matched. */
789 PyObject_SetAttrString(di->py_inst, "samplenum",
4564e8e5 790 PyLong_FromLong(di->abs_cur_samplenum));
21dfd91d
UH
791
792 if (di->match_array && di->match_array->len > 0) {
793 py_matched = PyTuple_New(di->match_array->len);
794 for (i = 0; i < di->match_array->len; i++)
795 PyTuple_SetItem(py_matched, i, PyBool_FromLong(di->match_array->data[i]));
796 PyObject_SetAttrString(di->py_inst, "matched", py_matched);
797 match_array_free(di);
798 } else {
799 PyObject_SetAttrString(di->py_inst, "matched", Py_None);
800 }
801
802 py_pinvalues = get_current_pinvalues(di);
803
804 g_mutex_unlock(&di->data_mutex);
805
514b2edc
UH
806 PyGILState_Release(gstate);
807
21dfd91d
UH
808 return py_pinvalues;
809 }
810
811 /* No match, reset state for the next chunk. */
812 di->got_new_samples = FALSE;
813 di->handled_all_samples = TRUE;
4564e8e5
UH
814 di->abs_start_samplenum = 0;
815 di->abs_end_samplenum = 0;
21dfd91d
UH
816 di->inbuf = NULL;
817 di->inbuflen = 0;
818
819 /* Signal the main thread that we handled all samples. */
820 g_cond_signal(&di->handled_all_samples_cond);
821
04383ea8
GS
822 /*
823 * When termination of wait() and decode() was requested,
824 * then exit the loop after releasing the mutex.
825 */
826 if (di->want_wait_terminate) {
827 srd_dbg("%s: %s: Will return from wait().",
828 di->inst_id, __func__);
829 g_mutex_unlock(&di->data_mutex);
514b2edc 830 goto err;
04383ea8
GS
831 }
832
21dfd91d
UH
833 g_mutex_unlock(&di->data_mutex);
834 }
835
514b2edc
UH
836 PyGILState_Release(gstate);
837
21dfd91d 838 Py_RETURN_NONE;
514b2edc
UH
839
840err:
841 PyGILState_Release(gstate);
842
843 return NULL;
21dfd91d
UH
844}
845
846/**
847 * Return whether the specified channel was supplied to the decoder.
848 *
849 * @param self TODO. Must not be NULL.
850 * @param args TODO. Must not be NULL.
851 *
852 * @retval Py_True The channel has been supplied by the frontend.
853 * @retval Py_False The channel has been supplied by the frontend.
854 * @retval NULL An error occurred.
855 */
856static PyObject *Decoder_has_channel(PyObject *self, PyObject *args)
857{
858 int idx, max_idx;
859 struct srd_decoder_inst *di;
860 PyObject *py_channel;
514b2edc 861 PyGILState_STATE gstate;
21dfd91d
UH
862
863 if (!self || !args)
864 return NULL;
865
514b2edc
UH
866 gstate = PyGILState_Ensure();
867
21dfd91d
UH
868 if (!(di = srd_inst_find_by_obj(NULL, self))) {
869 PyErr_SetString(PyExc_Exception, "decoder instance not found");
514b2edc 870 goto err;
21dfd91d
UH
871 }
872
873 /* Parse the argument of self.has_channel() into 'py_channel'. */
874 if (!PyArg_ParseTuple(args, "O", &py_channel)) {
875 /* Let Python raise this exception. */
514b2edc 876 goto err;
21dfd91d
UH
877 }
878
879 if (!PyLong_Check(py_channel)) {
880 PyErr_SetString(PyExc_Exception, "channel index not a number");
514b2edc 881 goto err;
21dfd91d
UH
882 }
883
884 idx = PyLong_AsLong(py_channel);
885 max_idx = g_slist_length(di->decoder->channels)
886 + g_slist_length(di->decoder->opt_channels) - 1;
887
888 if (idx < 0 || idx > max_idx) {
889 srd_err("Invalid channel index %d/%d.", idx, max_idx);
890 PyErr_SetString(PyExc_Exception, "invalid channel");
514b2edc 891 goto err;
21dfd91d
UH
892 }
893
514b2edc
UH
894 PyGILState_Release(gstate);
895
21dfd91d 896 return (di->dec_channelmap[idx] == -1) ? Py_False : Py_True;
514b2edc
UH
897
898err:
899 PyGILState_Release(gstate);
900
901 return NULL;
21dfd91d
UH
902}
903
d0a0ed03
BV
904static PyMethodDef Decoder_methods[] = {
905 {"put", Decoder_put, METH_VARARGS,
86528298 906 "Accepts a dictionary with the following keys: startsample, endsample, data"},
7ee0c40b
BV
907 {"register", (PyCFunction)Decoder_register, METH_VARARGS|METH_KEYWORDS,
908 "Register a new output stream"},
21dfd91d
UH
909 {"wait", Decoder_wait, METH_VARARGS,
910 "Wait for one or more conditions to occur"},
911 {"has_channel", Decoder_has_channel, METH_VARARGS,
912 "Report whether a channel was supplied"},
d0a0ed03
BV
913 {NULL, NULL, 0, NULL}
914};
915
21dfd91d
UH
916/**
917 * Create the sigrokdecode.Decoder type.
918 *
201a85a8 919 * @return The new type object.
21dfd91d 920 *
201a85a8
DE
921 * @private
922 */
923SRD_PRIV PyObject *srd_Decoder_type_new(void)
924{
925 PyType_Spec spec;
926 PyType_Slot slots[] = {
927 { Py_tp_doc, "sigrok Decoder base class" },
928 { Py_tp_methods, Decoder_methods },
929 { Py_tp_new, (void *)&PyType_GenericNew },
930 { 0, NULL }
931 };
514b2edc
UH
932 PyObject *py_obj;
933 PyGILState_STATE gstate;
934
935 gstate = PyGILState_Ensure();
936
201a85a8
DE
937 spec.name = "sigrokdecode.Decoder";
938 spec.basicsize = sizeof(srd_Decoder);
939 spec.itemsize = 0;
940 spec.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE;
941 spec.slots = slots;
942
514b2edc
UH
943 py_obj = PyType_FromSpec(&spec);
944
945 PyGILState_Release(gstate);
946
947 return py_obj;
201a85a8 948}