2 * Copyright 2017 Philippe Proulx <pproulx@efficios.com>
4 * Permission is hereby granted, free of charge, to any person obtaining a copy
5 * of this software and associated documentation files (the "Software"), to deal
6 * in the Software without restriction, including without limitation the rights
7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 * copies of the Software, and to permit persons to whom the Software is
9 * furnished to do so, subject to the following conditions:
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23 #include <babeltrace/babeltrace.h>
24 #include <babeltrace/babeltrace-internal.h>
25 #include <babeltrace/common-internal.h>
26 #include <plugins-common.h>
27 #include <babeltrace/assert-internal.h>
33 #define PRINTF_COUNT(_what_sing, _what_plur, _var, args...) \
35 if (counter->count._var != 0 || !counter->hide_zero) { \
36 printf("%15" PRIu64 " %s\n", \
37 counter->count._var, \
38 counter->count._var == 1 ? _what_sing : _what_plur); \
43 uint64_t get_total_count(struct counter
*counter
)
45 return counter
->count
.event
+
46 counter
->count
.stream_begin
+
47 counter
->count
.stream_end
+
48 counter
->count
.packet_begin
+
49 counter
->count
.packet_end
+
50 counter
->count
.inactivity
+
55 void print_count(struct counter
*counter
)
57 uint64_t total
= get_total_count(counter
);
59 PRINTF_COUNT("event", "events", event
);
60 PRINTF_COUNT("stream beginning", "stream beginnings", stream_begin
);
61 PRINTF_COUNT("stream end", "stream ends", stream_end
);
62 PRINTF_COUNT("packet beginning", "packet beginnings", packet_begin
);
63 PRINTF_COUNT("packet end", "packet ends", packet_end
);
64 PRINTF_COUNT("inactivity", "inactivities", inactivity
);
66 if (counter
->count
.other
> 0) {
67 PRINTF_COUNT(" other (unknown) notification",
68 " other (unknown) notifications", other
);
71 printf("%s%15" PRIu64
" notification%s (TOTAL)%s\n",
72 bt_common_color_bold(), total
, total
== 1 ? "" : "s",
73 bt_common_color_reset());
74 counter
->last_printed_total
= total
;
78 void try_print_count(struct counter
*counter
, uint64_t notif_count
)
80 if (counter
->step
== 0) {
85 counter
->at
+= notif_count
;
87 if (counter
->at
>= counter
->step
) {
95 void try_print_last(struct counter
*counter
)
97 const uint64_t total
= get_total_count(counter
);
99 if (total
!= counter
->last_printed_total
) {
100 print_count(counter
);
104 void destroy_private_counter_data(struct counter
*counter
)
106 bt_object_put_ref(counter
->notif_iter
);
111 void counter_finalize(struct bt_self_component_sink
*comp
)
113 struct counter
*counter
;
116 counter
= bt_self_component_get_data(
117 bt_self_component_sink_borrow_self_component(comp
));
119 try_print_last(counter
);
120 bt_object_put_ref(counter
->notif_iter
);
125 enum bt_self_component_status
counter_init(
126 struct bt_self_component_sink
*component
,
127 struct bt_value
*params
, UNUSED_VAR
void *init_method_data
)
129 enum bt_self_component_status ret
;
130 struct counter
*counter
= g_new0(struct counter
, 1);
131 struct bt_value
*step
= NULL
;
132 struct bt_value
*hide_zero
= NULL
;
135 ret
= BT_SELF_COMPONENT_STATUS_NOMEM
;
139 ret
= bt_self_component_sink_add_input_port(component
,
141 if (ret
!= BT_SELF_COMPONENT_STATUS_OK
) {
145 counter
->last_printed_total
= -1ULL;
146 counter
->step
= 1000;
147 step
= bt_value_map_borrow_entry_value(params
, "step");
148 if (step
&& bt_value_is_integer(step
)) {
151 val
= bt_value_integer_get(step
);
153 counter
->step
= (uint64_t) val
;
157 hide_zero
= bt_value_map_borrow_entry_value(params
, "hide-zero");
158 if (hide_zero
&& bt_value_is_bool(hide_zero
)) {
161 val
= bt_value_bool_get(hide_zero
);
162 counter
->hide_zero
= (bool) val
;
165 bt_self_component_set_data(
166 bt_self_component_sink_borrow_self_component(component
),
171 destroy_private_counter_data(counter
);
178 enum bt_self_component_status
counter_port_connected(
179 struct bt_self_component_sink
*comp
,
180 struct bt_self_component_port_input
*self_port
,
181 struct bt_port_output
*other_port
)
183 enum bt_self_component_status status
= BT_SELF_COMPONENT_STATUS_OK
;
184 struct counter
*counter
;
185 struct bt_self_component_port_input_notification_iterator
*iterator
;
187 counter
= bt_self_component_get_data(
188 bt_self_component_sink_borrow_self_component(comp
));
190 iterator
= bt_self_component_port_input_notification_iterator_create(
193 status
= BT_SELF_COMPONENT_STATUS_NOMEM
;
197 BT_OBJECT_MOVE_REF(counter
->notif_iter
, iterator
);
204 enum bt_self_component_status
counter_consume(
205 struct bt_self_component_sink
*comp
)
207 enum bt_self_component_status ret
= BT_SELF_COMPONENT_STATUS_OK
;
208 struct counter
*counter
;
209 enum bt_notification_iterator_status it_ret
;
210 uint64_t notif_count
;
211 bt_notification_array notifs
;
213 counter
= bt_self_component_get_data(
214 bt_self_component_sink_borrow_self_component(comp
));
217 if (unlikely(!counter
->notif_iter
)) {
218 try_print_last(counter
);
219 ret
= BT_SELF_COMPONENT_STATUS_END
;
223 /* Consume notifications */
224 it_ret
= bt_self_component_port_input_notification_iterator_next(
225 counter
->notif_iter
, ¬ifs
, ¬if_count
);
227 ret
= BT_SELF_COMPONENT_STATUS_ERROR
;
232 case BT_NOTIFICATION_ITERATOR_STATUS_OK
:
236 for (i
= 0; i
< notif_count
; i
++) {
237 struct bt_notification
*notif
= notifs
[i
];
240 switch (bt_notification_get_type(notif
)) {
241 case BT_NOTIFICATION_TYPE_EVENT
:
242 counter
->count
.event
++;
244 case BT_NOTIFICATION_TYPE_INACTIVITY
:
245 counter
->count
.inactivity
++;
247 case BT_NOTIFICATION_TYPE_STREAM_BEGIN
:
248 counter
->count
.stream_begin
++;
250 case BT_NOTIFICATION_TYPE_STREAM_END
:
251 counter
->count
.stream_end
++;
253 case BT_NOTIFICATION_TYPE_PACKET_BEGIN
:
254 counter
->count
.packet_begin
++;
256 case BT_NOTIFICATION_TYPE_PACKET_END
:
257 counter
->count
.packet_end
++;
260 counter
->count
.other
++;
263 bt_object_put_ref(notif
);
266 ret
= BT_SELF_COMPONENT_STATUS_OK
;
269 case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN
:
270 ret
= BT_SELF_COMPONENT_STATUS_AGAIN
;
272 case BT_NOTIFICATION_ITERATOR_STATUS_END
:
273 try_print_last(counter
);
274 ret
= BT_SELF_COMPONENT_STATUS_END
;
276 case BT_NOTIFICATION_ITERATOR_STATUS_NOMEM
:
277 ret
= BT_SELF_COMPONENT_STATUS_NOMEM
;
283 try_print_count(counter
, notif_count
);