X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Fbindings%2Fpython%2Fbt2%2Fbt2%2Fnative_bt_component_class.i;h=93aaf1b7f9f4c65054e3d4addb23c5abfaf6172d;hb=d24d56638469189904fb6ddbb3c725817b3e9417;hp=6deb22ef5ae22f10815dc704786cb892b24e41af;hpb=c594ab036e1797da45594dce8dfc07d90f2bb81f;p=babeltrace.git diff --git a/src/bindings/python/bt2/bt2/native_bt_component_class.i b/src/bindings/python/bt2/bt2/native_bt_component_class.i index 6deb22ef..93aaf1b7 100644 --- a/src/bindings/python/bt2/bt2/native_bt_component_class.i +++ b/src/bindings/python/bt2/bt2/native_bt_component_class.i @@ -22,401 +22,17 @@ * THE SOFTWARE. */ -/* From component-class-const.h */ - -typedef enum bt_component_class_status { - BT_COMPONENT_CLASS_STATUS_OK = 0, - BT_COMPONENT_CLASS_STATUS_NOMEM = -12, -} bt_component_class_status; - -typedef enum bt_component_class_type { - BT_COMPONENT_CLASS_TYPE_SOURCE = 0, - BT_COMPONENT_CLASS_TYPE_FILTER = 1, - BT_COMPONENT_CLASS_TYPE_SINK = 2, -} bt_component_class_type; - -extern const char *bt_component_class_get_name( - const bt_component_class *component_class); - -extern const char *bt_component_class_get_description( - const bt_component_class *component_class); - -extern const char *bt_component_class_get_help( - const bt_component_class *component_class); - -extern bt_component_class_type bt_component_class_get_type( - const bt_component_class *component_class); - -bt_bool bt_component_class_is_source( - const bt_component_class *component_class); - -bt_bool bt_component_class_is_filter( - const bt_component_class *component_class); - -bt_bool bt_component_class_is_sink( - const bt_component_class *component_class); - -extern void bt_component_class_get_ref( - const bt_component_class *component_class); - -extern void bt_component_class_put_ref( - const bt_component_class *component_class); - -/* From component-class-source-const.h */ - -const bt_component_class * -bt_component_class_source_as_component_class_const( - const bt_component_class_source *comp_cls_source); - -extern void bt_component_class_source_get_ref( - const bt_component_class_source *component_class_source); - -extern void bt_component_class_source_put_ref( - const bt_component_class_source *component_class_source); - -/* From component-class-source.h */ - -typedef bt_self_component_status -(*bt_component_class_source_init_method)( - bt_self_component_source *self_component, - const bt_value *params, void *init_method_data); - -typedef void (*bt_component_class_source_finalize_method)( - bt_self_component_source *self_component); - -typedef bt_self_message_iterator_status -(*bt_component_class_source_message_iterator_init_method)( - bt_self_message_iterator *message_iterator, - bt_self_component_source *self_component, - bt_self_component_port_output *port); - -typedef void -(*bt_component_class_source_message_iterator_finalize_method)( - bt_self_message_iterator *message_iterator); - -typedef bt_self_message_iterator_status -(*bt_component_class_source_message_iterator_next_method)( - bt_self_message_iterator *message_iterator, - bt_message_array_const msgs, uint64_t capacity, - uint64_t *count); - -typedef bt_self_message_iterator_status -(*bt_component_class_source_message_iterator_seek_ns_from_origin_method)( - bt_self_message_iterator *message_iterator, - int64_t ns_from_origin); - -typedef bt_self_message_iterator_status -(*bt_component_class_source_message_iterator_seek_beginning_method)( - bt_self_message_iterator *message_iterator); - -typedef bt_bool -(*bt_component_class_source_message_iterator_can_seek_ns_from_origin_method)( - bt_self_message_iterator *message_iterator, - int64_t ns_from_origin); - -typedef bt_bool -(*bt_component_class_source_message_iterator_can_seek_beginning_method)( - bt_self_message_iterator *message_iterator); - -typedef bt_query_status (*bt_component_class_source_query_method)( - bt_self_component_class_source *comp_class, - const bt_query_executor *query_executor, - const char *object, const bt_value *params, - bt_logging_level log_level, - const bt_value **result); - -typedef bt_self_component_status -(*bt_component_class_source_output_port_connected_method)( - bt_self_component_source *self_component, - bt_self_component_port_output *self_port, - const bt_port_input *other_port); - -bt_component_class *bt_component_class_source_as_component_class( - bt_component_class_source *comp_cls_source); - -extern -bt_component_class_source *bt_component_class_source_create( - const char *name, - bt_component_class_source_message_iterator_next_method method); - -extern bt_component_class_status -bt_component_class_source_set_init_method( - bt_component_class_source *comp_class, - bt_component_class_source_init_method method); - -extern bt_component_class_status -bt_component_class_source_set_finalize_method( - bt_component_class_source *comp_class, - bt_component_class_source_finalize_method method); - -extern bt_component_class_status -bt_component_class_source_set_output_port_connected_method( - bt_component_class_source *comp_class, - bt_component_class_source_output_port_connected_method method); - -extern bt_component_class_status -bt_component_class_source_set_query_method( - bt_component_class_source *comp_class, - bt_component_class_source_query_method method); - -extern bt_component_class_status -bt_component_class_source_set_message_iterator_init_method( - bt_component_class_source *comp_class, - bt_component_class_source_message_iterator_init_method method); - -extern bt_component_class_status -bt_component_class_source_set_message_iterator_finalize_method( - bt_component_class_source *comp_class, - bt_component_class_source_message_iterator_finalize_method method); - -extern bt_component_class_status -bt_component_class_source_set_message_iterator_seek_ns_from_origin_method( - bt_component_class_source *comp_class, - bt_component_class_source_message_iterator_seek_ns_from_origin_method method); - -extern bt_component_class_status -bt_component_class_source_set_message_iterator_seek_beginning_method( - bt_component_class_source *comp_class, - bt_component_class_source_message_iterator_seek_beginning_method method); - -extern bt_bool -bt_component_class_source_set_message_iterator_can_seek_ns_from_origin_method( - bt_component_class_source *comp_class, - bt_component_class_source_message_iterator_can_seek_ns_from_origin_method method); - -extern bt_bool -bt_component_class_source_set_message_iterator_can_seek_beginning_method( - bt_component_class_source *comp_class, - bt_component_class_source_message_iterator_can_seek_beginning_method method); - -/* From component-class-filter-const.h */ - -const bt_component_class * -bt_component_class_filter_as_component_class_const( - const bt_component_class_filter *comp_cls_filter); - -extern void bt_component_class_filter_get_ref( - const bt_component_class_filter *component_class_filter); - -extern void bt_component_class_filter_put_ref( - const bt_component_class_filter *component_class_filter); - -/* From component-class-filter.h */ - -typedef bt_self_component_status -(*bt_component_class_filter_init_method)( - bt_self_component_filter *self_component, - const bt_value *params, void *init_method_data); - -typedef void (*bt_component_class_filter_finalize_method)( - bt_self_component_filter *self_component); - -typedef bt_self_message_iterator_status -(*bt_component_class_filter_message_iterator_init_method)( - bt_self_message_iterator *message_iterator, - bt_self_component_filter *self_component, - bt_self_component_port_output *port); - -typedef void -(*bt_component_class_filter_message_iterator_finalize_method)( - bt_self_message_iterator *message_iterator); - -typedef bt_self_message_iterator_status -(*bt_component_class_filter_message_iterator_next_method)( - bt_self_message_iterator *message_iterator, - bt_message_array_const msgs, uint64_t capacity, - uint64_t *count); - -typedef bt_self_message_iterator_status -(*bt_component_class_filter_message_iterator_seek_ns_from_origin_method)( - bt_self_message_iterator *message_iterator, - int64_t ns_from_origin); - -typedef bt_self_message_iterator_status -(*bt_component_class_filter_message_iterator_seek_beginning_method)( - bt_self_message_iterator *message_iterator); - -typedef bt_bool -(*bt_component_class_filter_message_iterator_can_seek_ns_from_origin_method)( - bt_self_message_iterator *message_iterator, - int64_t ns_from_origin); - -typedef bt_bool -(*bt_component_class_filter_message_iterator_can_seek_beginning_method)( - bt_self_message_iterator *message_iterator); - -typedef bt_query_status -(*bt_component_class_filter_query_method)( - bt_self_component_class_filter *comp_class, - const bt_query_executor *query_executor, - const char *object, const bt_value *params, - bt_logging_level log_level, - const bt_value **result); - -typedef bt_self_component_status -(*bt_component_class_filter_input_port_connected_method)( - bt_self_component_filter *self_component, - bt_self_component_port_input *self_port, - const bt_port_output *other_port); - -typedef bt_self_component_status -(*bt_component_class_filter_output_port_connected_method)( - bt_self_component_filter *self_component, - bt_self_component_port_output *self_port, - const bt_port_input *other_port); - -bt_component_class *bt_component_class_filter_as_component_class( - bt_component_class_filter *comp_cls_filter); - -extern -bt_component_class_filter *bt_component_class_filter_create( - const char *name, - bt_component_class_filter_message_iterator_next_method method); - -extern bt_component_class_status -bt_component_class_filter_set_init_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_init_method method); - -extern bt_component_class_status -bt_component_class_filter_set_finalize_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_finalize_method method); - -extern bt_component_class_status -bt_component_class_filter_set_input_port_connected_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_input_port_connected_method method); - -extern bt_component_class_status -bt_component_class_filter_set_output_port_connected_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_output_port_connected_method method); - -extern bt_component_class_status -bt_component_class_filter_set_query_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_query_method method); - -extern bt_component_class_status -bt_component_class_filter_set_message_iterator_init_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_message_iterator_init_method method); - -extern bt_component_class_status -bt_component_class_filter_set_message_iterator_finalize_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_message_iterator_finalize_method method); - -extern bt_component_class_status -bt_component_class_filter_set_message_iterator_seek_ns_from_origin_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_message_iterator_seek_ns_from_origin_method method); - -extern bt_component_class_status -bt_component_class_filter_set_message_iterator_seek_beginning_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_message_iterator_seek_beginning_method method); - -extern bt_bool -bt_component_class_filter_set_message_iterator_can_seek_ns_from_origin_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_message_iterator_can_seek_ns_from_origin_method method); - -extern bt_bool -bt_component_class_filter_set_message_iterator_can_seek_beginning_method( - bt_component_class_filter *comp_class, - bt_component_class_filter_message_iterator_can_seek_beginning_method method); - -/* From component-class-sink-const.h */ - -const bt_component_class * -bt_component_class_sink_as_component_class_const( - const bt_component_class_sink *comp_cls_sink); - -extern void bt_component_class_sink_get_ref( - const bt_component_class_sink *component_class_sink); - -extern void bt_component_class_sink_put_ref( - const bt_component_class_sink *component_class_sink); - -/* From component-class-sink.h */ - -typedef bt_self_component_status (*bt_component_class_sink_init_method)( - bt_self_component_sink *self_component, - const bt_value *params, void *init_method_data); - -typedef void (*bt_component_class_sink_finalize_method)( - bt_self_component_sink *self_component); - -typedef bt_query_status -(*bt_component_class_sink_query_method)( - bt_self_component_class_sink *comp_class, - const bt_query_executor *query_executor, - const char *object, const bt_value *params, - bt_logging_level log_level, - const bt_value **result); - -typedef bt_self_component_status -(*bt_component_class_sink_input_port_connected_method)( - bt_self_component_sink *self_component, - bt_self_component_port_input *self_port, - const bt_port_output *other_port); - -typedef bt_self_component_status -(*bt_component_class_sink_graph_is_configured_method)( - bt_self_component_sink *self_component); - -typedef bt_self_component_status (*bt_component_class_sink_consume_method)( - bt_self_component_sink *self_component); - -bt_component_class *bt_component_class_sink_as_component_class( - bt_component_class_sink *comp_cls_sink); - -extern -bt_component_class_sink *bt_component_class_sink_create( - const char *name, - bt_component_class_sink_consume_method method); - -extern bt_component_class_status bt_component_class_sink_set_init_method( - bt_component_class_sink *comp_class, - bt_component_class_sink_init_method method); - -extern bt_component_class_status bt_component_class_sink_set_finalize_method( - bt_component_class_sink *comp_class, - bt_component_class_sink_finalize_method method); - -extern bt_component_class_status -bt_component_class_sink_set_input_port_connected_method( - bt_component_class_sink *comp_class, - bt_component_class_sink_input_port_connected_method method); - -extern bt_component_class_status -bt_component_class_sink_set_graph_is_configured_method( - bt_component_class_sink *comp_class, - bt_component_class_sink_graph_is_configured_method method); - -extern bt_component_class_status bt_component_class_sink_set_query_method( - bt_component_class_sink *comp_class, - bt_component_class_sink_query_method method); - -/* From self-component-class-source.h */ - -const bt_component_class_source * -bt_self_component_class_source_as_component_class_source( - bt_self_component_class_source *self_comp_cls_source); - -/* From self-component-class-filter.h */ - -const bt_component_class_filter * -bt_self_component_class_filter_as_component_class_filter( - bt_self_component_class_filter *self_comp_cls_filter); - -/* From self-component-class-sink.h */ - -const bt_component_class_sink * -bt_self_component_class_sink_as_component_class_sink( - bt_self_component_class_sink *self_comp_cls_sink); +%include +%include +%include +%include +%include +%include +%include +%include +%include +%include +%include %{ /* @@ -427,7 +43,7 @@ bt_self_component_class_sink_as_component_class_sink( * to create a user Python component anyway. * * This hash table is written to when a user-defined Python component - * class is created by one of the bt_py3_component_class_*_create() + * class is created by one of the bt_bt2_component_class_*_create() * functions. * * This function is read from when a user calls bt_component_create() @@ -439,7 +55,8 @@ bt_self_component_class_sink_as_component_class_sink( static GHashTable *bt_cc_ptr_to_py_cls; -static void register_cc_ptr_to_py_cls(struct bt_component_class *bt_cc, +static +void register_cc_ptr_to_py_cls(struct bt_component_class *bt_cc, PyObject *py_cls) { if (!bt_cc_ptr_to_py_cls) { @@ -457,7 +74,8 @@ static void register_cc_ptr_to_py_cls(struct bt_component_class *bt_cc, (gpointer) py_cls); } -static PyObject *lookup_cc_ptr_to_py_cls(const bt_component_class *bt_cc) +static +PyObject *lookup_cc_ptr_to_py_cls(const bt_component_class *bt_cc) { if (!bt_cc_ptr_to_py_cls) { BT_LOGW("Cannot look up Python component class because hash table is NULL: " @@ -479,10 +97,12 @@ static PyObject *py_mod_bt2_exc_error_type = NULL; static PyObject *py_mod_bt2_exc_try_again_type = NULL; static PyObject *py_mod_bt2_exc_stop_type = NULL; static PyObject *py_mod_bt2_exc_msg_iter_canceled_type = NULL; -static PyObject *py_mod_bt2_exc_invalid_query_object_type = NULL; -static PyObject *py_mod_bt2_exc_invalid_query_params_type = NULL; +static PyObject *py_mod_bt2_exc_invalid_object_type = NULL; +static PyObject *py_mod_bt2_exc_invalid_params_type = NULL; +static PyObject *py_mod_bt2_exc_unsupported_type = NULL; -static void bt_py3_cc_init_from_bt2(void) +static +void bt_bt2_cc_init_from_bt2(void) { /* * This is called once the bt2 package is loaded. @@ -503,21 +123,25 @@ static void bt_py3_cc_init_from_bt2(void) py_mod_bt2_exc_stop_type = PyObject_GetAttrString(py_mod_bt2, "Stop"); BT_ASSERT(py_mod_bt2_exc_stop_type); - py_mod_bt2_exc_invalid_query_object_type = - PyObject_GetAttrString(py_mod_bt2, "InvalidQueryObject"); - BT_ASSERT(py_mod_bt2_exc_invalid_query_object_type); - py_mod_bt2_exc_invalid_query_params_type = - PyObject_GetAttrString(py_mod_bt2, "InvalidQueryParams"); - BT_ASSERT(py_mod_bt2_exc_invalid_query_params_type); + py_mod_bt2_exc_invalid_object_type = + PyObject_GetAttrString(py_mod_bt2, "InvalidObject"); + BT_ASSERT(py_mod_bt2_exc_invalid_object_type); + py_mod_bt2_exc_invalid_params_type = + PyObject_GetAttrString(py_mod_bt2, "InvalidParams"); + BT_ASSERT(py_mod_bt2_exc_invalid_params_type); + py_mod_bt2_exc_unsupported_type = + PyObject_GetAttrString(py_mod_bt2, "Unsupported"); + BT_ASSERT(py_mod_bt2_exc_unsupported_type); } -static void bt_py3_cc_exit_handler(void) +static +void bt_bt2_cc_exit_handler(void) { /* * This is an exit handler (set by the bt2 package). * * We only give back the references that we took in - * bt_py3_cc_init_from_bt2() here. The global variables continue + * bt_bt2_cc_init_from_bt2() here. The global variables continue * to exist for the code of this file, but they are now borrowed * references. If this code is executed, it means that somehow * the modules are still loaded, so it should be safe to use @@ -531,15 +155,16 @@ static void bt_py3_cc_exit_handler(void) Py_XDECREF(py_mod_bt2_exc_try_again_type); Py_XDECREF(py_mod_bt2_exc_stop_type); Py_XDECREF(py_mod_bt2_exc_msg_iter_canceled_type); - Py_XDECREF(py_mod_bt2_exc_invalid_query_object_type); - Py_XDECREF(py_mod_bt2_exc_invalid_query_params_type); + Py_XDECREF(py_mod_bt2_exc_invalid_object_type); + Py_XDECREF(py_mod_bt2_exc_invalid_params_type); } /* Library destructor */ __attribute__((destructor)) -static void bt_py3_native_comp_class_dtor(void) { +static +void native_comp_class_dtor(void) { /* Destroy component class association hash table */ if (bt_cc_ptr_to_py_cls) { BT_LOGD_STR("Destroying native component class to Python component class hash table."); @@ -547,145 +172,42 @@ static void bt_py3_native_comp_class_dtor(void) { } } - -// TODO: maybe we can wrap code in the Python methods (e.g. _query_from_native) -// in a try catch and print the error there instead, it would be simpler. -static -void bt2_py_loge_exception(void) +static inline +void log_exception(int log_level) { - PyObject *type = NULL; - PyObject *value = NULL; - PyObject *traceback = NULL; - PyObject *traceback_module = NULL; - PyObject *format_exception_func = NULL; - PyObject *exc_str_list = NULL; - GString *msg_buf = NULL; - Py_ssize_t i; + GString *gstr; BT_ASSERT(PyErr_Occurred() != NULL); - - PyErr_Fetch(&type, &value, &traceback); - - BT_ASSERT(type != NULL); - - /* - * traceback can be NULL, when we fail to call a Python function from the - * native code (there is no Python stack at that point). E.g.: - * - * TypeError: _query_from_native() takes 5 positional arguments but 8 were given - */ - - - /* Make sure `value` is what we expected - an instance of `type`. */ - PyErr_NormalizeException(&type, &value, &traceback); - - traceback_module = PyImport_ImportModule("traceback"); - if (!traceback_module) { - BT_LOGE_STR("Failed to log Python exception (could not import traceback module)."); + gstr = bt_py_common_format_exception(BT_LOG_OUTPUT_LEVEL); + if (!gstr) { + /* bt_py_common_format_exception() logs errors */ goto end; } - format_exception_func = PyObject_GetAttrString(traceback_module, - traceback ? "format_exception" : "format_exception_only"); - if (!format_exception_func) { - BT_LOGE_STR("Failed to log Python exception (could not find format_exception)."); - goto end; - } - - if (!PyCallable_Check(format_exception_func)) { - BT_LOGE_STR("Failed to log Python exception (format_exception is not callable)."); - goto end; - } - - exc_str_list = PyObject_CallFunctionObjArgs(format_exception_func, type, value, traceback, NULL); - if (!exc_str_list) { - PyErr_Print(); - BT_LOGE_STR("Failed to log Python exception (call to format_exception failed)."); - goto end; - } - - msg_buf = g_string_new(NULL); - - for (i = 0; i < PyList_Size(exc_str_list); i++) { - PyObject *exc_str = PyList_GetItem(exc_str_list, i); - const char *str = PyUnicode_AsUTF8(exc_str); - if (!str) { - BT_LOGE_STR("Failed to log Python exception (failed to convert exception to string)."); - goto end; - } - - g_string_append(msg_buf, str); - } - - BT_LOGE_STR(msg_buf->str); + BT_LOG_WRITE(log_level, BT_LOG_TAG, "%s", gstr->str); end: - if (msg_buf) { - g_string_free(msg_buf, TRUE); + if (gstr) { + g_string_free(gstr, TRUE); } - Py_XDECREF(exc_str_list); - Py_XDECREF(format_exception_func); - Py_XDECREF(traceback_module); - - /* PyErr_Restore takes our references. */ - PyErr_Restore(type, value, traceback); } -static bt_self_component_status bt_py3_exc_to_self_component_status(void) +static inline +void loge_exception(void) { - bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK; - PyObject *exc = PyErr_Occurred(); - - if (!exc) { - goto end; - } - - if (PyErr_GivenExceptionMatches(exc, - py_mod_bt2_exc_try_again_type)) { - status = BT_SELF_COMPONENT_STATUS_AGAIN; - } else if (PyErr_GivenExceptionMatches(exc, - py_mod_bt2_exc_stop_type)) { - status = BT_SELF_COMPONENT_STATUS_END; - } else { - bt2_py_loge_exception(); - status = BT_SELF_COMPONENT_STATUS_ERROR; - } - -end: - PyErr_Clear(); - return status; + log_exception(BT_LOG_ERROR); } -/* Component class proxy methods (delegate to the attached Python object) */ - -static bt_self_message_iterator_status -bt_py3_exc_to_self_message_iterator_status(void) +static inline +void logw_exception(void) { - enum bt_self_message_iterator_status status = - BT_SELF_MESSAGE_ITERATOR_STATUS_OK; - PyObject *exc = PyErr_Occurred(); - - if (!exc) { - goto end; - } - - if (PyErr_GivenExceptionMatches(exc, py_mod_bt2_exc_stop_type)) { - status = BT_SELF_MESSAGE_ITERATOR_STATUS_END; - } else if (PyErr_GivenExceptionMatches(exc, py_mod_bt2_exc_try_again_type)) { - status = BT_SELF_MESSAGE_ITERATOR_STATUS_AGAIN; - } else { - bt2_py_loge_exception(); - status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR; - } - -end: - PyErr_Clear(); - return status; + log_exception(BT_LOG_WARN); } -static enum bt_query_status bt_py3_exc_to_query_status(void) +static inline +int py_exc_to_status(void) { - enum bt_query_status status = BT_QUERY_STATUS_OK; + int status = __BT_FUNC_STATUS_OK; PyObject *exc = PyErr_Occurred(); if (!exc) { @@ -693,17 +215,24 @@ static enum bt_query_status bt_py3_exc_to_query_status(void) } if (PyErr_GivenExceptionMatches(exc, - py_mod_bt2_exc_invalid_query_object_type)) { - status = BT_QUERY_STATUS_INVALID_OBJECT; + py_mod_bt2_exc_try_again_type)) { + status = __BT_FUNC_STATUS_AGAIN; } else if (PyErr_GivenExceptionMatches(exc, - py_mod_bt2_exc_invalid_query_params_type)) { - status = BT_QUERY_STATUS_INVALID_PARAMS; + py_mod_bt2_exc_stop_type)) { + status = __BT_FUNC_STATUS_END; } else if (PyErr_GivenExceptionMatches(exc, - py_mod_bt2_exc_try_again_type)) { - status = BT_QUERY_STATUS_AGAIN; + py_mod_bt2_exc_invalid_object_type)) { + status = __BT_FUNC_STATUS_INVALID_OBJECT; + } else if (PyErr_GivenExceptionMatches(exc, + py_mod_bt2_exc_invalid_params_type)) { + status = __BT_FUNC_STATUS_INVALID_PARAMS; + } else if (PyErr_GivenExceptionMatches(exc, + py_mod_bt2_exc_unsupported_type)) { + status = __BT_FUNC_STATUS_UNSUPPORTED; } else { - bt2_py_loge_exception(); - status = BT_QUERY_STATUS_ERROR; + /* Unknown exception: convert to general error */ + logw_exception(); + status = __BT_FUNC_STATUS_ERROR; } end: @@ -711,8 +240,10 @@ end: return status; } -static bt_self_component_status -bt_py3_component_class_init( +/* Component class proxy methods (delegate to the attached Python object) */ + +static +bt_component_class_init_method_status component_class_init( bt_self_component *self_component, void *self_component_v, swig_type_info *self_comp_cls_type_swig_type, @@ -721,7 +252,7 @@ bt_py3_component_class_init( { const bt_component *component = bt_self_component_as_component(self_component); const bt_component_class *component_class = bt_component_borrow_class_const(component); - bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK; + bt_component_class_init_method_status status = __BT_FUNC_STATUS_OK; PyObject *py_cls = NULL; PyObject *py_comp = NULL; PyObject *py_params_ptr = NULL; @@ -771,10 +302,9 @@ bt_py3_component_class_init( py_comp = PyObject_CallMethod(py_cls, "_init_from_native", "(OO)", py_comp_ptr, py_params_ptr); if (!py_comp) { - bt2_py_loge_exception(); - BT_LOGE("Failed to call Python class's _init_from_native() method: " + BT_LOGW("Failed to call Python class's _init_from_native() method: " "py-cls-addr=%p", py_cls); - + logw_exception(); goto error; } @@ -788,7 +318,7 @@ bt_py3_component_class_init( goto end; error: - status = BT_SELF_COMPONENT_STATUS_ERROR; + status = __BT_FUNC_STATUS_ERROR; /* * Clear any exception: we're returning a bad status anyway. If @@ -810,43 +340,47 @@ end: * of that class. */ -static bt_self_component_status -bt_py3_component_class_source_init(bt_self_component_source *self_component_source, +static +bt_component_class_init_method_status component_class_source_init( + bt_self_component_source *self_component_source, const bt_value *params, void *init_method_data) { bt_self_component *self_component = bt_self_component_source_as_self_component(self_component_source); - return bt_py3_component_class_init( + return component_class_init( self_component, self_component_source, SWIGTYPE_p_bt_self_component_source, params, init_method_data); } -static bt_self_component_status -bt_py3_component_class_filter_init(bt_self_component_filter *self_component_filter, +static +bt_component_class_init_method_status component_class_filter_init( + bt_self_component_filter *self_component_filter, const bt_value *params, void *init_method_data) { bt_self_component *self_component = bt_self_component_filter_as_self_component(self_component_filter); - return bt_py3_component_class_init( + return component_class_init( self_component, self_component_filter, SWIGTYPE_p_bt_self_component_filter, params, init_method_data); } -static bt_self_component_status -bt_py3_component_class_sink_init(bt_self_component_sink *self_component_sink, +static +bt_component_class_init_method_status component_class_sink_init( + bt_self_component_sink *self_component_sink, const bt_value *params, void *init_method_data) { bt_self_component *self_component = bt_self_component_sink_as_self_component(self_component_sink); - return bt_py3_component_class_init( + return component_class_init( self_component, self_component_sink, SWIGTYPE_p_bt_self_component_sink, params, init_method_data); } -static void bt_py3_component_class_finalize(bt_self_component *self_component) +static +void component_class_finalize(bt_self_component *self_component) { PyObject *py_comp = bt_self_component_get_data(self_component); BT_ASSERT(py_comp); @@ -856,7 +390,8 @@ static void bt_py3_component_class_finalize(bt_self_component *self_component) "_finalize", NULL); if (PyErr_Occurred()) { - BT_LOGW("User's _finalize() method raised an exception: ignoring."); + BT_LOGW("User component's _finalize() method raised an exception: ignoring:"); + logw_exception(); } /* @@ -869,29 +404,78 @@ static void bt_py3_component_class_finalize(bt_self_component *self_component) Py_DECREF(py_comp); } -static void -bt_py3_component_class_source_finalize(bt_self_component_source *self_component_source) +static +void component_class_source_finalize(bt_self_component_source *self_component_source) { bt_self_component *self_component = bt_self_component_source_as_self_component(self_component_source); - bt_py3_component_class_finalize(self_component); + component_class_finalize(self_component); } -static void -bt_py3_component_class_filter_finalize(bt_self_component_filter *self_component_filter) +static +void component_class_filter_finalize(bt_self_component_filter *self_component_filter) { bt_self_component *self_component = bt_self_component_filter_as_self_component(self_component_filter); - bt_py3_component_class_finalize(self_component); + component_class_finalize(self_component); } -static void -bt_py3_component_class_sink_finalize(bt_self_component_sink *self_component_sink) +static +void component_class_sink_finalize(bt_self_component_sink *self_component_sink) { bt_self_component *self_component = bt_self_component_sink_as_self_component(self_component_sink); - bt_py3_component_class_finalize(self_component); + component_class_finalize(self_component); +} + +static +bt_bool component_class_can_seek_beginning( + bt_self_message_iterator *self_message_iterator) +{ + PyObject *py_iter; + PyObject *py_result = NULL; + bt_bool can_seek_beginning = false; + + py_iter = bt_self_message_iterator_get_data(self_message_iterator); + BT_ASSERT(py_iter); + + py_result = PyObject_GetAttrString(py_iter, "_can_seek_beginning_from_native"); + + BT_ASSERT(!py_result || PyBool_Check(py_result)); + + if (py_result) { + can_seek_beginning = PyObject_IsTrue(py_result); + } else { + /* + * Once can_seek_beginning can report errors, convert the + * exception to a status. For now, log and return false; + */ + loge_exception(); + PyErr_Clear(); + } + + Py_XDECREF(py_result); + + return can_seek_beginning; } -static bt_self_component_status -bt_py3_component_class_port_connected( +static +bt_component_class_message_iterator_seek_beginning_method_status +component_class_seek_beginning(bt_self_message_iterator *self_message_iterator) +{ + PyObject *py_iter; + PyObject *py_result; + bt_component_class_message_iterator_seek_beginning_method_status status; + + py_iter = bt_self_message_iterator_get_data(self_message_iterator); + BT_ASSERT(py_iter); + py_result = PyObject_CallMethod(py_iter, "_seek_beginning_from_native", + NULL); + BT_ASSERT(!py_result || py_result == Py_None); + status = py_exc_to_status(); + Py_XDECREF(py_result); + return status; +} + +static +bt_component_class_port_connected_method_status component_class_port_connected( bt_self_component *self_component, void *self_component_port, swig_type_info *self_component_port_swig_type, @@ -899,7 +483,7 @@ bt_py3_component_class_port_connected( const void *other_port, swig_type_info *other_port_swig_type) { - bt_self_component_status status; + bt_component_class_port_connected_method_status status; PyObject *py_comp = NULL; PyObject *py_self_port_ptr = NULL; PyObject *py_other_port_ptr = NULL; @@ -907,12 +491,11 @@ bt_py3_component_class_port_connected( py_comp = bt_self_component_get_data(self_component); BT_ASSERT(py_comp); - py_self_port_ptr = SWIG_NewPointerObj(SWIG_as_voidptr(self_component_port), self_component_port_swig_type, 0); if (!py_self_port_ptr) { BT_LOGF_STR("Failed to create a SWIG pointer object."); - status = BT_SELF_COMPONENT_STATUS_NOMEM; + status = __BT_FUNC_STATUS_MEMORY_ERROR; goto end; } @@ -920,34 +503,32 @@ bt_py3_component_class_port_connected( other_port_swig_type, 0); if (!py_other_port_ptr) { BT_LOGF_STR("Failed to create a SWIG pointer object."); - status = BT_SELF_COMPONENT_STATUS_NOMEM; + status = __BT_FUNC_STATUS_MEMORY_ERROR; goto end; } py_method_result = PyObject_CallMethod(py_comp, "_port_connected_from_native", "(OiO)", py_self_port_ptr, self_component_port_type, py_other_port_ptr); - BT_ASSERT(!py_method_result || py_method_result == Py_None); - - status = bt_py3_exc_to_self_component_status(); + status = py_exc_to_status(); end: Py_XDECREF(py_self_port_ptr); Py_XDECREF(py_other_port_ptr); Py_XDECREF(py_method_result); - return status; } -static bt_self_component_status -bt_py3_component_class_source_output_port_connected( +static +bt_component_class_port_connected_method_status +component_class_source_output_port_connected( bt_self_component_source *self_component_source, bt_self_component_port_output *self_component_port_output, const bt_port_input *other_port_input) { bt_self_component *self_component = bt_self_component_source_as_self_component(self_component_source); - return bt_py3_component_class_port_connected( + return component_class_port_connected( self_component, self_component_port_output, SWIGTYPE_p_bt_self_component_port_output, @@ -956,15 +537,16 @@ bt_py3_component_class_source_output_port_connected( SWIGTYPE_p_bt_port_input); } -static bt_self_component_status -bt_py3_component_class_filter_input_port_connected( +static +bt_component_class_port_connected_method_status +component_class_filter_input_port_connected( bt_self_component_filter *self_component_filter, bt_self_component_port_input *self_component_port_input, const bt_port_output *other_port_output) { bt_self_component *self_component = bt_self_component_filter_as_self_component(self_component_filter); - return bt_py3_component_class_port_connected( + return component_class_port_connected( self_component, self_component_port_input, SWIGTYPE_p_bt_self_component_port_input, @@ -973,15 +555,16 @@ bt_py3_component_class_filter_input_port_connected( SWIGTYPE_p_bt_port_output); } -static bt_self_component_status -bt_py3_component_class_filter_output_port_connected( +static +bt_component_class_port_connected_method_status +component_class_filter_output_port_connected( bt_self_component_filter *self_component_filter, bt_self_component_port_output *self_component_port_output, const bt_port_input *other_port_input) { bt_self_component *self_component = bt_self_component_filter_as_self_component(self_component_filter); - return bt_py3_component_class_port_connected( + return component_class_port_connected( self_component, self_component_port_output, SWIGTYPE_p_bt_self_component_port_output, @@ -990,15 +573,16 @@ bt_py3_component_class_filter_output_port_connected( SWIGTYPE_p_bt_port_input); } -static bt_self_component_status -bt_py3_component_class_sink_input_port_connected( +static +bt_component_class_port_connected_method_status +component_class_sink_input_port_connected( bt_self_component_sink *self_component_sink, bt_self_component_port_input *self_component_port_input, const bt_port_output *other_port_output) { bt_self_component *self_component = bt_self_component_sink_as_self_component(self_component_sink); - return bt_py3_component_class_port_connected( + return component_class_port_connected( self_component, self_component_port_input, SWIGTYPE_p_bt_self_component_port_input, @@ -1007,29 +591,27 @@ bt_py3_component_class_sink_input_port_connected( SWIGTYPE_p_bt_port_output); } -static bt_self_component_status -bt_py3_component_class_sink_graph_is_configured(bt_self_component_sink *self_component_sink) +static +bt_component_class_sink_graph_is_configured_method_status +component_class_sink_graph_is_configured( + bt_self_component_sink *self_component_sink) { PyObject *py_comp = NULL; PyObject *py_method_result = NULL; - bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK; + bt_component_class_sink_graph_is_configured_method_status status = __BT_FUNC_STATUS_OK; bt_self_component *self_component = bt_self_component_sink_as_self_component(self_component_sink); py_comp = bt_self_component_get_data(self_component); py_method_result = PyObject_CallMethod(py_comp, "_graph_is_configured_from_native", NULL); - BT_ASSERT(!py_method_result || py_method_result == Py_None); - - status = bt_py3_exc_to_self_component_status(); - + status = py_exc_to_status(); Py_XDECREF(py_method_result); - return status; } -static bt_query_status -bt_py3_component_class_query( +static +bt_component_class_query_method_status component_class_query( const bt_component_class *component_class, const bt_query_executor *query_executor, const char *object, const bt_value *params, @@ -1042,7 +624,7 @@ bt_py3_component_class_query( PyObject *py_query_func = NULL; PyObject *py_object = NULL; PyObject *py_results_addr = NULL; - bt_query_status status = BT_QUERY_STATUS_OK; + bt_component_class_query_method_status status = __BT_FUNC_STATUS_OK; py_cls = lookup_cc_ptr_to_py_cls(component_class); if (!py_cls) { @@ -1074,11 +656,10 @@ bt_py3_component_class_query( py_results_addr = PyObject_CallMethod(py_cls, "_query_from_native", "(OOOi)", py_query_exec_ptr, py_object, py_params_ptr, (int) log_level); - if (!py_results_addr) { - BT_LOGE("Failed to call Python class's _query_from_native() method: " + BT_LOGW("Failed to call Python class's _query_from_native() method: " "py-cls-addr=%p", py_cls); - status = bt_py3_exc_to_query_status(); + status = py_exc_to_status(); goto end; } @@ -1094,7 +675,7 @@ bt_py3_component_class_query( error: PyErr_Clear(); - status = BT_QUERY_STATUS_ERROR; + status = __BT_FUNC_STATUS_ERROR; end: Py_XDECREF(py_params_ptr); @@ -1105,8 +686,8 @@ end: return status; } -static bt_query_status -bt_py3_component_class_source_query( +static +bt_component_class_query_method_status component_class_source_query( bt_self_component_class_source *self_component_class_source, const bt_query_executor *query_executor, const char *object, const bt_value *params, @@ -1115,11 +696,12 @@ bt_py3_component_class_source_query( { const bt_component_class_source *component_class_source = bt_self_component_class_source_as_component_class_source(self_component_class_source); const bt_component_class *component_class = bt_component_class_source_as_component_class_const(component_class_source); - return bt_py3_component_class_query(component_class, query_executor, object, params, log_level, result); + + return component_class_query(component_class, query_executor, object, params, log_level, result); } -static bt_query_status -bt_py3_component_class_filter_query( +static +bt_component_class_query_method_status component_class_filter_query( bt_self_component_class_filter *self_component_class_filter, const bt_query_executor *query_executor, const char *object, const bt_value *params, @@ -1128,11 +710,12 @@ bt_py3_component_class_filter_query( { const bt_component_class_filter *component_class_filter = bt_self_component_class_filter_as_component_class_filter(self_component_class_filter); const bt_component_class *component_class = bt_component_class_filter_as_component_class_const(component_class_filter); - return bt_py3_component_class_query(component_class, query_executor, object, params, log_level, result); + + return component_class_query(component_class, query_executor, object, params, log_level, result); } -static bt_query_status -bt_py3_component_class_sink_query( +static +bt_component_class_query_method_status component_class_sink_query( bt_self_component_class_sink *self_component_class_sink, const bt_query_executor *query_executor, const char *object, const bt_value *params, @@ -1141,16 +724,18 @@ bt_py3_component_class_sink_query( { const bt_component_class_sink *component_class_sink = bt_self_component_class_sink_as_component_class_sink(self_component_class_sink); const bt_component_class *component_class = bt_component_class_sink_as_component_class_const(component_class_sink); - return bt_py3_component_class_query(component_class, query_executor, object, params, log_level, result); + + return component_class_query(component_class, query_executor, object, params, log_level, result); } -static bt_self_message_iterator_status -bt_py3_component_class_message_iterator_init( +static +bt_component_class_message_iterator_init_method_status +component_class_message_iterator_init( bt_self_message_iterator *self_message_iterator, bt_self_component *self_component, bt_self_component_port_output *self_component_port_output) { - bt_self_message_iterator_status status = BT_SELF_MESSAGE_ITERATOR_STATUS_OK; + bt_component_class_message_iterator_init_method_status status = __BT_FUNC_STATUS_OK; PyObject *py_comp_cls = NULL; PyObject *py_iter_cls = NULL; PyObject *py_iter_ptr = NULL; @@ -1192,7 +777,7 @@ bt_py3_component_class_message_iterator_init( if (!py_iter) { BT_LOGE("Failed to call Python class's __new__() method: " "py-cls-addr=%p", py_iter_cls); - bt2_py_loge_exception(); + loge_exception(); goto error; } @@ -1208,17 +793,19 @@ bt_py3_component_class_message_iterator_init( * user Python component object from which the iterator was * created). */ - py_component_port_output_ptr = SWIG_NewPointerObj(SWIG_as_voidptr(self_component_port_output), + py_component_port_output_ptr = SWIG_NewPointerObj( + SWIG_as_voidptr(self_component_port_output), SWIGTYPE_p_bt_self_component_port_output, 0); if (!py_component_port_output_ptr) { BT_LOGE_STR("Failed to create a SWIG pointer object."); goto error; } - py_init_method_result = PyObject_CallMethod(py_iter, "_init_from_native", "O", py_component_port_output_ptr); + py_init_method_result = PyObject_CallMethod(py_iter, + "_init_from_native", "O", py_component_port_output_ptr); if (!py_init_method_result) { - BT_LOGE_STR("User's __init__() method failed."); - bt2_py_loge_exception(); + BT_LOGW_STR("User's __init__() method failed:"); + logw_exception(); goto error; } @@ -1245,13 +832,13 @@ bt_py3_component_class_message_iterator_init( goto end; error: - status = bt_py3_exc_to_self_message_iterator_status(); - if (status == BT_SELF_MESSAGE_ITERATOR_STATUS_OK) { + status = py_exc_to_status(); + if (status == __BT_FUNC_STATUS_OK) { /* * Looks like there wasn't any exception from the Python * side, but we're still in an error state here. */ - status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR; + status = __BT_FUNC_STATUS_ERROR; } /* @@ -1271,28 +858,32 @@ end: return status; } -static bt_self_message_iterator_status -bt_py3_component_class_source_message_iterator_init( +static +bt_component_class_message_iterator_init_method_status +component_class_source_message_iterator_init( bt_self_message_iterator *self_message_iterator, bt_self_component_source *self_component_source, bt_self_component_port_output *self_component_port_output) { bt_self_component *self_component = bt_self_component_source_as_self_component(self_component_source); - return bt_py3_component_class_message_iterator_init(self_message_iterator, self_component, self_component_port_output); + + return component_class_message_iterator_init(self_message_iterator, self_component, self_component_port_output); } -static bt_self_message_iterator_status -bt_py3_component_class_filter_message_iterator_init( +static +bt_component_class_message_iterator_init_method_status +component_class_filter_message_iterator_init( bt_self_message_iterator *self_message_iterator, bt_self_component_filter *self_component_filter, bt_self_component_port_output *self_component_port_output) { bt_self_component *self_component = bt_self_component_filter_as_self_component(self_component_filter); - return bt_py3_component_class_message_iterator_init(self_message_iterator, self_component, self_component_port_output); + + return component_class_message_iterator_init(self_message_iterator, self_component, self_component_port_output); } -static void -bt_py3_component_class_message_iterator_finalize( +static +void component_class_message_iterator_finalize( bt_self_message_iterator *message_iterator) { PyObject *py_message_iter = bt_self_message_iterator_get_data(message_iterator); @@ -1305,7 +896,8 @@ bt_py3_component_class_message_iterator_finalize( "_finalize", NULL); if (PyErr_Occurred()) { - BT_LOGW("User's _finalize() method raised an exception: ignoring."); + BT_LOGW("User's _finalize() method raised an exception: ignoring:"); + logw_exception(); } /* @@ -1320,13 +912,14 @@ bt_py3_component_class_message_iterator_finalize( /* Valid for both sources and filters. */ -static bt_self_message_iterator_status -bt_py3_component_class_message_iterator_next( - bt_self_message_iterator *message_iterator, - bt_message_array_const msgs, uint64_t capacity, - uint64_t *count) +static +bt_component_class_message_iterator_next_method_status +component_class_message_iterator_next( + bt_self_message_iterator *message_iterator, + bt_message_array_const msgs, uint64_t capacity, + uint64_t *count) { - bt_self_message_iterator_status status = BT_SELF_MESSAGE_ITERATOR_STATUS_OK; + bt_component_class_message_iterator_next_method_status status = __BT_FUNC_STATUS_OK; PyObject *py_message_iter = bt_self_message_iterator_get_data(message_iterator); PyObject *py_method_result = NULL; @@ -1334,8 +927,8 @@ bt_py3_component_class_message_iterator_next( py_method_result = PyObject_CallMethod(py_message_iter, "_next_from_native", NULL); if (!py_method_result) { - status = bt_py3_exc_to_self_message_iterator_status(); - BT_ASSERT(status != BT_SELF_MESSAGE_ITERATOR_STATUS_OK); + status = py_exc_to_status(); + BT_ASSERT(status != __BT_FUNC_STATUS_OK); goto end; } @@ -1356,24 +949,24 @@ end: return status; } -static bt_self_component_status -bt_py3_component_class_sink_consume(bt_self_component_sink *self_component_sink) +static +bt_component_class_sink_consume_method_status +component_class_sink_consume(bt_self_component_sink *self_component_sink) { bt_self_component *self_component = bt_self_component_sink_as_self_component(self_component_sink); PyObject *py_comp = bt_self_component_get_data(self_component); PyObject *py_method_result = NULL; - bt_self_component_status status; + bt_component_class_sink_consume_method_status status; BT_ASSERT(py_comp); py_method_result = PyObject_CallMethod(py_comp, "_consume", NULL); - - status = bt_py3_exc_to_self_component_status(); - if (!py_method_result && status == BT_SELF_COMPONENT_STATUS_OK) { + status = py_exc_to_status(); + if (!py_method_result && status == __BT_FUNC_STATUS_OK) { /* Pretty sure this should never happen, but just in case */ BT_LOGE("User's _consume() method failed without raising an exception: " "status=%d", status); - status = BT_SELF_COMPONENT_STATUS_ERROR; + status = __BT_FUNC_STATUS_ERROR; } Py_XDECREF(py_method_result); @@ -1381,7 +974,7 @@ bt_py3_component_class_sink_consume(bt_self_component_sink *self_component_sink) } static -int bt_py3_component_class_set_help_and_desc( +int component_class_set_help_and_desc( bt_component_class *component_class, const char *description, const char *help) { @@ -1412,7 +1005,7 @@ end: } static -bt_component_class_source *bt_py3_component_class_source_create( +bt_component_class_source *bt_bt2_component_class_source_create( PyObject *py_cls, const char *name, const char *description, const char *help) { @@ -1421,9 +1014,8 @@ bt_component_class_source *bt_py3_component_class_source_create( int ret; BT_ASSERT(py_cls); - component_class_source = bt_component_class_source_create(name, - bt_py3_component_class_message_iterator_next); + component_class_message_iterator_next); if (!component_class_source) { BT_LOGE_STR("Cannot create source component class."); goto end; @@ -1431,26 +1023,31 @@ bt_component_class_source *bt_py3_component_class_source_create( component_class = bt_component_class_source_as_component_class(component_class_source); - if (bt_py3_component_class_set_help_and_desc(component_class, description, help)) { + if (component_class_set_help_and_desc(component_class, description, help)) { goto end; } - ret = bt_component_class_source_set_init_method(component_class_source, bt_py3_component_class_source_init); + ret = bt_component_class_source_set_init_method(component_class_source, component_class_source_init); BT_ASSERT(ret == 0); - ret = bt_component_class_source_set_finalize_method (component_class_source, bt_py3_component_class_source_finalize); + ret = bt_component_class_source_set_finalize_method(component_class_source, component_class_source_finalize); + BT_ASSERT(ret == 0); + ret = bt_component_class_source_set_message_iterator_can_seek_beginning_method(component_class_source, + component_class_can_seek_beginning); + BT_ASSERT(ret == 0); + ret = bt_component_class_source_set_message_iterator_seek_beginning_method(component_class_source, + component_class_seek_beginning); BT_ASSERT(ret == 0); ret = bt_component_class_source_set_output_port_connected_method(component_class_source, - bt_py3_component_class_source_output_port_connected); + component_class_source_output_port_connected); BT_ASSERT(ret == 0); - ret = bt_component_class_source_set_query_method(component_class_source, bt_py3_component_class_source_query); + ret = bt_component_class_source_set_query_method(component_class_source, component_class_source_query); BT_ASSERT(ret == 0); ret = bt_component_class_source_set_message_iterator_init_method( - component_class_source, bt_py3_component_class_source_message_iterator_init); + component_class_source, component_class_source_message_iterator_init); BT_ASSERT(ret == 0); ret = bt_component_class_source_set_message_iterator_finalize_method( - component_class_source, bt_py3_component_class_message_iterator_finalize); + component_class_source, component_class_message_iterator_finalize); BT_ASSERT(ret == 0); - register_cc_ptr_to_py_cls(component_class, py_cls); end: @@ -1458,7 +1055,7 @@ end: } static -bt_component_class_filter *bt_py3_component_class_filter_create( +bt_component_class_filter *bt_bt2_component_class_filter_create( PyObject *py_cls, const char *name, const char *description, const char *help) { @@ -1467,9 +1064,8 @@ bt_component_class_filter *bt_py3_component_class_filter_create( int ret; BT_ASSERT(py_cls); - component_class_filter = bt_component_class_filter_create(name, - bt_py3_component_class_message_iterator_next); + component_class_message_iterator_next); if (!component_class_filter) { BT_LOGE_STR("Cannot create filter component class."); goto end; @@ -1477,29 +1073,34 @@ bt_component_class_filter *bt_py3_component_class_filter_create( component_class = bt_component_class_filter_as_component_class(component_class_filter); - if (bt_py3_component_class_set_help_and_desc(component_class, description, help)) { + if (component_class_set_help_and_desc(component_class, description, help)) { goto end; } - ret = bt_component_class_filter_set_init_method(component_class_filter, bt_py3_component_class_filter_init); + ret = bt_component_class_filter_set_init_method(component_class_filter, component_class_filter_init); + BT_ASSERT(ret == 0); + ret = bt_component_class_filter_set_finalize_method (component_class_filter, component_class_filter_finalize); + BT_ASSERT(ret == 0); + ret = bt_component_class_filter_set_message_iterator_can_seek_beginning_method(component_class_filter, + component_class_can_seek_beginning); BT_ASSERT(ret == 0); - ret = bt_component_class_filter_set_finalize_method (component_class_filter, bt_py3_component_class_filter_finalize); + ret = bt_component_class_filter_set_message_iterator_seek_beginning_method(component_class_filter, + component_class_seek_beginning); BT_ASSERT(ret == 0); ret = bt_component_class_filter_set_input_port_connected_method(component_class_filter, - bt_py3_component_class_filter_input_port_connected); + component_class_filter_input_port_connected); BT_ASSERT(ret == 0); ret = bt_component_class_filter_set_output_port_connected_method(component_class_filter, - bt_py3_component_class_filter_output_port_connected); + component_class_filter_output_port_connected); BT_ASSERT(ret == 0); - ret = bt_component_class_filter_set_query_method(component_class_filter, bt_py3_component_class_filter_query); + ret = bt_component_class_filter_set_query_method(component_class_filter, component_class_filter_query); BT_ASSERT(ret == 0); ret = bt_component_class_filter_set_message_iterator_init_method( - component_class_filter, bt_py3_component_class_filter_message_iterator_init); + component_class_filter, component_class_filter_message_iterator_init); BT_ASSERT(ret == 0); ret = bt_component_class_filter_set_message_iterator_finalize_method( - component_class_filter, bt_py3_component_class_message_iterator_finalize); + component_class_filter, component_class_message_iterator_finalize); BT_ASSERT(ret == 0); - register_cc_ptr_to_py_cls(component_class, py_cls); end: @@ -1507,7 +1108,7 @@ end: } static -bt_component_class_sink *bt_py3_component_class_sink_create( +bt_component_class_sink *bt_bt2_component_class_sink_create( PyObject *py_cls, const char *name, const char *description, const char *help) { @@ -1516,8 +1117,7 @@ bt_component_class_sink *bt_py3_component_class_sink_create( int ret; BT_ASSERT(py_cls); - - component_class_sink = bt_component_class_sink_create(name, bt_py3_component_class_sink_consume); + component_class_sink = bt_component_class_sink_create(name, component_class_sink_consume); if (!component_class_sink) { BT_LOGE_STR("Cannot create sink component class."); @@ -1526,23 +1126,22 @@ bt_component_class_sink *bt_py3_component_class_sink_create( component_class = bt_component_class_sink_as_component_class(component_class_sink); - if (bt_py3_component_class_set_help_and_desc(component_class, description, help)) { + if (component_class_set_help_and_desc(component_class, description, help)) { goto end; } - ret = bt_component_class_sink_set_init_method(component_class_sink, bt_py3_component_class_sink_init); + ret = bt_component_class_sink_set_init_method(component_class_sink, component_class_sink_init); BT_ASSERT(ret == 0); - ret = bt_component_class_sink_set_finalize_method(component_class_sink, bt_py3_component_class_sink_finalize); + ret = bt_component_class_sink_set_finalize_method(component_class_sink, component_class_sink_finalize); BT_ASSERT(ret == 0); ret = bt_component_class_sink_set_input_port_connected_method(component_class_sink, - bt_py3_component_class_sink_input_port_connected); + component_class_sink_input_port_connected); BT_ASSERT(ret == 0); ret = bt_component_class_sink_set_graph_is_configured_method(component_class_sink, - bt_py3_component_class_sink_graph_is_configured); + component_class_sink_graph_is_configured); BT_ASSERT(ret == 0); - ret = bt_component_class_sink_set_query_method(component_class_sink, bt_py3_component_class_sink_query); + ret = bt_component_class_sink_set_query_method(component_class_sink, component_class_sink_query); BT_ASSERT(ret == 0); - register_cc_ptr_to_py_cls(component_class, py_cls); end: @@ -1550,14 +1149,14 @@ end: } %} -struct bt_component_class_source *bt_py3_component_class_source_create( +struct bt_component_class_source *bt_bt2_component_class_source_create( PyObject *py_cls, const char *name, const char *description, const char *help); -struct bt_component_class_filter *bt_py3_component_class_filter_create( +struct bt_component_class_filter *bt_bt2_component_class_filter_create( PyObject *py_cls, const char *name, const char *description, const char *help); -struct bt_component_class_sink *bt_py3_component_class_sink_create( +struct bt_component_class_sink *bt_bt2_component_class_sink_create( PyObject *py_cls, const char *name, const char *description, const char *help); -void bt_py3_cc_init_from_bt2(void); -void bt_py3_cc_exit_handler(void); +void bt_bt2_cc_init_from_bt2(void); +void bt_bt2_cc_exit_handler(void);