[babeltrace.git] / src / bindings / python / bt2 / bt2 / native_bt_graph.i.h

/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

static
void graph_listener_removed(void *py_callable)
{
	BT_ASSERT(py_callable);
	Py_DECREF(py_callable);
}

static bt_graph_listener_func_status port_added_listener(
	const void *component,
	swig_type_info *component_swig_type,
	bt_component_class_type component_class_type,
	const void *port,
	swig_type_info *port_swig_type,
	bt_port_type port_type,
	void *py_callable)
{
	PyObject *py_component_ptr = NULL;
	PyObject *py_port_ptr = NULL;
	PyObject *py_res = NULL;
	bt_graph_listener_func_status status;

	py_component_ptr = SWIG_NewPointerObj(SWIG_as_voidptr(component), component_swig_type, 0);
	if (!py_component_ptr) {
		BT_LOGF_STR("Failed to create component SWIG pointer object.");
		status = __BT_FUNC_STATUS_MEMORY_ERROR;
		goto end;
	}

	py_port_ptr = SWIG_NewPointerObj(SWIG_as_voidptr(port), port_swig_type, 0);
	if (!py_port_ptr) {
		BT_LOGF_STR("Failed to create port SWIG pointer object.");
		status = __BT_FUNC_STATUS_MEMORY_ERROR;
		goto end;
	}

	py_res = PyObject_CallFunction(py_callable, "(OiOi)",
		py_component_ptr, component_class_type, py_port_ptr, port_type);
	if (!py_res) {
		loge_exception_append_cause_clear(
			"Graph's port added listener (Python)",
			BT_LOG_OUTPUT_LEVEL);
		status = __BT_FUNC_STATUS_ERROR;
		goto end;
	}

	BT_ASSERT(py_res == Py_None);
	status = __BT_FUNC_STATUS_OK;

end:
	Py_XDECREF(py_res);
	Py_XDECREF(py_port_ptr);
	Py_XDECREF(py_component_ptr);
	return status;
}

static
bt_graph_listener_func_status
source_component_output_port_added_listener(const bt_component_source *component_source,
					    const bt_port_output *port_output, void *py_callable)
{
	return port_added_listener(
		component_source, SWIGTYPE_p_bt_component_source, BT_COMPONENT_CLASS_TYPE_SOURCE,
		port_output, SWIGTYPE_p_bt_port_output, BT_PORT_TYPE_OUTPUT, py_callable);
}

static
bt_graph_listener_func_status
filter_component_input_port_added_listener(const bt_component_filter *component_filter,
					   const bt_port_input *port_input, void *py_callable)
{
	return port_added_listener(
		component_filter, SWIGTYPE_p_bt_component_filter, BT_COMPONENT_CLASS_TYPE_FILTER,
		port_input, SWIGTYPE_p_bt_port_input, BT_PORT_TYPE_INPUT, py_callable);
}

static
bt_graph_listener_func_status
filter_component_output_port_added_listener(const bt_component_filter *component_filter,
					    const bt_port_output *port_output, void *py_callable)
{
	return port_added_listener(
		component_filter, SWIGTYPE_p_bt_component_filter, BT_COMPONENT_CLASS_TYPE_FILTER,
		port_output, SWIGTYPE_p_bt_port_output, BT_PORT_TYPE_OUTPUT, py_callable);
}

static
bt_graph_listener_func_status
sink_component_input_port_added_listener(const bt_component_sink *component_sink,
					 const bt_port_input *port_input, void *py_callable)
{
	return port_added_listener(
		component_sink, SWIGTYPE_p_bt_component_sink, BT_COMPONENT_CLASS_TYPE_SINK,
		port_input, SWIGTYPE_p_bt_port_input, BT_PORT_TYPE_INPUT, py_callable);
}

