babeltrace2 0.1.0

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

@@ -0,0 +1,147 @@
+module Babeltrace2
+
+  attach_function :bt_self_message_iterator_borrow_component,
+                  [ :bt_self_message_iterator_handle ],
+                  :bt_self_component_handle
+
+  attach_function :bt_self_message_iterator_borrow_port,
+                  [ :bt_self_message_iterator_handle ],
+                  :bt_self_component_port_output_handle
+
+  attach_function :bt_self_message_iterator_set_data,
+                  [ :bt_self_message_iterator_handle, :pointer ],
+                  :void
+
+  attach_function :bt_self_message_iterator_get_data,
+                  [ :bt_self_message_iterator_handle ],
+                  :pointer
+
+  attach_function :bt_self_message_iterator_is_interrupted,
+                  [ :bt_self_message_iterator_handle ],
+                  :bt_bool
+
+  class BTSelfMessageIterator < BTMessageIterator
+    def get_component
+      handle = Babeltrace2.bt_self_message_iterator_borrow_component(@handle)
+      BTSelfComponent.from_handle(handle)
+    end
+    alias component get_component
+
+    def get_port
+      handle = Babeltrace2.bt_self_message_iterator_borrow_port(@handle)
+      BTSelfComponentPortOutput.new(handle, retain: true)
+    end
+    alias port get_port
+
+    def set_data(user_data)
+      Babeltrace2.bt_self_message_iterator_set_data(@handle, user_data)
+      self
+    end
+
+    def data=(user_data)
+      Babeltrace2.bt_self_message_iterator_set_data(@handle, user_data)
+      user_data
+    end
+
+    def get_data
+      Babeltrace2.bt_self_message_iterator_get_data(@handle)
+    end
+    alias data get_data
+
+    def is_interrupted
+      Babeltrace2.bt_self_message_iterator_is_interrupted(@handle) != BT_FALSE
+    end
+    alias interrupted? is_interrupted
+
+    def create_message_stream_beginning(stream)
+      BTMessageStreamBeginning.new(self_message_iterator: @handle, stream: stream)
+    end
+    alias create_stream_beginning_message create_message_stream_beginning
+    alias create_stream_beginning create_message_stream_beginning
+
+    def create_message_stream_end(stream)
+      BTMessageStreamEnd.new(self_message_iterator: @handle, stream: stream)
+    end
+    alias create_stream_end_message create_message_stream_end
+    alias create_stream_end create_message_stream_end
+
+    def create_message_event(event_class, stream_or_packet, clock_snapshot_value: nil)
+      case stream_or_packet
+      when BTStream
+        stream = stream_or_packet
+        packet = nil
+      when BTPacket
+        stream = nil
+        packet = stream_or_packet
+      else
+        raise "invalid stream or packet"
+      end
+      BTMessageEvent.new(self_message_iterator: @handle, event_class: event_class,
+                         stream: stream, clock_snapshot_value: clock_snapshot_value,
+                         packet: packet)
+    end
+    alias create_event_message create_message_event
+    alias create_event create_message_event
+
+    def create_message_packet_beginning(packet, clock_snapshot_value: nil)
+      BTMessagePacketBeginning.new(self_message_iterator: @handle, packet: packet,
+                                   clock_snapshot_value: clock_snapshot_value)
+    end
+    alias create_packet_beginning_message create_message_packet_beginning
+    alias create_packet_beginning create_message_packet_beginning
+
+    def create_message_packet_end(packet, clock_snapshot_value: nil)
+      BTMessagePacketEnd.new(self_message_iterator: @handle, packet: packet,
+                             clock_snapshot_value: clock_snapshot_value)
+    end
+    alias create_packet_end_message create_message_packet_end
+    alias create_packet_end create_message_packet_end
+
+    def create_message_discarded_events(stream, beginning_clock_snapshot_value: nil,
+                                                end_clock_snapshot_value: nil)
+      BTMessageDiscardedEvents.new(self_message_iterator: @handle, stream: stream,
+        beginning_clock_snapshot_value: beginning_clock_snapshot_value,
+        end_clock_snapshot_value: end_clock_snapshot_value)
+    end
+    alias create_discarded_events_message create_message_discarded_events
+    alias create_discarded_events create_message_discarded_events
+
+    def create_message_discarded_packets(stream, beginning_clock_snapshot_value: nil,
+                                                 end_clock_snapshot_value: nil)
+      BTMessageDiscardedPackets.new(self_message_iterator: @handle, stream: stream,
+        beginning_clock_snapshot_value: beginning_clock_snapshot_value,
+        end_clock_snapshot_value: end_clock_snapshot_value)
+    end
+    alias create_discarded_packets_message create_message_discarded_packets
+    alias create_discarded_packets create_message_discarded_packets
+
+    def create_message_message_iterator_inactivity(clock_class, clock_snapshot_value)
+      BTMessageMessageIteratorInactivity.new(self_message_iterator: @handle,
+                                             clock_class: clock_class,
+                                             clock_snapshot_value: clock_snapshot_value)
+    end
+    alias create_message_iterator_inactivity_message create_message_message_iterator_inactivity
+    alias create_message_iterator_inactivity create_message_message_iterator_inactivity
+  end
+
+  attach_function :bt_self_message_iterator_configuration_set_can_seek_forward,
+                  [ :bt_self_message_iterator_configuration_handle, :bt_bool ],
+                  :void
+
+  class BTSelfMessageIterator
+    class Configuration < BTObject
+      def set_can_seek_forward(can_seek_forward)
+        Babeltrace2.bt_self_message_iterator_configuration_set_can_seek_forward(
+          @handle, can_seek_forward ? BT_TRUE : BT_FALSE)
+        self
+      end
+
+      def can_seek_forward=(can_seek_forward)
+        Babeltrace2.bt_self_message_iterator_configuration_set_can_seek_forward(
+          @handle, can_seek_forward ? BT_TRUE : BT_FALSE)
+        can_seek_forward
+      end
+    end
+  end
+  BTSelfMessageIteratorConfiguration = BTSelfMessageIterator::Configuration
+end

