lib/component/sink.c

   1 /*
   2  * sink.c
   3  *
   4  * Babeltrace Sink Component
   5  *
   6  * Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   7  *
   8  * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
   9  *
  10  * Permission is hereby granted, free of charge, to any person obtaining a copy
  11  * of this software and associated documentation files (the "Software"), to deal
  12  * in the Software without restriction, including without limitation the rights
  13  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  14  * copies of the Software, and to permit persons to whom the Software is
  15  * furnished to do so, subject to the following conditions:
  16  *
  17  * The above copyright notice and this permission notice shall be included in
  18  * all copies or substantial portions of the Software.
  19  *
  20  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  23  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  25  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  26  * SOFTWARE.
  27  */
  28
  29 #include <babeltrace/compiler.h>
  30 #include <babeltrace/values.h>
  31 #include <babeltrace/component/component-sink-internal.h>
  32 #include <babeltrace/component/component-internal.h>
  33 #include <babeltrace/component/notification/notification.h>
  34
  35 BT_HIDDEN
  36 enum bt_component_status bt_component_sink_validate(
  37                 struct bt_component *component)
  38 {
  39         enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
  40
  41         if (!component) {
  42                 ret = BT_COMPONENT_STATUS_INVALID;
  43                 goto end;
  44         }
  45
  46         if (!component->class) {
  47                 ret = BT_COMPONENT_STATUS_INVALID;
  48                 goto end;
  49         }
  50
  51         if (component->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
  52                 ret = BT_COMPONENT_STATUS_INVALID;
  53                 goto end;
  54         }
  55 end:
  56         return ret;
  57 }
  58
  59 static
  60 void bt_component_sink_destroy(struct bt_component *component)
  61 {
  62         struct bt_component_sink *sink = container_of(component,
  63                         struct bt_component_sink, parent);
  64
  65         if (sink->input_ports) {
  66                 g_ptr_array_free(sink->input_ports, TRUE);
  67         }
  68 }
  69
  70 BT_HIDDEN
  71 struct bt_component *bt_component_sink_create(
  72                 struct bt_component_class *class, struct bt_value *params)
  73 {
  74         struct bt_component_sink *sink = NULL;
  75         enum bt_component_status ret;
  76
  77         sink = g_new0(struct bt_component_sink, 1);
  78         if (!sink) {
  79                 goto end;
  80         }
  81
  82         sink->parent.class = bt_get(class);
  83         ret = bt_component_init(&sink->parent, bt_component_sink_destroy);
  84         if (ret != BT_COMPONENT_STATUS_OK) {
  85                 goto error;
  86         }
  87
  88 /*
  89         ret = bt_component_sink_register_notification_type(&sink->parent,
  90                 BT_NOTIFICATION_TYPE_EVENT);
  91         if (ret != BT_COMPONENT_STATUS_OK) {
  92                 goto error;
  93         }
  94 */
  95         ret = bt_component_init_input_ports(&sink->parent,
  96                         &sink->input_ports);
  97         if (ret) {
  98                 goto error;
  99         }
 100
 101 end:
 102         return sink ? &sink->parent : NULL;
 103 error:
 104         BT_PUT(sink);
 105         return NULL;
 106 }
 107
 108 BT_HIDDEN
 109 enum bt_component_status bt_component_sink_consume(
 110                 struct bt_component *component)
 111 {
 112         enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
 113         struct bt_component_class_sink *sink_class = NULL;
 114
 115         if (!component) {
 116                 ret = BT_COMPONENT_STATUS_INVALID;
 117                 goto end;
 118         }
 119
 120         if (bt_component_get_class_type(component) != BT_COMPONENT_CLASS_TYPE_SINK) {
 121                 ret = BT_COMPONENT_STATUS_UNSUPPORTED;
 122                 goto end;
 123         }
 124
 125         sink_class = container_of(component->class, struct bt_component_class_sink, parent);
 126         assert(sink_class->methods.consume);
 127         ret = sink_class->methods.consume(component);
 128 end:
 129         return ret;
 130 }
 131 /*
 132 static
 133 enum bt_component_status bt_component_sink_register_notification_type(
 134                 struct bt_component *component, enum bt_notification_type type)
 135 {
 136         enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
 137         struct bt_component_sink *sink = NULL;
 138
 139         if (!component) {
 140                 ret = BT_COMPONENT_STATUS_INVALID;
 141                 goto end;
 142         }
 143
 144         if (bt_component_get_class_type(component) != BT_COMPONENT_CLASS_TYPE_SINK) {
 145                 ret = BT_COMPONENT_STATUS_UNSUPPORTED;
 146                 goto end;
 147         }
 148
 149         if (type <= BT_NOTIFICATION_TYPE_UNKNOWN ||
 150                 type >= BT_NOTIFICATION_TYPE_NR) {
 151                 ret = BT_COMPONENT_STATUS_INVALID;
 152                 goto end;
 153         }
 154         sink = container_of(component, struct bt_component_sink, parent);
 155         if (type == BT_NOTIFICATION_TYPE_ALL) {
 156                 sink->registered_notifications_mask = ~(notification_mask_t) 0;
 157         } else {
 158                 sink->registered_notifications_mask |=
 159                         (notification_mask_t) 1 << type;
 160         }
 161 end:
 162         return ret;
 163 }
 164 */
 165
 166 int bt_component_sink_get_input_port_count(struct bt_component *component)
 167 {
 168         int ret;
 169         struct bt_component_sink *sink;
 170
 171         if (!component) {
 172                 ret = -1;
 173                 goto end;
 174         }
 175
 176         if (component->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
 177                 ret = -1;
 178                 goto end;
 179         }
 180
 181         sink = container_of(component, struct bt_component_sink, parent);
 182         ret = sink->input_ports->len;
 183 end:
 184         return ret;
 185 }
 186
 187 struct bt_port *bt_component_sink_get_input_port(
 188                 struct bt_component *component, const char *name)
 189 {
 190         struct bt_component_sink *sink;
 191         struct bt_port *ret_port = NULL;
 192
 193         if (!component || !name ||
 194                         component->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
 195                 goto end;
 196         }
 197
 198         sink = container_of(component, struct bt_component_sink, parent);
 199         ret_port = bt_component_get_port(sink->input_ports, name);
 200 end:
 201         return ret_port;
 202 }
 203
 204 struct bt_port *bt_component_sink_get_input_port_at_index(
 205                 struct bt_component *component, int index)
 206 {
 207         struct bt_port *port = NULL;
 208         struct bt_component_sink *sink;
 209
 210         if (!component ||
 211                         component->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
 212                 goto end;
 213         }
 214
 215         sink = container_of(component, struct bt_component_sink, parent);
 216         port = bt_component_get_port_at_index(sink->input_ports, index);
 217 end:
 218         return port;
 219 }
 220
 221 struct bt_port *bt_component_sink_get_default_input_port(
 222                 struct bt_component *component)
 223 {
 224         return bt_component_sink_get_input_port(component,
 225                         DEFAULT_INPUT_PORT_NAME);
 226 }
 227
 228 struct bt_port *bt_component_sink_add_input_port(
 229                 struct bt_component *component, const char *name)
 230 {
 231         struct bt_port *port;
 232         struct bt_component_sink *sink;
 233
 234         if (!component ||
 235                         component->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
 236                 port = NULL;
 237                 goto end;
 238         }
 239
 240         sink = container_of(component, struct bt_component_sink, parent);
 241         port = bt_component_add_port(component, sink->input_ports,
 242                         BT_PORT_TYPE_INPUT, name);
 243 end:
 244         return port;
 245 }
 246
 247 enum bt_component_status bt_component_sink_remove_input_port(
 248                 struct bt_component *component, const char *name)
 249 {
 250         enum bt_component_status status;
 251         struct bt_component_sink *sink;
 252
 253         if (!component ||
 254                         component->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
 255                 status = BT_COMPONENT_STATUS_INVALID;
 256                 goto end;
 257         }
 258
 259         sink = container_of(component, struct bt_component_sink, parent);
 260         status = bt_component_remove_port(component, sink->input_ports,
 261                         name);
 262 end:
 263         return status;
 264 }