babeltrace2 0.1.0

data/lib/babeltrace2/graph/interrupter.rb

@@ -0,0 +1,57 @@
+module Babeltrace2
+
+  attach_function :bt_interrupter_create,
+                  [],
+                  :bt_interrupter_handle
+
+  attach_function :bt_interrupter_set,
+                  [ :bt_interrupter_handle ],
+                  :void
+
+  attach_function :bt_interrupter_reset,
+                  [ :bt_interrupter_handle ],
+                  :void
+
+  attach_function :bt_interrupter_is_set,
+                  [ :bt_interrupter_handle ],
+                  :bt_bool
+
+  attach_function :bt_interrupter_get_ref,
+                  [ :bt_interrupter_handle ],
+                  :void
+
+  attach_function :bt_interrupter_put_ref,
+                  [ :bt_interrupter_handle ],
+                  :void
+
+  class BTInterrupter < BTSharedObject
+    @get_ref = :bt_interrupter_get_ref
+    @put_ref = :bt_interrupter_put_ref
+
+    def initialize(handle = nil, retain: true, auto_release: true)
+      if handle
+        super(handle, retain: retain, auto_release: auto_release)
+      else
+        handle = Babeltrace2.bt_interrupter_create()
+        raise Babeltrace2.process_error if handle.null?
+        super(handle)
+      end
+    end
+
+    def set
+      Babeltrace2.bt_interrupter_set(@handle)
+    end
+    alias set! set
+
+    def reset
+      Babeltrace2.bt_interrupter_reset(@handle)
+    end
+    alias reset! reset
+
+    def is_set
+      Babeltrace2.bt_interrupter_is_set(@handle) != BT_FALSE
+    end
+    alias set? is_set
+  end
+
+end

data/lib/babeltrace2/graph/message-iterator-class.rb

