[babeltrace.git] / src / lib / graph / message / message.h

/*
 * SPDX-License-Identifier: MIT
 *
 * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
 * Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 */

#ifndef BABELTRACE_GRAPH_MESSAGE_MESSAGE_INTERNAL_H
#define BABELTRACE_GRAPH_MESSAGE_MESSAGE_INTERNAL_H

/* Protection: this file uses BT_LIB_LOG*() macros directly */
#ifndef BT_LIB_LOG_SUPPORTED
# error Please include "lib/logging.h" before including this file.
#endif

#include "common/macros.h"
#include "lib/object.h"
#include "common/assert.h"
#include <babeltrace2/graph/graph.h>
#include <babeltrace2/graph/message.h>
#include <babeltrace2/trace-ir/stream.h>
#include "lib/object-pool.h"
#include <babeltrace2/types.h>

/* Protection: this file uses BT_LIB_LOG*() macros directly */
#ifndef BT_LIB_LOG_SUPPORTED
# error Please include "lib/logging.h" before including this file.
#endif

typedef struct bt_stream *(*get_stream_func)(
		struct bt_message *message);

struct bt_message {
	struct bt_object base;
	enum bt_message_type type;
	bt_bool frozen;

	/* Owned by this; keeps the graph alive while the msg. is alive */
	struct bt_graph *graph;
};

#define _BT_ASSERT_PRE_MSG_IS_TYPE_COND(_msg, _type)			\
	(((struct bt_message *) (_msg))->type == (_type))

#define _BT_ASSERT_PRE_MSG_IS_TYPE_FMT					\
	"Message has the wrong type: expected-type=%s, %![msg-]+n"

#define BT_ASSERT_PRE_MSG_IS_TYPE(_msg, _type)				\
	BT_ASSERT_PRE(							\
		_BT_ASSERT_PRE_MSG_IS_TYPE_COND((_msg), (_type)),	\
		_BT_ASSERT_PRE_MSG_IS_TYPE_FMT,				\
		bt_message_type_string(_type), (_msg))

#define BT_ASSERT_PRE_DEV_MSG_IS_TYPE(_msg, _type)			\
	BT_ASSERT_PRE_DEV(						\
		_BT_ASSERT_PRE_MSG_IS_TYPE_COND((_msg), (_type)),	\
		_BT_ASSERT_PRE_MSG_IS_TYPE_FMT,				\
		bt_message_type_string(_type), (_msg))

BT_HIDDEN
void bt_message_init(struct bt_message *message,
		enum bt_message_type type,
		bt_object_release_func release,
		struct bt_graph *graph);

static inline
void bt_message_reset(struct bt_message *message)
{
	BT_ASSERT_DBG(message);

#ifdef BT_DEV_MODE
	message->frozen = BT_FALSE;
#endif
}

static inline
struct bt_message *bt_message_create_from_pool(
		struct bt_object_pool *pool, struct bt_graph *graph)
{
	struct bt_message *msg = bt_object_pool_create_object(pool);

	if (G_UNLIKELY(!msg)) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Cannot allocate one message from message pool: "
			"%![pool-]+o, %![graph-]+g", pool, graph);
		goto error;
	}

	if (G_LIKELY(!msg->graph)) {
		msg->graph = graph;
	}

	goto end;

error:
	BT_ASSERT(!msg);

end:
	return msg;
}

static inline void _bt_message_freeze(struct bt_message *message)
{
	message->frozen = BT_TRUE;
}

BT_HIDDEN
void bt_message_unlink_graph(struct bt_message *msg);

#ifdef BT_DEV_MODE
# define bt_message_freeze		_bt_message_freeze
#else
# define bt_message_freeze(_x)
#endif /* BT_DEV_MODE */

static inline
const char *bt_message_type_string(enum bt_message_type type)
{
	switch (type) {
	case BT_MESSAGE_TYPE_EVENT:
		return "EVENT";
	case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
		return "MESSAGE_ITERATOR_INACTIVITY";
	case BT_MESSAGE_TYPE_STREAM_BEGINNING:
		return "STREAM_BEGINNING";
	case BT_MESSAGE_TYPE_STREAM_END:
		return "STREAM_END";
	case BT_MESSAGE_TYPE_PACKET_BEGINNING:
		return "PACKET_BEGINNING";
	case BT_MESSAGE_TYPE_PACKET_END:
		return "PACKET_END";
	case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
		return "DISCARDED_EVENTS";
	default:
		return "(unknown)";
	}
}

#endif /* BABELTRACE_GRAPH_MESSAGE_MESSAGE_INTERNAL_H */
Commit	Line	Data
d6e69534	1	/*
0235b0db MJ	2	* SPDX-License-Identifier: MIT
0235b0db MJ	3	*
d6e69534 PP	4	* Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
d6e69534 PP	5	* Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com>
d6e69534 PP	6	*/
d6e69534 PP	7
0235b0db MJ	8	#ifndef BABELTRACE_GRAPH_MESSAGE_MESSAGE_INTERNAL_H
	9	#define BABELTRACE_GRAPH_MESSAGE_MESSAGE_INTERNAL_H
	10
