tests/lib: remove `test_bt_values` and `test_graph_topo`
[babeltrace.git] / src / plugins / utils / counter / counter.c
CommitLineData
358340ec
PP
1/*
2 * Copyright 2017 Philippe Proulx <pproulx@efficios.com>
3 *
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:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
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
20 * SOFTWARE.
21 */
22
82e80647 23#define BT_COMP_LOG_SELF_COMP (counter->self_comp)
9d87d0ea 24#define BT_LOG_OUTPUT_LEVEL (counter->log_level)
350ad6c1 25#define BT_LOG_TAG "PLUGIN/FLT.UTILS.COUNTER"
82e80647 26#include "plugins/comp-logging.h"
fdd3a2da 27
3fadfbc0 28#include <babeltrace2/babeltrace.h>
91d81473 29#include "common/macros.h"
578e048b 30#include "common/common.h"
578e048b 31#include "common/assert.h"
358340ec
PP
32#include <inttypes.h>
33#include <stdint.h>
34
35#include "counter.h"
36
0e847b69 37#define PRINTF_COUNT(_what, _var, args...) \
358340ec
PP
38 do { \
39 if (counter->count._var != 0 || !counter->hide_zero) { \
0e847b69 40 printf("%15" PRIu64 " %s message%s\n", \
358340ec 41 counter->count._var, \
0e847b69
PP
42 (_what), \
43 counter->count._var == 1 ? "" : "s"); \
358340ec
PP
44 } \
45 } while (0)
46
5badd463
PP
47static
48const char * const in_port_name = "in";
49
358340ec
PP
50static
51uint64_t get_total_count(struct counter *counter)
52{
53 return counter->count.event +
54 counter->count.stream_begin +
55 counter->count.stream_end +
0e847b69
PP
56 counter->count.stream_activity_begin +
57 counter->count.stream_activity_end +
358340ec
PP
58 counter->count.packet_begin +
59 counter->count.packet_end +
0e847b69
PP
60 counter->count.disc_events +
61 counter->count.disc_packets +
b9fd9cbb 62 counter->count.msg_iter_inactivity +
358340ec
PP
63 counter->count.other;
64}
65
66static
d4393e08 67void print_count(struct counter *counter)
358340ec 68{
d4393e08
PP
69 uint64_t total = get_total_count(counter);
70
0e847b69
PP
71 PRINTF_COUNT("Event", event);
72 PRINTF_COUNT("Stream beginning", stream_begin);
73 PRINTF_COUNT("Stream end", stream_end);
74 PRINTF_COUNT("Stream activity beginning", stream_activity_begin);
75 PRINTF_COUNT("Stream activity end", stream_activity_end);
76 PRINTF_COUNT("Packet beginning", packet_begin);
77 PRINTF_COUNT("Packet end", packet_end);
78 PRINTF_COUNT("Discarded event", disc_events);
79 PRINTF_COUNT("Discarded packet", disc_packets);
80 PRINTF_COUNT("Message iterator inactivity", msg_iter_inactivity);
358340ec
PP
81
82 if (counter->count.other > 0) {
0e847b69 83 PRINTF_COUNT("Other (unknown)", other);
358340ec
PP
84 }
85
d6e69534 86 printf("%s%15" PRIu64 " message%s (TOTAL)%s\n",
358340ec
PP
87 bt_common_color_bold(), total, total == 1 ? "" : "s",
88 bt_common_color_reset());
89 counter->last_printed_total = total;
90}
91
92static
d6e69534 93void try_print_count(struct counter *counter, uint64_t msg_count)
358340ec 94{
358340ec
PP
95 if (counter->step == 0) {
96 /* No update */
97 return;
98 }
99
d6e69534 100 counter->at += msg_count;
358340ec 101
d4393e08
PP
102 if (counter->at >= counter->step) {
103 counter->at = 0;
104 print_count(counter);
358340ec
PP
105 putchar('\n');
106 }
107}
108
109static
110void try_print_last(struct counter *counter)
111{
112 const uint64_t total = get_total_count(counter);
113
114 if (total != counter->last_printed_total) {
d4393e08 115 print_count(counter);
358340ec
PP
116 }
117}
118
119void destroy_private_counter_data(struct counter *counter)
120{
d6e69534 121 bt_self_component_port_input_message_iterator_put_ref(counter->msg_iter);
358340ec
PP
122 g_free(counter);
123}
124
d94d92ac 125BT_HIDDEN
b19ff26f 126void counter_finalize(bt_self_component_sink *comp)
358340ec
PP
127{
128 struct counter *counter;
129
d94d92ac
PP
130 BT_ASSERT(comp);
131 counter = bt_self_component_get_data(
707b7d35 132 bt_self_component_sink_as_self_component(comp));
f6ccaed9 133 BT_ASSERT(counter);
358340ec 134 try_print_last(counter);
d6e69534 135 bt_self_component_port_input_message_iterator_put_ref(counter->msg_iter);
358340ec
PP
136 g_free(counter);
137}
138
d94d92ac 139BT_HIDDEN
4cdfc5e8 140bt_self_component_status counter_init(
b19ff26f
PP
141 bt_self_component_sink *component,
142 const bt_value *params,
c88dd1cb 143 __attribute__((unused)) void *init_method_data)
358340ec 144{
4cdfc5e8 145 bt_self_component_status ret;
358340ec 146 struct counter *counter = g_new0(struct counter, 1);
b19ff26f
PP
147 const bt_value *step = NULL;
148 const bt_value *hide_zero = NULL;
358340ec
PP
149
150 if (!counter) {
d94d92ac
PP
151 ret = BT_SELF_COMPONENT_STATUS_NOMEM;
152 goto error;
358340ec
PP
153 }
154
82e80647
PP
155 counter->self_comp =
156 bt_self_component_sink_as_self_component(component);
9d87d0ea 157 counter->log_level = bt_component_get_logging_level(
82e80647 158 bt_self_component_as_component(counter->self_comp));
d94d92ac 159 ret = bt_self_component_sink_add_input_port(component,
358340ec 160 "in", NULL, NULL);
d94d92ac
PP
161 if (ret != BT_SELF_COMPONENT_STATUS_OK) {
162 goto error;
358340ec
PP
163 }
164
165 counter->last_printed_total = -1ULL;
a5bc50e4 166 counter->step = 10000;
05e21286 167 step = bt_value_map_borrow_entry_value_const(params, "step");
fdd3a2da
PP
168 if (step) {
169 if (!bt_value_is_unsigned_integer(step)) {
82e80647 170 BT_COMP_LOGE("`step` parameter: expecting an unsigned integer value: "
fdd3a2da
PP
171 "type=%s", bt_common_value_type_string(
172 bt_value_get_type(step)));
173 goto error;
358340ec 174 }
fdd3a2da
PP
175
176 counter->step = bt_value_unsigned_integer_get(step);
358340ec
PP
177 }
178
05e21286 179 hide_zero = bt_value_map_borrow_entry_value_const(params, "hide-zero");
fdd3a2da
PP
180 if (hide_zero) {
181 if (!bt_value_is_bool(hide_zero)) {
82e80647 182 BT_COMP_LOGE("`hide-zero` parameter: expecting a boolean value: "
fdd3a2da
PP
183 "type=%s", bt_common_value_type_string(
184 bt_value_get_type(hide_zero)));
185 goto error;
186 }
358340ec 187
fdd3a2da 188 counter->hide_zero = (bool) bt_value_bool_get(hide_zero);
358340ec
PP
189 }
190
d94d92ac 191 bt_self_component_set_data(
707b7d35 192 bt_self_component_sink_as_self_component(component),
d94d92ac 193 counter);
358340ec
PP
194 goto end;
195
196error:
197 destroy_private_counter_data(counter);
fdd3a2da 198 ret = BT_SELF_COMPONENT_STATUS_ERROR;
358340ec
PP
199
200end:
358340ec
PP
201 return ret;
202}
203
d94d92ac 204BT_HIDDEN
5badd463
PP
205bt_self_component_status counter_graph_is_configured(
206 bt_self_component_sink *comp)
358340ec 207{
4cdfc5e8 208 bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
358340ec 209 struct counter *counter;
d6e69534 210 bt_self_component_port_input_message_iterator *iterator;
358340ec 211
d94d92ac 212 counter = bt_self_component_get_data(
707b7d35 213 bt_self_component_sink_as_self_component(comp));
f6ccaed9 214 BT_ASSERT(counter);
d6e69534 215 iterator = bt_self_component_port_input_message_iterator_create(
5badd463
PP
216 bt_self_component_sink_borrow_input_port_by_name(comp,
217 in_port_name));
d94d92ac
PP
218 if (!iterator) {
219 status = BT_SELF_COMPONENT_STATUS_NOMEM;
358340ec
PP
220 goto end;
221 }
222
d6e69534
PP
223 BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF(
224 counter->msg_iter, iterator);
358340ec
PP
225
226end:
bf55043c 227 return status;
358340ec
PP
228}
229
d94d92ac 230BT_HIDDEN
4cdfc5e8 231bt_self_component_status counter_consume(
b19ff26f 232 bt_self_component_sink *comp)
358340ec 233{
4cdfc5e8 234 bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK;
358340ec 235 struct counter *counter;
4cdfc5e8 236 bt_message_iterator_status it_ret;
d6e69534
PP
237 uint64_t msg_count;
238 bt_message_array_const msgs;
358340ec 239
d94d92ac 240 counter = bt_self_component_get_data(
707b7d35 241 bt_self_component_sink_as_self_component(comp));
f6ccaed9 242 BT_ASSERT(counter);
358340ec 243
91d81473 244 if (G_UNLIKELY(!counter->msg_iter)) {
358340ec 245 try_print_last(counter);
d94d92ac 246 ret = BT_SELF_COMPONENT_STATUS_END;
358340ec
PP
247 goto end;
248 }
249
d6e69534
PP
250 /* Consume messages */
251 it_ret = bt_self_component_port_input_message_iterator_next(
252 counter->msg_iter, &msgs, &msg_count);
358340ec 253 if (it_ret < 0) {
d94d92ac 254 ret = BT_SELF_COMPONENT_STATUS_ERROR;
358340ec
PP
255 goto end;
256 }
257
258 switch (it_ret) {
d6e69534 259 case BT_MESSAGE_ITERATOR_STATUS_OK:
358340ec 260 {
d4393e08
PP
261 uint64_t i;
262
d6e69534
PP
263 for (i = 0; i < msg_count; i++) {
264 const bt_message *msg = msgs[i];
d4393e08 265
d6e69534
PP
266 BT_ASSERT(msg);
267 switch (bt_message_get_type(msg)) {
268 case BT_MESSAGE_TYPE_EVENT:
d4393e08
PP
269 counter->count.event++;
270 break;
0e847b69
PP
271 case BT_MESSAGE_TYPE_PACKET_BEGINNING:
272 counter->count.packet_begin++;
273 break;
274 case BT_MESSAGE_TYPE_PACKET_END:
275 counter->count.packet_end++;
276 break;
b9fd9cbb
FD
277 case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
278 counter->count.msg_iter_inactivity++;
d4393e08 279 break;
d6e69534 280 case BT_MESSAGE_TYPE_STREAM_BEGINNING:
d4393e08
PP
281 counter->count.stream_begin++;
282 break;
d6e69534 283 case BT_MESSAGE_TYPE_STREAM_END:
d4393e08
PP
284 counter->count.stream_end++;
285 break;
0e847b69
PP
286 case BT_MESSAGE_TYPE_STREAM_ACTIVITY_BEGINNING:
287 counter->count.stream_activity_begin++;
d4393e08 288 break;
0e847b69
PP
289 case BT_MESSAGE_TYPE_STREAM_ACTIVITY_END:
290 counter->count.stream_activity_end++;
291 break;
292 case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
293 counter->count.disc_events++;
294 break;
295 case BT_MESSAGE_TYPE_DISCARDED_PACKETS:
296 counter->count.disc_packets++;
d4393e08 297 break;
d4393e08
PP
298 default:
299 counter->count.other++;
358340ec 300 }
d4393e08 301
d6e69534 302 bt_message_put_ref(msg);
358340ec 303 }
d94d92ac
PP
304
305 ret = BT_SELF_COMPONENT_STATUS_OK;
306 break;
358340ec 307 }
d6e69534 308 case BT_MESSAGE_ITERATOR_STATUS_AGAIN:
d94d92ac
PP
309 ret = BT_SELF_COMPONENT_STATUS_AGAIN;
310 goto end;
d6e69534 311 case BT_MESSAGE_ITERATOR_STATUS_END:
d94d92ac
PP
312 try_print_last(counter);
313 ret = BT_SELF_COMPONENT_STATUS_END;
314 goto end;
d6e69534 315 case BT_MESSAGE_ITERATOR_STATUS_NOMEM:
d94d92ac
PP
316 ret = BT_SELF_COMPONENT_STATUS_NOMEM;
317 goto end;
358340ec
PP
318 default:
319 break;
320 }
321
d6e69534 322 try_print_count(counter, msg_count);
358340ec
PP
323
324end:
358340ec
PP
325 return ret;
326}
This page took 0.051966 seconds and 4 git commands to generate.