lib: strictly type function return status enumerations

[babeltrace.git] / src / plugins / utils / counter / counter.h

diff --git a/src/plugins/utils/counter/counter.h b/src/plugins/utils/counter/counter.h
index c29f0fd5335ff046a410ef35ec08637fff596bde..3d8b8b6e1d2da0c0fb7a3f11bf50cfbf45fdd488 100644 (file)
--- a/src/plugins/utils/counter/counter.h
+++ b/src/plugins/utils/counter/counter.h
@@ -27,6 +27,7 @@
 #include <babeltrace2/babeltrace.h>
 #include <stdbool.h>
 #include <stdint.h>
+#include "common/macros.h"
 
 struct counter {
        bt_self_component_port_input_message_iterator *msg_iter;
@@ -47,10 +48,12 @@ struct counter {
        uint64_t at;
        uint64_t step;
        bool hide_zero;
+       bt_logging_level log_level;
+       bt_self_component *self_comp;
 };
 
 BT_HIDDEN
-bt_self_component_status counter_init(
+bt_component_class_init_method_status counter_init(
                bt_self_component_sink *component,
                const bt_value *params, void *init_method_data);
 
@@ -58,10 +61,10 @@ BT_HIDDEN
 void counter_finalize(bt_self_component_sink *component);
 
 BT_HIDDEN
-bt_self_component_status counter_graph_is_configured(
+bt_component_class_sink_graph_is_configured_method_status counter_graph_is_configured(
                bt_self_component_sink *component);
 
 BT_HIDDEN
-bt_self_component_status counter_consume(bt_self_component_sink *component);
+bt_component_class_sink_consume_method_status counter_consume(bt_self_component_sink *component);
 
 #endif /* BABELTRACE_PLUGINS_UTILS_COUNTER_H */