babeltrace2 0.1.0

data/lib/babeltrace2/graph/component-class.rb

@@ -0,0 +1,134 @@
+module Babeltrace2
+  BT_COMPONENT_CLASS_TYPE_SOURCE = 1 << 0
+  BT_COMPONENT_CLASS_TYPE_FILTER = 1 << 1
+  BT_COMPONENT_CLASS_TYPE_SINK = 1 << 2
+
+  BTComponentClassType = enum :bt_component_class_type,
+    [ :BT_COMPONENT_CLASS_TYPE_SOURCE,
+       BT_COMPONENT_CLASS_TYPE_SOURCE,
+      :BT_COMPONENT_CLASS_TYPE_FILTER,
+       BT_COMPONENT_CLASS_TYPE_FILTER,
+      :BT_COMPONENT_CLASS_TYPE_SINK,
+       BT_COMPONENT_CLASS_TYPE_SINK ]
+
+  attach_function :bt_component_class_get_type,
+                  [:bt_component_class_handle],
+                  :bt_component_class_type
+  attach_function :bt_component_class_get_name,
+                  [:bt_component_class_handle],
+                  :string
+  attach_function :bt_component_class_get_description,
+                  [:bt_component_class_handle],
+                  :string
+  attach_function :bt_component_class_get_help,
+                  [:bt_component_class_handle],
+                  :string
+  attach_function :bt_component_class_get_ref,
+                  [:bt_component_class_handle],
+                  :void
+  attach_function :bt_component_class_put_ref,
+                  [:bt_component_class_handle],
+                  :void
+
+  class BTComponentClass < BTSharedObject
+    Type = BTComponentClassType
+    @get_ref = :bt_component_class_get_ref
+    @put_ref = :bt_component_class_put_ref
+
+    def self.from_handle(handle, retain: true, auto_release: true)
+      case Babeltrace2.bt_component_class_get_type(handle)
+      when :BT_COMPONENT_CLASS_TYPE_SOURCE
+        handle = BTComponentClassSourceHandle.new(handle)
+        BTComponentClassSource
+      when :BT_COMPONENT_CLASS_TYPE_FILTER
+        handle = BTComponentClassFilterHandle.new(handle)
+        BTComponentClassFilter
+      when :BT_COMPONENT_CLASS_TYPE_SINK
+        handle = BTComponentClassSinkHandle.new(handle)
+        BTComponentClassSink
+      else
+        raise Error.new("Unknown component class type")
+      end.new(handle, retain: retain, auto_release: auto_release)
+    end
+
+    def get_type
+      Babeltrace2.bt_component_class_get_type(@handle)
+    end
+    alias type get_type
+
+    def is_source
+      get_type == :BT_COMPONENT_CLASS_TYPE_SOURCE
+    end
+    alias source? is_source
+
+    def is_filter
+      get_type == :BT_COMPONENT_CLASS_TYPE_FILTER
+    end
+    alias filter? is_filter
+
+    def is_sink
+      get_type == :BT_COMPONENT_CLASS_TYPE_SINK
+    end
+    alias sink? is_sink
+
+    def get_name
+      Babeltrace2.bt_component_class_get_name(@handle)
+    end
+    alias name get_name
+
+    def get_description
+      Babeltrace2.bt_component_class_get_description(@handle)
+    end
+    alias description get_description
+
+    def get_help
+      Babeltrace2.bt_component_class_get_help(@handle)
+    end
+    alias help get_help
+  end
+
+  attach_function :bt_component_class_source_get_ref,
+                  [:bt_component_class_source_handle],
+                  :void
+  attach_function :bt_component_class_source_put_ref,
+                  [:bt_component_class_source_handle],
+                  :void
+
+  class BTComponentClass
+    class Source < BTComponentClass
+      @get_ref = :bt_component_class_source_get_ref
+      @put_ref = :bt_component_class_source_put_ref
+    end
+  end
+  BTComponentClassSource = BTComponentClass::Source
+
+  attach_function :bt_component_class_filter_get_ref,
+                  [:bt_component_class_filter_handle],
+                  :void
+  attach_function :bt_component_class_filter_put_ref,
+                  [:bt_component_class_filter_handle],
+                  :void
+
+  class BTComponentClass
+    class Filter < BTComponentClass
+      @get_ref = :bt_component_class_filter_get_ref
+      @put_ref = :bt_component_class_filter_put_ref
+    end
+  end
+  BTComponentClassFilter = BTComponentClass::Filter
+
+  attach_function :bt_component_class_sink_get_ref,
+                  [:bt_component_class_sink_handle],
+                  :void
+  attach_function :bt_component_class_sink_put_ref,
+                  [:bt_component_class_sink_handle],
+                  :void
+
+  class BTComponentClass
+    class Sink < BTComponentClass
+    @get_ref = :bt_component_class_sink_get_ref
+    @put_ref = :bt_component_class_sink_put_ref
+    end
+  end
+  BTComponentClassSink = BTComponentClass::Sink
+end