@@ -0,0 +1,374 @@
+module Babeltrace2
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+
+  BTMessageIteratorClassCanSeekBeginningMethodStatus =
+    enum :bt_message_iterator_class_can_seek_beginning_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_ERROR ]
+
+  callback :bt_message_iterator_class_can_seek_beginning_method,
+           [ :bt_self_message_iterator_handle,
+             :pointer ],
+           :bt_message_iterator_class_can_seek_beginning_method_status
+
+  def self._wrap_message_iterator_class_can_seek_beginning_method(handle, method)
+    id = handle.to_i
+    method_wrapper = lambda { |self_message_iterator, can_seek_beginning|
+      begin
+        csb = method.call(BTSelfMessageIterator.new(self_message_iterator,
+                            retain: false, auto_release: false))
+        can_seek_beginning.write_int(csb ? BT_TRUE : BT_FALSE)
+        :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_OK
+      rescue Exception => e
+        Babeltrace2.stack_ruby_error(e, source: self_message_iterator)
+        :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_BEGINNING_METHOD_STATUS_ERROR
+      end
+    }
+    @@callbacks[id][:can_seek_beginning_method] = method_wrapper
+    method_wrapper
+  end
+
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorClassCanSeekNSFromOriginMethodStatus =
+    enum :bt_message_iterator_class_can_seek_ns_from_origin_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR ]
+
+  callback :bt_message_iterator_class_can_seek_ns_from_origin_method,
+           [ :bt_self_message_iterator_handle,
+             :int64, :pointer ],
+           :bt_message_iterator_class_can_seek_ns_from_origin_method_status
+
+  def self._wrap_message_iterator_class_can_seek_ns_from_origin_method(handle, method)
+    id = handle.to_i
+    method_wrapper = lambda { |self_message_iterator, ns_from_origin, can_seek_beginning|
+      begin
+        csb = method.call(BTSelfMessageIterator.new(self_message_iterator,
+                            retain: false, auto_release: false), ns_from_origin)
+        can_seek_beginning.write_int(csb ? BT_TRUE : BT_FALSE)
+        :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK
+      rescue Exception => e
+        Babeltrace2.stack_ruby_error(e, source: self_message_iterator)
+        :BT_MESSAGE_ITERATOR_CLASS_CAN_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR
+      end
+    }
+    @@callbacks[id][:can_seek_ns_from_origin_method] = method_wrapper
+    method_wrapper
+  end
+
+  callback :bt_message_iterator_class_finalize_method,
+           [ :bt_self_message_iterator_handle ],
+           :void
+
+  def self._wrap_message_iterator_class_finalize_method(handle, method)
+    id = handle.to_i
+    method_wrapper = lambda { |self_message_iterator|
+      begin
+        method.call(BTSelfMessageIterator.new(self_message_iterator,
+                            retain: false, auto_release: false))
+      rescue Exception => e
+        puts e
+      end
+    }
+    @@callbacks[id][:finalize_method] = method_wrapper
+    method_wrapper
+  end
+
+  BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorClassInitializeMethodStatus =
+    enum :bt_message_iterator_class_initialize_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_ERROR ]
+
+  callback :bt_message_iterator_class_initialize_method,
+           [ :bt_self_message_iterator_handle,
+             :bt_self_message_iterator_configuration_handle,
+             :bt_self_component_port_output_handle ],
+           :bt_message_iterator_class_initialize_method_status
+
+  def self._wrap_message_iterator_class_initialize_method(handle, method)
+    id = handle.to_i
+    method_wrapper = lambda { |self_message_iterator, configuration, port|
+      begin
+        method.call(BTSelfMessageIterator.new(self_message_iterator,
+                      retain: false, auto_release: false),
+                    BTSelfMessageIteratorConfiguration.new(configuration),
+                    BTSelfComponentPortOutput.new(port,
+                      retain: false, auto_release: false))
+        :BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK
+      rescue Exception => e
+        Babeltrace2.stack_ruby_error(e, source: self_message_iterator)
+        :BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_ERROR
+      end
+    }
+    @@callbacks[id][:initialize_method] = method_wrapper
+    method_wrapper
+  end
+
+  BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_END = BT_FUNC_STATUS_END
+  BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorClassNextMethodStatus =
+    enum :bt_message_iterator_class_next_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_END,
+       BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_END,
+      :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR ]
+
+  callback :bt_message_iterator_class_next_method,
+           [ :bt_self_message_iterator_handle,
+             :bt_message_array_const,
+             :uint64, :pointer ],
+           :bt_message_iterator_class_next_method_status
+
+  def self._wrap_message_iterator_class_next_method(method)
+    lambda { |self_message_iterator, messages, capacity, count|
+      begin
+        mess = method.call(BTSelfMessageIterator.new(self_message_iterator,
+                             retain: false, auto_release: false),
+                           capacity)
+        if mess.size < capacity
+          mess.each { |m| bt_message_get_ref(m.handle) }
+          messages.write_array_of_pointer(mess.collect(&:handle))
+          count.write_uint64(mess.size)
+          :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK
+        else
+          puts "Too many messages for capacity"
+          :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR
+        end
+      rescue StopIteration
+        :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_END
+      rescue Exception => e
+        Babeltrace2.stack_ruby_error(e, source: self_message_iterator)
+        :BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR
+      end
+    }
+  end
+
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorClassSeekBeginningMethodStatus =
+    enum :bt_message_iterator_class_seek_beginning_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_ERROR ]
+
+  callback :bt_message_iterator_class_seek_beginning_method,
+           [ :bt_self_message_iterator_handle ],
+           :bt_message_iterator_class_seek_beginning_method_status
+
+  def self._wrap_message_iterator_class_seek_beginning_method(handle, method)
+    id = handle.to_i
+    method_wrapper = lambda { |self_message_iterator|
+      begin
+        method.call(BTSelfMessageIterator.new(self_message_iterator,
+                      retain: false, auto_release: false))
+        :BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_OK
+      rescue Exception => e
+        Babeltrace2.stack_ruby_error(e, source: self_message_iterator)
+        :BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_ERROR
+      end
+    }
+    @@callbacks[id][:seek_beginning_method] = method_wrapper
+    method_wrapper
+  end
+
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorClassSeekNSFromOriginMethodStatus =
+    enum :bt_message_iterator_class_seek_ns_from_origin_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR ]
+
+  callback :bt_message_iterator_class_seek_ns_from_origin_method,
+           [ :bt_self_message_iterator_handle,
+             :int64 ],
+           :bt_message_iterator_class_seek_ns_from_origin_method_status
+
+  def self._wrap_message_iterator_class_seek_ns_from_origin_method(handle, method)
+    id = handle.to_i
+    method_wrapper = lambda { |self_message_iterator, ns_from_origin|
+      begin
+        method.call(BTSelfMessageIterator.new(self_message_iterator,
+          retain: false, auto_release: false), ns_from_origin)
+        :BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_OK
+      rescue Exception => e
+        Babeltrace2.stack_ruby_error(e, source: self_message_iterator)
+        :BT_MESSAGE_ITERATOR_CLASS_SEEK_NS_FROM_ORIGIN_METHOD_STATUS_ERROR
+      end
+    }
+    @@callbacks[id][:seek_ns_from_origin_method] = method_wrapper
+    method_wrapper
+  end
+
+  attach_function :bt_message_iterator_class_create,
+                  [ :bt_message_iterator_class_next_method ],
+                  :bt_message_iterator_class_handle
+
+  BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK = BT_FUNC_STATUS_OK
+  BTMessageIteratorClassSetMethodStatus =
+    enum :bt_message_iterator_class_set_method_status,
+    [ :BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK ]
+
+  attach_function :bt_message_iterator_class_set_finalize_method,
+                  [ :bt_message_iterator_class_handle,
+                    :bt_message_iterator_class_finalize_method ],
+                  :bt_message_iterator_class_set_method_status
+
+  attach_function :bt_message_iterator_class_set_initialize_method,
+                  [ :bt_message_iterator_class_handle,
+                    :bt_message_iterator_class_initialize_method ],
+                  :bt_message_iterator_class_set_method_status
+
+  attach_function :bt_message_iterator_class_set_seek_beginning_methods,
+                  [ :bt_message_iterator_class_handle,
+                    :bt_message_iterator_class_seek_beginning_method,
+                    :bt_message_iterator_class_can_seek_beginning_method ],
+                  :bt_message_iterator_class_set_method_status
+
+  attach_function :bt_message_iterator_class_set_seek_ns_from_origin_methods,
+                  [ :bt_message_iterator_class_handle,
+                    :bt_message_iterator_class_seek_ns_from_origin_method,
+                    :bt_message_iterator_class_can_seek_ns_from_origin_method ],
+                  :bt_message_iterator_class_set_method_status
+
+  attach_function :bt_message_iterator_class_get_ref,
+                  [ :bt_message_iterator_class_handle],
+                  :void
+
+  attach_function :bt_message_iterator_class_put_ref,
+                  [ :bt_message_iterator_class_handle],
+                  :void
+
+  class BTMessageIteratorClass < BTSharedObject
+    CanSeekBeginningMethodStatus = BTMessageIteratorClassCanSeekBeginningMethodStatus
+    CanSeekNSFromOriginMethodStatus = BTMessageIteratorClassCanSeekNSFromOriginMethodStatus
+    InitializeMethodStatus = BTMessageIteratorClassInitializeMethodStatus
+    NextMethodStatus = BTMessageIteratorClassNextMethodStatus
+    SeekBeginningMethodStatus = BTMessageIteratorClassSeekBeginningMethodStatus
+    SeekNSFromOriginMethodStatus = BTMessageIteratorClassSeekNSFromOriginMethodStatus
+    SetMethodStatus = BTMessageIteratorClassSetMethodStatus
+    @get_ref = :bt_message_iterator_class_get_ref
+    @put_ref = :bt_message_iterator_class_put_ref
+
+    def initialize(handle = nil, retain: true, auto_release: true,
+                   next_method: nil)
+      if handle
+        super(handle, retain: retain, auto_release: auto_release)
+      else
+        raise ArgumentError, "invalid value for next_method" unless next_method
+        next_method = Babeltrace2._wrap_message_iterator_class_next_method(next_method)
+        handle = Babeltrace2.bt_message_iterator_class_create(next_method)
+        raise Babeltrace2.process_error if handle.null?
+        Babeltrace2._callbacks[handle.to_i][:next_method] = next_method
+        super(handle, retain: false)
+      end
+    end
+
+    def set_finalize_method(method)
+      method = Babeltrace2._wrap_message_iterator_class_finalize_method(@handle, method)
+      res = Babeltrace2.bt_message_iterator_class_set_finalize_method(@handle, method)
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK
+      self
+    end
+
+    def finalize_method=(method)
+      set_finalize_method(method)
+      method
+    end
+
+    def set_initialize_method(method)
+      method = Babeltrace2._wrap_message_iterator_class_initialize_method(@handle, method)
+      res = Babeltrace2.bt_message_iterator_class_set_initialize_method(@handle, method)
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK
+      self
+    end
+
+    def initialize_method=(method)
+      set_initialize_method(method)
+      method
+    end
+
+    def set_seek_beginning_methods(seek_method, can_seek_method: nil)
+      seek_method = Babeltrace2._wrap_message_iterator_class_seek_beginning_method(
+                      @handle, seek_method)
+      can_seek_method =
+        Babeltrace2._wrap_message_iterator_class_can_seek_beginning_method(
+          @handle, can_seek_method) if can_seek_method
+      res = Babeltrace2.bt_message_iterator_class_set_seek_beginning_methods(
+        @handle, seek_method, can_seek_method)
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK
+      self
+    end
+
+    def seek_beginning_method=(seek_method)
+      set_seek_beginning_methods(seek_method)
+      seek_method
+    end
+
+    def set_seek_ns_from_origin_methods(seek_method, can_seek_method: nil)
+      seek_method = Babeltrace2._wrap_message_iterator_class_seek_ns_from_origin_method(
+                      @handle, seek_method)
+      can_seek_method =
+        Babeltrace2._wrap_message_iterator_class_can_seek_ns_from_origin_method(
+          @handle, can_seek_method) if can_seek_method
+      res = Babeltrace2.bt_message_iterator_class_set_seek_ns_from_origin_methods(
+        @handle, seek_method, can_seek_method)
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK
+      self
+    end
+
+    def seek_ns_from_origin_method=(seek_method)
+      set_seek_ns_from_origin_methods(seek_method)
+      seek_method
+    end
+  end
+end

