rbx-trepanning 0.0.1-universal-rubinius

data/app/brkptmgr.rb

@@ -0,0 +1,138 @@
+# Copyright (C) 2010 Rocky Bernstein <rockyb@rubyforge.net>
+require 'set'
+require 'rubygems'; require 'require_relative'
+require_relative 'breakpoint'
+class BreakpointMgr
+
+  attr_reader :list
+  attr_reader :set
+
+  def initialize
+    @list = []
+    @set = Set.new
+  end
+
+  def <<(brkpt)
+    @list << brkpt
+    @set.add(set_key(brkpt))
+  end
+
+  def [](index)
+    raise TypeError, 
+    "index #{index} should be a Fixnum, is #{index.class}" unless
+      index.is_a?(Fixnum)
+    @list.detect {|bp| bp.id == index }
+  end
+
+  alias detect []
+
+  def delete(index)
+    bp = detect(index)
+    if bp
+      delete_by_brkpt(bp)
+      return bp
+    else
+      return nil
+    end
+  end
+
+  def delete_by_brkpt(delete_bp)
+    @list = @list.reject{|candidate| candidate == delete_bp}
+    @set  = Set.new(@list.map{|bp| set_key(bp)})
+    delete_bp.remove! unless @set.member?(set_key(delete_bp))
+    return delete_bp
+  end
+
+  def add(*args)
+    brkpt = Trepanning::Breakpoint.new(*args)
+    @list << brkpt
+    @set.add(set_key(brkpt))
+    return brkpt
+  end
+
+  def empty?
+    @list.empty?
+  end
+
+  # def line_breaks(container)
+  #   result = {}
+  #   @list.each do |bp|
+  #     if bp.source_container == container
+  #       bp.source_location.each do |line|
+  #         result[line] = bp 
+  #       end
+  #     end
+  #   end
+  #   result
+  # end
+
+  def find(meth, ip)
+    @list.detect do |bp| 
+      if bp.enabled? && bp.ip == ip
+        begin
+          return bp ## if bp.condition?(bind)
+        rescue
+        end 
+      end
+    end
+  end
+
+  def max
+    @list.map{|bp| bp.id}.max || 0
+  end
+
+  # Key used in @set to list unique instruction-sequence offsets.
+  def set_key(bp)
+    [bp.method, bp.ip]
+  end
+
+  def size
+    @list.size
+  end
+
+  def reset
+    @list.each{|bp| bp.remove!}
+    @list = []
+    @set  = Set.new
+  end
+
+end
+if __FILE__ == $0
+  def bp_status(brkpts, i)
+    puts "list size: #{brkpts.list.size}"
+    puts "set size: #{brkpts.set.size}"
+    puts "max: #{brkpts.max}"
+    p brkpts
+    puts "--- #{i} ---"
+  end
+
+  meth = Rubinius::CompiledMethod.of_sender
+
+  brkpts = BreakpointMgr.new
+  brkpts.add("<start>", meth, 0, 0, 1)
+  p brkpts[2]
+  bp_status(brkpts, 1)
+  # offset = frame.pc_offset
+  b2 = Trepanning::Breakpoint.new("<2nd one>", meth, 0, 5, 2)
+  brkpts << b2
+  p brkpts.find(b2.meth, b2.ip)
+  p brkpts[2]
+  puts '--- 2 ---'
+  ## p brkpts.line_breaks(iseq.source_container)
+  p brkpts.delete(2)
+  p brkpts[2]
+  bp_status(brkpts, 3)
+
+  # Two of the same breakpoints but delete 1 and see that the
+  # other still stays
+  b2 = Trepanning::Breakpoint.new("<dup brkpt>", meth, 0, 0, 0)
+  brkpts << b2
+  bp_status(brkpts, 4)
+  b3 = Trepanning::Breakpoint.new("<dup brkpt>", meth, 0, 0, 0)
+  brkpts << b3
+  bp_status(brkpts, 5)
+  brkpts.delete_by_brkpt(b2)
+  bp_status(brkpts, 6)
+  brkpts.delete_by_brkpt(b3)
+  bp_status(brkpts, 7)
+end