Commit | Line | Data |
---|---|---|
e0dfa761 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 | ||
3fadfbc0 | 23 | #include <babeltrace2/babeltrace.h> |
91d81473 | 24 | #include "common/macros.h" |
578e048b | 25 | #include "common/assert.h" |
c2b71c92 JG |
26 | #include "dummy.h" |
27 | ||
5badd463 PP |
28 | static |
29 | const char * const in_port_name = "in"; | |
30 | ||
c2b71c92 JG |
31 | void destroy_private_dummy_data(struct dummy *dummy) |
32 | { | |
d6e69534 | 33 | bt_self_component_port_input_message_iterator_put_ref(dummy->msg_iter); |
c2b71c92 | 34 | g_free(dummy); |
d94d92ac | 35 | |
c2b71c92 JG |
36 | } |
37 | ||
d94d92ac | 38 | BT_HIDDEN |
b19ff26f | 39 | void dummy_finalize(bt_self_component_sink *comp) |
c2b71c92 JG |
40 | { |
41 | struct dummy *dummy; | |
42 | ||
d94d92ac PP |
43 | BT_ASSERT(comp); |
44 | dummy = bt_self_component_get_data( | |
707b7d35 | 45 | bt_self_component_sink_as_self_component(comp)); |
f6ccaed9 | 46 | BT_ASSERT(dummy); |
c2b71c92 JG |
47 | destroy_private_dummy_data(dummy); |
48 | } | |
49 | ||
d94d92ac | 50 | BT_HIDDEN |
4cdfc5e8 | 51 | bt_self_component_status dummy_init( |
b19ff26f PP |
52 | bt_self_component_sink *component, |
53 | const bt_value *params, | |
c88dd1cb | 54 | __attribute__((unused)) void *init_method_data) |
c2b71c92 | 55 | { |
4cdfc5e8 | 56 | bt_self_component_status ret; |
c2b71c92 JG |
57 | struct dummy *dummy = g_new0(struct dummy, 1); |
58 | ||
59 | if (!dummy) { | |
d94d92ac | 60 | ret = BT_SELF_COMPONENT_STATUS_NOMEM; |
c2b71c92 JG |
61 | goto end; |
62 | } | |
63 | ||
d94d92ac | 64 | ret = bt_self_component_sink_add_input_port(component, |
147337a3 | 65 | "in", NULL, NULL); |
d94d92ac | 66 | if (ret != BT_SELF_COMPONENT_STATUS_OK) { |
c2b71c92 JG |
67 | goto error; |
68 | } | |
d19348b6 | 69 | |
d94d92ac | 70 | bt_self_component_set_data( |
707b7d35 | 71 | bt_self_component_sink_as_self_component(component), dummy); |
d19348b6 PP |
72 | goto end; |
73 | ||
c2b71c92 JG |
74 | error: |
75 | destroy_private_dummy_data(dummy); | |
d19348b6 PP |
76 | |
77 | end: | |
c2b71c92 JG |
78 | return ret; |
79 | } | |
80 | ||
d94d92ac | 81 | BT_HIDDEN |
5badd463 PP |
82 | bt_self_component_status dummy_graph_is_configured( |
83 | bt_self_component_sink *comp) | |
c2b71c92 | 84 | { |
4cdfc5e8 | 85 | bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK; |
c2b71c92 | 86 | struct dummy *dummy; |
d6e69534 | 87 | bt_self_component_port_input_message_iterator *iterator; |
c2b71c92 | 88 | |
d94d92ac | 89 | dummy = bt_self_component_get_data( |
707b7d35 | 90 | bt_self_component_sink_as_self_component(comp)); |
f6ccaed9 | 91 | BT_ASSERT(dummy); |
d6e69534 | 92 | iterator = bt_self_component_port_input_message_iterator_create( |
5badd463 PP |
93 | bt_self_component_sink_borrow_input_port_by_name(comp, |
94 | in_port_name)); | |
d94d92ac PP |
95 | if (!iterator) { |
96 | status = BT_SELF_COMPONENT_STATUS_NOMEM; | |
c2b71c92 JG |
97 | goto end; |
98 | } | |
99 | ||
d6e69534 PP |
100 | BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF( |
101 | dummy->msg_iter, iterator); | |
0d8b4d8e | 102 | |
c2b71c92 | 103 | end: |
bf55043c | 104 | return status; |
c2b71c92 | 105 | } |
e0dfa761 | 106 | |
d94d92ac | 107 | BT_HIDDEN |
4cdfc5e8 | 108 | bt_self_component_status dummy_consume( |
b19ff26f | 109 | bt_self_component_sink *component) |
e0dfa761 | 110 | { |
4cdfc5e8 | 111 | bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK; |
d6e69534 | 112 | bt_message_array_const msgs; |
d4393e08 | 113 | uint64_t count; |
c2b71c92 | 114 | struct dummy *dummy; |
4cdfc5e8 | 115 | bt_message_iterator_status it_ret; |
d4393e08 | 116 | uint64_t i; |
e0dfa761 | 117 | |
d94d92ac | 118 | dummy = bt_self_component_get_data( |
707b7d35 | 119 | bt_self_component_sink_as_self_component(component)); |
f6ccaed9 | 120 | BT_ASSERT(dummy); |
e0dfa761 | 121 | |
91d81473 | 122 | if (G_UNLIKELY(!dummy->msg_iter)) { |
d94d92ac | 123 | ret = BT_SELF_COMPONENT_STATUS_END; |
d19348b6 | 124 | goto end; |
e0dfa761 PP |
125 | } |
126 | ||
d6e69534 PP |
127 | /* Consume one message */ |
128 | it_ret = bt_self_component_port_input_message_iterator_next( | |
129 | dummy->msg_iter, &msgs, &count); | |
d19348b6 | 130 | switch (it_ret) { |
d6e69534 | 131 | case BT_MESSAGE_ITERATOR_STATUS_OK: |
d94d92ac | 132 | ret = BT_SELF_COMPONENT_STATUS_OK; |
d4393e08 PP |
133 | |
134 | for (i = 0; i < count; i++) { | |
d6e69534 | 135 | bt_message_put_ref(msgs[i]); |
d4393e08 PP |
136 | } |
137 | ||
138 | break; | |
d6e69534 | 139 | case BT_MESSAGE_ITERATOR_STATUS_AGAIN: |
d94d92ac | 140 | ret = BT_SELF_COMPONENT_STATUS_AGAIN; |
d19348b6 | 141 | goto end; |
d6e69534 | 142 | case BT_MESSAGE_ITERATOR_STATUS_END: |
d94d92ac | 143 | ret = BT_SELF_COMPONENT_STATUS_END; |
d19348b6 | 144 | goto end; |
d6e69534 | 145 | case BT_MESSAGE_ITERATOR_STATUS_ERROR: |
d94d92ac PP |
146 | ret = BT_SELF_COMPONENT_STATUS_ERROR; |
147 | goto end; | |
d6e69534 | 148 | case BT_MESSAGE_ITERATOR_STATUS_NOMEM: |
d94d92ac | 149 | ret = BT_SELF_COMPONENT_STATUS_NOMEM; |
d4393e08 | 150 | goto end; |
d19348b6 PP |
151 | default: |
152 | break; | |
e0dfa761 | 153 | } |
d19348b6 | 154 | |
e0dfa761 | 155 | end: |
e0dfa761 PP |
156 | return ret; |
157 | } |