data/lib/babeltrace2/graph/component-descriptor-set.rb

@@ -0,0 +1,78 @@
+module Babeltrace2
+
+  attach_function :bt_component_descriptor_set_create,
+                  [],
+                  :bt_component_descriptor_set_handle
+
+  BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_OK = BT_FUNC_STATUS_OK
+  BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_MEMORY_ERROR = BT_FUNC_STATUS_MEMORY_ERROR
+  BTComponentDescriptorSetAddDescriptorStatus =
+    enum :bt_component_descriptor_set_add_descriptor_status,
+    [ :BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_OK,
+       BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_OK,
+      :BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_MEMORY_ERROR,
+       BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_MEMORY_ERROR ]
+
+  attach_function :bt_component_descriptor_set_add_descriptor,
+                  [ :bt_component_descriptor_set_handle,
+                    :bt_component_class_handle,
+                    :bt_value_handle ],
+                  :bt_component_descriptor_set_add_descriptor_status
+
+  attach_function :bt_component_descriptor_set_add_descriptor_with_initialize_method_data,
+                  [ :bt_component_descriptor_set_handle,
+                    :bt_component_class_handle,
+                    :bt_value_handle,
+                    :pointer ],
+                  :bt_component_descriptor_set_add_descriptor_status
+
+  attach_function :bt_component_descriptor_set_get_ref,
+                  [ :bt_component_descriptor_set_handle ],
+                  :void
+
+  attach_function :bt_component_descriptor_set_put_ref,
+                  [ :bt_component_descriptor_set_handle ],
+                  :void
+
+  class BTComponent
+    DescriptorSetAddDescriptorStatus = BTComponentDescriptorSetAddDescriptorStatus
+    class DescriptorSet < BTSharedObject
+      AddDescriptorStatus = BTComponentDescriptorSetAddDescriptorStatus
+      @get_ref = :bt_component_descriptor_set_get_ref
+      @put_ref = :bt_component_descriptor_set_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_component_descriptor_set_create()
+          raise Babeltrace2.process_error if handle.null?
+          super(handle)
+        end
+      end
+
+      def add_descriptor(component_class, params: {}, initialize_method_data: nil)
+        params = BTValue.from_value(params)
+        raise "invalid value" unless params.kind_of?(BTValueMap)
+        res = if initialize_method_data
+            Babeltrace2.bt_component_descriptor_set_add_descriptor_with_initialize_method_data(@handle, component_class, params, initialize_method_data)
+          else
+            Babeltrace2.bt_component_descriptor_set_add_descriptor(@handle, component_class, params)
+          end
+        raise Babeltrace2.process_error(res) if res != :BT_COMPONENT_DESCRIPTOR_SET_ADD_DESCRIPTOR_STATUS_OK
+        self
+      end
+
+      def get_greatest_operative_mip_version(logging_level = BTLogging.default_level)
+        ptr = FFI::MemoryPointer.new(:uint64)
+        res = Babeltrace2.bt_get_greatest_operative_mip_version(
+                @handle, logging_level, ptr)
+        raise Babeltrace2.process_error(res) if res != :BT_GET_GREATEST_OPERATIVE_MIP_VERSION_STATUS_OK
+        ptr.read_uint64
+      end
+      alias greatest_operative_mip_version get_greatest_operative_mip_version
+    end
+  end
+  BTComponentDescriptorSet = BTComponent::DescriptorSet
+
+end