870631a2 PP	11	/* Protection: this file uses BT_LIB_LOG() macros directly /
	12	#ifndef BT_LIB_LOG_SUPPORTED
	13	# error Please include "lib/logging.h" before including this file.
	14	#endif
	15
91d81473	16	#include "common/macros.h"
578e048b MJ	17	#include "lib/object.h"
578e048b MJ	18	#include "common/assert.h"
3fadfbc0	19	#include <babeltrace2/graph/graph.h>
43c59509	20	#include <babeltrace2/graph/message.h>
3fadfbc0	21	#include <babeltrace2/trace-ir/stream.h>
578e048b	22	#include "lib/object-pool.h"
3fadfbc0	23	#include <babeltrace2/types.h>
d6e69534	24
f137d351 PP	25	/* Protection: this file uses BT_LIB_LOG() macros directly /
	26	#ifndef BT_LIB_LOG_SUPPORTED
	27	# error Please include "lib/logging.h" before including this file.
	28	#endif
	29
d6e69534 PP	30	typedef struct bt_stream (get_stream_func)(
	31	struct bt_message *message);
	32
	33	struct bt_message {
	34	struct bt_object base;
	35	enum bt_message_type type;
d6e69534 PP	36	bt_bool frozen;
	37
	38	/* Owned by this; keeps the graph alive while the msg. is alive */
	39	struct bt_graph *graph;
	40	};
	41
bdb288b3 PP	42	#define _BT_ASSERT_PRE_MSG_IS_TYPE_COND(_msg, _type) \
	43	(((struct bt_message *) (_msg))->type == (_type))
	44
	45	#define _BT_ASSERT_PRE_MSG_IS_TYPE_FMT \
	46	"Message has the wrong type: expected-type=%s, %![msg-]+n"
	47
	48	#define BT_ASSERT_PRE_MSG_IS_TYPE(_msg, _type) \
	49	BT_ASSERT_PRE( \
	50	_BT_ASSERT_PRE_MSG_IS_TYPE_COND((_msg), (_type)), \
	51	_BT_ASSERT_PRE_MSG_IS_TYPE_FMT, \
	52	bt_message_type_string(_type), (_msg))
	53
	54	#define BT_ASSERT_PRE_DEV_MSG_IS_TYPE(_msg, _type) \
	55	BT_ASSERT_PRE_DEV( \
	56	_BT_ASSERT_PRE_MSG_IS_TYPE_COND((_msg), (_type)), \
	57	_BT_ASSERT_PRE_MSG_IS_TYPE_FMT, \
	58	bt_message_type_string(_type), (_msg))
d6e69534 PP	59
	60	BT_HIDDEN
	61	void bt_message_init(struct bt_message *message,
	62	enum bt_message_type type,
	63	bt_object_release_func release,
	64	struct bt_graph *graph);
	65
	66	static inline
	67	void bt_message_reset(struct bt_message *message)
	68	{
98b15851	69	BT_ASSERT_DBG(message);
d6e69534 PP	70
	71	#ifdef BT_DEV_MODE
	72	message->frozen = BT_FALSE;
d6e69534 PP	73	#endif
	74	}
	75
	76	static inline
	77	struct bt_message *bt_message_create_from_pool(
	78	struct bt_object_pool pool, struct bt_graph graph)
	79	{
	80	struct bt_message *msg = bt_object_pool_create_object(pool);
	81
91d81473	82	if (G_UNLIKELY(!msg)) {
870631a2 PP	83	BT_LIB_LOGE_APPEND_CAUSE(
870631a2 PP	84	"Cannot allocate one message from message pool: "
d6e69534	85	"%![pool-]+o, %![graph-]+g", pool, graph);
d6e69534 PP	86	goto error;
	87	}
	88
91d81473	89	if (G_LIKELY(!msg->graph)) {
d6e69534 PP	90	msg->graph = graph;
	91	}
	92
	93	goto end;
	94
	95	error:
	96	BT_ASSERT(!msg);
	97
	98	end:
	99	return msg;
	100	}
	101
	102	static inline void _bt_message_freeze(struct bt_message *message)
	103	{
	104	message->frozen = BT_TRUE;
	105	}
	106
	107	BT_HIDDEN
	108	void bt_message_unlink_graph(struct bt_message *msg);
	109
	110	#ifdef BT_DEV_MODE
	111	# define bt_message_freeze _bt_message_freeze
	112	#else
	113	# define bt_message_freeze(_x)
	114	#endif /* BT_DEV_MODE */
	115
	116	static inline
	117	const char *bt_message_type_string(enum bt_message_type type)
	118	{
	119	switch (type) {
	120	case BT_MESSAGE_TYPE_EVENT:
8a432889	121	return "EVENT";
b9fd9cbb	122	case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
8a432889	123	return "MESSAGE_ITERATOR_INACTIVITY";
d6e69534	124	case BT_MESSAGE_TYPE_STREAM_BEGINNING:
8a432889	125	return "STREAM_BEGINNING";
d6e69534	126	case BT_MESSAGE_TYPE_STREAM_END:
8a432889	127	return "STREAM_END";
d6e69534	128	case BT_MESSAGE_TYPE_PACKET_BEGINNING:
8a432889	129	return "PACKET_BEGINNING";
d6e69534	130	case BT_MESSAGE_TYPE_PACKET_END:
8a432889	131	return "PACKET_END";
4c833281	132	case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
8a432889	133	return "DISCARDED_EVENTS";
d6e69534 PP	134	default:
	135	return "(unknown)";
	136	}
	137	}
	138
	139	#endif /* BABELTRACE_GRAPH_MESSAGE_MESSAGE_INTERNAL_H */