data/lib/babeltrace2/integer-range-set.rb

@@ -0,0 +1,275 @@
+module Babeltrace2
+
+  attach_function :bt_integer_range_unsigned_get_lower,
+                  [:bt_integer_range_unsigned_handle],
+                  :uint64
+
+  attach_function :bt_integer_range_unsigned_get_upper,
+                  [:bt_integer_range_unsigned_handle],
+                  :uint64
+
+  attach_function :bt_integer_range_unsigned_is_equal,
+                  [:bt_integer_range_unsigned_handle, :bt_integer_range_unsigned_handle],
+                  :bt_bool
+
+  class BTIntegerRange < BTObject
+    def value
+      lower..upper
+    end
+
+    def include?(val)
+      lower <= val && upper >= val
+    end
+  end
+
+  class BTIntegerRangeUnsigned < BTIntegerRange
+    def get_lower
+      Babeltrace2.bt_integer_range_unsigned_get_lower(@handle)
+    end
+    alias lower get_lower
+    alias min get_lower
+
+    def get_upper
+      Babeltrace2.bt_integer_range_unsigned_get_upper(@handle)
+    end
+    alias upper get_upper
+    alias max get_upper
+
+    def is_equal(other)
+      Babeltrace2.bt_integer_range_unsigned_is_equal(other) != BT_FALSE
+    end
+    alias == is_equal
+  end
+
+  attach_function :bt_integer_range_signed_get_lower,
+                  [:bt_integer_range_signed_handle],
+                  :int64
+
+  attach_function :bt_integer_range_signed_get_upper,
+                  [:bt_integer_range_signed_handle],
+                  :int64
+
+  attach_function :bt_integer_range_signed_is_equal,
+                  [:bt_integer_range_signed_handle, :bt_integer_range_signed_handle],
+                  :bt_bool
+
+  class BTIntegerRangeSigned < BTIntegerRange
+    def get_lower
+      Babeltrace2.bt_integer_range_signed_get_lower(@handle)
+    end
+    alias lower get_lower
+
+    def get_upper
+      Babeltrace2.bt_integer_range_signed_get_upper(@handle)
+    end
+    alias upper get_upper
+
+    def is_equal(other)
+      Babeltrace2.bt_integer_range_signed_is_equal(other) != BT_FALSE
+    end
+    alias == is_equal
+  end
+
+  BTIntegerRangeSetAddRangeStatus = enum :bt_integer_range_set_add_range_status,
+    [ :BT_INTEGER_RANGE_SET_ADD_RANGE_STATUS_OK, BT_FUNC_STATUS_OK,
+      :BT_INTEGER_RANGE_SET_ADD_RANGE_STATUS_MEMORY_ERROR, BT_FUNC_STATUS_MEMORY_ERROR]
+
+  attach_function :bt_integer_range_set_get_range_count,
+                  [:bt_integer_range_set_handle],
+                  :uint64
+
+  class BTIntegerRangeSet < BTSharedObject
+    AddRangeStatus = BTIntegerRangeSetAddRangeStatus
+
+    def self.from_value(value)
+      case value
+      when self
+        return value
+      when Array
+        range = self.new
+        range.push(*value)
+      else
+        range = self.new
+        range.add_range(value)
+      end
+    end
+
+    def get_range_count
+      return Babeltrace2.bt_integer_range_set_get_range_count(@handle)
+    end
+    alias range_count get_range_count
+    alias size get_range_count
+
+    def get_ranges
+      range_count.times.collect { |i|
+        get_range(i)
+      }
+    end
+    alias ranges get_ranges
+    alias to_a get_ranges
+
+    def push(*args)
+      args.each { |arg| add_range(arg) }
+      self
+    end
+
+    def each
+      if block_given?
+        range_count.times.each { |i|
+          yield get_range(i)
+        }
+      else
+        to_enum(:each)
+      end
+    end
+
+    def value
+      each.collect(&:value).to_a
+    end
+
+    private
+    def range_arg_get(lower, upper)
+      if not upper
+        case lower
+        when BTIntegerRange, Range, Array
+          upper = lower.max
+          lower = lower.min
+        else
+          upper = lower
+        end
+      end
+      [lower, upper]
+    end
+  end
+
+  attach_function :bt_integer_range_set_unsigned_create,
+                  [],
+                  :bt_integer_range_set_unsigned_handle
+
+  attach_function :bt_integer_range_set_unsigned_add_range,
+                  [:bt_integer_range_set_unsigned_handle, :uint64, :uint64],
+                  :bt_integer_range_set_add_range_status
+
+  attach_function :bt_integer_range_set_unsigned_borrow_range_by_index_const,
+                  [:bt_integer_range_set_unsigned_handle, :uint64],
+                  :bt_integer_range_unsigned_handle
+
+  attach_function :bt_integer_range_set_unsigned_is_equal,
+                  [:bt_integer_range_set_unsigned_handle, :bt_integer_range_set_unsigned_handle],
+                  :bt_bool
+
+  attach_function :bt_integer_range_set_unsigned_get_ref,
+                  [:bt_integer_range_set_unsigned_handle],
+                  :void
+
+  attach_function :bt_integer_range_set_unsigned_put_ref,
+                  [:bt_integer_range_set_unsigned_handle],
+                  :void
+
+  class BTIntegerRangeSet
+    class Unsigned < BTIntegerRangeSet
+      @get_ref = :bt_integer_range_set_unsigned_get_ref
+      @put_ref = :bt_integer_range_set_unsigned_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_integer_range_set_unsigned_create()
+          raise Babeltrace2.process_error if handle.null?
+          super(handle)
+        end
+      end
+
+      def add_range(lower, upper = nil)
+        res = Babeltrace2.bt_integer_range_set_unsigned_add_range(
+                @handle, *range_arg_get(lower, upper))
+        raise Babeltrace2.process_error(res) if res != :BT_INTEGER_RANGE_SET_ADD_RANGE_STATUS_OK
+        self
+      end
+      alias add add_range
+
+      def get_range(index)
+        count = range_count
+        index += count if count < 0
+        return nil if index >= count || index < 0
+        BTIntegerRangeUnsigned.new(
+          Babeltrace2.bt_integer_range_set_unsigned_borrow_range_by_index_const(
+            @handle, index))
+      end
+      alias [] get_range
+
+      def is_equal(other)
+        Babeltrace2.bt_integer_range_set_unsigned_is_equal(@handle, other) != BT_FALSE
+      end
+      alias == is_equal
+    end
+  end
+  BTIntegerRangeSetUnsigned = BTIntegerRangeSet::Unsigned
+
+  attach_function :bt_integer_range_set_signed_create,
+                  [],
+                  :bt_integer_range_set_signed_handle
+
+  attach_function :bt_integer_range_set_signed_add_range,
+                  [:bt_integer_range_set_signed_handle, :int64, :int64],
+                  :bt_integer_range_set_add_range_status
+
+  attach_function :bt_integer_range_set_signed_borrow_range_by_index_const,
+                  [:bt_integer_range_set_signed_handle, :uint64],
+                  :bt_integer_range_signed_handle
+
+  attach_function :bt_integer_range_set_signed_is_equal,
+                  [:bt_integer_range_set_signed_handle, :bt_integer_range_set_signed_handle],
+                  :bt_bool
+
+  attach_function :bt_integer_range_set_signed_get_ref,
+                  [:bt_integer_range_set_signed_handle],
+                  :void
+
+  attach_function :bt_integer_range_set_signed_put_ref,
+                  [:bt_integer_range_set_signed_handle],
+                  :void
+
+  class BTIntegerRangeSet
+    class Signed < BTIntegerRangeSet
+      @get_ref = :bt_integer_range_set_signed_get_ref
+      @put_ref = :bt_integer_range_set_signed_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_integer_range_set_signed_create()
+          raise Babeltrace2.process_error if handle.null?
+          super(handle)
+        end
+      end
+
+      def add_range(lower, upper = nil)
+        res = Babeltrace2.bt_integer_range_set_signed_add_range(
+                @handle, *range_arg_get(lower, upper))
+        raise Babeltrace2.process_error(res) if res != :BT_INTEGER_RANGE_SET_ADD_RANGE_STATUS_OK
+        self
+      end
+      alias add add_range
+
+      def get_range(index)
+        count = range_count
+        index += count if count < 0
+        return nil if index >= count || index < 0
+        BTIntegerRangeSigned.new(
+          Babeltrace2.bt_integer_range_set_signed_borrow_range_by_index_const(
+            @handle, index))
+      end
+      alias [] get_range
+
+      def is_equal(other)
+        Babeltrace2.bt_integer_range_set_signed_is_equal(@handle, other) != BT_FALSE
+      end
+      alias == is_equal
+    end
+  end
+  BTIntegerRangeSetSigned = BTIntegerRangeSet::Signed
+
+end

