X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fport-output-const.h;h=0afa640fe84db23c504951d92734cdcbcfdd6daf;hb=9f7a7546c45e7501e749d19ebad3a31b7d715ef6;hp=4d44914e86faaaf848ecae036f8cd090ec57b2d6;hpb=96ffdc460181984c548d8191bbc074df505cc379;p=deliverable%2Fbabeltrace.git

diff --git a/include/babeltrace/graph/port-output-const.h b/include/babeltrace/graph/port-output-const.h
index 4d44914e8..0afa640fe 100644
--- a/include/babeltrace/graph/port-output-const.h
+++ b/include/babeltrace/graph/port-output-const.h
@@ -2,10 +2,9 @@
 #define BABELTRACE_GRAPH_PORT_OUTPUT_CONST_H
 
 /*
+ * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
  * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
- * Author: Jérémie Galarneau <jeremie.galarneau@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
@@ -25,22 +24,38 @@
  * SOFTWARE.
  */
 
+/* For bt_port, bt_port_output, __BT_UPCAST_CONST */
+#include <babeltrace/types.h>
+
 #include <stdint.h>
 
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-struct bt_port;
-struct bt_port_output;
-
 static inline
-const struct bt_port *bt_port_output_as_port_const(
-		const struct bt_port_output *port_output)
+const bt_port *bt_port_output_as_port_const(const bt_port_output *port_output)
 {
-	return (const void *) port_output;
+	return __BT_UPCAST_CONST(bt_port, port_output);
 }
 
+extern void bt_port_output_get_ref(const bt_port_output *port_output);
+
+extern void bt_port_output_put_ref(const bt_port_output *port_output);
+
+#define BT_PORT_OUTPUT_PUT_REF_AND_RESET(_var)		\
+	do {						\
+		bt_port_output_put_ref(_var);		\
+		(_var) = NULL;				\
+	} while (0)
+
+#define BT_PORT_OUTPUT_MOVE_REF(_var_dst, _var_src)	\
+	do {						\
+		bt_port_output_put_ref(_var_dst);	\
+		(_var_dst) = (_var_src);		\
+		(_var_src) = NULL;			\
+	} while (0)
+
 #ifdef __cplusplus
 }
 #endif