@@ -120,7 +120,7 @@ def IRB.init_config(ap_path)
120
120
puts 'processing time: %fs' % ( now - time ) if IRB . conf [ :MEASURE ]
121
121
result
122
122
}
123
- @CONF [ :MEASURE_PROC ] [ :STACKPROF ] = proc { |context , code , line_no , &block |
123
+ @CONF [ :MEASURE_PROC ] [ :STACKPROF ] = proc { |context , code , line_no , arg , &block |
124
124
success = false
125
125
begin
126
126
require 'stackprof'
@@ -130,7 +130,7 @@ def IRB.init_config(ap_path)
130
130
end
131
131
if success
132
132
result = nil
133
- stackprof_result = StackProf . run ( mode : :cpu ) do
133
+ stackprof_result = StackProf . run ( mode : arg ? arg : :cpu ) do
134
134
result = block . ( )
135
135
end
136
136
StackProf ::Report . new ( stackprof_result ) . print_text if IRB . conf [ :MEASURE ]
@@ -146,17 +146,17 @@ def IRB.init_config(ap_path)
146
146
@CONF [ :AT_EXIT ] = [ ]
147
147
end
148
148
149
- def IRB . set_measure_callback ( type = nil )
149
+ def IRB . set_measure_callback ( type = nil , arg = nil )
150
150
added = nil
151
151
if type
152
152
type_sym = type . upcase . to_sym
153
153
if IRB . conf [ :MEASURE_PROC ] [ type_sym ]
154
- added = [ type_sym , IRB . conf [ :MEASURE_PROC ] [ type_sym ] ]
154
+ added = [ type_sym , IRB . conf [ :MEASURE_PROC ] [ type_sym ] , arg ]
155
155
end
156
156
elsif IRB . conf [ :MEASURE_PROC ] [ :CUSTOM ]
157
- added = [ :CUSTOM , IRB . conf [ :MEASURE_PROC ] [ :CUSTOM ] ]
157
+ added = [ :CUSTOM , IRB . conf [ :MEASURE_PROC ] [ :CUSTOM ] , arg ]
158
158
else
159
- added = [ :TIME , IRB . conf [ :MEASURE_PROC ] [ :TIME ] ]
159
+ added = [ :TIME , IRB . conf [ :MEASURE_PROC ] [ :TIME ] , arg ]
160
160
end
161
161
IRB . conf [ :MEASURE_CALLBACKS ] << added if added
162
162
added
@@ -167,7 +167,7 @@ def IRB.unset_measure_callback(type = nil)
167
167
IRB . conf [ :MEASURE_CALLBACKS ] . clear
168
168
else
169
169
type_sym = type . upcase . to_sym
170
- IRB . conf [ :MEASURE_CALLBACKS ] . reject! { |t , c | t == type_sym }
170
+ IRB . conf [ :MEASURE_CALLBACKS ] . reject! { |t , | t == type_sym }
171
171
end
172
172
end
173
173
0 commit comments