static
PyObject *bt_bt2_graph_add_port_added_listener(struct bt_graph *graph,
		PyObject *py_callable)
{
	PyObject *py_listener_ids = NULL;
	PyObject *py_listener_id = NULL;
	bt_listener_id listener_id;
	bt_graph_add_listener_status status;
	const char * const module_name =
		"graph_add_port_added_listener() (Python)";

	BT_ASSERT(graph);
	BT_ASSERT(py_callable);

	/*
	 * Behind the scene, we will be registering 4 different listeners and
	 * return all of their ids.
	 */
	py_listener_ids = PyTuple_New(4);
	if (!py_listener_ids) {
		BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
			"Failed to allocate one PyTuple.");
		goto error;
	}

	/* source output port */
	status = bt_graph_add_source_component_output_port_added_listener(
		graph, source_component_output_port_added_listener,
		graph_listener_removed, py_callable, &listener_id);
	if (status != __BT_FUNC_STATUS_OK) {
		/*
		 * bt_graph_add_source_component_output_port_added_listener has
		 * already logged/appended an error cause.
		 */
		goto error;
	}

	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
	if (!py_listener_id) {
		BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
			"Failed to allocate one PyLong.");
		goto error;
	}

	PyTuple_SET_ITEM(py_listener_ids, 0, py_listener_id);
	py_listener_id = NULL;

	/* filter input port */
	status = bt_graph_add_filter_component_input_port_added_listener(
		graph, filter_component_input_port_added_listener,
		graph_listener_removed, py_callable, &listener_id);
	if (status != __BT_FUNC_STATUS_OK) {
		/*
		 * bt_graph_add_filter_component_input_port_added_listener has
		 * already logged/appended an error cause.
		 */
		goto error;
	}

	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
	if (!py_listener_id) {
		BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
			"Failed to allocate one PyLong.");
		goto error;
	}

	PyTuple_SET_ITEM(py_listener_ids, 1, py_listener_id);
	py_listener_id = NULL;

	/* filter output port */
	status = bt_graph_add_filter_component_output_port_added_listener(
		graph, filter_component_output_port_added_listener,
		graph_listener_removed, py_callable, &listener_id);
	if (status != __BT_FUNC_STATUS_OK) {
		/*
		 * bt_graph_add_filter_component_output_port_added_listener has
		 * already logged/appended an error cause.
		 */
		goto error;
	}

	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
	if (!py_listener_id) {
		BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
			"Failed to allocate one PyLong.");
		goto error;
	}

	PyTuple_SET_ITEM(py_listener_ids, 2, py_listener_id);
	py_listener_id = NULL;

	/* sink input port */
	status = bt_graph_add_sink_component_input_port_added_listener(
		graph, sink_component_input_port_added_listener,
		graph_listener_removed, py_callable, &listener_id);
	if (status != __BT_FUNC_STATUS_OK) {
		/*
		 * bt_graph_add_sink_component_input_port_added_listener has
		 * already logged/appended an error cause.
		 */
		goto error;
	}

	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
	if (!py_listener_id) {
		BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
			"Failed to allocate one PyLong.");
		goto error;
	}

	PyTuple_SET_ITEM(py_listener_ids, 3, py_listener_id);
	py_listener_id = NULL;

	Py_INCREF(py_callable);
	Py_INCREF(py_callable);
	Py_INCREF(py_callable);
	Py_INCREF(py_callable);

	goto end;

error:
	Py_XDECREF(py_listener_ids);
	py_listener_ids = Py_None;
	Py_INCREF(py_listener_ids);

end:
	Py_XDECREF(py_listener_id);
	return py_listener_ids;
}

static
bt_graph_add_component_status
bt_bt2_graph_add_source_component(
		bt_graph *graph,
		const bt_component_class_source *component_class,
		const char *name, const bt_value *params,
		PyObject *obj, bt_logging_level log_level,
		const bt_component_source **component)
{
	return bt_graph_add_source_component_with_initialize_method_data(graph,
		component_class, name, params, obj == Py_None ? NULL : obj,
		log_level, component);
}

static
bt_graph_add_component_status
bt_bt2_graph_add_filter_component(
		bt_graph *graph,
		const bt_component_class_filter *component_class,
		const char *name, const bt_value *params,
		PyObject *obj, bt_logging_level log_level,
		const bt_component_filter **component)
{
	return bt_graph_add_filter_component_with_initialize_method_data(graph,
		component_class, name, params, obj == Py_None ? NULL : obj,
		log_level, component);
}