data/lib/babeltrace2/graph/message-iterator.rb

@@ -0,0 +1,253 @@
+module Babeltrace2
+  BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorCreateFromMessageIteratorStatus =
+    enum :bt_message_iterator_create_from_message_iterator_status,
+    [ :BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_create_from_message_iterator,
+                  [ :bt_self_message_iterator_handle,
+                    :bt_self_component_port_input_handle,
+                    :pointer ],
+                  :bt_message_iterator_create_from_message_iterator_status
+
+  BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorCreateFromSinkComponentStatus =
+    enum :bt_message_iterator_create_from_sink_component_status,
+    [ :BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_create_from_sink_component,
+                  [ :bt_self_component_sink_handle,
+                    :bt_self_component_port_input_handle,
+                    :pointer ],
+                  :bt_message_iterator_create_from_sink_component_status
+
+  attach_function :bt_message_iterator_borrow_component,
+                  [ :bt_message_iterator_handle ],
+                  :bt_component_handle
+
+  BT_MESSAGE_ITERATOR_NEXT_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_NEXT_STATUS_END = BT_FUNC_STATUS_END
+  BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_NEXT_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorNextStatus = enum :bt_message_iterator_next_status,
+    [ :BT_MESSAGE_ITERATOR_NEXT_STATUS_OK,
+       BT_MESSAGE_ITERATOR_NEXT_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_NEXT_STATUS_END,
+       BT_MESSAGE_ITERATOR_NEXT_STATUS_END,
+      :BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_NEXT_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_NEXT_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_next,
+                  [ :bt_message_iterator_handle,
+                    :pointer, :pointer ],
+                  :bt_message_iterator_next_status
+
+  BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorCanSeekBeginningStatus =
+    enum :bt_message_iterator_can_seek_beginning_status,
+    [ :BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_can_seek_beginning,
+                  [ :bt_message_iterator_handle,
+                    :pointer ],
+                  :bt_message_iterator_can_seek_beginning_status
+
+  BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorSeekBeginningStatus =
+    enum :bt_message_iterator_seek_beginning_status,
+    [ :BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_OK,
+       BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_seek_beginning,
+                  [ :bt_message_iterator_handle ],
+                  :bt_message_iterator_seek_beginning_status
+
+  BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorCanSeekNSFromOriginStatus =
+    enum :bt_message_iterator_can_seek_ns_from_origin_status,
+    [ :BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_OK,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_can_seek_ns_from_origin,
+                  [ :bt_message_iterator_handle,
+                    :int64, :pointer ],
+                  :bt_message_iterator_can_seek_ns_from_origin_status
+
+  BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN = BT_FUNC_STATUS_AGAIN
+  BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_ERROR = BT_FUNC_STATUS_ERROR
+  BTMessageIteratorSeekNSFromOriginStatus =
+    enum :bt_message_iterator_seek_ns_from_origin_status,
+    [ :BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_OK,
+       BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_OK,
+      :BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN,
+       BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN,
+      :BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_MEMORY_ERROR,
+       BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_MEMORY_ERROR,
+      :BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_ERROR,
+       BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_ERROR ]
+
+  attach_function :bt_message_iterator_seek_ns_from_origin,
+                  [ :bt_message_iterator_handle,
+                    :int64 ],
+                  :bt_message_iterator_seek_ns_from_origin_status
+
+  attach_function :bt_message_iterator_can_seek_forward,
+                  [ :bt_message_iterator_handle ],
+                  :bt_bool
+
+  attach_function :bt_message_iterator_get_ref,
+                  [ :bt_message_iterator_handle ],
+                  :void
+
+  attach_function :bt_message_iterator_put_ref,
+                  [ :bt_message_iterator_handle ],
+                  :void
+
+  class BTMessageIterator < BTSharedObject
+    CreateFromMessageIteratorStatus = BTMessageIteratorCreateFromMessageIteratorStatus
+    CreateFromSinkComponentStatus = BTMessageIteratorCreateFromSinkComponentStatus
+    NextStatus = BTMessageIteratorNextStatus
+    CanSeekBeginningStatus = BTMessageIteratorCanSeekBeginningStatus
+    SeekBeginningStatus = BTMessageIteratorSeekBeginningStatus
+    CanSeekNSFromOriginStatus = BTMessageIteratorCanSeekNSFromOriginStatus
+    @get_ref = :bt_message_iterator_get_ref
+    @put_ref = :bt_message_iterator_put_ref
+
+    def self.create_from_message_iterator(self_message_iterator, port)
+      ptr = FFI::MemoryPointer.new(:pointer)
+      res = Babeltrace2.bt_message_iterator_create_from_message_iterator(self_message_iterator, port, ptr)
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK
+      BTMessageIterator.new(BTMessageIteratorHandle.new(ptr.read_pointer), retain: false)
+    end
+
+    def create_message_iterator(port)
+      BTMessageIterator.create_from_message_iterator(self, port)
+    end
+
+    def self.create_from_sink_component(self_component_sink, port)
+      ptr = FFI::MemoryPointer.new(:pointer)
+      res = Babeltrace2.bt_message_iterator_create_from_sink_component(self_component_sink, port, ptr)
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK
+      BTMessageIterator.new(BTMessageIteratorHandle.new(ptr.read_pointer), retain: false)
+    end
+
+    def get_component
+      handle = Babeltrace2.bt_message_iterator_borrow_component(@handle)
+      BTComponent.from_handle(handle)
+    end
+    alias component get_component
+
+    def next_messages
+      ptr_messages = FFI::MemoryPointer.new(:pointer)
+      ptr_count = FFI::MemoryPointer.new(:uint64)
+      while ((res = Babeltrace2.bt_message_iterator_next(@handle, ptr_messages, ptr_count)) == :BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN)
+        puts "waiting"
+        sleep BT_SLEEP_TIME
+      end
+      case res
+      when :BT_MESSAGE_ITERATOR_NEXT_STATUS_OK
+        count = ptr_count.read_uint64
+        messages = ptr_messages.read_pointer
+        return messages.read_array_of_pointer(count).collect { |h|
+          BTMessage.from_handle(BTMessageHandle.new(h), retain: false)
+        }
+      when :BT_MESSAGE_ITERATOR_NEXT_STATUS_END
+        raise StopIteration
+      else
+        raise Babeltrace2.process_error(res)
+      end
+    end
+
+    def can_seek_beginning
+      ptr = FFI::MemoryPointer.new(:int)
+      while ((res = Babeltrace2.bt_message_iterator_can_seek_beginning(@handle, ptr)) == :BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_AGAIN)
+        sleep BT_SLEEP_TIME
+      end
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_STATUS_OK
+      ptr.read_int != BT_FALSE
+    end
+    alias can_seek_beginning? can_seek_beginning
+
+    def seek_beginning
+      raise "invalid operation" unless can_seek_beginning?
+      while ((res = Babeltrace2.bt_message_iterator_seek_beginning(@handle)) == :BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_AGAIN)
+        sleep BT_SLEEP_TIME
+      end
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_SEEK_BEGINNING_STATUS_OK
+      self
+    end
+
+    def can_seek_ns_from_origin(ns)
+      ptr = FFI::MemoryPointer.new(:int)
+      while ((res = Babeltrace2.bt_message_iterator_can_seek_ns_from_origin(@handle, ns, ptr)) == :BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN)
+        sleep BT_SLEEP_TIME
+      end
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_CAN_SEEK_NS_FROM_ORIGIN_STATUS_OK
+      ptr.read_int != BT_FALSE
+    end
+    alias can_seek_ns_from_origin? can_seek_ns_from_origin
+
+    def seek_ns_from_origin(ns)
+      raise "invalid operation" unless can_seek_ns_from_origin(ns)
+      while ((res = Babeltrace2.bt_message_iterator_seek_ns_from_origin@handle, ns) == :BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_AGAIN)
+        sleep BT_SLEEP_TIME
+      end
+      raise Babeltrace2.process_error(res) if res != :BT_MESSAGE_ITERATOR_SEEK_NS_FROM_ORIGIN_STATUS_OK
+      self
+    end
+
+    def can_seek_forward
+      Babeltrace2.bt_message_iterator_can_seek_forward(@handle) != BT_FALSE
+    end
+    alias can_seek_forward? can_seek_forward
+  end
+end