data/lib/babeltrace2/logging-defs.rb

@@ -0,0 +1,19 @@
+module Babeltrace2
+
+  BT_LOGGING_LEVEL_TRACE = 1
+  private_constant :BT_LOGGING_LEVEL_TRACE
+  BT_LOGGING_LEVEL_DEBUG = 2
+  private_constant :BT_LOGGING_LEVEL_DEBUG
+  BT_LOGGING_LEVEL_INFO = 3
+  private_constant :BT_LOGGING_LEVEL_INFO
+  BT_LOGGING_LEVEL_WARNING = 4
+  private_constant :BT_LOGGING_LEVEL_WARNING
+  BT_LOGGING_LEVEL_ERROR = 5
+  private_constant :BT_LOGGING_LEVEL_ERROR
+  BT_LOGGING_LEVEL_FATAL = 6
+  private_constant :BT_LOGGING_LEVEL_FATAL
+  BT_LOGGING_LEVEL_NONE = 0xff
+  private_constant :BT_LOGGING_LEVEL_NONE
+  
+end
+

data/lib/babeltrace2/logging.rb

@@ -0,0 +1,77 @@
+module Babeltrace2
+  BTLoggingLevel = enum :bt_logging_level,
+    [ :BT_LOGGING_LEVEL_TRACE, BT_LOGGING_LEVEL_TRACE,
+      :BT_LOGGING_LEVEL_DEBUG, BT_LOGGING_LEVEL_DEBUG,
+      :BT_LOGGING_LEVEL_INFO, BT_LOGGING_LEVEL_INFO,
+      :BT_LOGGING_LEVEL_WARNING, BT_LOGGING_LEVEL_WARNING,
+      :BT_LOGGING_LEVEL_ERROR, BT_LOGGING_LEVEL_ERROR,
+      :BT_LOGGING_LEVEL_FATAL, BT_LOGGING_LEVEL_FATAL,
+      :BT_LOGGING_LEVEL_NONE, BT_LOGGING_LEVEL_NONE ]
+
+  attach_function :bt_logging_set_global_level,
+                  [:bt_logging_level],
+                  :void
+
+  attach_function :bt_logging_get_global_level,
+                  [],
+                  :bt_logging_level
+
+  attach_function :bt_logging_get_minimal_level,
+                  [],
+                  :bt_logging_level
+
+  module BTLogging
+    class << self
+      def set_global_level(logging_level)
+        Babeltrace2.bt_logging_set_global_level(logging_level)
+        self
+      end
+
+      def global_level=(logging_level)
+        set_global_level(logging_level)
+        logging_level
+      end
+
+      def get_global_level
+        Babeltrace2.bt_logging_get_global_level
+      end
+      alias global_level get_global_level
+
+      def get_minimal_level
+        Babeltrace2.bt_logging_get_minimal_level
+      end
+      alias minimal_level get_minimal_level
+
+      def get_default_level
+        if ENV["BABELTRACE_RUBY_BT2_LOG_LEVEL"]
+          get_level_from_string(ENV["BABELTRACE_RUBY_BT2_LOG_LEVEL"])
+        else
+          get_global_level
+        end
+      end
+      alias default_level get_default_level
+
+      private
+      def get_level_from_string(str)
+        case str
+        when "TRACE", "T"
+          :BT_LOGGING_LEVEL_TRACE
+        when "DEBUG", "D"
+          :BT_LOGGING_LEVEL_DEBUG
+        when "INFO", "I"
+          :BT_LOGGING_LEVEL_INFO
+        when "WARN", "WARNING", "W"
+          :BT_LOGGING_LEVEL_WARNING
+        when "ERROR", "E"
+          :BT_LOGGING_LEVEL_ERROR
+        when "FATAL", "F"
+          :BT_LOGGING_LEVEL_FATAL
+        when "NONE", "N"
+          :BT_LOGGING_LEVEL_NONE
+        else
+          -1
+        end
+      end
+    end
+  end
+end