lib: make public reference count functions have strict types
[babeltrace.git] / plugins / utils / counter / counter.c
index e782a7b7b58bdc760de8b84816eb2f32b881659c..ba4c521e10408a157c7394bd4c810c1cdd5bcd78 100644 (file)
@@ -103,67 +103,69 @@ void try_print_last(struct counter *counter)
 
 void destroy_private_counter_data(struct counter *counter)
 {
-       bt_put(counter->notif_iter);
+       bt_self_component_port_input_notification_iterator_put_ref(counter->notif_iter);
        g_free(counter);
 }
 
-void counter_finalize(struct bt_private_component *component)
+BT_HIDDEN
+void counter_finalize(struct bt_self_component_sink *comp)
 {
        struct counter *counter;
 
-       BT_ASSERT(component);
-       counter = bt_private_component_get_user_data(component);
+       BT_ASSERT(comp);
+       counter = bt_self_component_get_data(
+                       bt_self_component_sink_as_self_component(comp));
        BT_ASSERT(counter);
        try_print_last(counter);
-       bt_put(counter->notif_iter);
+       bt_self_component_port_input_notification_iterator_put_ref(counter->notif_iter);
        g_free(counter);
 }
 
-enum bt_component_status counter_init(struct bt_private_component *component,
-               struct bt_value *params, UNUSED_VAR void *init_method_data)
+BT_HIDDEN
+enum bt_self_component_status counter_init(
+               struct bt_self_component_sink *component,
+               const struct bt_value *params,
+               UNUSED_VAR void *init_method_data)
 {
-       enum bt_component_status ret;
+       enum bt_self_component_status ret;
        struct counter *counter = g_new0(struct counter, 1);
-       struct bt_value *step = NULL;
-       struct bt_value *hide_zero = NULL;
+       const struct bt_value *step = NULL;
+       const struct bt_value *hide_zero = NULL;
 
        if (!counter) {
-               ret = BT_COMPONENT_STATUS_NOMEM;
-               goto end;
+               ret = BT_SELF_COMPONENT_STATUS_NOMEM;
+               goto error;
        }
 
-       ret = bt_private_component_sink_add_input_private_port(component,
+       ret = bt_self_component_sink_add_input_port(component,
                "in", NULL, NULL);
-       if (ret != BT_COMPONENT_STATUS_OK) {
-               goto end;
+       if (ret != BT_SELF_COMPONENT_STATUS_OK) {
+               goto error;
        }
 
        counter->last_printed_total = -1ULL;
        counter->step = 1000;
-       step = bt_value_map_borrow(params, "step");
+       step = bt_value_map_borrow_entry_value_const(params, "step");
        if (step && bt_value_is_integer(step)) {
                int64_t val;
 
-               (void) bt_value_integer_get(step, &val);
-
+               val = bt_value_integer_get(step);
                if (val >= 0) {
                        counter->step = (uint64_t) val;
                }
        }
 
-       hide_zero = bt_value_map_borrow(params, "hide-zero");
+       hide_zero = bt_value_map_borrow_entry_value_const(params, "hide-zero");
        if (hide_zero && bt_value_is_bool(hide_zero)) {
                bt_bool val;
 
-               (void) bt_value_bool_get(hide_zero, &val);
+               val = bt_value_bool_get(hide_zero);
                counter->hide_zero = (bool) val;
        }
 
-       ret = bt_private_component_set_user_data(component, counter);
-       if (ret != BT_COMPONENT_STATUS_OK) {
-               goto error;
-       }
-
+       bt_self_component_set_data(
+               bt_self_component_sink_as_self_component(component),
+               counter);
        goto end;
 
 error:
@@ -173,74 +175,68 @@ end:
        return ret;
 }
 
-enum bt_component_status counter_port_connected(
-               struct bt_private_component *component,
-               struct bt_private_port *self_port,
-               struct bt_port *other_port)
+BT_HIDDEN
+enum bt_self_component_status counter_port_connected(
+               struct bt_self_component_sink *comp,
+               struct bt_self_component_port_input *self_port,
+               const struct bt_port_output *other_port)
 {
-       enum bt_component_status status = BT_COMPONENT_STATUS_OK;
+       enum bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
        struct counter *counter;
-       struct bt_notification_iterator *iterator;
-       struct bt_private_connection *connection;
-       enum bt_connection_status conn_status;
+       struct bt_self_component_port_input_notification_iterator *iterator;
 
-       counter = bt_private_component_get_user_data(component);
+       counter = bt_self_component_get_data(
+               bt_self_component_sink_as_self_component(comp));
        BT_ASSERT(counter);
-       connection = bt_private_port_get_private_connection(self_port);
-       BT_ASSERT(connection);
-       conn_status = bt_private_connection_create_notification_iterator(
-               connection, &iterator);
-       if (conn_status != BT_CONNECTION_STATUS_OK) {
-               status = BT_COMPONENT_STATUS_ERROR;
+       iterator = bt_self_component_port_input_notification_iterator_create(
+               self_port);
+       if (!iterator) {
+               status = BT_SELF_COMPONENT_STATUS_NOMEM;
                goto end;
        }
 
-       BT_MOVE(counter->notif_iter, iterator);
+       BT_SELF_COMPONENT_PORT_INPUT_NOTIFICATION_ITERATOR_MOVE_REF(
+               counter->notif_iter, iterator);
 
 end:
-       bt_put(connection);
        return status;
 }
 
-enum bt_component_status counter_consume(struct bt_private_component *component)
+BT_HIDDEN
+enum bt_self_component_status counter_consume(
+               struct bt_self_component_sink *comp)
 {
-       enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
+       enum bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK;
        struct counter *counter;
        enum bt_notification_iterator_status it_ret;
        uint64_t notif_count;
-       bt_notification_array notifs;
+       bt_notification_array_const notifs;
 
-       counter = bt_private_component_get_user_data(component);
+       counter = bt_self_component_get_data(
+                       bt_self_component_sink_as_self_component(comp));
        BT_ASSERT(counter);
 
        if (unlikely(!counter->notif_iter)) {
                try_print_last(counter);
-               ret = BT_COMPONENT_STATUS_END;
+               ret = BT_SELF_COMPONENT_STATUS_END;
                goto end;
        }
 
        /* Consume notifications */
-       it_ret = bt_private_connection_notification_iterator_next(
+       it_ret = bt_self_component_port_input_notification_iterator_next(
                counter->notif_iter, &notifs, &notif_count);
        if (it_ret < 0) {
-               ret = BT_COMPONENT_STATUS_ERROR;
+               ret = BT_SELF_COMPONENT_STATUS_ERROR;
                goto end;
        }
 
        switch (it_ret) {
-       case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
-               ret = BT_COMPONENT_STATUS_AGAIN;
-               goto end;
-       case BT_NOTIFICATION_ITERATOR_STATUS_END:
-               try_print_last(counter);
-               ret = BT_COMPONENT_STATUS_END;
-               goto end;
        case BT_NOTIFICATION_ITERATOR_STATUS_OK:
        {
                uint64_t i;
 
                for (i = 0; i < notif_count; i++) {
-                       struct bt_notification *notif = notifs[i];
+                       const struct bt_notification *notif = notifs[i];
 
                        BT_ASSERT(notif);
                        switch (bt_notification_get_type(notif)) {
@@ -250,13 +246,13 @@ enum bt_component_status counter_consume(struct bt_private_component *component)
                        case BT_NOTIFICATION_TYPE_INACTIVITY:
                                counter->count.inactivity++;
                                break;
-                       case BT_NOTIFICATION_TYPE_STREAM_BEGIN:
+                       case BT_NOTIFICATION_TYPE_STREAM_BEGINNING:
                                counter->count.stream_begin++;
                                break;
                        case BT_NOTIFICATION_TYPE_STREAM_END:
                                counter->count.stream_end++;
                                break;
-                       case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
+                       case BT_NOTIFICATION_TYPE_PACKET_BEGINNING:
                                counter->count.packet_begin++;
                                break;
                        case BT_NOTIFICATION_TYPE_PACKET_END:
@@ -266,9 +262,22 @@ enum bt_component_status counter_consume(struct bt_private_component *component)
                                counter->count.other++;
                        }
 
-                       bt_put(notif);
+                       bt_notification_put_ref(notif);
                }
+
+               ret = BT_SELF_COMPONENT_STATUS_OK;
+               break;
        }
+       case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
+               ret = BT_SELF_COMPONENT_STATUS_AGAIN;
+               goto end;
+       case BT_NOTIFICATION_ITERATOR_STATUS_END:
+               try_print_last(counter);
+               ret = BT_SELF_COMPONENT_STATUS_END;
+               goto end;
+       case BT_NOTIFICATION_ITERATOR_STATUS_NOMEM:
+               ret = BT_SELF_COMPONENT_STATUS_NOMEM;
+               goto end;
        default:
                break;
        }
This page took 0.025884 seconds and 4 git commands to generate.