Set notification iterator methods to the component class

[babeltrace.git] / plugins / trimmer / iterator.h

diff --git a/plugins/trimmer/iterator.h b/plugins/trimmer/iterator.h
index a517b1f03e77118e71c993e31f5c88d022a54943..9bebf45982c1b507813d4e6fed186c8182c13cc0 100644 (file)
--- a/plugins/trimmer/iterator.h
+++ b/plugins/trimmer/iterator.h
@@ -28,20 +28,23 @@
  */
 
 #include "trimmer.h"
-#include <babeltrace/plugin/notification/notification.h>
+#include <babeltrace/component/notification/notification.h>
+#include <babeltrace/component/notification/iterator.h>
 
 struct trimmer_iterator {
        /* Input iterators associated with this output iterator. */
        GPtrArray *input_iterator_group;
        struct bt_notification *current_notification;
-       GHashTable *stream_to_packet_start_notification;
 };
 
 BT_HIDDEN
-enum bt_component_status trimmer_iterator_init(
+enum bt_notification_iterator_status trimmer_iterator_init(
                struct bt_component *component,
                struct bt_notification_iterator *iterator);
 
+BT_HIDDEN
+void trimmer_iterator_destroy(struct bt_notification_iterator *it);
+
 BT_HIDDEN
 struct bt_notification *trimmer_iterator_get(
                struct bt_notification_iterator *iterator);