2 * The MIT License (MIT)
4 * Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
25 #include "logging/comp-logging.h"
26 #include "compat/glib.h"
29 * This hash table associates a BT component class object address to a
30 * user-defined Python class (PyObject *). The keys and values are NOT
31 * owned by this hash table. The Python class objects are owned by the
32 * Python module, which should not be unloaded until it is not possible
33 * to create a user Python component anyway.
35 * This hash table is written to when a user-defined Python component
36 * class is created by one of the bt_bt2_component_class_*_create()
39 * This function is read from when a user calls bt_component_create()
40 * with a component class pointer created by one of the functions above.
41 * In this case, the original Python class needs to be found to
42 * instantiate it and associate the created Python component object with
43 * a BT component object instance.
46 static GHashTable
*bt_cc_ptr_to_py_cls
;
49 void bt_bt2_unregister_cc_ptr_to_py_cls(const bt_component_class
*comp_cls
)
53 if (!bt_cc_ptr_to_py_cls
) {
57 existed
= g_hash_table_remove(bt_cc_ptr_to_py_cls
, comp_cls
);
62 void register_cc_ptr_to_py_cls(struct bt_component_class
*bt_cc
,
65 if (!bt_cc_ptr_to_py_cls
) {
67 * Lazy-initializing this GHashTable because GLib
68 * might not be initialized yet and it needs to be
69 * before we call g_hash_table_new()
71 BT_LOGD_STR("Creating native component class to Python component class hash table.");
72 bt_cc_ptr_to_py_cls
= g_hash_table_new(g_direct_hash
, g_direct_equal
);
73 BT_ASSERT(bt_cc_ptr_to_py_cls
);
76 g_hash_table_insert(bt_cc_ptr_to_py_cls
, (gpointer
) bt_cc
,
81 PyObject
*lookup_cc_ptr_to_py_cls(const bt_component_class
*bt_cc
)
83 if (!bt_cc_ptr_to_py_cls
) {
84 BT_LOGW("Cannot look up Python component class because hash table is NULL: "
85 "comp-cls-addr=%p", bt_cc
);
89 return (PyObject
*) g_hash_table_lookup(bt_cc_ptr_to_py_cls
,
90 (gconstpointer
) bt_cc
);
93 /* Library destructor */
95 __attribute__((destructor
))
97 void native_comp_class_dtor(void) {
98 /* Destroy component class association hash table */
99 if (bt_cc_ptr_to_py_cls
) {
100 BT_LOGD_STR("Destroying native component class to Python component class hash table.");
101 g_hash_table_destroy(bt_cc_ptr_to_py_cls
);
102 bt_cc_ptr_to_py_cls
= NULL
;
107 int py_exc_to_status(bt_self_component_class
*self_component_class
,
108 bt_self_component
*self_component
,
109 bt_self_message_iterator
*self_message_iterator
,
110 const char *module_name
, int active_log_level
)
112 int status
= __BT_FUNC_STATUS_OK
;
113 PyObject
*exc
= PyErr_Occurred();
119 if (PyErr_GivenExceptionMatches(exc
,
120 py_mod_bt2_exc_try_again_type
)) {
121 status
= __BT_FUNC_STATUS_AGAIN
;
122 } else if (PyErr_GivenExceptionMatches(exc
,
123 py_mod_bt2_exc_stop_type
)) {
124 status
= __BT_FUNC_STATUS_END
;
125 } else if (PyErr_GivenExceptionMatches(exc
,
126 py_mod_bt2_exc_unknown_object_type
)) {
127 status
= __BT_FUNC_STATUS_UNKNOWN_OBJECT
;
130 * Unknown exception: convert to general error.
132 * Because we only want to fetch the log level when
133 * we actually get an exception, and not systematically
134 * when we call py_exc_to_status() (as py_exc_to_status()
135 * can return `__BT_FUNC_STATUS_OK`), we get it here
136 * depending on the actor's type.
138 if (self_component
) {
139 active_log_level
= get_self_component_log_level(
141 } else if (self_message_iterator
) {
142 active_log_level
= get_self_message_iterator_log_level(
143 self_message_iterator
);
146 BT_ASSERT(active_log_level
!= -1);
147 log_exception_and_maybe_append_error(BT_LOG_WARNING
,
148 active_log_level
, true,
149 self_component_class
, self_component
,
150 self_message_iterator
, module_name
);
152 if (PyErr_GivenExceptionMatches(exc
,
153 py_mod_bt2_exc_memory_error
)) {
154 status
= __BT_FUNC_STATUS_MEMORY_ERROR
;
156 status
= __BT_FUNC_STATUS_ERROR
;
166 int py_exc_to_status_component_class(
167 bt_self_component_class
*self_component_class
,
168 int active_log_level
)
170 return py_exc_to_status(self_component_class
, NULL
, NULL
, NULL
,
175 int py_exc_to_status_component(bt_self_component
*self_component
)
177 return py_exc_to_status(NULL
, self_component
, NULL
, NULL
, -1);
181 int py_exc_to_status_message_iterator(
182 bt_self_message_iterator
*self_message_iterator
)
184 return py_exc_to_status(NULL
, NULL
, self_message_iterator
, NULL
, -1);
188 bool bt_bt2_is_python_component_class(const bt_component_class
*comp_cls
)
190 return bt_g_hash_table_contains(bt_cc_ptr_to_py_cls
, comp_cls
);
193 /* Component class proxy methods (delegate to the attached Python object) */
196 bt_component_class_init_method_status
component_class_init(
197 bt_self_component
*self_component
,
198 void *self_component_v
,
199 swig_type_info
*self_comp_cls_type_swig_type
,
200 const bt_value
*params
,
201 void *init_method_data
)
203 const bt_component
*component
= bt_self_component_as_component(self_component
);
204 const bt_component_class
*component_class
= bt_component_borrow_class_const(component
);
205 bt_component_class_init_method_status status
= __BT_FUNC_STATUS_OK
;
206 PyObject
*py_cls
= NULL
;
207 PyObject
*py_comp
= NULL
;
208 PyObject
*py_params_ptr
= NULL
;
209 PyObject
*py_comp_ptr
= NULL
;
210 bt_logging_level log_level
= get_self_component_log_level(
213 BT_ASSERT(self_component
);
214 BT_ASSERT(self_component_v
);
215 BT_ASSERT(self_comp_cls_type_swig_type
);
218 * Get the user-defined Python class which created this
219 * component's class in the first place (borrowed
222 py_cls
= lookup_cc_ptr_to_py_cls(component_class
);
224 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
225 "Cannot find Python class associated to native component class: "
226 "comp-cls-addr=%p", component_class
);
230 /* Parameters pointer -> SWIG pointer Python object */
231 py_params_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(params
),
232 SWIGTYPE_p_bt_value
, 0);
233 if (!py_params_ptr
) {
234 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
235 "Failed to create a SWIG pointer object.");
239 py_comp_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(self_component_v
),
240 self_comp_cls_type_swig_type
, 0);
242 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
243 "Failed to create a SWIG pointer object.");
248 * Do the equivalent of this:
250 * py_comp = py_cls._bt_init_from_native(py_comp_ptr,
251 * py_params_ptr, init_method_data ? init_method_data : Py_None)
253 * _UserComponentType._bt_init_from_native() calls the Python
254 * component object's __init__() function.
256 * We don't take any reference on `init_method_data` which, if
257 * not `NULL`, is assumed to be a `PyObject *`: the user's
258 * __init__() function will eventually take a reference if
259 * needed. If `init_method_data` is `NULL`, then we pass
260 * `Py_None` as the initialization's Python object.
262 py_comp
= PyObject_CallMethod(py_cls
,
263 "_bt_init_from_native", "(OOO)", py_comp_ptr
, py_params_ptr
,
264 init_method_data
? init_method_data
: Py_None
);
266 BT_COMP_LOG_CUR_LVL(BT_LOG_WARNING
, log_level
, self_component
,
267 "Failed to call Python class's _bt_init_from_native() method: "
268 "py-cls-addr=%p", py_cls
);
269 status
= py_exc_to_status_component(self_component
);
274 * Our user Python component object is now fully created and
275 * initialized by the user. Since we just created it, this
276 * native component is its only (persistent) owner.
278 bt_self_component_set_data(self_component
, py_comp
);
283 status
= __BT_FUNC_STATUS_ERROR
;
286 * Clear any exception: we're returning a bad status anyway. If
287 * this call originated from Python (creation from a plugin's
288 * component class, for example), then the user gets an
289 * appropriate creation error.
295 Py_XDECREF(py_params_ptr
);
296 Py_XDECREF(py_comp_ptr
);
301 bt_component_class_get_supported_mip_versions_method_status
302 component_class_get_supported_mip_versions(
303 const bt_component_class
*component_class
,
304 bt_self_component_class
*self_component_class
,
305 const bt_value
*params
, void *init_method_data
,
306 bt_logging_level log_level
,
307 bt_integer_range_set_unsigned
*supported_versions
)
310 PyObject
*py_cls
= NULL
;
311 PyObject
*py_params_ptr
= NULL
;
312 PyObject
*py_range_set_addr
= NULL
;
313 bt_integer_range_set_unsigned
*ret_range_set
= NULL
;
314 bt_component_class_get_supported_mip_versions_method_status status
=
317 py_cls
= lookup_cc_ptr_to_py_cls(component_class
);
319 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
320 "Cannot find Python class associated to native component class: "
321 "comp-cls-addr=%p", component_class
);
325 py_params_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(params
),
326 SWIGTYPE_p_bt_value
, 0);
327 if (!py_params_ptr
) {
328 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
329 "Failed to create a SWIG pointer object.");
334 * We don't take any reference on `init_method_data` which, if
335 * not `NULL`, is assumed to be a `PyObject *`: the user's
336 * _user_get_supported_mip_versions() function will eventually
337 * take a reference if needed. If `init_method_data` is `NULL`,
338 * then we pass `Py_None` as the initialization's Python object.
340 py_range_set_addr
= PyObject_CallMethod(py_cls
,
341 "_bt_get_supported_mip_versions_from_native", "(OOi)",
342 py_params_ptr
, init_method_data
? init_method_data
: Py_None
,
344 if (!py_range_set_addr
) {
345 BT_LOG_WRITE_CUR_LVL(BT_LOG_WARNING
, log_level
, BT_LOG_TAG
,
346 "Failed to call Python class's _bt_get_supported_mip_versions_from_native() method: "
347 "py-cls-addr=%p", py_cls
);
348 status
= py_exc_to_status_component_class(self_component_class
,
354 * The returned object, on success, is an integer object
355 * (PyLong) containing the address of a BT unsigned integer
356 * range set object (new reference).
358 ret_range_set
= PyLong_AsVoidPtr(py_range_set_addr
);
359 BT_ASSERT(!PyErr_Occurred());
360 BT_ASSERT(ret_range_set
);
362 /* Copy returned ranges to input range set */
363 for (i
= 0; i
< bt_integer_range_set_get_range_count(
364 bt_integer_range_set_unsigned_as_range_set_const(ret_range_set
));
366 const bt_integer_range_unsigned
*range
=
367 bt_integer_range_set_unsigned_borrow_range_by_index_const(
369 bt_integer_range_set_add_range_status add_range_status
;
371 add_range_status
= bt_integer_range_set_unsigned_add_range(
373 bt_integer_range_unsigned_get_lower(range
),
374 bt_integer_range_unsigned_get_upper(range
));
375 if (add_range_status
) {
376 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
377 "Failed to add range to supported MIP versions range set.");
386 status
= __BT_FUNC_STATUS_ERROR
;
389 Py_XDECREF(py_params_ptr
);
390 Py_XDECREF(py_range_set_addr
);
391 bt_integer_range_set_unsigned_put_ref(ret_range_set
);
396 bt_component_class_get_supported_mip_versions_method_status
397 component_class_source_get_supported_mip_versions(
398 bt_self_component_class_source
*self_component_class_source
,
399 const bt_value
*params
, void *init_method_data
,
400 bt_logging_level log_level
,
401 bt_integer_range_set_unsigned
*supported_versions
)
403 const bt_component_class_source
*component_class_source
= bt_self_component_class_source_as_component_class_source(self_component_class_source
);
404 const bt_component_class
*component_class
= bt_component_class_source_as_component_class_const(component_class_source
);
405 bt_self_component_class
*self_component_class
= bt_self_component_class_source_as_self_component_class(self_component_class_source
);
407 return component_class_get_supported_mip_versions(
408 component_class
, self_component_class
,
409 params
, init_method_data
, log_level
, supported_versions
);
413 bt_component_class_get_supported_mip_versions_method_status
414 component_class_filter_get_supported_mip_versions(
415 bt_self_component_class_filter
*self_component_class_filter
,
416 const bt_value
*params
, void *init_method_data
,
417 bt_logging_level log_level
,
418 bt_integer_range_set_unsigned
*supported_versions
)
420 const bt_component_class_filter
*component_class_filter
= bt_self_component_class_filter_as_component_class_filter(self_component_class_filter
);
421 const bt_component_class
*component_class
= bt_component_class_filter_as_component_class_const(component_class_filter
);
422 bt_self_component_class
*self_component_class
= bt_self_component_class_filter_as_self_component_class(self_component_class_filter
);
424 return component_class_get_supported_mip_versions(
425 component_class
, self_component_class
,
426 params
, init_method_data
, log_level
, supported_versions
);
430 bt_component_class_get_supported_mip_versions_method_status
431 component_class_sink_get_supported_mip_versions(
432 bt_self_component_class_sink
*self_component_class_sink
,
433 const bt_value
*params
, void *init_method_data
,
434 bt_logging_level log_level
,
435 bt_integer_range_set_unsigned
*supported_versions
)
437 const bt_component_class_sink
*component_class_sink
= bt_self_component_class_sink_as_component_class_sink(self_component_class_sink
);
438 const bt_component_class
*component_class
= bt_component_class_sink_as_component_class_const(component_class_sink
);
439 bt_self_component_class
*self_component_class
= bt_self_component_class_sink_as_self_component_class(self_component_class_sink
);
441 return component_class_get_supported_mip_versions(
442 component_class
, self_component_class
,
443 params
, init_method_data
, log_level
, supported_versions
);
447 * Method of bt_component_class_source to initialize a bt_self_component_source
452 bt_component_class_init_method_status
component_class_source_init(
453 bt_self_component_source
*self_component_source
,
454 const bt_value
*params
, void *init_method_data
)
456 bt_self_component
*self_component
= bt_self_component_source_as_self_component(self_component_source
);
457 return component_class_init(
459 self_component_source
,
460 SWIGTYPE_p_bt_self_component_source
,
461 params
, init_method_data
);
465 bt_component_class_init_method_status
component_class_filter_init(
466 bt_self_component_filter
*self_component_filter
,
467 const bt_value
*params
, void *init_method_data
)
469 bt_self_component
*self_component
= bt_self_component_filter_as_self_component(self_component_filter
);
470 return component_class_init(
472 self_component_filter
,
473 SWIGTYPE_p_bt_self_component_filter
,
474 params
, init_method_data
);
478 bt_component_class_init_method_status
component_class_sink_init(
479 bt_self_component_sink
*self_component_sink
,
480 const bt_value
*params
, void *init_method_data
)
482 bt_self_component
*self_component
= bt_self_component_sink_as_self_component(self_component_sink
);
483 return component_class_init(
486 SWIGTYPE_p_bt_self_component_sink
,
487 params
, init_method_data
);
491 void component_class_finalize(bt_self_component
*self_component
)
493 PyObject
*py_comp
= bt_self_component_get_data(self_component
);
496 /* Call user's _user_finalize() method */
497 PyObject
*py_method_result
= PyObject_CallMethod(py_comp
,
498 "_user_finalize", NULL
);
500 if (PyErr_Occurred()) {
501 bt_logging_level log_level
= get_self_component_log_level(
504 BT_COMP_LOG_CUR_LVL(BT_LOG_WARNING
, log_level
, self_component
,
505 "User component's _user_finalize() method raised an exception: ignoring:");
506 logw_exception(log_level
);
510 * Ignore any exception raised by the _user_finalize() method
511 * because it won't change anything at this point: the component
512 * is being destroyed anyway.
515 Py_XDECREF(py_method_result
);
520 void component_class_source_finalize(bt_self_component_source
*self_component_source
)
522 bt_self_component
*self_component
= bt_self_component_source_as_self_component(self_component_source
);
523 component_class_finalize(self_component
);
527 void component_class_filter_finalize(bt_self_component_filter
*self_component_filter
)
529 bt_self_component
*self_component
= bt_self_component_filter_as_self_component(self_component_filter
);
530 component_class_finalize(self_component
);
534 void component_class_sink_finalize(bt_self_component_sink
*self_component_sink
)
536 bt_self_component
*self_component
= bt_self_component_sink_as_self_component(self_component_sink
);
537 component_class_finalize(self_component
);
541 bt_component_class_message_iterator_can_seek_beginning_method_status
542 component_class_can_seek_beginning(
543 bt_self_message_iterator
*self_message_iterator
, bt_bool
*can_seek
)
546 PyObject
*py_result
= NULL
;
547 bt_component_class_message_iterator_can_seek_beginning_method_status status
;
548 py_iter
= bt_self_message_iterator_get_data(self_message_iterator
);
551 py_result
= PyObject_CallMethod(py_iter
,
552 "_bt_can_seek_beginning_from_native", NULL
);
554 BT_ASSERT(!py_result
|| PyBool_Check(py_result
));
557 *can_seek
= PyObject_IsTrue(py_result
);
558 status
= BT_COMPONENT_CLASS_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_METHOD_STATUS_OK
;
560 status
= py_exc_to_status_message_iterator(self_message_iterator
);
561 BT_ASSERT(status
!= __BT_FUNC_STATUS_OK
);
564 Py_XDECREF(py_result
);
570 bt_component_class_message_iterator_seek_beginning_method_status
571 component_class_seek_beginning(bt_self_message_iterator
*self_message_iterator
)
575 bt_component_class_message_iterator_seek_beginning_method_status status
;
577 py_iter
= bt_self_message_iterator_get_data(self_message_iterator
);
579 py_result
= PyObject_CallMethod(py_iter
, "_bt_seek_beginning_from_native",
581 BT_ASSERT(!py_result
|| py_result
== Py_None
);
582 status
= py_exc_to_status_message_iterator(self_message_iterator
);
583 Py_XDECREF(py_result
);
588 bt_component_class_port_connected_method_status
component_class_port_connected(
589 bt_self_component
*self_component
,
590 void *self_component_port
,
591 swig_type_info
*self_component_port_swig_type
,
592 bt_port_type self_component_port_type
,
593 const void *other_port
,
594 swig_type_info
*other_port_swig_type
)
596 bt_component_class_port_connected_method_status status
;
597 PyObject
*py_comp
= NULL
;
598 PyObject
*py_self_port_ptr
= NULL
;
599 PyObject
*py_other_port_ptr
= NULL
;
600 PyObject
*py_method_result
= NULL
;
601 bt_logging_level log_level
= get_self_component_log_level(
604 py_comp
= bt_self_component_get_data(self_component
);
606 py_self_port_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(self_component_port
),
607 self_component_port_swig_type
, 0);
608 if (!py_self_port_ptr
) {
609 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
610 "Failed to create a SWIG pointer object.");
611 status
= __BT_FUNC_STATUS_MEMORY_ERROR
;
615 py_other_port_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(other_port
),
616 other_port_swig_type
, 0);
617 if (!py_other_port_ptr
) {
618 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
619 "Failed to create a SWIG pointer object.");
620 status
= __BT_FUNC_STATUS_MEMORY_ERROR
;
623 py_method_result
= PyObject_CallMethod(py_comp
,
624 "_bt_port_connected_from_native", "(OiO)", py_self_port_ptr
,
625 self_component_port_type
, py_other_port_ptr
);
626 BT_ASSERT(!py_method_result
|| py_method_result
== Py_None
);
627 status
= py_exc_to_status_component(self_component
);
630 Py_XDECREF(py_self_port_ptr
);
631 Py_XDECREF(py_other_port_ptr
);
632 Py_XDECREF(py_method_result
);
637 bt_component_class_port_connected_method_status
638 component_class_source_output_port_connected(
639 bt_self_component_source
*self_component_source
,
640 bt_self_component_port_output
*self_component_port_output
,
641 const bt_port_input
*other_port_input
)
643 bt_self_component
*self_component
= bt_self_component_source_as_self_component(self_component_source
);
645 return component_class_port_connected(
647 self_component_port_output
,
648 SWIGTYPE_p_bt_self_component_port_output
,
651 SWIGTYPE_p_bt_port_input
);
655 bt_component_class_port_connected_method_status
656 component_class_filter_input_port_connected(
657 bt_self_component_filter
*self_component_filter
,
658 bt_self_component_port_input
*self_component_port_input
,
659 const bt_port_output
*other_port_output
)
661 bt_self_component
*self_component
= bt_self_component_filter_as_self_component(self_component_filter
);
663 return component_class_port_connected(
665 self_component_port_input
,
666 SWIGTYPE_p_bt_self_component_port_input
,
669 SWIGTYPE_p_bt_port_output
);
673 bt_component_class_port_connected_method_status
674 component_class_filter_output_port_connected(
675 bt_self_component_filter
*self_component_filter
,
676 bt_self_component_port_output
*self_component_port_output
,
677 const bt_port_input
*other_port_input
)
679 bt_self_component
*self_component
= bt_self_component_filter_as_self_component(self_component_filter
);
681 return component_class_port_connected(
683 self_component_port_output
,
684 SWIGTYPE_p_bt_self_component_port_output
,
687 SWIGTYPE_p_bt_port_input
);
691 bt_component_class_port_connected_method_status
692 component_class_sink_input_port_connected(
693 bt_self_component_sink
*self_component_sink
,
694 bt_self_component_port_input
*self_component_port_input
,
695 const bt_port_output
*other_port_output
)
697 bt_self_component
*self_component
= bt_self_component_sink_as_self_component(self_component_sink
);
699 return component_class_port_connected(
701 self_component_port_input
,
702 SWIGTYPE_p_bt_self_component_port_input
,
705 SWIGTYPE_p_bt_port_output
);
709 bt_component_class_sink_graph_is_configured_method_status
710 component_class_sink_graph_is_configured(
711 bt_self_component_sink
*self_component_sink
)
713 PyObject
*py_comp
= NULL
;
714 PyObject
*py_method_result
= NULL
;
715 bt_component_class_sink_graph_is_configured_method_status status
= __BT_FUNC_STATUS_OK
;
716 bt_self_component
*self_component
= bt_self_component_sink_as_self_component(self_component_sink
);
718 py_comp
= bt_self_component_get_data(self_component
);
719 py_method_result
= PyObject_CallMethod(py_comp
,
720 "_bt_graph_is_configured_from_native", NULL
);
721 BT_ASSERT(!py_method_result
|| py_method_result
== Py_None
);
722 status
= py_exc_to_status_component(self_component
);
723 Py_XDECREF(py_method_result
);
728 bt_component_class_query_method_status
component_class_query(
729 const bt_component_class
*component_class
,
730 bt_self_component_class
*self_component_class
,
731 bt_private_query_executor
*priv_query_executor
,
732 const char *object
, const bt_value
*params
, void *method_data
,
733 const bt_value
**result
)
735 PyObject
*py_cls
= NULL
;
736 PyObject
*py_params_ptr
= NULL
;
737 PyObject
*py_priv_query_exec_ptr
= NULL
;
738 PyObject
*py_query_func
= NULL
;
739 PyObject
*py_object
= NULL
;
740 PyObject
*py_results_addr
= NULL
;
741 bt_component_class_query_method_status status
= __BT_FUNC_STATUS_OK
;
742 const bt_query_executor
*query_exec
=
743 bt_private_query_executor_as_query_executor_const(
744 priv_query_executor
);
745 bt_logging_level log_level
=
746 bt_query_executor_get_logging_level(query_exec
);
749 * If there's any `method_data`, assume this component class is
750 * getting queried from Python, so that `method_data` is a
751 * Python object to pass to the user's _user_query() method.
753 BT_ASSERT(!method_data
||
754 bt_bt2_is_python_component_class(component_class
));
756 py_cls
= lookup_cc_ptr_to_py_cls(component_class
);
758 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
759 "Cannot find Python class associated to native component class: "
760 "comp-cls-addr=%p", component_class
);
764 py_params_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(params
),
765 SWIGTYPE_p_bt_value
, 0);
766 if (!py_params_ptr
) {
767 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
768 "Failed to create a SWIG pointer object.");
772 py_priv_query_exec_ptr
= SWIG_NewPointerObj(
773 SWIG_as_voidptr(priv_query_executor
),
774 SWIGTYPE_p_bt_private_query_executor
, 0);
775 if (!py_priv_query_exec_ptr
) {
776 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
777 "Failed to create a SWIG pointer object.");
781 py_object
= SWIG_FromCharPtr(object
);
783 BT_LOG_WRITE_CUR_LVL(BT_LOG_ERROR
, log_level
, BT_LOG_TAG
,
784 "Failed to create a Python string.");
789 * We don't take any reference on `method_data` which, if not
790 * `NULL`, is assumed to be a `PyObject *`: the user's
791 * _user_query() function will eventually take a reference if
792 * needed. If `method_data` is `NULL`, then we pass `Py_None` as
793 * the initialization's Python object.
795 py_results_addr
= PyObject_CallMethod(py_cls
,
796 "_bt_query_from_native", "(OOOO)", py_priv_query_exec_ptr
,
797 py_object
, py_params_ptr
,
798 method_data
? method_data
: Py_None
);
799 if (!py_results_addr
) {
800 status
= py_exc_to_status_component_class(self_component_class
,
803 static const char *fmt
=
804 "Failed to call Python class's _bt_query_from_native() method: py-cls-addr=%p";
805 BT_LOG_WRITE_CUR_LVL(BT_LOG_WARNING
, log_level
, BT_LOG_TAG
,
807 BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_COMPONENT_CLASS(
808 self_component_class
, fmt
, py_cls
);
814 * The returned object, on success, is an integer object
815 * (PyLong) containing the address of a BT value object (new
818 *result
= PyLong_AsVoidPtr(py_results_addr
);
819 BT_ASSERT(!PyErr_Occurred());
825 status
= __BT_FUNC_STATUS_ERROR
;
828 Py_XDECREF(py_params_ptr
);
829 Py_XDECREF(py_priv_query_exec_ptr
);
830 Py_XDECREF(py_query_func
);
831 Py_XDECREF(py_object
);
832 Py_XDECREF(py_results_addr
);
837 bt_component_class_query_method_status
component_class_source_query(
838 bt_self_component_class_source
*self_component_class_source
,
839 bt_private_query_executor
*priv_query_executor
,
840 const char *object
, const bt_value
*params
, void *method_data
,
841 const bt_value
**result
)
843 const bt_component_class_source
*component_class_source
= bt_self_component_class_source_as_component_class_source(self_component_class_source
);
844 const bt_component_class
*component_class
= bt_component_class_source_as_component_class_const(component_class_source
);
845 bt_self_component_class
*self_component_class
= bt_self_component_class_source_as_self_component_class(self_component_class_source
);
847 return component_class_query(component_class
, self_component_class
,
848 priv_query_executor
, object
, params
, method_data
, result
);
852 bt_component_class_query_method_status
component_class_filter_query(
853 bt_self_component_class_filter
*self_component_class_filter
,
854 bt_private_query_executor
*priv_query_executor
,
855 const char *object
, const bt_value
*params
, void *method_data
,
856 const bt_value
**result
)
858 const bt_component_class_filter
*component_class_filter
= bt_self_component_class_filter_as_component_class_filter(self_component_class_filter
);
859 const bt_component_class
*component_class
= bt_component_class_filter_as_component_class_const(component_class_filter
);
860 bt_self_component_class
*self_component_class
= bt_self_component_class_filter_as_self_component_class(self_component_class_filter
);
862 return component_class_query(component_class
, self_component_class
,
863 priv_query_executor
, object
, params
, method_data
, result
);
867 bt_component_class_query_method_status
component_class_sink_query(
868 bt_self_component_class_sink
*self_component_class_sink
,
869 bt_private_query_executor
*priv_query_executor
,
870 const char *object
, const bt_value
*params
, void *method_data
,
871 const bt_value
**result
)
873 const bt_component_class_sink
*component_class_sink
= bt_self_component_class_sink_as_component_class_sink(self_component_class_sink
);
874 const bt_component_class
*component_class
= bt_component_class_sink_as_component_class_const(component_class_sink
);
875 bt_self_component_class
*self_component_class
= bt_self_component_class_sink_as_self_component_class(self_component_class_sink
);
877 return component_class_query(component_class
, self_component_class
,
878 priv_query_executor
, object
, params
, method_data
, result
);
882 bt_component_class_message_iterator_init_method_status
883 component_class_message_iterator_init(
884 bt_self_message_iterator
*self_message_iterator
,
885 bt_self_component
*self_component
,
886 bt_self_component_port_output
*self_component_port_output
)
888 bt_component_class_message_iterator_init_method_status status
= __BT_FUNC_STATUS_OK
;
889 PyObject
*py_comp_cls
= NULL
;
890 PyObject
*py_iter_cls
= NULL
;
891 PyObject
*py_iter_ptr
= NULL
;
892 PyObject
*py_component_port_output_ptr
= NULL
;
893 PyObject
*py_init_method_result
= NULL
;
894 PyObject
*py_iter
= NULL
;
896 bt_logging_level log_level
= get_self_component_log_level(
899 py_comp
= bt_self_component_get_data(self_component
);
901 /* Find user's Python message iterator class */
902 py_comp_cls
= PyObject_GetAttrString(py_comp
, "__class__");
904 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
905 "Cannot get Python object's `__class__` attribute.");
909 py_iter_cls
= PyObject_GetAttrString(py_comp_cls
, "_iter_cls");
911 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
912 "Cannot get Python class's `_iter_cls` attribute.");
916 py_iter_ptr
= SWIG_NewPointerObj(SWIG_as_voidptr(self_message_iterator
),
917 SWIGTYPE_p_bt_self_message_iterator
, 0);
919 const char *err
= "Failed to create a SWIG pointer object.";
921 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
923 BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_MESSAGE_ITERATOR(
924 self_message_iterator
, err
);
929 * Create object with borrowed native message iterator
932 * py_iter = py_iter_cls.__new__(py_iter_cls, py_iter_ptr)
934 py_iter
= PyObject_CallMethod(py_iter_cls
, "__new__",
935 "(OO)", py_iter_cls
, py_iter_ptr
);
937 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
938 "Failed to call Python class's __new__() method: "
939 "py-cls-addr=%p", py_iter_cls
);
946 * py_iter.__init__(self_output_port)
948 * through the _init_for_native helper static method.
950 * At this point, py_iter._ptr is set, so this initialization
951 * function has access to self._component (which gives it the
952 * user Python component object from which the iterator was
955 py_component_port_output_ptr
= SWIG_NewPointerObj(
956 SWIG_as_voidptr(self_component_port_output
),
957 SWIGTYPE_p_bt_self_component_port_output
, 0);
958 if (!py_component_port_output_ptr
) {
959 const char *err
= "Failed to create a SWIG pointer object.";
961 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
963 BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_MESSAGE_ITERATOR(
964 self_message_iterator
, err
);
968 py_init_method_result
= PyObject_CallMethod(py_iter
,
969 "_bt_init_from_native", "O", py_component_port_output_ptr
);
970 if (!py_init_method_result
) {
971 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR
, log_level
, self_component
,
972 "User's __init__() method failed:");
977 * Since the Python code can never instantiate a user-defined
978 * message iterator class, the native message iterator
979 * object does NOT belong to a user Python message iterator
980 * object (borrowed reference). However this Python object is
981 * owned by this native message iterator object.
983 * In the Python world, the lifetime of the native message
984 * iterator is managed by a _GenericMessageIterator
987 * _GenericMessageIterator instance:
988 * owns a native bt_message_iterator object (iter)
989 * owns a _UserMessageIterator instance (py_iter)
990 * self._ptr is a borrowed reference to the
991 * native bt_private_connection_private_message_iterator
994 bt_self_message_iterator_set_data(self_message_iterator
, py_iter
);
999 /* Handling of errors that cause a Python exception to be set. */
1000 status
= py_exc_to_status_message_iterator(self_message_iterator
);
1001 BT_ASSERT(status
!= __BT_FUNC_STATUS_OK
);
1005 /* Handling of errors that don't cause a Python exception to be set. */
1006 status
= __BT_FUNC_STATUS_ERROR
;
1009 BT_ASSERT(!PyErr_Occurred());
1011 Py_XDECREF(py_comp_cls
);
1012 Py_XDECREF(py_iter_cls
);
1013 Py_XDECREF(py_iter_ptr
);
1014 Py_XDECREF(py_component_port_output_ptr
);
1015 Py_XDECREF(py_init_method_result
);
1016 Py_XDECREF(py_iter
);
1021 bt_component_class_message_iterator_init_method_status
1022 component_class_source_message_iterator_init(
1023 bt_self_message_iterator
*self_message_iterator
,
1024 bt_self_component_source
*self_component_source
,
1025 bt_self_component_port_output
*self_component_port_output
)
1027 bt_self_component
*self_component
= bt_self_component_source_as_self_component(self_component_source
);
1029 return component_class_message_iterator_init(self_message_iterator
, self_component
, self_component_port_output
);
1033 bt_component_class_message_iterator_init_method_status
1034 component_class_filter_message_iterator_init(
1035 bt_self_message_iterator
*self_message_iterator
,
1036 bt_self_component_filter
*self_component_filter
,
1037 bt_self_component_port_output
*self_component_port_output
)
1039 bt_self_component
*self_component
= bt_self_component_filter_as_self_component(self_component_filter
);
1041 return component_class_message_iterator_init(self_message_iterator
, self_component
, self_component_port_output
);
1045 void component_class_message_iterator_finalize(
1046 bt_self_message_iterator
*message_iterator
)
1048 PyObject
*py_message_iter
= bt_self_message_iterator_get_data(
1050 PyObject
*py_method_result
= NULL
;
1052 BT_ASSERT(py_message_iter
);
1054 /* Call user's _user_finalize() method */
1055 py_method_result
= PyObject_CallMethod(py_message_iter
,
1056 "_user_finalize", NULL
);
1058 if (PyErr_Occurred()) {
1059 bt_self_component
*self_comp
=
1060 bt_self_message_iterator_borrow_component(
1062 bt_logging_level log_level
= get_self_component_log_level(
1065 BT_COMP_LOG_CUR_LVL(BT_LOG_WARNING
, log_level
, self_comp
,
1066 "User's _user_finalize() method raised an exception: ignoring:");
1067 logw_exception(get_self_message_iterator_log_level(
1072 * Ignore any exception raised by the _user_finalize() method
1073 * because it won't change anything at this point: the component
1074 * is being destroyed anyway.
1077 Py_XDECREF(py_method_result
);
1078 Py_DECREF(py_message_iter
);
1081 /* Valid for both sources and filters. */
1084 bt_component_class_message_iterator_next_method_status
1085 component_class_message_iterator_next(
1086 bt_self_message_iterator
*message_iterator
,
1087 bt_message_array_const msgs
, uint64_t capacity
,
1090 bt_component_class_message_iterator_next_method_status status
= __BT_FUNC_STATUS_OK
;
1091 PyObject
*py_message_iter
= bt_self_message_iterator_get_data(message_iterator
);
1092 PyObject
*py_method_result
= NULL
;
1094 BT_ASSERT(py_message_iter
);
1095 py_method_result
= PyObject_CallMethod(py_message_iter
,
1096 "_bt_next_from_native", NULL
);
1097 if (!py_method_result
) {
1098 status
= py_exc_to_status_message_iterator(message_iterator
);
1099 BT_ASSERT(status
!= __BT_FUNC_STATUS_OK
);
1104 * The returned object, on success, is an integer object
1105 * (PyLong) containing the address of a native message
1106 * object (which is now ours).
1108 msgs
[0] = PyLong_AsVoidPtr(py_method_result
);
1111 /* Clear potential overflow error; should never happen */
1112 BT_ASSERT(!PyErr_Occurred());
1116 Py_XDECREF(py_method_result
);
1121 bt_component_class_sink_consume_method_status
1122 component_class_sink_consume(bt_self_component_sink
*self_component_sink
)
1124 bt_self_component
*self_component
= bt_self_component_sink_as_self_component(self_component_sink
);
1125 PyObject
*py_comp
= bt_self_component_get_data(self_component
);
1126 PyObject
*py_method_result
= NULL
;
1127 bt_component_class_sink_consume_method_status status
;
1130 py_method_result
= PyObject_CallMethod(py_comp
,
1131 "_user_consume", NULL
);
1132 status
= py_exc_to_status_component(self_component
);
1133 BT_ASSERT(py_method_result
|| status
!= __BT_FUNC_STATUS_OK
);
1134 Py_XDECREF(py_method_result
);
1139 int component_class_set_help_and_desc(
1140 bt_component_class
*component_class
,
1141 const char *description
, const char *help
)
1146 ret
= bt_component_class_set_description(component_class
, description
);
1148 BT_LOGE("Cannot set component class's description: "
1149 "comp-cls-addr=%p", component_class
);
1155 ret
= bt_component_class_set_help(component_class
, help
);
1157 BT_LOGE("Cannot set component class's help text: "
1158 "comp-cls-addr=%p", component_class
);
1170 bt_component_class_source
*bt_bt2_component_class_source_create(
1171 PyObject
*py_cls
, const char *name
, const char *description
,
1174 bt_component_class_source
*component_class_source
;
1175 bt_component_class
*component_class
;
1179 component_class_source
= bt_component_class_source_create(name
,
1180 component_class_message_iterator_next
);
1181 if (!component_class_source
) {
1182 BT_LOGE_STR("Cannot create source component class.");
1186 component_class
= bt_component_class_source_as_component_class(component_class_source
);
1188 if (component_class_set_help_and_desc(component_class
, description
, help
)) {
1192 ret
= bt_component_class_source_set_init_method(component_class_source
, component_class_source_init
);
1193 BT_ASSERT(ret
== 0);
1194 ret
= bt_component_class_source_set_finalize_method(component_class_source
, component_class_source_finalize
);
1195 BT_ASSERT(ret
== 0);
1196 ret
= bt_component_class_source_set_message_iterator_can_seek_beginning_method(component_class_source
,
1197 component_class_can_seek_beginning
);
1198 BT_ASSERT(ret
== 0);
1199 ret
= bt_component_class_source_set_message_iterator_seek_beginning_method(component_class_source
,
1200 component_class_seek_beginning
);
1201 BT_ASSERT(ret
== 0);
1202 ret
= bt_component_class_source_set_output_port_connected_method(component_class_source
,
1203 component_class_source_output_port_connected
);
1204 BT_ASSERT(ret
== 0);
1205 ret
= bt_component_class_source_set_query_method(component_class_source
, component_class_source_query
);
1206 BT_ASSERT(ret
== 0);
1207 ret
= bt_component_class_source_set_get_supported_mip_versions_method(component_class_source
, component_class_source_get_supported_mip_versions
);
1208 BT_ASSERT(ret
== 0);
1209 ret
= bt_component_class_source_set_message_iterator_init_method(
1210 component_class_source
, component_class_source_message_iterator_init
);
1211 BT_ASSERT(ret
== 0);
1212 ret
= bt_component_class_source_set_message_iterator_finalize_method(
1213 component_class_source
, component_class_message_iterator_finalize
);
1214 BT_ASSERT(ret
== 0);
1215 register_cc_ptr_to_py_cls(component_class
, py_cls
);
1218 return component_class_source
;
1222 bt_component_class_filter
*bt_bt2_component_class_filter_create(
1223 PyObject
*py_cls
, const char *name
, const char *description
,
1226 bt_component_class
*component_class
;
1227 bt_component_class_filter
*component_class_filter
;
1231 component_class_filter
= bt_component_class_filter_create(name
,
1232 component_class_message_iterator_next
);
1233 if (!component_class_filter
) {
1234 BT_LOGE_STR("Cannot create filter component class.");
1238 component_class
= bt_component_class_filter_as_component_class(component_class_filter
);
1240 if (component_class_set_help_and_desc(component_class
, description
, help
)) {
1244 ret
= bt_component_class_filter_set_init_method(component_class_filter
, component_class_filter_init
);
1245 BT_ASSERT(ret
== 0);
1246 ret
= bt_component_class_filter_set_finalize_method (component_class_filter
, component_class_filter_finalize
);
1247 BT_ASSERT(ret
== 0);
1248 ret
= bt_component_class_filter_set_message_iterator_can_seek_beginning_method(component_class_filter
,
1249 component_class_can_seek_beginning
);
1250 BT_ASSERT(ret
== 0);
1251 ret
= bt_component_class_filter_set_message_iterator_seek_beginning_method(component_class_filter
,
1252 component_class_seek_beginning
);
1253 BT_ASSERT(ret
== 0);
1254 ret
= bt_component_class_filter_set_input_port_connected_method(component_class_filter
,
1255 component_class_filter_input_port_connected
);
1256 BT_ASSERT(ret
== 0);
1257 ret
= bt_component_class_filter_set_output_port_connected_method(component_class_filter
,
1258 component_class_filter_output_port_connected
);
1259 BT_ASSERT(ret
== 0);
1260 ret
= bt_component_class_filter_set_query_method(component_class_filter
, component_class_filter_query
);
1261 BT_ASSERT(ret
== 0);
1262 ret
= bt_component_class_filter_set_get_supported_mip_versions_method(component_class_filter
, component_class_filter_get_supported_mip_versions
);
1263 BT_ASSERT(ret
== 0);
1264 ret
= bt_component_class_filter_set_message_iterator_init_method(
1265 component_class_filter
, component_class_filter_message_iterator_init
);
1266 BT_ASSERT(ret
== 0);
1267 ret
= bt_component_class_filter_set_message_iterator_finalize_method(
1268 component_class_filter
, component_class_message_iterator_finalize
);
1269 BT_ASSERT(ret
== 0);
1270 register_cc_ptr_to_py_cls(component_class
, py_cls
);
1273 return component_class_filter
;
1277 bt_component_class_sink
*bt_bt2_component_class_sink_create(
1278 PyObject
*py_cls
, const char *name
, const char *description
,
1281 bt_component_class_sink
*component_class_sink
;
1282 bt_component_class
*component_class
;
1286 component_class_sink
= bt_component_class_sink_create(name
, component_class_sink_consume
);
1288 if (!component_class_sink
) {
1289 BT_LOGE_STR("Cannot create sink component class.");
1293 component_class
= bt_component_class_sink_as_component_class(component_class_sink
);
1295 if (component_class_set_help_and_desc(component_class
, description
, help
)) {
1299 ret
= bt_component_class_sink_set_init_method(component_class_sink
, component_class_sink_init
);
1300 BT_ASSERT(ret
== 0);
1301 ret
= bt_component_class_sink_set_finalize_method(component_class_sink
, component_class_sink_finalize
);
1302 BT_ASSERT(ret
== 0);
1303 ret
= bt_component_class_sink_set_input_port_connected_method(component_class_sink
,
1304 component_class_sink_input_port_connected
);
1305 BT_ASSERT(ret
== 0);
1306 ret
= bt_component_class_sink_set_graph_is_configured_method(component_class_sink
,
1307 component_class_sink_graph_is_configured
);
1308 BT_ASSERT(ret
== 0);
1309 ret
= bt_component_class_sink_set_query_method(component_class_sink
, component_class_sink_query
);
1310 BT_ASSERT(ret
== 0);
1311 ret
= bt_component_class_sink_set_get_supported_mip_versions_method(component_class_sink
, component_class_sink_get_supported_mip_versions
);
1312 BT_ASSERT(ret
== 0);
1313 register_cc_ptr_to_py_cls(component_class
, py_cls
);
1316 return component_class_sink
;