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

/*
 * SPDX-License-Identifier: MIT
 *
 * Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
 */

static
bt_message_iterator_create_from_message_iterator_status
bt_bt2_message_iterator_create_from_message_iterator(
		bt_self_message_iterator *self_msg_iter,
		bt_self_component_port_input *input_port,
		bt_message_iterator **message_iterator)
{
	bt_message_iterator_create_from_message_iterator_status
		status;

	status = bt_message_iterator_create_from_message_iterator(
		self_msg_iter, input_port, message_iterator);

	if (status != BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK) {
		*message_iterator = NULL;
	}

	return status;
}

static
bt_message_iterator_create_from_sink_component_status
bt_bt2_message_iterator_create_from_sink_component(
		bt_self_component_sink *self_comp,
		bt_self_component_port_input *input_port,
		bt_message_iterator **message_iterator)
{
	bt_message_iterator_create_from_sink_component_status
		status;

	status = bt_message_iterator_create_from_sink_component(
		self_comp, input_port, message_iterator);

	if (status != BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK) {
		*message_iterator = NULL;
	}

	return status;
}

static PyObject *bt_bt2_get_user_component_from_user_msg_iter(
		bt_self_message_iterator *self_message_iterator)
{
	bt_self_component *self_component = bt_self_message_iterator_borrow_component(self_message_iterator);
	PyObject *py_comp;

	BT_ASSERT_DBG(self_component);
	py_comp = bt_self_component_get_data(self_component);
	BT_ASSERT_DBG(py_comp);

	/* Return new reference */
	Py_INCREF(py_comp);
	return py_comp;
}

static inline
PyObject *create_pylist_from_messages(bt_message_array_const messages,
		uint64_t message_count)
{
	uint64_t i;
	PyObject *py_msg_list = PyList_New(message_count);

	BT_ASSERT(py_msg_list);

	for (i = 0; i < message_count; i++) {
		PyList_SET_ITEM(py_msg_list, i,
				SWIG_NewPointerObj(SWIG_as_voidptr(messages[i]),
					SWIGTYPE_p_bt_message, 0));
	}

	return py_msg_list;
}

static
PyObject *get_msg_range_common(bt_message_iterator_next_status status,
		bt_message_array_const messages, uint64_t message_count)
{
	PyObject *py_status;
	PyObject *py_return_tuple;
	PyObject *py_msg_list;

	py_return_tuple = PyTuple_New(2);
	BT_ASSERT(py_return_tuple);

	/* Set tuple[0], status. */
	py_status = SWIG_From_long_SS_long(status);
	PyTuple_SET_ITEM(py_return_tuple, 0, py_status);

	/* Set tuple[1], message list on success, None otherwise. */
	if (status == __BT_FUNC_STATUS_OK) {
		py_msg_list = create_pylist_from_messages(messages, message_count);
	} else {
		py_msg_list = Py_None;
		Py_INCREF(py_msg_list);
	}

	PyTuple_SET_ITEM(py_return_tuple, 1, py_msg_list);

	return py_return_tuple;
}

static PyObject *bt_bt2_self_component_port_input_get_msg_range(
		bt_message_iterator *iter)
{
	bt_message_array_const messages;
	uint64_t message_count = 0;
	bt_message_iterator_next_status status;

	status = bt_message_iterator_next(iter,
		&messages, &message_count);
	return get_msg_range_common(status, messages, message_count);
}
Commit	Line	Data
	1	/*
	2	* SPDX-License-Identifier: MIT
	3	*
	4	* Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
	5	*/
	6
	7	static
	8	bt_message_iterator_create_from_message_iterator_status
	9	bt_bt2_message_iterator_create_from_message_iterator(
	10	bt_self_message_iterator *self_msg_iter,
	11	bt_self_component_port_input *input_port,
	12	bt_message_iterator **message_iterator)
	13	{
	14	bt_message_iterator_create_from_message_iterator_status
	15	status;
	16
	17	status = bt_message_iterator_create_from_message_iterator(
	18	self_msg_iter, input_port, message_iterator);
	19
	20	if (status != BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK) {
	21	*message_iterator = NULL;
	22	}
	23
	24	return status;
	25	}
	26
	27	static
	28	bt_message_iterator_create_from_sink_component_status
	29	bt_bt2_message_iterator_create_from_sink_component(
	30	bt_self_component_sink *self_comp,
	31	bt_self_component_port_input *input_port,
	32	bt_message_iterator **message_iterator)
	33	{
	34	bt_message_iterator_create_from_sink_component_status
	35	status;
	36
	37	status = bt_message_iterator_create_from_sink_component(
	38	self_comp, input_port, message_iterator);
	39
	40	if (status != BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK) {
	41	*message_iterator = NULL;
	42	}
	43
	44	return status;
	45	}
	46
	47	static PyObject *bt_bt2_get_user_component_from_user_msg_iter(
	48	bt_self_message_iterator *self_message_iterator)
	49	{
	50	bt_self_component *self_component = bt_self_message_iterator_borrow_component(self_message_iterator);
	51	PyObject *py_comp;
	52
	53	BT_ASSERT_DBG(self_component);
	54	py_comp = bt_self_component_get_data(self_component);
	55	BT_ASSERT_DBG(py_comp);
	56
	57	/* Return new reference */
	58	Py_INCREF(py_comp);
	59	return py_comp;
	60	}
	61
	62	static inline
	63	PyObject *create_pylist_from_messages(bt_message_array_const messages,
	64	uint64_t message_count)
	65	{
	66	uint64_t i;
	67	PyObject *py_msg_list = PyList_New(message_count);
	68
	69	BT_ASSERT(py_msg_list);
	70
	71	for (i = 0; i < message_count; i++) {
	72	PyList_SET_ITEM(py_msg_list, i,
	73	SWIG_NewPointerObj(SWIG_as_voidptr(messages[i]),
	74	SWIGTYPE_p_bt_message, 0));
	75	}
	76
	77	return py_msg_list;
	78	}
	79
	80	static
	81	PyObject *get_msg_range_common(bt_message_iterator_next_status status,
	82	bt_message_array_const messages, uint64_t message_count)
	83	{
	84	PyObject *py_status;
	85	PyObject *py_return_tuple;
	86	PyObject *py_msg_list;
	87
	88	py_return_tuple = PyTuple_New(2);
	89	BT_ASSERT(py_return_tuple);
	90
	91	/* Set tuple[0], status. */
	92	py_status = SWIG_From_long_SS_long(status);
	93	PyTuple_SET_ITEM(py_return_tuple, 0, py_status);
	94
	95	/* Set tuple[1], message list on success, None otherwise. */
	96	if (status == __BT_FUNC_STATUS_OK) {
	97	py_msg_list = create_pylist_from_messages(messages, message_count);
	98	} else {
	99	py_msg_list = Py_None;
	100	Py_INCREF(py_msg_list);
	101	}
	102
	103	PyTuple_SET_ITEM(py_return_tuple, 1, py_msg_list);
	104
	105	return py_return_tuple;
	106	}
	107
	108	static PyObject *bt_bt2_self_component_port_input_get_msg_range(
	109	bt_message_iterator *iter)
	110	{
	111	bt_message_array_const messages;
	112	uint64_t message_count = 0;
	113	bt_message_iterator_next_status status;
	114
	115	status = bt_message_iterator_next(iter,
	116	&messages, &message_count);
	117	return get_msg_range_common(status, messages, message_count);
	118	}