data/lib/babeltrace2/graph/component.rb

@@ -0,0 +1,362 @@
+module Babeltrace2
+
+  attach_function :bt_component_get_class_type,
+                  [ :bt_component_handle ],
+                  :bt_component_class_type
+
+  attach_function :bt_component_borrow_class_const,
+                  [ :bt_component_handle ],
+                  :bt_component_class_handle
+
+  attach_function :bt_component_get_name,
+                  [ :bt_component_handle ],
+                  :string
+
+  attach_function :bt_component_get_logging_level,
+                  [ :bt_component_handle ],
+                  :bt_logging_level
+
+  attach_function :bt_component_get_ref,
+                  [ :bt_component_handle ],
+                  :void
+
+  attach_function :bt_component_put_ref,
+                  [ :bt_component_handle ],
+                  :void
+
+  class BTComponent < BTSharedObject
+    ClassType = BTComponentClassType
+    @get_ref = :bt_component_get_ref
+    @put_ref = :bt_component_put_ref
+
+    def self.from_handle(handle, retain: true, auto_release: true)
+      case Babeltrace2.bt_component_get_class_type(handle)
+      when :BT_COMPONENT_CLASS_TYPE_SOURCE
+        handle = BTComponentSourceHandle.new(handle)
+        BTComponentSource
+      when :BT_COMPONENT_CLASS_TYPE_FILTER
+        handle = BTComponentFilterHandle.new(handle)
+        BTComponentFilter
+      when :BT_COMPONENT_CLASS_TYPE_SINK
+        handle = BTComponentSinkHandle.new(handle)
+        BTComponentSink
+      else
+        raise Error.new("unknown component class type")
+      end.new(handle, retain: retain, auto_release: auto_release)
+    end
+
+    def get_class_type
+      Babeltrace2.bt_component_get_class_type(@handle)
+    end
+    alias class_type get_class_type
+
+    def is_source
+      get_class_type == :BT_COMPONENT_CLASS_TYPE_SOURCE
+    end
+    alias source? is_source
+
+    def is_filter
+      get_class_type == :BT_COMPONENT_CLASS_TYPE_FILTER
+    end
+    alias filter? is_filter
+
+    def is_sink
+      get_class_type == :BT_COMPONENT_CLASS_TYPE_SINK
+    end
+    alias sink? is_sink
+
+    def get_class
+      handle = Babeltrace2.bt_component_borrow_class_const(@handle)
+      BTComponentClass.from_handle(handle)
+    end
+
+    def get_name
+      Babeltrace2.bt_component_get_name(@handle)
+    end
+    alias name get_name
+
+    def get_logging_level
+      Babeltrace2.bt_component_get_logging_level(@handle)
+    end
+    alias logging_level get_logging_level
+  end
+
+  attach_function :bt_component_source_borrow_class_const,
+                  [ :bt_component_source_handle ],
+                  :bt_component_class_source_handle
+
+  attach_function :bt_component_source_get_output_port_count,
+                  [ :bt_component_source_handle ],
+                  :uint64
+
+  attach_function :bt_component_source_borrow_output_port_by_index_const,
+                  [ :bt_component_source_handle,
+                    :uint64 ],
+                  :bt_port_output_handle
+
+  attach_function :bt_component_source_borrow_output_port_by_name_const,
+                  [ :bt_component_source_handle,
+                    :string ],
+                  :bt_port_output_handle
+
+  attach_function :bt_component_source_get_ref,
+                  [ :bt_component_source_handle ],
+                  :void
+
+  attach_function :bt_component_source_put_ref,
+                  [ :bt_component_source_handle ],
+                  :void
+
+  class BTComponent::Source < BTComponent
+    @get_ref = :bt_component_source_get_ref
+    @put_ref = :bt_component_source_put_ref
+    def get_class
+      handle = Babeltrace2.bt_component_source_borrow_class_const(@handle)
+      BTComponentClassSource.new(handle, retain: true)
+    end
+
+    def get_output_port_count
+      Babeltrace2.bt_component_source_get_output_port_count(@handle)
+    end
+    alias output_port_count get_output_port_count
+
+    def get_output_port_by_index(index)
+      count = get_output_port_count
+      index += count if index < 0
+      return nil if index >= count || index < 0
+      handle = Babeltrace2.bt_component_source_borrow_output_port_by_index_const(@handle, index)
+      BTPortOutput.new(handle, retain: true)
+    end
+
+    def get_output_port_by_name(name)
+      handle = Babeltrace2.bt_component_source_borrow_output_port_by_name_const(@handle, name)
+      return nil if handle.null?
+      BTPortOutput.new(handle, retain: true)
+    end
+
+    def get_output_port(port)
+      case port
+      when String
+        get_output_port_by_name(port)
+      when Integer
+        get_output_port_by_index(port)
+      else
+        raise TypeError, "wrong type for port query"
+      end
+    end
+    alias output_port get_output_port
+
+    def output_ports
+      output_port_count.times.collect { |index|
+        handle = Babeltrace2.bt_component_source_borrow_output_port_by_index_const(
+                   @handle, index)
+        BTPortOutput.new(handle, retain: true)
+      }
+    end
+  end
+  BTComponentSource = BTComponent::Source
+
+  attach_function :bt_component_filter_borrow_class_const,
+                  [ :bt_component_filter_handle ],
+                  :bt_component_class_filter_handle
+
+  attach_function :bt_component_filter_get_input_port_count,
+                  [ :bt_component_filter_handle ],
+                  :uint64
+
+  attach_function :bt_component_filter_borrow_input_port_by_index_const,
+                  [ :bt_component_filter_handle,
+                    :uint64 ],
+                  :bt_port_input_handle
+
+  attach_function :bt_component_filter_borrow_input_port_by_name_const,
+                  [ :bt_component_filter_handle,
+                    :string ],
+                  :bt_port_input_handle
+
+  attach_function :bt_component_filter_get_output_port_count,
+                  [ :bt_component_filter_handle ],
+                  :uint64
+
+  attach_function :bt_component_filter_borrow_output_port_by_index_const,
+                  [ :bt_component_filter_handle,
+                    :uint64 ],
+                  :bt_port_output_handle
+
+  attach_function :bt_component_filter_borrow_output_port_by_name_const,
+                  [ :bt_component_filter_handle,
+                    :string ],
+                  :bt_port_output_handle
+
+  attach_function :bt_component_filter_get_ref,
+                  [ :bt_component_filter_handle ],
+                  :void
+
+  attach_function :bt_component_filter_put_ref,
+                  [ :bt_component_filter_handle ],
+                  :void
+
+  class BTComponent::Filter < BTComponent
+    @get_ref = :bt_component_filter_get_ref
+    @put_ref = :bt_component_filter_put_ref
+    def get_class
+      handle = Babeltrace2.bt_component_filter_borrow_class_const(@handle)
+      BTComponentClassFilter.new(handle, retain: true)
+    end
+
+    def get_output_port_count
+      Babeltrace2.bt_component_filter_get_output_port_count(@handle)
+    end
+    alias output_port_count get_output_port_count
+
+    def get_output_port_by_index(index)
+      count = get_output_port_count
+      index += count if index < 0
+      return nil if index >= count || index < 0
+      handle = Babeltrace2.bt_component_filter_borrow_output_port_by_index_const(@handle, index)
+      BTPortOutput.new(handle, retain: true)
+    end
+
+    def get_output_port_by_name(name)
+      handle = Babeltrace2.bt_component_filter_borrow_output_port_by_name_const(@handle, name)
+      return nil if handle.null?
+      BTPortOutput.new(handle, retain: true)
+    end
+
+    def get_output_port(port)
+      case port
+      when String
+        get_output_port_by_name(port)
+      when Integer
+        get_output_port_by_index(port)
+      else
+        raise TypeError, "wrong type for port query"
+      end
+    end
+
+    def output_ports
+      output_port_count.times.collect { |index|
+        handle = Babeltrace2.bt_component_filter_borrow_output_port_by_index_const(
+                   @handle, index)
+        BTPortOutput.new(handle, retain: true)
+      }
+    end
+    alias output_port get_output_port
+
+    def get_input_port_count
+      Babeltrace2.bt_component_filter_get_input_port_count(@handle)
+    end
+    alias input_port_count get_input_port_count
+
+    def get_input_port_by_index(index)
+      count = get_input_port_count
+      index += count if index < 0
+      return nil if index >= count || index < 0
+      handle = Babeltrace2.bt_component_filter_borrow_input_port_by_index_const(@handle, index)
+      BTPortInput.new(handle, retain: true)
+    end
+
+    def get_input_port_by_name(name)
+      handle = Babeltrace2.bt_component_filter_borrow_input_port_by_name_const(@handle, name)
+      return nil if handle.null?
+      BTPortInput.new(handle, retain: true)
+    end
+
+    def get_input_port(port)
+      case port
+      when String
+        get_input_port_by_name(port)
+      when Integer
+        get_input_port_by_index(port)
+      else
+        raise TypeError, "wrong type for port query"
+      end
+    end
+    alias input_port get_input_port
+
+    def input_ports
+      input_port_count.times.collect { |index|
+        handle = Babeltrace2.bt_component_filter_borrow_input_port_by_index_const(
+                   @handle, index)
+        BTPortInput.new(handle, retain: true)
+      }
+    end
+  end
+  BTComponentFilter = BTComponent::Filter
+
+  attach_function :bt_component_sink_borrow_class_const,
+                  [ :bt_component_sink_handle ],
+                  :bt_component_class_sink_handle
+
+  attach_function :bt_component_sink_get_input_port_count,
+                  [ :bt_component_sink_handle ],
+                  :uint64
+
+  attach_function :bt_component_sink_borrow_input_port_by_index_const,
+                  [ :bt_component_sink_handle,
+                    :uint64 ],
+                  :bt_port_input_handle
+
+  attach_function :bt_component_sink_borrow_input_port_by_name_const,
+                  [ :bt_component_sink_handle,
+                    :string ],
+                  :bt_port_input_handle
+
+  attach_function :bt_component_sink_get_ref,
+                  [ :bt_component_sink_handle ],
+                  :void
+
+  attach_function :bt_component_sink_put_ref,
+                  [ :bt_component_sink_handle ],
+                  :void
+
+  class BTComponent::Sink < BTComponent
+    @get_ref = :bt_component_sink_get_ref
+    @put_ref = :bt_component_sink_put_ref
+    def get_class
+      handle = Babeltrace2.bt_component_sink_borrow_class_const(@handle)
+      BTComponentClassSink.new(handle, retain: true)
+    end
+
+    def get_input_port_count
+      Babeltrace2.bt_component_sink_get_input_port_count(@handle)
+    end
+    alias input_port_count get_input_port_count
+
+    def get_input_port_by_index(index)
+      count = get_input_port_count
+      index += count if index < 0
+      return nil if index >= count || index < 0
+      handle = Babeltrace2.bt_component_sink_borrow_input_port_by_index_const(@handle, index)
+      BTPortInput.new(handle, retain: true)
+    end
+
+    def get_input_port_by_name(name)
+      handle = Babeltrace2.bt_component_sink_borrow_input_port_by_name_const(@handle, name)
+      return nil if handle.null?
+      BTPortInput.new(handle, retain: true)
+    end
+
+    def get_input_port(port)
+      case port
+      when String
+        get_input_port_by_name(port)
+      when Integer
+        get_input_port_by_index(port)
+      else
+        raise TypeError, "wrong type for port query"
+      end
+    end
+    alias input_port get_input_port
+
+    def input_ports
+      input_port_count.times.collect { |index|
+        handle = Babeltrace2.bt_component_sink_borrow_input_port_by_index_const(
+                   @handle, index)
+        BTPortInput.new(handle, retain: true)
+      }
+    end
+  end
+  BTComponentSink = BTComponent::Sink
+
+end