static
bt_graph_add_component_status
bt_bt2_graph_add_sink_component(
		bt_graph *graph,
		const bt_component_class_sink *component_class,
		const char *name, const bt_value *params,
		PyObject *obj, bt_logging_level log_level,
		const bt_component_sink **component)
{
	return bt_graph_add_sink_component_with_initialize_method_data(graph,
		component_class, name, params, obj == Py_None ? NULL : obj,
		log_level, component);
}
Commit	Line	Data
4212232c PP	1	/*
	2	* The MIT License (MIT)
	3	*
	4	* Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
	5	*
	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:
	12	*
	13	* The above copyright notice and this permission notice shall be included in
	14	* all copies or substantial portions of the Software.
	15	*
	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
	22	* THE SOFTWARE.
	23	*/
	24
	25	static
	26	void graph_listener_removed(void *py_callable)
	27	{
	28	BT_ASSERT(py_callable);
	29	Py_DECREF(py_callable);
	30	}
	31
	32	static bt_graph_listener_func_status port_added_listener(
	33	const void *component,
	34	swig_type_info *component_swig_type,
	35	bt_component_class_type component_class_type,
	36	const void *port,
	37	swig_type_info *port_swig_type,
	38	bt_port_type port_type,
	39	void *py_callable)
	40	{
	41	PyObject *py_component_ptr = NULL;
	42	PyObject *py_port_ptr = NULL;
	43	PyObject *py_res = NULL;
	44	bt_graph_listener_func_status status;
	45
	46	py_component_ptr = SWIG_NewPointerObj(SWIG_as_voidptr(component), component_swig_type, 0);
	47	if (!py_component_ptr) {
	48	BT_LOGF_STR("Failed to create component SWIG pointer object.");
	49	status = __BT_FUNC_STATUS_MEMORY_ERROR;
	50	goto end;
	51	}
	52
	53	py_port_ptr = SWIG_NewPointerObj(SWIG_as_voidptr(port), port_swig_type, 0);
	54	if (!py_port_ptr) {
	55	BT_LOGF_STR("Failed to create port SWIG pointer object.");
	56	status = __BT_FUNC_STATUS_MEMORY_ERROR;
	57	goto end;
	58	}
	59
	60	py_res = PyObject_CallFunction(py_callable, "(OiOi)",
	61	py_component_ptr, component_class_type, py_port_ptr, port_type);
	62	if (!py_res) {
981f33dd	63	loge_exception_append_cause_clear(
84e438af	64	"Graph's port added listener (Python)",
4212232c	65	BT_LOG_OUTPUT_LEVEL);
4212232c PP	66	status = __BT_FUNC_STATUS_ERROR;
	67	goto end;
	68	}
	69
	70	BT_ASSERT(py_res == Py_None);
	71	status = __BT_FUNC_STATUS_OK;
	72
	73	end:
	74	Py_XDECREF(py_res);
	75	Py_XDECREF(py_port_ptr);
	76	Py_XDECREF(py_component_ptr);
	77	return status;
	78	}
	79
	80	static
	81	bt_graph_listener_func_status
	82	source_component_output_port_added_listener(const bt_component_source *component_source,
	83	const bt_port_output port_output, void py_callable)
	84	{
	85	return port_added_listener(
	86	component_source, SWIGTYPE_p_bt_component_source, BT_COMPONENT_CLASS_TYPE_SOURCE,
	87	port_output, SWIGTYPE_p_bt_port_output, BT_PORT_TYPE_OUTPUT, py_callable);
	88	}
	89
	90	static
	91	bt_graph_listener_func_status
	92	filter_component_input_port_added_listener(const bt_component_filter *component_filter,
	93	const bt_port_input port_input, void py_callable)
	94	{
	95	return port_added_listener(
	96	component_filter, SWIGTYPE_p_bt_component_filter, BT_COMPONENT_CLASS_TYPE_FILTER,
	97	port_input, SWIGTYPE_p_bt_port_input, BT_PORT_TYPE_INPUT, py_callable);
	98	}
	99
	100	static
	101	bt_graph_listener_func_status
	102	filter_component_output_port_added_listener(const bt_component_filter *component_filter,
	103	const bt_port_output port_output, void py_callable)
	104	{
	105	return port_added_listener(
	106	component_filter, SWIGTYPE_p_bt_component_filter, BT_COMPONENT_CLASS_TYPE_FILTER,
	107	port_output, SWIGTYPE_p_bt_port_output, BT_PORT_TYPE_OUTPUT, py_callable);
	108	}
	109
	110	static
	111	bt_graph_listener_func_status
	112	sink_component_input_port_added_listener(const bt_component_sink *component_sink,
	113	const bt_port_input port_input, void py_callable)
	114	{
	115	return port_added_listener(
	116	component_sink, SWIGTYPE_p_bt_component_sink, BT_COMPONENT_CLASS_TYPE_SINK,
	117	port_input, SWIGTYPE_p_bt_port_input, BT_PORT_TYPE_INPUT, py_callable);
	118	}
	119
	120	static
	121	PyObject bt_bt2_graph_add_port_added_listener(struct bt_graph graph,
	122	PyObject *py_callable)
	123	{
	124	PyObject *py_listener_ids = NULL;
	125	PyObject *py_listener_id = NULL;
	126	bt_listener_id listener_id;
	127	bt_graph_add_listener_status status;
	128	const char * const module_name =
	129	"graph_add_port_added_listener() (Python)";
130
131	BT_ASSERT(graph);
132	BT_ASSERT(py_callable);
133
134	/*
135	* Behind the scene, we will be registering 4 different listeners and
136	* return all of their ids.
137	*/
138	py_listener_ids = PyTuple_New(4);
139	if (!py_listener_ids) {
140	BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
141	"Failed to allocate one PyTuple.");
142	goto error;
143	}
144
145	/* source output port */
146	status = bt_graph_add_source_component_output_port_added_listener(
147	graph, source_component_output_port_added_listener,
148	graph_listener_removed, py_callable, &listener_id);
149	if (status != __BT_FUNC_STATUS_OK) {
150	/*
151	* bt_graph_add_source_component_output_port_added_listener has
152	* already logged/appended an error cause.
153	*/
154	goto error;
155	}
156
157	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
158	if (!py_listener_id) {
159	BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
160	"Failed to allocate one PyLong.");
161	goto error;
162	}
163
164	PyTuple_SET_ITEM(py_listener_ids, 0, py_listener_id);
165	py_listener_id = NULL;
166
167	/* filter input port */
168	status = bt_graph_add_filter_component_input_port_added_listener(
169	graph, filter_component_input_port_added_listener,
170	graph_listener_removed, py_callable, &listener_id);
171	if (status != __BT_FUNC_STATUS_OK) {
172	/*
173	* bt_graph_add_filter_component_input_port_added_listener has
174	* already logged/appended an error cause.
175	*/
176	goto error;
177	}
178
179	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
180	if (!py_listener_id) {
181	BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
182	"Failed to allocate one PyLong.");
183	goto error;
184	}
185
186	PyTuple_SET_ITEM(py_listener_ids, 1, py_listener_id);
187	py_listener_id = NULL;
188
189	/* filter output port */
190	status = bt_graph_add_filter_component_output_port_added_listener(
191	graph, filter_component_output_port_added_listener,
192	graph_listener_removed, py_callable, &listener_id);
193	if (status != __BT_FUNC_STATUS_OK) {
194	/*
195	* bt_graph_add_filter_component_output_port_added_listener has
196	* already logged/appended an error cause.
197	*/
198	goto error;
199	}
200
201	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
202	if (!py_listener_id) {
203	BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
204	"Failed to allocate one PyLong.");
205	goto error;
206	}
207
208	PyTuple_SET_ITEM(py_listener_ids, 2, py_listener_id);
209	py_listener_id = NULL;
210
211	/* sink input port */
212	status = bt_graph_add_sink_component_input_port_added_listener(
213	graph, sink_component_input_port_added_listener,
214	graph_listener_removed, py_callable, &listener_id);
215	if (status != __BT_FUNC_STATUS_OK) {
216	/*
217	* bt_graph_add_sink_component_input_port_added_listener has
218	* already logged/appended an error cause.
219	*/
220	goto error;
221	}
222
223	py_listener_id = PyLong_FromUnsignedLongLong(listener_id);
224	if (!py_listener_id) {
225	BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(module_name,
226	"Failed to allocate one PyLong.");
227	goto error;
228	}
229
4212232c PP	230	PyTuple_SET_ITEM(py_listener_ids, 3, py_listener_id);
	231	py_listener_id = NULL;
	232
	233	Py_INCREF(py_callable);
	234	Py_INCREF(py_callable);
	235	Py_INCREF(py_callable);
	236	Py_INCREF(py_callable);
	237
	238	goto end;
	239
	240	error:
	241	Py_XDECREF(py_listener_ids);
	242	py_listener_ids = Py_None;
	243	Py_INCREF(py_listener_ids);
	244
	245	end:
4212232c PP	246	Py_XDECREF(py_listener_id);
	247	return py_listener_ids;
	248	}
	249
66964f3f PP	250	static
	251	bt_graph_add_component_status
	252	bt_bt2_graph_add_source_component(
	253	bt_graph *graph,
	254	const bt_component_class_source *component_class,
	255	const char name, const bt_value params,
	256	PyObject *obj, bt_logging_level log_level,
	257	const bt_component_source **component)
	258	{
21a9f056	259	return bt_graph_add_source_component_with_initialize_method_data(graph,
66964f3f PP	260	component_class, name, params, obj == Py_None ? NULL : obj,
	261	log_level, component);
	262	}
	263
	264	static
	265	bt_graph_add_component_status
	266	bt_bt2_graph_add_filter_component(
	267	bt_graph *graph,
	268	const bt_component_class_filter *component_class,
	269	const char name, const bt_value params,
	270	PyObject *obj, bt_logging_level log_level,
	271	const bt_component_filter **component)
	272	{
21a9f056	273	return bt_graph_add_filter_component_with_initialize_method_data(graph,
66964f3f PP	274	component_class, name, params, obj == Py_None ? NULL : obj,
	275	log_level, component);
	276	}
	277
	278	static
	279	bt_graph_add_component_status
	280	bt_bt2_graph_add_sink_component(
	281	bt_graph *graph,
	282	const bt_component_class_sink *component_class,
	283	const char name, const bt_value params,
	284	PyObject *obj, bt_logging_level log_level,
	285	const bt_component_sink **component)
	286	{
21a9f056	287	return bt_graph_add_sink_component_with_initialize_method_data(graph,
66964f3f PP	288	component_class, name, params, obj == Py_None ? NULL : obj,
	289	log_level, component);
	290	}